fix(langfuse.py): serialize message for logging

This commit is contained in:
Krrish Dholakia 2023-12-12 21:40:55 -08:00
parent 7856d65d44
commit f9dfeb502a
2 changed files with 13 additions and 5 deletions

View file

@ -58,7 +58,7 @@ class LangFuseLogger:
model=kwargs['model'],
modelParameters=optional_params,
prompt=prompt,
completion=response_obj['choices'][0]['message'],
completion=response_obj['choices'][0]['message'].json(),
usage=Usage(
prompt_tokens=response_obj['usage']['prompt_tokens'],
completion_tokens=response_obj['usage']['completion_tokens']