Update models.py

models.py

@@ -1,37 +1,57 @@
 from typing import List, Optional
-from pydantic import BaseModel
-
-class Location(BaseModel):
-  latitude: float
-  longitude: float
-  place_name: str
-  historical_significance: str
-
-class ArtPeriod(BaseModel):
-  name: str
-  start_year: int
-  end_year: int
-  key_characteristics: List[str]
-
-class Artist(BaseModel):
-  name: str
-  birth_year: Optional[int]
-  death_year: Optional[int]
-  nationality: str
-  primary_medium: List[str]
-
-class Artwork(BaseModel):
-  title: str
-  year: Optional[int]
-  medium: str
-  current_location: Optional[Location]
-  significance: str
-
-class ArtHistoryResponse(BaseModel):
-  time_period: ArtPeriod
-  locations: List[Location]
-  key_artists: List[Artist]
-  notable_works: List[Artwork]
-  cultural_context: str
-  historical_events: List[str]
-  influence_on_later_periods: str
+from pydantic import BaseModel, Field, conint, confloat
+
+class PotentialValue(BaseModel):
+  value: str = Field(..., description="The specific value within the axis")
+  relevance_score: confloat(ge=0, le=100) = Field(..., description="Relevance score from 0-100")
+  contextual_rationale: str = Field(..., description="Explanation of the value's relevance")
+
+class InnovativeValue(BaseModel):
+  value: str = Field(..., description="The innovative value within the emergent axis")
+  innovation_score: confloat(ge=0, le=100) = Field(..., description="Innovation score from 0-100")
+  discovery_potential: str = Field(..., description="Description of potential discoveries")
+
+class StandardAxis(BaseModel):
+  name: str = Field(..., description="Name of the standard axis")
+  current_values: List[str] = Field(default_factory=list, description="Currently selected values")
+  potential_values: List[PotentialValue] = Field(..., description="Potential values for exploration")
+  axis_constraints: List[str] = Field(..., description="Constraints for this axis")
+
+class EmergentAxis(BaseModel):
+  name: str = Field(..., description="Name of the emergent axis")
+  parent_axis: str = Field(..., description="Parent axis from which this emerged")
+  innovative_values: List[InnovativeValue] = Field(..., description="Innovative values for exploration")
+
+class ZoomTrajectory(BaseModel):
+  target_axis: str = Field(..., description="Target axis for zooming")
+  zoom_value: str = Field(..., description="Specific value to zoom into")
+  unlocked_dimensions: List[str] = Field(..., description="New dimensions unlocked by zooming")
+  depth_increment: conint(ge=1, le=3) = Field(..., description="Depth increase (1-3)")
+
+class DezoomPathway(BaseModel):
+  removal_tuple: dict = Field(..., description="Axis-value pair to remove")
+  contextual_expansion: str = Field(..., description="Description of broader context")
+  new_possibility_vectors: List[str] = Field(..., description="New possibilities unlocked")
+
+class ExplorationSummary(BaseModel):
+  current_context: str = Field(..., description="Current exploration context")
+  complexity_level: conint(ge=0, le=10) = Field(..., description="Complexity level (0-10)")
+
+class KnowledgeAxes(BaseModel):
+  standard_axes: List[StandardAxis] = Field(..., description="Standard exploration axes")
+  emergent_axes: List[EmergentAxis] = Field(default_factory=list, description="Emergent exploration axes")
+
+class NavigationStrategies(BaseModel):
+  zoom_trajectories: List[ZoomTrajectory] = Field(..., description="Paths for deeper exploration")
+  dezoom_pathways: List[DezoomPathway] = Field(..., description="Paths for broader exploration")
+
+class MetaInsights(BaseModel):
+  exploration_efficiency: confloat(ge=0, le=100) = Field(..., description="Overall efficiency score")
+  knowledge_gap_indicators: List[str] = Field(..., description="Identified knowledge gaps")
+  recommended_next_steps: List[str] = Field(..., description="Recommended next steps")
+
+class ExplorationResponse(BaseModel):
+  exploration_summary: ExplorationSummary = Field(..., description="Summary of exploration state")
+  knowledge_axes: KnowledgeAxes = Field(..., description="Exploration axes")
+  navigation_strategies: NavigationStrategies = Field(..., description="Navigation options")
+  meta_insights: MetaInsights = Field(..., description="Meta-level insights")