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

Merge branch 'main' into litellm_azure_batch_apis

Browse source
This commit is contained in:
Krish Dholakia 2024-08-22 19:07:54 -07:00 • committed by GitHub
commit 11cbf60e4f
38 changed files with 1078 additions and 159 deletions
Download patch file Download diff file Expand all files Collapse all files

6
litellm/main.py
View file

@ -943,7 +943,9 @@ def completion(
output_cost_per_token=output_cost_per_token,
cooldown_time=cooldown_time,
text_completion=kwargs.get("text_completion"),
azure_ad_token_provider=kwargs.get("azure_ad_token_provider"),
user_continue_message=kwargs.get("user_continue_message"),
)
logging.update_environment_variables(
model=model,
@ -3247,6 +3249,10 @@ def embedding(
"model_config",
"cooldown_time",
"tags",
"azure_ad_token_provider",
"tenant_id",
"client_id",
"client_secret",
"extra_headers",
]
default_params = openai_params + litellm_params