@vscode/chat-lib 0.1.1 → 0.1.2-0

package/dist/src/_internal/extension/completions-core/vscode-node/lib/src/prompt/components/recentEdits.js

@@ -0,0 +1,99 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.RecentEdits = void 0;
+exports.editIsTooCloseToCursor = editIsTooCloseToCursor;
+const jsx_runtime_1 = require("../../../../prompt/jsx-runtime//jsx-runtime");
+/*---------------------------------------------------------------------------------------------
+ *  Copyright (c) Microsoft Corporation. All rights reserved.
+ *  Licensed under the MIT License. See License.txt in the project root for license information.
+ *--------------------------------------------------------------------------------------------*/
+/** @jsxRuntime automatic */
+/** @jsxImportSource ../../../../prompt/jsx-runtime/ */
+const components_1 = require("../../../../prompt/src/components/components");
+const languageMarker_1 = require("../../../../prompt/src/languageMarker");
+const componentsCompletionsPromptFactory_1 = require("../completionsPromptFactory/componentsCompletionsPromptFactory");
+function editIsTooCloseToCursor(edit, filterByCursorLine = false, cursorLine = undefined, activeDocDistanceLimitFromCursor) {
+    if (filterByCursorLine) {
+        if (cursorLine === undefined || activeDocDistanceLimitFromCursor === undefined) {
+            throw new Error('cursorLine and activeDocDistanceLimitFromCursor are required when filterByCursorLine is true');
+        }
+    }
+    const startLineNumber = edit.startLine - 1;
+    const endLineNumber = edit.endLine - 1;
+    if (filterByCursorLine &&
+        (Math.abs(startLineNumber - cursorLine) <= activeDocDistanceLimitFromCursor ||
+            Math.abs(endLineNumber - cursorLine) <= activeDocDistanceLimitFromCursor)) {
+        // skip over a diff that's too close to the cursor
+        // this isn't cached since the cursor moves
+        return true;
+    }
+    return false;
+}
+/**
+ * Render the most recent edits in the prompt.
+ * @param props
+ * @param context
+ * @returns a <Text> element containing recent edit summaries, or undefined if there are no recent edits
+ */
+const RecentEdits = (props, context) => {
+    const [prompt, setPrompt] = context.useState();
+    context.useData(componentsCompletionsPromptFactory_1.isCompletionRequestData, async (request) => {
+        if (!request.document) {
+            return;
+        }
+        const recentEditProvider = props.recentEditsProvider;
+        if (recentEditProvider.isEnabled()) {
+            recentEditProvider.start();
+        }
+        else {
+            return;
+        }
+        const recentEditsConfig = recentEditProvider.config;
+        const recentEdits = recentEditProvider.getRecentEdits();
+        const filesIncluded = new Set();
+        const tdm = props.tdms;
+        const editSummaries = [];
+        // Walk backwards through the recent edits (most recent first) until we hit the max files or max edits, whichever comes first
+        for (let i = recentEdits.length - 1; i >= 0; i--) {
+            // if we've hit the max edits, stop
+            if (editSummaries.length >= recentEditsConfig.maxEdits) {
+                break;
+            }
+            const edit = recentEdits[i];
+            // If the file is excluded, skip it
+            if (!(await tdm.getTextDocument({ uri: edit.file }))) {
+                continue;
+            }
+            // If adding an edit from this file would exceed the max files, skip it
+            const isNewFile = !filesIncluded.has(edit.file);
+            const projectedFileCount = filesIncluded.size + (isNewFile ? 1 : 0);
+            if (projectedFileCount > recentEditsConfig.maxFiles) {
+                break;
+            }
+            const filterByCursorLine = edit.file === request.document?.uri;
+            const activeDocCursorLine = filterByCursorLine ? request.position.line : undefined;
+            // Check if the edit is too close to the cursor line, if applicable, in which case we skip it
+            const editTooClose = editIsTooCloseToCursor(edit, filterByCursorLine, activeDocCursorLine, recentEditsConfig.activeDocDistanceLimitFromCursor);
+            if (editTooClose) {
+                continue;
+            }
+            const summarizedEdit = recentEditProvider.getEditSummary(edit);
+            if (summarizedEdit) {
+                filesIncluded.add(edit.file);
+                const relativePathOrUri = tdm.getRelativePath({ uri: edit.file });
+                editSummaries.unshift((0, languageMarker_1.newLineEnded)(`File: ${relativePathOrUri}`) + (0, languageMarker_1.newLineEnded)(summarizedEdit));
+            }
+        }
+        if (editSummaries.length === 0) {
+            setPrompt(undefined);
+            return;
+        }
+        const newPrompt = (0, languageMarker_1.newLineEnded)('These are recently edited files. Do not suggest code that has been deleted.') +
+            editSummaries.join('') +
+            (0, languageMarker_1.newLineEnded)('End of recent edits');
+        setPrompt(newPrompt);
+    });
+    return prompt ? ((0, jsx_runtime_1.jsx)(components_1.Chunk, { children: (0, jsx_runtime_1.jsx)(components_1.Text, { children: prompt }) })) : undefined;
+};
+exports.RecentEdits = RecentEdits;
+//# sourceMappingURL=recentEdits.js.map

