fix: Adjust timeout handling in litellm.completion method

This commit is contained in:
Paul Gauthier 2025-02-06 11:37:44 -08:00 committed by Paul Gauthier (aider)
parent 39855f4d2b
commit 685e63b9da

View file

@ -601,6 +601,7 @@ class Model(ModelSettings):
# dump(kwargs)
hash_object = hashlib.sha1(key)
# don't pass timeout here, add it to kwargs (if not already present) ai!
res = litellm.completion(timeout=request_timeout, **kwargs)
return hash_object, res