add jina rerank http timout parameter (#10476)

2024-11-11 13:28:11 +08:00
parent b8b6cd409a
commit 0c1307b083
1 changed files with 1 additions and 0 deletions
--- a/api/core/model_runtime/model_providers/jina/rerank/rerank.py
+++ b/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()