package/dist/src/_internal/extension/completions-core/vscode-node/lib/src/prompt/components/recentEdits.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"recentEdits.js","sourceRoot":"","sources":["../../../../../../../../../../src/_internal/extension/completions-core/vscode-node/lib/src/prompt/components/recentEdits.tsx"],"names":[],"mappings":";;;AAiBA,wDA2BC;;AA5CD;;;gGAGgG;AAChG,4BAA4B;AAC5B,uDAAuD;AAEvD,6EAAiH;AACjH,0EAAqE;AAErE,uHAGwE;AAIxE,SAAgB,sBAAsB,CACrC,IAAgB,EAChB,qBAA8B,KAAK,EACnC,aAAiC,SAAS,EAC1C,gCAAoD;IAEpD,IAAI,kBAAkB,EAAE,CAAC;QACxB,IAAI,UAAU,KAAK,SAAS,IAAI,gCAAgC,KAAK,SAAS,EAAE,CAAC;YAChF,MAAM,IAAI,KAAK,CACd,8FAA8F,CAC9F,CAAC;QACH,CAAC;IACF,CAAC;IAED,MAAM,eAAe,GAAG,IAAI,CAAC,SAAS,GAAG,CAAC,CAAC;IAC3C,MAAM,aAAa,GAAG,IAAI,CAAC,OAAO,GAAG,CAAC,CAAC;IAEvC,IACC,kBAAkB;QAClB,CAAC,IAAI,CAAC,GAAG,CAAC,eAAe,GAAG,UAAW,CAAC,IAAI,gCAAiC;YAC5E,IAAI,CAAC,GAAG,CAAC,aAAa,GAAG,UAAW,CAAC,IAAI,gCAAiC,CAAC,EAC3E,CAAC;QACF,kDAAkD;QAClD,2CAA2C;QAC3C,OAAO,IAAI,CAAC;IACb,CAAC;IACD,OAAO,KAAK,CAAC;AACd,CAAC;AAOD;;;;;GAKG;AACI,MAAM,WAAW,GAAG,CAAC,KAAuB,EAAE,OAAyB,EAAE,EAAE;IACjF,MAAM,CAAC,MAAM,EAAE,SAAS,CAAC,GAAG,OAAO,CAAC,QAAQ,EAAsB,CAAC;IAEnE,OAAO,CAAC,OAAO,CAAC,4DAAuB,EAAE,KAAK,EAAE,OAA8B,EAAE,EAAE;QACjF,IAAI,CAAC,OAAO,CAAC,QAAQ,EAAE,CAAC;YAAC,OAAO;QAAC,CAAC;QAElC,MAAM,kBAAkB,GAAG,KAAK,CAAC,mBAAmB,CAAC;QAErD,IAAI,kBAAkB,CAAC,SAAS,EAAE,EAAE,CAAC;YACpC,kBAAkB,CAAC,KAAK,EAAE,CAAC;QAC5B,CAAC;aAAM,CAAC;YACP,OAAO;QACR,CAAC;QAED,MAAM,iBAAiB,GAAI,kBAA8C,CAAC,MAAM,CAAC;QACjF,MAAM,WAAW,GAAG,kBAAkB,CAAC,cAAc,EAAE,CAAC;QAExD,MAAM,aAAa,GAAG,IAAI,GAAG,EAAU,CAAC;QACxC,MAAM,GAAG,GAAG,KAAK,CAAC,IAAI,CAAC;QACvB,MAAM,aAAa,GAAa,EAAE,CAAC;QAEnC,6HAA6H;QAC7H,KAAK,IAAI,CAAC,GAAG,WAAW,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC,IAAI,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC;YAClD,mCAAmC;YACnC,IAAI,aAAa,CAAC,MAAM,IAAI,iBAAiB,CAAC,QAAQ,EAAE,CAAC;gBAAC,MAAM;YAAC,CAAC;YAElE,MAAM,IAAI,GAAG,WAAW,CAAC,CAAC,CAAC,CAAC;YAE5B,mCAAmC;YACnC,IAAI,CAAC,CAAC,MAAM,GAAG,CAAC,eAAe,CAAC,EAAE,GAAG,EAAE,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC,EAAE,CAAC;gBAAC,SAAS;YAAC,CAAC;YAEnE,uEAAuE;YACvE,MAAM,SAAS,GAAG,CAAC,aAAa,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAChD,MAAM,kBAAkB,GAAG,aAAa,CAAC,IAAI,GAAG,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;YACpE,IAAI,kBAAkB,GAAG,iBAAiB,CAAC,QAAQ,EAAE,CAAC;gBAAC,MAAM;YAAC,CAAC;YAE/D,MAAM,kBAAkB,GAAG,IAAI,CAAC,IAAI,KAAK,OAAO,CAAC,QAAQ,EAAE,GAAG,CAAC;YAC/D,MAAM,mBAAmB,GAAG,kBAAkB,CAAC,CAAC,CAAC,OAAO,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,SAAS,CAAC;YAEnF,6FAA6F;YAC7F,MAAM,YAAY,GAAG,sBAAsB,CAC1C,IAAI,EACJ,kBAAkB,EAClB,mBAAmB,EACnB,iBAAiB,CAAC,gCAAgC,CAClD,CAAC;YACF,IAAI,YAAY,EAAE,CAAC;gBAClB,SAAS;YACV,CAAC;YAED,MAAM,cAAc,GAAG,kBAAkB,CAAC,cAAc,CAAC,IAAI,CAAC,CAAC;YAC/D,IAAI,cAAc,EAAE,CAAC;gBACpB,aAAa,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;gBAC7B,MAAM,iBAAiB,GAAG,GAAG,CAAC,eAAe,CAAC,EAAE,GAAG,EAAE,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC;gBAClE,aAAa,CAAC,OAAO,CAAC,IAAA,6BAAY,EAAC,SAAS,iBAAiB,EAAE,CAAC,GAAG,IAAA,6BAAY,EAAC,cAAc,CAAC,CAAC,CAAC;YAClG,CAAC;QACF,CAAC;QAED,IAAI,aAAa,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAChC,SAAS,CAAC,SAAS,CAAC,CAAC;YACrB,OAAO;QACR,CAAC;QAED,MAAM,SAAS,GACd,IAAA,6BAAY,EAAC,6EAA6E,CAAC;YAC3F,aAAa,CAAC,IAAI,CAAC,EAAE,CAAC;YACtB,IAAA,6BAAY,EAAC,qBAAqB,CAAC,CAAC;QAErC,SAAS,CAAC,SAAS,CAAC,CAAC;IACtB,CAAC,CAAC,CAAC;IAEH,OAAO,MAAM,CAAC,CAAC,CAAC,CACf,uBAAC,kBAAK,cACL,uBAAC,iBAAI,cAAE,MAAM,GAAQ,GACd,CACR,CAAC,CAAC,CAAC,SAAS,CAAC;AACf,CAAC,CAAC;AA5EW,QAAA,WAAW,eA4EtB"}

package/dist/src/_internal/extension/completions-core/vscode-node/lib/src/prompt/components/similarFiles.d.ts

@@ -0,0 +1,12 @@
+/** @jsxRuntime automatic */
+/** @jsxImportSource ../../../../prompt/jsx-runtime/ */
+import { IInstantiationService } from '../../../../../../../util/vs/platform/instantiation/common/instantiation';
+import { ComponentContext, PromptElementProps } from '../../../../prompt/src/components/components';
+import { ICompletionsTextDocumentManagerService } from '../../textDocumentManager';
+type SimilarFilesProps = {
+    instantiationService: IInstantiationService;
+    tdms: ICompletionsTextDocumentManagerService;
+} & PromptElementProps;
+export declare const SimilarFiles: (props: SimilarFilesProps, context: ComponentContext) => import("../../../../prompt/jsx-runtime//jsx-runtime").JSX.Element;
+export {};
+//# sourceMappingURL=similarFiles.d.ts.map

package/dist/src/_internal/extension/completions-core/vscode-node/lib/src/prompt/components/similarFiles.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"similarFiles.d.ts","sourceRoot":"","sources":["../../../../../../../../../../src/_internal/extension/completions-core/vscode-node/lib/src/prompt/components/similarFiles.tsx"],"names":[],"mappings":"AAIA,4BAA4B;AAC5B,uDAAuD;AAEvD,OAAO,EAAE,qBAAqB,EAAE,MAAM,0EAA0E,CAAC;AACjH,OAAO,EAAS,gBAAgB,EAAE,kBAAkB,EAAQ,MAAM,8CAA8C,CAAC;AAOjH,OAAO,EAAE,sCAAsC,EAAE,MAAM,2BAA2B,CAAC;AASnF,KAAK,iBAAiB,GAAG;IACxB,oBAAoB,EAAE,qBAAqB,CAAC;IAC5C,IAAI,EAAE,sCAAsC,CAAC;CAC7C,GAAG,kBAAkB,CAAC;AAQvB,eAAO,MAAM,YAAY,GAAI,OAAO,iBAAiB,EAAE,SAAS,gBAAgB,sEAwE/E,CAAC"}

package/dist/src/_internal/extension/completions-core/vscode-node/lib/src/prompt/components/similarFiles.js

@@ -0,0 +1,58 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.SimilarFiles = void 0;
+const jsx_runtime_1 = require("../../../../prompt/jsx-runtime//jsx-runtime");
+const components_1 = require("../../../../prompt/src/components/components");
+const similarFiles_1 = require("../../../../prompt/src/snippetInclusion/similarFiles");
+const snippets_1 = require("../../../../prompt/src/snippetInclusion/snippets");
+const similarFileOptionsProvider_1 = require("../../experiments/similarFileOptionsProvider");
+const componentsCompletionsPromptFactory_1 = require("../completionsPromptFactory/componentsCompletionsPromptFactory");
+const prompt_1 = require("../prompt");
+const neighborFiles_1 = require("../similarFiles/neighborFiles");
+const SimilarFiles = (props, context) => {
+    const [document, setDocument] = context.useState();
+    const [similarFiles, setSimilarFiles] = context.useState([]);
+    context.useData(componentsCompletionsPromptFactory_1.isCompletionRequestData, async (requestData) => {
+        if (requestData.document.uri !== document?.uri) {
+            setSimilarFiles([]);
+        }
+        setDocument(requestData.document);
+        let files = neighborFiles_1.NeighborSource.defaultEmptyResult();
+        if (!requestData.turnOffSimilarFiles) {
+            files = await props.instantiationService.invokeFunction(async (acc) => await neighborFiles_1.NeighborSource.getNeighborFilesAndTraits(acc, requestData.document.uri, requestData.document.detectedLanguageId, requestData.telemetryData, requestData.cancellationToken, requestData.data));
+        }
+        const similarFiles = await produceSimilarFiles(requestData.telemetryData, requestData.document, requestData, files);
+        setSimilarFiles(similarFiles);
+    });
+    async function produceSimilarFiles(telemetryData, doc, requestData, files) {
+        const promptOptions = props.instantiationService.invokeFunction(prompt_1.getPromptOptions, telemetryData, doc.detectedLanguageId);
+        const similarSnippets = await findSimilarSnippets(promptOptions, telemetryData, doc, requestData, files);
+        return similarSnippets
+            .filter(s => s.snippet.length > 0)
+            .sort((a, b) => a.score - b.score)
+            .map(s => {
+            return { ...(0, snippets_1.announceSnippet)(s), score: s.score };
+        });
+    }
+    async function findSimilarSnippets(promptOptions, telemetryData, doc, requestData, files) {
+        const similarFilesOptions = promptOptions.similarFilesOptions ||
+            props.instantiationService.invokeFunction(similarFileOptionsProvider_1.getSimilarFilesOptions, telemetryData, doc.detectedLanguageId);
+        const tdm = props.tdms;
+        const relativePath = tdm.getRelativePath(doc);
+        const docInfo = {
+            uri: doc.uri,
+            source: doc.getText(),
+            offset: doc.offsetAt(requestData.position),
+            relativePath,
+            languageId: doc.detectedLanguageId,
+        };
+        return await (0, similarFiles_1.getSimilarSnippets)(docInfo, Array.from(files.docs.values()), similarFilesOptions);
+    }
+    return (0, jsx_runtime_1.jsxs)(jsx_runtime_1.Fragment, { children: [...similarFiles.map((file, index) => (0, jsx_runtime_1.jsx)(SimilarFile, { snippet: file }))] });
+};
+exports.SimilarFiles = SimilarFiles;
+// TODO: change Chunk for KeepTogether
+const SimilarFile = (props, context) => {
+    return ((0, jsx_runtime_1.jsxs)(components_1.Chunk, { children: [(0, jsx_runtime_1.jsx)(components_1.Text, { children: props.snippet.headline }), (0, jsx_runtime_1.jsx)(components_1.Text, { children: props.snippet.snippet })] }));
+};
+//# sourceMappingURL=similarFiles.js.map

