From 7dadca6a990b7a760ae1f4eea7bb6d8a6ca79da0 Mon Sep 17 00:00:00 2001
From: Krrish Dholakia <krrishdholakia@gmail.com>
Date: Fri, 7 Jun 2024 09:01:07 -0700
Subject: [PATCH] fix(bedrock_httpx.py): fix linting errors

---
 .pre-commit-config.yaml       | 16 ++++++++--------
 litellm/llms/bedrock_httpx.py | 12 ++++++------
 2 files changed, 14 insertions(+), 14 deletions(-)

diff --git a/.pre-commit-config.yaml b/.pre-commit-config.yaml
index e8bb1ff66a..cc41d85f14 100644
--- a/.pre-commit-config.yaml
+++ b/.pre-commit-config.yaml
@@ -16,11 +16,11 @@ repos:
         name: Check if files match
         entry: python3 ci_cd/check_files_match.py
         language: system
-# -   repo: local
-#     hooks:
-#     -   id: mypy
-#         name: mypy
-#         entry: python3 -m mypy --ignore-missing-imports
-#         language: system
-#         types: [python]
-#         files: ^litellm/
\ No newline at end of file
+-   repo: local
+    hooks:
+    -   id: mypy
+        name: mypy
+        entry: python3 -m mypy --ignore-missing-imports
+        language: system
+        types: [python]
+        files: ^litellm/
\ No newline at end of file
diff --git a/litellm/llms/bedrock_httpx.py b/litellm/llms/bedrock_httpx.py
index 7aba78d7ce..9e44f16cfc 100644
--- a/litellm/llms/bedrock_httpx.py
+++ b/litellm/llms/bedrock_httpx.py
@@ -1879,7 +1879,7 @@ class AWSEventStreamDecoder:
         elif "stopReason" in chunk_data:
             finish_reason = map_finish_reason(chunk_data.get("stopReason", "stop"))
         elif "usage" in chunk_data:
-            usage = ConverseTokenUsageBlock(**chunk_data["usage"])
+            usage = ConverseTokenUsageBlock(**chunk_data["usage"])  # type: ignore
         response = GenericStreamingChunk(
             text=text,
             tool_str=tool_str,
@@ -1929,11 +1929,11 @@ class AWSEventStreamDecoder:
             is_finished = True
             finish_reason = chunk_data["completionReason"]
         return GenericStreamingChunk(
-            **{
-                "text": text,
-                "is_finished": is_finished,
-                "finish_reason": finish_reason,
-            }
+            text=text,
+            is_finished=is_finished,
+            finish_reason=finish_reason,
+            tool_str="",
+            usage=None,
         )
 
     def iter_bytes(self, iterator: Iterator[bytes]) -> Iterator[GenericStreamingChunk]: