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

use litellm_parent_otel_span as litellm_param

Browse source
This commit is contained in:
Ishaan Jaff 2024-06-07 08:54:28 -07:00
parent 8370f81aa6
commit 0f99d47d87
2 changed files with 7 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

3
litellm/main.py
View file

@ -600,6 +600,7 @@ def completion(
client = kwargs.get("client", None)
### Admin Controls ###
no_log = kwargs.get("no-log", False)
litellm_parent_otel_span = kwargs.get("litellm_parent_otel_span", None)
######## end of unpacking kwargs ###########
openai_params = [
"functions",
@ -689,6 +690,7 @@ def completion(
"allowed_model_region",
"model_config",
"fastest_response",
"litellm_parent_otel_span",
]
default_params = openai_params + litellm_params
@ -873,6 +875,7 @@ def completion(
input_cost_per_token=input_cost_per_token,
output_cost_per_second=output_cost_per_second,
output_cost_per_token=output_cost_per_token,
litellm_parent_otel_span=litellm_parent_otel_span,
)
logging.update_environment_variables(
model=model,