Skip to content

Commit 16c8288

Browse files
committed
updated snippet filtering and validation
1 parent 841818d commit 16c8288

File tree

2 files changed

+13
-14
lines changed

2 files changed

+13
-14
lines changed

core/autocomplete/templating/filtering.ts

Lines changed: 4 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -3,6 +3,7 @@ import { SnippetPayload } from "../snippets";
33
import {
44
AutocompleteCodeSnippet,
55
AutocompleteSnippet,
6+
AutocompleteSnippetType,
67
} from "../snippets/types";
78
import { HelperVars } from "../util/HelperVars";
89
import { formatOpenedFilesContext } from "./formatOpenedFilesContext";
@@ -137,17 +138,9 @@ export const getSnippets = (
137138
for (const { key } of snippetOrder) {
138139
// Special handling for recentlyOpenedFiles
139140
if (key === "recentlyOpenedFiles" && helper.options.useRecentlyOpened) {
140-
const recentlyOpenedFilesSnippets =
141-
payload.recentlyOpenedFileSnippets.filter(
142-
(snippet) =>
143-
!(snippet as AutocompleteCodeSnippet).filepath?.startsWith(
144-
"output:extension-output-Continue.continue",
145-
),
146-
);
147-
148141
// Custom trimming
149142
const processedSnippets = formatOpenedFilesContext(
150-
recentlyOpenedFilesSnippets,
143+
payload.recentlyOpenedFileSnippets,
151144
remainingTokenCount,
152145
helper,
153146
finalSnippets,
@@ -173,11 +166,8 @@ export const getSnippets = (
173166
// Normal processing for other snippet types
174167
const snippetsToProcess = snippets[key].filter(
175168
(snippet) =>
176-
!(snippet as AutocompleteCodeSnippet).filepath?.startsWith(
177-
"output:extension-output-Continue.continue",
178-
) &&
179-
((snippet as AutocompleteCodeSnippet).filepath === undefined ||
180-
!addedFilepaths.has((snippet as AutocompleteCodeSnippet).filepath)),
169+
snippet.type !== AutocompleteSnippetType.Code ||
170+
!addedFilepaths.has(snippet.filepath),
181171
);
182172

183173
for (const snippet of snippetsToProcess) {

core/autocomplete/templating/validation.ts

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,6 @@
11
import {
22
AutocompleteClipboardSnippet,
3+
AutocompleteCodeSnippet,
34
AutocompleteSnippet,
45
AutocompleteSnippetType,
56
} from "../snippets/types";
@@ -25,5 +26,13 @@ export const isValidSnippet = (snippet: AutocompleteSnippet): boolean => {
2526
return isValidClipboardSnippet(snippet);
2627
}
2728

29+
if (
30+
(snippet as AutocompleteCodeSnippet).filepath?.startsWith(
31+
"output:extension-output-Continue.continue",
32+
)
33+
) {
34+
return false;
35+
}
36+
2837
return true;
2938
};

0 commit comments

Comments
 (0)