@aws-sdk/client-bedrock-agent-runtime 3.696.0 → 3.699.0

package/dist-es/protocols/Aws_restJson1.js

@@ -68,6 +68,45 @@ export const se_InvokeFlowCommand = async (input, context) => {
     b.m("POST").h(headers).b(body);
     return b.build();
 };
+export const se_InvokeInlineAgentCommand = async (input, context) => {
+    const b = rb(input, context);
+    const headers = {
+        "content-type": "application/json",
+    };
+    b.bp("/agents/{sessionId}");
+    b.p("sessionId", () => input.sessionId, "{sessionId}", false);
+    let body;
+    body = JSON.stringify(take(input, {
+        actionGroups: (_) => _json(_),
+        customerEncryptionKeyArn: [],
+        enableTrace: [],
+        endSession: [],
+        foundationModel: [],
+        guardrailConfiguration: (_) => _json(_),
+        idleSessionTTLInSeconds: [],
+        inlineSessionState: (_) => se_InlineSessionState(_, context),
+        inputText: [],
+        instruction: [],
+        knowledgeBases: (_) => se_KnowledgeBases(_, context),
+        promptOverrideConfiguration: (_) => se_PromptOverrideConfiguration(_, context),
+    }));
+    b.m("POST").h(headers).b(body);
+    return b.build();
+};
+export const se_OptimizePromptCommand = async (input, context) => {
+    const b = rb(input, context);
+    const headers = {
+        "content-type": "application/json",
+    };
+    b.bp("/optimize-prompt");
+    let body;
+    body = JSON.stringify(take(input, {
+        input: (_) => _json(_),
+        targetModelId: [],
+    }));
+    b.m("POST").h(headers).b(body);
+    return b.build();
+};
 export const se_RetrieveCommand = async (input, context) => {
     const b = rb(input, context);
     const headers = {
@@ -150,6 +189,30 @@ export const de_InvokeFlowCommand = async (output, context) => {
     contents.responseStream = de_FlowResponseStream(data, context);
     return contents;
 };
+export const de_InvokeInlineAgentCommand = async (output, context) => {
+    if (output.statusCode !== 200 && output.statusCode >= 300) {
+        return de_CommandError(output, context);
+    }
+    const contents = map({
+        $metadata: deserializeMetadata(output),
+        [_cT]: [, output.headers[_xabact]],
+        [_sI]: [, output.headers[_xabasi]],
+    });
+    const data = output.body;
+    contents.completion = de_InlineAgentResponseStream(data, context);
+    return contents;
+};
+export const de_OptimizePromptCommand = async (output, context) => {
+    if (output.statusCode !== 200 && output.statusCode >= 300) {
+        return de_CommandError(output, context);
+    }
+    const contents = map({
+        $metadata: deserializeMetadata(output),
+    });
+    const data = output.body;
+    contents.optimizedPrompt = de_OptimizedPromptStream(data, context);
+    return contents;
+};
 export const de_RetrieveCommand = async (output, context) => {
     if (output.statusCode !== 200 && output.statusCode >= 300) {
         return de_CommandError(output, context);
@@ -410,6 +473,121 @@ const de_FlowResponseStream = (output, context) => {
         return { $unknown: output };
     });
 };
+const de_InlineAgentResponseStream = (output, context) => {
+    return context.eventStreamMarshaller.deserialize(output, async (event) => {
+        if (event["chunk"] != null) {
+            return {
+                chunk: await de_InlineAgentPayloadPart_event(event["chunk"], context),
+            };
+        }
+        if (event["trace"] != null) {
+            return {
+                trace: await de_InlineAgentTracePart_event(event["trace"], context),
+            };
+        }
+        if (event["returnControl"] != null) {
+            return {
+                returnControl: await de_InlineAgentReturnControlPayload_event(event["returnControl"], context),
+            };
+        }
+        if (event["internalServerException"] != null) {
+            return {
+                internalServerException: await de_InternalServerException_event(event["internalServerException"], context),
+            };
+        }
+        if (event["validationException"] != null) {
+            return {
+                validationException: await de_ValidationException_event(event["validationException"], context),
+            };
+        }
+        if (event["resourceNotFoundException"] != null) {
+            return {
+                resourceNotFoundException: await de_ResourceNotFoundException_event(event["resourceNotFoundException"], context),
+            };
+        }
+        if (event["serviceQuotaExceededException"] != null) {
+            return {
+                serviceQuotaExceededException: await de_ServiceQuotaExceededException_event(event["serviceQuotaExceededException"], context),
+            };
+        }
+        if (event["throttlingException"] != null) {
+            return {
+                throttlingException: await de_ThrottlingException_event(event["throttlingException"], context),
+            };
+        }
+        if (event["accessDeniedException"] != null) {
+            return {
+                accessDeniedException: await de_AccessDeniedException_event(event["accessDeniedException"], context),
+            };
+        }
+        if (event["conflictException"] != null) {
+            return {
+                conflictException: await de_ConflictException_event(event["conflictException"], context),
+            };
+        }
+        if (event["dependencyFailedException"] != null) {
+            return {
+                dependencyFailedException: await de_DependencyFailedException_event(event["dependencyFailedException"], context),
+            };
+        }
+        if (event["badGatewayException"] != null) {
+            return {
+                badGatewayException: await de_BadGatewayException_event(event["badGatewayException"], context),
+            };
+        }
+        if (event["files"] != null) {
+            return {
+                files: await de_InlineAgentFilePart_event(event["files"], context),
+            };
+        }
+        return { $unknown: output };
+    });
+};
+const de_OptimizedPromptStream = (output, context) => {
+    return context.eventStreamMarshaller.deserialize(output, async (event) => {
+        if (event["optimizedPromptEvent"] != null) {
+            return {
+                optimizedPromptEvent: await de_OptimizedPromptEvent_event(event["optimizedPromptEvent"], context),
+            };
+        }
+        if (event["analyzePromptEvent"] != null) {
+            return {
+                analyzePromptEvent: await de_AnalyzePromptEvent_event(event["analyzePromptEvent"], context),
+            };
+        }
+        if (event["internalServerException"] != null) {
+            return {
+                internalServerException: await de_InternalServerException_event(event["internalServerException"], context),
+            };
+        }
+        if (event["throttlingException"] != null) {
+            return {
+                throttlingException: await de_ThrottlingException_event(event["throttlingException"], context),
+            };
+        }
+        if (event["validationException"] != null) {
+            return {
+                validationException: await de_ValidationException_event(event["validationException"], context),
+            };
+        }
+        if (event["dependencyFailedException"] != null) {
+            return {
+                dependencyFailedException: await de_DependencyFailedException_event(event["dependencyFailedException"], context),
+            };
+        }
+        if (event["accessDeniedException"] != null) {
+            return {
+                accessDeniedException: await de_AccessDeniedException_event(event["accessDeniedException"], context),
+            };
+        }
+        if (event["badGatewayException"] != null) {
+            return {
+                badGatewayException: await de_BadGatewayException_event(event["badGatewayException"], context),
+            };
+        }
+        return { $unknown: output };
+    });
+};
 const de_ResponseStream = (output, context) => {
     return context.eventStreamMarshaller.deserialize(output, async (event) => {
         if (event["chunk"] != null) {
@@ -487,6 +665,12 @@ const de_AccessDeniedException_event = async (output, context) => {
     };
     return de_AccessDeniedExceptionRes(parsedOutput, context);
 };
+const de_AnalyzePromptEvent_event = async (output, context) => {
+    const contents = {};
+    const data = await parseBody(output.body, context);
+    Object.assign(contents, _json(data));
+    return contents;
+};
 const de_BadGatewayException_event = async (output, context) => {
     const parsedOutput = {
         ...output,
@@ -532,6 +716,30 @@ const de_FlowTraceEvent_event = async (output, context) => {
     Object.assign(contents, de_FlowTraceEvent(data, context));
     return contents;
 };
+const de_InlineAgentFilePart_event = async (output, context) => {
+    const contents = {};
+    const data = await parseBody(output.body, context);
+    Object.assign(contents, de_InlineAgentFilePart(data, context));
+    return contents;
+};
+const de_InlineAgentPayloadPart_event = async (output, context) => {
+    const contents = {};
+    const data = await parseBody(output.body, context);
+    Object.assign(contents, de_InlineAgentPayloadPart(data, context));
+    return contents;
+};
+const de_InlineAgentReturnControlPayload_event = async (output, context) => {
+    const contents = {};
+    const data = await parseBody(output.body, context);
+    Object.assign(contents, _json(data));
+    return contents;
+};
+const de_InlineAgentTracePart_event = async (output, context) => {
+    const contents = {};
+    const data = await parseBody(output.body, context);
+    Object.assign(contents, de_InlineAgentTracePart(data, context));
+    return contents;
+};
 const de_InternalServerException_event = async (output, context) => {
     const parsedOutput = {
         ...output,
@@ -539,6 +747,12 @@ const de_InternalServerException_event = async (output, context) => {
     };
     return de_InternalServerExceptionRes(parsedOutput, context);
 };
+const de_OptimizedPromptEvent_event = async (output, context) => {
+    const contents = {};
+    const data = await parseBody(output.body, context);
+    Object.assign(contents, _json(data));
+    return contents;
+};
 const de_PayloadPart_event = async (output, context) => {
     const contents = {};
     const data = await parseBody(output.body, context);
@@ -688,6 +902,24 @@ const se_InferenceConfig = (input, context) => {
         textInferenceConfig: (_) => se_TextInferenceConfig(_, context),
     });
 };
+const se_InferenceConfiguration = (input, context) => {
+    return take(input, {
+        maximumLength: [],
+        stopSequences: _json,
+        temperature: __serializeFloat,
+        topK: [],
+        topP: __serializeFloat,
+    });
+};
+const se_InlineSessionState = (input, context) => {
+    return take(input, {
+        files: (_) => se_InputFiles(_, context),
+        invocationId: [],
+        promptSessionAttributes: _json,
+        returnControlInvocationResults: _json,
+        sessionAttributes: _json,
+    });
+};
 const se_InputFile = (input, context) => {
     return take(input, {
         name: [],
@@ -702,6 +934,13 @@ const se_InputFiles = (input, context) => {
         return se_InputFile(entry, context);
     });
 };
+const se_KnowledgeBase = (input, context) => {
+    return take(input, {
+        description: [],
+        knowledgeBaseId: [],
+        retrievalConfiguration: (_) => se_KnowledgeBaseRetrievalConfiguration(_, context),
+    });
+};
 const se_KnowledgeBaseConfiguration = (input, context) => {
     return take(input, {
         knowledgeBaseId: [],
@@ -729,6 +968,13 @@ const se_KnowledgeBaseRetrieveAndGenerateConfiguration = (input, context) => {
         retrievalConfiguration: (_) => se_KnowledgeBaseRetrievalConfiguration(_, context),
     });
 };
+const se_KnowledgeBases = (input, context) => {
+    return input
+        .filter((e) => e != null)
+        .map((entry) => {
+        return se_KnowledgeBase(entry, context);
+    });
+};
 const se_KnowledgeBaseVectorSearchConfiguration = (input, context) => {
     return take(input, {
         filter: (_) => se_RetrievalFilter(_, context),
@@ -744,6 +990,29 @@ const se_OrchestrationConfiguration = (input, context) => {
         queryTransformationConfiguration: _json,
     });
 };
+const se_PromptConfiguration = (input, context) => {
+    return take(input, {
+        basePromptTemplate: [],
+        inferenceConfiguration: (_) => se_InferenceConfiguration(_, context),
+        parserMode: [],
+        promptCreationMode: [],
+        promptState: [],
+        promptType: [],
+    });
+};
+const se_PromptConfigurations = (input, context) => {
+    return input
+        .filter((e) => e != null)
+        .map((entry) => {
+        return se_PromptConfiguration(entry, context);
+    });
+};
+const se_PromptOverrideConfiguration = (input, context) => {
+    return take(input, {
+        overrideLambda: [],
+        promptConfigurations: (_) => se_PromptConfigurations(_, context),
+    });
+};
 const se_RetrievalFilter = (input, context) => {
     return RetrievalFilter.visit(input, {
         andAll: (value) => ({ andAll: se_RetrievalFilterList(value, context) }),
@@ -933,6 +1202,23 @@ const de_InferenceConfiguration = (output, context) => {
         topP: __limitedParseFloat32,
     });
 };
+const de_InlineAgentFilePart = (output, context) => {
+    return take(output, {
+        files: (_) => de_OutputFiles(_, context),
+    });
+};
+const de_InlineAgentPayloadPart = (output, context) => {
+    return take(output, {
+        attribution: (_) => de_Attribution(_, context),
+        bytes: context.base64Decoder,
+    });
+};
+const de_InlineAgentTracePart = (output, context) => {
+    return take(output, {
+        sessionId: __expectString,
+        trace: (_) => de_Trace(__expectUnion(_), context),
+    });
+};
 const de_KnowledgeBaseLookupOutput = (output, context) => {
     return take(output, {
         retrievedReferences: (_) => de_RetrievedReferences(_, context),

package/dist-types/BedrockAgentRuntime.d.ts

@@ -4,6 +4,8 @@ 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, RetrieveAndGenerateCommandOutput } from "./commands/RetrieveAndGenerateCommand";
 import { RetrieveCommandInput, RetrieveCommandOutput } from "./commands/RetrieveCommand";
 export interface BedrockAgentRuntime {
@@ -31,6 +33,18 @@ export interface BedrockAgentRuntime {
     invokeFlow(args: InvokeFlowCommandInput, options?: __HttpHandlerOptions): Promise<InvokeFlowCommandOutput>;
     invokeFlow(args: InvokeFlowCommandInput, cb: (err: any, data?: InvokeFlowCommandOutput) => void): void;
     invokeFlow(args: InvokeFlowCommandInput, options: __HttpHandlerOptions, 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}
+     */
+    optimizePrompt(args: OptimizePromptCommandInput, options?: __HttpHandlerOptions): Promise<OptimizePromptCommandOutput>;
+    optimizePrompt(args: OptimizePromptCommandInput, cb: (err: any, data?: OptimizePromptCommandOutput) => void): void;
+    optimizePrompt(args: OptimizePromptCommandInput, options: __HttpHandlerOptions, cb: (err: any, data?: OptimizePromptCommandOutput) => void): void;
     /**
      * @see {@link RetrieveCommand}
      */

package/dist-types/BedrockAgentRuntimeClient.d.ts

@@ -12,6 +12,8 @@ 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, RetrieveAndGenerateCommandOutput } from "./commands/RetrieveAndGenerateCommand";
 import { RetrieveCommandInput, RetrieveCommandOutput } from "./commands/RetrieveCommand";
 import { ClientInputEndpointParameters, ClientResolvedEndpointParameters, EndpointParameters } from "./endpoint/EndpointParameters";
@@ -20,11 +22,11 @@ export { __Client };
 /**
  * @public
  */
-export type ServiceInputTypes = DeleteAgentMemoryCommandInput | GetAgentMemoryCommandInput | InvokeAgentCommandInput | InvokeFlowCommandInput | RetrieveAndGenerateCommandInput | RetrieveCommandInput;
+export type ServiceInputTypes = DeleteAgentMemoryCommandInput | GetAgentMemoryCommandInput | InvokeAgentCommandInput | InvokeFlowCommandInput | InvokeInlineAgentCommandInput | OptimizePromptCommandInput | RetrieveAndGenerateCommandInput | RetrieveCommandInput;
 /**
  * @public
  */
-export type ServiceOutputTypes = DeleteAgentMemoryCommandOutput | GetAgentMemoryCommandOutput | InvokeAgentCommandOutput | InvokeFlowCommandOutput | RetrieveAndGenerateCommandOutput | RetrieveCommandOutput;
+export type ServiceOutputTypes = DeleteAgentMemoryCommandOutput | GetAgentMemoryCommandOutput | InvokeAgentCommandOutput | InvokeFlowCommandOutput | InvokeInlineAgentCommandOutput | OptimizePromptCommandOutput | RetrieveAndGenerateCommandOutput | RetrieveCommandOutput;
 /**
  * @public
  */

package/dist-types/commands/InvokeAgentCommand.d.ts

@@ -242,10 +242,7 @@ declare const InvokeAgentCommand_base: {
  * //       },
  * //     },
  * //     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",
@@ -363,7 +360,7 @@ declare const InvokeAgentCommand_base: {
  * //           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
@@ -494,7 +491,7 @@ declare const InvokeAgentCommand_base: {
  * //           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: {
@@ -525,7 +522,7 @@ declare const InvokeAgentCommand_base: {
  * //           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: {
@@ -560,6 +557,9 @@ declare const InvokeAgentCommand_base: {
  * //           failureReason: "STRING_VALUE",
  * //         },
  * //       },
+ * //       agentId: "STRING_VALUE",
+ * //       agentAliasId: "STRING_VALUE",
+ * //       agentVersion: "STRING_VALUE",
  * //     },
  * //     returnControl: { // ReturnControlPayload
  * //       invocationInputs: [ // InvocationInputs

package/dist-types/commands/InvokeFlowCommand.d.ts

@@ -27,7 +27,7 @@ declare const InvokeFlowCommand_base: {
     getEndpointParameterInstructions(): import("@smithy/middleware-endpoint").EndpointParameterInstructions;
 };
 /**
- * <p>Invokes an alias of a flow to run the inputs that you specify and return the output of each node as a stream. If there's an error, the error is returned. For more information, see <a href="https://docs.aws.amazon.com/bedrock/latest/userguide/flows-test.html">Test a flow in Amazon Bedrock</a> in the Amazon Bedrock User Guide.</p>
+ * <p>Invokes an alias of a flow to run the inputs that you specify and return the output of each node as a stream. If there's an error, the error is returned. For more information, see <a href="https://docs.aws.amazon.com/bedrock/latest/userguide/flows-test.html">Test a flow in Amazon Bedrock</a> in the <a href="https://docs.aws.amazon.com/bedrock/latest/userguide/what-is-service.html">Amazon Bedrock User Guide</a>.</p>
  *          <note>
  *             <p>The CLI doesn't support streaming operations in Amazon Bedrock, including <code>InvokeFlow</code>.</p>
  *          </note>