@aws-sdk/client-bedrock-agent-runtime 3.895.0 → 3.898.0

package/dist-es/models/models_1.js

@@ -1,5 +1,20 @@
 import { SENSITIVE_STRING } from "@smithy/smithy-client";
-import { AgentActionGroupFilterSensitiveLog, CitationEventFilterSensitiveLog, CollaboratorConfigurationFilterSensitiveLog, ConversationHistoryFilterSensitiveLog, ExternalSourcesRetrieveAndGenerateConfigurationFilterSensitiveLog, GenerationConfigurationFilterSensitiveLog, ImplicitFilterConfigurationFilterSensitiveLog, InlineSessionStateFilterSensitiveLog, InputFileFilterSensitiveLog, InvocationResultMemberFilterSensitiveLog, OrchestrationConfigurationFilterSensitiveLog, VectorSearchRerankingConfigurationFilterSensitiveLog, } from "./models_0";
+import { AgentActionGroupFilterSensitiveLog, ByteContentDocFilterSensitiveLog, CitationFilterSensitiveLog, CollaboratorConfigurationFilterSensitiveLog, ConversationHistoryFilterSensitiveLog, ExternalSourcesGenerationConfigurationFilterSensitiveLog, GeneratedResponsePartFilterSensitiveLog, ImplicitFilterConfigurationFilterSensitiveLog, InlineSessionStateFilterSensitiveLog, InputFileFilterSensitiveLog, InvocationResultMemberFilterSensitiveLog, PromptTemplateFilterSensitiveLog, RetrievedReferenceFilterSensitiveLog, VectorSearchRerankingConfigurationFilterSensitiveLog, } from "./models_0";
+export const ExternalSourceType = {
+    BYTE_CONTENT: "BYTE_CONTENT",
+    S3: "S3",
+};
+export const QueryTransformationType = {
+    QUERY_DECOMPOSITION: "QUERY_DECOMPOSITION",
+};
+export const RetrieveAndGenerateType = {
+    EXTERNAL_SOURCES: "EXTERNAL_SOURCES",
+    KNOWLEDGE_BASE: "KNOWLEDGE_BASE",
+};
+export const GuadrailAction = {
+    INTERVENED: "INTERVENED",
+    NONE: "NONE",
+};
 export var RetrieveAndGenerateStreamResponseOutput;
 (function (RetrieveAndGenerateStreamResponseOutput) {
     RetrieveAndGenerateStreamResponseOutput.visit = (value, visitor) => {
@@ -101,6 +116,43 @@ export var RetrievalFilter;
         return visitor._(value.$unknown[0], value.$unknown[1]);
     };
 })(RetrievalFilter || (RetrievalFilter = {}));
+export const ExternalSourceFilterSensitiveLog = (obj) => ({
+    ...obj,
+    ...(obj.byteContent && { byteContent: ByteContentDocFilterSensitiveLog(obj.byteContent) }),
+});
+export const ExternalSourcesRetrieveAndGenerateConfigurationFilterSensitiveLog = (obj) => ({
+    ...obj,
+    ...(obj.sources && { sources: obj.sources.map((item) => ExternalSourceFilterSensitiveLog(item)) }),
+    ...(obj.generationConfiguration && {
+        generationConfiguration: ExternalSourcesGenerationConfigurationFilterSensitiveLog(obj.generationConfiguration),
+    }),
+});
+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,
+});
+export const RetrieveAndGenerateResponseFilterSensitiveLog = (obj) => ({
+    ...obj,
+    ...(obj.output && { output: SENSITIVE_STRING }),
+    ...(obj.citations && { citations: obj.citations.map((item) => CitationFilterSensitiveLog(item)) }),
+});
+export const CitationEventFilterSensitiveLog = (obj) => ({
+    ...obj,
+    ...(obj.citation && { citation: CitationFilterSensitiveLog(obj.citation) }),
+    ...(obj.generatedResponsePart && {
+        generatedResponsePart: GeneratedResponsePartFilterSensitiveLog(obj.generatedResponsePart),
+    }),
+    ...(obj.retrievedReferences && {
+        retrievedReferences: obj.retrievedReferences.map((item) => RetrievedReferenceFilterSensitiveLog(item)),
+    }),
+});
 export const RetrieveAndGenerateOutputEventFilterSensitiveLog = (obj) => ({
     ...obj,
 });

package/dist-es/protocols/Aws_restJson1.js

@@ -2080,6 +2080,14 @@ const de_AgentCollaboratorInvocationOutput = (output, context) => {
         output: _json,
     });
 };
