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: try:
result = VECTOR_DB_CLIENT.search( result = VECTOR_DB_CLIENT.search(
collection_name=collection_name, collection_name=collection_name,
vectors=query_embedding, vectors=[query_embedding],
limit=k, limit=k,
) )