package/dist/src/_internal/extension/completions-core/vscode-node/lib/src/prompt/components/similarFiles.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"similarFiles.js","sourceRoot":"","sources":["../../../../../../../../../../src/_internal/extension/completions-core/vscode-node/lib/src/prompt/components/similarFiles.tsx"],"names":[],"mappings":";;;;AAQA,6EAAiH;AAEjH,uFAA0F;AAC1F,+EAAmF;AACnF,6FAAsF;AAItF,uHAIwE;AACxE,sCAA6C;AAC7C,iEAAoF;AAa7E,MAAM,YAAY,GAAG,CAAC,KAAwB,EAAE,OAAyB,EAAE,EAAE;IACnF,MAAM,CAAC,QAAQ,EAAE,WAAW,CAAC,GAAG,OAAO,CAAC,QAAQ,EAA6B,CAAC;IAC9E,MAAM,CAAC,YAAY,EAAE,eAAe,CAAC,GAAG,OAAO,CAAC,QAAQ,CAAuB,EAAE,CAAC,CAAC;IAEnF,OAAO,CAAC,OAAO,CAAC,4DAAuB,EAAE,KAAK,EAAE,WAAkC,EAAE,EAAE;QACrF,IAAI,WAAW,CAAC,QAAQ,CAAC,GAAG,KAAK,QAAQ,EAAE,GAAG,EAAE,CAAC;YAChD,eAAe,CAAC,EAAE,CAAC,CAAC;QACrB,CAAC;QACD,WAAW,CAAC,WAAW,CAAC,QAAQ,CAAC,CAAC;QAElC,IAAI,KAAK,GAAkC,8BAAc,CAAC,kBAAkB,EAAE,CAAC;QAC/E,IAAI,CAAC,WAAW,CAAC,mBAAmB,EAAE,CAAC;YACtC,KAAK,GAAG,MAAM,KAAK,CAAC,oBAAoB,CAAC,cAAc,CAAC,KAAK,EAAC,GAAG,EAAC,EAAE,CAAC,MAAM,8BAAc,CAAC,yBAAyB,CAClH,GAAG,EACH,WAAW,CAAC,QAAQ,CAAC,GAAG,EACxB,WAAW,CAAC,QAAQ,CAAC,kBAAkB,EACvC,WAAW,CAAC,aAAa,EACzB,WAAW,CAAC,iBAAiB,EAC7B,WAAW,CAAC,IAAI,CAChB,CAAC,CAAC;QACJ,CAAC;QAED,MAAM,YAAY,GAAG,MAAM,mBAAmB,CAC7C,WAAW,CAAC,aAAa,EACzB,WAAW,CAAC,QAAQ,EACpB,WAAW,EACX,KAAK,CACL,CAAC;QACF,eAAe,CAAC,YAAY,CAAC,CAAC;IAC/B,CAAC,CAAC,CAAC;IAEH,KAAK,UAAU,mBAAmB,CACjC,aAA+B,EAC/B,GAAyB,EACzB,WAAkC,EAClC,KAEC;QAED,MAAM,aAAa,GAAG,KAAK,CAAC,oBAAoB,CAAC,cAAc,CAAC,yBAAgB,EAAE,aAAa,EAAE,GAAG,CAAC,kBAAkB,CAAC,CAAC;QACzH,MAAM,eAAe,GAAG,MAAM,mBAAmB,CAAC,aAAa,EAAE,aAAa,EAAE,GAAG,EAAE,WAAW,EAAE,KAAK,CAAC,CAAC;QACzG,OAAO,eAAe;aACpB,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,MAAM,GAAG,CAAC,CAAC;aACjC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,GAAG,CAAC,CAAC,KAAK,CAAC;aACjC,GAAG,CAAC,CAAC,CAAC,EAAE;YACR,OAAO,EAAE,GAAG,IAAA,0BAAe,EAAC,CAAC,CAAC,EAAE,KAAK,EAAE,CAAC,CAAC,KAAK,EAAE,CAAC;QAClD,CAAC,CAAC,CAAC;IACL,CAAC;IAED,KAAK,UAAU,mBAAmB,CACjC,aAA4B,EAC5B,aAA+B,EAC/B,GAAyB,EACzB,WAAkC,EAClC,KAAoC;QAEpC,MAAM,mBAAmB,GACxB,aAAa,CAAC,mBAAmB;YACjC,KAAK,CAAC,oBAAoB,CAAC,cAAc,CAAC,mDAAsB,EAAE,aAAa,EAAE,GAAG,CAAC,kBAAkB,CAAC,CAAC;QAC1G,MAAM,GAAG,GAAG,KAAK,CAAC,IAAI,CAAC;QACvB,MAAM,YAAY,GAAG,GAAG,CAAC,eAAe,CAAC,GAAG,CAAC,CAAC;QAC9C,MAAM,OAAO,GAA2B;YACvC,GAAG,EAAE,GAAG,CAAC,GAAG;YACZ,MAAM,EAAE,GAAG,CAAC,OAAO,EAAE;YACrB,MAAM,EAAE,GAAG,CAAC,QAAQ,CAAC,WAAW,CAAC,QAAQ,CAAC;YAC1C,YAAY;YACZ,UAAU,EAAE,GAAG,CAAC,kBAAkB;SAClC,CAAC;QACF,OAAO,MAAM,IAAA,iCAAkB,EAAC,OAAO,EAAE,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC,EAAE,mBAAmB,CAAC,CAAC;IAChG,CAAC;IAED,OAAO,gEAAM,YAAY,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,KAAK,EAAE,EAAE,CAAC,uBAAC,WAAW,IAAC,OAAO,EAAE,IAAI,GAAI,CAAC,IAAI,CAAC;AACpF,CAAC,CAAC;AAxEW,QAAA,YAAY,gBAwEvB;AAEF,sCAAsC;AACtC,MAAM,WAAW,GAAG,CAAC,KAAsC,EAAE,OAAyB,EAAE,EAAE;IACzF,OAAO,CACN,wBAAC,kBAAK,eACL,uBAAC,iBAAI,cAAE,KAAK,CAAC,OAAO,CAAC,QAAQ,GAAQ,EACrC,uBAAC,iBAAI,cAAE,KAAK,CAAC,OAAO,CAAC,OAAO,GAAQ,IAC7B,CACR,CAAC;AACH,CAAC,CAAC"}

package/dist/src/_internal/extension/completions-core/vscode-node/lib/src/prompt/components/splitContextPrompt.d.ts

@@ -0,0 +1,9 @@
+/** @jsxRuntime automatic */
+/** @jsxImportSource ../../../../prompt/jsx-runtime/ */
+import { ServicesAccessor } from '../../../../../../../util/vs/platform/instantiation/common/instantiation';
+/**
+ * Function that returns the prompt structure for a code completion request following the split context prompt design
+ * that optimizes for cache hits.
+ */
+export declare function splitContextCompletionsPrompt(accessor: ServicesAccessor): import("../../../../prompt/jsx-runtime//jsx-runtime").JSX.Element;
+//# sourceMappingURL=splitContextPrompt.d.ts.map

