aiexecode 1.0.72 → 1.0.74

package/src/system/conversation_trimmer.js

@@ -0,0 +1,133 @@
+/**
+ * 대화 히스토리 trim 및 정리 유틸리티
+ *
+ * orchestrator와 completion_judge에서 공통으로 사용하는
+ * 대화 관리 기능을 중앙집중화합니다.
+ */
+
+import { createDebugLogger } from "../util/debug_log.js";
+
+const debugLog = createDebugLogger('conversation_trimmer.log', 'conversation_trimmer');
+
+/**
+ * 대화에서 특정 call_id와 관련된 모든 항목을 제거합니다.
+ *
+ * @param {Array} conversation - 대화 배열
+ * @param {string} callIdToRemove - 제거할 call_id
+ */
+export function removeEntriesWithCallId(conversation, callIdToRemove) {
+    if (!callIdToRemove || typeof callIdToRemove !== 'string') {
+        return;
+    }
+
+    debugLog(`[removeEntriesWithCallId] Looking for entries with call_id: ${callIdToRemove}`);
+
+    // 1단계: 삭제할 항목들의 인덱스를 먼저 수집 (반복 중에는 배열 수정 금지)
+    const indicesToRemove = [];
+
+    for (let i = 0; i < conversation.length; i++) {
+        const entry = conversation[i];
+
+        // call_id 속성이 있고 값이 일치하면 삭제 목록에 추가
+        if (entry && entry.call_id === callIdToRemove) {
+            indicesToRemove.push(i);
+            debugLog(`[removeEntriesWithCallId] Marked index ${i} for removal (type: ${entry.type})`);
+        }
+    }
+
+    // 2단계: 수집된 인덱스를 역순으로 삭제 (뒤에서부터 지워야 인덱스가 꼬이지 않음)
+    for (let i = indicesToRemove.length - 1; i >= 0; i--) {
+        const indexToRemove = indicesToRemove[i];
+        conversation.splice(indexToRemove, 1);
+        debugLog(`[removeEntriesWithCallId] Removed entry at index ${indexToRemove}`);
+    }
+
+    debugLog(`[removeEntriesWithCallId] Total removed: ${indicesToRemove.length} entries`);
+}
+
+/**
+ * 고아(orphan) function_call_output을 제거합니다.
+ *
+ * 고아 output이란?
+ * - function_call은 없는데 function_call_output만 남아있는 경우
+ * - 예: function_call을 trim했는데 output은 뒤에 남아있는 경우
+ *
+ * @param {Array} conversation - 대화 배열
+ */
+export function cleanupOrphanOutputs(conversation) {
+    debugLog(`[cleanupOrphanOutputs] Starting cleanup, conversation length: ${conversation.length}`);
+
+    // 1단계: "어떤 function_call들이 대화에 있는가?" 파악
+    const existingCallIds = new Set();
+
+    for (const entry of conversation) {
+        if (entry?.type === 'function_call' && entry.call_id) {
+            existingCallIds.add(entry.call_id);
+        }
+    }
+
+    debugLog(`[cleanupOrphanOutputs] Found ${existingCallIds.size} function_calls`);
+
+    // 2단계: "고아 output 찾기" - call_id가 없거나, 대응하는 function_call이 없는 output
+    const orphanIndexes = [];
+
+    for (let i = 0; i < conversation.length; i++) {
+        const entry = conversation[i];
+
+        // output이 아니면 패스
+        if (entry?.type !== 'function_call_output') {
+            continue;
+        }
+
+        const outputCallId = entry.call_id;
+
+        // call_id가 없거나, 대응하는 function_call이 없으면 고아
+        const isOrphan = !outputCallId || !existingCallIds.has(outputCallId);
+
+        if (isOrphan) {
+            orphanIndexes.push(i);
+            debugLog(`[cleanupOrphanOutputs] Found orphan at index ${i}, call_id: ${outputCallId}`);
+        }
+    }
+
+    // 3단계: 고아들을 뒤에서부터 제거 (앞에서부터 지우면 인덱스가 밀림)
+    for (let i = orphanIndexes.length - 1; i >= 0; i--) {
+        const indexToRemove = orphanIndexes[i];
+        conversation.splice(indexToRemove, 1);
+    }
+
+    debugLog(`[cleanupOrphanOutputs] Removed ${orphanIndexes.length} orphans. New length: ${conversation.length}`);
+}
+
+/**
+ * 컨텍스트 윈도우 초과 시 대화를 줄입니다.
+ * 시스템 프롬프트(인덱스 0)를 제외한 가장 오래된 항목(인덱스 1)을 제거합니다.
+ *
+ * @param {Array} conversation - 대화 배열
+ * @returns {boolean} 제거 성공 여부
+ */
+export function trimConversation(conversation) {
+    // 시스템 프롬프트와 최소한 하나의 항목이 필요
+    if (conversation.length <= 2) {
+        debugLog(`[trimConversation] Cannot trim - conversation too short (${conversation.length} entries)`);
+        return false;
+    }
+
+    // 인덱스 1 (시스템 프롬프트 제외 첫 번째 항목) 선택
+    const targetEntry = conversation[1];
+    const targetCallId = targetEntry?.call_id;
+
+    debugLog(`[trimConversation] Target entry at index 1: type=${targetEntry?.type}, call_id=${targetCallId}`);
+
+    // 먼저 인덱스 1 항목 제거
+    conversation.splice(1, 1);
+    debugLog(`[trimConversation] Removed entry at index 1`);
+
+    // call_id가 있으면 같은 call_id를 가진 다른 항목들도 제거
+    if (targetCallId) {
+        removeEntriesWithCallId(conversation, targetCallId);
+    }
+
+    debugLog(`[trimConversation] Trim completed. New conversation length: ${conversation.length}`);
+    return true;
+}

