Преглед на файлове

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