+const de_AgentTraces = (output, context) => {
+    const retVal = (output || [])
+        .filter((e) => e != null)
+        .map((entry) => {
+        return de_TracePart(entry, context);
+    });
+    return retVal;
+};
 const de_ApiResult = (output, context) => {
     return take(output, {
         actionGroup: __expectString,
@@ -2207,6 +2215,16 @@ const de_FlowExecutionEvent = (output, context) => {
             flowOutputEvent: de_FlowExecutionOutputEvent(output.flowOutputEvent, context),
         };
     }
+    if (output.nodeActionEvent != null) {
+        return {
+            nodeActionEvent: de_NodeActionEvent(output.nodeActionEvent, context),
+        };
+    }
+    if (output.nodeDependencyEvent != null) {
+        return {
+            nodeDependencyEvent: de_NodeDependencyEvent(output.nodeDependencyEvent, context),
+        };
+    }
     if (output.nodeFailureEvent != null) {
         return {
             nodeFailureEvent: de_NodeFailureEvent(output.nodeFailureEvent, context),
@@ -2341,6 +2359,11 @@ const de_FlowTrace = (output, context) => {
             nodeActionTrace: de_FlowTraceNodeActionEvent(output.nodeActionTrace, context),
         };
     }
+    if (output.nodeDependencyTrace != null) {
+        return {
+            nodeDependencyTrace: de_FlowTraceDependencyEvent(output.nodeDependencyTrace, context),
+        };
+    }
     if (output.nodeInputTrace != null) {
         return {
             nodeInputTrace: de_FlowTraceNodeInputEvent(output.nodeInputTrace, context),
@@ -2360,6 +2383,13 @@ const de_FlowTraceConditionNodeResultEvent = (output, context) => {
         timestamp: (_) => __expectNonNull(__parseRfc3339DateTimeWithOffset(_)),
     });
 };
+const de_FlowTraceDependencyEvent = (output, context) => {
+    return take(output, {
+        nodeName: __expectString,
+        timestamp: (_) => __expectNonNull(__parseRfc3339DateTimeWithOffset(_)),
+        traceElements: (_) => de_TraceElements(__expectUnion(_), context),
+    });
+};
 const de_FlowTraceEvent = (output, context) => {
     return take(output, {
         trace: (_) => de_FlowTrace(__expectUnion(_), context),
@@ -2369,6 +2399,8 @@ const de_FlowTraceNodeActionEvent = (output, context) => {
     return take(output, {
         nodeName: __expectString,
         operationName: __expectString,
+        operationRequest: (_) => de_Document(_, context),
+        operationResponse: (_) => de_Document(_, context),
         requestId: __expectString,
         serviceName: __expectString,
         timestamp: (_) => __expectNonNull(__parseRfc3339DateTimeWithOffset(_)),
@@ -2391,8 +2423,12 @@ const de_FlowTraceNodeInputEvent = (output, context) => {
 };
 const de_FlowTraceNodeInputField = (output, context) => {
     return take(output, {
+        category: __expectString,
         content: (_) => de_FlowTraceNodeInputContent(__expectUnion(_), context),
+        executionChain: _json,
         nodeInputName: __expectString,
+        source: _json,
+        type: __expectString,
     });
 };
 const de_FlowTraceNodeInputFields = (output, context) => {
@@ -2421,7 +2457,9 @@ const de_FlowTraceNodeOutputEvent = (output, context) => {
 const de_FlowTraceNodeOutputField = (output, context) => {
     return take(output, {
         content: (_) => de_FlowTraceNodeOutputContent(__expectUnion(_), context),
+        next: _json,
         nodeOutputName: __expectString,
+        type: __expectString,
     });
 };
 const de_FlowTraceNodeOutputFields = (output, context) => {
@@ -2661,6 +2699,24 @@ const de_ModelInvocationInput = (output, context) => {
         type: __expectString,
     });
 };
+const de_NodeActionEvent = (output, context) => {
+    return take(output, {
+        nodeName: __expectString,
+        operationName: __expectString,
+        operationRequest: (_) => de_Document(_, context),
+        operationResponse: (_) => de_Document(_, context),
+        requestId: __expectString,
+        serviceName: __expectString,
+        timestamp: (_) => __expectNonNull(__parseRfc3339DateTimeWithOffset(_)),
+    });
+};
+const de_NodeDependencyEvent = (output, context) => {
+    return take(output, {
+        nodeName: __expectString,
+        timestamp: (_) => __expectNonNull(__parseRfc3339DateTimeWithOffset(_)),
+        traceElements: (_) => de_NodeTraceElements(__expectUnion(_), context),
+    });
+};
 const de_NodeExecutionContent = (output, context) => {
     if (output.document != null) {
         return {
@@ -2686,8 +2742,12 @@ const de_NodeInputEvent = (output, context) => {
 };
 const de_NodeInputField = (output, context) => {
     return take(output, {
+        category: __expectString,
         content: (_) => de_NodeExecutionContent(__expectUnion(_), context),
+        executionChain: _json,
         name: __expectString,
+        source: _json,
+        type: __expectString,
     });
 };
 const de_NodeInputFields = (output, context) => {
@@ -2709,6 +2769,8 @@ const de_NodeOutputField = (output, context) => {
     return take(output, {
         content: (_) => de_NodeExecutionContent(__expectUnion(_), context),
         name: __expectString,
+        next: _json,
+        type: __expectString,
     });
 };
 const de_NodeOutputFields = (output, context) => {
@@ -2719,6 +2781,14 @@ const de_NodeOutputFields = (output, context) => {
     });
     return retVal;
 };
+const de_NodeTraceElements = (output, context) => {
+    if (output.agentTraces != null) {
+        return {
+            agentTraces: de_AgentTraces(output.agentTraces, context),
+        };
+    }
+    return { $unknown: Object.entries(output)[0] };
+};
 const de_Observation = (output, context) => {
     return take(output, {
         actionGroupInvocationOutput: (_) => de_ActionGroupInvocationOutput(_, context),
@@ -3002,6 +3072,14 @@ const de_Trace = (output, context) => {
     }
     return { $unknown: Object.entries(output)[0] };
 };
+const de_TraceElements = (output, context) => {
+    if (output.agentTraces != null) {
+        return {
+            agentTraces: de_AgentTraces(output.agentTraces, context),
+        };
+    }
+    return { $unknown: Object.entries(output)[0] };
+};
 const de_TracePart = (output, context) => {
     return take(output, {
         agentAliasId: __expectString,