fixes langchain-ai#176

ardf · Sep 2, 2024 · f84b8df · f84b8df
1 parent d89fcf8
commit f84b8df
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/libs/aws/langchain_aws/llms/bedrock.py b/libs/aws/langchain_aws/llms/bedrock.py
@@ -144,8 +144,8 @@ def _stream_response_to_generation_chunk(
             return AIMessageChunk(
                 content="",
                 response_metadata={
-                    "stop_reason": stream_response["delta"]["stop_reason"],
-                    "stop_sequence": stream_response["delta"]["stop_sequence"],
+                    "stop_reason": stream_response["delta"].get("stop_reason"),
+                    "stop_sequence": stream_response["delta"].get("stop_sequence"),
                 },
             )
         else: