Fixes error when stop reason missing in response.

3coins · 3coins · commit 7ad4906241f3 · 2024-05-23T11:39:36.000-07:00
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

Original file line number	Diff line number	Diff line change
`@@ -257,7 +257,7 @@ def prepare_output(cls, provider: str, response: Any) -> dict:`
`257`	`257`	`"completion_tokens": completion_tokens,`
`258`	`258`	`"total_tokens": prompt_tokens + completion_tokens,`
`259`	`259`	`},`
`260`		`- "stop_reason": response_body["stop_reason"],`
	`260`	`+ "stop_reason": response_body.get("stop_reason"),`
`261`	`261`	`}`
`262`	`262`
`263`	`263`	`@classmethod`