fix: added missing parameters.

added missing parameters in `simple_send_with_retries` function.
This commit is contained in:
hypn4 2024-09-21 05:23:38 +09:00
parent d0bce02c00
commit 6dc846d41b

View file

@ -96,7 +96,7 @@ def send_completion(
@lazy_litellm_retry_decorator
def simple_send_with_retries(model_name, messages, extra_headers=None):
def simple_send_with_retries(model_name, messages, extra_headers=None, extra_body=None):
try:
kwargs = {
"model_name": model_name,