소스 검색

feat: optimize weaviate error msg (#441)

tags/0.3.6
John Wang 2 년 전
부모
커밋
951afcaaed
No account linked to committer's email address
1개의 변경된 파일10개의 추가작업 그리고 6개의 파일을 삭제
  1. 10
    6
      api/core/index/vector_index/weaviate_vector_index.py

+ 10
- 6
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

Loading…
취소
저장