@aws-sdk/client-bedrock-agent 3.702.0 → 3.705.0

package/dist-es/protocols/Aws_restJson1.js

@@ -4,6 +4,25 @@ import { _json, collectBody, decorateServiceException as __decorateServiceExcept
 import { v4 as generateIdempotencyToken } from "uuid";
 import { BedrockAgentServiceException as __BaseException } from "../models/BedrockAgentServiceException";
 import { AccessDeniedException, ConflictException, FlowNodeConfiguration, InternalServerException, PromptFlowNodeSourceConfiguration, PromptInferenceConfiguration, PromptTemplateConfiguration, ResourceNotFoundException, ServiceQuotaExceededException, ThrottlingException, Tool, ToolInputSchema, ValidationException, } from "../models/models_0";
+export const se_AssociateAgentCollaboratorCommand = async (input, context) => {
+    const b = rb(input, context);
+    const headers = {
+        "content-type": "application/json",
+    };
+    b.bp("/agents/{agentId}/agentversions/{agentVersion}/agentcollaborators");
+    b.p("agentId", () => input.agentId, "{agentId}", false);
+    b.p("agentVersion", () => input.agentVersion, "{agentVersion}", false);
+    let body;
+    body = JSON.stringify(take(input, {
+        agentDescriptor: (_) => _json(_),
+        clientToken: [true, (_) => _ ?? generateIdempotencyToken()],
+        collaborationInstruction: [],
+        collaboratorName: [],
+        relayConversationHistory: [],
+    }));
+    b.m("PUT").h(headers).b(body);
+    return b.build();
+};
 export const se_AssociateAgentKnowledgeBaseCommand = async (input, context) => {
     const b = rb(input, context);
     const headers = {
@@ -29,6 +48,7 @@ export const se_CreateAgentCommand = async (input, context) => {
     b.bp("/agents");
     let body;
     body = JSON.stringify(take(input, {
+        agentCollaboration: [],
         agentName: [],
         agentResourceRoleArn: [],
         clientToken: [true, (_) => _ ?? generateIdempotencyToken()],
@@ -316,6 +336,22 @@ export const se_DeleteKnowledgeBaseCommand = async (input, context) => {
     b.m("DELETE").h(headers).b(body);
     return b.build();
 };
+export const se_DeleteKnowledgeBaseDocumentsCommand = async (input, context) => {
+    const b = rb(input, context);
+    const headers = {
+        "content-type": "application/json",
+    };
+    b.bp("/knowledgebases/{knowledgeBaseId}/datasources/{dataSourceId}/documents/deleteDocuments");
+    b.p("knowledgeBaseId", () => input.knowledgeBaseId, "{knowledgeBaseId}", false);
+    b.p("dataSourceId", () => input.dataSourceId, "{dataSourceId}", false);
+    let body;
+    body = JSON.stringify(take(input, {
+        clientToken: [true, (_) => _ ?? generateIdempotencyToken()],
+        documentIdentifiers: (_) => _json(_),
+    }));
+    b.m("POST").h(headers).b(body);
+    return b.build();
+};
 export const se_DeletePromptCommand = async (input, context) => {
     const b = rb(input, context);
     const headers = {};
@@ -328,6 +364,17 @@ export const se_DeletePromptCommand = async (input, context) => {
     b.m("DELETE").h(headers).q(query).b(body);
     return b.build();
 };
+export const se_DisassociateAgentCollaboratorCommand = async (input, context) => {
+    const b = rb(input, context);
+    const headers = {};
+    b.bp("/agents/{agentId}/agentversions/{agentVersion}/agentcollaborators/{collaboratorId}");
+    b.p("agentId", () => input.agentId, "{agentId}", false);
+    b.p("agentVersion", () => input.agentVersion, "{agentVersion}", false);
+    b.p("collaboratorId", () => input.collaboratorId, "{collaboratorId}", false);
+    let body;
+    b.m("DELETE").h(headers).b(body);
+    return b.build();
+};
 export const se_DisassociateAgentKnowledgeBaseCommand = async (input, context) => {
     const b = rb(input, context);
     const headers = {};
@@ -369,6 +416,17 @@ export const se_GetAgentAliasCommand = async (input, context) => {
     b.m("GET").h(headers).b(body);
     return b.build();
 };
+export const se_GetAgentCollaboratorCommand = async (input, context) => {
+    const b = rb(input, context);
+    const headers = {};
+    b.bp("/agents/{agentId}/agentversions/{agentVersion}/agentcollaborators/{collaboratorId}");
+    b.p("agentId", () => input.agentId, "{agentId}", false);
+    b.p("agentVersion", () => input.agentVersion, "{agentVersion}", false);
+    b.p("collaboratorId", () => input.collaboratorId, "{collaboratorId}", false);
+    let body;
+    b.m("GET").h(headers).b(body);
+    return b.build();
+};
 export const se_GetAgentKnowledgeBaseCommand = async (input, context) => {
     const b = rb(input, context);
     const headers = {};
@@ -449,6 +507,21 @@ export const se_GetKnowledgeBaseCommand = async (input, context) => {
     b.m("GET").h(headers).b(body);
     return b.build();
 };
+export const se_GetKnowledgeBaseDocumentsCommand = async (input, context) => {
+    const b = rb(input, context);
+    const headers = {
+        "content-type": "application/json",
+    };
+    b.bp("/knowledgebases/{knowledgeBaseId}/datasources/{dataSourceId}/documents/getDocuments");
+    b.p("knowledgeBaseId", () => input.knowledgeBaseId, "{knowledgeBaseId}", false);
+    b.p("dataSourceId", () => input.dataSourceId, "{dataSourceId}", false);
+    let body;
+    body = JSON.stringify(take(input, {
+        documentIdentifiers: (_) => _json(_),
+    }));
+    b.m("POST").h(headers).b(body);
+    return b.build();
+};
 export const se_GetPromptCommand = async (input, context) => {
     const b = rb(input, context);
     const headers = {};
@@ -461,6 +534,22 @@ export const se_GetPromptCommand = async (input, context) => {
     b.m("GET").h(headers).q(query).b(body);
     return b.build();
 };
+export const se_IngestKnowledgeBaseDocumentsCommand = async (input, context) => {
+    const b = rb(input, context);
+    const headers = {
+        "content-type": "application/json",
+    };
+    b.bp("/knowledgebases/{knowledgeBaseId}/datasources/{dataSourceId}/documents");
+    b.p("knowledgeBaseId", () => input.knowledgeBaseId, "{knowledgeBaseId}", false);
+    b.p("dataSourceId", () => input.dataSourceId, "{dataSourceId}", false);
+    let body;
+    body = JSON.stringify(take(input, {
+        clientToken: [true, (_) => _ ?? generateIdempotencyToken()],
+        documents: (_) => se_KnowledgeBaseDocuments(_, context),
+    }));
+    b.m("PUT").h(headers).b(body);
+    return b.build();
+};
 export const se_ListAgentActionGroupsCommand = async (input, context) => {
     const b = rb(input, context);
     const headers = {
@@ -492,6 +581,22 @@ export const se_ListAgentAliasesCommand = async (input, context) => {
     b.m("POST").h(headers).b(body);
     return b.build();
 };
+export const se_ListAgentCollaboratorsCommand = async (input, context) => {
+    const b = rb(input, context);
+    const headers = {
+        "content-type": "application/json",
+    };
+    b.bp("/agents/{agentId}/agentversions/{agentVersion}/agentcollaborators");
+    b.p("agentId", () => input.agentId, "{agentId}", false);
+    b.p("agentVersion", () => input.agentVersion, "{agentVersion}", false);
+    let body;
+    body = JSON.stringify(take(input, {
+        maxResults: [],
+        nextToken: [],
+    }));
+    b.m("POST").h(headers).b(body);
+    return b.build();
+};
 export const se_ListAgentKnowledgeBasesCommand = async (input, context) => {
     const b = rb(input, context);
     const headers = {
@@ -608,6 +713,22 @@ export const se_ListIngestionJobsCommand = async (input, context) => {
     b.m("POST").h(headers).b(body);
     return b.build();
 };
+export const se_ListKnowledgeBaseDocumentsCommand = async (input, context) => {
+    const b = rb(input, context);
+    const headers = {
+        "content-type": "application/json",
+    };
+    b.bp("/knowledgebases/{knowledgeBaseId}/datasources/{dataSourceId}/documents");
+    b.p("knowledgeBaseId", () => input.knowledgeBaseId, "{knowledgeBaseId}", false);
+    b.p("dataSourceId", () => input.dataSourceId, "{dataSourceId}", false);
+    let body;
+    body = JSON.stringify(take(input, {
+        maxResults: [],
+        nextToken: [],
+    }));
+    b.m("POST").h(headers).b(body);
+    return b.build();
+};
 export const se_ListKnowledgeBasesCommand = async (input, context) => {
     const b = rb(input, context);
     const headers = {
@@ -724,6 +845,7 @@ export const se_UpdateAgentCommand = async (input, context) => {
     b.p("agentId", () => input.agentId, "{agentId}", false);
     let body;
     body = JSON.stringify(take(input, {
+        agentCollaboration: [],
         agentName: [],
         agentResourceRoleArn: [],
         customOrchestration: (_) => _json(_),
@@ -779,6 +901,25 @@ export const se_UpdateAgentAliasCommand = async (input, context) => {
     b.m("PUT").h(headers).b(body);
     return b.build();
 };
+export const se_UpdateAgentCollaboratorCommand = async (input, context) => {
+    const b = rb(input, context);
+    const headers = {
+        "content-type": "application/json",
+    };
+    b.bp("/agents/{agentId}/agentversions/{agentVersion}/agentcollaborators/{collaboratorId}");
+    b.p("agentId", () => input.agentId, "{agentId}", false);
+    b.p("agentVersion", () => input.agentVersion, "{agentVersion}", false);
+    b.p("collaboratorId", () => input.collaboratorId, "{collaboratorId}", false);
+    let body;
+    body = JSON.stringify(take(input, {
+        agentDescriptor: (_) => _json(_),
+        collaborationInstruction: [],
+        collaboratorName: [],
+        relayConversationHistory: [],
+    }));
+    b.m("PUT").h(headers).b(body);
+    return b.build();
+};
 export const se_UpdateAgentKnowledgeBaseCommand = async (input, context) => {
     const b = rb(input, context);
     const headers = {
@@ -900,6 +1041,20 @@ export const se_ValidateFlowDefinitionCommand = async (input, context) => {
     b.m("POST").h(headers).b(body);
     return b.build();
 };
+export const de_AssociateAgentCollaboratorCommand = 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, {
+        agentCollaborator: (_) => de_AgentCollaborator(_, context),
+    });
+    Object.assign(contents, doc);
+    return contents;
+};
 export const de_AssociateAgentKnowledgeBaseCommand = async (output, context) => {
     if (output.statusCode !== 200 && output.statusCode >= 300) {
         return de_CommandError(output, context);
@@ -1230,6 +1385,20 @@ export const de_DeleteKnowledgeBaseCommand = async (output, context) => {
     Object.assign(contents, doc);
     return contents;
 };
+export const de_DeleteKnowledgeBaseDocumentsCommand = async (output, context) => {
+    if (output.statusCode !== 202 && 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, {
+        documentDetails: (_) => de_KnowledgeBaseDocumentDetails(_, context),
+    });
+    Object.assign(contents, doc);
+    return contents;
+};
 export const de_DeletePromptCommand = async (output, context) => {
     if (output.statusCode !== 200 && output.statusCode >= 300) {
         return de_CommandError(output, context);
@@ -1245,6 +1414,16 @@ export const de_DeletePromptCommand = async (output, context) => {
     Object.assign(contents, doc);
     return contents;
 };
+export const de_DisassociateAgentCollaboratorCommand = async (output, context) => {
+    if (output.statusCode !== 204 && output.statusCode >= 300) {
+        return de_CommandError(output, context);
+    }
+    const contents = map({
+        $metadata: deserializeMetadata(output),
+    });
+    await collectBody(output.body, context);
+    return contents;
+};
 export const de_DisassociateAgentKnowledgeBaseCommand = async (output, context) => {
     if (output.statusCode !== 204 && output.statusCode >= 300) {
         return de_CommandError(output, context);
@@ -1297,6 +1476,20 @@ export const de_GetAgentAliasCommand = async (output, context) => {
     Object.assign(contents, doc);
     return contents;
 };
+export const de_GetAgentCollaboratorCommand = 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, {
+        agentCollaborator: (_) => de_AgentCollaborator(_, context),
+    });
+    Object.assign(contents, doc);
+    return contents;
+};
 export const de_GetAgentKnowledgeBaseCommand = async (output, context) => {
     if (output.statusCode !== 200 && output.statusCode >= 300) {
         return de_CommandError(output, context);
@@ -1436,6 +1629,20 @@ export const de_GetKnowledgeBaseCommand = async (output, context) => {
     Object.assign(contents, doc);
     return contents;
 };
+export const de_GetKnowledgeBaseDocumentsCommand = 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, {
+        documentDetails: (_) => de_KnowledgeBaseDocumentDetails(_, context),
+    });
+    Object.assign(contents, doc);
+    return contents;
+};
 export const de_GetPromptCommand = async (output, context) => {
     if (output.statusCode !== 200 && output.statusCode >= 300) {
         return de_CommandError(output, context);
@@ -1459,6 +1666,20 @@ export const de_GetPromptCommand = async (output, context) => {
     Object.assign(contents, doc);
     return contents;
 };
+export const de_IngestKnowledgeBaseDocumentsCommand = async (output, context) => {
+    if (output.statusCode !== 202 && 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, {
+        documentDetails: (_) => de_KnowledgeBaseDocumentDetails(_, context),
+    });
+    Object.assign(contents, doc);
+    return contents;
+};
 export const de_ListAgentActionGroupsCommand = async (output, context) => {
     if (output.statusCode !== 200 && output.statusCode >= 300) {
         return de_CommandError(output, context);
@@ -1489,6 +1710,21 @@ export const de_ListAgentAliasesCommand = async (output, context) => {
     Object.assign(contents, doc);
     return contents;
 };
+export const de_ListAgentCollaboratorsCommand = 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, {
+        agentCollaboratorSummaries: (_) => de_AgentCollaboratorSummaries(_, context),
+        nextToken: __expectString,
+    });
+    Object.assign(contents, doc);
+    return contents;
+};
 export const de_ListAgentKnowledgeBasesCommand = async (output, context) => {
     if (output.statusCode !== 200 && output.statusCode >= 300) {
         return de_CommandError(output, context);
@@ -1609,6 +1845,21 @@ export const de_ListIngestionJobsCommand = async (output, context) => {
     Object.assign(contents, doc);
     return contents;
 };
+export const de_ListKnowledgeBaseDocumentsCommand = 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, {
+        documentDetails: (_) => de_KnowledgeBaseDocumentDetails(_, context),
+        nextToken: __expectString,
+    });
+    Object.assign(contents, doc);
+    return contents;
+};
 export const de_ListKnowledgeBasesCommand = async (output, context) => {
     if (output.statusCode !== 200 && output.statusCode >= 300) {
         return de_CommandError(output, context);
@@ -1775,6 +2026,20 @@ export const de_UpdateAgentAliasCommand = async (output, context) => {
     Object.assign(contents, doc);
     return contents;
 };
+export const de_UpdateAgentCollaboratorCommand = 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, {
+        agentCollaborator: (_) => de_AgentCollaborator(_, context),
+    });
+    Object.assign(contents, doc);
+    return contents;
+};
 export const de_UpdateAgentKnowledgeBaseCommand = async (output, context) => {
     if (output.statusCode !== 200 && output.statusCode >= 300) {
         return de_CommandError(output, context);
@@ -2029,6 +2294,12 @@ const de_ValidationExceptionRes = async (parsedOutput, context) => {
     });
     return __decorateServiceException(exception, parsedOutput.body);
 };
+const se_ByteContentDoc = (input, context) => {
+    return take(input, {
+        data: context.base64Encoder,
+        mimeType: [],
+    });
+};
 const se_ChatPromptTemplateConfiguration = (input, context) => {
     return take(input, {
         inputVariables: _json,
@@ -2037,6 +2308,28 @@ const se_ChatPromptTemplateConfiguration = (input, context) => {
         toolConfiguration: (_) => se_ToolConfiguration(_, context),
     });
 };
+const se_CustomContent = (input, context) => {
+    return take(input, {
+        customDocumentIdentifier: _json,
+        inlineContent: (_) => se_InlineContent(_, context),
+        s3Location: _json,
+        sourceType: [],
+    });
+};
+const se_DocumentContent = (input, context) => {
+    return take(input, {
+        custom: (_) => se_CustomContent(_, context),
+        dataSourceType: [],
+        s3: _json,
+    });
+};
+const se_DocumentMetadata = (input, context) => {
+    return take(input, {
+        inlineAttributes: (_) => se_MetadataAttributes(_, context),
+        s3Location: _json,
+        type: [],
+    });
+};
 const se_FlowDefinition = (input, context) => {
     return take(input, {
         connections: _json,
@@ -2085,9 +2378,52 @@ const se_InferenceConfiguration = (input, context) => {
         topP: __serializeFloat,
     });
 };
+const se_InlineContent = (input, context) => {
+    return take(input, {
+        byteContent: (_) => se_ByteContentDoc(_, context),
+        textContent: _json,
+        type: [],
+    });
+};
+const se_KnowledgeBaseDocument = (input, context) => {
+    return take(input, {
+        content: (_) => se_DocumentContent(_, context),
+        metadata: (_) => se_DocumentMetadata(_, context),
+    });
+};
+const se_KnowledgeBaseDocuments = (input, context) => {
+    return input
+        .filter((e) => e != null)
+        .map((entry) => {
+        return se_KnowledgeBaseDocument(entry, context);
+    });
+};
+const se_MetadataAttribute = (input, context) => {
+    return take(input, {
+        key: [],
+        value: (_) => se_MetadataAttributeValue(_, context),
+    });
+};
+const se_MetadataAttributes = (input, context) => {
+    return input
+        .filter((e) => e != null)
+        .map((entry) => {
+        return se_MetadataAttribute(entry, context);
+    });
+};
+const se_MetadataAttributeValue = (input, context) => {
+    return take(input, {
+        booleanValue: [],
+        numberValue: __serializeFloat,
+        stringListValue: _json,
+        stringValue: [],
+        type: [],
+    });
+};
 const se_PromptConfiguration = (input, context) => {
     return take(input, {
         basePromptTemplate: [],
+        foundationModel: [],
         inferenceConfiguration: (_) => se_InferenceConfiguration(_, context),
         parserMode: [],
         promptCreationMode: [],
@@ -2225,6 +2561,7 @@ const de_ActionGroupSummary = (output, context) => {
 const de_Agent = (output, context) => {
     return take(output, {
         agentArn: __expectString,
+        agentCollaboration: __expectString,
         agentId: __expectString,
         agentName: __expectString,
         agentResourceRoleArn: __expectString,
@@ -2315,6 +2652,41 @@ const de_AgentAliasSummary = (output, context) => {
         updatedAt: (_) => __expectNonNull(__parseRfc3339DateTimeWithOffset(_)),
     });
 };
+const de_AgentCollaborator = (output, context) => {
+    return take(output, {
+        agentDescriptor: _json,
+        agentId: __expectString,
+        agentVersion: __expectString,
+        clientToken: __expectString,
+        collaborationInstruction: __expectString,
+        collaboratorId: __expectString,
+        collaboratorName: __expectString,
+        createdAt: (_) => __expectNonNull(__parseRfc3339DateTimeWithOffset(_)),
+        lastUpdatedAt: (_) => __expectNonNull(__parseRfc3339DateTimeWithOffset(_)),
+        relayConversationHistory: __expectString,
+    });
+};
+const de_AgentCollaboratorSummaries = (output, context) => {
+    const retVal = (output || [])
+        .filter((e) => e != null)
+        .map((entry) => {
+        return de_AgentCollaboratorSummary(entry, context);
+    });
+    return retVal;
+};
+const de_AgentCollaboratorSummary = (output, context) => {
+    return take(output, {
+        agentDescriptor: _json,
+        agentId: __expectString,
+        agentVersion: __expectString,
+        collaborationInstruction: __expectString,
+        collaboratorId: __expectString,
+        collaboratorName: __expectString,
+        createdAt: (_) => __expectNonNull(__parseRfc3339DateTimeWithOffset(_)),
+        lastUpdatedAt: (_) => __expectNonNull(__parseRfc3339DateTimeWithOffset(_)),
+        relayConversationHistory: __expectString,
+    });
+};
 const de_AgentKnowledgeBase = (output, context) => {
     return take(output, {
         agentId: __expectString,
@@ -2364,6 +2736,7 @@ const de_AgentSummary = (output, context) => {
 const de_AgentVersion = (output, context) => {
     return take(output, {
         agentArn: __expectString,
+        agentCollaboration: __expectString,
         agentId: __expectString,
         agentName: __expectString,
         agentResourceRoleArn: __expectString,
@@ -2644,6 +3017,24 @@ const de_KnowledgeBase = (output, context) => {
         updatedAt: (_) => __expectNonNull(__parseRfc3339DateTimeWithOffset(_)),
     });
 };
+const de_KnowledgeBaseDocumentDetail = (output, context) => {
+    return take(output, {
+        dataSourceId: __expectString,
+        identifier: _json,
+        knowledgeBaseId: __expectString,
+        status: __expectString,
+        statusReason: __expectString,
+        updatedAt: (_) => __expectNonNull(__parseRfc3339DateTimeWithOffset(_)),
+    });
+};
+const de_KnowledgeBaseDocumentDetails = (output, context) => {
+    const retVal = (output || [])
+        .filter((e) => e != null)
+        .map((entry) => {
+        return de_KnowledgeBaseDocumentDetail(entry, context);
+    });
+    return retVal;
+};
 const de_KnowledgeBaseSummaries = (output, context) => {
     const retVal = (output || [])
         .filter((e) => e != null)
@@ -2664,6 +3055,7 @@ const de_KnowledgeBaseSummary = (output, context) => {
 const de_PromptConfiguration = (output, context) => {
     return take(output, {
         basePromptTemplate: __expectString,
+        foundationModel: __expectString,
         inferenceConfiguration: (_) => de_InferenceConfiguration(_, context),
         parserMode: __expectString,
         promptCreationMode: __expectString,