jyapayne/dify
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

add jina rerank http timout parameter (#10476)

Browse source
This commit is contained in:
Jyong 2024-11-11 13:28:11 +08:00 • committed by GitHub
commit 0c1307b083
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 1 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

1
api/core/model_runtime/model_providers/jina/rerank/rerank.py
View file

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