From 951afcaaed2b7278b5fd7bcea9ac9c5f722fe999 Mon Sep 17 00:00:00 2001 From: John Wang Date: Sun, 25 Jun 2023 17:05:56 +0800 Subject: [PATCH] feat: optimize weaviate error msg (#441) --- .../index/vector_index/weaviate_vector_index.py | 16 ++++++++++------ 1 file changed, 10 insertions(+), 6 deletions(-) diff --git a/api/core/index/vector_index/weaviate_vector_index.py b/api/core/index/vector_index/weaviate_vector_index.py index e9eae4468c..df12dd9c53 100644 --- a/api/core/index/vector_index/weaviate_vector_index.py +++ b/api/core/index/vector_index/weaviate_vector_index.py @@ -1,5 +1,6 @@ from typing import Optional, cast +import requests import weaviate from langchain.embeddings.base import Embeddings from langchain.schema import Document, BaseRetriever @@ -34,12 +35,15 @@ class WeaviateVectorIndex(BaseVectorIndex): weaviate.connect.connection.has_grpc = False - client = weaviate.Client( - url=config.endpoint, - auth_client_secret=auth_config, - timeout_config=(5, 60), - startup_period=None - ) + try: + client = weaviate.Client( + url=config.endpoint, + auth_client_secret=auth_config, + timeout_config=(5, 60), + startup_period=None + ) + except requests.exceptions.ConnectionError: + raise ConnectionError("Vector database connection error") client.batch.configure( # `batch_size` takes an `int` value to enable auto-batching