feat(client-bedrock-agent-runtime): InvokeInlineAgent API release to …

…help invoke runtime agents without any dependency on preconfigured agents.
aws · Nov 22, 2024 · 372482a · 372482a
1 parent 3ef966f
commit 372482a
Show file tree

Hide file tree

Showing 9 changed files with 3,557 additions and 121 deletions.
diff --git a/clients/client-bedrock-agent-runtime/README.md b/clients/client-bedrock-agent-runtime/README.md
@@ -234,6 +234,14 @@ InvokeFlow
 
 [Command API Reference](https://docs.aws.amazon.com/AWSJavaScriptSDK/v3/latest/client/bedrock-agent-runtime/command/InvokeFlowCommand/) / [Input](https://docs.aws.amazon.com/AWSJavaScriptSDK/v3/latest/Package/-aws-sdk-client-bedrock-agent-runtime/Interface/InvokeFlowCommandInput/) / [Output](https://docs.aws.amazon.com/AWSJavaScriptSDK/v3/latest/Package/-aws-sdk-client-bedrock-agent-runtime/Interface/InvokeFlowCommandOutput/)
 
+</details>
+<details>
+<summary>
+InvokeInlineAgent
+</summary>
+
+[Command API Reference](https://docs.aws.amazon.com/AWSJavaScriptSDK/v3/latest/client/bedrock-agent-runtime/command/InvokeInlineAgentCommand/) / [Input](https://docs.aws.amazon.com/AWSJavaScriptSDK/v3/latest/Package/-aws-sdk-client-bedrock-agent-runtime/Interface/InvokeInlineAgentCommandInput/) / [Output](https://docs.aws.amazon.com/AWSJavaScriptSDK/v3/latest/Package/-aws-sdk-client-bedrock-agent-runtime/Interface/InvokeInlineAgentCommandOutput/)
+
 </details>
 <details>
 <summary>

diff --git a/clients/client-bedrock-agent-runtime/src/BedrockAgentRuntime.ts b/clients/client-bedrock-agent-runtime/src/BedrockAgentRuntime.ts
@@ -15,6 +15,11 @@ import {
 } from "./commands/GetAgentMemoryCommand";
 import { InvokeAgentCommand, InvokeAgentCommandInput, InvokeAgentCommandOutput } from "./commands/InvokeAgentCommand";
 import { InvokeFlowCommand, InvokeFlowCommandInput, InvokeFlowCommandOutput } from "./commands/InvokeFlowCommand";
+import {
+  InvokeInlineAgentCommand,
+  InvokeInlineAgentCommandInput,
+  InvokeInlineAgentCommandOutput,
+} from "./commands/InvokeInlineAgentCommand";
 import {
   OptimizePromptCommand,
   OptimizePromptCommandInput,
@@ -32,6 +37,7 @@ const commands = {
   GetAgentMemoryCommand,
   InvokeAgentCommand,
   InvokeFlowCommand,
+  InvokeInlineAgentCommand,
   OptimizePromptCommand,
   RetrieveCommand,
   RetrieveAndGenerateCommand,
@@ -91,6 +97,23 @@ export interface BedrockAgentRuntime {
     cb: (err: any, data?: InvokeFlowCommandOutput) => void
   ): void;
 
+  /**
+   * @see {@link InvokeInlineAgentCommand}
+   */
+  invokeInlineAgent(
+    args: InvokeInlineAgentCommandInput,
+    options?: __HttpHandlerOptions
+  ): Promise<InvokeInlineAgentCommandOutput>;
+  invokeInlineAgent(
+    args: InvokeInlineAgentCommandInput,
+    cb: (err: any, data?: InvokeInlineAgentCommandOutput) => void
+  ): void;
+  invokeInlineAgent(
+    args: InvokeInlineAgentCommandInput,
+    options: __HttpHandlerOptions,
+    cb: (err: any, data?: InvokeInlineAgentCommandOutput) => void
+  ): void;
+
   /**
    * @see {@link OptimizePromptCommand}
    */

diff --git a/clients/client-bedrock-agent-runtime/src/BedrockAgentRuntimeClient.ts b/clients/client-bedrock-agent-runtime/src/BedrockAgentRuntimeClient.ts
@@ -63,6 +63,7 @@ import { DeleteAgentMemoryCommandInput, DeleteAgentMemoryCommandOutput } from ".
 import { GetAgentMemoryCommandInput, GetAgentMemoryCommandOutput } from "./commands/GetAgentMemoryCommand";
 import { InvokeAgentCommandInput, InvokeAgentCommandOutput } from "./commands/InvokeAgentCommand";
 import { InvokeFlowCommandInput, InvokeFlowCommandOutput } from "./commands/InvokeFlowCommand";
+import { InvokeInlineAgentCommandInput, InvokeInlineAgentCommandOutput } from "./commands/InvokeInlineAgentCommand";
 import { OptimizePromptCommandInput, OptimizePromptCommandOutput } from "./commands/OptimizePromptCommand";
 import {
   RetrieveAndGenerateCommandInput,
@@ -88,6 +89,7 @@ export type ServiceInputTypes =
   | GetAgentMemoryCommandInput
   | InvokeAgentCommandInput
   | InvokeFlowCommandInput
+  | InvokeInlineAgentCommandInput
   | OptimizePromptCommandInput
   | RetrieveAndGenerateCommandInput
   | RetrieveCommandInput;
@@ -100,6 +102,7 @@ export type ServiceOutputTypes =
   | GetAgentMemoryCommandOutput
   | InvokeAgentCommandOutput
   | InvokeFlowCommandOutput
+  | InvokeInlineAgentCommandOutput
   | OptimizePromptCommandOutput
   | RetrieveAndGenerateCommandOutput
   | RetrieveCommandOutput;

diff --git a/clients/client-bedrock-agent-runtime/src/commands/InvokeAgentCommand.ts b/clients/client-bedrock-agent-runtime/src/commands/InvokeAgentCommand.ts
@@ -252,10 +252,7 @@ export interface InvokeAgentCommandOutput extends InvokeAgentResponse, __Metadat
  * //       },
  * //     },
  * //     trace: { // TracePart
- * //       agentId: "STRING_VALUE",
- * //       agentAliasId: "STRING_VALUE",
  * //       sessionId: "STRING_VALUE",
- * //       agentVersion: "STRING_VALUE",
  * //       trace: { // Trace Union: only one key present
  * //         guardrailTrace: { // GuardrailTrace
  * //           action: "INTERVENED" || "NONE",
@@ -373,7 +370,7 @@ export interface InvokeAgentCommandOutput extends InvokeAgentResponse, __Metadat
  * //           modelInvocationInput: { // ModelInvocationInput
  * //             traceId: "STRING_VALUE",
  * //             text: "STRING_VALUE",
- * //             type: "PRE_PROCESSING" || "ORCHESTRATION" || "KNOWLEDGE_BASE_RESPONSE_GENERATION" || "POST_PROCESSING",
+ * //             type: "PRE_PROCESSING" || "ORCHESTRATION" || "KNOWLEDGE_BASE_RESPONSE_GENERATION" || "POST_PROCESSING" || "ROUTING_CLASSIFIER",
  * //             overrideLambda: "STRING_VALUE",
  * //             promptCreationMode: "DEFAULT" || "OVERRIDDEN",
  * //             inferenceConfiguration: { // InferenceConfiguration
@@ -504,7 +501,7 @@ export interface InvokeAgentCommandOutput extends InvokeAgentResponse, __Metadat
  * //           modelInvocationInput: {
  * //             traceId: "STRING_VALUE",
  * //             text: "STRING_VALUE",
- * //             type: "PRE_PROCESSING" || "ORCHESTRATION" || "KNOWLEDGE_BASE_RESPONSE_GENERATION" || "POST_PROCESSING",
+ * //             type: "PRE_PROCESSING" || "ORCHESTRATION" || "KNOWLEDGE_BASE_RESPONSE_GENERATION" || "POST_PROCESSING" || "ROUTING_CLASSIFIER",
  * //             overrideLambda: "STRING_VALUE",
  * //             promptCreationMode: "DEFAULT" || "OVERRIDDEN",
  * //             inferenceConfiguration: {
@@ -535,7 +532,7 @@ export interface InvokeAgentCommandOutput extends InvokeAgentResponse, __Metadat
  * //           modelInvocationInput: {
  * //             traceId: "STRING_VALUE",
  * //             text: "STRING_VALUE",
- * //             type: "PRE_PROCESSING" || "ORCHESTRATION" || "KNOWLEDGE_BASE_RESPONSE_GENERATION" || "POST_PROCESSING",
+ * //             type: "PRE_PROCESSING" || "ORCHESTRATION" || "KNOWLEDGE_BASE_RESPONSE_GENERATION" || "POST_PROCESSING" || "ROUTING_CLASSIFIER",
  * //             overrideLambda: "STRING_VALUE",
  * //             promptCreationMode: "DEFAULT" || "OVERRIDDEN",
  * //             inferenceConfiguration: {
@@ -570,6 +567,9 @@ export interface InvokeAgentCommandOutput extends InvokeAgentResponse, __Metadat
  * //           failureReason: "STRING_VALUE",
  * //         },
  * //       },
+ * //       agentId: "STRING_VALUE",
+ * //       agentAliasId: "STRING_VALUE",
+ * //       agentVersion: "STRING_VALUE",
  * //     },
  * //     returnControl: { // ReturnControlPayload
  * //       invocationInputs: [ // InvocationInputs