Fixes error when stop reason missing in response.

langchain-ai · May 23, 2024 · 7ad4906 · 7ad4906
1 parent 9adc14d
commit 7ad4906
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/libs/aws/langchain_aws/llms/bedrock.py b/libs/aws/langchain_aws/llms/bedrock.py
@@ -257,7 +257,7 @@ def prepare_output(cls, provider: str, response: Any) -> dict:
                 "completion_tokens": completion_tokens,
                 "total_tokens": prompt_tokens + completion_tokens,
             },
-            "stop_reason": response_body["stop_reason"],
+            "stop_reason": response_body.get("stop_reason"),
         }
 
     @classmethod