Skip to content

Commit cd45de3

Browse files
committed
fix(langchain): capture usage on streamed gemini responses
1 parent 342efe9 commit cd45de3

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

langfuse/langchain/CallbackHandler.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1036,7 +1036,9 @@ def _parse_usage(response: LLMResult) -> Any:
10361036
llm_usage = _parse_usage_model(
10371037
generation_chunk.generation_info["usage_metadata"]
10381038
)
1039-
break
1039+
1040+
if llm_usage is not None:
1041+
break
10401042

10411043
message_chunk = getattr(generation_chunk, "message", {})
10421044
response_metadata = getattr(message_chunk, "response_metadata", {})

0 commit comments

Comments
 (0)