Unverified Commit 951afcaa authored by John Wang's avatar John Wang Committed by GitHub

feat: optimize weaviate error msg (#441)

parent 3241e401
from typing import Optional, cast from typing import Optional, cast
import requests
import weaviate import weaviate
from langchain.embeddings.base import Embeddings from langchain.embeddings.base import Embeddings
from langchain.schema import Document, BaseRetriever from langchain.schema import Document, BaseRetriever
...@@ -34,12 +35,15 @@ class WeaviateVectorIndex(BaseVectorIndex): ...@@ -34,12 +35,15 @@ class WeaviateVectorIndex(BaseVectorIndex):
weaviate.connect.connection.has_grpc = False weaviate.connect.connection.has_grpc = False
try:
client = weaviate.Client( client = weaviate.Client(
url=config.endpoint, url=config.endpoint,
auth_client_secret=auth_config, auth_client_secret=auth_config,
timeout_config=(5, 60), timeout_config=(5, 60),
startup_period=None startup_period=None
) )
except requests.exceptions.ConnectionError:
raise ConnectionError("Vector database connection error")
client.batch.configure( client.batch.configure(
# `batch_size` takes an `int` value to enable auto-batching # `batch_size` takes an `int` value to enable auto-batching
......
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