Skip to content

Commit b2b9515

Browse files
author
AWS
committed
Agents for Amazon Bedrock Runtime Update: Support for Custom Orchestration within InlineAgents
1 parent f65eea9 commit b2b9515

File tree

2 files changed

+74
-16
lines changed

2 files changed

+74
-16
lines changed

‎.changes/next-release/feature-AgentsforAmazonBedrockRuntime-f95518e.json

+6
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,6 @@
1+
{
2+
"type": "feature",
3+
"category": "Agents for Amazon Bedrock Runtime",
4+
"contributor": "",
5+
"description": "Support for Custom Orchestration within InlineAgents"
6+
}

‎services/bedrockagentruntime/src/main/resources/codegen-resources/service-2.json

+68-16
Original file line numberDiff line numberDiff line change
@@ -65,8 +65,8 @@
6565
"input":{"shape":"DeleteAgentMemoryRequest"},
6666
"output":{"shape":"DeleteAgentMemoryResponse"},
6767
"errors":[
68-
{"shape":"ConflictException"},
6968
{"shape":"ResourceNotFoundException"},
69+
{"shape":"ConflictException"},
7070
{"shape":"ValidationException"},
7171
{"shape":"InternalServerException"},
7272
{"shape":"DependencyFailedException"},
@@ -88,8 +88,8 @@
8888
"input":{"shape":"DeleteSessionRequest"},
8989
"output":{"shape":"DeleteSessionResponse"},
9090
"errors":[
91-
{"shape":"ConflictException"},
9291
{"shape":"ResourceNotFoundException"},
92+
{"shape":"ConflictException"},
9393
{"shape":"ValidationException"},
9494
{"shape":"InternalServerException"},
9595
{"shape":"ThrottlingException"},
@@ -108,8 +108,8 @@
108108
"input":{"shape":"EndSessionRequest"},
109109
"output":{"shape":"EndSessionResponse"},
110110
"errors":[
111-
{"shape":"ConflictException"},
112111
{"shape":"ResourceNotFoundException"},
112+
{"shape":"ConflictException"},
113113
{"shape":"ValidationException"},
114114
{"shape":"InternalServerException"},
115115
{"shape":"ThrottlingException"},
@@ -128,8 +128,8 @@
128128
"input":{"shape":"GenerateQueryRequest"},
129129
"output":{"shape":"GenerateQueryResponse"},
130130
"errors":[
131-
{"shape":"ConflictException"},
132131
{"shape":"ResourceNotFoundException"},
132+
{"shape":"ConflictException"},
133133
{"shape":"ValidationException"},
134134
{"shape":"InternalServerException"},
135135
{"shape":"DependencyFailedException"},
@@ -150,8 +150,8 @@
150150
"input":{"shape":"GetAgentMemoryRequest"},
151151
"output":{"shape":"GetAgentMemoryResponse"},
152152
"errors":[
153-
{"shape":"ConflictException"},
154153
{"shape":"ResourceNotFoundException"},
154+
{"shape":"ConflictException"},
155155
{"shape":"ValidationException"},
156156
{"shape":"InternalServerException"},
157157
{"shape":"DependencyFailedException"},
@@ -209,8 +209,8 @@
209209
"output":{"shape":"InvokeAgentResponse"},
210210
"errors":[
211211
{"shape":"ModelNotReadyException"},
212-
{"shape":"ConflictException"},
213212
{"shape":"ResourceNotFoundException"},
213+
{"shape":"ConflictException"},
214214
{"shape":"ValidationException"},
215215
{"shape":"InternalServerException"},
216216
{"shape":"DependencyFailedException"},
@@ -219,7 +219,7 @@
219219
{"shape":"AccessDeniedException"},
220220
{"shape":"ServiceQuotaExceededException"}
221221
],
222-
"documentation":"<note> </note> <p>Sends a prompt for the agent to process and respond to. Note the following fields for the request:</p> <ul> <li> <p>To continue the same conversation with an agent, use the same <code>sessionId</code> value in the request.</p> </li> <li> <p>To activate trace enablement, turn <code>enableTrace</code> to <code>true</code>. Trace enablement helps you follow the agent's reasoning process that led it to the information it processed, the actions it took, and the final result it yielded. For more information, see <a href=\"https://docs.aws.amazon.com/bedrock/latest/userguide/agents-test.html#trace-events\">Trace enablement</a>.</p> </li> <li> <p>End a conversation by setting <code>endSession</code> to <code>true</code>.</p> </li> <li> <p>In the <code>sessionState</code> object, you can include attributes for the session or prompt or, if you configured an action group to return control, results from invocation of the action group.</p> </li> </ul> <p>The response contains both <b>chunk</b> and <b>trace</b> attributes.</p> <p>The final response is returned in the <code>bytes</code> field of the <code>chunk</code> object. The <code>InvokeAgent</code> returns one chunk for the entire interaction.</p> <ul> <li> <p>The <code>attribution</code> object contains citations for parts of the response.</p> </li> <li> <p>If you set <code>enableTrace</code> to <code>true</code> in the request, you can trace the agent's steps and reasoning process that led it to the response.</p> </li> <li> <p>If the action predicted was configured to return control, the response returns parameters for the action, elicited from the user, in the <code>returnControl</code> field.</p> </li> <li> <p>Errors are also surfaced in the response.</p> </li> </ul>"
222+
"documentation":"<p><note> </note> <p>Sends a prompt for the agent to process and respond to. Note the following fields for the request:</p> <ul> <li> <p>To continue the same conversation with an agent, use the same <code>sessionId</code> value in the request.</p> </li> <li> <p>To activate trace enablement, turn <code>enableTrace</code> to <code>true</code>. Trace enablement helps you follow the agent's reasoning process that led it to the information it processed, the actions it took, and the final result it yielded. For more information, see <a href=\"https://docs.aws.amazon.com/bedrock/latest/userguide/agents-test.html#trace-events\">Trace enablement</a>.</p> </li> <li> <p>End a conversation by setting <code>endSession</code> to <code>true</code>.</p> </li> <li> <p>In the <code>sessionState</code> object, you can include attributes for the session or prompt or, if you configured an action group to return control, results from invocation of the action group.</p> </li> </ul> <p>The response contains both <b>chunk</b> and <b>trace</b> attributes.</p> <p>The final response is returned in the <code>bytes</code> field of the <code>chunk</code> object. The <code>InvokeAgent</code> returns one chunk for the entire interaction.</p> <ul> <li> <p>The <code>attribution</code> object contains citations for parts of the response.</p> </li> <li> <p>If you set <code>enableTrace</code> to <code>true</code> in the request, you can trace the agent's steps and reasoning process that led it to the response.</p> </li> <li> <p>If the action predicted was configured to return control, the response returns parameters for the action, elicited from the user, in the <code>returnControl</code> field.</p> </li> <li> <p>Errors are also surfaced in the response.</p> </li> </ul></p>"
223223
},
224224
"InvokeFlow":{
225225
"name":"InvokeFlow",
@@ -231,8 +231,8 @@
231231
"input":{"shape":"InvokeFlowRequest"},
232232
"output":{"shape":"InvokeFlowResponse"},
233233
"errors":[
234-
{"shape":"ConflictException"},
235234
{"shape":"ResourceNotFoundException"},
235+
{"shape":"ConflictException"},
236236
{"shape":"ValidationException"},
237237
{"shape":"InternalServerException"},
238238
{"shape":"DependencyFailedException"},
@@ -253,8 +253,8 @@
253253
"input":{"shape":"InvokeInlineAgentRequest"},
254254
"output":{"shape":"InvokeInlineAgentResponse"},
255255
"errors":[
256-
{"shape":"ConflictException"},
257256
{"shape":"ResourceNotFoundException"},
257+
{"shape":"ConflictException"},
258258
{"shape":"ValidationException"},
259259
{"shape":"InternalServerException"},
260260
{"shape":"DependencyFailedException"},
@@ -365,8 +365,8 @@
365365
"input":{"shape":"PutInvocationStepRequest"},
366366
"output":{"shape":"PutInvocationStepResponse"},
367367
"errors":[
368-
{"shape":"ConflictException"},
369368
{"shape":"ResourceNotFoundException"},
369+
{"shape":"ConflictException"},
370370
{"shape":"ValidationException"},
371371
{"shape":"InternalServerException"},
372372
{"shape":"ThrottlingException"},
@@ -386,8 +386,8 @@
386386
"input":{"shape":"RerankRequest"},
387387
"output":{"shape":"RerankResponse"},
388388
"errors":[
389-
{"shape":"ConflictException"},
390389
{"shape":"ResourceNotFoundException"},
390+
{"shape":"ConflictException"},
391391
{"shape":"ValidationException"},
392392
{"shape":"InternalServerException"},
393393
{"shape":"DependencyFailedException"},
@@ -408,8 +408,8 @@
408408
"input":{"shape":"RetrieveRequest"},
409409
"output":{"shape":"RetrieveResponse"},
410410
"errors":[
411-
{"shape":"ConflictException"},
412411
{"shape":"ResourceNotFoundException"},
412+
{"shape":"ConflictException"},
413413
{"shape":"ValidationException"},
414414
{"shape":"InternalServerException"},
415415
{"shape":"DependencyFailedException"},
@@ -430,8 +430,8 @@
430430
"input":{"shape":"RetrieveAndGenerateRequest"},
431431
"output":{"shape":"RetrieveAndGenerateResponse"},
432432
"errors":[
433-
{"shape":"ConflictException"},
434433
{"shape":"ResourceNotFoundException"},
434+
{"shape":"ConflictException"},
435435
{"shape":"ValidationException"},
436436
{"shape":"InternalServerException"},
437437
{"shape":"DependencyFailedException"},
@@ -452,8 +452,8 @@
452452
"input":{"shape":"RetrieveAndGenerateStreamRequest"},
453453
"output":{"shape":"RetrieveAndGenerateStreamResponse"},
454454
"errors":[
455-
{"shape":"ConflictException"},
456455
{"shape":"ResourceNotFoundException"},
456+
{"shape":"ConflictException"},
457457
{"shape":"ValidationException"},
458458
{"shape":"InternalServerException"},
459459
{"shape":"DependencyFailedException"},
@@ -1485,6 +1485,16 @@
14851485
"type":"string",
14861486
"enum":["RETURN_CONTROL"]
14871487
},
1488+
"CustomOrchestration":{
1489+
"type":"structure",
1490+
"members":{
1491+
"executor":{
1492+
"shape":"OrchestrationExecutor",
1493+
"documentation":"<p>The structure of the executor invoking the actions in custom orchestration. </p>"
1494+
}
1495+
},
1496+
"documentation":"<p>Contains details of the custom orchestration configured for the agent. </p>"
1497+
},
14881498
"CustomOrchestrationTrace":{
14891499
"type":"structure",
14901500
"members":{
@@ -3343,6 +3353,18 @@
33433353
"InlineAgentTracePart":{
33443354
"type":"structure",
33453355
"members":{
3356+
"callerChain":{
3357+
"shape":"CallerChain",
3358+
"documentation":"<p>The caller chain for the trace part.</p>"
3359+
},
3360+
"collaboratorName":{
3361+
"shape":"Name",
3362+
"documentation":"<p>The collaborator name for the trace part.</p>"
3363+
},
3364+
"eventTime":{
3365+
"shape":"SyntheticTimestamp_date_time",
3366+
"documentation":"<p>The time that trace occurred. </p>"
3367+
},
33463368
"sessionId":{
33473369
"shape":"SessionId",
33483370
"documentation":"<p>The unique identifier of the session with the agent.</p>"
@@ -3822,6 +3844,10 @@
38223844
"shape":"AgentCollaboration",
38233845
"documentation":"<p> Defines how the inline collaborator agent handles information across multiple collaborator agents to coordinate a final response. The inline collaborator agent can also be the supervisor. </p>"
38243846
},
3847+
"agentName":{
3848+
"shape":"Name",
3849+
"documentation":"<p>The name for the agent.</p>"
3850+
},
38253851
"bedrockModelConfigurations":{
38263852
"shape":"InlineBedrockModelConfigurations",
38273853
"documentation":"<p>Model settings for the request.</p>"
@@ -3834,13 +3860,17 @@
38343860
"shape":"Collaborators",
38353861
"documentation":"<p> List of collaborator inline agents. </p>"
38363862
},
3863+
"customOrchestration":{
3864+
"shape":"CustomOrchestration",
3865+
"documentation":"<p>Contains details of the custom orchestration configured for the agent. </p>"
3866+
},
38373867
"customerEncryptionKeyArn":{
38383868
"shape":"KmsKeyArn",
38393869
"documentation":"<p> The Amazon Resource Name (ARN) of the Amazon Web Services KMS key to use to encrypt your inline agent. </p>"
38403870
},
38413871
"enableTrace":{
38423872
"shape":"Boolean",
3843-
"documentation":"<p> Specifies whether to turn on the trace or not to track the agent's reasoning process. For more information, see <a href=\"https://docs.aws.amazon.com/bedrock/latest/userguide/trace-events.html\">Using trace</a>. <pre><code> &lt;/p&gt; </code></pre>"
3873+
"documentation":"<p> Specifies whether to turn on the trace or not to track the agent's reasoning process. For more information, see <a href=\"https://docs.aws.amazon.com/bedrock/latest/userguide/trace-events.html\">Using trace</a>. </p>"
38443874
},
38453875
"endSession":{
38463876
"shape":"Boolean",
@@ -3874,6 +3904,10 @@
38743904
"shape":"KnowledgeBases",
38753905
"documentation":"<p> Contains information of the knowledge bases to associate with. </p>"
38763906
},
3907+
"orchestrationType":{
3908+
"shape":"OrchestrationType",
3909+
"documentation":"<p>Specifies the type of orchestration strategy for the agent. This is set to DEFAULT orchestration type, by default. </p>"
3910+
},
38773911
"promptOverrideConfiguration":{
38783912
"shape":"PromptOverrideConfiguration",
38793913
"documentation":"<p> Configurations for advanced prompts used to override the default prompts to enhance the accuracy of the inline agent. </p>"
@@ -3900,7 +3934,7 @@
39003934
"members":{
39013935
"completion":{
39023936
"shape":"InlineAgentResponseStream",
3903-
"documentation":"<p> <pre><code> &lt;/p&gt; </code></pre>"
3937+
"documentation":"<p>The inline agent's response to the user prompt. </p>"
39043938
},
39053939
"contentType":{
39063940
"shape":"MimeType",
@@ -4698,6 +4732,17 @@
46984732
},
46994733
"documentation":"<p>Settings for how the model processes the prompt prior to retrieval and generation.</p>"
47004734
},
4735+
"OrchestrationExecutor":{
4736+
"type":"structure",
4737+
"members":{
4738+
"lambda":{
4739+
"shape":"LambdaArn",
4740+
"documentation":"<p>The Amazon Resource Name (ARN) of the Lambda function containing the business logic that is carried out upon invoking the action. </p>"
4741+
}
4742+
},
4743+
"documentation":"<p>The structure of the executor invoking the actions in custom orchestration.</p>",
4744+
"union":true
4745+
},
47014746
"OrchestrationModelInvocationOutput":{
47024747
"type":"structure",
47034748
"members":{
@@ -4749,6 +4794,13 @@
47494794
"sensitive":true,
47504795
"union":true
47514796
},
4797+
"OrchestrationType":{
4798+
"type":"string",
4799+
"enum":[
4800+
"DEFAULT",
4801+
"CUSTOM_ORCHESTRATION"
4802+
]
4803+
},
47524804
"OutputFile":{
47534805
"type":"structure",
47544806
"members":{

0 commit comments

Comments
 (0)