package/dist/src/_internal/extension/completions-core/vscode-node/lib/src/prompt/components/splitContextPrompt.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"splitContextPrompt.d.ts","sourceRoot":"","sources":["../../../../../../../../../../src/_internal/extension/completions-core/vscode-node/lib/src/prompt/components/splitContextPrompt.tsx"],"names":[],"mappings":"AAIA,4BAA4B;AAC5B,uDAAuD;AAEvD,OAAO,EAAyB,gBAAgB,EAAE,MAAM,0EAA0E,CAAC;AAWnI;;;GAGG;AACH,wBAAgB,6BAA6B,CAAC,QAAQ,EAAE,gBAAgB,qEAmBvE"}

package/dist/src/_internal/extension/completions-core/vscode-node/lib/src/prompt/components/splitContextPrompt.js

@@ -0,0 +1,31 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.splitContextCompletionsPrompt = splitContextCompletionsPrompt;
+const jsx_runtime_1 = require("../../../../prompt/jsx-runtime//jsx-runtime");
+/*---------------------------------------------------------------------------------------------
+ *  Copyright (c) Microsoft Corporation. All rights reserved.
+ *  Licensed under the MIT License. See License.txt in the project root for license information.
+ *--------------------------------------------------------------------------------------------*/
+/** @jsxRuntime automatic */
+/** @jsxImportSource ../../../../prompt/jsx-runtime/ */
+const instantiation_1 = require("../../../../../../../util/vs/platform/instantiation/common/instantiation");
+const textDocumentManager_1 = require("../../textDocumentManager");
+const recentEditsProvider_1 = require("../recentEdits/recentEditsProvider");
+const codeSnippets_1 = require("./codeSnippets");
+const completionsContext_1 = require("./completionsContext");
+const currentFile_1 = require("./currentFile");
+const marker_1 = require("./marker");
+const recentEdits_1 = require("./recentEdits");
+const similarFiles_1 = require("./similarFiles");
+const traits_1 = require("./traits");
+/**
+ * Function that returns the prompt structure for a code completion request following the split context prompt design
+ * that optimizes for cache hits.
+ */
+function splitContextCompletionsPrompt(accessor) {
+    const instantiationService = accessor.get(instantiation_1.IInstantiationService);
+    const tdms = accessor.get(textDocumentManager_1.ICompletionsTextDocumentManagerService);
+    const recentEditsProvider = accessor.get(recentEditsProvider_1.ICompletionsRecentEditsProviderService);
+    return ((0, jsx_runtime_1.jsxs)(jsx_runtime_1.Fragment, { children: [(0, jsx_runtime_1.jsxs)(completionsContext_1.StableCompletionsContext, { children: [(0, jsx_runtime_1.jsx)(marker_1.DocumentMarker, { tdms: tdms, weight: 0.7 }), (0, jsx_runtime_1.jsx)(traits_1.Traits, { weight: 0.6 }), (0, jsx_runtime_1.jsx)(codeSnippets_1.CodeSnippets, { tdms: tdms, weight: 0.9 }), (0, jsx_runtime_1.jsx)(similarFiles_1.SimilarFiles, { tdms: tdms, instantiationService: instantiationService, weight: 0.8 })] }), (0, jsx_runtime_1.jsx)(currentFile_1.DocumentSuffix, { weight: 1 }), (0, jsx_runtime_1.jsx)(completionsContext_1.AdditionalCompletionsContext, { children: (0, jsx_runtime_1.jsx)(recentEdits_1.RecentEdits, { tdms: tdms, recentEditsProvider: recentEditsProvider, weight: 0.99 }) }), (0, jsx_runtime_1.jsx)(currentFile_1.DocumentPrefix, { weight: 1 })] }));
+}
+//# sourceMappingURL=splitContextPrompt.js.map

package/dist/src/_internal/extension/completions-core/vscode-node/lib/src/prompt/components/splitContextPrompt.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"splitContextPrompt.js","sourceRoot":"","sources":["../../../../../../../../../../src/_internal/extension/completions-core/vscode-node/lib/src/prompt/components/splitContextPrompt.tsx"],"names":[],"mappings":";;AAsBA,sEAmBC;;AAzCD;;;gGAGgG;AAChG,4BAA4B;AAC5B,uDAAuD;AAEvD,4GAAmI;AACnI,mEAAmF;AACnF,4EAA4F;AAC5F,iDAA8C;AAC9C,6DAA8F;AAC9F,+CAA+D;AAC/D,qCAA0C;AAC1C,+CAA4C;AAC5C,iDAA8C;AAC9C,qCAAkC;AAElC;;;GAGG;AACH,SAAgB,6BAA6B,CAAC,QAA0B;IACvE,MAAM,oBAAoB,GAAG,QAAQ,CAAC,GAAG,CAAC,qCAAqB,CAAC,CAAC;IACjE,MAAM,IAAI,GAAG,QAAQ,CAAC,GAAG,CAAC,4DAAsC,CAAC,CAAC;IAClE,MAAM,mBAAmB,GAAG,QAAQ,CAAC,GAAG,CAAC,4DAAsC,CAAC,CAAC;IACjF,OAAO,CACN,6DACC,wBAAC,6CAAwB,eACxB,uBAAC,uBAAc,IAAC,IAAI,EAAE,IAAI,EAAE,MAAM,EAAE,GAAG,GAAI,EAC3C,uBAAC,eAAM,IAAC,MAAM,EAAE,GAAG,GAAI,EACvB,uBAAC,2BAAY,IAAC,IAAI,EAAE,IAAI,EAAE,MAAM,EAAE,GAAG,GAAI,EACzC,uBAAC,2BAAY,IAAC,IAAI,EAAE,IAAI,EAAE,oBAAoB,EAAE,oBAAoB,EAAE,MAAM,EAAE,GAAG,GAAI,IAC3D,EAC3B,uBAAC,4BAAc,IAAC,MAAM,EAAE,CAAC,GAAI,EAC7B,uBAAC,iDAA4B,cAC5B,uBAAC,yBAAW,IAAC,IAAI,EAAE,IAAI,EAAE,mBAAmB,EAAE,mBAAmB,EAAE,MAAM,EAAE,IAAI,GAAI,GACrD,EAC/B,uBAAC,4BAAc,IAAC,MAAM,EAAE,CAAC,GAAI,IAC3B,CACH,CAAC;AACH,CAAC"}

package/dist/src/_internal/extension/completions-core/vscode-node/lib/src/prompt/components/splitContextPromptRenderer.d.ts

@@ -0,0 +1,15 @@
+/** @jsxRuntime automatic */
+/** @jsxImportSource ../../../../prompt/jsx-runtime/ */
+import { ComponentStatistics, PromptSnapshotNode } from '../../../../prompt/src/components/components';
+import { CompletionsPromptRenderer } from './completionsPromptRenderer';
+import { ElidedBlock, WeightedBlock } from './elision';
+export declare class SplitContextPromptRenderer extends CompletionsPromptRenderer {
+    protected formatPrefix: (elidedBlocks: ElidedBlock[]) => string;
+    protected formatContext: ((elidedBlocks: ElidedBlock[]) => string[]) | undefined;
+    processSnapshot(snapshot: PromptSnapshotNode, delimiter: string): {
+        prefixBlocks: WeightedBlock[];
+        suffixBlock: WeightedBlock;
+        componentStatistics: ComponentStatistics[];
+    };
+}
+//# sourceMappingURL=splitContextPromptRenderer.d.ts.map

package/dist/src/_internal/extension/completions-core/vscode-node/lib/src/prompt/components/splitContextPromptRenderer.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"splitContextPromptRenderer.d.ts","sourceRoot":"","sources":["../../../../../../../../../../src/_internal/extension/completions-core/vscode-node/lib/src/prompt/components/splitContextPromptRenderer.tsx"],"names":[],"mappings":"AAIA,4BAA4B;AAC5B,uDAAuD;AAEvD,OAAO,EAAE,mBAAmB,EAAE,kBAAkB,EAAE,MAAM,8CAA8C,CAAC;AAGvG,OAAO,EACN,yBAAyB,EAGzB,MAAM,6BAA6B,CAAC;AAErC,OAAO,EAAE,WAAW,EAAuC,aAAa,EAAE,MAAM,WAAW,CAAC;AAW5F,qBAAa,0BAA2B,SAAQ,yBAAyB;IACxE,UAAmB,YAAY,EAAE,CAAC,YAAY,EAAE,WAAW,EAAE,KAAK,MAAM,CAAoB;IAC5F,UAAmB,aAAa,EAAE,CAAC,CAAC,YAAY,EAAE,WAAW,EAAE,KAAK,MAAM,EAAE,CAAC,GAAG,SAAS,CAAqB;IAErG,eAAe,CACvB,QAAQ,EAAE,kBAAkB,EAC5B,SAAS,EAAE,MAAM,GACf;QACF,YAAY,EAAE,aAAa,EAAE,CAAC;QAC9B,WAAW,EAAE,aAAa,CAAC;QAC3B,mBAAmB,EAAE,mBAAmB,EAAE,CAAC;KAC3C;CAqFD"}

