Преглед на файлове

Fix vdb merge error (#2650)

tags/0.5.8
waltcow преди 1 година
родител
ревизия
f322d9bddb
No account linked to committer's email address
променени са 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…
Отказ
Запис