Commit 48552a5b authored by John Wang's avatar John Wang

Merge branch 'feat/upgrade-langchain' into deploy/dev

parents 2149ae9d a2d498f0
...@@ -6,6 +6,7 @@ from typing import List, Any, cast ...@@ -6,6 +6,7 @@ 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
...@@ -120,7 +121,13 @@ class BaseVectorIndex(BaseIndex): ...@@ -120,7 +121,13 @@ class BaseVectorIndex(BaseIndex):
def recreate_dataset(self, dataset: Dataset): def recreate_dataset(self, dataset: Dataset):
logging.info(f"Recreating dataset {dataset.id}") logging.info(f"Recreating dataset {dataset.id}")
self.delete()
try:
self.delete()
except UnexpectedStatusCodeException as e:
if e.status_code != 400:
# 400 means index not exists
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,
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment