File tree Expand file tree Collapse file tree 1 file changed +1
-3
lines changed
Expand file tree Collapse file tree 1 file changed +1
-3
lines changed Original file line number Diff line number Diff 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
You can’t perform that action at this time.
0 commit comments