Feat: add exponential back-off for Chat LiteLLM (#9880)
### What problem does this PR solve? Add exponential back-off for Chat LiteLLM. #9858. ### Type of change - [x] New Feature (non-breaking change which adds functionality)
This commit is contained in:
parent
1bc33009c7
commit
b58e882eaa
1 changed files with 1 additions and 0 deletions
|
|
@ -1534,6 +1534,7 @@ class LiteLLMBase(ABC):
|
||||||
"model": self.model_name,
|
"model": self.model_name,
|
||||||
"messages": history,
|
"messages": history,
|
||||||
"api_key": self.api_key,
|
"api_key": self.api_key,
|
||||||
|
"num_retries": self.max_retries,
|
||||||
**kwargs,
|
**kwargs,
|
||||||
}
|
}
|
||||||
if stream:
|
if stream:
|
||||||
|
|
|
||||||
Loading…
Add table
Reference in a new issue