| from langchain.embeddings.base import Embeddings | from langchain.embeddings.base import Embeddings | ||||
| from langchain.schema import Document, BaseRetriever | from langchain.schema import Document, BaseRetriever | ||||
| from langchain.vectorstores import VectorStore | from langchain.vectorstores import VectorStore | ||||
| from weaviate import UnexpectedStatusCodeException | |||||
| from core.index.base import BaseIndex | from core.index.base import BaseIndex | ||||
| from extensions.ext_database import db | from extensions.ext_database import db | ||||
| try: | try: | ||||
| self.delete() | self.delete() | ||||
| except UnexpectedStatusCodeException as e: | |||||
| if e.status_code != 400: | |||||
| # 400 means index not exists | |||||
| raise e | |||||
| except Exception as e: | |||||
| raise e | |||||
| dataset_documents = db.session.query(DatasetDocument).filter( | dataset_documents = db.session.query(DatasetDocument).filter( | ||||
| DatasetDocument.dataset_id == dataset.id, | DatasetDocument.dataset_id == dataset.id, | ||||
| try: | try: | ||||
| self.delete() | self.delete() | ||||
| except UnexpectedStatusCodeException as e: | |||||
| if e.status_code != 400: | |||||
| # 400 means index not exists | |||||
| raise e | |||||
| except Exception as e: | |||||
| raise e | |||||
| dataset_documents = db.session.query(DatasetDocument).filter( | dataset_documents = db.session.query(DatasetDocument).filter( | ||||
| DatasetDocument.dataset_id == dataset.id, | DatasetDocument.dataset_id == dataset.id, |