From a6ebffb69504576a805f3b9f09732ad344751b89 Mon Sep 17 00:00:00 2001
From: Piyush Jain <piyushjain@duck.com>
Date: Thu, 8 Jun 2023 14:16:01 -0700
Subject: [PATCH] Fixes model arguments for amazon models (#5896)

Fixes #5713
#### Who can review?

Tag maintainers/contributors who might be interested:
@hwchase17
@agola11
@aarora79
@rsgrewal-aws
---
 langchain/llms/bedrock.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/langchain/llms/bedrock.py b/langchain/llms/bedrock.py
index 22296030..b87f8483 100644
--- a/langchain/llms/bedrock.py
+++ b/langchain/llms/bedrock.py
@@ -20,6 +20,10 @@ class LLMInputOutputAdapter:
         input_body = {**model_kwargs}
         if provider == "anthropic" or provider == "ai21":
             input_body["prompt"] = prompt
+        elif provider == "amazon":
+            input_body = dict()
+            input_body["inputText"] = prompt
+            input_body["textGenerationConfig"] = {**model_kwargs}
         else:
             input_body["inputText"] = prompt