pacnpal/Roo-Code
1
0
Fork 0
mirror of https://github.com/pacnpal/Roo-Code.git synced 2025-12-20 12:21:13 -05:00
Code Issues Packages Projects Releases Wiki Activity

Only show warning if truncation comment is in new content

Browse Source
This commit is contained in:
Saoud Rizwan
2024-10-17 06:02:16 -04:00
parent 0d09897aa6
commit 717d83b5cb
2 changed files with 15 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
src/core/Cline.ts
View File

@@ -1075,8 +1075,7 @@ export class Cline {
await this.diffViewProvider.update(newContent, true)
await delay(300) // wait for diff view to update
this.diffViewProvider.scrollToFirstDiff()
showOmissionWarning(newContent)
showOmissionWarning(this.diffViewProvider.originalContent || "", newContent)
const completeMessage = JSON.stringify({
...sharedMessageProps,

21
src/integrations/editor/detect-omission.ts
View File

@@ -2,12 +2,14 @@ import * as vscode from "vscode"
/**
* Detects potential AI-generated code omissions in the given file content.
* @param fileContent The content of the file to check.
* @param originalFileContent The original content of the file.
* @param newFileContent The new content of the file to check.
* @returns True if a potential omission is detected, false otherwise.
*/
function detectCodeOmission(fileContent: string): boolean {
const lines = fileContent.split("\n")
const omissionKeywords = ["remain", "remains", "unchanged", "rest", "previous", "..."]
function detectCodeOmission(originalFileContent: string, newFileContent: string): boolean {
const originalLines = originalFileContent.split("\n")
const newLines = newFileContent.split("\n")
const omissionKeywords = ["remain", "remains", "unchanged", "rest", "previous", "existing", "..."]
const commentPatterns = [
/^\s*\/\//, // Single-line comment for most languages
@@ -16,24 +18,27 @@ function detectCodeOmission(fileContent: string): boolean {
/^\s*<!--/, // HTML comment opening
]
for (const line of lines) {
for (const line of newLines) {
if (commentPatterns.some((pattern) => pattern.test(line))) {
const words = line.toLowerCase().split(/\s+/)
if (omissionKeywords.some((keyword) => words.includes(keyword))) {
if (!originalLines.includes(line)) {
return true
}
}
}
}
return false
}
/**
* Shows a warning in VSCode if a potential code omission is detected.
* @param fileContent The content of the file to check.
* @param originalFileContent The original content of the file.
* @param newFileContent The new content of the file to check.
*/
export function showOmissionWarning(fileContent: string): void {
if (detectCodeOmission(fileContent)) {
export function showOmissionWarning(originalFileContent: string, newFileContent: string): void {
if (detectCodeOmission(originalFileContent, newFileContent)) {
vscode.window
.showWarningMessage(
"Potential code truncation detected. This happens when the AI reaches its max output limit.",