phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-24 10:14:26 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

(feat) completion:openai pass OpenAI client

Browse source
This commit is contained in:
ishaan-jaff 2023-11-28 16:05:01 -08:00
parent 01c38d37fa
commit 8ac7801283
1 changed files with 3 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

5
litellm/main.py
View file

@ -498,7 +498,7 @@ def completion(
logging_obj=logging,
acompletion=acompletion,
timeout=timeout,
client=optional_params.pop("client", None)
client=optional_params.pop("client", None) # pass AsyncAzureOpenAI, AzureOpenAI client
)
## LOGGING
@ -568,7 +568,8 @@ def completion(
litellm_params=litellm_params,
logger_fn=logger_fn,
timeout=timeout,
custom_prompt_dict=custom_prompt_dict
custom_prompt_dict=custom_prompt_dict,
client=optional_params.pop("client", None) # pass AsyncOpenAI, OpenAI client
)
except Exception as e:
## LOGGING - log the original exception returned