mirror of
https://github.com/dataforcanada/d4c-service-geo-assistant.git
synced 2026-06-13 14:31:01 +02:00
Daniel Wiesmann
11 lines
345 B
Python
11 lines
345 B
Python
from langchain.agents import AgentState as BaseAgentState
|
|
from geojson_pydantic import FeatureCollection
|
|
from typing import Optional
|
|
from pydantic import Field
|
|
|
|
|
|
class AgentState(BaseAgentState):
|
|
feature_collection: Optional[FeatureCollection] = Field(
|
|
default=None, description="FeatureCollection to be used for the analysis"
|
|
)
|