package/dist/src/_internal/extension/completions-core/vscode-node/lib/src/prompt/components/splitContextPromptRenderer.js

@@ -0,0 +1,113 @@
+"use strict";
+/*---------------------------------------------------------------------------------------------
+ *  Copyright (c) Microsoft Corporation. All rights reserved.
+ *  Licensed under the MIT License. See License.txt in the project root for license information.
+ *--------------------------------------------------------------------------------------------*/
+/** @jsxRuntime automatic */
+/** @jsxImportSource ../../../../prompt/jsx-runtime/ */
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.SplitContextPromptRenderer = void 0;
+const walker_1 = require("../../../../prompt/src/components/walker");
+const completionsContext_1 = require("./completionsContext");
+const completionsPromptRenderer_1 = require("./completionsPromptRenderer");
+const currentFile_1 = require("./currentFile");
+const elision_1 = require("./elision");
+let contextIndex = 0;
+function resetContextIndex() {
+    contextIndex = 0;
+}
+function getNextContextIndex() {
+    return contextIndex++;
+}
+class SplitContextPromptRenderer extends completionsPromptRenderer_1.CompletionsPromptRenderer {
+    constructor() {
+        super(...arguments);
+        this.formatPrefix = elision_1.makePrefixPrompt;
+        this.formatContext = elision_1.makeContextPrompt;
+    }
+    processSnapshot(snapshot, delimiter) {
+        const prefixBlocks = [];
+        const suffixBlocks = [];
+        const componentStatistics = [];
+        // Store the status of the required prefix node
+        let foundPrefix = false;
+        resetContextIndex();
+        const walker = new walker_1.SnapshotWalker(snapshot, splitContextTransformers);
+        walker.walkSnapshot((node, _parent, context) => {
+            if (node === snapshot) {
+                return true;
+            }
+            if (node.statistics.updateDataTimeMs && node.statistics.updateDataTimeMs > 0) {
+                componentStatistics.push({
+                    componentPath: node.path,
+                    updateDataTimeMs: node.statistics.updateDataTimeMs,
+                });
+            }
+            // Check for the presence of required prefix node
+            if (node.name === currentFile_1.BeforeCursor.name) {
+                foundPrefix = true;
+            }
+            if (node.value === undefined || node.value === '') {
+                // No need to process this node as it only adds whitespace
+                return true;
+            }
+            const chunks = context.chunks;
+            const type = context.type;
+            if (type === 'suffix') {
+                // Suffix handling: Mark the child node with content as suffix
+                suffixBlocks.push({
+                    value: (0, completionsPromptRenderer_1.normalizeLineEndings)(node.value),
+                    type: 'suffix',
+                    weight: context.weight,
+                    componentPath: node.path,
+                    nodeStatistics: node.statistics,
+                    chunks,
+                    source: context.source,
+                });
+            }
+            else {
+                const isPrefix = type === 'prefix';
+                // Add delimiter to non-prefix nodes
+                const nodeValueWithDelimiter = isPrefix || node.value.endsWith(delimiter) ? node.value : node.value + delimiter;
+                prefixBlocks.push({
+                    type: isPrefix ? 'prefix' : 'context',
+                    value: (0, completionsPromptRenderer_1.normalizeLineEndings)(nodeValueWithDelimiter),
+                    weight: context.weight,
+                    componentPath: node.path,
+                    nodeStatistics: node.statistics,
+                    chunks,
+                    source: context.source,
+                    index: isPrefix ? undefined : context.index, // index only set for context nodes
+                });
+            }
+            return true;
+        });
+        if (!foundPrefix) {
+            throw new Error(`Node of type ${currentFile_1.BeforeCursor.name} not found`);
+        }
+        if (suffixBlocks.length > 1) {
+            throw new Error(`Only one suffix is allowed`);
+        }
+        const suffixBlock = suffixBlocks.length === 1
+            ? suffixBlocks[0]
+            : {
+                componentPath: '',
+                value: '',
+                weight: 1,
+                nodeStatistics: {},
+                type: 'suffix',
+            };
+        return { prefixBlocks, suffixBlock, componentStatistics };
+    }
+}
+exports.SplitContextPromptRenderer = SplitContextPromptRenderer;
+const splitContextTransformers = [
+    ...completionsPromptRenderer_1.transformers,
+    (node, _, context) => {
+        if ((0, completionsContext_1.isContextNode)(node)) {
+            return { ...context, index: getNextContextIndex() };
+        }
+        return context;
+    },
+];
+//# sourceMappingURL=splitContextPromptRenderer.js.map

package/dist/src/_internal/extension/completions-core/vscode-node/lib/src/prompt/components/splitContextPromptRenderer.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"splitContextPromptRenderer.js","sourceRoot":"","sources":["../../../../../../../../../../src/_internal/extension/completions-core/vscode-node/lib/src/prompt/components/splitContextPromptRenderer.tsx"],"names":[],"mappings":";AAAA;;;gGAGgG;AAChG,4BAA4B;AAC5B,uDAAuD;;;AAGvD,qEAAkG;AAClG,6DAAqD;AACrD,2EAIqC;AACrC,+CAA6C;AAC7C,uCAA4F;AAE5F,IAAI,YAAY,GAAG,CAAC,CAAC;AACrB,SAAS,iBAAiB;IACzB,YAAY,GAAG,CAAC,CAAC;AAClB,CAAC;AAED,SAAS,mBAAmB;IAC3B,OAAO,YAAY,EAAE,CAAC;AACvB,CAAC;AAED,MAAa,0BAA2B,SAAQ,qDAAyB;IAAzE;;QACoB,iBAAY,GAA4C,0BAAgB,CAAC;QACzE,kBAAa,GAA4D,2BAAiB,CAAC;IA8F/G,CAAC;IA5FS,eAAe,CACvB,QAA4B,EAC5B,SAAiB;QAMjB,MAAM,YAAY,GAAoB,EAAE,CAAC;QACzC,MAAM,YAAY,GAAoB,EAAE,CAAC;QACzC,MAAM,mBAAmB,GAA0B,EAAE,CAAC;QAEtD,+CAA+C;QAC/C,IAAI,WAAW,GAAG,KAAK,CAAC;QAExB,iBAAiB,EAAE,CAAC;QACpB,MAAM,MAAM,GAAG,IAAI,uBAAc,CAAC,QAAQ,EAAE,wBAAwB,CAAC,CAAC;QACtE,MAAM,CAAC,YAAY,CAAC,CAAC,IAAI,EAAE,OAAO,EAAE,OAAO,EAAE,EAAE;YAC9C,IAAI,IAAI,KAAK,QAAQ,EAAE,CAAC;gBACvB,OAAO,IAAI,CAAC;YACb,CAAC;YAED,IAAI,IAAI,CAAC,UAAU,CAAC,gBAAgB,IAAI,IAAI,CAAC,UAAU,CAAC,gBAAgB,GAAG,CAAC,EAAE,CAAC;gBAC9E,mBAAmB,CAAC,IAAI,CAAC;oBACxB,aAAa,EAAE,IAAI,CAAC,IAAI;oBACxB,gBAAgB,EAAE,IAAI,CAAC,UAAU,CAAC,gBAAgB;iBAClD,CAAC,CAAC;YACJ,CAAC;YAED,iDAAiD;YACjD,IAAI,IAAI,CAAC,IAAI,KAAK,0BAAY,CAAC,IAAI,EAAE,CAAC;gBACrC,WAAW,GAAG,IAAI,CAAC;YACpB,CAAC;YAED,IAAI,IAAI,CAAC,KAAK,KAAK,SAAS,IAAI,IAAI,CAAC,KAAK,KAAK,EAAE,EAAE,CAAC;gBACnD,0DAA0D;gBAC1D,OAAO,IAAI,CAAC;YACb,CAAC;YAED,MAAM,MAAM,GAAG,OAAO,CAAC,MAAiC,CAAC;YACzD,MAAM,IAAI,GAAG,OAAO,CAAC,IAA0B,CAAC;YAChD,IAAI,IAAI,KAAK,QAAQ,EAAE,CAAC;gBACvB,8DAA8D;gBAC9D,YAAY,CAAC,IAAI,CAAC;oBACjB,KAAK,EAAE,IAAA,gDAAoB,EAAC,IAAI,CAAC,KAAK,CAAC;oBACvC,IAAI,EAAE,QAAQ;oBACd,MAAM,EAAE,OAAO,CAAC,MAAgB;oBAChC,aAAa,EAAE,IAAI,CAAC,IAAI;oBACxB,cAAc,EAAE,IAAI,CAAC,UAAU;oBAC/B,MAAM;oBACN,MAAM,EAAE,OAAO,CAAC,MAAM;iBACtB,CAAC,CAAC;YACJ,CAAC;iBAAM,CAAC;gBACP,MAAM,QAAQ,GAAG,IAAI,KAAK,QAAQ,CAAC;gBAEnC,oCAAoC;gBACpC,MAAM,sBAAsB,GAC3B,QAAQ,IAAI,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,GAAG,SAAS,CAAC;gBAClF,YAAY,CAAC,IAAI,CAAC;oBACjB,IAAI,EAAE,QAAQ,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,SAAS;oBACrC,KAAK,EAAE,IAAA,gDAAoB,EAAC,sBAAsB,CAAC;oBACnD,MAAM,EAAE,OAAO,CAAC,MAAgB;oBAChC,aAAa,EAAE,IAAI,CAAC,IAAI;oBACxB,cAAc,EAAE,IAAI,CAAC,UAAU;oBAC/B,MAAM;oBACN,MAAM,EAAE,OAAO,CAAC,MAAM;oBACtB,KAAK,EAAE,QAAQ,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAE,OAAO,CAAC,KAAgB,EAAE,mCAAmC;iBAC5F,CAAC,CAAC;YACJ,CAAC;YACD,OAAO,IAAI,CAAC;QACb,CAAC,CAAC,CAAC;QAEH,IAAI,CAAC,WAAW,EAAE,CAAC;YAClB,MAAM,IAAI,KAAK,CAAC,gBAAgB,0BAAY,CAAC,IAAI,YAAY,CAAC,CAAC;QAChE,CAAC;QACD,IAAI,YAAY,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC7B,MAAM,IAAI,KAAK,CAAC,4BAA4B,CAAC,CAAC;QAC/C,CAAC;QAED,MAAM,WAAW,GAChB,YAAY,CAAC,MAAM,KAAK,CAAC;YACxB,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC;YACjB,CAAC,CAAC;gBACD,aAAa,EAAE,EAAE;gBACjB,KAAK,EAAE,EAAE;gBACT,MAAM,EAAE,CAAC;gBACT,cAAc,EAAE,EAAE;gBAClB,IAAI,EAAE,QAAQ;aACd,CAAC;QAEJ,OAAO,EAAE,YAAY,EAAE,WAAW,EAAE,mBAAmB,EAAE,CAAC;IAC3D,CAAC;CACD;AAhGD,gEAgGC;AAED,MAAM,wBAAwB,GAA6B;IAC1D,GAAG,wCAAY;IACf,CAAC,IAAI,EAAE,CAAC,EAAE,OAAO,EAAE,EAAE;QACpB,IAAI,IAAA,kCAAa,EAAC,IAAI,CAAC,EAAE,CAAC;YACzB,OAAO,EAAE,GAAG,OAAO,EAAE,KAAK,EAAE,mBAAmB,EAAE,EAAE,CAAC;QACrD,CAAC;QACD,OAAO,OAAO,CAAC;IAChB,CAAC;CACD,CAAC"}

