add completion call id

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

View file

@ -646,6 +646,7 @@ def get_litellm_params(
custom_api_base=None,
litellm_call_id=None,
model_alias_map=None,
completion_call_id=None
):
litellm_params = {
"return_async": return_async,
@ -657,6 +658,7 @@ def get_litellm_params(
"custom_api_base": custom_api_base,
"litellm_call_id": litellm_call_id,
"model_alias_map": model_alias_map,
"completion_call_id": completion_call_id,
"stream_response": {} # litellm_call_id: ModelResponse Dict
}