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

fix logic for deep copying otel spans / traces

Browse source
This commit is contained in:
Ishaan Jaff 2024-06-07 15:10:05 -07:00
parent 1915cd326c
commit 15c0d334e8
1 changed files with 3 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

8
litellm/proxy/utils.py
View file

@ -83,15 +83,13 @@ def safe_deep_copy(data):
if isinstance(data, dict): if isinstance(data, dict):
# remove litellm_parent_otel_span since this is not picklable # remove litellm_parent_otel_span since this is not picklable
if "metadata" in data and "litellm_parent_otel_span" in data["metadata"]: if "metadata" in data and "litellm_parent_otel_span" in data["metadata"]:
data["metadata"].pop("litellm_parent_otel_span") litellm_parent_otel_span = data["metadata"].pop("litellm_parent_otel_span")
new_data = copy.deepcopy(data) new_data = copy.deepcopy(data)
# Step 2: re-add the litellm_parent_otel_span after doing a deep copy # Step 2: re-add the litellm_parent_otel_span after doing a deep copy
if isinstance(data, dict): if isinstance(data, dict):
if "metadata" in data and "litellm_parent_otel_span" in data["metadata"]: if "metadata" in data:
data["metadata"]["litellm_parent_otel_span"] = data["metadata"][ data["metadata"]["litellm_parent_otel_span"] = litellm_parent_otel_span
"litellm_parent_otel_span"
]
return new_data return new_data