fix simplify - pass litellm_parent_otel_span

This commit is contained in:
Ishaan Jaff 2024-06-07 13:48:21 -07:00
parent 42df97db3c
commit 7ef7bc8a9a
6 changed files with 5 additions and 9 deletions

View file

@ -73,7 +73,8 @@ def print_verbose(print_statement):
def safe_deep_copy(data):
if isinstance(data, dict):
# remove litellm_parent_otel_span since this is not picklable
data.pop("litellm_parent_otel_span", None)
if "metadata" in data and "litellm_parent_otel_span" in data["metadata"]:
data["metadata"].pop("litellm_parent_otel_span")
new_data = copy.deepcopy(data)
return new_data