Parcourir la source

Refactor BaseVectorIndex delete method (#1853)

tags/0.4.0
waltcow il y a 1 an
Parent
révision
06d2d8cea3
Aucun compte lié à l'adresse e-mail de l'auteur
1 fichiers modifiés avec 4 ajouts et 9 suppressions
  1. 4
    9
      api/core/index/vector_index/base.py

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

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

try:
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(
DatasetDocument.dataset_id == dataset.id,
@@ -194,10 +191,8 @@ class BaseVectorIndex(BaseIndex):

try:
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(
DatasetDocument.dataset_id == dataset.id,

Chargement…
Annuler
Enregistrer