Skip to content

Commit 6e5a4e1

Browse files
committed
fix(langchain): filter string keys from usage object
1 parent 91b77c0 commit 6e5a4e1

1 file changed

Lines changed: 2 additions & 0 deletions

File tree

langfuse/langchain/CallbackHandler.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -880,6 +880,8 @@ def _parse_usage_model(usage: typing.Union[pydantic.BaseModel, dict]):
880880
"token_count"
881881
]
882882

883+
usage_model = {k: v for k, v in usage_model.items() if not isinstance(v, str)}
884+
883885
return usage_model if usage_model else None
884886

885887

0 commit comments

Comments
 (0)