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

add completion call id

Browse source
This commit is contained in:
Krrish Dholakia 2023-08-31 12:54:34 -07:00
parent ef947bf144
commit 66bfd70253
3 changed files with 5 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
litellm/main.py
View file

@ -94,6 +94,7 @@ def completion(
custom_api_base=None,
litellm_call_id=None,
litellm_logging_obj=None,
completion_call_id=None, # this is an optional param to tag individual completion calls
# model specific optional params
# used by text-bison only
top_k=40,
@ -153,6 +154,7 @@ def completion(
custom_api_base=custom_api_base,
litellm_call_id=litellm_call_id,
model_alias_map=litellm.model_alias_map,
completion_call_id=completion_call_id
)
logging.update_environment_variables(optional_params=optional_params, litellm_params=litellm_params)
if custom_llm_provider == "azure":