heyio 4.3.2 → 4.3.4

package/dist/daemon/cli.js

@@ -80,7 +80,7 @@ var init_constants = __esm({
   "packages/shared/dist/constants.js"() {
     "use strict";
     APP_NAME = "io";
-    APP_VERSION = "4.3.2";
+    APP_VERSION = "4.3.4";
     API_PORT = 7777;
     API_HOST = "0.0.0.0";
     DEFAULT_MODEL = "gpt-4.1-mini";
@@ -68368,6 +68368,7 @@ async function refreshModelPricing(logger2) {
   }
   const db = await getDatabase();
   const now = nowIso();
+  await db.execute("DELETE FROM model_pricing WHERE id LIKE '%/%'");
   for (const model of modelMap.values()) {
     if (model.tokenInputMultiplier == null) {
       continue;
@@ -68473,7 +68474,7 @@ async function upsertModel(db, model) {
 }
 function rowToModelPricing(row) {
   return {
-    id: asString(row.id),
+    id: stripVendorPrefix(asString(row.id)),
     displayName: asString(row.display_name),
     premiumMultiplier: asNullableNumber(row.premium_multiplier),
     tokenInputMultiplier: asNullableNumber(row.token_input_multiplier),
@@ -69532,18 +69533,21 @@ var init_manager2 = __esm({
 });
 
 // packages/daemon/src/squad/model-selector.ts
-import { CopilotClient as CopilotClient2, approveAll as approveAll2 } from "@github/copilot-sdk";
+function classifyTaskComplexity(taskDescription) {
+  const tiers = ["ultra", "premium", "standard", "fast", "trivial"];
+  for (const tier of tiers) {
+    if (TIER_KEYWORDS[tier].some((pattern) => pattern.test(taskDescription))) {
+      return tier;
+    }
+  }
+  return "standard";
+}
 async function selectModelForTask(taskDescription) {
   const classifierModel = await getCheapestAvailableModel();
   if (!classifierModel) {
     throw new Error("No models available in pricing database");
   }
-  let tier;
-  try {
-    tier = await classifyTaskComplexity(taskDescription, classifierModel.id);
-  } catch {
-    return classifierModel.id;
-  }
+  const tier = classifyTaskComplexity(taskDescription);
   const selectedModel = await getCheapestInTier(tier);
   if (selectedModel) {
     return selectedModel.id;
@@ -69557,46 +69561,18 @@ async function selectModelForTask(taskDescription) {
   }
   return classifierModel.id;
 }
-async function classifyTaskComplexity(taskDescription, modelId) {
-  let client2 = null;
-  try {
-    client2 = new CopilotClient2();
-    await client2.start();
-    const session = await client2.createSession({
-      model: modelId,
-      onPermissionRequest: approveAll2,
-      systemMessage: { content: CLASSIFICATION_PROMPT }
-    });
-    try {
-      const response = await session.sendAndWait({ prompt: `Task: ${taskDescription}` }, 15e3);
-      const raw = (response.text ?? "").trim().toLowerCase();
-      const tier = VALID_TIERS.find((t) => raw.includes(t));
-      return tier ?? "standard";
-    } finally {
-      await session.disconnect().catch(() => void 0);
-    }
-  } finally {
-    if (client2) {
-      await client2.stop().catch(() => void 0);
-    }
-  }
-}
-var VALID_TIERS, CLASSIFICATION_PROMPT;
+var TIER_KEYWORDS;
 var init_model_selector = __esm({
   "packages/daemon/src/squad/model-selector.ts"() {
     "use strict";
     init_registry();
-    VALID_TIERS = ["trivial", "fast", "standard", "premium", "ultra"];
-    CLASSIFICATION_PROMPT = `You are a task complexity classifier. Given a task description, classify its complexity into exactly one tier.
-
-Tiers (from simplest to most complex):
-- trivial: Typos, renames, comment changes, config tweaks, formatting
-- fast: Simple bug fixes, small features, documentation updates, single-file changes
-- standard: Feature implementation, multi-file changes, moderate refactoring
-- premium: Architecture changes, complex refactoring, security work, performance optimization
-- ultra: System-wide redesigns, critical infrastructure, cross-cutting concerns
-
-Reply with ONLY the tier name (one word, lowercase). Nothing else.`;
+    TIER_KEYWORDS = {
+      ultra: [/system[- ]wide/i, /redesign/i, /infrastructure/i, /cross[- ]cutting/i, /migration/i],
+      premium: [/architect/i, /security/i, /performance/i, /complex refactor/i, /optimization/i],
+      standard: [/implement/i, /feature/i, /multi[- ]file/i, /refactor/i, /integration/i],
+      fast: [/fix/i, /bug/i, /update/i, /documentation/i, /single[- ]file/i, /small/i],
+      trivial: [/typo/i, /rename/i, /comment/i, /config/i, /format/i]
+    };
   }
 });
 
@@ -69637,8 +69613,8 @@ import { mkdir as mkdir9, readFile as readFile7, readdir as readdir5, stat as st
 import { dirname as dirname8, extname as extname3, isAbsolute, join as join11, relative as relative3, resolve as resolve4 } from "node:path";
 import { promisify as promisify3 } from "node:util";
 import {
-  CopilotClient as CopilotClient3,
-  approveAll as approveAll3,
+  CopilotClient as CopilotClient2,
+  approveAll as approveAll2,
   defineTool
 } from "@github/copilot-sdk";
 function createEmptyUsage() {
@@ -69897,15 +69873,15 @@ async function executeAgentTask(member, task, worktreePath, options2) {
   const mcpServerNote = options2?.mcpServers?.length ? `Available MCP server labels: ${options2.mcpServers.join(", ")}.` : "No additional MCP servers were configured for this run.";
   let client2 = null;
   try {
-    client2 = new CopilotClient3({ workingDirectory: worktreePath });
+    client2 = new CopilotClient2({ workingDirectory: worktreePath });
     await client2.start();
     const model = member.model ? stripVendorPrefix(member.model) : await selectModelForTask(task.description);
-    const session = await client2.createSession({
+    const sessionPromise = client2.createSession({
       model,
       workingDirectory: worktreePath,
       tools,
       availableTools: ["custom:*"],
-      onPermissionRequest: approveAll3,
+      onPermissionRequest: approveAll2,
       systemMessage: {
         content: `${member.systemPrompt}
 
@@ -69915,6 +69891,15 @@ ${historyContext}
 ${mcpServerNote}${options2?.instancePromptSuffix ?? ""}`
       }
     });
+    const session = await Promise.race([
+      sessionPromise,
+      new Promise(
+        (_, reject) => setTimeout(
+          () => reject(new Error(`Session creation timed out after ${SESSION_CREATE_TIMEOUT_MS}ms`)),
+          SESSION_CREATE_TIMEOUT_MS
+        )
+      )
+    ]);
     session.on("assistant.usage", (event) => {
       usageEvents.push(event.data);
       mergeUsage(usage, event.data);
@@ -69953,7 +69938,7 @@ Work only inside the current worktree. Summarize the concrete changes you made,
     }
   }
 }
-var execAsync3, MAX_FILE_SIZE, MAX_LIST_RESULTS, MAX_SEARCH_RESULTS;
+var execAsync3, MAX_FILE_SIZE, MAX_LIST_RESULTS, MAX_SEARCH_RESULTS, SESSION_CREATE_TIMEOUT_MS;
 var init_agent = __esm({
   "packages/daemon/src/execution/agent.ts"() {
     "use strict";
@@ -69965,6 +69950,7 @@ var init_agent = __esm({
     MAX_FILE_SIZE = 2e5;
     MAX_LIST_RESULTS = 200;
     MAX_SEARCH_RESULTS = 100;
+    SESSION_CREATE_TIMEOUT_MS = 3e4;
   }
 });
 
@@ -70056,7 +70042,7 @@ import { exec as exec4 } from "node:child_process";
 import { access, readFile as readFile8 } from "node:fs/promises";
 import { join as join12 } from "node:path";
 import { promisify as promisify4 } from "node:util";
-import { CopilotClient as CopilotClient4, approveAll as approveAll4 } from "@github/copilot-sdk";
+import { CopilotClient as CopilotClient3, approveAll as approveAll3 } from "@github/copilot-sdk";
 async function fileExists(path) {
   try {
     await access(path);
@@ -70150,13 +70136,13 @@ Return strict JSON in this shape:
 }`;
   let client2 = null;
   try {
-    client2 = new CopilotClient4({ workingDirectory: repoPath });
+    client2 = new CopilotClient3({ workingDirectory: repoPath });
     await client2.start();
     const model = await selectModelForTask(`Create implementation plan: ${objective.description}`);
     const session = await client2.createSession({
       model,
       workingDirectory: repoPath,
-      onPermissionRequest: approveAll4,
+      onPermissionRequest: approveAll3,
       systemMessage: {
         content: `${TEAM_LEAD_PROMPT}
 
@@ -70276,7 +70262,7 @@ var init_pr = __esm({
 // packages/daemon/src/execution/qa.ts
 import { exec as exec6 } from "node:child_process";
 import { promisify as promisify6 } from "node:util";
-import { CopilotClient as CopilotClient5, approveAll as approveAll5 } from "@github/copilot-sdk";
+import { CopilotClient as CopilotClient4, approveAll as approveAll4 } from "@github/copilot-sdk";
 function extractJsonObject2(content) {
   const fenced = content.match(/```(?:json)?\s*([\s\S]*?)```/i);
   if (fenced?.[1]) {
@@ -70314,13 +70300,13 @@ Return strict JSON:
 }`;
   let client2 = null;
   try {
-    client2 = new CopilotClient5({ workingDirectory: worktreePath });
+    client2 = new CopilotClient4({ workingDirectory: worktreePath });
     await client2.start();
     const model = qaMember.model ? stripVendorPrefix(qaMember.model) : await selectModelForTask(`QA review: ${objective.description}`);
     const session = await client2.createSession({
       model,
       workingDirectory: worktreePath,
-      onPermissionRequest: approveAll5,
+      onPermissionRequest: approveAll4,
       systemMessage: {
         content: QA_PROMPT
       }
@@ -70404,7 +70390,7 @@ var init_qa = __esm({
 });
 
 // packages/daemon/src/execution/review.ts
-import { CopilotClient as CopilotClient6, approveAll as approveAll6 } from "@github/copilot-sdk";
+import { CopilotClient as CopilotClient5, approveAll as approveAll5 } from "@github/copilot-sdk";
 function extractJsonObject3(content) {
   const fenced = content.match(/```(?:json)?\s*([\s\S]*?)```/i);
   if (fenced?.[1]) {
@@ -70449,12 +70435,12 @@ Return strict JSON:
 }`;
   let client2 = null;
   try {
-    client2 = new CopilotClient6();
+    client2 = new CopilotClient5();
     await client2.start();
     const model = teamLead.model ? stripVendorPrefix(teamLead.model) : await selectModelForTask(`Code review: ${objective.description}`);
     const session = await client2.createSession({
       model,
-      onPermissionRequest: approveAll6,
+      onPermissionRequest: approveAll5,
       systemMessage: {
         content: `${TEAM_LEAD_PROMPT}
 
@@ -70616,60 +70602,49 @@ async function executePendingTasks(objective, members, worktreePath, mcpServers,
   if (pendingTasks.length === 0) {
     return getTasksForObjective(objective.id);
   }
-  const results = await Promise.allSettled(
-    pendingTasks.map(async (task) => {
-      const member = members.find((candidate) => candidate.id === task.assigneeId);
-      if (!member) {
-        const failedTask = await markTaskFailed(
-          task.id,
-          "No squad member matched the task assignee."
-        );
-        eventBus.emit(EVENT_NAMES.TASK_FAILED, {
-          task: failedTask,
-          agentName: "unassigned",
-          reason: failedTask.result ?? "No assignee"
-        });
-        throw new Error(`Task ${task.id} has no matching assignee`);
-      }
-      const inProgressTask = await updateTaskStatus(
+  for (const task of pendingTasks) {
+    const member = members.find((candidate) => candidate.id === task.assigneeId);
+    if (!member) {
+      const failedTask = await markTaskFailed(
         task.id,
-        "in_progress",
-        task.result ?? void 0
+        "No squad member matched the task assignee."
       );
-      const startedTask = inProgressTask ?? task;
-      eventBus.emit(EVENT_NAMES.TASK_STARTED, { task: startedTask, agentName: member.name });
-      eventBus.emit(EVENT_NAMES.AGENT_EXECUTING, {
-        squadId: objective.squadId,
-        agentId: member.id,
-        taskId: task.id
-      });
-      const execution = await executeAgentTask(member, task, worktreePath, {
-        mcpServers,
-        instancePromptSuffix
+      eventBus.emit(EVENT_NAMES.TASK_FAILED, {
+        task: failedTask,
+        agentName: "unassigned",
+        reason: failedTask.result ?? "No assignee"
       });
-      if (!execution.success) {
-        const failedTask = await markTaskFailed(task.id, execution.result);
-        eventBus.emit(EVENT_NAMES.TASK_FAILED, {
-          task: failedTask,
-          agentName: member.name,
-          reason: execution.result
-        });
-        throw new Error(execution.result);
-      }
-      const completedTask = await markTaskComplete(task.id, execution.result);
-      await extractLearnings(member.id, member.squadId, execution.result);
-      eventBus.emit(EVENT_NAMES.TASK_COMPLETED, { task: completedTask, agentName: member.name });
-      eventBus.emit(EVENT_NAMES.AGENT_COMPLETED, {
-        squadId: objective.squadId,
-        agentId: member.id,
-        taskId: task.id
+      throw new Error(`Task ${task.id} has no matching assignee`);
+    }
+    const inProgressTask = await updateTaskStatus(task.id, "in_progress", task.result ?? void 0);
+    const startedTask = inProgressTask ?? task;
+    eventBus.emit(EVENT_NAMES.TASK_STARTED, { task: startedTask, agentName: member.name });
+    eventBus.emit(EVENT_NAMES.AGENT_EXECUTING, {
+      squadId: objective.squadId,
+      agentId: member.id,
+      taskId: task.id
+    });
+    const execution = await executeAgentTask(member, task, worktreePath, {
+      mcpServers,
+      instancePromptSuffix
+    });
+    if (!execution.success) {
+      const failedTask = await markTaskFailed(task.id, execution.result);
+      eventBus.emit(EVENT_NAMES.TASK_FAILED, {
+        task: failedTask,
+        agentName: member.name,
+        reason: execution.result
       });
-      return completedTask;
-    })
-  );
-  const failed = results.find((result) => result.status === "rejected");
-  if (failed && failed.status === "rejected") {
-    throw failed.reason instanceof Error ? failed.reason : new Error(String(failed.reason));
+      throw new Error(execution.result);
+    }
+    const completedTask = await markTaskComplete(task.id, execution.result);
+    await extractLearnings(member.id, member.squadId, execution.result);
+    eventBus.emit(EVENT_NAMES.TASK_COMPLETED, { task: completedTask, agentName: member.name });
+    eventBus.emit(EVENT_NAMES.AGENT_COMPLETED, {
+      squadId: objective.squadId,
+      agentId: member.id,
+      taskId: task.id
+    });
   }
   return getTasksForObjective(objective.id);
 }