diff --git a/backend/agent/api.py b/backend/agent/api.py index caa5935..927290c 100644 --- a/backend/agent/api.py +++ b/backend/agent/api.py @@ -552,6 +552,7 @@ async def route_workflow(request: RouterSelectRequest): return await select_workflow( db, message=request.message, recent_messages=request.recent_messages[-4:], attachments=request.attachments, library_slug=request.library_slug, router_model=request.model, chat_model=request.chat_model, + web_search_enabled=request.web_search_enabled, confidence_threshold=request.confidence_threshold, ) finally: diff --git a/backend/agent/schemas.py b/backend/agent/schemas.py index 9d2d8ac..bc05bde 100644 --- a/backend/agent/schemas.py +++ b/backend/agent/schemas.py @@ -122,6 +122,7 @@ class RouterSelectRequest(BaseModel): recent_messages: List[Dict[str, Any]] = Field(default_factory=list) attachments: List[Dict[str, Any]] = Field(default_factory=list) library_slug: Optional[str] = None + web_search_enabled: bool = False model: Optional[str] = None chat_model: Optional[str] = None confidence_threshold: float = 0.55