From cc933530f7c5b398ba9880e08386606caf2bc60c Mon Sep 17 00:00:00 2001
From: vedantsahai18 <vedantsahai18@gmail.com>
Date: Sun, 5 Jan 2025 19:32:53 -0500
Subject: [PATCH] chore: docs query test fix

---
 agents-api/tests/test_docs_queries.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/agents-api/tests/test_docs_queries.py b/agents-api/tests/test_docs_queries.py
index b6b23438a..6690badfd 100644
--- 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,