fix(langfuse.py): fix metadata logging for langfuse

This commit is contained in:
Krrish Dholakia 2023-11-30 13:45:26 -08:00
parent 5ff3cf6007
commit 1600ad11fc
2 changed files with 6 additions and 5 deletions

View file

@ -36,7 +36,7 @@ class LangFuseLogger:
print_verbose(
f"Langfuse Logging - Enters logging function for model {kwargs}"
)
metadata = kwargs.get("metadata", {})
metadata = kwargs["litellm_params"].get("metadata", {})
prompt = [kwargs.get('messages')]
optional_params = kwargs.get("optional_params", {})
@ -50,7 +50,6 @@ class LangFuseLogger:
pass
# end of processing langfuse ########################
self.Langfuse.generation(InitialGeneration(
name=metadata.get("generation_name", "litellm-completion"),
startTime=start_time,