HOW TO ADD LOG IN CONTINUE

[chenglu66]

Validations

• [x] I believe this is a way to improve. I'll try to join the Continue Discord for questions
• [x] I'm not able to find an open issue that requests the same enhancement

Problem

when i use edit function，i found there was a bug in it ，so i want to add some log to debug。but i don‘t know how to add log。I try this way “”“ import { ChatMessage, DiffLine, ILLM, Prediction } from "../"; import { filterCodeBlockLines, filterEnglishLinesAtEnd, filterEnglishLinesAtStart, filterLeadingAndTrailingNewLineInsertion, removeTrailingWhitespace, skipLines, stopAtLines, } from "../autocomplete/filtering/streamTransforms/lineStream"; import { streamDiff } from "../diff/streamDiff"; import { streamLines } from "../diff/util"; import { gptEditPrompt } from "../llm/templates/edit"; import { Telemetry } from "../util/posthog"; import { getCoreLogPath } from "../util/paths"; // 引入日志文件路径获取函数

function constructPrompt( prefix: string, highlighted: string, suffix: string, llm: ILLM, userInput: string, language: string | undefined, ): string | ChatMessage[] { const template = llm.promptTemplates?.edit ?? gptEditPrompt; return llm.renderPromptTemplate(template, [], { userInput, prefix, codeToEdit: highlighted, suffix, language: language ?? "", }); }

export async function* addIndentation( diffLineGenerator: AsyncGenerator<DiffLine>, indentation: string, ): AsyncGenerator<DiffLine> { for await (const diffLine of diffLineGenerator) { yield { ...diffLine, line: indentation + diffLine.line, }; } }

function modelIsInept(model: string): boolean { return !(model.includes("gpt") || model.includes("claude")); }

export async function* streamDiffLines( prefix: string, highlighted: string, suffix: string, llm: ILLM, input: string, language: string | undefined, onlyOneInsertion: boolean, overridePrompt: ChatMessage[] | undefined, ): AsyncGenerator<DiffLine> { const logPath = getCoreLogPath(); // 获取日志文件路径 const logStream = fs.createWriteStream(logPath, { flags: 'a' }); // 创建日志文件流

void Telemetry.capture( "inlineEdit", { model: llm.model, provider: llm.providerName, }, true, );

logStream.write(Starting streamDiffLines with model: ${llm.model}\n); // 添加日志

// Strip common indentation for the LLM, then add back after generation let oldLines = highlighted.length > 0 ? highlighted.split("\n") : // When highlighted is empty, we need to combine last line of prefix and first line of suffix to determine the line being edited [(prefix + suffix).split("\n")[prefix.split("\n").length - 1]];

// But if that line is empty, we can assume we are insertion-only if (oldLines.length === 1 && oldLines[0].trim() === "") { oldLines = []; }

const prompt = overridePrompt ?? constructPrompt(prefix, highlighted, suffix, llm, input, language); const inept = modelIsInept(llm.model);

const prediction: Prediction = { type: "content", content: highlighted, };

const completion = typeof prompt === "string" ? llm.streamComplete(prompt, new AbortController().signal, { raw: true, prediction, }) : llm.streamChat(prompt, new AbortController().signal, { prediction, });

let lines = streamLines(completion);

lines = filterEnglishLinesAtStart(lines); lines = filterCodeBlockLines(lines); lines = stopAtLines(lines, () => {}); lines = skipLines(lines); lines = removeTrailingWhitespace(lines); if (inept) { // lines = fixCodeLlamaFirstLineIndentation(lines); lines = filterEnglishLinesAtEnd(lines); }

let diffLines = streamDiff(oldLines, lines); diffLines = filterLeadingAndTrailingNewLineInsertion(diffLines); if (highlighted.length === 0) { const line = prefix.split("\n").slice(-1)[0]; const indentation = line.slice(0, line.length - line.trimStart().length); diffLines = addIndentation(diffLines, indentation); }

let seenGreen = false; for await (const diffLine of diffLines) { logStream.write(Generated diffLine: ${JSON.stringify(diffLine)}\n); // 添加日志 yield diffLine; if (diffLine.type === "new") { seenGreen = true; } else if (onlyOneInsertion && seenGreen && diffLine.type === "same") { logStream.write(Breaking loop due to onlyOneInsertion condition\n); // 添加日志 break; } }

logStream.end(); // 关闭日志文件流 } ”“” but it does not work 。

Solution

can you instruct me how to add log