(feat) show langfuse logging tags better through proxy

This commit is contained in:
ishaan-jaff 2024-02-06 13:09:48 -08:00 committed by Krrish Dholakia
parent 31f85d65ae
commit a73d57b32b

View file

@ -252,8 +252,14 @@ class LangFuseLogger:
print_verbose(f"trace: {cost}") print_verbose(f"trace: {cost}")
if supports_tags: if supports_tags:
for key, value in metadata.items(): for key, value in metadata.items():
tags.append(f"{key}:{value}") if key in [
"user_api_key",
"user_api_key_user_id",
]:
tags.append(f"{key}:{value}")
if "cache_hit" in kwargs: if "cache_hit" in kwargs:
if kwargs["cache_hit"] is None:
kwargs["cache_hit"] = False
tags.append(f"cache_hit:{kwargs['cache_hit']}") tags.append(f"cache_hit:{kwargs['cache_hit']}")
trace_params.update({"tags": tags}) trace_params.update({"tags": tags})