浏览代码

Refactor BaseVectorIndex delete method (#1853)

waltcow 1 年之前
父节点
当前提交
06d2d8cea3
共有 1 个文件被更改,包括 4 次插入9 次删除
  1. 4 9
      api/core/index/vector_index/base.py

+ 4 - 9
api/core/index/vector_index/base.py

@@ -6,7 +6,6 @@ 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
 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
@@ -139,10 +138,8 @@ class BaseVectorIndex(BaseIndex):
 
 
         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,
@@ -194,10 +191,8 @@ class BaseVectorIndex(BaseIndex):
 
 
         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,