package/dist/src/_internal/extension/completions-core/vscode-node/lib/src/prompt/components/traits.d.ts

@@ -0,0 +1,5 @@
+/** @jsxRuntime automatic */
+/** @jsxImportSource ../../../../prompt/jsx-runtime/ */
+import { ComponentContext, PromptElementProps } from '../../../../prompt/src/components/components';
+export declare const Traits: (_props: PromptElementProps, context: ComponentContext) => import("../../../../prompt/jsx-runtime//jsx-runtime").JSX.Element | undefined;
+//# sourceMappingURL=traits.d.ts.map

package/dist/src/_internal/extension/completions-core/vscode-node/lib/src/prompt/components/traits.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"traits.d.ts","sourceRoot":"","sources":["../../../../../../../../../../src/_internal/extension/completions-core/vscode-node/lib/src/prompt/components/traits.tsx"],"names":[],"mappings":"AAIA,4BAA4B;AAC5B,uDAAuD;AAEvD,OAAO,EAAE,gBAAgB,EAAE,kBAAkB,EAAQ,MAAM,8CAA8C,CAAC;AAQ1G,eAAO,MAAM,MAAM,GAAI,QAAQ,kBAAkB,EAAE,SAAS,gBAAgB,kFA8B3E,CAAC"}

package/dist/src/_internal/extension/completions-core/vscode-node/lib/src/prompt/components/traits.js

@@ -0,0 +1,33 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.Traits = void 0;
+const jsx_runtime_1 = require("../../../../prompt/jsx-runtime//jsx-runtime");
+/*---------------------------------------------------------------------------------------------
+ *  Copyright (c) Microsoft Corporation. All rights reserved.
+ *  Licensed under the MIT License. See License.txt in the project root for license information.
+ *--------------------------------------------------------------------------------------------*/
+/** @jsxRuntime automatic */
+/** @jsxImportSource ../../../../prompt/jsx-runtime/ */
+const components_1 = require("../../../../prompt/src/components/components");
+const prompt_1 = require("../../../../prompt/src/prompt");
+const componentsCompletionsPromptFactory_1 = require("../completionsPromptFactory/componentsCompletionsPromptFactory");
+const Traits = (_props, context) => {
+    const [traits, setTraits] = context.useState();
+    const [languageId, setLanguageId] = context.useState();
+    context.useData(componentsCompletionsPromptFactory_1.isCompletionRequestData, (data) => {
+        if (data.traits !== traits) {
+            setTraits(data.traits);
+        }
+        const normalizedLanguageId = (0, prompt_1.normalizeLanguageId)(data.document.detectedLanguageId);
+        if (normalizedLanguageId !== languageId) {
+            setLanguageId(normalizedLanguageId);
+        }
+    });
+    if (!traits || traits.length === 0 || !languageId) {
+        return;
+    }
+    // TODO: use a `KeepTogether` elision that removes the header if no traits are present
+    return ((0, jsx_runtime_1.jsxs)(jsx_runtime_1.Fragment, { children: [(0, jsx_runtime_1.jsx)(components_1.Text, { children: 'Consider this related information:\n' }), ...traits.map(trait => ((0, jsx_runtime_1.jsx)(components_1.Text, { source: trait, children: `${trait.name}: ${trait.value}` }, trait.id)))] }));
+};
+exports.Traits = Traits;
+//# sourceMappingURL=traits.js.map

package/dist/src/_internal/extension/completions-core/vscode-node/lib/src/prompt/components/traits.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"traits.js","sourceRoot":"","sources":["../../../../../../../../../../src/_internal/extension/completions-core/vscode-node/lib/src/prompt/components/traits.tsx"],"names":[],"mappings":";;;;AAAA;;;gGAGgG;AAChG,4BAA4B;AAC5B,uDAAuD;AAEvD,6EAA0G;AAC1G,0DAAoE;AACpE,uHAGwE;AAGjE,MAAM,MAAM,GAAG,CAAC,MAA0B,EAAE,OAAyB,EAAE,EAAE;IAC/E,MAAM,CAAC,MAAM,EAAE,SAAS,CAAC,GAAG,OAAO,CAAC,QAAQ,EAAiB,CAAC;IAC9D,MAAM,CAAC,UAAU,EAAE,aAAa,CAAC,GAAG,OAAO,CAAC,QAAQ,EAAU,CAAC;IAE/D,OAAO,CAAC,OAAO,CAAC,4DAAuB,EAAE,CAAC,IAA2B,EAAE,EAAE;QACxE,IAAI,IAAI,CAAC,MAAM,KAAK,MAAM,EAAE,CAAC;YAC5B,SAAS,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QACxB,CAAC;QAED,MAAM,oBAAoB,GAAG,IAAA,4BAAmB,EAAC,IAAI,CAAC,QAAQ,CAAC,kBAAkB,CAAC,CAAC;QACnF,IAAI,oBAAoB,KAAK,UAAU,EAAE,CAAC;YACzC,aAAa,CAAC,oBAAoB,CAAC,CAAC;QACrC,CAAC;IACF,CAAC,CAAC,CAAC;IAEH,IAAI,CAAC,MAAM,IAAI,MAAM,CAAC,MAAM,KAAK,CAAC,IAAI,CAAC,UAAU,EAAE,CAAC;QACnD,OAAO;IACR,CAAC;IAED,sFAAsF;IACtF,OAAO,CACN,6DACC,uBAAC,iBAAI,cAAE,sCAAsC,GAAQ,KACjD,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,CAAC,CACvB,uBAAC,iBAAI,IAAgB,MAAM,EAAE,KAAK,YAChC,GAAG,KAAK,CAAC,IAAI,KAAK,KAAK,CAAC,KAAK,EAAE,IADtB,KAAK,CAAC,EAAE,CAEZ,CACP,CAAC,IACA,CACH,CAAC;AACH,CAAC,CAAC;AA9BW,QAAA,MAAM,UA8BjB"}

