@wolfx/oh-my-openagent 3.17.13 → 3.17.14

package/dist/features/background-agent/manager.d.ts

@@ -84,6 +84,7 @@ export declare class BackgroundManager {
     private startTask;
     getTask(id: string): BackgroundTask | undefined;
     getTasksByParentSession(sessionID: string): BackgroundTask[];
+    private updateBackgroundTaskMarker;
     getAllDescendantTasks(sessionID: string): BackgroundTask[];
     findBySession(sessionID: string): BackgroundTask | undefined;
     private resolveTaskAttemptBySession;

package/dist/features/background-agent/process-cleanup.d.ts

@@ -1,3 +1,7 @@
+/** @internal test-only */
+export declare function __disableScheduledForcedExitForTesting(): void;
+/** @internal test-only */
+export declare function __enableScheduledForcedExitForTesting(): void;
 interface CleanupTarget {
     shutdown(): void | Promise<void>;
 }

package/dist/features/run-continuation-state/types.d.ts

@@ -1,4 +1,4 @@
-export type ContinuationMarkerSource = "todo" | "stop";
+export type ContinuationMarkerSource = "todo" | "stop" | "background-task";
 export type ContinuationMarkerState = "idle" | "active" | "stopped";
 export interface ContinuationMarkerSourceEntry {
     state: ContinuationMarkerState;

package/dist/features/skill-mcp-manager/types.d.ts

@@ -20,6 +20,7 @@ export interface SkillMcpClientInfo {
     skillName: string;
     sessionID: string;
     scope?: SkillScope | "local";
+    directory?: string;
 }
 export interface SkillMcpServerContext {
     config: ClaudeCodeMcpServer;

package/dist/index.js

@@ -19612,6 +19612,7 @@ function toLogLabel(cacheLabel) {
 }
 function createJsonFileCacheStore(options) {
   let memoryValue;
+  let writtenInCurrentProcess = false;
   function getCacheFilePath() {
     return join10(options.getCacheDir(), options.filename);
   }
@@ -19646,6 +19647,12 @@ function createJsonFileCacheStore(options) {
     }
   }
   function has() {
+    if (memoryValue !== undefined && memoryValue !== null) {
+      return true;
+    }
+    if (writtenInCurrentProcess) {
+      return true;
+    }
     return existsSync10(getCacheFilePath());
   }
   function write(value) {
@@ -19654,6 +19661,7 @@ function createJsonFileCacheStore(options) {
     try {
       writeFileSync2(cacheFile, options.serialize?.(value) ?? JSON.stringify(value, null, 2));
       memoryValue = value;
+      writtenInCurrentProcess = true;
       log(`[${options.logPrefix}] ${options.cacheLabel} written`, options.describe(value));
     } catch (error) {
       log(`[${options.logPrefix}] Error writing ${toLogLabel(options.cacheLabel)}`, {
@@ -19663,6 +19671,7 @@ function createJsonFileCacheStore(options) {
   }
   function resetMemory() {
     memoryValue = undefined;
+    writtenInCurrentProcess = false;
   }
   return {
     read,
@@ -19673,6 +19682,7 @@ function createJsonFileCacheStore(options) {
 }
 
 // src/shared/connected-providers-cache.ts
+var providerModelsCacheWrittenInCurrentProcess = false;
 var CONNECTED_PROVIDERS_CACHE_FILE = "connected-providers.json";
 var PROVIDER_MODELS_CACHE_FILE = "provider-models.json";
 function isRecord(value) {
@@ -19712,6 +19722,9 @@ function createConnectedProvidersCacheStore(getCacheDir2 = getOmoOpenCodeCacheDi
     return providerModelsCacheStore.read();
   }
   function hasProviderModelsCache() {
+    if (providerModelsCacheWrittenInCurrentProcess) {
+      return true;
+    }
     return providerModelsCacheStore.has();
   }
   function writeProviderModelsCache(data) {
@@ -19719,6 +19732,7 @@ function createConnectedProvidersCacheStore(getCacheDir2 = getOmoOpenCodeCacheDi
       ...data,
       updatedAt: new Date().toISOString()
     });
+    providerModelsCacheWrittenInCurrentProcess = true;
   }
   async function updateConnectedProvidersCache(client) {
     if (!client?.provider?.list) {
@@ -19767,6 +19781,7 @@ function createConnectedProvidersCacheStore(getCacheDir2 = getOmoOpenCodeCacheDi
   function _resetMemCacheForTesting() {
     connectedProvidersCacheStore.resetMemory();
     providerModelsCacheStore.resetMemory();
+    providerModelsCacheWrittenInCurrentProcess = false;
   }
   return {
     readConnectedProvidersCache,
@@ -66659,7 +66674,7 @@ async function injectContinuation(args) {
     log(`[${HOOK_NAME}] Skipped injection: continuation stopped for session`, { sessionID });
     return;
   }
-  const hasRunningBgTasks = backgroundManager ? backgroundManager.getTasksByParentSession(sessionID).some((task) => task.status === "running") : false;
+  const hasRunningBgTasks = backgroundManager ? backgroundManager.getTasksByParentSession(sessionID).some((task) => task.status === "running" || task.status === "pending") : false;
   if (hasRunningBgTasks) {
     log(`[${HOOK_NAME}] Skipped injection: background tasks running`, { sessionID });
     return;
@@ -66865,7 +66880,7 @@ async function handleSessionIdle(args) {
     }
     state2.abortDetectedAt = undefined;
   }
-  const hasRunningBgTasks = backgroundManager ? backgroundManager.getTasksByParentSession(sessionID).some((task) => task.status === "running") : false;
+  const hasRunningBgTasks = backgroundManager ? backgroundManager.getTasksByParentSession(sessionID).some((task) => task.status === "running" || task.status === "pending") : false;
   if (hasRunningBgTasks) {
     log(`[${HOOK_NAME}] Skipped: background tasks running`, { sessionID });
     return;
@@ -99494,6 +99509,7 @@ async function resolveRecentPromptContextForSession(ctx, sessionID) {
 }
 
 // src/hooks/atlas/boulder-continuation-injector.ts
+var ACTIVE_BACKGROUND_TASK_STATUSES = new Set(["pending", "running"]);
 async function injectBoulderContinuation(input) {
   const {
     ctx,
@@ -99508,7 +99524,7 @@ async function injectBoulderContinuation(input) {
     backgroundManager,
     sessionState
   } = input;
-  const hasRunningBgTasks = backgroundManager ? backgroundManager.getTasksByParentSession(sessionID).some((t) => t.status === "running") : false;
+  const hasRunningBgTasks = backgroundManager ? backgroundManager.getTasksByParentSession(sessionID).some((t) => ACTIVE_BACKGROUND_TASK_STATUSES.has(t.status)) : false;
   if (hasRunningBgTasks) {
     log(`[${HOOK_NAME7}] Skipped injection: background tasks running`, { sessionID });
     return "skipped_background_tasks";
@@ -122078,7 +122094,8 @@ function createSkillMcpTool(options) {
         serverName: args.mcp_name,
         skillName: found.skill.name,
         sessionID,
-        scope: found.skill.scope
+        scope: found.skill.scope,
+        directory: toolContext.directory
       };
       const context = {
         config: found.config,
@@ -130091,11 +130108,17 @@ async function tryFallbackRetry(args) {
 }
 
 // src/features/background-agent/process-cleanup.ts
-function scheduleForcedExit(cleanupResult, exitCode) {
+var _scheduleForcedExitEnabled = true;
+function scheduleForcedExit(cleanupResult, exitCode, exitAfterCleanup = false) {
+  if (!_scheduleForcedExitEnabled)
+    return;
   process.exitCode = exitCode;
   const exitTimeout = setTimeout(() => process.exit(), 6000);
   Promise.resolve(cleanupResult).finally(() => {
     clearTimeout(exitTimeout);
+    if (exitAfterCleanup) {
+      process.exit(exitCode);
+    }
   });
 }
 function registerProcessSignal(signal, handler, exitAfter) {
@@ -130110,8 +130133,9 @@ function registerProcessSignal(signal, handler, exitAfter) {
 }
 function registerErrorEvent(signal, handler) {
   const listener = (error92) => {
+    process.off(signal, listener);
     log(`[background-agent] ${signal} received during shutdown cleanup:`, error92);
-    scheduleForcedExit(handler(error92), 1);
+    scheduleForcedExit(handler(error92), 1, true);
   };
   process.on(signal, listener);
   return listener;
@@ -130950,6 +130974,7 @@ class BackgroundManager {
       }
       spawnReservation.commit();
       this.markPreStartDescendantReservation(task);
+      this.updateBackgroundTaskMarker(input.parentSessionId);
       this.processKey(key);
       return { ...task };
     } catch (error92) {
@@ -130997,6 +131022,7 @@ class BackgroundManager {
           if (item.task.sessionId) {
             await this.abortSessionWithLogging(item.task.sessionId, "startTask error cleanup");
           }
+          this.updateBackgroundTaskMarker(item.task.parentSessionId);
           this.markForNotification(item.task);
           this.enqueueNotificationForParent(item.task.parentSessionId, () => this.notifyParentSession(item.task)).catch((err) => {
             log("[background-agent] Failed to notify on startTask error:", err);
@@ -131251,6 +131277,15 @@ The fallback retry session is now created and can be inspected directly.
     }
     return tasks;
   }
+  updateBackgroundTaskMarker(parentSessionID) {
+    const tasks = this.getTasksByParentSession(parentSessionID);
+    const activeTasks = tasks.filter((t) => t.status === "running" || t.status === "pending");
+    if (activeTasks.length > 0) {
+      setContinuationMarkerSource(this.directory, parentSessionID, "background-task", "active", `${activeTasks.length} background task(s) active`);
+    } else {
+      setContinuationMarkerSource(this.directory, parentSessionID, "background-task", "idle");
+    }
+  }
   getAllDescendantTasks(sessionID) {
     const result = [];
     const directChildren = this.getTasksByParentSession(sessionID);
@@ -131831,6 +131866,9 @@ The fallback retry session is now created and can be inspected directly.
     if (task.sessionId) {
       SessionCategoryRegistry.remove(task.sessionId);
     }
+    if (task.parentSessionId) {
+      this.updateBackgroundTaskMarker(task.parentSessionId);
+    }
     this.markForNotification(task);
     this.enqueueNotificationForParent(task.parentSessionId, () => this.notifyParentSession(task)).catch((err) => {
       log("[background-agent] Error in notifyParentSession for errored task:", { taskId: task.id, error: err });
@@ -132065,6 +132103,9 @@ ${originalText}`;
       SessionCategoryRegistry.remove(task.sessionId);
     }
     removeTaskToastTracking(task.id);
+    if (task.parentSessionId) {
+      this.updateBackgroundTaskMarker(task.parentSessionId);
+    }
     if (options?.skipNotification) {
       this.cleanupPendingByParent(task);
       this.scheduleTaskRemoval(task.id);
@@ -132144,6 +132185,9 @@ ${originalText}`;
       await this.abortSessionWithLogging(task.sessionId, `task completion (${source})`);
       SessionCategoryRegistry.remove(task.sessionId);
     }
+    if (task.parentSessionId) {
+      this.updateBackgroundTaskMarker(task.parentSessionId);
+    }
     try {
       await this.enqueueNotificationForParent(task.parentSessionId, () => this.notifyParentSession(task));
       log(`[background-agent] Task completed via ${source}:`, task.id);
@@ -132327,6 +132371,9 @@ ${originalText}`;
           }
         }
         this.cleanupPendingByParent(task);
+        if (task.parentSessionId) {
+          this.updateBackgroundTaskMarker(task.parentSessionId);
+        }
         this.markForNotification(task);
         this.enqueueNotificationForParent(task.parentSessionId, () => this.notifyParentSession(task)).catch((err) => {
           log("[background-agent] Error in notifyParentSession for stale-pruned task:", { taskId: task.id, error: err });
@@ -132381,6 +132428,9 @@ ${originalText}`;
     if (task.sessionId) {
       SessionCategoryRegistry.remove(task.sessionId);
     }
+    if (task.parentSessionId) {
+      this.updateBackgroundTaskMarker(task.parentSessionId);
+    }
     this.markForNotification(task);
     this.enqueueNotificationForParent(task.parentSessionId, () => this.notifyParentSession(task)).catch((err) => {
       log("[background-agent] Error in notifyParentSession for crashed task:", { taskId: task.id, error: err });
@@ -137489,7 +137539,8 @@ async function createStdioClient(params) {
     command,
     args,
     env: mergedEnv,
-    stderr: "ignore"
+    stderr: "ignore",
+    ...info.directory ? { cwd: info.directory } : {}
   });
   const client2 = stdioClientDependencies.createClient({ name: `skill-mcp-${info.skillName}-${info.serverName}`, version: "1.0.0" }, { capabilities: {} });
   try {
@@ -153320,7 +153371,6 @@ init_agent_display_names();
 init_agent_display_names();
 
 // src/plugin/ultrawork-db-model-override.ts
-import { Database } from "bun:sqlite";
 import { join as join101 } from "path";
 import { existsSync as existsSync91 } from "fs";
 function getDbPath() {
@@ -153399,7 +153449,13 @@ function retryViaMicrotask(db, messageId, targetModel, variant, attempt) {
   });
 }
 function scheduleDeferredModelOverride(messageId, targetModel, variant) {
-  queueMicrotask(() => {
+  queueMicrotask(async () => {
+    const sqliteModule = await import("bun:sqlite").catch(() => null);
+    const Database = sqliteModule?.Database;
+    if (typeof Database !== "function") {
+      log("[ultrawork-db-override] bun:sqlite unavailable, skipping deferred override", { messageId });
+      return;
+    }
     const dbPath = getDbPath();
     if (!existsSync91(dbPath)) {
       log("[ultrawork-db-override] DB not found, skipping deferred override");

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@wolfx/oh-my-openagent",
-  "version": "3.17.13",
+  "version": "3.17.14",
   "description": "A fork of oh-my-openagent",
   "main": "./dist/index.js",
   "types": "dist/index.d.ts",
@@ -39,7 +39,7 @@
     "ai",
     "llm"
   ],
-  "author": "wolfx",
+  "author": "YeonGyu-Kim",
   "license": "SUL-1.0",
   "repository": {
     "type": "git",