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

fix: lunary callback tags

Browse source
This commit is contained in:
hughcrt 2024-06-11 21:30:11 -03:00
parent 5994ca572c
commit 93cb18c633
1 changed files with 2 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

6
litellm/integrations/lunary.py
View file

@ -105,19 +105,17 @@ class LunaryLogger:
end_time=datetime.now(timezone.utc), end_time=datetime.now(timezone.utc),
error=None, error=None,
): ):
# Method definition
try: try:
print_verbose(f"Lunary Logging - Logging request for model {model}") print_verbose(f"Lunary Logging - Logging request for model {model}")
litellm_params = kwargs.get("litellm_params", {}) litellm_params = kwargs.get("litellm_params", {})
optional_params = kwargs.get("optional_params", {}) optional_params = kwargs.get("optional_params", {})
metadata = litellm_params.get("metadata", {}) or {} metadata = litellm_params.get("metadata", {})
if optional_params: if optional_params:
# merge into extra
extra = {**extra, **optional_params} extra = {**extra, **optional_params}
tags = litellm_params.pop("tags", None) or [] tags = metadata.get("metadata", None)
if extra: if extra:
extra.pop("extra_body", None) extra.pop("extra_body", None)