@aws/lsp-codewhisperer 0.0.96 → 0.0.98

package/node_modules/@amzn/codewhisperer-streaming/dist-cjs/protocols/Aws_restJson1.js

@@ -1,6 +1,6 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.de_SendMessageCommand = exports.de_GenerateTaskAssistPlanCommand = exports.de_GenerateAssistantResponseCommand = exports.de_ExportResultArchiveCommand = exports.se_SendMessageCommand = exports.se_GenerateTaskAssistPlanCommand = exports.se_GenerateAssistantResponseCommand = exports.se_ExportResultArchiveCommand = void 0;
+exports.de_SendMessageCommand = exports.de_InvokeMCPCommand = exports.de_GenerateTaskAssistPlanCommand = exports.de_GenerateAssistantResponseCommand = exports.de_ExportResultArchiveCommand = exports.se_SendMessageCommand = exports.se_InvokeMCPCommand = exports.se_GenerateTaskAssistPlanCommand = exports.se_GenerateAssistantResponseCommand = exports.se_ExportResultArchiveCommand = void 0;
 const CodeWhispererStreamingServiceException_1 = require("../models/CodeWhispererStreamingServiceException");
 const models_0_1 = require("../models/models_0");
 const core_1 = require("@aws-sdk/core");
@@ -27,9 +27,10 @@ const se_ExportResultArchiveCommand = async (input, context) => {
 exports.se_ExportResultArchiveCommand = se_ExportResultArchiveCommand;
 const se_GenerateAssistantResponseCommand = async (input, context) => {
     const b = (0, core_2.requestBuilder)(input, context);
-    const headers = {
+    const headers = (0, smithy_client_1.map)({}, smithy_client_1.isSerializableHeaderValue, {
         'content-type': 'application/json',
-    };
+        [_xakam]: input[_aM],
+    });
     b.bp("/generateAssistantResponse");
     let body;
     body = JSON.stringify((0, smithy_client_1.take)(input, {
@@ -60,6 +61,25 @@ const se_GenerateTaskAssistPlanCommand = async (input, context) => {
     return b.build();
 };
 exports.se_GenerateTaskAssistPlanCommand = se_GenerateTaskAssistPlanCommand;
+const se_InvokeMCPCommand = async (input, context) => {
+    const b = (0, core_2.requestBuilder)(input, context);
+    const headers = {
+        'content-type': 'application/json',
+    };
+    b.bp("/mcp");
+    let body;
+    body = JSON.stringify((0, smithy_client_1.take)(input, {
+        'id': _ => se_Document(_, context),
+        'jsonrpc': [],
+        'method': [],
+        'params': _ => se_SensitiveDocument(_, context),
+    }));
+    b.m("POST")
+        .h(headers)
+        .b(body);
+    return b.build();
+};
+exports.se_InvokeMCPCommand = se_InvokeMCPCommand;
 const se_SendMessageCommand = async (input, context) => {
     const b = (0, core_2.requestBuilder)(input, context);
     const headers = {
@@ -116,6 +136,24 @@ const de_GenerateTaskAssistPlanCommand = async (output, context) => {
     return contents;
 };
 exports.de_GenerateTaskAssistPlanCommand = de_GenerateTaskAssistPlanCommand;
+const de_InvokeMCPCommand = async (output, context) => {
+    if (output.statusCode !== 200 && output.statusCode >= 300) {
+        return de_CommandError(output, context);
+    }
+    const contents = (0, smithy_client_1.map)({
+        $metadata: deserializeMetadata(output),
+    });
+    const data = (0, smithy_client_1.expectNonNull)(((0, smithy_client_1.expectObject)(await (0, core_1.parseJsonBody)(output.body, context))), "body");
+    const doc = (0, smithy_client_1.take)(data, {
+        'error': _ => de_JsonRpcError(_, context),
+        'id': _ => de_Document(_, context),
+        'jsonrpc': smithy_client_1.expectString,
+        'result': _ => de_SensitiveDocument(_, context),
+    });
+    Object.assign(contents, doc);
+    return contents;
+};
+exports.de_InvokeMCPCommand = de_InvokeMCPCommand;
 const de_SendMessageCommand = async (output, context) => {
     if (output.statusCode !== 200 && output.statusCode >= 300) {
         return de_CommandError(output, context);
@@ -156,6 +194,9 @@ const de_CommandError = async (output, context) => {
         case "ServiceQuotaExceededException":
         case "com.amazon.aws.codewhisperer#ServiceQuotaExceededException":
             throw await de_ServiceQuotaExceededExceptionRes(parsedOutput, context);
+        case "ServiceUnavailableException":
+        case "com.amazon.aws.codewhisperer#ServiceUnavailableException":
+            throw await de_ServiceUnavailableExceptionRes(parsedOutput, context);
         case "DryRunOperationException":
         case "com.amazon.aws.codewhisperer#DryRunOperationException":
             throw await de_DryRunOperationExceptionRes(parsedOutput, context);
@@ -251,6 +292,19 @@ const de_ServiceQuotaExceededExceptionRes = async (parsedOutput, context) => {
     });
     return (0, smithy_client_1.decorateServiceException)(exception, parsedOutput.body);
 };
+const de_ServiceUnavailableExceptionRes = async (parsedOutput, context) => {
+    const contents = (0, smithy_client_1.map)({});
+    const data = parsedOutput.body;
+    const doc = (0, smithy_client_1.take)(data, {
+        'message': smithy_client_1.expectString,
+    });
+    Object.assign(contents, doc);
+    const exception = new models_0_1.ServiceUnavailableException({
+        $metadata: deserializeMetadata(parsedOutput),
+        ...contents
+    });
+    return (0, smithy_client_1.decorateServiceException)(exception, parsedOutput.body);
+};
 const de_ThrottlingExceptionRes = async (parsedOutput, context) => {
     const contents = (0, smithy_client_1.map)({});
     const data = parsedOutput.body;
@@ -291,6 +345,11 @@ const de_ChatResponseStream = (output, context) => {
                 assistantResponseEvent: await de_AssistantResponseEvent_event(event["assistantResponseEvent"], context),
             };
         }
+        if (event["reasoningContentEvent"] != null) {
+            return {
+                reasoningContentEvent: await de_ReasoningContentEvent_event(event["reasoningContentEvent"], context),
+            };
+        }
         if (event["dryRunSucceedEvent"] != null) {
             return {
                 dryRunSucceedEvent: await de_DryRunSucceedEvent_event(event["dryRunSucceedEvent"], context),
@@ -336,6 +395,21 @@ const de_ChatResponseStream = (output, context) => {
                 toolResultEvent: await de_ToolResultEvent_event(event["toolResultEvent"], context),
             };
         }
+        if (event["metadataEvent"] != null) {
+            return {
+                metadataEvent: await de_MetadataEvent_event(event["metadataEvent"], context),
+            };
+        }
+        if (event["meteringEvent"] != null) {
+            return {
+                meteringEvent: await de_MeteringEvent_event(event["meteringEvent"], context),
+            };
+        }
+        if (event["contextUsageEvent"] != null) {
+            return {
+                contextUsageEvent: await de_ContextUsageEvent_event(event["contextUsageEvent"], context),
+            };
+        }
         if (event["citationEvent"] != null) {
             return {
                 citationEvent: await de_CitationEvent_event(event["citationEvent"], context),
@@ -410,6 +484,12 @@ const de_CodeReferenceEvent_event = async (output, context) => {
     Object.assign(contents, (0, smithy_client_1._json)(data));
     return contents;
 };
+const de_ContextUsageEvent_event = async (output, context) => {
+    const contents = {};
+    const data = await (0, core_1.parseJsonBody)(output.body, context);
+    Object.assign(contents, de_ContextUsageEvent(data, context));
+    return contents;
+};
 const de_DryRunSucceedEvent_event = async (output, context) => {
     const contents = {};
     const data = await (0, core_1.parseJsonBody)(output.body, context);
@@ -447,6 +527,24 @@ const de_MessageMetadataEvent_event = async (output, context) => {
     Object.assign(contents, (0, smithy_client_1._json)(data));
     return contents;
 };
+const de_MetadataEvent_event = async (output, context) => {
+    const contents = {};
+    const data = await (0, core_1.parseJsonBody)(output.body, context);
+    Object.assign(contents, de_MetadataEvent(data, context));
+    return contents;
+};
+const de_MeteringEvent_event = async (output, context) => {
+    const contents = {};
+    const data = await (0, core_1.parseJsonBody)(output.body, context);
+    Object.assign(contents, de_MeteringEvent(data, context));
+    return contents;
+};
+const de_ReasoningContentEvent_event = async (output, context) => {
+    const contents = {};
+    const data = await (0, core_1.parseJsonBody)(output.body, context);
+    Object.assign(contents, de_ReasoningContentEvent(data, context));
+    return contents;
+};
 const de_SupplementaryWebLinksEvent_event = async (output, context) => {
     const contents = {};
     const data = await (0, core_1.parseJsonBody)(output.body, context);
@@ -473,9 +571,11 @@ const de_InteractionComponentsEvent_event = async (output, context) => {
 };
 const se_AssistantResponseMessage = (input, context) => {
     return (0, smithy_client_1.take)(input, {
+        'cachePoint': smithy_client_1._json,
         'content': [],
         'followupPrompt': smithy_client_1._json,
         'messageId': [],
+        'reasoningContent': _ => se_ReasoningContent(_, context),
         'references': smithy_client_1._json,
         'supplementaryWebLinks': smithy_client_1._json,
         'toolUses': _ => se_ToolUses(_, context),
@@ -495,6 +595,8 @@ const se_ChatMessage = (input, context) => {
 };
 const se_ConversationState = (input, context) => {
     return (0, smithy_client_1.take)(input, {
+        'agentContinuationId': [],
+        'agentTaskType': [],
         'chatTriggerType': [],
         'conversationId': [],
         'currentMessage': _ => se_ChatMessage(_, context),
@@ -520,11 +622,19 @@ const se_ImageSource = (input, context) => {
         _: (name, value) => ({ name: value })
     });
 };
+const se_ReasoningContent = (input, context) => {
+    return models_0_1.ReasoningContent.visit(input, {
+        reasoningText: value => ({ "reasoningText": (0, smithy_client_1._json)(value) }),
+        redactedContent: value => ({ "redactedContent": context.base64Encoder(value) }),
+        _: (name, value) => ({ name: value })
+    });
+};
 const se_SensitiveDocument = (input, context) => {
     return input;
 };
 const se_Tool = (input, context) => {
     return models_0_1.Tool.visit(input, {
+        cachePoint: value => ({ "cachePoint": (0, smithy_client_1._json)(value) }),
         toolSpecification: value => ({ "toolSpecification": se_ToolSpecification(value, context) }),
         _: (name, value) => ({ name: value })
     });
@@ -584,6 +694,8 @@ const se_ToolUses = (input, context) => {
 };
 const se_UserInputMessage = (input, context) => {
     return (0, smithy_client_1.take)(input, {
+        'cachePoint': smithy_client_1._json,
+        'clientCacheConfig': smithy_client_1._json,
         'content': [],
         'images': _ => se_ImageBlocks(_, context),
         'modelId': [],
@@ -607,11 +719,38 @@ const se_UserInputMessageContext = (input, context) => {
         'userSettings': smithy_client_1._json,
     });
 };
+const se_Document = (input, context) => {
+    return input;
+};
 const de_BinaryPayloadEvent = (output, context) => {
     return (0, smithy_client_1.take)(output, {
         'bytes': context.base64Decoder,
     });
 };
+const de_ContextUsageEvent = (output, context) => {
+    return (0, smithy_client_1.take)(output, {
+        'contextUsagePercentage': smithy_client_1.limitedParseFloat32,
+    });
+};
+const de_MetadataEvent = (output, context) => {
+    return (0, smithy_client_1.take)(output, {
+        'tokenUsage': (_) => de_TokenUsage(_, context),
+    });
+};
+const de_MeteringEvent = (output, context) => {
+    return (0, smithy_client_1.take)(output, {
+        'unit': smithy_client_1.expectString,
+        'unitPlural': smithy_client_1.expectString,
+        'usage': smithy_client_1.limitedParseDouble,
+    });
+};
+const de_ReasoningContentEvent = (output, context) => {
+    return (0, smithy_client_1.take)(output, {
+        'redactedContent': context.base64Decoder,
+        'signature': smithy_client_1.expectString,
+        'text': smithy_client_1.expectString,
+    });
+};
 const de_SensitiveDocument = (output, context) => {
     return output;
 };
@@ -644,6 +783,23 @@ const de_ToolResultEvent = (output, context) => {
         'toolResult': (_) => de_ToolResult(_, context),
     });
 };
+const de_TokenUsage = (output, context) => {
+    return (0, smithy_client_1.take)(output, {
+        'cacheReadInputTokens': smithy_client_1.expectInt32,
+        'cacheWriteInputTokens': smithy_client_1.expectInt32,
+        'contextUsagePercentage': smithy_client_1.limitedParseFloat32,
+        'outputTokens': smithy_client_1.expectInt32,
+        'totalTokens': smithy_client_1.expectInt32,
+        'uncachedInputTokens': smithy_client_1.expectInt32,
+    });
+};
+const de_JsonRpcError = (output, context) => {
+    return (0, smithy_client_1.take)(output, {
+        'code': smithy_client_1.expectInt32,
+        'data': (_) => de_Document(_, context),
+        'message': smithy_client_1.expectString,
+    });
+};
 const de_InteractionComponentsEvent = (output, context) => {
     return (0, smithy_client_1.take)(output, {
         'interactionComponentEntries': (_) => de_InteractionComponentEntryList(_, context),
@@ -677,6 +833,9 @@ const de_InteractionComponentEntryList = (output, context) => {
     });
     return retVal;
 };
+const de_Document = (output, context) => {
+    return output;
+};
 const deserializeMetadata = (output) => ({
     httpStatusCode: output.statusCode,
     requestId: output.headers["x-amzn-requestid"] ?? output.headers["x-amzn-request-id"] ?? output.headers["x-amz-request-id"],
@@ -684,5 +843,7 @@ const deserializeMetadata = (output) => ({
     cfId: output.headers["x-amz-cf-id"],
 });
 const collectBodyString = (streamBody, context) => (0, smithy_client_1.collectBody)(streamBody, context).then(body => context.utf8Encoder(body));
+const _aM = "agentMode";
 const _cI = "conversationId";
 const _xacci = "x-amzn-codewhisperer-conversation-id";
+const _xakam = "x-amzn-kiro-agent-mode";

package/node_modules/@amzn/codewhisperer-streaming/dist-es/CodeWhispererStreaming.js

@@ -2,12 +2,14 @@ import { CodeWhispererStreamingClient, } from "./CodeWhispererStreamingClient";
 import { ExportResultArchiveCommand, } from "./commands/ExportResultArchiveCommand";
 import { GenerateAssistantResponseCommand, } from "./commands/GenerateAssistantResponseCommand";
 import { GenerateTaskAssistPlanCommand, } from "./commands/GenerateTaskAssistPlanCommand";
+import { InvokeMCPCommand, } from "./commands/InvokeMCPCommand";
 import { SendMessageCommand, } from "./commands/SendMessageCommand";
 import { createAggregatedClient } from "@smithy/smithy-client";
 const commands = {
     ExportResultArchiveCommand,
     GenerateAssistantResponseCommand,
     GenerateTaskAssistPlanCommand,
+    InvokeMCPCommand,
     SendMessageCommand,
 };
 export class CodeWhispererStreaming extends CodeWhispererStreamingClient {

package/node_modules/@amzn/codewhisperer-streaming/dist-es/commands/InvokeMCPCommand.js

@@ -0,0 +1,18 @@
+import { InvokeMCPRequestFilterSensitiveLog, InvokeMCPResponseFilterSensitiveLog, } from "../models/models_0";
+import { de_InvokeMCPCommand, se_InvokeMCPCommand, } from "../protocols/Aws_restJson1";
+import { getSerdePlugin } from "@smithy/middleware-serde";
+import { Command as $Command } from "@smithy/smithy-client";
+export { $Command };
+export class InvokeMCPCommand extends $Command.classBuilder()
+    .m(function (Command, cs, config, o) {
+    return [
+        getSerdePlugin(config, this.serialize, this.deserialize),
+    ];
+})
+    .s("AmazonCodeWhispererStreamingService", "InvokeMCP", {})
+    .n("CodeWhispererStreamingClient", "InvokeMCPCommand")
+    .f(InvokeMCPRequestFilterSensitiveLog, InvokeMCPResponseFilterSensitiveLog)
+    .ser(se_InvokeMCPCommand)
+    .de(de_InvokeMCPCommand)
+    .build() {
+}

package/node_modules/@amzn/codewhisperer-streaming/dist-es/commands/index.js

@@ -1,4 +1,5 @@
 export * from "./ExportResultArchiveCommand";
 export * from "./GenerateAssistantResponseCommand";
 export * from "./GenerateTaskAssistPlanCommand";
+export * from "./InvokeMCPCommand";
 export * from "./SendMessageCommand";