zenox 1.4.2 → 1.4.3

package/dist/background/index.d.ts

@@ -5,4 +5,4 @@
  */
 export { BackgroundManager } from "./manager";
 export { createBackgroundTools, type BackgroundTools } from "./tools";
-export type { BackgroundTask, TaskStatus, LaunchInput, CompletionNotification, } from "./types";
+export type { BackgroundTask, TaskStatus, LaunchInput, CompletionNotification, ParentModel, } from "./types";

package/dist/background/types.d.ts

@@ -3,6 +3,10 @@
  * Minimal interfaces for fire-and-forget parallel agent execution
  */
 export type TaskStatus = "running" | "completed" | "failed" | "cancelled";
+export interface ParentModel {
+    providerID: string;
+    modelID: string;
+}
 export interface BackgroundTask {
     id: string;
     sessionID: string;
@@ -14,6 +18,7 @@ export interface BackgroundTask {
     completedAt?: Date;
     error?: string;
     parentAgent?: string;
+    parentModel?: ParentModel;
 }
 export interface LaunchInput {
     agent: string;
@@ -21,6 +26,7 @@ export interface LaunchInput {
     description: string;
     parentSessionID: string;
     parentAgent?: string;
+    parentModel?: ParentModel;
 }
 export interface CompletionNotification {
     allComplete: boolean;
@@ -28,4 +34,5 @@ export interface CompletionNotification {
     completedTasks: BackgroundTask[];
     runningCount: number;
     parentAgent?: string;
+    parentModel?: ParentModel;
 }

package/dist/index.js

@@ -835,18 +835,26 @@ function getOrchestrationPrompt(agent) {
 }
 
 // src/orchestration/session-agent-tracker.ts
-var sessionAgentMap = new Map;
-function setSessionAgent(sessionID, agent) {
-  if (agent) {
-    sessionAgentMap.set(sessionID, agent);
-  }
+var sessionContextMap = new Map;
+function setSessionContext(sessionID, context) {
+  const existing = sessionContextMap.get(sessionID) ?? {};
+  sessionContextMap.set(sessionID, {
+    agent: context.agent ?? existing.agent,
+    model: context.model ?? existing.model
+  });
+}
+function getSessionContext(sessionID) {
+  return sessionContextMap.get(sessionID);
 }
 function getSessionAgent(sessionID) {
-  return sessionAgentMap.get(sessionID);
+  return sessionContextMap.get(sessionID)?.agent;
+}
+function getSessionModel(sessionID) {
+  return sessionContextMap.get(sessionID)?.model;
 }
 function clearSessionAgent(sessionID) {
   if (sessionID) {
-    sessionAgentMap.delete(sessionID);
+    sessionContextMap.delete(sessionID);
   }
 }
 function getOrchestrationAgentType(agent) {
@@ -4995,7 +5003,8 @@ class BackgroundManager {
       prompt: input.prompt,
       status: "running",
       startedAt: new Date,
-      parentAgent: input.parentAgent
+      parentAgent: input.parentAgent,
+      parentModel: input.parentModel
     };
     this.tasks.set(task.id, task);
     if (this.toastManager) {
@@ -5134,12 +5143,14 @@ ${runningTasks.length} task(s) still running. Continue working.
 </system-reminder>`;
     }
     const parentAgent = completedTasks[0]?.parentAgent;
+    const parentModel = completedTasks[0]?.parentModel;
     return {
       allComplete,
       message,
       completedTasks,
       runningCount: runningTasks.length,
-      parentAgent
+      parentAgent,
+      parentModel
     };
   }
   listActiveTasks() {
@@ -17490,12 +17501,14 @@ Use for independent research tasks that benefit from parallelism.`,
     },
     async execute(args, context) {
       try {
+        const parentModel = getSessionModel(context.sessionID);
         const task = await manager.launch(client, {
           agent: args.agent,
           description: args.description,
           prompt: args.prompt,
           parentSessionID: context.sessionID,
-          parentAgent: context.agent
+          parentAgent: context.agent,
+          parentModel
         });
         const activeTasks = manager.listActiveTasks();
         return `Background task launched successfully.
@@ -17930,12 +17943,15 @@ Current incomplete tasks:
 ${todoList}
 
 ${statusLine}`;
-      const sendPrompt = async (omitAgent = false) => {
+      const sessionContext = getSessionContext(sessionID);
+      const model = sessionContext?.model;
+      const sendPrompt = async (omitContext = false) => {
         await ctx.client.session.prompt({
           path: { id: sessionID },
           body: {
             noReply: false,
-            ...omitAgent || !agent ? {} : { agent },
+            ...omitContext || !agent ? {} : { agent },
+            ...omitContext || !model ? {} : { model },
             parts: [{ type: "text", text: prompt }]
           }
         });
@@ -17944,7 +17960,7 @@ ${statusLine}`;
         await sendPrompt();
       } catch (err) {
         const errorMsg = err instanceof Error ? err.message : String(err);
-        if (errorMsg.includes("agent") || errorMsg.includes("undefined")) {
+        if (errorMsg.includes("agent") || errorMsg.includes("model") || errorMsg.includes("undefined")) {
           try {
             await sendPrompt(true);
           } catch {
@@ -18349,7 +18365,7 @@ var ZenoxPlugin = async (ctx) => {
       ...codeIntelligenceTools
     },
     "chat.message": async (input, output) => {
-      setSessionAgent(input.sessionID, input.agent);
+      setSessionContext(input.sessionID, { agent: input.agent, model: input.model });
       const message = output.message;
       if (firstMessageVariantGate.shouldOverride(input.sessionID)) {
         const variant = resolveAgentVariant(pluginConfig, input.agent);
@@ -18402,19 +18418,20 @@ var ZenoxPlugin = async (ctx) => {
         if (notification) {
           const mainSessionID = backgroundManager.getMainSession();
           if (mainSessionID) {
-            const sendNotification = async (omitAgent = false) => {
+            const sendNotification = async (omitContext = false) => {
               try {
                 await ctx.client.session.prompt({
                   path: { id: mainSessionID },
                   body: {
                     noReply: !notification.allComplete,
-                    ...omitAgent ? {} : { agent: notification.parentAgent },
+                    ...omitContext ? {} : { agent: notification.parentAgent },
+                    ...omitContext ? {} : { model: notification.parentModel },
                     parts: [{ type: "text", text: notification.message }]
                   }
                 });
               } catch (err) {
                 const errorMsg = err instanceof Error ? err.message : String(err);
-                if (!omitAgent && (errorMsg.includes("agent.name") || errorMsg.includes("undefined"))) {
+                if (!omitContext && (errorMsg.includes("agent") || errorMsg.includes("model") || errorMsg.includes("undefined"))) {
                   return sendNotification(true);
                 }
               }

package/dist/orchestration/session-agent-tracker.d.ts

@@ -1,17 +1,39 @@
 /**
- * Session-Agent Tracker
+ * Session Context Tracker
  *
- * Tracks which agent is active for each session, allowing the system transform
- * hook to inject agent-specific prompts.
+ * Tracks agent and model context for each session, allowing:
+ * - System transform hook to inject agent-specific prompts
+ * - Todo enforcer and background tasks to preserve model context
  *
  * Flow:
- * 1. chat.message hook fires with { sessionID, agent }
- * 2. We store the mapping: sessionID -> agentName
+ * 1. chat.message hook fires with { sessionID, agent, model }
+ * 2. We store the mapping: sessionID -> { agent, model }
  * 3. experimental.chat.system.transform fires with { sessionID }
- * 4. We look up the agent and inject the appropriate prompt
+ * 4. We look up the context and inject the appropriate prompt
+ * 5. Todo enforcer / background tasks use model context when sending prompts
  */
+export interface SessionModel {
+    providerID: string;
+    modelID: string;
+}
+export interface SessionContext {
+    agent?: string;
+    model?: SessionModel;
+}
 /**
- * Record which agent is active for a session.
+ * Record context (agent + model) for a session.
+ * Called from chat.message hook.
+ */
+export declare function setSessionContext(sessionID: string, context: {
+    agent?: string;
+    model?: SessionModel;
+}): void;
+/**
+ * Get the full context for a session.
+ */
+export declare function getSessionContext(sessionID: string): SessionContext | undefined;
+/**
+ * Record which agent is active for a session (legacy helper).
  * Called from chat.message hook.
  */
 export declare function setSessionAgent(sessionID: string, agent: string | undefined): void;
@@ -20,6 +42,10 @@ export declare function setSessionAgent(sessionID: string, agent: string | undef
  * Called from experimental.chat.system.transform hook.
  */
 export declare function getSessionAgent(sessionID: string): string | undefined;
+/**
+ * Get the active model for a session.
+ */
+export declare function getSessionModel(sessionID: string): SessionModel | undefined;
 /**
  * Clear tracking for a session (on deletion).
  */

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "zenox",
-  "version": "1.4.2",
+  "version": "1.4.3",
   "description": "OpenCode plugin with specialized agents (explorer, librarian, oracle, ui-planner), background tasks for parallel execution, and smart orchestration",
   "author": "Ayush",
   "license": "MIT",