package/dist/src/_internal/extension/completions-core/vscode-node/lib/src/prompt/contextProviderRegistry.d.ts

@@ -0,0 +1,90 @@
+import { CancellationToken, DocumentSelector } from 'vscode-languageserver-protocol';
+import { ILanguageContextProviderService } from '../../../../../../platform/languageContextProvider/common/languageContextProviderService';
+import { IInstantiationService, ServicesAccessor } from '../../../../../../util/vs/platform/instantiation/common/instantiation';
+import { ContextItemUsageDetails, ContextProvider, DocumentContext, ResolutionStatus, SupportedContextItem, UsageStatus } from '../../../types/src';
+import { ICompletionsLogTargetService } from '../logger';
+import { TelemetryWithExp } from '../telemetry';
+import { ICompletionsRuntimeModeService } from '../util/runtimeMode';
+import { SupportedContextItemWithId } from './contextProviders/contextItemSchemas';
+import { ICompletionsContextProviderService } from './contextProviderStatistics';
+export interface ResolvedContextItem<T extends SupportedContextItemWithId = SupportedContextItemWithId> {
+    providerId: string;
+    matchScore: number;
+    resolution: ResolutionStatus;
+    resolutionTimeMs: number;
+    data: T[];
+}
+export interface ContextProviderTelemetry {
+    providerId: string;
+    matched: boolean;
+    resolution: ResolutionStatus;
+    resolutionTimeMs: number;
+    usage: UsageStatus;
+    usageDetails?: ContextItemUsageDetails[];
+    numResolvedItems: number;
+    numUsedItems?: number;
+    numPartiallyUsedItems?: number;
+}
+export declare const ICompletionsContextProviderRegistryService: import("../../../../../../util/common/services").ServiceIdentifier<ICompletionsContextProviderRegistryService>;
+export interface ICompletionsContextProviderRegistryService {
+    readonly _serviceBrand: undefined;
+    registerContextProvider<T extends SupportedContextItem>(provider: ContextProvider<T>): void;
+    unregisterContextProvider(providerId: string): void;
+    providers: ContextProvider<SupportedContextItem>[];
+    resolveAllProviders(completionId: string, opportunityId: string, documentContext: DocumentContext, telemetryData: TelemetryWithExp, completionToken?: CancellationToken, data?: unknown): Promise<ResolvedContextItem[]>;
+}
+export type ActiveExperiments = Map<string, string | number | boolean | string[]>;
+export declare const ICompletionsDefaultContextProviders: import("../../../../../../util/common/services").ServiceIdentifier<ICompletionsDefaultContextProviders>;
+export interface ICompletionsDefaultContextProviders {
+    readonly _serviceBrand: undefined;
+    getIds(): string[];
+    add(id: string): void;
+}
+export declare class DefaultContextProvidersContainer implements ICompletionsDefaultContextProviders {
+    _serviceBrand: undefined;
+    private ids;
+    add(id: string): void;
+    getIds(): string[];
+}
+type ContextProviderMatchFunction = (instantiationService: IInstantiationService, documentSelector: DocumentSelector, documentContext: DocumentContext) => Promise<number> | number;
+export declare class CoreContextProviderRegistry implements ICompletionsContextProviderRegistryService {
+    private match;
+    private registryService;
+    private runtimeMode;
+    protected instantiationService: IInstantiationService;
+    protected logTarget: ICompletionsLogTargetService;
+    protected contextProviderStatistics: ICompletionsContextProviderService;
+    _serviceBrand: undefined;
+    constructor(match: ContextProviderMatchFunction, registryService: ILanguageContextProviderService, runtimeMode: ICompletionsRuntimeModeService, instantiationService: IInstantiationService, logTarget: ICompletionsLogTargetService, contextProviderStatistics: ICompletionsContextProviderService);
+    registerContextProvider<T extends SupportedContextItem>(_provider: ContextProvider<T>): void;
+    unregisterContextProvider(_providerId: string): void;
+    get providers(): ContextProvider<SupportedContextItem>[];
+    /**
+     * Resolves all context providers for the given context.
+     * Items returned will need to be filtered by schema.
+     */
+    resolveAllProviders(completionId: string, opportunityId: string, documentContext: DocumentContext, telemetryData: TelemetryWithExp, completionCancellationToken?: CancellationToken, data?: unknown): Promise<ResolvedContextItem[]>;
+    private matchProviders;
+}
+export declare class MutableContextProviderRegistry extends CoreContextProviderRegistry {
+    private _providers;
+    constructor(match: ContextProviderMatchFunction, registryService: ILanguageContextProviderService, runtimeMode: ICompletionsRuntimeModeService, instantiationService: IInstantiationService, logTarget: ICompletionsLogTargetService, contextProviderStatistics: ICompletionsContextProviderService);
+    registerContextProvider<T extends SupportedContextItem>(provider: ContextProvider<T>): void;
+    unregisterContextProvider(providerId: string): void;
+    get providers(): ContextProvider<SupportedContextItem>[];
+}
+export declare class CachedContextProviderRegistry implements ICompletionsContextProviderRegistryService {
+    _serviceBrand: undefined;
+    private _cachedContextItems;
+    private readonly delegate;
+    constructor(registry: new (match: ContextProviderMatchFunction) => CoreContextProviderRegistry, match: ContextProviderMatchFunction, instantiationService: IInstantiationService);
+    registerContextProvider<T extends SupportedContextItem>(provider: ContextProvider<T>): void;
+    unregisterContextProvider(providerId: string): void;
+    get providers(): ContextProvider<SupportedContextItem>[];
+    resolveAllProviders(completionId: string, opportunityId: string, documentContext: DocumentContext, telemetryData: TelemetryWithExp, completionToken?: CancellationToken, data?: unknown): Promise<ResolvedContextItem[]>;
+}
+export declare function telemetrizeContextItems(contextProvider: ICompletionsContextProviderService, completionId: string, resolvedContextItems: ResolvedContextItem[]): ContextProviderTelemetry[];
+export declare function matchContextItems(resolvedContextItem: ResolvedContextItem): boolean;
+export declare function useContextProviderAPI(accessor: ServicesAccessor, languageId: string, telemetryData: TelemetryWithExp): boolean;
+export {};
+//# sourceMappingURL=contextProviderRegistry.d.ts.map

