Skip to content

Commit d1b5d42

Browse files
authored
1 parent 88bc75f commit d1b5d42

File tree

2 files changed

+12
-13
lines changed

2 files changed

+12
-13
lines changed

src/vs/workbench/contrib/chat/common/chatViewModel.ts

+1-12
Original file line numberDiff line numberDiff line change
@@ -274,24 +274,13 @@ export class ChatViewModel extends Disposable implements IChatViewModel {
274274
const renderer = new marked.Renderer();
275275
renderer.code = (value, languageId) => {
276276
languageId ??= '';
277-
const newText = this.fixCodeText(value, languageId);
278-
this.codeBlockModelCollection.update(this._model.sessionId, model, codeBlockIndex++, { text: newText, languageId });
277+
this.codeBlockModelCollection.update(this._model.sessionId, model, codeBlockIndex++, { text: value, languageId });
279278
return '';
280279
};
281280

282281
marked.parse(this.ensureFencedCodeBlocksTerminated(content), { renderer });
283282
}
284283

285-
private fixCodeText(text: string, languageId: string): string {
286-
if (languageId === 'php') {
287-
if (!text.trim().startsWith('<')) {
288-
return `<?php\n${text}\n?>`;
289-
}
290-
}
291-
292-
return text;
293-
}
294-
295284
/**
296285
* Marked doesn't consistently render fenced code blocks that aren't terminated.
297286
*

src/vs/workbench/contrib/chat/common/codeBlockModelCollection.ts

+11-1
Original file line numberDiff line numberDiff line change
@@ -64,7 +64,7 @@ export class CodeBlockModelCollection extends Disposable {
6464
const entry = this.getOrCreate(sessionId, chat, codeBlockIndex);
6565

6666
const extractedVulns = extractVulnerabilitiesFromText(content.text);
67-
const newText = extractedVulns.newText;
67+
const newText = fixCodeText(extractedVulns.newText, content.languageId);
6868
this.setVulns(sessionId, chat, codeBlockIndex, extractedVulns.vulnerabilities);
6969

7070
const textModel = (await entry.model).textEditorModel;
@@ -137,3 +137,13 @@ export class CodeBlockModelCollection extends Disposable {
137137
};
138138
}
139139
}
140+
141+
function fixCodeText(text: string, languageId: string | undefined): string {
142+
if (languageId === 'php') {
143+
if (!text.trim().startsWith('<')) {
144+
return `<?php\n${text}`;
145+
}
146+
}
147+
148+
return text;
149+
}

0 commit comments

Comments
 (0)