@aws-sdk/client-bedrock-agent-runtime 3.675.0 → 3.676.0

package/dist-cjs/index.js

@@ -114,6 +114,7 @@ __export(src_exports, {
   ModelInvocationInputFilterSensitiveLog: () => ModelInvocationInputFilterSensitiveLog,
   NodeType: () => NodeType,
   ObservationFilterSensitiveLog: () => ObservationFilterSensitiveLog,
+  OrchestrationConfigurationFilterSensitiveLog: () => OrchestrationConfigurationFilterSensitiveLog,
   OrchestrationModelInvocationOutputFilterSensitiveLog: () => OrchestrationModelInvocationOutputFilterSensitiveLog,
   OrchestrationTrace: () => OrchestrationTrace,
   OrchestrationTraceFilterSensitiveLog: () => OrchestrationTraceFilterSensitiveLog,
@@ -1243,6 +1244,10 @@ var GenerationConfigurationFilterSensitiveLog = /* @__PURE__ */ __name((obj) =>
   ...obj,
   ...obj.promptTemplate && { promptTemplate: PromptTemplateFilterSensitiveLog(obj.promptTemplate) }
 }), "GenerationConfigurationFilterSensitiveLog");
+var OrchestrationConfigurationFilterSensitiveLog = /* @__PURE__ */ __name((obj) => ({
+  ...obj,
+  ...obj.promptTemplate && { promptTemplate: PromptTemplateFilterSensitiveLog(obj.promptTemplate) }
+}), "OrchestrationConfigurationFilterSensitiveLog");
 var RetrieveAndGenerateOutputFilterSensitiveLog = /* @__PURE__ */ __name((obj) => ({
   ...obj
 }), "RetrieveAndGenerateOutputFilterSensitiveLog");
@@ -1317,6 +1322,9 @@ var KnowledgeBaseRetrieveAndGenerateConfigurationFilterSensitiveLog = /* @__PURE
   },
   ...obj.generationConfiguration && {
     generationConfiguration: GenerationConfigurationFilterSensitiveLog(obj.generationConfiguration)
+  },
+  ...obj.orchestrationConfiguration && {
+    orchestrationConfiguration: OrchestrationConfigurationFilterSensitiveLog(obj.orchestrationConfiguration)
   }
 }), "KnowledgeBaseRetrieveAndGenerateConfigurationFilterSensitiveLog");
 var RetrieveRequestFilterSensitiveLog = /* @__PURE__ */ __name((obj) => ({
@@ -2097,7 +2105,7 @@ var se_KnowledgeBaseRetrieveAndGenerateConfiguration = /* @__PURE__ */ __name((i
     generationConfiguration: (_) => se_GenerationConfiguration(_, context),
     knowledgeBaseId: [],
     modelArn: [],
-    orchestrationConfiguration: import_smithy_client._json,
+    orchestrationConfiguration: (_) => se_OrchestrationConfiguration(_, context),
     retrievalConfiguration: (_) => se_KnowledgeBaseRetrievalConfiguration(_, context)
   });
 }, "se_KnowledgeBaseRetrieveAndGenerateConfiguration");
@@ -2108,6 +2116,14 @@ var se_KnowledgeBaseVectorSearchConfiguration = /* @__PURE__ */ __name((input, c
     overrideSearchType: []
   });
 }, "se_KnowledgeBaseVectorSearchConfiguration");
