feat: optimize weaviate error msg (#441)

pull/442/head
John Wang 3 years ago committed by GitHub
parent 3241e4015b
commit 951afcaaed
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -1,5 +1,6 @@
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):
weaviate.connect.connection.has_grpc = False weaviate.connect.connection.has_grpc = False
client = weaviate.Client( try:
url=config.endpoint, client = weaviate.Client(
auth_client_secret=auth_config, url=config.endpoint,
timeout_config=(5, 60), auth_client_secret=auth_config,
startup_period=None timeout_config=(5, 60),
) 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

Loading…
Cancel
Save