Skip to content

Commit 28f64cc

Browse files
authored
Fix BedrockAugmentedLLM config and generate_structured() (#253)
1 parent 99bce86 commit 28f64cc

File tree

1 file changed

+10
-3
lines changed

1 file changed

+10
-3
lines changed

src/mcp_agent/workflows/llm/augmented_llm_bedrock.py

Lines changed: 10 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -69,6 +69,13 @@ def __init__(self, *args, **kwargs):
6969
if self.context.config.bedrock:
7070
if hasattr(self.context.config.bedrock, "default_model"):
7171
default_model = self.context.config.bedrock.default_model
72+
else:
73+
self.logger.error(
74+
"Bedrock configuration not found. Please provide Bedrock configuration."
75+
)
76+
raise ValueError(
77+
"Bedrock configuration not found. Please provide Bedrock configuration."
78+
)
7279

7380
self.default_request_params = self.default_request_params or RequestParams(
7481
model=default_model,
@@ -377,7 +384,7 @@ async def request_completion_task(
377384
aws_access_key_id=request.config.aws_access_key_id,
378385
aws_secret_access_key=request.config.aws_secret_access_key,
379386
aws_session_token=request.config.aws_session_token,
380-
region_name=request.config.bedrock.aws_region,
387+
region_name=request.config.aws_region,
381388
)
382389
else:
383390
session = Session()
@@ -413,7 +420,7 @@ async def request_structured_completion_task(
413420
aws_access_key_id=request.config.aws_access_key_id,
414421
aws_secret_access_key=request.config.aws_secret_access_key,
415422
aws_session_token=request.config.aws_session_token,
416-
region_name=request.config.bedrock.aws_region,
423+
region_name=request.config.aws_region,
417424
)
418425
else:
419426
session = Session()
@@ -424,7 +431,7 @@ async def request_structured_completion_task(
424431
# Extract structured data from natural language
425432
structured_response = client.chat.completions.create(
426433
modelId=request.model,
427-
messages=[{"role": "user", "content": [{"text": request.response_str}]}],
434+
messages=[{"role": "user", "content": request.response_str}],
428435
response_model=response_model,
429436
)
430437

0 commit comments

Comments
 (0)