Commit e14ffcca authored by John Wang's avatar John Wang

fix: delete document bug

parent 9e40dbd6
import json import json
import logging import logging
from abc import abstractmethod from abc import abstractmethod
from typing import List, Any, Tuple, cast from typing import List, Any, cast
from langchain.embeddings.base import Embeddings from langchain.embeddings.base import Embeddings
from langchain.schema import Document, BaseRetriever from langchain.schema import Document, BaseRetriever
...@@ -101,6 +101,7 @@ class BaseVectorIndex(BaseIndex): ...@@ -101,6 +101,7 @@ class BaseVectorIndex(BaseIndex):
def delete_by_ids(self, ids: list[str]) -> None: def delete_by_ids(self, ids: list[str]) -> None:
if self._is_origin(): if self._is_origin():
self.recreate_dataset(self.dataset) self.recreate_dataset(self.dataset)
return
vector_store = self._get_vector_store() vector_store = self._get_vector_store()
vector_store = cast(self._get_vector_store_class(), vector_store) vector_store = cast(self._get_vector_store_class(), vector_store)
......
...@@ -90,6 +90,7 @@ class QdrantVectorIndex(BaseVectorIndex): ...@@ -90,6 +90,7 @@ class QdrantVectorIndex(BaseVectorIndex):
def delete_by_document_id(self, document_id: str): def delete_by_document_id(self, document_id: str):
if self._is_origin(): if self._is_origin():
self.recreate_dataset(self.dataset) self.recreate_dataset(self.dataset)
return
vector_store = self._get_vector_store() vector_store = self._get_vector_store()
vector_store = cast(self._get_vector_store_class(), vector_store) vector_store = cast(self._get_vector_store_class(), vector_store)
......
...@@ -111,6 +111,7 @@ class WeaviateVectorIndex(BaseVectorIndex): ...@@ -111,6 +111,7 @@ class WeaviateVectorIndex(BaseVectorIndex):
def delete_by_document_id(self, document_id: str): def delete_by_document_id(self, document_id: str):
if self._is_origin(): if self._is_origin():
self.recreate_dataset(self.dataset) self.recreate_dataset(self.dataset)
return
vector_store = self._get_vector_store() vector_store = self._get_vector_store()
vector_store = cast(self._get_vector_store_class(), vector_store) vector_store = cast(self._get_vector_store_class(), vector_store)
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment