From 01e6e9a2bacf665846ba4e9805b0c33600b6af0d Mon Sep 17 00:00:00 2001 From: Evan Lohn Date: Thu, 30 Jan 2025 10:05:05 -0800 Subject: [PATCH] fixed errors on import --- backend/onyx/agents/agent_search/models.py | 9 +++++++++ 1 file changed, 9 insertions(+) diff --git a/backend/onyx/agents/agent_search/models.py b/backend/onyx/agents/agent_search/models.py index e06211cd1..5dc67d329 100644 --- a/backend/onyx/agents/agent_search/models.py +++ b/backend/onyx/agents/agent_search/models.py @@ -97,6 +97,9 @@ class GraphInputs(BaseModel): files: list[InMemoryChatFile] | None = None structured_response_format: dict | None = None + class Config: + arbitrary_types_allowed = True + class GraphTooling(BaseModel): """Tools and LLMs available to the graph""" @@ -108,6 +111,9 @@ class GraphTooling(BaseModel): force_use_tool: ForceUseTool using_tool_calling_llm: bool = False + class Config: + arbitrary_types_allowed = True + class GraphPersistence(BaseModel): """Configuration for data persistence""" @@ -117,6 +123,9 @@ class GraphPersistence(BaseModel): use_agentic_persistence: bool = True db_session: Session | None = None + class Config: + arbitrary_types_allowed = True + @model_validator(mode="after") def validate_db_session(self) -> "GraphPersistence": if self.use_agentic_persistence and self.db_session is None: