Pārlūkot izejas kodu

add jina rerank http timout parameter (#10476)

Jyong 4 mēneši atpakaļ
vecāks
revīzija
0c1307b083

+ 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()