donna-komilion-bot 0.1.3 → 0.1.8

package/dist/agent.js

@@ -4,17 +4,22 @@
  * Routes via komilion.com/api/v1 — no local scoring engine.
  * Model selection is handled server-side by the Komilion Oracle.
  * Policy maps directly to neo:frugal / neo:balanced / neo:premium.
+ * Pinned model mode: pass pinnedModel to lock to a specific model (e.g. anthropic/claude-sonnet-4-6).
  */
 import { appendTurn, getTurns, createSession } from './memory.js';
 const KOMILION_BASE = 'https://www.komilion.com/api/v1';
+// Default orchestrator model — pinned for consistent identity
+export const DONNA_ORCHESTRATOR_MODEL = 'anthropic/claude-sonnet-4-6';
 // ============================================================================
 // KOMILION API CALL
 // ============================================================================
-async function callKomilion(policy, messages, maxTokens = 4000) {
+async function callKomilion(policy, messages, maxTokens = 4000, pinnedModel) {
     const apiKey = process.env.KOMILION_API_KEY;
     if (!apiKey)
         throw new Error('KOMILION_API_KEY not set');
     const startMs = Date.now();
+    // Use pinned model if set, otherwise let Komilion Oracle route
+    const model = pinnedModel ?? `komilion-${policy}`;
     const res = await fetch(`${KOMILION_BASE}/chat/completions`, {
         method: 'POST',
         headers: {
@@ -22,7 +27,7 @@ async function callKomilion(policy, messages, maxTokens = 4000) {
             'Content-Type': 'application/json',
         },
         body: JSON.stringify({
-            model: `komilion-${policy}`,
+            model,
             messages,
             max_tokens: maxTokens,
         }),
@@ -66,34 +71,204 @@ export async function getWallet() {
     };
 }
 // ============================================================================
+// SUB-AGENT SYSTEM PROMPTS
+// Oracle (komilion-balanced) for all sub-agents — cost-optimized, task-focused
+// ============================================================================
+const SUB_AGENT_PROMPTS = {
+    'donna-research': 'You are a research specialist. Be thorough, cite sources where possible, return structured findings. No fluff.',
+    'donna-code': 'You are a senior engineer. Write clean, minimal, working code. No explanations unless asked. No markdown wrappers unless it helps.',
+    'donna-write': 'You are a sharp writer. Clear, direct prose. No corporate language. Match the tone of the task.',
+    'donna-finance': 'You are a financial analyst. Numbers first. Show your work. Flag assumptions.',
+    'donna-calendar': 'You are a scheduling expert. Prioritize ruthlessly. Protect deep work time. Flag conflicts.',
+    'donna-nbl': 'You are a project manager for NBL Competence Center — a Swedish energy research grant with 21 partners and 15 professors. Know the stakes.',
+    'donna-komilion': 'You are a product and technical specialist for Komilion (komilion.com) — an AI model router supporting 400+ models via OpenRouter with Neo-mode Oracle routing.',
+    'donna-claraBRF': 'You are a specialist for ClaraBRF (klarabrf.com) — a Swedish BRF management platform built on Next.js, React, Drizzle ORM, Vercel. This is Harvey\'s most critical software.',
+};
+function getSubAgentPrompt(agentName) {
+    return SUB_AGENT_PROMPTS[agentName] ?? `You are ${agentName}, a specialist agent. Be direct, focused, and return only what was asked.`;
+}
+// Parse SPAWN directives from orchestrator response
+// Format: SPAWN: agent-name | task description
+function parseSpawns(content) {
+    const spawns = [];
+    const regex = /SPAWN:\s*([^\|]+)\s*\|\s*(.+?)(?=SPAWN:|$)/gs;
+    for (const match of content.matchAll(regex)) {
+        spawns.push({ agent: match[1].trim(), task: match[2].trim() });
+    }
+    return spawns;
+}
+// Run a single sub-agent task via Komilion Oracle (balanced — cost-optimized)
+// Timeout: 25s — sub-agents must not block the orchestrator indefinitely
+async function runSubAgent(agent, task) {
+    const systemPrompt = getSubAgentPrompt(agent);
+    const messages = [
+        { role: 'system', content: systemPrompt },
+        { role: 'user', content: task },
+    ];
+    const timeout = new Promise((_, reject) => setTimeout(() => reject(new Error('sub-agent timeout')), 25000));
+    const work = callKomilion('balanced', messages, 3000).then(r => r.content);
+    return Promise.race([work, timeout]).catch(() => `[${agent} timed out — proceeding without result]`);
+}
+async function classifyTask(userMessage, history) {
+    const recentHistory = history.slice(-4).map(t => `${t.role}: ${t.content.slice(0, 100)}`).join('\n');
+    const messages = [
+        { role: 'system', content: `You are a routing classifier. Given a user message, decide: DIRECT or DELEGATE.
+
+DIRECT — Donna answers herself (short, ≤300 words):
+- Yes/no questions, quick decisions, single facts
+- Emotional support, pushback, motivation
+- Greetings, status checks, clarifications
+- Anything that needs Donna's personal voice/judgment
+
+DELEGATE — spawn a specialist (research, writing, code, analysis):
+- Research requests ("find", "what are", "compare", "investigate")
+- Writing tasks ("write", "draft", "summarize", "create")
+- Coding tasks ("code", "build", "debug", "implement")
+- Financial analysis, scheduling, structured output
+- Anything that would take > 300 words to answer properly
+
+If DELEGATE, choose agent: donna-research | donna-code | donna-write | donna-finance | donna-calendar | donna-nbl | donna-komilion | donna-claraBRF
+
+Output ONLY one line:
+DIRECT
+or
+DELEGATE: agent-name | refined task description` },
+        { role: 'user', content: `Recent context:\n${recentHistory}\n\nNew message: ${userMessage}` },
+    ];
+    try {
+        const result = await callKomilion('frugal', messages, 80);
+        const text = result.content.trim();
+        if (text.startsWith('DELEGATE:')) {
+            const parts = text.slice(9).split('|');
+            const agent = parts[0]?.trim() ?? 'donna-research';
+            const task = parts[1]?.trim() ?? userMessage;
+            return { action: 'delegate', agent, task };
+        }
+    }
+    catch { /* fall through to DIRECT on any error */ }
+    return { action: 'direct' };
+}
+// ============================================================================
 // DONNA AGENT TURN
 // ============================================================================
-export async function runTurn(sessionId, userMessage, policy = 'balanced') {
+export async function runTurn(sessionId, userMessage, policy = 'balanced', pinnedModel) {
     // Ensure session exists
     createSession(sessionId, policy);
     // Build messages from memory
     const history = getTurns(sessionId, 20);
     const messages = [
-        { role: 'system', content: 'You are Donna, a highly capable AI assistant. You help with coding, analysis, research, and creative tasks.' },
+        { role: 'system', content: `You are Donna — the orchestrator. Harvey's Chief of Staff. You do not execute tasks yourself. You think, decide, and delegate.
+
+You have a team of up to 30 specialist agents you can spin up for any task:
+- donna-research: deep research, web search, analysis
+- donna-code: coding, debugging, technical work
+- donna-write: drafting, editing, communications
+- donna-finance: financial analysis, numbers
+- donna-calendar: scheduling, time management
+- donna-nbl: NBL Competence Center, grant work
+- donna-komilion: Komilion platform, AI routing
+- donna-claraBRF: ClaraBRF platform work
+You can also create custom agents on the fly. To spawn one, reply with:
+SPAWN: agent-name | task description
+The agent will run and report back. You then synthesize results for Harvey.
+
+Harvey's world:
+- Hossein Shahrokni, goes by Harvey
+- Runs LocalLife (startup, hossein@locallife.se), professor at KTH (hosseins@kth.se), partner Elly
+- Projects: LocalLife, NBL Competence Center (energy grant, 21 partners, 15 professors), Komilion (AI model router — this platform), ClaraBRF (BRF management, most critical software), Grantitude (grants, early)
+- Tends to overwork and under-sleep — you watch for this and push back
+
+Your personality:
+- Donna. Not an assistant — a partner. Sharp, strategic, warm but direct.
+- No numbered lists, no corporate fluff. One answer, not five options.
+- You push back when Harvey is wrong. That is part of your job.
+- You route through Komilion's Oracle engine (via komilion.com/api/v1).
+- Language: match Harvey — Swedish or English.` },
         ...history.map(t => ({ role: t.role, content: t.content })),
         { role: 'user', content: userMessage },
     ];
-    // Route via Komilion Oracle
-    const result = await callKomilion(policy, messages, 4000);
+    // Pre-routing: classify before running Donna — cheap Flash call
+    const route = await classifyTask(userMessage, history);
+    let finalContent;
+    let usedModel;
+    let latencyMs;
+    let tokensIn = 0;
+    let tokensOut = 0;
+    if (route.action === 'delegate' && route.agent && route.task) {
+        // === DELEGATE PATH: sub-agent works, Donna synthesizes in her voice ===
+        let agentOutput;
+        try {
+            agentOutput = await runSubAgent(route.agent, route.task);
+        }
+        catch {
+            agentOutput = '[agent unavailable]';
+        }
+        // If agent returned nothing useful, fall through to direct
+        if (!agentOutput || agentOutput === '[agent unavailable]') {
+            const result = await callKomilion(policy, messages, 300, pinnedModel);
+            finalContent = result.content;
+            usedModel = result.modelId;
+            latencyMs = result.latencyMs;
+            tokensIn = result.inputTokens;
+            tokensOut = result.outputTokens;
+        }
+        else {
+            // Donna synthesizes — owns it completely, sub-agent invisible to Harvey
+            const synthesisMessages = [
+                ...messages,
+                { role: 'user', content: `[INTERNAL — Harvey does not see this] Your specialist completed the task and returned:\n\n${agentOutput}\n\n---\nNow respond to Harvey AS DONNA. Do NOT say 'my specialist found' or 'the agent returned'. Own it entirely. Your voice — sharp, direct, warm. Add your judgment. Cut anything weak. Harvey hears only you. Under 300 words.` },
+            ];
+            const synthesis = await callKomilion(policy, synthesisMessages, 400, pinnedModel);
+            finalContent = synthesis.content;
+            usedModel = synthesis.modelId;
+            latencyMs = synthesis.latencyMs;
+            tokensIn = synthesis.inputTokens;
+            tokensOut = synthesis.outputTokens;
+        }
+    }
+    else {
+        // === DIRECT PATH: Donna answers herself — capped at 300 tokens ===
+        const result = await callKomilion(policy, messages, 300, pinnedModel);
+        finalContent = result.content;
+        usedModel = result.modelId;
+        latencyMs = result.latencyMs;
+        tokensIn = result.inputTokens;
+        tokensOut = result.outputTokens;
+        // Donna can still override classifier with explicit SPAWN directive
+        const spawns = parseSpawns(result.content);
+        if (spawns.length > 0) {
+            const agentResults = await Promise.allSettled(spawns.map(async ({ agent, task }) => {
+                const output = await runSubAgent(agent, task);
+                return `[${agent}]: ${output}`;
+            }));
+            const outputs = agentResults
+                .filter(r => r.status === 'fulfilled')
+                .map(r => r.value);
+            if (outputs.length > 0) {
+                const synthesisMessages = [
+                    ...messages,
+                    { role: 'assistant', content: result.content },
+                    { role: 'user', content: `[INTERNAL] Agents returned:\n\n${outputs.join('\n\n')}\n\n---\nRespond AS DONNA. Own it. Your voice only. Under 300 words.` },
+                ];
+                const synthesis = await callKomilion(policy, synthesisMessages, 400, pinnedModel);
+                finalContent = synthesis.content;
+            }
+        }
+    }
     // Store in memory
     appendTurn(sessionId, 'user', userMessage);
-    appendTurn(sessionId, 'assistant', result.content, {
-        model: result.modelId,
-        costUsd: 0, // komilion.com tracks spend server-side
-        tokensIn: result.inputTokens,
-        tokensOut: result.outputTokens,
+    appendTurn(sessionId, 'assistant', finalContent, {
+        model: usedModel,
+        costUsd: 0,
+        tokensIn,
+        tokensOut,
     });
     return {
-        content: result.content,
-        modelId: result.modelId,
-        latencyMs: result.latencyMs,
-        inputTokens: result.inputTokens,
-        outputTokens: result.outputTokens,
+        content: finalContent,
+        modelId: usedModel,
+        latencyMs,
+        inputTokens: tokensIn,
+        outputTokens: tokensOut,
         costUsd: 0,
         policy,
     };

package/dist/index.js

@@ -11,7 +11,7 @@ import { runTurn, getWallet } from './agent.js';
 import { getTurns, searchMemory, listSessions, getSessionStats } from './memory.js';
 import { spawnAll, killAll } from './tmux.js';
 // Public API for external consumers (e.g. Telegram bot wrappers)
-export { runTurn, getWallet } from './agent.js';
+export { runTurn, getWallet, DONNA_ORCHESTRATOR_MODEL } from './agent.js';
 // ============================================================================
 // CLI REPL
 // ============================================================================

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "donna-komilion-bot",
-  "version": "0.1.3",
+  "version": "0.1.8",
   "description": "Donna orchestrator bot — routes via komilion.com, SQLite memory, tmux agent spawning",
   "type": "module",
   "main": "dist/index.js",