瀏覽代碼

Fix vdb merge error (#2650)

tags/0.5.8
waltcow 1 年之前
父節點
當前提交
f322d9bddb
沒有連結到貢獻者的電子郵件帳戶。
共有 2 個檔案被更改,包括 24 行新增15 行删除
  1. 23
    14
      api/core/rag/datasource/vdb/qdrant/qdrant_vector.py
  2. 1
    1
      api/requirements.txt

+ 23
- 14
api/core/rag/datasource/vdb/qdrant/qdrant_vector.py 查看文件

@@ -231,21 +231,30 @@ class QdrantVector(BaseVector):

def delete(self):
from qdrant_client.http import models
filter = models.Filter(
must=[
models.FieldCondition(
key="group_id",
match=models.MatchValue(value=self._group_id),
from qdrant_client.http.exceptions import UnexpectedResponse
try:
filter = models.Filter(
must=[
models.FieldCondition(
key="group_id",
match=models.MatchValue(value=self._group_id),
),
],
)
self._client.delete(
collection_name=self._collection_name,
points_selector=FilterSelector(
filter=filter
),
],
)
self._client.delete(
collection_name=self._collection_name,
points_selector=FilterSelector(
filter=filter
),
)

)
except UnexpectedResponse as e:
# Collection does not exist, so return
if e.status_code == 404:
return
# Some other error occurred, so re-raise the exception
else:
raise e
def delete_by_ids(self, ids: list[str]) -> None:

from qdrant_client.http import models

+ 1
- 1
api/requirements.txt 查看文件

@@ -52,7 +52,7 @@ safetensors==0.3.2
zhipuai==1.0.7
werkzeug~=3.0.1
pymilvus==2.3.0
qdrant-client==1.6.4
qdrant-client==1.7.3
cohere~=4.44
pyyaml~=6.0.1
numpy~=1.25.2

Loading…
取消
儲存