From c2ca4b1e1c86fd48526a93ca4806dca02a3ee0b0 Mon Sep 17 00:00:00 2001 From: Piyush Jain Date: Wed, 3 Apr 2024 21:37:22 -0700 Subject: [PATCH] Fixed liniting. --- libs/aws/langchain_aws/chat_models/bedrock.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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