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,
)