Переглянути джерело

Refactor BaseVectorIndex delete method (#1853)

tags/0.4.0
waltcow 1 рік тому
джерело
коміт
06d2d8cea3
Аккаунт користувача з таким Email не знайдено
1 змінених файлів з 4 додано та 9 видалено
  1. 4
    9
      api/core/index/vector_index/base.py

+ 4
- 9
api/core/index/vector_index/base.py Переглянути файл

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,

Завантаження…
Відмінити
Зберегти