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

package/dist-es/protocols/Aws_restJson1.js

@@ -48,6 +48,7 @@ export const se_InvokeAgentCommand = async (input, context) => {
         inputText: [],
         memoryId: [],
         sessionState: (_) => se_SessionState(_, context),
+        streamingConfigurations: (_) => _json(_),
     }));
     b.m("POST").h(headers).b(body);
     return b.build();
@@ -107,6 +108,22 @@ export const se_OptimizePromptCommand = async (input, context) => {
     b.m("POST").h(headers).b(body);
     return b.build();
 };
+export const se_RerankCommand = async (input, context) => {
+    const b = rb(input, context);
+    const headers = {
+        "content-type": "application/json",
+    };
+    b.bp("/rerank");
+    let body;
+    body = JSON.stringify(take(input, {
+        nextToken: [],
+        queries: (_) => _json(_),
+        rerankingConfiguration: (_) => se_RerankingConfiguration(_, context),
+        sources: (_) => se_RerankSourcesList(_, context),
+    }));
+    b.m("POST").h(headers).b(body);
+    return b.build();
+};
 export const se_RetrieveCommand = async (input, context) => {
     const b = rb(input, context);
     const headers = {
@@ -116,6 +133,7 @@ export const se_RetrieveCommand = async (input, context) => {
     b.p("knowledgeBaseId", () => input.knowledgeBaseId, "{knowledgeBaseId}", false);
     let body;
     body = JSON.stringify(take(input, {
+        guardrailConfiguration: (_) => _json(_),
         nextToken: [],
         retrievalConfiguration: (_) => se_KnowledgeBaseRetrievalConfiguration(_, context),
         retrievalQuery: (_) => _json(_),
@@ -139,6 +157,22 @@ export const se_RetrieveAndGenerateCommand = async (input, context) => {
     b.m("POST").h(headers).b(body);
     return b.build();
 };
+export const se_RetrieveAndGenerateStreamCommand = async (input, context) => {
+    const b = rb(input, context);
+    const headers = {
+        "content-type": "application/json",
+    };
+    b.bp("/retrieveAndGenerateStream");
+    let body;
+    body = JSON.stringify(take(input, {
+        input: (_) => _json(_),
+        retrieveAndGenerateConfiguration: (_) => se_RetrieveAndGenerateConfiguration(_, context),
+        sessionConfiguration: (_) => _json(_),
+        sessionId: [],
+    }));
+    b.m("POST").h(headers).b(body);
+    return b.build();
+};
 export const de_DeleteAgentMemoryCommand = async (output, context) => {
     if (output.statusCode !== 202 && output.statusCode >= 300) {
         return de_CommandError(output, context);
@@ -213,6 +247,21 @@ export const de_OptimizePromptCommand = async (output, context) => {
     contents.optimizedPrompt = de_OptimizedPromptStream(data, context);
     return contents;
 };
+export const de_RerankCommand = async (output, context) => {
+    if (output.statusCode !== 200 && output.statusCode >= 300) {
+        return de_CommandError(output, context);
+    }
+    const contents = map({
+        $metadata: deserializeMetadata(output),
+    });
+    const data = __expectNonNull(__expectObject(await parseBody(output.body, context)), "body");
+    const doc = take(data, {
+        nextToken: __expectString,
+        results: (_) => de_RerankResultsList(_, context),
+    });
+    Object.assign(contents, doc);
+    return contents;
+};
 export const de_RetrieveCommand = async (output, context) => {
     if (output.statusCode !== 200 && output.statusCode >= 300) {
         return de_CommandError(output, context);
@@ -222,6 +271,7 @@ export const de_RetrieveCommand = async (output, context) => {
     });
     const data = __expectNonNull(__expectObject(await parseBody(output.body, context)), "body");
     const doc = take(data, {
+        guardrailAction: __expectString,
         nextToken: __expectString,
         retrievalResults: (_) => de_KnowledgeBaseRetrievalResults(_, context),
     });
@@ -245,6 +295,18 @@ export const de_RetrieveAndGenerateCommand = async (output, context) => {
     Object.assign(contents, doc);
     return contents;
 };
+export const de_RetrieveAndGenerateStreamCommand = async (output, context) => {
+    if (output.statusCode !== 200 && output.statusCode >= 300) {
+        return de_CommandError(output, context);
+    }
+    const contents = map({
+        $metadata: deserializeMetadata(output),
+        [_sI]: [, output.headers[_xabkbsi]],
+    });
+    const data = output.body;
+    contents.stream = de_RetrieveAndGenerateStreamResponseOutput(data, context);
+    return contents;
+};
 const de_CommandError = async (output, context) => {
     const parsedOutput = {
         ...output,
@@ -658,6 +720,71 @@ const de_ResponseStream = (output, context) => {
         return { $unknown: output };
     });
 };
+const de_RetrieveAndGenerateStreamResponseOutput = (output, context) => {
+    return context.eventStreamMarshaller.deserialize(output, async (event) => {
+        if (event["output"] != null) {
+            return {
+                output: await de_RetrieveAndGenerateOutputEvent_event(event["output"], context),
+            };
+        }
+        if (event["citation"] != null) {
+            return {
+                citation: await de_CitationEvent_event(event["citation"], context),
+            };
+        }
+        if (event["guardrail"] != null) {
+            return {
+                guardrail: await de_GuardrailEvent_event(event["guardrail"], 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),
+            };
+        }
+        return { $unknown: output };
+    });
+};
 const de_AccessDeniedException_event = async (output, context) => {
     const parsedOutput = {
         ...output,
@@ -678,6 +805,12 @@ const de_BadGatewayException_event = async (output, context) => {
     };
     return de_BadGatewayExceptionRes(parsedOutput, context);
 };
+const de_CitationEvent_event = async (output, context) => {
+    const contents = {};
+    const data = await parseBody(output.body, context);
+    Object.assign(contents, de_CitationEvent(data, context));
+    return contents;
+};
 const de_ConflictException_event = async (output, context) => {
     const parsedOutput = {
         ...output,
@@ -716,6 +849,12 @@ const de_FlowTraceEvent_event = async (output, context) => {
     Object.assign(contents, de_FlowTraceEvent(data, context));
     return contents;
 };
+const de_GuardrailEvent_event = async (output, context) => {
+    const contents = {};
+    const data = await parseBody(output.body, context);
+    Object.assign(contents, _json(data));
+    return contents;
+};
 const de_InlineAgentFilePart_event = async (output, context) => {
     const contents = {};
     const data = await parseBody(output.body, context);
@@ -766,6 +905,12 @@ const de_ResourceNotFoundException_event = async (output, context) => {
     };
     return de_ResourceNotFoundExceptionRes(parsedOutput, context);
 };
+const de_RetrieveAndGenerateOutputEvent_event = async (output, context) => {
+    const contents = {};
+    const data = await parseBody(output.body, context);
+    Object.assign(contents, _json(data));
+    return contents;
+};
 const de_ReturnControlPayload_event = async (output, context) => {
     const contents = {};
     const data = await parseBody(output.body, context);
@@ -811,6 +956,18 @@ const se_AdditionalModelRequestFields = (input, context) => {
 const se_AdditionalModelRequestFieldsValue = (input, context) => {
     return input;
 };
+const se_BedrockRerankingConfiguration = (input, context) => {
+    return take(input, {
+        modelConfiguration: (_) => se_BedrockRerankingModelConfiguration(_, context),
+        numberOfResults: [],
+    });
+};
+const se_BedrockRerankingModelConfiguration = (input, context) => {
+    return take(input, {
+        additionalModelRequestFields: (_) => se_AdditionalModelRequestFields(_, context),
+        modelArn: [],
+    });
+};
 const se_ByteContentDoc = (input, context) => {
     return take(input, {
         contentType: [],
@@ -978,8 +1135,10 @@ const se_KnowledgeBases = (input, context) => {
 const se_KnowledgeBaseVectorSearchConfiguration = (input, context) => {
     return take(input, {
         filter: (_) => se_RetrievalFilter(_, context),
+        implicitFilterConfiguration: _json,
         numberOfResults: [],
         overrideSearchType: [],
+        rerankingConfiguration: (_) => se_VectorSearchRerankingConfiguration(_, context),
     });
 };
 const se_OrchestrationConfiguration = (input, context) => {
@@ -1013,6 +1172,32 @@ const se_PromptOverrideConfiguration = (input, context) => {
         promptConfigurations: (_) => se_PromptConfigurations(_, context),
     });
 };
+const se_RerankDocument = (input, context) => {
+    return take(input, {
+        jsonDocument: (_) => se_Document(_, context),
+        textDocument: _json,
+        type: [],
+    });
+};
+const se_RerankingConfiguration = (input, context) => {
+    return take(input, {
+        bedrockRerankingConfiguration: (_) => se_BedrockRerankingConfiguration(_, context),
+        type: [],
+    });
+};
+const se_RerankSource = (input, context) => {
+    return take(input, {
+        inlineDocumentSource: (_) => se_RerankDocument(_, context),
+        type: [],
+    });
+};
+const se_RerankSourcesList = (input, context) => {
+    return input
+        .filter((e) => e != null)
+        .map((entry) => {
+        return se_RerankSource(entry, context);
+    });
+};
 const se_RetrievalFilter = (input, context) => {
     return RetrievalFilter.visit(input, {
         andAll: (value) => ({ andAll: se_RetrievalFilterList(value, context) }),
@@ -1063,6 +1248,25 @@ const se_TextInferenceConfig = (input, context) => {
         topP: __serializeFloat,
     });
 };
+const se_VectorSearchBedrockRerankingConfiguration = (input, context) => {
+    return take(input, {
+        metadataConfiguration: _json,
+        modelConfiguration: (_) => se_VectorSearchBedrockRerankingModelConfiguration(_, context),
+        numberOfRerankedResults: [],
+    });
+};
+const se_VectorSearchBedrockRerankingModelConfiguration = (input, context) => {
+    return take(input, {
+        additionalModelRequestFields: (_) => se_AdditionalModelRequestFields(_, context),
+        modelArn: [],
+    });
+};
+const se_VectorSearchRerankingConfiguration = (input, context) => {
+    return take(input, {
+        bedrockRerankingConfiguration: (_) => se_VectorSearchBedrockRerankingConfiguration(_, context),
+        type: [],
+    });
+};
 const se_Document = (input, context) => {
     return input;
 };
@@ -1077,6 +1281,11 @@ const de_Citation = (output, context) => {
         retrievedReferences: (_) => de_RetrievedReferences(_, context),
     });
 };
+const de_CitationEvent = (output, context) => {
+    return take(output, {
+        citation: (_) => de_Citation(_, context),
+    });
+};
 const de_Citations = (output, context) => {
     const retVal = (output || [])
         .filter((e) => e != null)
@@ -1362,6 +1571,28 @@ const de_PreProcessingTrace = (output, context) => {
     }
     return { $unknown: Object.entries(output)[0] };
 };
+const de_RerankDocument = (output, context) => {
+    return take(output, {
+        jsonDocument: (_) => de_Document(_, context),
+        textDocument: _json,
+        type: __expectString,
+    });
+};
+const de_RerankResult = (output, context) => {
+    return take(output, {
+        document: (_) => de_RerankDocument(_, context),
+        index: __expectInt32,
+        relevanceScore: __limitedParseFloat32,
+    });
+};
+const de_RerankResultsList = (output, context) => {
+    const retVal = (output || [])
+        .filter((e) => e != null)
+        .map((entry) => {
+        return de_RerankResult(entry, context);
+    });
+    return retVal;
+};
 const de_RetrievalResultMetadata = (output, context) => {
     return Object.entries(output).reduce((acc, [key, value]) => {
         if (value === null) {
@@ -1390,6 +1621,11 @@ const de_RetrievedReferences = (output, context) => {
     return retVal;
 };
 const de_Trace = (output, context) => {
+    if (output.customOrchestrationTrace != null) {
+        return {
+            customOrchestrationTrace: _json(output.customOrchestrationTrace),
+        };
+    }
     if (output.failureTrace != null) {
         return {
             failureTrace: _json(output.failureTrace),
@@ -1445,3 +1681,4 @@ const _sI = "sessionId";
 const _xabact = "x-amzn-bedrock-agent-content-type";
 const _xabami = "x-amz-bedrock-agent-memory-id";
 const _xabasi = "x-amz-bedrock-agent-session-id";
+const _xabkbsi = "x-amzn-bedrock-knowledge-base-session-id";

package/dist-types/BedrockAgentRuntime.d.ts

@@ -6,7 +6,9 @@ import { InvokeAgentCommandInput, InvokeAgentCommandOutput } from "./commands/In
 import { InvokeFlowCommandInput, InvokeFlowCommandOutput } from "./commands/InvokeFlowCommand";
 import { InvokeInlineAgentCommandInput, InvokeInlineAgentCommandOutput } from "./commands/InvokeInlineAgentCommand";
 import { OptimizePromptCommandInput, OptimizePromptCommandOutput } from "./commands/OptimizePromptCommand";
+import { RerankCommandInput, RerankCommandOutput } from "./commands/RerankCommand";
 import { RetrieveAndGenerateCommandInput, RetrieveAndGenerateCommandOutput } from "./commands/RetrieveAndGenerateCommand";
+import { RetrieveAndGenerateStreamCommandInput, RetrieveAndGenerateStreamCommandOutput } from "./commands/RetrieveAndGenerateStreamCommand";
 import { RetrieveCommandInput, RetrieveCommandOutput } from "./commands/RetrieveCommand";
 export interface BedrockAgentRuntime {
     /**
@@ -45,6 +47,12 @@ export interface BedrockAgentRuntime {
     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 RerankCommand}
+     */
+    rerank(args: RerankCommandInput, options?: __HttpHandlerOptions): Promise<RerankCommandOutput>;
+    rerank(args: RerankCommandInput, cb: (err: any, data?: RerankCommandOutput) => void): void;
+    rerank(args: RerankCommandInput, options: __HttpHandlerOptions, cb: (err: any, data?: RerankCommandOutput) => void): void;
     /**
      * @see {@link RetrieveCommand}
      */
@@ -57,6 +65,12 @@ export interface BedrockAgentRuntime {
     retrieveAndGenerate(args: RetrieveAndGenerateCommandInput, options?: __HttpHandlerOptions): Promise<RetrieveAndGenerateCommandOutput>;
     retrieveAndGenerate(args: RetrieveAndGenerateCommandInput, cb: (err: any, data?: RetrieveAndGenerateCommandOutput) => void): void;
     retrieveAndGenerate(args: RetrieveAndGenerateCommandInput, options: __HttpHandlerOptions, cb: (err: any, data?: RetrieveAndGenerateCommandOutput) => void): void;
+    /**
+     * @see {@link RetrieveAndGenerateStreamCommand}
+     */
+    retrieveAndGenerateStream(args: RetrieveAndGenerateStreamCommandInput, options?: __HttpHandlerOptions): Promise<RetrieveAndGenerateStreamCommandOutput>;
+    retrieveAndGenerateStream(args: RetrieveAndGenerateStreamCommandInput, cb: (err: any, data?: RetrieveAndGenerateStreamCommandOutput) => void): void;
+    retrieveAndGenerateStream(args: RetrieveAndGenerateStreamCommandInput, options: __HttpHandlerOptions, cb: (err: any, data?: RetrieveAndGenerateStreamCommandOutput) => void): void;
 }
 /**
  * <p>Contains APIs related to model invocation and querying of knowledge bases.</p>

package/dist-types/BedrockAgentRuntimeClient.d.ts

@@ -14,7 +14,9 @@ import { InvokeAgentCommandInput, InvokeAgentCommandOutput } from "./commands/In
 import { InvokeFlowCommandInput, InvokeFlowCommandOutput } from "./commands/InvokeFlowCommand";
 import { InvokeInlineAgentCommandInput, InvokeInlineAgentCommandOutput } from "./commands/InvokeInlineAgentCommand";
 import { OptimizePromptCommandInput, OptimizePromptCommandOutput } from "./commands/OptimizePromptCommand";
+import { RerankCommandInput, RerankCommandOutput } from "./commands/RerankCommand";
 import { RetrieveAndGenerateCommandInput, RetrieveAndGenerateCommandOutput } from "./commands/RetrieveAndGenerateCommand";
+import { RetrieveAndGenerateStreamCommandInput, RetrieveAndGenerateStreamCommandOutput } from "./commands/RetrieveAndGenerateStreamCommand";
 import { RetrieveCommandInput, RetrieveCommandOutput } from "./commands/RetrieveCommand";
 import { ClientInputEndpointParameters, ClientResolvedEndpointParameters, EndpointParameters } from "./endpoint/EndpointParameters";
 import { RuntimeExtension, RuntimeExtensionsConfig } from "./runtimeExtensions";
@@ -22,11 +24,11 @@ export { __Client };
 /**
  * @public
  */
-export type ServiceInputTypes = DeleteAgentMemoryCommandInput | GetAgentMemoryCommandInput | InvokeAgentCommandInput | InvokeFlowCommandInput | InvokeInlineAgentCommandInput | OptimizePromptCommandInput | RetrieveAndGenerateCommandInput | RetrieveCommandInput;
+export type ServiceInputTypes = DeleteAgentMemoryCommandInput | GetAgentMemoryCommandInput | InvokeAgentCommandInput | InvokeFlowCommandInput | InvokeInlineAgentCommandInput | OptimizePromptCommandInput | RerankCommandInput | RetrieveAndGenerateCommandInput | RetrieveAndGenerateStreamCommandInput | RetrieveCommandInput;
 /**
  * @public
  */
-export type ServiceOutputTypes = DeleteAgentMemoryCommandOutput | GetAgentMemoryCommandOutput | InvokeAgentCommandOutput | InvokeFlowCommandOutput | InvokeInlineAgentCommandOutput | OptimizePromptCommandOutput | RetrieveAndGenerateCommandOutput | RetrieveCommandOutput;
+export type ServiceOutputTypes = DeleteAgentMemoryCommandOutput | GetAgentMemoryCommandOutput | InvokeAgentCommandOutput | InvokeFlowCommandOutput | InvokeInlineAgentCommandOutput | OptimizePromptCommandOutput | RerankCommandOutput | RetrieveAndGenerateCommandOutput | RetrieveAndGenerateStreamCommandOutput | RetrieveCommandOutput;
 /**
  * @public
  */

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

@@ -178,6 +178,43 @@ declare const InvokeAgentCommand_base: {
  *                 "<RetrievalFilter>",
  *               ],
  *             },
+ *             rerankingConfiguration: { // VectorSearchRerankingConfiguration
+ *               type: "BEDROCK_RERANKING_MODEL", // required
+ *               bedrockRerankingConfiguration: { // VectorSearchBedrockRerankingConfiguration
+ *                 modelConfiguration: { // VectorSearchBedrockRerankingModelConfiguration
+ *                   modelArn: "STRING_VALUE", // required
+ *                   additionalModelRequestFields: { // AdditionalModelRequestFields
+ *                     "<keys>": "DOCUMENT_VALUE",
+ *                   },
+ *                 },
+ *                 numberOfRerankedResults: Number("int"),
+ *                 metadataConfiguration: { // MetadataConfigurationForReranking
+ *                   selectionMode: "SELECTIVE" || "ALL", // required
+ *                   selectiveModeConfiguration: { // RerankingMetadataSelectiveModeConfiguration Union: only one key present
+ *                     fieldsToInclude: [ // FieldsForReranking
+ *                       { // FieldForReranking
+ *                         fieldName: "STRING_VALUE", // required
+ *                       },
+ *                     ],
+ *                     fieldsToExclude: [
+ *                       {
+ *                         fieldName: "STRING_VALUE", // required
+ *                       },
+ *                     ],
+ *                   },
+ *                 },
+ *               },
+ *             },
+ *             implicitFilterConfiguration: { // ImplicitFilterConfiguration
+ *               metadataAttributes: [ // MetadataAttributeSchemaList // required
+ *                 { // MetadataAttributeSchema
+ *                   key: "STRING_VALUE", // required
+ *                   type: "STRING" || "NUMBER" || "BOOLEAN" || "STRING_LIST", // required
+ *                   description: "STRING_VALUE", // required
+ *                 },
+ *               ],
+ *               modelArn: "STRING_VALUE", // required
+ *             },
  *           },
  *         },
  *       },
@@ -190,6 +227,10 @@ declare const InvokeAgentCommand_base: {
  *   enableTrace: true || false,
  *   inputText: "STRING_VALUE",
  *   memoryId: "STRING_VALUE",
+ *   streamingConfigurations: { // StreamingConfigurations
+ *     streamFinalResponse: true || false,
+ *     applyGuardrailInterval: Number("int"),
+ *   },
  * };
  * const command = new InvokeAgentCommand(input);
  * const response = await client.send(command);
@@ -215,7 +256,7 @@ declare const InvokeAgentCommand_base: {
  * //                   text: "STRING_VALUE", // required
  * //                 },
  * //                 location: { // RetrievalResultLocation
- * //                   type: "S3" || "WEB" || "CONFLUENCE" || "SALESFORCE" || "SHAREPOINT", // required
+ * //                   type: "S3" || "WEB" || "CONFLUENCE" || "SALESFORCE" || "SHAREPOINT" || "CUSTOM", // required
  * //                   s3Location: { // RetrievalResultS3Location
  * //                     uri: "STRING_VALUE",
  * //                   },
@@ -231,6 +272,9 @@ declare const InvokeAgentCommand_base: {
  * //                   sharePointLocation: { // RetrievalResultSharePointLocation
  * //                     url: "STRING_VALUE",
  * //                   },
+ * //                   customDocumentLocation: { // RetrievalResultCustomDocumentLocation
+ * //                     id: "STRING_VALUE",
+ * //                   },
  * //                 },
  * //                 metadata: { // RetrievalResultMetadata
  * //                   "<keys>": "DOCUMENT_VALUE",
@@ -449,7 +493,7 @@ declare const InvokeAgentCommand_base: {
  * //                     text: "STRING_VALUE", // required
  * //                   },
  * //                   location: {
- * //                     type: "S3" || "WEB" || "CONFLUENCE" || "SALESFORCE" || "SHAREPOINT", // required
+ * //                     type: "S3" || "WEB" || "CONFLUENCE" || "SALESFORCE" || "SHAREPOINT" || "CUSTOM", // required
  * //                     s3Location: {
  * //                       uri: "STRING_VALUE",
  * //                     },
@@ -465,6 +509,9 @@ declare const InvokeAgentCommand_base: {
  * //                     sharePointLocation: {
  * //                       url: "STRING_VALUE",
  * //                     },
+ * //                     customDocumentLocation: {
+ * //                       id: "STRING_VALUE",
+ * //                     },
  * //                   },
  * //                   metadata: {
  * //                     "<keys>": "DOCUMENT_VALUE",
@@ -556,6 +603,12 @@ declare const InvokeAgentCommand_base: {
  * //           traceId: "STRING_VALUE",
  * //           failureReason: "STRING_VALUE",
  * //         },
+ * //         customOrchestrationTrace: { // CustomOrchestrationTrace
+ * //           traceId: "STRING_VALUE",
+ * //           event: { // CustomOrchestrationTraceEvent
+ * //             text: "STRING_VALUE",
+ * //           },
+ * //         },
  * //       },
  * //       agentId: "STRING_VALUE",
  * //       agentAliasId: "STRING_VALUE",

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

@@ -215,6 +215,43 @@ declare const InvokeInlineAgentCommand_base: {
  *               "<RetrievalFilter>",
  *             ],
  *           },
+ *           rerankingConfiguration: { // VectorSearchRerankingConfiguration
+ *             type: "BEDROCK_RERANKING_MODEL", // required
+ *             bedrockRerankingConfiguration: { // VectorSearchBedrockRerankingConfiguration
+ *               modelConfiguration: { // VectorSearchBedrockRerankingModelConfiguration
+ *                 modelArn: "STRING_VALUE", // required
+ *                 additionalModelRequestFields: { // AdditionalModelRequestFields
+ *                   "<keys>": "DOCUMENT_VALUE",
+ *                 },
+ *               },
+ *               numberOfRerankedResults: Number("int"),
+ *               metadataConfiguration: { // MetadataConfigurationForReranking
+ *                 selectionMode: "SELECTIVE" || "ALL", // required
+ *                 selectiveModeConfiguration: { // RerankingMetadataSelectiveModeConfiguration Union: only one key present
+ *                   fieldsToInclude: [ // FieldsForReranking
+ *                     { // FieldForReranking
+ *                       fieldName: "STRING_VALUE", // required
+ *                     },
+ *                   ],
+ *                   fieldsToExclude: [
+ *                     {
+ *                       fieldName: "STRING_VALUE", // required
+ *                     },
+ *                   ],
+ *                 },
+ *               },
+ *             },
+ *           },
+ *           implicitFilterConfiguration: { // ImplicitFilterConfiguration
+ *             metadataAttributes: [ // MetadataAttributeSchemaList // required
+ *               { // MetadataAttributeSchema
+ *                 key: "STRING_VALUE", // required
+ *                 type: "STRING" || "NUMBER" || "BOOLEAN" || "STRING_LIST", // required
+ *                 description: "STRING_VALUE", // required
+ *               },
+ *             ],
+ *             modelArn: "STRING_VALUE", // required
+ *           },
  *         },
  *       },
  *     },
@@ -269,7 +306,7 @@ declare const InvokeInlineAgentCommand_base: {
  * //                   text: "STRING_VALUE", // required
  * //                 },
  * //                 location: { // RetrievalResultLocation
- * //                   type: "S3" || "WEB" || "CONFLUENCE" || "SALESFORCE" || "SHAREPOINT", // required
+ * //                   type: "S3" || "WEB" || "CONFLUENCE" || "SALESFORCE" || "SHAREPOINT" || "CUSTOM", // required
  * //                   s3Location: { // RetrievalResultS3Location
  * //                     uri: "STRING_VALUE",
  * //                   },
@@ -285,6 +322,9 @@ declare const InvokeInlineAgentCommand_base: {
  * //                   sharePointLocation: { // RetrievalResultSharePointLocation
  * //                     url: "STRING_VALUE",
  * //                   },
+ * //                   customDocumentLocation: { // RetrievalResultCustomDocumentLocation
+ * //                     id: "STRING_VALUE",
+ * //                   },
  * //                 },
  * //                 metadata: { // RetrievalResultMetadata
  * //                   "<keys>": "DOCUMENT_VALUE",
@@ -503,7 +543,7 @@ declare const InvokeInlineAgentCommand_base: {
  * //                     text: "STRING_VALUE", // required
  * //                   },
  * //                   location: {
- * //                     type: "S3" || "WEB" || "CONFLUENCE" || "SALESFORCE" || "SHAREPOINT", // required
+ * //                     type: "S3" || "WEB" || "CONFLUENCE" || "SALESFORCE" || "SHAREPOINT" || "CUSTOM", // required
  * //                     s3Location: {
  * //                       uri: "STRING_VALUE",
  * //                     },
@@ -519,6 +559,9 @@ declare const InvokeInlineAgentCommand_base: {
  * //                     sharePointLocation: {
  * //                       url: "STRING_VALUE",
  * //                     },
+ * //                     customDocumentLocation: {
+ * //                       id: "STRING_VALUE",
+ * //                     },
  * //                   },
  * //                   metadata: {
  * //                     "<keys>": "DOCUMENT_VALUE",
@@ -610,6 +653,12 @@ declare const InvokeInlineAgentCommand_base: {
  * //           traceId: "STRING_VALUE",
  * //           failureReason: "STRING_VALUE",
  * //         },
+ * //         customOrchestrationTrace: { // CustomOrchestrationTrace
+ * //           traceId: "STRING_VALUE",
+ * //           event: { // CustomOrchestrationTraceEvent
+ * //             text: "STRING_VALUE",
+ * //           },
+ * //         },
  * //       },
  * //     },
  * //     returnControl: { // InlineAgentReturnControlPayload