소스 검색

fix weaviate delete_by_ids (#2565)

tags/0.5.7
Bowen Liang 1 년 전
부모
커밋
562ca45e07
No account linked to committer's email address
1개의 변경된 파일5개의 추가작업 그리고 4개의 파일을 삭제
  1. 5
    4
      api/core/rag/datasource/vdb/weaviate/weaviate_vector.py

+ 5
- 4
api/core/rag/datasource/vdb/weaviate/weaviate_vector.py 파일 보기

@@ -150,10 +150,11 @@ class WeaviateVector(BaseVector):
return True

def delete_by_ids(self, ids: list[str]) -> None:
self._client.data_object.delete(
ids,
class_name=self._collection_name
)
for uuid in ids:
self._client.data_object.delete(
class_name=self._collection_name,
uuid=uuid,
)

def search_by_vector(self, query_vector: list[float], **kwargs: Any) -> list[Document]:
"""Look up similar documents by embedding vector in Weaviate."""

Loading…
취소
저장