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