Parcourir la source

add jina rerank http timout parameter (#10476)

Jyong il y a 5 mois
Parent
commit
0c1307b083
1 fichiers modifiés avec 1 ajouts et 0 suppressions
  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()