package/dist/src/_internal/extension/completions-core/vscode-node/lib/src/prompt/contextProviderRegistry.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"contextProviderRegistry.d.ts","sourceRoot":"","sources":["../../../../../../../../../src/_internal/extension/completions-core/vscode-node/lib/src/prompt/contextProviderRegistry.ts"],"names":[],"mappings":"AAKA,OAAO,EAAE,iBAAiB,EAA2B,gBAAgB,EAAE,MAAM,gCAAgC,CAAC;AAC9G,OAAO,EAAE,+BAA+B,EAAkB,MAAM,0FAA0F,CAAC;AAG3J,OAAO,EAAE,qBAAqB,EAAE,gBAAgB,EAAE,MAAM,uEAAuE,CAAC;AAChI,OAAO,EACN,uBAAuB,EACvB,eAAe,EACf,eAAe,EACf,gBAAgB,EAGhB,oBAAoB,EACpB,WAAW,EACX,MAAM,oBAAoB,CAAC;AAI5B,OAAO,EAAE,4BAA4B,EAAU,MAAM,WAAW,CAAC;AACjE,OAAO,EAAE,gBAAgB,EAAE,MAAM,cAAc,CAAC;AAChD,OAAO,EAAE,8BAA8B,EAAE,MAAM,qBAAqB,CAAC;AAMrE,OAAO,EAGN,0BAA0B,EAC1B,MAAM,uCAAuC,CAAC;AAC/C,OAAO,EAAE,kCAAkC,EAAE,MAAM,6BAA6B,CAAC;AAEjF,MAAM,WAAW,mBAAmB,CAAC,CAAC,SAAS,0BAA0B,GAAG,0BAA0B;IACrG,UAAU,EAAE,MAAM,CAAC;IACnB,UAAU,EAAE,MAAM,CAAC;IACnB,UAAU,EAAE,gBAAgB,CAAC;IAC7B,gBAAgB,EAAE,MAAM,CAAC;IACzB,IAAI,EAAE,CAAC,EAAE,CAAC;CACV;AAED,MAAM,WAAW,wBAAwB;IACxC,UAAU,EAAE,MAAM,CAAC;IACnB,OAAO,EAAE,OAAO,CAAC;IACjB,UAAU,EAAE,gBAAgB,CAAC;IAC7B,gBAAgB,EAAE,MAAM,CAAC;IACzB,KAAK,EAAE,WAAW,CAAC;IACnB,YAAY,CAAC,EAAE,uBAAuB,EAAE,CAAC;IACzC,gBAAgB,EAAE,MAAM,CAAC;IACzB,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,qBAAqB,CAAC,EAAE,MAAM,CAAC;CAC/B;AAED,eAAO,MAAM,0CAA0C,gHAAoH,CAAC;AAC5K,MAAM,WAAW,0CAA0C;IAC1D,QAAQ,CAAC,aAAa,EAAE,SAAS,CAAC;IAElC,uBAAuB,CAAC,CAAC,SAAS,oBAAoB,EAAE,QAAQ,EAAE,eAAe,CAAC,CAAC,CAAC,GAAG,IAAI,CAAC;IAC5F,yBAAyB,CAAC,UAAU,EAAE,MAAM,GAAG,IAAI,CAAC;IACpD,SAAS,EAAE,eAAe,CAAC,oBAAoB,CAAC,EAAE,CAAC;IACnD,mBAAmB,CAClB,YAAY,EAAE,MAAM,EACpB,aAAa,EAAE,MAAM,EACrB,eAAe,EAAE,eAAe,EAChC,aAAa,EAAE,gBAAgB,EAC/B,eAAe,CAAC,EAAE,iBAAiB,EACnC,IAAI,CAAC,EAAE,OAAO,GACZ,OAAO,CAAC,mBAAmB,EAAE,CAAC,CAAC;CAClC;AAED,MAAM,MAAM,iBAAiB,GAAG,GAAG,CAAC,MAAM,EAAE,MAAM,GAAG,MAAM,GAAG,OAAO,GAAG,MAAM,EAAE,CAAC,CAAC;AAElF,eAAO,MAAM,mCAAmC,yGAAsG,CAAC;AACvJ,MAAM,WAAW,mCAAmC;IACnD,QAAQ,CAAC,aAAa,EAAE,SAAS,CAAC;IAClC,MAAM,IAAI,MAAM,EAAE,CAAC;IACnB,GAAG,CAAC,EAAE,EAAE,MAAM,GAAG,IAAI,CAAC;CACtB;AAED,qBAAa,gCAAiC,YAAW,mCAAmC;IACnF,aAAa,EAAE,SAAS,CAAC;IAEjC,OAAO,CAAC,GAAG,CAAgB;IAE3B,GAAG,CAAC,EAAE,EAAE,MAAM;IAId,MAAM,IAAI,MAAM,EAAE;CAGlB;AAED,KAAK,4BAA4B,GAAG,CACnC,oBAAoB,EAAE,qBAAqB,EAC3C,gBAAgB,EAAE,gBAAgB,EAClC,eAAe,EAAE,eAAe,KAC5B,OAAO,CAAC,MAAM,CAAC,GAAG,MAAM,CAAC;AAG9B,qBAAa,2BAA4B,YAAW,0CAA0C;IAI5F,OAAO,CAAC,KAAK;IACoB,OAAO,CAAC,eAAe;IACxB,OAAO,CAAC,WAAW;IAC5B,SAAS,CAAC,oBAAoB,EAAE,qBAAqB;IAC9C,SAAS,CAAC,SAAS,EAAE,4BAA4B;IAC3C,SAAS,CAAC,yBAAyB,EAAE,kCAAkC;IARpG,aAAa,EAAE,SAAS,CAAC;gBAGxB,KAAK,EAAE,4BAA4B,EACF,eAAe,EAAE,+BAA+B,EACjD,WAAW,EAAE,8BAA8B,EAClD,oBAAoB,EAAE,qBAAqB,EACpC,SAAS,EAAE,4BAA4B,EACjC,yBAAyB,EAAE,kCAAkC;IAG5G,uBAAuB,CAAC,CAAC,SAAS,oBAAoB,EAAE,SAAS,EAAE,eAAe,CAAC,CAAC,CAAC;IAIrF,yBAAyB,CAAC,WAAW,EAAE,MAAM;IAI7C,IAAI,SAAS,IAAI,eAAe,CAAC,oBAAoB,CAAC,EAAE,CAEvD;IAED;;;OAGG;IACG,mBAAmB,CACxB,YAAY,EAAE,MAAM,EACpB,aAAa,EAAE,MAAM,EACrB,eAAe,EAAE,eAAe,EAChC,aAAa,EAAE,gBAAgB,EAC/B,2BAA2B,CAAC,EAAE,iBAAiB,EAC/C,IAAI,CAAC,EAAE,OAAO,GACZ,OAAO,CAAC,mBAAmB,EAAE,CAAC;YA6KnB,cAAc;CAoB5B;AAED,qBAAa,8BAA+B,SAAQ,2BAA2B;IAE9E,OAAO,CAAC,UAAU,CAA+C;gBAGhE,KAAK,EAAE,4BAA4B,EACF,eAAe,EAAE,+BAA+B,EACjD,WAAW,EAAE,8BAA8B,EACpD,oBAAoB,EAAE,qBAAqB,EACpC,SAAS,EAAE,4BAA4B,EACjC,yBAAyB,EAAE,kCAAkC;IAKzF,uBAAuB,CAAC,CAAC,SAAS,oBAAoB,EAAE,QAAQ,EAAE,eAAe,CAAC,CAAC,CAAC;IAYpF,yBAAyB,CAAC,UAAU,EAAE,MAAM;IAIrD,IAAa,SAAS,4CAErB;CACD;AAED,qBAAa,6BAA8B,YAAW,0CAA0C;IACvF,aAAa,EAAE,SAAS,CAAC;IAGjC,OAAO,CAAC,mBAAmB,CAAkE;IAE7F,OAAO,CAAC,QAAQ,CAAC,QAAQ,CAA8B;gBAGtD,QAAQ,EAAE,KAAK,KAAK,EAAE,4BAA4B,KAAK,2BAA2B,EAClF,KAAK,EAAE,4BAA4B,EACZ,oBAAoB,EAAE,qBAAqB;IAKnE,uBAAuB,CAAC,CAAC,SAAS,oBAAoB,EAAE,QAAQ,EAAE,eAAe,CAAC,CAAC,CAAC,GAAG,IAAI;IAI3F,yBAAyB,CAAC,UAAU,EAAE,MAAM,GAAG,IAAI;IAInD,IAAI,SAAS,IAAI,eAAe,CAAC,oBAAoB,CAAC,EAAE,CAEvD;IAEK,mBAAmB,CACxB,YAAY,EAAE,MAAM,EACpB,aAAa,EAAE,MAAM,EACrB,eAAe,EAAE,eAAe,EAChC,aAAa,EAAE,gBAAgB,EAC/B,eAAe,CAAC,EAAE,iBAAiB,EACnC,IAAI,CAAC,EAAE,OAAO,GACZ,OAAO,CAAC,mBAAmB,EAAE,CAAC;CAsBjC;AAED,wBAAgB,uBAAuB,CACtC,eAAe,EAAE,kCAAkC,EACnD,YAAY,EAAE,MAAM,EACpB,oBAAoB,EAAE,mBAAmB,EAAE,8BAmD3C;AAED,wBAAgB,iBAAiB,CAAC,mBAAmB,EAAE,mBAAmB,GAAG,OAAO,CAEnF;AAuCD,wBAAgB,qBAAqB,CAAC,QAAQ,EAAE,gBAAgB,EAAE,UAAU,EAAE,MAAM,EAAE,aAAa,EAAE,gBAAgB,WAEpH"}