Skip to content

Commit e3b21ab

Browse files
committed
fix(langchain): take only int values in parsed usage
1 parent e286441 commit e3b21ab

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

langfuse/langchain/CallbackHandler.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -965,7 +965,7 @@ def _parse_usage_model(usage: typing.Union[pydantic.BaseModel, dict]) -> Any:
965965
if "input" in usage_model:
966966
usage_model["input"] = max(0, usage_model["input"] - value)
967967

968-
usage_model = {k: v for k, v in usage_model.items() if not isinstance(v, str)}
968+
usage_model = {k: v for k, v in usage_model.items() if isinstance(v, int)}
969969

970970
return usage_model if usage_model else None
971971

0 commit comments

Comments
 (0)