package/src/system/tool_approval.js

@@ -43,8 +43,7 @@ async function willDefinitelyFail(toolName, args) {
         try {
             const absolutePath = resolve(args.file_path);
 
-            // 스냅샷만 확인 (실제 파일을 읽지 않음)
-            // 편집 도구는 반드시 먼저 read_file로 읽어야 함
+            // 스냅샷 확인 (파일을 먼저 read_file로 읽었는지 확인)
             const snapshot = getFileSnapshot(absolutePath);
             const content = snapshot?.content;
 
@@ -53,7 +52,7 @@ async function willDefinitelyFail(toolName, args) {
                 return { willFail: true, reason: 'File not read yet or empty' };
             }
 
-            // old_string이 파일에 없는 경우
+            // old_string이 스냅샷에 없는 경우
             const oldString = args.old_string || '';
             if (!content.includes(oldString)) {
                 return { willFail: true, reason: 'old_string not found in file' };
@@ -64,13 +63,12 @@ async function willDefinitelyFail(toolName, args) {
         }
     }
 
-    // edit_file_range: 스냅샷이 없는 경우
+    // edit_file_range: 스냅샷이 없거나 라인 범위가 잘못된 경우
     if (toolName === 'edit_file_range') {
         try {
             const absolutePath = resolve(args.file_path);
 
-            // 스냅샷만 확인 (실제 파일을 읽지 않음)
-            // 편집 도구는 반드시 먼저 read_file로 읽어야 함
+            // 스냅샷 확인 (파일을 먼저 read_file로 읽었는지 확인)
             const snapshot = getFileSnapshot(absolutePath);
             const content = snapshot?.content;
 
@@ -78,6 +76,19 @@ async function willDefinitelyFail(toolName, args) {
             if (content === undefined || content === null) {
                 return { willFail: true, reason: 'File not read yet or empty' };
             }
+
+            // 라인 범위 검증
+            const lines = content.split('\n');
+            const totalLines = content === '' ? 0 :
+                (content.endsWith('\n') ? lines.length - 1 : lines.length);
+
+            if (args.start_line < 1 || args.end_line < args.start_line - 1) {
+                return { willFail: true, reason: 'Invalid line range' };
+            }
+
+            if (args.start_line > totalLines + 1 || args.end_line > totalLines) {
+                return { willFail: true, reason: `Line range exceeds file length (${totalLines} lines)` };
+            }
         } catch (error) {
             return { willFail: true, reason: `Validation error: ${error.message}` };
         }

package/src/tools/code_editor.js

@@ -442,7 +442,6 @@ export async function edit_file_replace({ file_path, old_string, new_string }) {
         return {
             operation_successful: true,
             target_file_path: absolutePath,
-            absolute_file_path: absolutePath,
             replacement_count: replacementCount,
             fileSnapshot: fileSnapshot,  // UI 히스토리에서 정확한 diff 표시를 위해 편집 전 스냅샷 포함
             file_stats: {
@@ -654,7 +653,6 @@ export async function edit_file_range({ file_path, start_line, end_line, new_con
             operation_successful: true,
             operation_type: operationType,
             target_file_path: absolutePath,
-            absolute_file_path: absolutePath,
             fileSnapshot: fileSnapshot,  // UI 히스토리에서 정확한 diff 표시를 위해 편집 전 스냅샷 포함
             file_stats: {
                 updated_content: newFileContent  // 수정 후 전체 파일 내용

package/src/tools/file_reader.js

@@ -1,5 +1,6 @@
 import { safeReadFile } from '../util/safe_fs.js';
 import { resolve } from 'path';
+import { createHash } from 'crypto';
 import { trackFileRead, saveFileSnapshot } from '../system/file_integrity.js';
 import { createDebugLogger } from '../util/debug_log.js';
 import { toDisplayPath } from '../util/path_helper.js';
@@ -77,6 +78,11 @@ export async function read_file({ filePath }) {
         saveFileSnapshot(absolutePath, content);
         debugLog(`Snapshot saved`);
 
+        // MD5 해시 계산
+        debugLog(`Calculating MD5 hash...`);
+        const md5Hash = createHash('md5').update(content).digest('hex');
+        debugLog(`MD5 hash: ${md5Hash}`);
+
         debugLog('========== read_file SUCCESS END ==========');
 
         return {
@@ -84,8 +90,7 @@ export async function read_file({ filePath }) {
             target_file_path: absolutePath,
             total_line_count: totalLines,
             file_content: content,
-            file_lines: content === '' ? [] :
-                (content.endsWith('\n') ? lines.slice(0, -1) : lines)
+            md5_hash: md5Hash
         };
     } catch (error) {
         debugLog(`========== read_file EXCEPTION ==========`);
@@ -200,6 +205,11 @@ export async function read_file_range({ filePath, startLine, endLine }) {
         saveFileSnapshot(absolutePath, content);
         debugLog(`Snapshot saved`);
 
+        // MD5 해시 계산 (전체 파일 내용에 대해)
+        debugLog(`Calculating MD5 hash...`);
+        const md5Hash = createHash('md5').update(content).digest('hex');
+        debugLog(`MD5 hash: ${md5Hash}`);
+
         // 라인 번호 유효성 검사 (1부터 시작)
         debugLog(`Validating line range...`);
         if (startLine < 1) {
@@ -225,16 +235,6 @@ export async function read_file_range({ filePath, startLine, endLine }) {
         const selectedLines = actualLines.slice(startLine - 1, endLine);
         debugLog(`Extracted ${selectedLines.length} lines`);
 
-        const numberedLines = selectedLines.map((line, index) => ({
-            line_number: startLine + index,
-            line_content: line
-        }));
-
-        // file_content에 라인 번호 추가
-        const contentWithLineNumbers = selectedLines
-            .map((line, index) => `${startLine + index}| ${line}`)
-            .join('\n');
-
         debugLog('========== read_file_range SUCCESS END ==========');
 
         return {
@@ -246,7 +246,8 @@ export async function read_file_range({ filePath, startLine, endLine }) {
                 start_line: startLine,
                 end_line: Math.min(endLine, totalLines)
             },
-            file_content: contentWithLineNumbers
+            file_content: selectedLines.join('\n'),
+            md5_hash: md5Hash
         };
     } catch (error) {
         debugLog(`========== read_file_range EXCEPTION ==========`);
@@ -299,7 +300,7 @@ export const readFileRangeSchema = {
         },
         "format_tool_result": (result) => {
             if (result.operation_successful) {
-                const lineCount = result.file_lines?.length || 0;
+                const lineCount = result.file_content ? result.file_content.split('\n').length : 0;
                 return {
                     type: 'formatted',
                     parts: [

package/src/util/exit_handler.js

@@ -27,6 +27,15 @@ export async function performExit(options = {}) {
         await uiEvents.waitForRender();
     }
 
+    // UI 인스턴스 정리 전에 터미널 상태 복구
+    // 이렇게 하면 React cleanup이 실행되지 않아도 터미널 상태가 복구됨
+    if (process.stdout && process.stdout.write) {
+        // 커서 표시 복구 (App.js:447과 동일)
+        process.stdout.write('\x1B[?25h');
+        // Line wrapping 복구 (index.js:54와 동일)
+        process.stdout.write('\x1b[?7h');
+    }
+
     // UI 인스턴스 정리
     if (uiInstance) {
         uiInstance.unmount();

package/src/util/output_formatter.js

@@ -22,16 +22,20 @@ export function clampOutput(text) {
 
 /**
  * 파일 읽기 결과에 줄 번호 추가
- * @param {Object} result - 파일 읽기 실행 결과
+ * @param {string} content - 파일 내용 (문자열)
+ * @param {number} startLine - 시작 줄 번호 (기본값: 1)
  * @returns {string} 줄 번호가 포함된 파일 내용
  */
-export function formatReadFileStdout(result) {
-    let stdout = '';
-    if (result.file_lines) {
-        result.file_lines.map((line, line_number) => {
-            stdout += `${line_number + 1}|${line}\n`;
-        });
+export function formatReadFileStdout(content, startLine = 1) {
+    if (!content || typeof content !== 'string') {
+        return '';
     }
+
+    const lines = content.split('\n');
+    let stdout = '';
+    lines.forEach((line, index) => {
+        stdout += `${startLine + index}|${line}\n`;
+    });
     return stdout;
 }
 
@@ -42,10 +46,5 @@ export function formatReadFileStdout(result) {
  * @returns {string} 포맷된 출력
  */
 export function formatToolStdout(operation, result) {
-    switch (operation) {
-        case 'read_file':
-            return formatReadFileStdout(result);
-        default:
-            return JSON.stringify(result, null, 2);
-    }
+    return JSON.stringify(result, null, 2);
 }