Pedro Bento commited on
Commit
1711059
·
1 Parent(s): 3271610

Fix to BedrockChat model invoke

Browse files
Files changed (1) hide show
  1. tdagent/grchat.py +1 -1
tdagent/grchat.py CHANGED
@@ -209,7 +209,7 @@ def create_bedrock_llm(
209
  try:
210
  bedrock_client = boto3.client("bedrock-runtime", **boto3_config)
211
  llm = ChatBedrock(
212
- model_id=bedrock_model_id,
213
  client=bedrock_client,
214
  model_kwargs={"temperature": temperature, "max_tokens": max_tokens},
215
  )
 
209
  try:
210
  bedrock_client = boto3.client("bedrock-runtime", **boto3_config)
211
  llm = ChatBedrock(
212
+ model=bedrock_model_id,
213
  client=bedrock_client,
214
  model_kwargs={"temperature": temperature, "max_tokens": max_tokens},
215
  )