Skip to content

Commit 64bf782

Browse files
committed
feat: update getCurrentDateAndModelPrompt
1 parent 21a3add commit 64bf782

1 file changed

Lines changed: 6 additions & 3 deletions

File tree

src/prompt.ts

Lines changed: 6 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -281,14 +281,17 @@ function readToolDocs(extensionRoot: string, options: PromptToolOptions = {}): s
281281
return docs.join("\n\n");
282282
}
283283

284-
function getCurrentDatePrompt(date = new Date()): string {
285-
return `今天是${date.getFullYear()}${date.getMonth() + 1}${date.getDate()}日。随着对话的进行,时间在流逝。`;
284+
function getCurrentDateAndModelPrompt(model?: string): string {
285+
const date = new Date();
286+
let prompt = `今天是${date.getFullYear()}${date.getMonth() + 1}${date.getDate()}日。随着对话的进行,时间在流逝。`;
287+
prompt += model ? `\n当前LLM模型为${model},对话中可通过/model命令切换模型。` : "";
288+
return prompt;
286289
}
287290

288291
export function getSystemPrompt(projectRoot: string, options: PromptToolOptions = {}): string {
289292
const toolDocs = readToolDocs(getExtensionRoot(), options);
290293
const basePrompt = toolDocs ? `${SYSTEM_PROMPT_BASE}\n\n# Available Tools\n\n${toolDocs}` : SYSTEM_PROMPT_BASE;
291-
return `${basePrompt}\n\n${getCurrentDatePrompt()}\n\n${getRuntimeContext(projectRoot)}`;
294+
return `${basePrompt}\n\n${getCurrentDateAndModelPrompt(options.model)}\n\n${getRuntimeContext(projectRoot)}`;
292295
}
293296

294297
export function getCompactPrompt(sessionMessages: SessionMessage[]): string {

0 commit comments

Comments
 (0)