Procházet zdrojové kódy

修改删除向量库返回的数据

weiyu před 8 měsíci
rodič
revize
7cfccdff81
1 změnil soubory, kde provedl 1 přidání a 1 odebrání
  1. 1 1
      rag/vector_db/milvus_vector.py

+ 1 - 1
rag/vector_db/milvus_vector.py

@@ -379,7 +379,7 @@ class HybridRetriever:
         if not results:
             print(f"No data found for chunk id: {chunk_id}")
             # return "delete_no_result", []
-            return "delete_success", 0
+            return "delete_success", [0]
         
         # 提取主键值
         primary_keys = [result["pk"] for result in results]