chore: docs query test fix

julep-ai · Jan 6, 2025 · cc93353 · cc93353
1 parent 0e689d9
commit cc93353
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/agents-api/tests/test_docs_queries.py b/agents-api/tests/test_docs_queries.py
@@ -272,7 +272,7 @@ async def _(
     result = await search_docs_by_embedding(
         developer_id=developer.id,
         owners=[("agent", agent.id)],
-        query_embedding=query_embedding,
+        embedding=query_embedding,
         k=3,  # Add k parameter
         metadata_filter={"test": "test"},  # Add metadata filter
         connection_pool=pool,