@@ -69,6 +69,13 @@ def __init__(self, *args, **kwargs):
69
69
if self .context .config .bedrock :
70
70
if hasattr (self .context .config .bedrock , "default_model" ):
71
71
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
+ )
72
79
73
80
self .default_request_params = self .default_request_params or RequestParams (
74
81
model = default_model ,
@@ -377,7 +384,7 @@ async def request_completion_task(
377
384
aws_access_key_id = request .config .aws_access_key_id ,
378
385
aws_secret_access_key = request .config .aws_secret_access_key ,
379
386
aws_session_token = request .config .aws_session_token ,
380
- region_name = request .config .bedrock . aws_region ,
387
+ region_name = request .config .aws_region ,
381
388
)
382
389
else :
383
390
session = Session ()
@@ -413,7 +420,7 @@ async def request_structured_completion_task(
413
420
aws_access_key_id = request .config .aws_access_key_id ,
414
421
aws_secret_access_key = request .config .aws_secret_access_key ,
415
422
aws_session_token = request .config .aws_session_token ,
416
- region_name = request .config .bedrock . aws_region ,
423
+ region_name = request .config .aws_region ,
417
424
)
418
425
else :
419
426
session = Session ()
@@ -424,7 +431,7 @@ async def request_structured_completion_task(
424
431
# Extract structured data from natural language
425
432
structured_response = client .chat .completions .create (
426
433
modelId = request .model ,
427
- messages = [{"role" : "user" , "content" : [{ "text" : request .response_str }] }],
434
+ messages = [{"role" : "user" , "content" : request .response_str }],
428
435
response_model = response_model ,
429
436
)
430
437
0 commit comments