diff --git a/libs/aws/langchain_aws/chat_models/bedrock.py b/libs/aws/langchain_aws/chat_models/bedrock.py index 4213fcfc..72f725c7 100644 --- a/libs/aws/langchain_aws/chat_models/bedrock.py +++ b/libs/aws/langchain_aws/chat_models/bedrock.py @@ -331,7 +331,7 @@ def _generate( ) -> ChatResult: completion = "" llm_output: Dict[str, Any] = {"model_id": self.model_id} - usage_info = {} + usage_info: Dict[str, Any] = {} if self.streaming: for chunk in self._stream(messages, stop, run_manager, **kwargs): completion += chunk.text