heyio 4.2.7 → 4.3.0

package/dist/daemon/cli.js

@@ -80,10 +80,10 @@ var init_constants = __esm({
   "packages/shared/dist/constants.js"() {
     "use strict";
     APP_NAME = "io";
-    APP_VERSION = "4.2.7";
+    APP_VERSION = "4.3.0";
     API_PORT = 7777;
     API_HOST = "0.0.0.0";
-    DEFAULT_MODEL = "gpt-4o";
+    DEFAULT_MODEL = "gpt-4.1-mini";
     SESSION_RESET_THRESHOLD = 50;
     SCHEDULER_INTERVAL_MS = 6e4;
     QA_MAX_REVISIONS = 3;
@@ -68341,12 +68341,15 @@ async function refreshModelPricing(logger2) {
   const db = await getDatabase();
   const now = nowIso();
   for (const model of modelMap.values()) {
+    if (model.tokenInputMultiplier == null) {
+      continue;
+    }
     const tier = computeTierFromMultiplier(model.premiumMultiplier ?? null);
     await upsertModel(db, {
       id: model.id,
       displayName: model.displayName,
       premiumMultiplier: model.premiumMultiplier ?? null,
-      tokenInputMultiplier: model.tokenInputMultiplier ?? null,
+      tokenInputMultiplier: model.tokenInputMultiplier,
       tokenOutputMultiplier: model.tokenOutputMultiplier ?? null,
       cachedInputMultiplier: model.cachedInputMultiplier ?? null,
       tier,
@@ -68357,6 +68360,18 @@ async function refreshModelPricing(logger2) {
   }
   return result;
 }
+async function getModelsForTier(tier) {
+  const db = await getDatabase();
+  const result = await db.execute({
+    sql: "SELECT * FROM model_pricing WHERE tier = ? AND available = 1 ORDER BY premium_multiplier ASC NULLS LAST",
+    args: [tier]
+  });
+  return result.rows.map(rowToModelPricing);
+}
+async function getCheapestInTier(tier) {
+  const models = await getModelsForTier(tier);
+  return models[0] ?? null;
+}
 async function getModelPricing(modelId) {
   const db = await getDatabase();
   const result = await db.execute({
@@ -68377,6 +68392,16 @@ async function getModelPricing(modelId) {
   }
   return null;
 }
+async function getCheapestAvailableModel() {
+  const tiers = ["trivial", "fast", "standard", "premium", "ultra"];
+  for (const tier of tiers) {
+    const model = await getCheapestInTier(tier);
+    if (model) {
+      return model;
+    }
+  }
+  return null;
+}
 function calculateTokenUnitCost(inputTokens, outputTokens, inputMultiplier, outputMultiplier) {
   if (inputMultiplier === null || outputMultiplier === null) {
     return 0;
@@ -68384,6 +68409,14 @@ function calculateTokenUnitCost(inputTokens, outputTokens, inputMultiplier, outp
   const tokenUnits = inputTokens * inputMultiplier + outputTokens * outputMultiplier;
   return tokenUnits * TOKEN_UNIT_PRICE;
 }
+function getNextTierUp(tier) {
+  const order = ["trivial", "fast", "standard", "premium", "ultra"];
+  const idx = order.indexOf(tier);
+  if (idx < 0 || idx >= order.length - 1) {
+    return null;
+  }
+  return order[idx + 1];
+}
 async function upsertModel(db, model) {
   await db.execute({
     sql: `INSERT INTO model_pricing (id, display_name, premium_multiplier, token_input_multiplier, token_output_multiplier, cached_input_multiplier, tier, available, updated_at)
@@ -69470,6 +69503,75 @@ var init_manager2 = __esm({
   }
 });
 
+// packages/daemon/src/squad/model-selector.ts
+import { CopilotClient as CopilotClient2, approveAll as approveAll2 } from "@github/copilot-sdk";
+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 selectedModel = await getCheapestInTier(tier);
+  if (selectedModel) {
+    return selectedModel.id;
+  }
+  const nextTier = getNextTierUp(tier);
+  if (nextTier) {
+    const escalatedModel = await getCheapestInTier(nextTier);
+    if (escalatedModel) {
+      return escalatedModel.id;
+    }
+  }
+  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 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.`;
+  }
+});
+
 // packages/daemon/src/execution/history.ts
 function summarizeTaskResult(taskResult) {
   const condensed = taskResult.replace(/\s+/g, " ").trim();
@@ -69507,8 +69609,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 CopilotClient2,
-  approveAll as approveAll2,
+  CopilotClient as CopilotClient3,
+  approveAll as approveAll3,
   defineTool
 } from "@github/copilot-sdk";
 function createEmptyUsage() {
@@ -69767,14 +69869,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 CopilotClient2({ workingDirectory: worktreePath });
+    client2 = new CopilotClient3({ workingDirectory: worktreePath });
     await client2.start();
+    const model = member.model ?? await selectModelForTask(task.description);
     const session = await client2.createSession({
-      model: member.model ?? DEFAULT_MODEL,
+      model,
       workingDirectory: worktreePath,
       tools,
       availableTools: ["custom:*"],
-      onPermissionRequest: approveAll2,
+      onPermissionRequest: approveAll3,
       systemMessage: {
         content: `${member.systemPrompt}
 
@@ -69826,8 +69929,8 @@ var execAsync3, MAX_FILE_SIZE, MAX_LIST_RESULTS, MAX_SEARCH_RESULTS;
 var init_agent = __esm({
   "packages/daemon/src/execution/agent.ts"() {
     "use strict";
-    init_dist();
     init_registry();
+    init_model_selector();
     init_store2();
     init_history();
     execAsync3 = promisify3(exec3);
@@ -69925,7 +70028,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 CopilotClient3, approveAll as approveAll3 } from "@github/copilot-sdk";
+import { CopilotClient as CopilotClient4, approveAll as approveAll4 } from "@github/copilot-sdk";
 async function fileExists(path) {
   try {
     await access(path);
@@ -70019,12 +70122,13 @@ Return strict JSON in this shape:
 }`;
   let client2 = null;
   try {
-    client2 = new CopilotClient3({ workingDirectory: repoPath });
+    client2 = new CopilotClient4({ workingDirectory: repoPath });
     await client2.start();
+    const model = await selectModelForTask(`Create implementation plan: ${objective.description}`);
     const session = await client2.createSession({
-      model: DEFAULT_MODEL,
+      model,
       workingDirectory: repoPath,
-      onPermissionRequest: approveAll3,
+      onPermissionRequest: approveAll4,
       systemMessage: {
         content: `${TEAM_LEAD_PROMPT}
 
@@ -70059,7 +70163,7 @@ var execAsync4, README_CANDIDATES, MAX_REPO_CONTEXT_LENGTH;
 var init_planning = __esm({
   "packages/daemon/src/execution/planning.ts"() {
     "use strict";
-    init_dist();
+    init_model_selector();
     init_roles();
     execAsync4 = promisify4(exec4);
     README_CANDIDATES = ["README.md", "readme.md"];
@@ -70144,7 +70248,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 CopilotClient4, approveAll as approveAll4 } from "@github/copilot-sdk";
+import { CopilotClient as CopilotClient5, approveAll as approveAll5 } from "@github/copilot-sdk";
 function extractJsonObject2(content) {
   const fenced = content.match(/```(?:json)?\s*([\s\S]*?)```/i);
   if (fenced?.[1]) {
@@ -70182,12 +70286,13 @@ Return strict JSON:
 }`;
   let client2 = null;
   try {
-    client2 = new CopilotClient4({ workingDirectory: worktreePath });
+    client2 = new CopilotClient5({ workingDirectory: worktreePath });
     await client2.start();
+    const model = qaMember.model ?? await selectModelForTask(`QA review: ${objective.description}`);
     const session = await client2.createSession({
-      model: qaMember.model ?? DEFAULT_MODEL,
+      model,
       workingDirectory: worktreePath,
-      onPermissionRequest: approveAll4,
+      onPermissionRequest: approveAll5,
       systemMessage: {
         content: QA_PROMPT
       }
@@ -70261,6 +70366,7 @@ var init_qa = __esm({
     "use strict";
     init_dist();
     init_event_bus();
+    init_model_selector();
     init_roles();
     init_store2();
     execAsync6 = promisify6(exec6);
@@ -70269,7 +70375,7 @@ var init_qa = __esm({
 });
 
 // packages/daemon/src/execution/review.ts
-import { CopilotClient as CopilotClient5, approveAll as approveAll5 } from "@github/copilot-sdk";
+import { CopilotClient as CopilotClient6, approveAll as approveAll6 } from "@github/copilot-sdk";
 function extractJsonObject3(content) {
   const fenced = content.match(/```(?:json)?\s*([\s\S]*?)```/i);
   if (fenced?.[1]) {
@@ -70314,11 +70420,12 @@ Return strict JSON:
 }`;
   let client2 = null;
   try {
-    client2 = new CopilotClient5();
+    client2 = new CopilotClient6();
     await client2.start();
+    const model = teamLead.model ?? await selectModelForTask(`Code review: ${objective.description}`);
     const session = await client2.createSession({
-      model: teamLead.model ?? DEFAULT_MODEL,
-      onPermissionRequest: approveAll5,
+      model,
+      onPermissionRequest: approveAll6,
       systemMessage: {
         content: `${TEAM_LEAD_PROMPT}
 
@@ -70352,7 +70459,7 @@ You are conducting a final coordination review before QA.`
 var init_review = __esm({
   "packages/daemon/src/execution/review.ts"() {
     "use strict";
-    init_dist();
+    init_model_selector();
     init_roles();
   }
 });
@@ -71725,13 +71832,13 @@ var init_orchestrator = __esm({
           });
           this.activeModel = model;
         } catch (error51) {
-          if (model !== DEFAULT_MODEL) {
+          if (model !== this.config.defaultModel) {
             this.activeSession = await createSession({
-              model: DEFAULT_MODEL,
+              model: this.config.defaultModel,
               systemPrompt,
               tools: createBoundOrchestratorTools(this.config)
             });
-            this.activeModel = DEFAULT_MODEL;
+            this.activeModel = this.config.defaultModel;
           } else {
             throw error51;
           }
@@ -85183,9 +85290,16 @@ var init_bot = __esm({
         if (this.started) {
           return;
         }
+        this.bot.catch((err) => {
+          this.logger.error({ err: err.error }, "Telegram bot error: %s", err.message);
+        });
         this.registerHandlers();
-        this.bot.start().catch((error51) => {
-          this.logger.error({ err: error51 }, "Telegram polling failed");
+        this.bot.start({
+          onStart: () => {
+            this.logger.info("Telegram bot connected and polling");
+          }
+        }).catch((error51) => {
+          this.logger.error({ err: error51 }, "Telegram polling failed to start");
         });
         this.started = true;
       }
@@ -85195,11 +85309,26 @@ var init_bot = __esm({
         }
         this.bot.stop();
         this.started = false;
+        this.logger.info("Telegram bot stopped");
       }
       async sendText(chatId, text) {
         await this.bot.api.sendMessage(chatId, text);
       }
+      isAuthorized(userId) {
+        if (!this.config.telegramUserId) {
+          return false;
+        }
+        return String(userId) === this.config.telegramUserId;
+      }
       registerHandlers() {
+        this.bot.use(async (ctx, next) => {
+          const userId = ctx.from?.id;
+          if (!userId || !this.isAuthorized(userId)) {
+            this.logger.warn({ userId }, "Unauthorized Telegram message, ignoring");
+            return;
+          }
+          await next();
+        });
         this.bot.command("start", async (ctx) => {
           await ctx.reply(
             "Hello from Io. Send me a message and I will route it through the daemon orchestrator."
@@ -85406,8 +85535,13 @@ async function main() {
     setChatOrchestrator(orchestrator2);
     const apiServer = createApiServer(config2);
     const telegramBot = createTelegramBot(config2, orchestrator2);
-    telegramBot?.start();
-    createTelegramNotifier(telegramBot, config2, eventBus);
+    if (telegramBot) {
+      telegramBot.start();
+      createTelegramNotifier(telegramBot, config2, eventBus);
+      logger2.info("Telegram bot initialized");
+    } else {
+      logger2.info("Telegram bot disabled (no token configured)");
+    }
     registerShutdownHandlers(logger2, async () => {
       if (pricingRefreshTimer) {
         clearInterval(pricingRefreshTimer);