Spaces:
Running
Running
Pedro Bento
commited on
Commit
·
1711059
1
Parent(s):
3271610
Fix to BedrockChat model invoke
Browse files- 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 |
-
|
| 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 |
)
|