瀏覽代碼

fix: shorten connection timeout to pypi.org for deprecation check for weaviate client (#21131)

tags/1.5.0
kurokobo 4 月之前
父節點
當前提交
4689e8953e
沒有連結到貢獻者的電子郵件帳戶。
共有 1 個檔案被更改,包括 6 行新增0 行删除
  1. 6
    0
      api/core/rag/datasource/vdb/weaviate/weaviate_vector.py

+ 6
- 0
api/core/rag/datasource/vdb/weaviate/weaviate_vector.py 查看文件

@@ -41,6 +41,12 @@ class WeaviateVector(BaseVector):

weaviate.connect.connection.has_grpc = False

# Fix to minimize the performance impact of the deprecation check in weaviate-client 3.24.0,
# by changing the connection timeout to pypi.org from 1 second to 0.001 seconds.
# TODO: This can be removed once weaviate-client is updated to 3.26.7 or higher,
# which does not contain the deprecation check.
weaviate.connect.connection.PYPI_TIMEOUT = 0.001

try:
client = weaviate.Client(
url=config.endpoint, auth_client_secret=auth_config, timeout_config=(5, 60), startup_period=None

Loading…
取消
儲存