diff --git a/litellm/main.py b/litellm/main.py index 9d79a33c4a8..8bff2ff4022 100644 --- a/litellm/main.py +++ b/litellm/main.py @@ -341,6 +341,7 @@ async def acompletion( custom_llm_provider=custom_llm_provider, original_exception=e, completion_kwargs=completion_kwargs, + extra_kwargs=kwargs, ) @@ -2137,6 +2138,7 @@ def completion( custom_llm_provider=custom_llm_provider, original_exception=e, completion_kwargs=args, + extra_kwargs=kwargs, ) @@ -2498,6 +2500,7 @@ async def aembedding(*args, **kwargs): custom_llm_provider=custom_llm_provider, original_exception=e, completion_kwargs=args, + extra_kwargs=kwargs, ) @@ -2939,7 +2942,10 @@ def embedding( ) ## Map to OpenAI Exception raise exception_type( - model=model, original_exception=e, custom_llm_provider=custom_llm_provider + model=model, + original_exception=e, + custom_llm_provider=custom_llm_provider, + extra_kwargs=kwargs, ) @@ -3033,6 +3039,7 @@ async def atext_completion(*args, **kwargs): custom_llm_provider=custom_llm_provider, original_exception=e, completion_kwargs=args, + extra_kwargs=kwargs, ) @@ -3370,6 +3377,7 @@ async def aimage_generation(*args, **kwargs): custom_llm_provider=custom_llm_provider, original_exception=e, completion_kwargs=args, + extra_kwargs=kwargs, ) @@ -3569,6 +3577,7 @@ def image_generation( custom_llm_provider=custom_llm_provider, original_exception=e, completion_kwargs=locals(), + extra_kwargs=kwargs, ) @@ -3618,6 +3627,7 @@ async def atranscription(*args, **kwargs): custom_llm_provider=custom_llm_provider, original_exception=e, completion_kwargs=args, + extra_kwargs=kwargs, )