add jina rerank http timout parameter (#10476)

This commit is contained in:
Jyong 2024-11-11 13:28:11 +08:00 committed by GitHub
parent b8b6cd409a
commit 0c1307b083
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

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