Refactor BaseVectorIndex delete method (#1853)

This commit is contained in:
waltcow 2023-12-30 21:49:01 +08:00 committed by GitHub
parent 936c3cc4d7
commit 06d2d8cea3
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -6,7 +6,6 @@ from typing import List, Any, cast
from langchain.embeddings.base import Embeddings
from langchain.schema import Document, BaseRetriever
from langchain.vectorstores import VectorStore
from weaviate import UnexpectedStatusCodeException
from core.index.base import BaseIndex
from extensions.ext_database import db
@ -139,9 +138,7 @@ class BaseVectorIndex(BaseIndex):
try:
self.delete()
except UnexpectedStatusCodeException as e:
if e.status_code != 400:
# 400 means index not exists
except Exception as e:
raise e
dataset_documents = db.session.query(DatasetDocument).filter(
@ -194,9 +191,7 @@ class BaseVectorIndex(BaseIndex):
try:
self.delete()
except UnexpectedStatusCodeException as e:
if e.status_code != 400:
# 400 means index not exists
except Exception as e:
raise e
dataset_documents = db.session.query(DatasetDocument).filter(