@aws-sdk/client-bedrock-agent-runtime 3.540.0 → 3.543.0

package/dist-cjs/index.js

@@ -36,6 +36,7 @@ __export(src_exports, {
   FailureTraceFilterSensitiveLog: () => FailureTraceFilterSensitiveLog,
   FinalResponseFilterSensitiveLog: () => FinalResponseFilterSensitiveLog,
   GeneratedResponsePartFilterSensitiveLog: () => GeneratedResponsePartFilterSensitiveLog,
+  GenerationConfigurationFilterSensitiveLog: () => GenerationConfigurationFilterSensitiveLog,
   InternalServerException: () => InternalServerException,
   InvocationInputFilterSensitiveLog: () => InvocationInputFilterSensitiveLog,
   InvocationType: () => InvocationType,
@@ -45,7 +46,10 @@ __export(src_exports, {
   KnowledgeBaseLookupInputFilterSensitiveLog: () => KnowledgeBaseLookupInputFilterSensitiveLog,
   KnowledgeBaseLookupOutputFilterSensitiveLog: () => KnowledgeBaseLookupOutputFilterSensitiveLog,
   KnowledgeBaseQueryFilterSensitiveLog: () => KnowledgeBaseQueryFilterSensitiveLog,
+  KnowledgeBaseRetrievalConfigurationFilterSensitiveLog: () => KnowledgeBaseRetrievalConfigurationFilterSensitiveLog,
   KnowledgeBaseRetrievalResultFilterSensitiveLog: () => KnowledgeBaseRetrievalResultFilterSensitiveLog,
+  KnowledgeBaseRetrieveAndGenerateConfigurationFilterSensitiveLog: () => KnowledgeBaseRetrieveAndGenerateConfigurationFilterSensitiveLog,
+  KnowledgeBaseVectorSearchConfigurationFilterSensitiveLog: () => KnowledgeBaseVectorSearchConfigurationFilterSensitiveLog,
   ModelInvocationInputFilterSensitiveLog: () => ModelInvocationInputFilterSensitiveLog,
   ObservationFilterSensitiveLog: () => ObservationFilterSensitiveLog,
   OrchestrationTrace: () => OrchestrationTrace,
@@ -59,16 +63,20 @@ __export(src_exports, {
   PreProcessingParsedResponseFilterSensitiveLog: () => PreProcessingParsedResponseFilterSensitiveLog,
   PreProcessingTrace: () => PreProcessingTrace,
   PreProcessingTraceFilterSensitiveLog: () => PreProcessingTraceFilterSensitiveLog,
+  PromptTemplateFilterSensitiveLog: () => PromptTemplateFilterSensitiveLog,
   PromptType: () => PromptType,
   RationaleFilterSensitiveLog: () => RationaleFilterSensitiveLog,
   RepromptResponseFilterSensitiveLog: () => RepromptResponseFilterSensitiveLog,
   ResourceNotFoundException: () => ResourceNotFoundException,
   ResponseStream: () => ResponseStream,
   ResponseStreamFilterSensitiveLog: () => ResponseStreamFilterSensitiveLog,
+  RetrievalFilter: () => RetrievalFilter,
+  RetrievalFilterFilterSensitiveLog: () => RetrievalFilterFilterSensitiveLog,
   RetrievalResultContentFilterSensitiveLog: () => RetrievalResultContentFilterSensitiveLog,
   RetrievalResultLocationFilterSensitiveLog: () => RetrievalResultLocationFilterSensitiveLog,
   RetrievalResultLocationType: () => RetrievalResultLocationType,
   RetrieveAndGenerateCommand: () => RetrieveAndGenerateCommand,
+  RetrieveAndGenerateConfigurationFilterSensitiveLog: () => RetrieveAndGenerateConfigurationFilterSensitiveLog,
   RetrieveAndGenerateInputFilterSensitiveLog: () => RetrieveAndGenerateInputFilterSensitiveLog,
   RetrieveAndGenerateOutputFilterSensitiveLog: () => RetrieveAndGenerateOutputFilterSensitiveLog,
   RetrieveAndGenerateRequestFilterSensitiveLog: () => RetrieveAndGenerateRequestFilterSensitiveLog,
@@ -533,6 +541,34 @@ var SearchType = {
 var RetrieveAndGenerateType = {
   KNOWLEDGE_BASE: "KNOWLEDGE_BASE"
 };
+var RetrievalFilter;
+((RetrievalFilter2) => {
+  RetrievalFilter2.visit = /* @__PURE__ */ __name((value, visitor) => {
+    if (value.equals !== void 0)
+      return visitor.equals(value.equals);
+    if (value.notEquals !== void 0)
+      return visitor.notEquals(value.notEquals);
+    if (value.greaterThan !== void 0)
+      return visitor.greaterThan(value.greaterThan);
+    if (value.greaterThanOrEquals !== void 0)
+      return visitor.greaterThanOrEquals(value.greaterThanOrEquals);
+    if (value.lessThan !== void 0)
+      return visitor.lessThan(value.lessThan);
+    if (value.lessThanOrEquals !== void 0)
+      return visitor.lessThanOrEquals(value.lessThanOrEquals);
+    if (value.in !== void 0)
+      return visitor.in(value.in);
+    if (value.notIn !== void 0)
+      return visitor.notIn(value.notIn);
+    if (value.startsWith !== void 0)
+      return visitor.startsWith(value.startsWith);
+    if (value.andAll !== void 0)
+      return visitor.andAll(value.andAll);
+    if (value.orAll !== void 0)
+      return visitor.orAll(value.orAll);
+    return visitor._(value.$unknown[0], value.$unknown[1]);
+  }, "visit");
+})(RetrievalFilter || (RetrievalFilter = {}));
 var ActionGroupInvocationInputFilterSensitiveLog = /* @__PURE__ */ __name((obj) => ({
   ...obj,
   ...obj.actionGroupName && { actionGroupName: import_smithy_client.SENSITIVE_STRING },
@@ -563,7 +599,8 @@ var RetrievalResultLocationFilterSensitiveLog = /* @__PURE__ */ __name((obj) =>
 var RetrievedReferenceFilterSensitiveLog = /* @__PURE__ */ __name((obj) => ({
   ...obj,
   ...obj.content && { content: import_smithy_client.SENSITIVE_STRING },
-  ...obj.location && { location: import_smithy_client.SENSITIVE_STRING }
+  ...obj.location && { location: import_smithy_client.SENSITIVE_STRING },
+  ...obj.metadata && { metadata: import_smithy_client.SENSITIVE_STRING }
 }), "RetrievedReferenceFilterSensitiveLog");
 var CitationFilterSensitiveLog = /* @__PURE__ */ __name((obj) => ({
   ...obj,
@@ -727,10 +764,14 @@ var InvokeAgentResponseFilterSensitiveLog = /* @__PURE__ */ __name((obj) => ({
 var RetrieveAndGenerateInputFilterSensitiveLog = /* @__PURE__ */ __name((obj) => ({
   ...obj
 }), "RetrieveAndGenerateInputFilterSensitiveLog");
-var RetrieveAndGenerateRequestFilterSensitiveLog = /* @__PURE__ */ __name((obj) => ({
+var PromptTemplateFilterSensitiveLog = /* @__PURE__ */ __name((obj) => ({
   ...obj,
-  ...obj.input && { input: import_smithy_client.SENSITIVE_STRING }
-}), "RetrieveAndGenerateRequestFilterSensitiveLog");
+  ...obj.textPromptTemplate && { textPromptTemplate: import_smithy_client.SENSITIVE_STRING }
+}), "PromptTemplateFilterSensitiveLog");
+var GenerationConfigurationFilterSensitiveLog = /* @__PURE__ */ __name((obj) => ({
+  ...obj,
+  ...obj.promptTemplate && { promptTemplate: PromptTemplateFilterSensitiveLog(obj.promptTemplate) }
+}), "GenerationConfigurationFilterSensitiveLog");
 var RetrieveAndGenerateOutputFilterSensitiveLog = /* @__PURE__ */ __name((obj) => ({
   ...obj
 }), "RetrieveAndGenerateOutputFilterSensitiveLog");
@@ -742,19 +783,85 @@ var RetrieveAndGenerateResponseFilterSensitiveLog = /* @__PURE__ */ __name((obj)
 var KnowledgeBaseQueryFilterSensitiveLog = /* @__PURE__ */ __name((obj) => ({
   ...obj
 }), "KnowledgeBaseQueryFilterSensitiveLog");
-var RetrieveRequestFilterSensitiveLog = /* @__PURE__ */ __name((obj) => ({
-  ...obj,
-  ...obj.retrievalQuery && { retrievalQuery: import_smithy_client.SENSITIVE_STRING }
-}), "RetrieveRequestFilterSensitiveLog");
 var KnowledgeBaseRetrievalResultFilterSensitiveLog = /* @__PURE__ */ __name((obj) => ({
   ...obj,
   ...obj.content && { content: import_smithy_client.SENSITIVE_STRING },
-  ...obj.location && { location: import_smithy_client.SENSITIVE_STRING }
+  ...obj.location && { location: import_smithy_client.SENSITIVE_STRING },
+  ...obj.metadata && { metadata: import_smithy_client.SENSITIVE_STRING }
 }), "KnowledgeBaseRetrievalResultFilterSensitiveLog");
 var RetrieveResponseFilterSensitiveLog = /* @__PURE__ */ __name((obj) => ({
   ...obj,
   ...obj.retrievalResults && { retrievalResults: import_smithy_client.SENSITIVE_STRING }
 }), "RetrieveResponseFilterSensitiveLog");
+var RetrievalFilterFilterSensitiveLog = /* @__PURE__ */ __name((obj) => {
+  if (obj.equals !== void 0)
+    return { equals: obj.equals };
+  if (obj.notEquals !== void 0)
+    return { notEquals: obj.notEquals };
+  if (obj.greaterThan !== void 0)
+    return { greaterThan: obj.greaterThan };
+  if (obj.greaterThanOrEquals !== void 0)
+    return { greaterThanOrEquals: obj.greaterThanOrEquals };
+  if (obj.lessThan !== void 0)
+    return { lessThan: obj.lessThan };
+  if (obj.lessThanOrEquals !== void 0)
+    return { lessThanOrEquals: obj.lessThanOrEquals };
+  if (obj.in !== void 0)
+    return { in: obj.in };
+  if (obj.notIn !== void 0)
+    return { notIn: obj.notIn };
+  if (obj.startsWith !== void 0)
+    return { startsWith: obj.startsWith };
+  if (obj.andAll !== void 0)
+    return { andAll: import_smithy_client.SENSITIVE_STRING };
+  if (obj.orAll !== void 0)
+    return { orAll: import_smithy_client.SENSITIVE_STRING };
+  if (obj.$unknown !== void 0)
+    return { [obj.$unknown[0]]: "UNKNOWN" };
+}, "RetrievalFilterFilterSensitiveLog");
+var KnowledgeBaseVectorSearchConfigurationFilterSensitiveLog = /* @__PURE__ */ __name((obj) => ({
+  ...obj,
+  ...obj.filter && { filter: import_smithy_client.SENSITIVE_STRING }
+}), "KnowledgeBaseVectorSearchConfigurationFilterSensitiveLog");
+var KnowledgeBaseRetrievalConfigurationFilterSensitiveLog = /* @__PURE__ */ __name((obj) => ({
+  ...obj,
+  ...obj.vectorSearchConfiguration && {
+    vectorSearchConfiguration: KnowledgeBaseVectorSearchConfigurationFilterSensitiveLog(obj.vectorSearchConfiguration)
+  }
+}), "KnowledgeBaseRetrievalConfigurationFilterSensitiveLog");
+var KnowledgeBaseRetrieveAndGenerateConfigurationFilterSensitiveLog = /* @__PURE__ */ __name((obj) => ({
+  ...obj,
+  ...obj.retrievalConfiguration && {
+    retrievalConfiguration: KnowledgeBaseRetrievalConfigurationFilterSensitiveLog(obj.retrievalConfiguration)
+  },
+  ...obj.generationConfiguration && {
+    generationConfiguration: GenerationConfigurationFilterSensitiveLog(obj.generationConfiguration)
+  }
+}), "KnowledgeBaseRetrieveAndGenerateConfigurationFilterSensitiveLog");
+var RetrieveRequestFilterSensitiveLog = /* @__PURE__ */ __name((obj) => ({
+  ...obj,
+  ...obj.retrievalQuery && { retrievalQuery: import_smithy_client.SENSITIVE_STRING },
+  ...obj.retrievalConfiguration && {
+    retrievalConfiguration: KnowledgeBaseRetrievalConfigurationFilterSensitiveLog(obj.retrievalConfiguration)
+  }
+}), "RetrieveRequestFilterSensitiveLog");
+var RetrieveAndGenerateConfigurationFilterSensitiveLog = /* @__PURE__ */ __name((obj) => ({
+  ...obj,
+  ...obj.knowledgeBaseConfiguration && {
+    knowledgeBaseConfiguration: KnowledgeBaseRetrieveAndGenerateConfigurationFilterSensitiveLog(
+      obj.knowledgeBaseConfiguration
+    )
+  }
+}), "RetrieveAndGenerateConfigurationFilterSensitiveLog");
+var RetrieveAndGenerateRequestFilterSensitiveLog = /* @__PURE__ */ __name((obj) => ({
+  ...obj,
+  ...obj.input && { input: import_smithy_client.SENSITIVE_STRING },
+  ...obj.retrieveAndGenerateConfiguration && {
+    retrieveAndGenerateConfiguration: RetrieveAndGenerateConfigurationFilterSensitiveLog(
+      obj.retrieveAndGenerateConfiguration
+    )
+  }
+}), "RetrieveAndGenerateRequestFilterSensitiveLog");
 
 // src/protocols/Aws_restJson1.ts
 var import_core2 = require("@aws-sdk/core");
@@ -792,7 +899,7 @@ var se_RetrieveCommand = /* @__PURE__ */ __name(async (input, context) => {
   body = JSON.stringify(
     (0, import_smithy_client.take)(input, {
       nextToken: [],
-      retrievalConfiguration: (_) => (0, import_smithy_client._json)(_),
+      retrievalConfiguration: (_) => se_KnowledgeBaseRetrievalConfiguration(_, context),
       retrievalQuery: (_) => (0, import_smithy_client._json)(_)
     })
   );
@@ -809,7 +916,7 @@ var se_RetrieveAndGenerateCommand = /* @__PURE__ */ __name(async (input, context
   body = JSON.stringify(
     (0, import_smithy_client.take)(input, {
       input: (_) => (0, import_smithy_client._json)(_),
-      retrieveAndGenerateConfiguration: (_) => (0, import_smithy_client._json)(_),
+      retrieveAndGenerateConfiguration: (_) => se_RetrieveAndGenerateConfiguration(_, context),
       sessionConfiguration: (_) => (0, import_smithy_client._json)(_),
       sessionId: []
     })
@@ -854,7 +961,7 @@ var de_RetrieveAndGenerateCommand = /* @__PURE__ */ __name(async (output, contex
   });
   const data = (0, import_smithy_client.expectNonNull)((0, import_smithy_client.expectObject)(await (0, import_core2.parseJsonBody)(output.body, context)), "body");
   const doc = (0, import_smithy_client.take)(data, {
-    citations: import_smithy_client._json,
+    citations: (_) => de_Citations(_, context),
     output: import_smithy_client._json,
     sessionId: import_smithy_client.expectString
   });
@@ -1168,6 +1275,79 @@ var de_ValidationException_event = /* @__PURE__ */ __name(async (output, context
   };
   return de_ValidationExceptionRes(parsedOutput, context);
 }, "de_ValidationException_event");
+var se_FilterAttribute = /* @__PURE__ */ __name((input, context) => {
+  return (0, import_smithy_client.take)(input, {
+    key: [],
+    value: (_) => se_FilterValue(_, context)
+  });
+}, "se_FilterAttribute");
+var se_FilterValue = /* @__PURE__ */ __name((input, context) => {
+  return input;
+}, "se_FilterValue");
+var se_KnowledgeBaseRetrievalConfiguration = /* @__PURE__ */ __name((input, context) => {
+  return (0, import_smithy_client.take)(input, {
+    vectorSearchConfiguration: (_) => se_KnowledgeBaseVectorSearchConfiguration(_, context)
+  });
+}, "se_KnowledgeBaseRetrievalConfiguration");
+var se_KnowledgeBaseRetrieveAndGenerateConfiguration = /* @__PURE__ */ __name((input, context) => {
+  return (0, import_smithy_client.take)(input, {
+    generationConfiguration: import_smithy_client._json,
+    knowledgeBaseId: [],
+    modelArn: [],
+    retrievalConfiguration: (_) => se_KnowledgeBaseRetrievalConfiguration(_, context)
+  });
+}, "se_KnowledgeBaseRetrieveAndGenerateConfiguration");
+var se_KnowledgeBaseVectorSearchConfiguration = /* @__PURE__ */ __name((input, context) => {
+  return (0, import_smithy_client.take)(input, {
+    filter: (_) => se_RetrievalFilter(_, context),
+    numberOfResults: [],
+    overrideSearchType: []
+  });
+}, "se_KnowledgeBaseVectorSearchConfiguration");
+var se_RetrievalFilter = /* @__PURE__ */ __name((input, context) => {
+  return RetrievalFilter.visit(input, {
+    andAll: (value) => ({ andAll: se_RetrievalFilterList(value, context) }),
+    equals: (value) => ({ equals: se_FilterAttribute(value, context) }),
+    greaterThan: (value) => ({ greaterThan: se_FilterAttribute(value, context) }),
+    greaterThanOrEquals: (value) => ({ greaterThanOrEquals: se_FilterAttribute(value, context) }),
+    in: (value) => ({ in: se_FilterAttribute(value, context) }),
+    lessThan: (value) => ({ lessThan: se_FilterAttribute(value, context) }),
+    lessThanOrEquals: (value) => ({ lessThanOrEquals: se_FilterAttribute(value, context) }),
+    notEquals: (value) => ({ notEquals: se_FilterAttribute(value, context) }),
+    notIn: (value) => ({ notIn: se_FilterAttribute(value, context) }),
+    orAll: (value) => ({ orAll: se_RetrievalFilterList(value, context) }),
+    startsWith: (value) => ({ startsWith: se_FilterAttribute(value, context) }),
+    _: (name, value) => ({ name: value })
+  });
+}, "se_RetrievalFilter");
+var se_RetrievalFilterList = /* @__PURE__ */ __name((input, context) => {
+  return input.filter((e) => e != null).map((entry) => {
+    return se_RetrievalFilter(entry, context);
+  });
+}, "se_RetrievalFilterList");
+var se_RetrieveAndGenerateConfiguration = /* @__PURE__ */ __name((input, context) => {
+  return (0, import_smithy_client.take)(input, {
+    knowledgeBaseConfiguration: (_) => se_KnowledgeBaseRetrieveAndGenerateConfiguration(_, context),
+    type: []
+  });
+}, "se_RetrieveAndGenerateConfiguration");
+var de_Attribution = /* @__PURE__ */ __name((output, context) => {
+  return (0, import_smithy_client.take)(output, {
+    citations: (_) => de_Citations(_, context)
+  });
+}, "de_Attribution");
+var de_Citation = /* @__PURE__ */ __name((output, context) => {
+  return (0, import_smithy_client.take)(output, {
+    generatedResponsePart: import_smithy_client._json,
+    retrievedReferences: (_) => de_RetrievedReferences(_, context)
+  });
+}, "de_Citation");
+var de_Citations = /* @__PURE__ */ __name((output, context) => {
+  const retVal = (output || []).filter((e) => e != null).map((entry) => {
+    return de_Citation(entry, context);
+  });
+  return retVal;
+}, "de_Citations");
 var de_InferenceConfiguration = /* @__PURE__ */ __name((output, context) => {
   return (0, import_smithy_client.take)(output, {
     maximumLength: import_smithy_client.expectInt32,
@@ -1177,10 +1357,16 @@ var de_InferenceConfiguration = /* @__PURE__ */ __name((output, context) => {
     topP: import_smithy_client.limitedParseFloat32
   });
 }, "de_InferenceConfiguration");
+var de_KnowledgeBaseLookupOutput = /* @__PURE__ */ __name((output, context) => {
+  return (0, import_smithy_client.take)(output, {
+    retrievedReferences: (_) => de_RetrievedReferences(_, context)
+  });
+}, "de_KnowledgeBaseLookupOutput");
 var de_KnowledgeBaseRetrievalResult = /* @__PURE__ */ __name((output, context) => {
   return (0, import_smithy_client.take)(output, {
     content: import_smithy_client._json,
     location: import_smithy_client._json,
+    metadata: (_) => de_RetrievalResultMetadata(_, context),
     score: import_smithy_client.limitedParseDouble
   });
 }, "de_KnowledgeBaseRetrievalResult");
@@ -1201,6 +1387,16 @@ var de_ModelInvocationInput = /* @__PURE__ */ __name((output, context) => {
     type: import_smithy_client.expectString
   });
 }, "de_ModelInvocationInput");
+var de_Observation = /* @__PURE__ */ __name((output, context) => {
+  return (0, import_smithy_client.take)(output, {
+    actionGroupInvocationOutput: import_smithy_client._json,
+    finalResponse: import_smithy_client._json,
+    knowledgeBaseLookupOutput: (_) => de_KnowledgeBaseLookupOutput(_, context),
+    repromptResponse: import_smithy_client._json,
+    traceId: import_smithy_client.expectString,
+    type: import_smithy_client.expectString
+  });
+}, "de_Observation");
 var de_OrchestrationTrace = /* @__PURE__ */ __name((output, context) => {
   if (output.invocationInput != null) {
     return {
@@ -1214,7 +1410,7 @@ var de_OrchestrationTrace = /* @__PURE__ */ __name((output, context) => {
   }
   if (output.observation != null) {
     return {
-      observation: (0, import_smithy_client._json)(output.observation)
+      observation: de_Observation(output.observation, context)
     };
   }
   if (output.rationale != null) {
@@ -1226,7 +1422,7 @@ var de_OrchestrationTrace = /* @__PURE__ */ __name((output, context) => {
 }, "de_OrchestrationTrace");
 var de_PayloadPart = /* @__PURE__ */ __name((output, context) => {
   return (0, import_smithy_client.take)(output, {
-    attribution: import_smithy_client._json,
+    attribution: (_) => de_Attribution(_, context),
     bytes: context.base64Decoder
   });
 }, "de_PayloadPart");
@@ -1256,6 +1452,31 @@ var de_PreProcessingTrace = /* @__PURE__ */ __name((output, context) => {
   }
   return { $unknown: Object.entries(output)[0] };
 }, "de_PreProcessingTrace");
+var de_RetrievalResultMetadata = /* @__PURE__ */ __name((output, context) => {
+  return Object.entries(output).reduce((acc, [key, value]) => {
+    if (value === null) {
+      return acc;
+    }
+    acc[key] = de_RetrievalResultMetadataValue(value, context);
+    return acc;
+  }, {});
+}, "de_RetrievalResultMetadata");
+var de_RetrievalResultMetadataValue = /* @__PURE__ */ __name((output, context) => {
+  return output;
+}, "de_RetrievalResultMetadataValue");
+var de_RetrievedReference = /* @__PURE__ */ __name((output, context) => {
+  return (0, import_smithy_client.take)(output, {
+    content: import_smithy_client._json,
+    location: import_smithy_client._json,
+    metadata: (_) => de_RetrievalResultMetadata(_, context)
+  });
+}, "de_RetrievedReference");
+var de_RetrievedReferences = /* @__PURE__ */ __name((output, context) => {
+  const retVal = (output || []).filter((e) => e != null).map((entry) => {
+    return de_RetrievedReference(entry, context);
+  });
+  return retVal;
+}, "de_RetrievedReferences");
 var de_Trace = /* @__PURE__ */ __name((output, context) => {
   if (output.failureTrace != null) {
     return {
@@ -1401,6 +1622,7 @@ var paginateRetrieve = (0, import_core.createPaginator)(BedrockAgentRuntimeClien
   ResponseStream,
   SearchType,
   RetrieveAndGenerateType,
+  RetrievalFilter,
   ActionGroupInvocationInputFilterSensitiveLog,
   ActionGroupInvocationOutputFilterSensitiveLog,
   InvokeAgentRequestFilterSensitiveLog,
@@ -1433,12 +1655,19 @@ var paginateRetrieve = (0, import_core.createPaginator)(BedrockAgentRuntimeClien
   ResponseStreamFilterSensitiveLog,
   InvokeAgentResponseFilterSensitiveLog,
   RetrieveAndGenerateInputFilterSensitiveLog,
-  RetrieveAndGenerateRequestFilterSensitiveLog,
+  PromptTemplateFilterSensitiveLog,
+  GenerationConfigurationFilterSensitiveLog,
   RetrieveAndGenerateOutputFilterSensitiveLog,
   RetrieveAndGenerateResponseFilterSensitiveLog,
   KnowledgeBaseQueryFilterSensitiveLog,
-  RetrieveRequestFilterSensitiveLog,
   KnowledgeBaseRetrievalResultFilterSensitiveLog,
-  RetrieveResponseFilterSensitiveLog
+  RetrieveResponseFilterSensitiveLog,
+  RetrievalFilterFilterSensitiveLog,
+  KnowledgeBaseVectorSearchConfigurationFilterSensitiveLog,
+  KnowledgeBaseRetrievalConfigurationFilterSensitiveLog,
+  KnowledgeBaseRetrieveAndGenerateConfigurationFilterSensitiveLog,
+  RetrieveRequestFilterSensitiveLog,
+  RetrieveAndGenerateConfigurationFilterSensitiveLog,
+  RetrieveAndGenerateRequestFilterSensitiveLog
 });
 

package/dist-es/models/models_0.js

@@ -223,6 +223,34 @@ export const SearchType = {
 export const RetrieveAndGenerateType = {
     KNOWLEDGE_BASE: "KNOWLEDGE_BASE",
 };
+export var RetrievalFilter;
+(function (RetrievalFilter) {
+    RetrievalFilter.visit = (value, visitor) => {
+        if (value.equals !== undefined)
+            return visitor.equals(value.equals);
+        if (value.notEquals !== undefined)
+            return visitor.notEquals(value.notEquals);
+        if (value.greaterThan !== undefined)
+            return visitor.greaterThan(value.greaterThan);
+        if (value.greaterThanOrEquals !== undefined)
+            return visitor.greaterThanOrEquals(value.greaterThanOrEquals);
+        if (value.lessThan !== undefined)
+            return visitor.lessThan(value.lessThan);
+        if (value.lessThanOrEquals !== undefined)
+            return visitor.lessThanOrEquals(value.lessThanOrEquals);
+        if (value.in !== undefined)
+            return visitor.in(value.in);
+        if (value.notIn !== undefined)
+            return visitor.notIn(value.notIn);
+        if (value.startsWith !== undefined)
+            return visitor.startsWith(value.startsWith);
+        if (value.andAll !== undefined)
+            return visitor.andAll(value.andAll);
+        if (value.orAll !== undefined)
+            return visitor.orAll(value.orAll);
+        return visitor._(value.$unknown[0], value.$unknown[1]);
+    };
+})(RetrievalFilter || (RetrievalFilter = {}));
 export const ActionGroupInvocationInputFilterSensitiveLog = (obj) => ({
     ...obj,
     ...(obj.actionGroupName && { actionGroupName: SENSITIVE_STRING }),
@@ -254,6 +282,7 @@ export const RetrievedReferenceFilterSensitiveLog = (obj) => ({
     ...obj,
     ...(obj.content && { content: SENSITIVE_STRING }),
     ...(obj.location && { location: SENSITIVE_STRING }),
+    ...(obj.metadata && { metadata: SENSITIVE_STRING }),
 });
 export const CitationFilterSensitiveLog = (obj) => ({
     ...obj,
@@ -417,9 +446,13 @@ export const InvokeAgentResponseFilterSensitiveLog = (obj) => ({
 export const RetrieveAndGenerateInputFilterSensitiveLog = (obj) => ({
     ...obj,
 });
-export const RetrieveAndGenerateRequestFilterSensitiveLog = (obj) => ({
+export const PromptTemplateFilterSensitiveLog = (obj) => ({
     ...obj,
-    ...(obj.input && { input: SENSITIVE_STRING }),
+    ...(obj.textPromptTemplate && { textPromptTemplate: SENSITIVE_STRING }),
+});
+export const GenerationConfigurationFilterSensitiveLog = (obj) => ({
+    ...obj,
+    ...(obj.promptTemplate && { promptTemplate: PromptTemplateFilterSensitiveLog(obj.promptTemplate) }),
 });
 export const RetrieveAndGenerateOutputFilterSensitiveLog = (obj) => ({
     ...obj,
@@ -432,16 +465,78 @@ export const RetrieveAndGenerateResponseFilterSensitiveLog = (obj) => ({
 export const KnowledgeBaseQueryFilterSensitiveLog = (obj) => ({
     ...obj,
 });
-export const RetrieveRequestFilterSensitiveLog = (obj) => ({
-    ...obj,
-    ...(obj.retrievalQuery && { retrievalQuery: SENSITIVE_STRING }),
-});
 export const KnowledgeBaseRetrievalResultFilterSensitiveLog = (obj) => ({
     ...obj,
     ...(obj.content && { content: SENSITIVE_STRING }),
     ...(obj.location && { location: SENSITIVE_STRING }),
+    ...(obj.metadata && { metadata: SENSITIVE_STRING }),
 });
 export const RetrieveResponseFilterSensitiveLog = (obj) => ({
     ...obj,
     ...(obj.retrievalResults && { retrievalResults: SENSITIVE_STRING }),
 });
+export const RetrievalFilterFilterSensitiveLog = (obj) => {
+    if (obj.equals !== undefined)
+        return { equals: obj.equals };
+    if (obj.notEquals !== undefined)
+        return { notEquals: obj.notEquals };
+    if (obj.greaterThan !== undefined)
+        return { greaterThan: obj.greaterThan };
+    if (obj.greaterThanOrEquals !== undefined)
+        return { greaterThanOrEquals: obj.greaterThanOrEquals };
+    if (obj.lessThan !== undefined)
+        return { lessThan: obj.lessThan };
+    if (obj.lessThanOrEquals !== undefined)
+        return { lessThanOrEquals: obj.lessThanOrEquals };
+    if (obj.in !== undefined)
+        return { in: obj.in };
+    if (obj.notIn !== undefined)
+        return { notIn: obj.notIn };
+    if (obj.startsWith !== undefined)
+        return { startsWith: obj.startsWith };
+    if (obj.andAll !== undefined)
+        return { andAll: SENSITIVE_STRING };
+    if (obj.orAll !== undefined)
+        return { orAll: SENSITIVE_STRING };
+    if (obj.$unknown !== undefined)
+        return { [obj.$unknown[0]]: "UNKNOWN" };
+};
+export const KnowledgeBaseVectorSearchConfigurationFilterSensitiveLog = (obj) => ({
+    ...obj,
+    ...(obj.filter && { filter: SENSITIVE_STRING }),
+});
+export const KnowledgeBaseRetrievalConfigurationFilterSensitiveLog = (obj) => ({
+    ...obj,
+    ...(obj.vectorSearchConfiguration && {
+        vectorSearchConfiguration: KnowledgeBaseVectorSearchConfigurationFilterSensitiveLog(obj.vectorSearchConfiguration),
+    }),
+});
+export const KnowledgeBaseRetrieveAndGenerateConfigurationFilterSensitiveLog = (obj) => ({
+    ...obj,
+    ...(obj.retrievalConfiguration && {
+        retrievalConfiguration: KnowledgeBaseRetrievalConfigurationFilterSensitiveLog(obj.retrievalConfiguration),
+    }),
+    ...(obj.generationConfiguration && {
+        generationConfiguration: GenerationConfigurationFilterSensitiveLog(obj.generationConfiguration),
+    }),
+});
+export const RetrieveRequestFilterSensitiveLog = (obj) => ({
+    ...obj,
+    ...(obj.retrievalQuery && { retrievalQuery: SENSITIVE_STRING }),
+    ...(obj.retrievalConfiguration && {
+        retrievalConfiguration: KnowledgeBaseRetrievalConfigurationFilterSensitiveLog(obj.retrievalConfiguration),
+    }),
+});
+export const RetrieveAndGenerateConfigurationFilterSensitiveLog = (obj) => ({
+    ...obj,
+    ...(obj.knowledgeBaseConfiguration && {
+        knowledgeBaseConfiguration: KnowledgeBaseRetrieveAndGenerateConfigurationFilterSensitiveLog(obj.knowledgeBaseConfiguration),
+    }),
+});
+export const RetrieveAndGenerateRequestFilterSensitiveLog = (obj) => ({
+    ...obj,
+    ...(obj.input && { input: SENSITIVE_STRING }),
+    ...(obj.retrieveAndGenerateConfiguration && {
+        retrieveAndGenerateConfiguration: RetrieveAndGenerateConfigurationFilterSensitiveLog(obj.retrieveAndGenerateConfiguration),
+    }),
+});