浏览代码

add jina rerank http timout parameter (#10476)

Jyong 5 月之前
父节点
当前提交
0c1307b083
共有 1 个文件被更改,包括 1 次插入0 次删除
  1. 1 0
      api/core/model_runtime/model_providers/jina/rerank/rerank.py

+ 1 - 0
api/core/model_runtime/model_providers/jina/rerank/rerank.py

@@ -55,6 +55,7 @@ class JinaRerankModel(RerankModel):
                 base_url + "/rerank",
                 json={"model": model, "query": query, "documents": docs, "top_n": top_n},
                 headers={"Authorization": f"Bearer {credentials.get('api_key')}"},
+                timeout=20,
             )
             response.raise_for_status()
             results = response.json()