From e39056bf1f84680368d3fabd2744fa5ad0176b9c Mon Sep 17 00:00:00 2001 From: James Braza Date: Fri, 27 Sep 2024 16:09:42 -0700 Subject: [PATCH] For autogenerated `Router` kwargs, specifying `timeout` of 60-sec (#501) --- paperqa/llms.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/paperqa/llms.py b/paperqa/llms.py index 6c9b7401..50cf8eee 100644 --- a/paperqa/llms.py +++ b/paperqa/llms.py @@ -391,7 +391,7 @@ def maybe_set_config_attribute(cls, data: dict[str, Any]) -> dict[str, Any]: ), } ], - "router_kwargs": {"num_retries": 3, "retry_after": 5}, + "router_kwargs": {"num_retries": 3, "retry_after": 5, "timeout": 60}, } # we only support one "model name" for now, here we validate model_list = data["config"]["model_list"]