aiexecode 1.0.126 → 1.0.128

package/src/system/output_helper.js

@@ -1,35 +1,70 @@
 /**
  * Output Helper - UI 이벤트 발생
+ * Pipe mode에서는 stderr로 출력하거나 무시
  */
 
 import { uiEvents } from './ui_events.js';
 
+// Pipe mode 감지
+const isPipeMode = () => process.app_custom?.pipeMode === true;
+
 export function logSystem(message) {
-    uiEvents.addSystemMessage(message);
+    if (isPipeMode()) {
+        console.error(`[SYSTEM] ${message}`);
+    } else {
+        uiEvents.addSystemMessage(message);
+    }
 }
 
 export function logError(message) {
-    uiEvents.addErrorMessage(message);
+    if (isPipeMode()) {
+        console.error(`[ERROR] ${message}`);
+    } else {
+        uiEvents.addErrorMessage(message);
+    }
 }
 
 export function logAssistantMessage(message) {
-    uiEvents.addAssistantMessage(message);
+    if (isPipeMode()) {
+        console.error(`[ASSISTANT] ${message}`);
+    } else {
+        uiEvents.addAssistantMessage(message);
+    }
 }
 
 export function logToolCall(toolName, args) {
-    uiEvents.startToolExecution(toolName, args);
+    if (isPipeMode()) {
+        console.error(`[TOOL] ${toolName}`);
+    } else {
+        uiEvents.startToolExecution(toolName, args);
+    }
 }
 
 export function logToolResult(toolName, stdout, originalResult = null, toolInput = null, fileSnapshot = null) {
-    uiEvents.addToolResult(toolName, { stdout, originalResult, fileSnapshot }, toolInput);
+    if (isPipeMode()) {
+        // Pipe mode에서는 도구 결과를 간략하게 stderr로
+        const preview = stdout?.substring(0, 200) || '';
+        console.error(`[TOOL_RESULT] ${toolName}: ${preview}${stdout?.length > 200 ? '...' : ''}`);
+    } else {
+        uiEvents.addToolResult(toolName, { stdout, originalResult, fileSnapshot }, toolInput);
+    }
 }
 
 export function logCodeExecution(language, code) {
-    uiEvents.startCodeExecution(language, code);
+    if (isPipeMode()) {
+        console.error(`[CODE] ${language}`);
+    } else {
+        uiEvents.startCodeExecution(language, code);
+    }
 }
 
 export function logCodeResult(stdout, stderr, exitCode) {
-    uiEvents.addCodeResult(stdout, stderr, exitCode);
+    if (isPipeMode()) {
+        if (stdout) console.error(`[CODE_STDOUT] ${stdout.substring(0, 200)}${stdout.length > 200 ? '...' : ''}`);
+        if (stderr) console.error(`[CODE_STDERR] ${stderr.substring(0, 200)}${stderr.length > 200 ? '...' : ''}`);
+    } else {
+        uiEvents.addCodeResult(stdout, stderr, exitCode);
+    }
 }
 
 export function logIteration(iterationNumber) {
@@ -38,9 +73,17 @@ export function logIteration(iterationNumber) {
 }
 
 export function logMissionComplete() {
-    uiEvents.missionCompleted(0);
+    if (isPipeMode()) {
+        console.error('[COMPLETE] Mission completed');
+    } else {
+        uiEvents.missionCompleted(0);
+    }
 }
 
 export function logConversationRestored(count) {
-    uiEvents.conversationRestored(count);
+    if (isPipeMode()) {
+        console.error(`[RESTORED] ${count} conversation(s) restored`);
+    } else {
+        uiEvents.conversationRestored(count);
+    }
 }

package/src/system/session.js

@@ -1,7 +1,8 @@
 // Agent Loop: 계획·실행·검증 사이클을 수행하는 핵심 모듈
 import { orchestrateMission, continueOrchestratorConversation, resetOrchestratorConversation, recordOrchestratorToolResult, getOrchestratorConversation, restoreOrchestratorConversation } from "../ai_based/orchestrator.js";
 import { judgeMissionCompletion } from "../ai_based/completion_judge.js";
-import { execPythonCode, execShellScript } from "./code_executer.js";
+import { execPythonCode, execShellScript, resetShellCwd, getPersistentShell } from "./code_executer.js";
+import { runInBackground, getBackgroundProcess } from "./background_process.js";
 import { FILE_READER_FUNCTIONS } from "../tools/file_reader.js";
 import { RIPGREP_FUNCTIONS } from "../tools/ripgrep.js";
 import { GLOB_FUNCTIONS } from "../tools/glob.js";
@@ -30,6 +31,47 @@ import {
 
 const debugLog = createDebugLogger('session.log', 'session');
 
+// Pipe mode helper: UI 이벤트를 조건부로 처리
+const isPipeMode = () => process.app_custom?.pipeMode === true;
+
+// Pipe mode에서는 stderr로 출력, 일반 모드에서는 uiEvents 사용
+const safeUiEvents = {
+    addSystemMessage: (msg) => {
+        if (isPipeMode()) {
+            console.error(`[SYSTEM] ${msg}`);
+        } else {
+            uiEvents.addSystemMessage(msg);
+        }
+    },
+    addErrorMessage: (msg) => {
+        if (isPipeMode()) {
+            console.error(`[ERROR] ${msg}`);
+        } else {
+            uiEvents.addErrorMessage(msg);
+        }
+    },
+    sessionStart: (msg) => {
+        if (!isPipeMode()) {
+            uiEvents.sessionStart(msg);
+        }
+    },
+    sessionEnd: () => {
+        if (!isPipeMode()) {
+            uiEvents.sessionEnd();
+        }
+    },
+    on: (event, handler) => {
+        if (!isPipeMode()) {
+            uiEvents.on(event, handler);
+        }
+    },
+    off: (event, handler) => {
+        if (!isPipeMode()) {
+            uiEvents.off(event, handler);
+        }
+    }
+};
+
 
 /**
  * 서브미션 이름을 추론하는 헬퍼 함수
@@ -397,7 +439,7 @@ async function processOrchestratorResponses(params) {
             debugLog(`No processable output, reasoningOnlyResponses: ${reasoningOnlyResponses}`);
             if (reasoningOnlyResponses >= MAX_REASONING_ONLY_RESPONSES) {
                 debugLog(`Reached MAX_REASONING_ONLY_RESPONSES (${MAX_REASONING_ONLY_RESPONSES}), treating as stall`);
-                uiEvents.addSystemMessage('⚠ Orchestrator stalled (reasoning only) - treating as completion');
+                safeUiEvents.addSystemMessage('⚠ Orchestrator stalled (reasoning only) - treating as completion');
                 stallDetected = true;
                 break;
             }
@@ -816,14 +858,14 @@ export async function runSession(options) {
         if (!sessionInterrupted) {
             sessionInterrupted = true;
             abortCurrentRequest(); // API 요청 즉시 중단
-            uiEvents.addErrorMessage('Session interrupted by user');
+            safeUiEvents.addErrorMessage('Session interrupted by user');
             debugLog('[runSession] Session interrupted by user');
         }
     };
-    uiEvents.on('session:interrupt', handleInterrupt);
+    safeUiEvents.on('session:interrupt', handleInterrupt);
 
     // 세션 시작 알림
-    uiEvents.sessionStart('Running agent session...');
+    safeUiEvents.sessionStart('Running agent session...');
 
     // catch 블록에서도 접근 가능하도록 try 블록 밖에서 선언
     let currentSessionData = null;
@@ -895,6 +937,11 @@ export async function runSession(options) {
                 restoreTrimmedFileReadsFromSession({ trimmedFileReads: conversationState.trimmedFileReads });
                 debugLog(`[runSession] Restored ${conversationState.trimmedFileReads.length} trimmed file reads from previous session`);
             }
+            // Shell CWD 복원 (세션 이어가기 시 이전 작업 디렉토리 유지)
+            if (conversationState.shellCwd) {
+                await resetShellCwd(conversationState.shellCwd);
+                debugLog(`[runSession] Restored shell cwd to: ${conversationState.shellCwd}`);
+            }
             // logSuccess('✓ Conversation state restored');
         } else {
             // logSystem('ℹ Starting with fresh conversation state');
@@ -917,6 +964,11 @@ export async function runSession(options) {
                 updateCurrentTodos([]);
                 debugLog(`[runSession] Starting fresh session - no todos in memory`);
             }
+
+            // 새 세션 시작 시 쉘의 작업 디렉토리를 프로젝트 루트로 리셋
+            // (이전 미션에서 cd한 디렉토리가 남아있지 않도록)
+            await resetShellCwd();
+            debugLog(`[runSession] Shell cwd reset to: ${process.cwd()}`);
         }
 
         // Python 사용 가능 여부 확인
@@ -932,7 +984,23 @@ export async function runSession(options) {
             ...TODO_WRITE_FUNCTIONS,
             ...SKILL_FUNCTIONS,  // 스킬 호출 도구
             ...mcpToolFunctions,
-            "bash": async (args) => execShellScript(args.script)
+            "bash": async (args) => {
+                if (args.background) {
+                    // 백그라운드 실행
+                    const result = await runInBackground(args.script);
+                    return {
+                        stdout: `Background process started: ${result.id} (pid: ${result.pid})`,
+                        stderr: '',
+                        code: 0,
+                        background: true,
+                        processId: result.id,
+                        pid: result.pid
+                    };
+                } else {
+                    // 일반 실행
+                    return execShellScript(args.script);
+                }
+            }
         };
 
         // Python이 있는 경우에만 Python 관련 도구 추가
@@ -1059,6 +1127,9 @@ export async function runSession(options) {
                 currentSessionData.orchestratorConversation = getOrchestratorConversation();
                 currentSessionData.orchestratorRequestOptions = null;
 
+                // Shell CWD를 세션에 저장 (세션 이어가기 시 복원용)
+                currentSessionData.shellCwd = await getPersistentShell().getCwd();
+
                 // Todos를 세션에 저장
                 saveTodosToSession(currentSessionData);
 
@@ -1093,6 +1164,9 @@ export async function runSession(options) {
                 currentSessionData.orchestratorConversation = getOrchestratorConversation();
                 currentSessionData.orchestratorRequestOptions = null;
 
+                // Shell CWD를 세션에 저장
+                currentSessionData.shellCwd = await getPersistentShell().getCwd();
+
                 // 인터럽트 시에도 Todos와 TrimmedFileReads 저장 (다음 세션에서 복원 가능하도록)
                 saveTodosToSession(currentSessionData);
                 saveTrimmedFileReadsToSession(currentSessionData);
@@ -1132,7 +1206,7 @@ export async function runSession(options) {
                 const message = stallDetected
                     ? '⚠ Session ended due to orchestrator stall'
                     : '✅ No function calls detected - mission complete';
-                uiEvents.addSystemMessage(message);
+                safeUiEvents.addSystemMessage(message);
                 debugLog(`[ITERATION ${iteration_count}] ${message}`);
 
                 // completion_judge 없이 종료되는 경우에도 Todos와 TrimmedFileReads 저장
@@ -1142,6 +1216,7 @@ export async function runSession(options) {
                 currentSessionData.toolUsageHistory = toolUsageHistory;
                 currentSessionData.orchestratorConversation = getOrchestratorConversation();
                 currentSessionData.orchestratorRequestOptions = null;
+                currentSessionData.shellCwd = await getPersistentShell().getCwd();
                 saveTodosToSession(currentSessionData);
                 saveTrimmedFileReadsToSession(currentSessionData);
                 debugLog(`[ITERATION ${iteration_count}] Saving session on no-function-call exit`);
@@ -1188,6 +1263,9 @@ export async function runSession(options) {
         currentSessionData.orchestratorConversation = result.orchestratorConversation;
         currentSessionData.orchestratorRequestOptions = null; // 필요시 orchestrator에서 가져올 수 있음
 
+        // Shell CWD를 세션에 저장
+        currentSessionData.shellCwd = await getPersistentShell().getCwd();
+
         // 루프 정상 종료 시에도 Todos와 TrimmedFileReads 저장
         saveTodosToSession(currentSessionData);
         saveTrimmedFileReadsToSession(currentSessionData);
@@ -1252,12 +1330,13 @@ export async function runSession(options) {
             consolidatedErrorMessage = `[Session] ${userFriendlyMessage}`;
         }
 
-        uiEvents.addErrorMessage(consolidatedErrorMessage);
+        safeUiEvents.addErrorMessage(consolidatedErrorMessage);
 
         // 에러 발생 시에도 Todos와 TrimmedFileReads 저장 시도
         if (currentSessionData) {
             currentSessionData.completed_at = new Date().toISOString();
             currentSessionData.mission_solved = false;
+            currentSessionData.shellCwd = await getPersistentShell().getCwd();
             saveTodosToSession(currentSessionData);
             saveTrimmedFileReadsToSession(currentSessionData);
             debugLog(`[runSession] Saving session on error`);
@@ -1270,9 +1349,9 @@ export async function runSession(options) {
         return null;
     } finally {
         // 세션 중단 이벤트 리스너 제거
-        uiEvents.off('session:interrupt', handleInterrupt);
+        safeUiEvents.off('session:interrupt', handleInterrupt);
 
         // 세션 종료 알림
-        uiEvents.sessionEnd();
+        safeUiEvents.sessionEnd();
     }
 }

package/src/system/session_memory.js

@@ -152,7 +152,8 @@ export function getLastConversationState(sessions) {
         orchestratorConversation: lastSession.orchestratorConversation || [],
         orchestratorRequestOptions: lastSession.orchestratorRequestOptions || null,
         currentTodos: lastSession.currentTodos || [],
-        trimmedFileReads: lastSession.trimmedFileReads || []
+        trimmedFileReads: lastSession.trimmedFileReads || [],
+        shellCwd: lastSession.shellCwd || null
     };
 }
 

package/src/util/exit_handler.js

@@ -1,4 +1,6 @@
 import { uiEvents } from '../system/ui_events.js';
+import { closePersistentShell } from '../system/code_executer.js';
+import { killAllBackgroundProcesses } from '../system/background_process.js';
 
 /**
  * Unified exit handler for the application
@@ -17,6 +19,12 @@ export async function performExit(options = {}) {
         uiEvents.addSystemMessage(`Goodbye! Session ID: ${sessionID}`);
     }
 
+    // 백그라운드 프로세스 정리
+    await killAllBackgroundProcesses();
+
+    // PersistentShell 정리
+    await closePersistentShell();
+
     // MCP Integration 정리
     if (mcpIntegration) {
         await mcpIntegration.cleanup();