Skip to content

Commit 156f224

Browse files
committed
choreL: bump version
1 parent 6f11d1c commit 156f224

File tree

1 file changed

+1
-3
lines changed

1 file changed

+1
-3
lines changed

posthog/ai/langchain/callbacks.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -603,7 +603,6 @@ def _parse_usage_model(
603603
# langchain-ibm https://pypi.org/project/langchain-ibm/
604604
("input_token_count", "input"),
605605
("generated_token_count", "output"),
606-
607606
]
608607

609608
parsed_usage = {}
@@ -634,7 +633,7 @@ def _parse_usage(response: LLMResult):
634633
if "usage" in generation:
635634
llm_usage = _parse_usage_model(generation["usage"])
636635
break
637-
636+
638637
for generation_chunk in generation:
639638
if generation_chunk.generation_info and ("usage_metadata" in generation_chunk.generation_info):
640639
llm_usage = _parse_usage_model(generation_chunk.generation_info["usage_metadata"])
@@ -660,7 +659,6 @@ def _parse_usage(response: LLMResult):
660659
llm_usage = _parse_usage_model(chunk_usage)
661660
break
662661

663-
664662
return llm_usage
665663

666664

0 commit comments

Comments
 (0)