feat: add timeout configuration for litellm requests

This commit is contained in:
Paul Gauthier 2024-11-30 13:03:20 -08:00 committed by Paul Gauthier (aider)
parent a3e7203331
commit ade4847c61

View file

@ -467,6 +467,10 @@ def main(argv=None, input=None, output=None, force_git_root=None, return_coder=F
litellm._lazy_module.client_session = httpx.Client(verify=False)
litellm._lazy_module.aclient_session = httpx.AsyncClient(verify=False)
if args.timeout:
litellm._load_litellm()
litellm._lazy_module.request_timeout = args.timeout
if args.dark_mode:
args.user_input_color = "#32FF32"
args.tool_error_color = "#FF3333"