This commit is contained in:
Timothy J. Baek
2024-10-08 17:03:42 -07:00
parent a4870f4c91
commit 3391a855f0

View File

@@ -71,7 +71,7 @@ def query_doc(
try:
result = VECTOR_DB_CLIENT.search(
collection_name=collection_name,
vectors=query_embedding,
vectors=[query_embedding],
limit=k,
)