hotfix(agents-api): fix in docs routes

julep-ai · Jan 11, 2025 · ffb934d · ffb934d
1 parent 0509485
commit ffb934d
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 0 deletions.
diff --git a/agents-api/agents_api/routers/docs/create_doc.py b/agents-api/agents_api/routers/docs/create_doc.py
@@ -44,6 +44,7 @@ async def create_agent_doc(
     agent_id: UUID,
     data: CreateDocRequest,
     x_developer_id: Annotated[UUID, Depends(get_developer_id)],
+    connection_pool: Any = None,  # FIXME: Placeholder that should be removed
 ) -> ResourceCreatedResponse:
     doc: Doc = await create_doc_query(
         developer_id=x_developer_id,

diff --git a/agents-api/agents_api/routers/docs/search_docs.py b/agents-api/agents_api/routers/docs/search_docs.py
@@ -133,6 +133,8 @@ async def search_agent_docs(
     x_developer_id: Annotated[UUID, Depends(get_developer_id)],
     search_params: (TextOnlyDocSearchRequest | VectorDocSearchRequest | HybridDocSearchRequest),
     agent_id: UUID,
+    connection_pool: Any = None,  # FIXME: Placeholder that should be removed
+
 ) -> DocSearchResponse:
     """
     Searches for documents associated with a specific agent.