Skip to content

Commit 28c9d3b

Browse files
author
awstools
committed
feat(client-bedrock-agent-runtime): Support Multi Agent Collaboration within Inline Agents
1 parent 01aec90 commit 28c9d3b

File tree

8 files changed

+896
-238
lines changed

8 files changed

+896
-238
lines changed

‎clients/client-bedrock-agent-runtime/src/commands/InvokeAgentCommand.ts‎

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1039,6 +1039,7 @@ export interface InvokeAgentCommandOutput extends InvokeAgentResponse, __Metadat
10391039
* // agentAliasArn: "STRING_VALUE",
10401040
* // },
10411041
* // ],
1042+
* // eventTime: new Date("TIMESTAMP"),
10421043
* // collaboratorName: "STRING_VALUE",
10431044
* // },
10441045
* // returnControl: {

‎clients/client-bedrock-agent-runtime/src/commands/InvokeInlineAgentCommand.ts‎

Lines changed: 214 additions & 63 deletions
Original file line numberDiff line numberDiff line change
@@ -65,66 +65,7 @@ export interface InvokeInlineAgentCommandOutput extends InvokeInlineAgentRespons
6565
* // const { BedrockAgentRuntimeClient, InvokeInlineAgentCommand } = require("@aws-sdk/client-bedrock-agent-runtime"); // CommonJS import
6666
* const client = new BedrockAgentRuntimeClient(config);
6767
* const input = { // InvokeInlineAgentRequest
68-
* sessionId: "STRING_VALUE", // required
6968
* customerEncryptionKeyArn: "STRING_VALUE",
70-
* endSession: true || false,
71-
* enableTrace: true || false,
72-
* inputText: "STRING_VALUE",
73-
* inlineSessionState: { // InlineSessionState
74-
* sessionAttributes: { // SessionAttributesMap
75-
* "<keys>": "STRING_VALUE",
76-
* },
77-
* promptSessionAttributes: { // PromptSessionAttributesMap
78-
* "<keys>": "STRING_VALUE",
79-
* },
80-
* returnControlInvocationResults: [ // ReturnControlInvocationResults
81-
* { // InvocationResultMember Union: only one key present
82-
* apiResult: { // ApiResult
83-
* actionGroup: "STRING_VALUE", // required
84-
* httpMethod: "STRING_VALUE",
85-
* apiPath: "STRING_VALUE",
86-
* confirmationState: "CONFIRM" || "DENY",
87-
* responseState: "FAILURE" || "REPROMPT",
88-
* httpStatusCode: Number("int"),
89-
* responseBody: { // ResponseBody
90-
* "<keys>": { // ContentBody
91-
* body: "STRING_VALUE",
92-
* },
93-
* },
94-
* agentId: "STRING_VALUE",
95-
* },
96-
* functionResult: { // FunctionResult
97-
* actionGroup: "STRING_VALUE", // required
98-
* confirmationState: "CONFIRM" || "DENY",
99-
* function: "STRING_VALUE",
100-
* responseBody: {
101-
* "<keys>": {
102-
* body: "STRING_VALUE",
103-
* },
104-
* },
105-
* responseState: "FAILURE" || "REPROMPT",
106-
* agentId: "STRING_VALUE",
107-
* },
108-
* },
109-
* ],
110-
* invocationId: "STRING_VALUE",
111-
* files: [ // InputFiles
112-
* { // InputFile
113-
* name: "STRING_VALUE", // required
114-
* source: { // FileSource
115-
* sourceType: "S3" || "BYTE_CONTENT", // required
116-
* s3Location: { // S3ObjectFile
117-
* uri: "STRING_VALUE", // required
118-
* },
119-
* byteContent: { // ByteContentFile
120-
* mediaType: "STRING_VALUE", // required
121-
* data: new Uint8Array(), // e.g. Buffer.from("") or new TextEncoder().encode("") // required
122-
* },
123-
* },
124-
* useCase: "CODE_INTERPRETER" || "CHAT", // required
125-
* },
126-
* ],
127-
* },
12869
* foundationModel: "STRING_VALUE", // required
12970
* instruction: "STRING_VALUE", // required
13071
* idleSessionTTLInSeconds: Number("int"),
@@ -284,20 +225,230 @@ export interface InvokeInlineAgentCommandOutput extends InvokeInlineAgentRespons
284225
* ],
285226
* },
286227
* parserMode: "DEFAULT" || "OVERRIDDEN",
228+
* foundationModel: "STRING_VALUE",
287229
* additionalModelRequestFields: "DOCUMENT_VALUE",
288230
* },
289231
* ],
290232
* overrideLambda: "STRING_VALUE",
291233
* },
292-
* bedrockModelConfigurations: { // InlineBedrockModelConfigurations
293-
* performanceConfig: { // PerformanceConfiguration
294-
* latency: "standard" || "optimized",
234+
* agentCollaboration: "SUPERVISOR" || "SUPERVISOR_ROUTER" || "DISABLED",
235+
* collaboratorConfigurations: [ // CollaboratorConfigurations
236+
* { // CollaboratorConfiguration
237+
* collaboratorName: "STRING_VALUE", // required
238+
* collaboratorInstruction: "STRING_VALUE", // required
239+
* agentAliasArn: "STRING_VALUE",
240+
* relayConversationHistory: "TO_COLLABORATOR" || "DISABLED",
295241
* },
296-
* },
242+
* ],
243+
* sessionId: "STRING_VALUE", // required
244+
* endSession: true || false,
245+
* enableTrace: true || false,
246+
* inputText: "STRING_VALUE",
297247
* streamingConfigurations: { // StreamingConfigurations
298248
* streamFinalResponse: true || false,
299249
* applyGuardrailInterval: Number("int"),
300250
* },
251+
* inlineSessionState: { // InlineSessionState
252+
* sessionAttributes: { // SessionAttributesMap
253+
* "<keys>": "STRING_VALUE",
254+
* },
255+
* promptSessionAttributes: { // PromptSessionAttributesMap
256+
* "<keys>": "STRING_VALUE",
257+
* },
258+
* returnControlInvocationResults: [ // ReturnControlInvocationResults
259+
* { // InvocationResultMember Union: only one key present
260+
* apiResult: { // ApiResult
261+
* actionGroup: "STRING_VALUE", // required
262+
* httpMethod: "STRING_VALUE",
263+
* apiPath: "STRING_VALUE",
264+
* confirmationState: "CONFIRM" || "DENY",
265+
* responseState: "FAILURE" || "REPROMPT",
266+
* httpStatusCode: Number("int"),
267+
* responseBody: { // ResponseBody
268+
* "<keys>": { // ContentBody
269+
* body: "STRING_VALUE",
270+
* },
271+
* },
272+
* agentId: "STRING_VALUE",
273+
* },
274+
* functionResult: { // FunctionResult
275+
* actionGroup: "STRING_VALUE", // required
276+
* confirmationState: "CONFIRM" || "DENY",
277+
* function: "STRING_VALUE",
278+
* responseBody: {
279+
* "<keys>": {
280+
* body: "STRING_VALUE",
281+
* },
282+
* },
283+
* responseState: "FAILURE" || "REPROMPT",
284+
* agentId: "STRING_VALUE",
285+
* },
286+
* },
287+
* ],
288+
* invocationId: "STRING_VALUE",
289+
* files: [ // InputFiles
290+
* { // InputFile
291+
* name: "STRING_VALUE", // required
292+
* source: { // FileSource
293+
* sourceType: "S3" || "BYTE_CONTENT", // required
294+
* s3Location: { // S3ObjectFile
295+
* uri: "STRING_VALUE", // required
296+
* },
297+
* byteContent: { // ByteContentFile
298+
* mediaType: "STRING_VALUE", // required
299+
* data: new Uint8Array(), // e.g. Buffer.from("") or new TextEncoder().encode("") // required
300+
* },
301+
* },
302+
* useCase: "CODE_INTERPRETER" || "CHAT", // required
303+
* },
304+
* ],
305+
* conversationHistory: { // ConversationHistory
306+
* messages: [ // Messages
307+
* { // Message
308+
* role: "user" || "assistant", // required
309+
* content: [ // ContentBlocks // required
310+
* { // ContentBlock Union: only one key present
311+
* text: "STRING_VALUE",
312+
* },
313+
* ],
314+
* },
315+
* ],
316+
* },
317+
* },
318+
* collaborators: [ // Collaborators
319+
* { // Collaborator
320+
* customerEncryptionKeyArn: "STRING_VALUE",
321+
* foundationModel: "STRING_VALUE", // required
322+
* instruction: "STRING_VALUE", // required
323+
* idleSessionTTLInSeconds: Number("int"),
324+
* actionGroups: [
325+
* {
326+
* actionGroupName: "STRING_VALUE", // required
327+
* description: "STRING_VALUE",
328+
* parentActionGroupSignature: "AMAZON.UserInput" || "AMAZON.CodeInterpreter",
329+
* actionGroupExecutor: {// Union: only one key present
330+
* lambda: "STRING_VALUE",
331+
* customControl: "RETURN_CONTROL",
332+
* },
333+
* apiSchema: {// Union: only one key present
334+
* s3: {
335+
* s3BucketName: "STRING_VALUE",
336+
* s3ObjectKey: "STRING_VALUE",
337+
* },
338+
* payload: "STRING_VALUE",
339+
* },
340+
* functionSchema: {// Union: only one key present
341+
* functions: [
342+
* {
343+
* name: "STRING_VALUE", // required
344+
* description: "STRING_VALUE",
345+
* parameters: {
346+
* "<keys>": {
347+
* description: "STRING_VALUE",
348+
* type: "string" || "number" || "integer" || "boolean" || "array", // required
349+
* required: true || false,
350+
* },
351+
* },
352+
* requireConfirmation: "ENABLED" || "DISABLED",
353+
* },
354+
* ],
355+
* },
356+
* },
357+
* ],
358+
* knowledgeBases: [
359+
* {
360+
* knowledgeBaseId: "STRING_VALUE", // required
361+
* description: "STRING_VALUE", // required
362+
* retrievalConfiguration: {
363+
* vectorSearchConfiguration: {
364+
* numberOfResults: Number("int"),
365+
* overrideSearchType: "HYBRID" || "SEMANTIC",
366+
* filter: "<RetrievalFilter>",
367+
* rerankingConfiguration: {
368+
* type: "BEDROCK_RERANKING_MODEL", // required
369+
* bedrockRerankingConfiguration: {
370+
* modelConfiguration: {
371+
* modelArn: "STRING_VALUE", // required
372+
* additionalModelRequestFields: {
373+
* "<keys>": "DOCUMENT_VALUE",
374+
* },
375+
* },
376+
* numberOfRerankedResults: Number("int"),
377+
* metadataConfiguration: {
378+
* selectionMode: "SELECTIVE" || "ALL", // required
379+
* selectiveModeConfiguration: {// Union: only one key present
380+
* fieldsToInclude: [
381+
* {
382+
* fieldName: "STRING_VALUE", // required
383+
* },
384+
* ],
385+
* fieldsToExclude: [
386+
* {
387+
* fieldName: "STRING_VALUE", // required
388+
* },
389+
* ],
390+
* },
391+
* },
392+
* },
393+
* },
394+
* implicitFilterConfiguration: {
395+
* metadataAttributes: [ // required
396+
* {
397+
* key: "STRING_VALUE", // required
398+
* type: "STRING" || "NUMBER" || "BOOLEAN" || "STRING_LIST", // required
399+
* description: "STRING_VALUE", // required
400+
* },
401+
* ],
402+
* modelArn: "STRING_VALUE", // required
403+
* },
404+
* },
405+
* },
406+
* },
407+
* ],
408+
* guardrailConfiguration: {
409+
* guardrailIdentifier: "STRING_VALUE", // required
410+
* guardrailVersion: "STRING_VALUE", // required
411+
* },
412+
* promptOverrideConfiguration: {
413+
* promptConfigurations: [ // required
414+
* {
415+
* promptType: "PRE_PROCESSING" || "ORCHESTRATION" || "KNOWLEDGE_BASE_RESPONSE_GENERATION" || "POST_PROCESSING" || "ROUTING_CLASSIFIER",
416+
* promptCreationMode: "DEFAULT" || "OVERRIDDEN",
417+
* promptState: "ENABLED" || "DISABLED",
418+
* basePromptTemplate: "STRING_VALUE",
419+
* inferenceConfiguration: {
420+
* temperature: Number("float"),
421+
* topP: Number("float"),
422+
* topK: Number("int"),
423+
* maximumLength: Number("int"),
424+
* stopSequences: [
425+
* "STRING_VALUE",
426+
* ],
427+
* },
428+
* parserMode: "DEFAULT" || "OVERRIDDEN",
429+
* foundationModel: "STRING_VALUE",
430+
* additionalModelRequestFields: "DOCUMENT_VALUE",
431+
* },
432+
* ],
433+
* overrideLambda: "STRING_VALUE",
434+
* },
435+
* agentCollaboration: "SUPERVISOR" || "SUPERVISOR_ROUTER" || "DISABLED",
436+
* collaboratorConfigurations: [
437+
* {
438+
* collaboratorName: "STRING_VALUE", // required
439+
* collaboratorInstruction: "STRING_VALUE", // required
440+
* agentAliasArn: "STRING_VALUE",
441+
* relayConversationHistory: "TO_COLLABORATOR" || "DISABLED",
442+
* },
443+
* ],
444+
* agentName: "STRING_VALUE",
445+
* },
446+
* ],
447+
* bedrockModelConfigurations: { // InlineBedrockModelConfigurations
448+
* performanceConfig: { // PerformanceConfiguration
449+
* latency: "standard" || "optimized",
450+
* },
451+
* },
301452
* };
302453
* const command = new InvokeInlineAgentCommand(input);
303454
* const response = await client.send(command);

‎clients/client-bedrock-agent-runtime/src/commands/TagResourceCommand.ts‎

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,8 @@ import {
1010
ServiceOutputTypes,
1111
} from "../BedrockAgentRuntimeClient";
1212
import { commonParams } from "../endpoint/EndpointParameters";
13-
import { TagResourceRequest, TagResourceResponse } from "../models/models_0";
13+
import { TagResourceRequest } from "../models/models_0";
14+
import { TagResourceResponse } from "../models/models_1";
1415
import { de_TagResourceCommand, se_TagResourceCommand } from "../protocols/Aws_restJson1";
1516

1617
/**

‎clients/client-bedrock-agent-runtime/src/commands/UntagResourceCommand.ts‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,7 @@ import {
1010
ServiceOutputTypes,
1111
} from "../BedrockAgentRuntimeClient";
1212
import { commonParams } from "../endpoint/EndpointParameters";
13-
import { UntagResourceRequest, UntagResourceResponse } from "../models/models_0";
13+
import { UntagResourceRequest, UntagResourceResponse } from "../models/models_1";
1414
import { de_UntagResourceCommand, se_UntagResourceCommand } from "../protocols/Aws_restJson1";
1515

1616
/**

0 commit comments

Comments
 (0)