+var se_OrchestrationConfiguration = /* @__PURE__ */ __name((input, context) => {
+  return (0, import_smithy_client.take)(input, {
+    additionalModelRequestFields: (_) => se_AdditionalModelRequestFields(_, context),
+    inferenceConfig: (_) => se_InferenceConfig(_, context),
+    promptTemplate: import_smithy_client._json,
+    queryTransformationConfiguration: import_smithy_client._json
+  });
+}, "se_OrchestrationConfiguration");
 var se_RetrievalFilter = /* @__PURE__ */ __name((input, context) => {
   return RetrievalFilter.visit(input, {
     andAll: (value) => ({ andAll: se_RetrievalFilterList(value, context) }),
@@ -2677,6 +2693,7 @@ var paginateRetrieve = (0, import_core.createPaginator)(BedrockAgentRuntimeClien
   ExternalSourceFilterSensitiveLog,
   ExternalSourcesRetrieveAndGenerateConfigurationFilterSensitiveLog,
   GenerationConfigurationFilterSensitiveLog,
+  OrchestrationConfigurationFilterSensitiveLog,
   RetrieveAndGenerateOutputFilterSensitiveLog,
   RetrieveAndGenerateResponseFilterSensitiveLog,
   KnowledgeBaseQueryFilterSensitiveLog,

package/dist-es/models/models_0.js

@@ -858,6 +858,10 @@ export const GenerationConfigurationFilterSensitiveLog = (obj) => ({
     ...obj,
     ...(obj.promptTemplate && { promptTemplate: PromptTemplateFilterSensitiveLog(obj.promptTemplate) }),
 });
+export const OrchestrationConfigurationFilterSensitiveLog = (obj) => ({
+    ...obj,
+    ...(obj.promptTemplate && { promptTemplate: PromptTemplateFilterSensitiveLog(obj.promptTemplate) }),
+});
 export const RetrieveAndGenerateOutputFilterSensitiveLog = (obj) => ({
     ...obj,
 });
@@ -933,6 +937,9 @@ export const KnowledgeBaseRetrieveAndGenerateConfigurationFilterSensitiveLog = (
     ...(obj.generationConfiguration && {
         generationConfiguration: GenerationConfigurationFilterSensitiveLog(obj.generationConfiguration),
     }),
+    ...(obj.orchestrationConfiguration && {
+        orchestrationConfiguration: OrchestrationConfigurationFilterSensitiveLog(obj.orchestrationConfiguration),
+    }),
 });
 export const RetrieveRequestFilterSensitiveLog = (obj) => ({
     ...obj,

package/dist-es/protocols/Aws_restJson1.js

@@ -713,7 +713,7 @@ const se_KnowledgeBaseRetrieveAndGenerateConfiguration = (input, context) => {
         generationConfiguration: (_) => se_GenerationConfiguration(_, context),
         knowledgeBaseId: [],
         modelArn: [],
-        orchestrationConfiguration: _json,
+        orchestrationConfiguration: (_) => se_OrchestrationConfiguration(_, context),
         retrievalConfiguration: (_) => se_KnowledgeBaseRetrievalConfiguration(_, context),
     });
 };
@@ -724,6 +724,14 @@ const se_KnowledgeBaseVectorSearchConfiguration = (input, context) => {
         overrideSearchType: [],
     });
 };
+const se_OrchestrationConfiguration = (input, context) => {
+    return take(input, {
+        additionalModelRequestFields: (_) => se_AdditionalModelRequestFields(_, context),
+        inferenceConfig: (_) => se_InferenceConfig(_, context),
+        promptTemplate: _json,
+        queryTransformationConfiguration: _json,
+    });
+};
 const se_RetrievalFilter = (input, context) => {
     return RetrievalFilter.visit(input, {
         andAll: (value) => ({ andAll: se_RetrievalFilterList(value, context) }),

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

@@ -125,6 +125,22 @@ declare const RetrieveAndGenerateCommand_base: {
  *         },
  *       },
  *       orchestrationConfiguration: { // OrchestrationConfiguration
+ *         promptTemplate: {
+ *           textPromptTemplate: "STRING_VALUE",
+ *         },
+ *         inferenceConfig: {
+ *           textInferenceConfig: {
+ *             temperature: Number("float"),
+ *             topP: Number("float"),
+ *             maxTokens: Number("int"),
+ *             stopSequences: [
+ *               "STRING_VALUE",
+ *             ],
+ *           },
+ *         },
+ *         additionalModelRequestFields: {
+ *           "<keys>": "DOCUMENT_VALUE",
+ *         },
  *         queryTransformationConfiguration: { // QueryTransformationConfiguration
  *           type: "QUERY_DECOMPOSITION", // required
  *         },

package/dist-types/models/models_0.d.ts

@@ -2229,7 +2229,7 @@ export interface OrchestrationModelInvocationOutput {
      */
     rawResponse?: RawResponse;
     /**
-     * <p>Contains information about the foundation model output.</p>
+     * <p>Contains information about the foundation model output from the orchestration step.</p>
      * @public
      */
     metadata?: Metadata;
@@ -2533,12 +2533,16 @@ export interface PostProcessingModelInvocationOutput {
      */
     parsedResponse?: PostProcessingParsedResponse;
     /**
-     * <p>Contains the raw output from the foundation model.</p>
+     * <p>
+     *             Details of the raw response from the foundation model output.
+     *         </p>
      * @public
      */
     rawResponse?: RawResponse;
     /**
-     * <p>Provides details of the foundation model.</p>
+     * <p>
+     *             Contains information about the foundation model output from the post-processing step.
+     *         </p>
      * @public
      */
     metadata?: Metadata;
@@ -2628,12 +2632,16 @@ export interface PreProcessingModelInvocationOutput {
      */
     parsedResponse?: PreProcessingParsedResponse;
     /**
-     * <p>Contains the raw output from the foundation model.</p>
+     * <p>
+     *             Details of the raw response from the foundation model output.
+     *         </p>
      * @public
      */
     rawResponse?: RawResponse;
     /**
-     * <p>Provides details of the foundation model.</p>
+     * <p>
+     *             Contains information about the foundation model output from the pre-processing step.
+     *         </p>
      * @public
      */
     metadata?: Metadata;
@@ -3583,12 +3591,27 @@ export interface QueryTransformationConfiguration {
  * @public
  */
 export interface OrchestrationConfiguration {
+    /**
+     * <p>Contains the template for the prompt that's sent to the model for response generation.</p>
+     * @public
+     */
+    promptTemplate?: PromptTemplate;
+    /**
+     * <p> Configuration settings for inference when using RetrieveAndGenerate to generate responses while using a knowledge base as a source. </p>
+     * @public
+     */
+    inferenceConfig?: InferenceConfig;
+    /**
+     * <p> Additional model parameters and corresponding values not included in the textInferenceConfig structure for a knowledge base. This allows users to provide custom model parameters specific to the language model being used. </p>
+     * @public
+     */
+    additionalModelRequestFields?: Record<string, __DocumentType>;
     /**
      * <p>To split up the prompt and retrieve multiple sources, set the transformation type to
      *     <code>QUERY_DECOMPOSITION</code>.</p>
      * @public
      */
-    queryTransformationConfiguration: QueryTransformationConfiguration | undefined;
+    queryTransformationConfiguration?: QueryTransformationConfiguration;
 }
 /**
  * @public
@@ -4673,6 +4696,10 @@ export declare const ExternalSourcesRetrieveAndGenerateConfigurationFilterSensit
  * @internal
  */
 export declare const GenerationConfigurationFilterSensitiveLog: (obj: GenerationConfiguration) => any;
+/**
+ * @internal
+ */
+export declare const OrchestrationConfigurationFilterSensitiveLog: (obj: OrchestrationConfiguration) => any;
 /**
  * @internal
  */

package/dist-types/ts3.4/models/models_0.d.ts

@@ -1436,9 +1436,10 @@ export interface QueryTransformationConfiguration {
   type: QueryTransformationType | undefined;
 }
 export interface OrchestrationConfiguration {
-  queryTransformationConfiguration:
-    | QueryTransformationConfiguration
-    | undefined;
+  promptTemplate?: PromptTemplate;
+  inferenceConfig?: InferenceConfig;
+  additionalModelRequestFields?: Record<string, __DocumentType>;
+  queryTransformationConfiguration?: QueryTransformationConfiguration;
 }
 export declare const RetrieveAndGenerateType: {
   readonly EXTERNAL_SOURCES: "EXTERNAL_SOURCES";
@@ -1965,6 +1966,9 @@ export declare const ExternalSourcesRetrieveAndGenerateConfigurationFilterSensit
 export declare const GenerationConfigurationFilterSensitiveLog: (
   obj: GenerationConfiguration
 ) => any;
+export declare const OrchestrationConfigurationFilterSensitiveLog: (
+  obj: OrchestrationConfiguration
+) => any;
 export declare const RetrieveAndGenerateOutputFilterSensitiveLog: (
   obj: RetrieveAndGenerateOutput
 ) => any;

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "@aws-sdk/client-bedrock-agent-runtime",
   "description": "AWS SDK for JavaScript Bedrock Agent Runtime Client for Node.js, Browser and React Native",
-  "version": "3.675.0",
+  "version": "3.676.0",
   "scripts": {
     "build": "concurrently 'yarn:build:cjs' 'yarn:build:es' 'yarn:build:types'",
     "build:cjs": "node ../../scripts/compilation/inline client-bedrock-agent-runtime",