gitea/infio-copilot
1
0
Fork 0
mirror of https://github.com/EthanMarti/infio-copilot.git synced 2026-01-16 08:21:55 +00:00
Code Issues Packages Projects Releases Wiki Activity

update current active file bug

Browse Source
This commit is contained in:
duanfuxiang 2025-06-04 18:43:40 +08:00
parent 05aefa0fc7
commit 2a14fd3b70
1 changed files with 40 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

44
src/utils/prompt-generator.ts
View File

@ -169,6 +169,7 @@ export class PromptGenerator {
await this.compileUserMessagePrompt({
isNewChat,
message: lastUserMessage,
messages,
useVaultSearch,
onQueryProgressChange,
})
@ -284,11 +285,13 @@ export class PromptGenerator {
private async compileUserMessagePrompt({
isNewChat,
message,
messages,
useVaultSearch,
onQueryProgressChange,
}: {
isNewChat: boolean
message: ChatUserMessage
messages?: ChatMessage[]
useVaultSearch?: boolean
onQueryProgressChange?: (queryProgress: QueryProgressState) => void
}): Promise<{
@ -298,9 +301,10 @@ export class PromptGenerator {
})[]
}> {
// Add environment details
const environmentDetails = isNewChat
? await this.getEnvironmentDetails()
: undefined
// const environmentDetails = isNewChat
// ? await this.getEnvironmentDetails()
// : undefined
const environmentDetails = await this.getEnvironmentDetails()
// if isToolCallReturn, add read_file_content to promptContent
if (message.content === null) {
@ -378,6 +382,7 @@ export class PromptGenerator {
))
.join('\n') : undefined
// current file
const currentFile = message.mentionables
.filter((m): m is MentionableFile => m.type === 'current-file')
.first()
@ -385,7 +390,38 @@ export class PromptGenerator {
? await getFileOrFolderContent(currentFile.file, this.app.vault)
: undefined
const currentFileContentPrompt = isNewChat && currentFileContent && this.settings.mode !== 'research'
// Check if current file content should be included
let shouldIncludeCurrentFile = false
if (currentFileContent && this.settings.mode !== 'research') {
if (isNewChat) {
// For new chats, always include current file content
shouldIncludeCurrentFile = true
} else {
// For continuing chats, check if current file content already exists in history
const currentFilePromptTag = `<current_file_content path="${currentFile.file.path}">`
const hasCurrentFileInHistory = messages?.some((msg) => {
if (msg.role === 'user' && msg.promptContent) {
if (typeof msg.promptContent === 'string') {
// Handle string type promptContent
return msg.promptContent.includes(currentFilePromptTag)
} else if (Array.isArray(msg.promptContent)) {
// Handle ContentPart[] type promptContent
return msg.promptContent.some((part) => {
if (part.type === 'text' && part.text) {
return part.text.includes(currentFilePromptTag)
}
return false
})
}
}
return false
}) || false
// Only include if not already in history
shouldIncludeCurrentFile = !hasCurrentFileInHistory
}
}
const currentFileContentPrompt = shouldIncludeCurrentFile
? `<current_file_content path="${currentFile.file.path}">\n${currentFileContent}\n</current_file_content>`
: undefined