mirror of
https://github.com/CopilotKit/CopilotKit.git
synced 2026-09-14 16:26:20 +08:00
Jordan Ritter
2482317ccc
320 files reformatted. One-time alignment to match the ruff format check added to CI in #4812.
50 lines
916 B
Python
50 lines
916 B
Python
from typing import Literal, TypedDict, List, Optional
|
|
from langgraph.graph import MessagesState
|
|
|
|
|
|
class Place(TypedDict):
|
|
"""A place."""
|
|
|
|
id: str
|
|
name: str
|
|
address: str
|
|
latitude: float
|
|
longitude: float
|
|
rating: float
|
|
description: Optional[str]
|
|
|
|
|
|
class Trip(TypedDict):
|
|
"""A trip."""
|
|
|
|
id: str
|
|
name: str
|
|
center_latitude: float
|
|
center_longitude: float
|
|
zoom: int # 13 for city, 15 for airport
|
|
places: List[Place]
|
|
|
|
|
|
class SearchProgress(TypedDict):
|
|
"""The progress of a search."""
|
|
|
|
query: str
|
|
results: list[str]
|
|
done: bool
|
|
|
|
|
|
class PlanningProgress(TypedDict):
|
|
"""The progress of a planning."""
|
|
|
|
trip: Trip
|
|
done: bool
|
|
|
|
|
|
class AgentState(MessagesState):
|
|
"""The state of the agent."""
|
|
|
|
selected_trip_id: Optional[str]
|
|
trips: List[Trip]
|
|
search_progress: List[SearchProgress]
|
|
planning_progress: List[PlanningProgress]
|