feat(client-bedrock-agent): Releasing SDK for Multi-Agent Collaboration.

aws · Dec 3, 2024 · 29558b0 · 29558b0
1 parent 9169d69
commit 29558b0
Show file tree

Hide file tree

Showing 21 changed files with 6,382 additions and 3,943 deletions.
diff --git a/clients/client-bedrock-agent/README.md b/clients/client-bedrock-agent/README.md
@@ -203,6 +203,14 @@ see LICENSE for more information.
 
 ## Client Commands (Operations List)
 
+<details>
+<summary>
+AssociateAgentCollaborator
+</summary>
+
+[Command API Reference](https://docs.aws.amazon.com/AWSJavaScriptSDK/v3/latest/client/bedrock-agent/command/AssociateAgentCollaboratorCommand/) / [Input](https://docs.aws.amazon.com/AWSJavaScriptSDK/v3/latest/Package/-aws-sdk-client-bedrock-agent/Interface/AssociateAgentCollaboratorCommandInput/) / [Output](https://docs.aws.amazon.com/AWSJavaScriptSDK/v3/latest/Package/-aws-sdk-client-bedrock-agent/Interface/AssociateAgentCollaboratorCommandOutput/)
+
+</details>
 <details>
 <summary>
 AssociateAgentKnowledgeBase
@@ -378,6 +386,14 @@ DeletePrompt
 
 [Command API Reference](https://docs.aws.amazon.com/AWSJavaScriptSDK/v3/latest/client/bedrock-agent/command/DeletePromptCommand/) / [Input](https://docs.aws.amazon.com/AWSJavaScriptSDK/v3/latest/Package/-aws-sdk-client-bedrock-agent/Interface/DeletePromptCommandInput/) / [Output](https://docs.aws.amazon.com/AWSJavaScriptSDK/v3/latest/Package/-aws-sdk-client-bedrock-agent/Interface/DeletePromptCommandOutput/)
 
+</details>
+<details>
+<summary>
+DisassociateAgentCollaborator
+</summary>
+
+[Command API Reference](https://docs.aws.amazon.com/AWSJavaScriptSDK/v3/latest/client/bedrock-agent/command/DisassociateAgentCollaboratorCommand/) / [Input](https://docs.aws.amazon.com/AWSJavaScriptSDK/v3/latest/Package/-aws-sdk-client-bedrock-agent/Interface/DisassociateAgentCollaboratorCommandInput/) / [Output](https://docs.aws.amazon.com/AWSJavaScriptSDK/v3/latest/Package/-aws-sdk-client-bedrock-agent/Interface/DisassociateAgentCollaboratorCommandOutput/)
+
 </details>
 <details>
 <summary>
@@ -410,6 +426,14 @@ GetAgentAlias
 
 [Command API Reference](https://docs.aws.amazon.com/AWSJavaScriptSDK/v3/latest/client/bedrock-agent/command/GetAgentAliasCommand/) / [Input](https://docs.aws.amazon.com/AWSJavaScriptSDK/v3/latest/Package/-aws-sdk-client-bedrock-agent/Interface/GetAgentAliasCommandInput/) / [Output](https://docs.aws.amazon.com/AWSJavaScriptSDK/v3/latest/Package/-aws-sdk-client-bedrock-agent/Interface/GetAgentAliasCommandOutput/)
 
+</details>
+<details>
+<summary>
+GetAgentCollaborator
+</summary>
+
+[Command API Reference](https://docs.aws.amazon.com/AWSJavaScriptSDK/v3/latest/client/bedrock-agent/command/GetAgentCollaboratorCommand/) / [Input](https://docs.aws.amazon.com/AWSJavaScriptSDK/v3/latest/Package/-aws-sdk-client-bedrock-agent/Interface/GetAgentCollaboratorCommandInput/) / [Output](https://docs.aws.amazon.com/AWSJavaScriptSDK/v3/latest/Package/-aws-sdk-client-bedrock-agent/Interface/GetAgentCollaboratorCommandOutput/)
+
 </details>
 <details>
 <summary>
@@ -514,6 +538,14 @@ ListAgentAliases
 
 [Command API Reference](https://docs.aws.amazon.com/AWSJavaScriptSDK/v3/latest/client/bedrock-agent/command/ListAgentAliasesCommand/) / [Input](https://docs.aws.amazon.com/AWSJavaScriptSDK/v3/latest/Package/-aws-sdk-client-bedrock-agent/Interface/ListAgentAliasesCommandInput/) / [Output](https://docs.aws.amazon.com/AWSJavaScriptSDK/v3/latest/Package/-aws-sdk-client-bedrock-agent/Interface/ListAgentAliasesCommandOutput/)
 
+</details>
+<details>
+<summary>
+ListAgentCollaborators
+</summary>
+
+[Command API Reference](https://docs.aws.amazon.com/AWSJavaScriptSDK/v3/latest/client/bedrock-agent/command/ListAgentCollaboratorsCommand/) / [Input](https://docs.aws.amazon.com/AWSJavaScriptSDK/v3/latest/Package/-aws-sdk-client-bedrock-agent/Interface/ListAgentCollaboratorsCommandInput/) / [Output](https://docs.aws.amazon.com/AWSJavaScriptSDK/v3/latest/Package/-aws-sdk-client-bedrock-agent/Interface/ListAgentCollaboratorsCommandOutput/)
+
 </details>
 <details>
 <summary>
@@ -682,6 +714,14 @@ UpdateAgentAlias
 
 [Command API Reference](https://docs.aws.amazon.com/AWSJavaScriptSDK/v3/latest/client/bedrock-agent/command/UpdateAgentAliasCommand/) / [Input](https://docs.aws.amazon.com/AWSJavaScriptSDK/v3/latest/Package/-aws-sdk-client-bedrock-agent/Interface/UpdateAgentAliasCommandInput/) / [Output](https://docs.aws.amazon.com/AWSJavaScriptSDK/v3/latest/Package/-aws-sdk-client-bedrock-agent/Interface/UpdateAgentAliasCommandOutput/)
 
+</details>
+<details>
+<summary>
+UpdateAgentCollaborator
+</summary>
+
+[Command API Reference](https://docs.aws.amazon.com/AWSJavaScriptSDK/v3/latest/client/bedrock-agent/command/UpdateAgentCollaboratorCommand/) / [Input](https://docs.aws.amazon.com/AWSJavaScriptSDK/v3/latest/Package/-aws-sdk-client-bedrock-agent/Interface/UpdateAgentCollaboratorCommandInput/) / [Output](https://docs.aws.amazon.com/AWSJavaScriptSDK/v3/latest/Package/-aws-sdk-client-bedrock-agent/Interface/UpdateAgentCollaboratorCommandOutput/)
+
 </details>
 <details>
 <summary>

diff --git a/clients/client-bedrock-agent/src/BedrockAgent.ts b/clients/client-bedrock-agent/src/BedrockAgent.ts
@@ -3,6 +3,11 @@ import { createAggregatedClient } from "@smithy/smithy-client";
 import { HttpHandlerOptions as __HttpHandlerOptions } from "@smithy/types";
 
 import { BedrockAgentClient, BedrockAgentClientConfig } from "./BedrockAgentClient";
+import {
+  AssociateAgentCollaboratorCommand,
+  AssociateAgentCollaboratorCommandInput,
+  AssociateAgentCollaboratorCommandOutput,
+} from "./commands/AssociateAgentCollaboratorCommand";
 import {
   AssociateAgentKnowledgeBaseCommand,
   AssociateAgentKnowledgeBaseCommandInput,
@@ -97,6 +102,11 @@ import {
   DeletePromptCommandInput,
   DeletePromptCommandOutput,
 } from "./commands/DeletePromptCommand";
+import {
+  DisassociateAgentCollaboratorCommand,
+  DisassociateAgentCollaboratorCommandInput,
+  DisassociateAgentCollaboratorCommandOutput,
+} from "./commands/DisassociateAgentCollaboratorCommand";
 import {
   DisassociateAgentKnowledgeBaseCommand,
   DisassociateAgentKnowledgeBaseCommandInput,
@@ -112,6 +122,11 @@ import {
   GetAgentAliasCommandInput,
   GetAgentAliasCommandOutput,
 } from "./commands/GetAgentAliasCommand";
+import {
+  GetAgentCollaboratorCommand,
+  GetAgentCollaboratorCommandInput,
+  GetAgentCollaboratorCommandOutput,
+} from "./commands/GetAgentCollaboratorCommand";
 import { GetAgentCommand, GetAgentCommandInput, GetAgentCommandOutput } from "./commands/GetAgentCommand";
 import {
   GetAgentKnowledgeBaseCommand,
@@ -170,6 +185,11 @@ import {
   ListAgentAliasesCommandInput,
   ListAgentAliasesCommandOutput,
 } from "./commands/ListAgentAliasesCommand";
+import {
+  ListAgentCollaboratorsCommand,
+  ListAgentCollaboratorsCommandInput,
+  ListAgentCollaboratorsCommandOutput,
+} from "./commands/ListAgentCollaboratorsCommand";
 import {
   ListAgentKnowledgeBasesCommand,
   ListAgentKnowledgeBasesCommandInput,
@@ -250,6 +270,11 @@ import {
   UpdateAgentAliasCommandInput,
   UpdateAgentAliasCommandOutput,
 } from "./commands/UpdateAgentAliasCommand";
+import {
+  UpdateAgentCollaboratorCommand,
+  UpdateAgentCollaboratorCommandInput,
+  UpdateAgentCollaboratorCommandOutput,
+} from "./commands/UpdateAgentCollaboratorCommand";
 import { UpdateAgentCommand, UpdateAgentCommandInput, UpdateAgentCommandOutput } from "./commands/UpdateAgentCommand";
 import {
   UpdateAgentKnowledgeBaseCommand,
@@ -284,6 +309,7 @@ import {
 } from "./commands/ValidateFlowDefinitionCommand";
 
 const commands = {
+  AssociateAgentCollaboratorCommand,
   AssociateAgentKnowledgeBaseCommand,
   CreateAgentCommand,
   CreateAgentActionGroupCommand,
@@ -306,10 +332,12 @@ const commands = {
   DeleteKnowledgeBaseCommand,
   DeleteKnowledgeBaseDocumentsCommand,
   DeletePromptCommand,
+  DisassociateAgentCollaboratorCommand,
   DisassociateAgentKnowledgeBaseCommand,
   GetAgentCommand,
   GetAgentActionGroupCommand,
   GetAgentAliasCommand,
+  GetAgentCollaboratorCommand,
   GetAgentKnowledgeBaseCommand,
   GetAgentVersionCommand,
   GetDataSourceCommand,
@@ -323,6 +351,7 @@ const commands = {
   IngestKnowledgeBaseDocumentsCommand,
   ListAgentActionGroupsCommand,
   ListAgentAliasesCommand,
+  ListAgentCollaboratorsCommand,
   ListAgentKnowledgeBasesCommand,
   ListAgentsCommand,
   ListAgentVersionsCommand,
@@ -344,6 +373,7 @@ const commands = {
   UpdateAgentCommand,
   UpdateAgentActionGroupCommand,
   UpdateAgentAliasCommand,
+  UpdateAgentCollaboratorCommand,
   UpdateAgentKnowledgeBaseCommand,
   UpdateDataSourceCommand,
   UpdateFlowCommand,
@@ -354,6 +384,23 @@ const commands = {
 };
 
 export interface BedrockAgent {
+  /**
+   * @see {@link AssociateAgentCollaboratorCommand}
+   */
+  associateAgentCollaborator(
+    args: AssociateAgentCollaboratorCommandInput,
+    options?: __HttpHandlerOptions
+  ): Promise<AssociateAgentCollaboratorCommandOutput>;
+  associateAgentCollaborator(
+    args: AssociateAgentCollaboratorCommandInput,
+    cb: (err: any, data?: AssociateAgentCollaboratorCommandOutput) => void
+  ): void;
+  associateAgentCollaborator(
+    args: AssociateAgentCollaboratorCommandInput,
+    options: __HttpHandlerOptions,
+    cb: (err: any, data?: AssociateAgentCollaboratorCommandOutput) => void
+  ): void;
+
   /**
    * @see {@link AssociateAgentKnowledgeBaseCommand}
    */
@@ -686,6 +733,23 @@ export interface BedrockAgent {
     cb: (err: any, data?: DeletePromptCommandOutput) => void
   ): void;
 
+  /**
+   * @see {@link DisassociateAgentCollaboratorCommand}
+   */
+  disassociateAgentCollaborator(
+    args: DisassociateAgentCollaboratorCommandInput,
+    options?: __HttpHandlerOptions
+  ): Promise<DisassociateAgentCollaboratorCommandOutput>;
+  disassociateAgentCollaborator(
+    args: DisassociateAgentCollaboratorCommandInput,
+    cb: (err: any, data?: DisassociateAgentCollaboratorCommandOutput) => void
+  ): void;
+  disassociateAgentCollaborator(
+    args: DisassociateAgentCollaboratorCommandInput,
+    options: __HttpHandlerOptions,
+    cb: (err: any, data?: DisassociateAgentCollaboratorCommandOutput) => void
+  ): void;
+
   /**
    * @see {@link DisassociateAgentKnowledgeBaseCommand}
    */
@@ -742,6 +806,23 @@ export interface BedrockAgent {
     cb: (err: any, data?: GetAgentAliasCommandOutput) => void
   ): void;
 
+  /**
+   * @see {@link GetAgentCollaboratorCommand}
+   */
+  getAgentCollaborator(
+    args: GetAgentCollaboratorCommandInput,
+    options?: __HttpHandlerOptions
+  ): Promise<GetAgentCollaboratorCommandOutput>;
+  getAgentCollaborator(
+    args: GetAgentCollaboratorCommandInput,
+    cb: (err: any, data?: GetAgentCollaboratorCommandOutput) => void
+  ): void;
+  getAgentCollaborator(
+    args: GetAgentCollaboratorCommandInput,
+    options: __HttpHandlerOptions,
+    cb: (err: any, data?: GetAgentCollaboratorCommandOutput) => void
+  ): void;
+
   /**
    * @see {@link GetAgentKnowledgeBaseCommand}
    */
@@ -930,6 +1011,23 @@ export interface BedrockAgent {
     cb: (err: any, data?: ListAgentAliasesCommandOutput) => void
   ): void;
 
+  /**
+   * @see {@link ListAgentCollaboratorsCommand}
+   */
+  listAgentCollaborators(
+    args: ListAgentCollaboratorsCommandInput,
+    options?: __HttpHandlerOptions
+  ): Promise<ListAgentCollaboratorsCommandOutput>;
+  listAgentCollaborators(
+    args: ListAgentCollaboratorsCommandInput,
+    cb: (err: any, data?: ListAgentCollaboratorsCommandOutput) => void
+  ): void;
+  listAgentCollaborators(
+    args: ListAgentCollaboratorsCommandInput,
+    options: __HttpHandlerOptions,
+    cb: (err: any, data?: ListAgentCollaboratorsCommandOutput) => void
+  ): void;
+
   /**
    * @see {@link ListAgentKnowledgeBasesCommand}
    */
@@ -1237,6 +1335,23 @@ export interface BedrockAgent {
     cb: (err: any, data?: UpdateAgentAliasCommandOutput) => void
   ): void;
 
+  /**
+   * @see {@link UpdateAgentCollaboratorCommand}
+   */
+  updateAgentCollaborator(
+    args: UpdateAgentCollaboratorCommandInput,
+    options?: __HttpHandlerOptions
+  ): Promise<UpdateAgentCollaboratorCommandOutput>;
+  updateAgentCollaborator(
+    args: UpdateAgentCollaboratorCommandInput,
+    cb: (err: any, data?: UpdateAgentCollaboratorCommandOutput) => void
+  ): void;
+  updateAgentCollaborator(
+    args: UpdateAgentCollaboratorCommandInput,
+    options: __HttpHandlerOptions,
+    cb: (err: any, data?: UpdateAgentCollaboratorCommandOutput) => void
+  ): void;
+
   /**
    * @see {@link UpdateAgentKnowledgeBaseCommand}
    */