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