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(utils.py): support image gen logging to langfuse

Browse source
This commit is contained in:
Krrish Dholakia 2024-02-16 16:12:52 -08:00
parent 5f9e141d1e
commit f57483ea70
5 changed files with 58 additions and 91 deletions
Download patch file Download diff file Expand all files Collapse all files

9
litellm/integrations/langfuse.py
View file

@ -110,9 +110,16 @@ class LangFuseLogger:
):
input = prompt
output = response_obj["data"]
elif response_obj is not None:
elif response_obj is not None and isinstance(
response_obj, litellm.ModelResponse
):
input = prompt
output = response_obj["choices"][0]["message"].json()
elif response_obj is not None and isinstance(
response_obj, litellm.ImageResponse
):
input = prompt
output = response_obj["data"]
print_verbose(f"OUTPUT IN LANGFUSE: {output}; original: {response_obj}")
if self._is_langfuse_v2():
self._log_langfuse_v2(