kiku-agent 1.0.11 → 1.2.0

package/server/agent/suggestionTui.js

@@ -0,0 +1,294 @@
+import readline from 'node:readline';
+import { C, getCategoryBadge, getStringDisplayWidth, getFullWidth } from './ui.js';
+import { isColorDisabled } from './theme.js';
+
+export const SUGGESTION_STUDIO_TABS = Object.freeze([
+  'ALL',
+  'DRAMA',
+  'MUSIC',
+  'FASHION',
+  'MILESTONE',
+  'ANALYSIS',
+  'TRIVIA'
+]);
+
+export function renderSuggestionStudioView({
+  suggestions = [],
+  selectedIndex = 0,
+  selectedIds = new Set(),
+  activeCategory = 'ALL',
+  termWidth = getFullWidth(),
+} = {}) {
+  const width = Math.max(50, Math.min(termWidth - 2, 90));
+  const innerWidth = width - 4;
+  const rows = [];
+
+  const filtered = activeCategory === 'ALL'
+    ? suggestions
+    : suggestions.filter(s => String(s.category || '').toUpperCase() === activeCategory);
+
+  const headerBg = isColorDisabled() ? '' : '\x1b[48;2;65;20;42m';
+  const headerFg = isColorDisabled() ? '' : '\x1b[38;2;255;220;235m';
+
+  rows.push(`  ${headerBg}${headerFg}${C.bold}  🌸 KIKU SUGGESTION STUDIO  ${C.reset}`);
+  rows.push(`${C.dim}  Interactive topic explorer • Select multiple with [Space] and launch with [Enter]${C.reset}`);
+  rows.push('');
+
+  const tabTokens = SUGGESTION_STUDIO_TABS.map(tab => {
+    const badge = tab === 'ALL' ? { icon: '✨', label: 'ALL', bg: '\x1b[48;2;60;25;40m', color: C.sakura } : getCategoryBadge(tab);
+    const isCurrent = tab === activeCategory;
+    if (isColorDisabled()) {
+      return isCurrent ? `[${badge.icon} ${badge.label}]` : `${badge.icon} ${badge.label}`;
+    }
+    if (isCurrent) {
+      return `\x1b[48;2;212;83;110m\x1b[38;2;255;255;255m\x1b[1m  ${badge.icon} ${badge.label}  \x1b[0m`;
+    }
+    return `${badge.bg || '\x1b[48;2;36;32;44m'}\x1b[38;2;180;180;200m ${badge.icon} ${badge.label} \x1b[0m`;
+  });
+  rows.push(`  ${tabTokens.join('  ')}`);
+  rows.push('');
+
+  if (filtered.length === 0) {
+    rows.push(`    ${C.dim}No suggestions found under ${activeCategory}. Press [Tab] to switch category.${C.reset}`);
+    rows.push('');
+  } else {
+    for (let i = 0; i < filtered.length; i++) {
+      const item = filtered[i];
+      const isSelected = i === selectedIndex;
+      const isChecked = selectedIds.has(item.id);
+      const badge = getCategoryBadge(item.category);
+
+      const pointer = isSelected ? `${C.bold}${C.gold}❯${C.reset}` : ' ';
+      const checkMark = isChecked
+        ? (isColorDisabled() ? '[◉]' : '\x1b[48;2;20;80;40m\x1b[38;2;80;240;140m\x1b[1m[◉]\x1b[0m')
+        : (isColorDisabled() ? '[ ]' : '\x1b[48;2;35;35;45m\x1b[38;2;160;160;180m[ ]\x1b[0m');
+      const numLabel = `${item.id}.`;
+      const catBadge = isColorDisabled()
+        ? `[${badge.icon} ${badge.label}]`
+        : `${badge.bg}${badge.color}${C.bold} ${badge.icon} ${badge.label} ${C.reset}`;
+
+      let qText = item.question;
+      const availableTextWidth = Math.max(10, innerWidth - 20);
+      if (getStringDisplayWidth(qText) > availableTextWidth) {
+        let truncated = '';
+        for (const char of qText) {
+          if (getStringDisplayWidth(truncated + char) > availableTextWidth - 2) break;
+          truncated += char;
+        }
+        qText = `${truncated}…`;
+      }
+
+      const qStyled = isSelected ? `${C.bold}${C.white}${qText}${C.reset}` : `${C.white}${qText}${C.reset}`;
+      rows.push(`  ${pointer} ${checkMark} ${C.bold}${numLabel}${C.reset} ${catBadge} ${qStyled}`);
+
+      if (isSelected) {
+        const descText = item.description || 'Explore deep multi-agent insights and citations.';
+        const cleanDesc = descText.length > innerWidth - 6 ? `${descText.slice(0, innerWidth - 9)}…` : descText;
+        const focusTag = isColorDisabled() ? 'Focus:' : '\x1b[48;2;45;20;32m\x1b[38;2;255;190;210m Focus: \x1b[0m';
+        rows.push(`      ${focusTag} ${C.slate}${cleanDesc}${C.reset}`);
+
+        const toolsStr = Array.isArray(item.tools) ? item.tools.join(', ') : 'search_works';
+        const depthStr = item.depth || 'Deep Analysis';
+        const estStr = `~${item.estimatedSeconds || 2}s`;
+
+        const toolTag = isColorDisabled() ? 'Tools:' : '\x1b[48;2;20;40;55m\x1b[38;2;120;220;255m Tools: \x1b[0m';
+        const estTag = isColorDisabled() ? 'Est:' : '\x1b[48;2;45;38;20m\x1b[38;2;255;210;80m Est: \x1b[0m';
+        const depthTag = isColorDisabled() ? 'Depth:' : '\x1b[48;2;20;50;35m\x1b[38;2;100;240;150m Depth: \x1b[0m';
+
+        rows.push(`      ${toolTag} ${C.slate}${toolsStr}${C.reset}   ${estTag} ${C.slate}${estStr}${C.reset}   ${depthTag} ${C.slate}${depthStr}${C.reset}`);
+      }
+    }
+  }
+
+  rows.push('');
+  const checkedCount = selectedIds.size;
+  const launchLabel = checkedCount > 0 ? `Launch Batch (${checkedCount})` : 'Launch (1)';
+  const navBar = isColorDisabled()
+    ? `[↑/↓] Nav • [Space] Select • [Enter] ${launchLabel} • [Tab] Tab • [A] All • [Esc] Cancel`
+    : `\x1b[48;2;36;28;44m\x1b[38;2;225;210;245m  [↑/↓] Nav  •  [Space] Select  •  [Enter] ${launchLabel}  •  [Tab] Tab  •  [A] All  •  [Esc] Cancel  \x1b[0m`;
+  rows.push(`  ${navBar}`);
+  rows.push(`  ${C.dim}[1-9] Quick Toggle  •  [R] Refresh  •  [Q] Exit${C.reset}`);
+
+  return rows;
+}
+
+
+export async function promptInteractiveSuggestion(suggestions = [], options = {}) {
+  const stdin = options.stdin || process.stdin;
+  const stdout = options.stdout || process.stdout;
+
+  
+  if (!stdin.isTTY || !stdout.isTTY || options.plain || process.env.KIKU_PLAIN === 'true') {
+    return { action: 'display', targets: null };
+  }
+
+  return new Promise((resolve) => {
+    let selectedIndex = 0;
+    const selectedIds = new Set();
+    let currentTabIdx = 0;
+    let activeCategory = SUGGESTION_STUDIO_TABS[0];
+    let isCleanedUp = false;
+    let renderedRowCount = 0;
+
+    const getFiltered = () => {
+      return activeCategory === 'ALL'
+        ? suggestions
+        : suggestions.filter(s => String(s.category || '').toUpperCase() === activeCategory);
+    };
+
+    const cleanup = () => {
+      if (isCleanedUp) return;
+      isCleanedUp = true;
+      stdin.removeListener('keypress', handleKey);
+      try { stdin.setRawMode(false); } catch {}
+     
+      if (renderedRowCount > 0) {
+        readline.cursorTo(stdout, 0);
+        readline.moveCursor(stdout, 0, -renderedRowCount + 1);
+        readline.clearScreenDown(stdout);
+      }
+    };
+
+    const render = () => {
+      if (isCleanedUp) return;
+      const termWidth = stdout.columns || 80;
+      const frameRows = renderSuggestionStudioView({
+        suggestions,
+        selectedIndex,
+        selectedIds,
+        activeCategory,
+        termWidth,
+      });
+
+      if (renderedRowCount > 0) {
+        readline.cursorTo(stdout, 0);
+        readline.moveCursor(stdout, 0, -renderedRowCount + 1);
+        readline.clearScreenDown(stdout);
+      }
+
+      stdout.write(frameRows.join('\n') + '\n');
+      renderedRowCount = frameRows.length;
+    };
+
+    const handleKey = (str, key) => {
+      if (isCleanedUp) return;
+
+     
+      if ((key?.ctrl && key?.name === 'c') || key?.name === 'escape' || str === 'q' || str === 'Q') {
+        cleanup();
+        return resolve({ action: 'cancel' });
+      }
+
+      const filtered = getFiltered();
+
+    
+      if (key?.name === 'up' || str === 'k') {
+        if (filtered.length > 0) {
+          selectedIndex = (selectedIndex - 1 + filtered.length) % filtered.length;
+          render();
+        }
+        return;
+      }
+
+      
+      if (key?.name === 'down' || str === 'j') {
+        if (filtered.length > 0) {
+          selectedIndex = (selectedIndex + 1) % filtered.length;
+          render();
+        }
+        return;
+      }
+
+     
+      if (key?.name === 'tab' || key?.name === 'right') {
+        currentTabIdx = (currentTabIdx + 1) % SUGGESTION_STUDIO_TABS.length;
+        activeCategory = SUGGESTION_STUDIO_TABS[currentTabIdx];
+        selectedIndex = 0;
+        render();
+        return;
+      }
+      if (key?.name === 'left') {
+        currentTabIdx = (currentTabIdx - 1 + SUGGESTION_STUDIO_TABS.length) % SUGGESTION_STUDIO_TABS.length;
+        activeCategory = SUGGESTION_STUDIO_TABS[currentTabIdx];
+        selectedIndex = 0;
+        render();
+        return;
+      }
+
+      
+      if (key?.name === 'space' || str === ' ') {
+        const item = filtered[selectedIndex];
+        if (item) {
+          if (selectedIds.has(item.id)) {
+            selectedIds.delete(item.id);
+          } else {
+            selectedIds.add(item.id);
+          }
+          render();
+        }
+        return;
+      }
+
+     
+      if (str === 'a' || str === 'A') {
+        const allChecked = filtered.every(s => selectedIds.has(s.id));
+        if (allChecked) {
+          filtered.forEach(s => selectedIds.delete(s.id));
+        } else {
+          filtered.forEach(s => selectedIds.add(s.id));
+        }
+        render();
+        return;
+      }
+
+    
+      if (str === 'r' || str === 'R') {
+        cleanup();
+        return resolve({ action: 'refresh' });
+      }
+
+      
+      if (/^[1-9]$/.test(str)) {
+        const num = Number.parseInt(str, 10);
+        const matchIdx = filtered.findIndex(s => s.id === num);
+        if (matchIdx >= 0) {
+          selectedIndex = matchIdx;
+          if (selectedIds.has(num)) {
+            selectedIds.delete(num);
+          } else {
+            selectedIds.add(num);
+          }
+          render();
+        }
+        return;
+      }
+
+     
+      if (key?.name === 'return' || str === '\r' || str === '\n') {
+        cleanup();
+        let targetStr = '';
+        if (selectedIds.size > 0) {
+          targetStr = Array.from(selectedIds).sort((a, b) => a - b).join(',');
+        } else {
+          const item = filtered[selectedIndex] || suggestions[0];
+          targetStr = item ? String(item.id) : '1';
+        }
+        return resolve({ action: 'run', targets: targetStr });
+      }
+    };
+
+  
+    readline.emitKeypressEvents(stdin);
+    try { stdin.setRawMode(true); } catch {}
+    stdin.on('keypress', handleKey);
+
+    if (options.signal) {
+      options.signal.addEventListener('abort', () => {
+        cleanup();
+        resolve({ action: 'cancel' });
+      }, { once: true });
+    }
+
+    render();
+  });
+}

package/server/agent/suggestions.js

@@ -16,44 +16,89 @@ export const CURATED_SUGGESTION_TEMPLATES = Object.freeze([
     category: 'DRAMA',
     question: "What are Ju Jingyi's upcoming 2026/2025 drama projects, characters, and release platforms?",
     description: "Detailed breakdown of upcoming dramas, character names, and broadcasting platforms.",
+    tools: ['search_works', 'web_search'],
+    depth: 'Deep Analysis',
+    estimatedSeconds: 3,
   },
   {
     category: 'MUSIC',
     question: "What are the most popular OSTs and solo singles released by Ju Jingyi, and what are their YouTube links?",
     description: "Discography highlights from In Blossom, Ruyi Pavilion, and Legend of Yunxi with video links.",
+    tools: ['search_discography', 'search_videos'],
+    depth: 'Media Scan',
+    estimatedSeconds: 2,
   },
   {
     category: 'FASHION',
     question: "What are Ju Jingyi's major luxury brand ambassadorships and global fashion partnerships in 2025-2026?",
     description: "Overview of Versace Global Brand Ambassador role, Maybelline, and top endorsements.",
+    tools: ['get_profile_and_fandom', 'search_works'],
+    depth: 'Portfolio Scan',
+    estimatedSeconds: 2,
   },
   {
     category: 'MILESTONE',
     question: "How did Ju Jingyi achieve back-to-back #1 victories in the SNH48 General Elections and transition to her solo studio?",
     description: "Chronological analysis of the SNH48 Hall of Fame elevation and independent studio milestones.",
+    tools: ['get_career_timeline', 'search_awards'],
+    depth: 'Historical Deep Dive',
+    estimatedSeconds: 3,
   },
   {
     category: 'ANALYSIS',
     question: "What acting awards and cultural influence honors has Ju Jingyi received across Chinese television and media?",
     description: "Awards summary spanning Weibo Night, Baidu Feast, Tencent Starlight, and drama recognitions.",
+    tools: ['search_awards', 'get_career_timeline'],
+    depth: 'Accolades Breakdown',
+    estimatedSeconds: 2,
   },
   {
     category: 'TRIVIA',
     question: "What is the meaning and background of Ju Jingyi's fan community name 'Mìjú' (蜜橘) and official fandom colors?",
     description: "Biographical details, fan community history, official colors, and community traditions.",
+    tools: ['get_profile_and_fandom'],
+    depth: 'Fandom Trivia',
+    estimatedSeconds: 1,
   },
   {
     category: 'DRAMA',
     question: "How did Ju Jingyi's performance in 'In Blossom' (花间令) as Yang Caiwei receive audience acclaim and streaming milestones?",
     description: "Deep dive into character duality, critical reception, and viewership records.",
+    tools: ['search_works', 'search_awards'],
+    depth: 'Performance Review',
+    estimatedSeconds: 2,
   },
   {
     category: 'MUSIC',
     question: "What are Ju Jingyi's live stage performances and music videos with the highest viewership?",
     description: "Top viral stage performances, traditional guofeng choreography, and music video releases.",
+    tools: ['search_discography', 'search_videos'],
+    depth: 'Visual Stages',
+    estimatedSeconds: 2,
   },
 ]);
 
+export async function mapConcurrent(items, limit, fn) {
+  const results = new Array(items.length);
+  const maxLimit = Math.max(1, limit || 1);
+  let nextIndex = 0;
+
+  const workers = Array.from({ length: Math.min(maxLimit, items.length) }, async () => {
+    while (nextIndex < items.length) {
+      const currentIndex = nextIndex++;
+      try {
+        const val = await fn(items[currentIndex], currentIndex);
+        results[currentIndex] = { status: 'fulfilled', value: val };
+      } catch (reason) {
+        results[currentIndex] = { status: 'rejected', reason };
+      }
+    }
+  });
+
+  await Promise.all(workers);
+  return results;
+}
+
 export class SuggestionEngine {
   constructor(options = {}) {
     this.client = options.client || null;
@@ -89,6 +134,9 @@ export class SuggestionEngine {
       category: item.category,
       question: item.question,
       description: item.description,
+      tools: item.tools || ['search_works'],
+      depth: item.depth || 'Standard',
+      estimatedSeconds: item.estimatedSeconds || 2,
       source: 'catalog',
     }));
 
@@ -98,6 +146,11 @@ export class SuggestionEngine {
   async generateSuggestions(options = {}) {
     const limit = Math.min(Math.max(options.limit || 5, 2), 8);
     const topic = (options.topic || '').trim();
+    const lang = String(options.lang || 'en').toLowerCase();
+    const langNames = { th: 'Thai', zh: 'Simplified Chinese', en: 'English', ja: 'Japanese', ko: 'Korean' };
+    const langInstruction = langNames[lang]
+      ? `Write every question and description entirely in ${langNames[lang]} (no English except proper nouns like drama titles).`
+      : '';
 
     if (!this.client || typeof this.client.chat !== 'function') {
       return this.generateCatalogSuggestions(options);
@@ -111,6 +164,7 @@ export class SuggestionEngine {
       const systemPrompt = `You are the Ju Jingyi intelligence coordinator for Kiku Agent.
 Generate ${limit} distinct, highly specific, and exciting research questions about Ju Jingyi that fans or researchers would want to explore.
 ${topicInstruction}
+${langInstruction}
 
 Output ONLY a valid JSON array of objects with the exact schema:
 [
@@ -123,7 +177,7 @@ Output ONLY a valid JSON array of objects with the exact schema:
 Do not include markdown code fences or any text other than the JSON array.`;
 
       const controller = new AbortController();
-      const timeoutId = setTimeout(() => controller.abort(), 6000);
+      const timeoutId = setTimeout(() => controller.abort(), 30000);
       if (options.signal) {
         options.signal.addEventListener('abort', () => controller.abort(), { once: true });
       }
@@ -153,6 +207,9 @@ Do not include markdown code fences or any text other than the JSON array.`;
           category: String(item.category || 'ANALYSIS').toUpperCase(),
           question: String(item.question || '').trim(),
           description: String(item.description || '').trim(),
+          tools: Array.isArray(item.tools) && item.tools.length > 0 ? item.tools : ['search_works'],
+          depth: item.depth || 'Deep Analysis',
+          estimatedSeconds: item.estimatedSeconds || 3,
           source: 'ai_model',
         })).filter(s => s.question.length > 5);
 
@@ -236,8 +293,17 @@ Do not include markdown code fences or any text other than the JSON array.`;
       })),
     });
 
+    const isLocalProvider = Boolean(
+      this.client?.provider === 'lmstudio' ||
+      this.client?.provider === 'ollama' ||
+      this.client?.isLocal ||
+      this.client?.getProviderInfo?.()?.isLocal
+    );
+    const defaultLimit = isLocalProvider ? 1 : this.maxWorkers;
+    const concurrency = Math.min(Math.max(options.concurrency || defaultLimit, 1), this.maxWorkers || 5);
+
     try {
-      const settled = await Promise.allSettled(tasks.map(async (task, index) => {
+      const settled = await mapConcurrent(tasks, concurrency, async (task, index) => {
         const workerId = `worker_${index + 1}`;
         const workerLabel = `[Worker ${index + 1}] ${task.category || 'Research'}`;
 
@@ -298,7 +364,7 @@ Do not include markdown code fences or any text other than the JSON array.`;
           toolCalls: result.toolCalls,
           iterations: result.iterations,
         };
-      }));
+      });
 
       const results = settled.map((entry, index) => {
         if (entry.status === 'fulfilled') return entry.value;

package/server/agent/telemetry.js

@@ -0,0 +1,119 @@
+export const MODEL_PRICING = {
+  lmstudio: {
+    default: { in: 0.0, out: 0.0 },
+    'google/gemma-4-e4b': { in: 0.0, out: 0.0 },
+    'google/gemma-4-e4b-Q4_K_M': { in: 0.0, out: 0.0 },
+    'google/gemma-4-e4b-Q6_K': { in: 0.0, out: 0.0 },
+    'google/gemma-4-e4b-Q8_0': { in: 0.0, out: 0.0 }
+  },
+  groq: {
+    default: { in: 0.15, out: 0.60 },
+    'openai/gpt-oss-120b': { in: 0.15, out: 0.60 },
+    'openai/gpt-oss-20b': { in: 0.075, out: 0.30 }
+  },
+  deepseek: {
+    default: { in: 0.30, out: 1.20 },
+    'deepseek-v4-flash': { in: 0.27, out: 1.10 },
+    'deepseek-v4-pro': { in: 0.99, out: 2.19 }
+  },
+  openai: {
+    default: { in: 0.20, out: 1.20 },
+    'gpt-5.6-luna': { in: 0.20, out: 1.20 },
+    'gpt-5.6-terra': { in: 2.00, out: 6.00 },
+    'gpt-5.6-sol': { in: 4.00, out: 12.00 }
+  },
+  gemini: {
+    default: { in: 0.75, out: 3.00 },
+    'gemini-3.7-flash': { in: 0.75, out: 3.00 },
+    'gemini-3.5-flash-lite': { in: 0.30, out: 1.20 },
+    'gemini-3.6-flash': { in: 0.75, out: 3.00 }
+  },
+  openrouter: {
+    default: { in: 0.20, out: 0.80 },
+    'openrouter/auto': { in: 0.20, out: 0.80 }
+  },
+  custom: {
+    default: { in: 0.0, out: 0.0 }
+  }
+};
+
+
+export function estimateTokens(text) {
+  if (!text || typeof text !== 'string') return 0;
+  const str = text.trim();
+  if (!str) return 0;
+
+ 
+  const cjkAndThai = str.match(/[\u4e00-\u9fa5\u0e00-\u0e7f\u3040-\u30ff\uac00-\ud7af]/g) || [];
+  const nonCjk = str.replace(/[\u4e00-\u9fa5\u0e00-\u0e7f\u3040-\u30ff\uac00-\ud7af]/g, ' ');
+  const words = nonCjk.split(/\s+/).filter(Boolean);
+
+ 
+  const cjkTokens = Math.ceil(cjkAndThai.length * 1.25);
+
+ 
+  const wordTokens = Math.ceil(words.length * 1.3);
+
+  return Math.max(1, cjkTokens + wordTokens);
+}
+
+
+export function calculateTokenCost(providerOrTokens, modelOrProvider, promptTokensOrModel = 0, completionTokens = 0) {
+  let pId, mId, prompt, completion;
+  if (typeof providerOrTokens === 'object' && providerOrTokens !== null) {
+    prompt = Number(providerOrTokens.prompt || 0);
+    completion = Number(providerOrTokens.completion || 0);
+    pId = String(modelOrProvider || 'lmstudio').toLowerCase().trim();
+    mId = String(promptTokensOrModel || 'default').trim();
+  } else {
+    pId = String(providerOrTokens || 'lmstudio').toLowerCase().trim();
+    mId = String(modelOrProvider || 'default').trim();
+    prompt = Number(promptTokensOrModel || 0);
+    completion = Number(completionTokens || 0);
+  }
+
+  const providerPricing = MODEL_PRICING[pId] || MODEL_PRICING.custom;
+  const rates = providerPricing[mId] || providerPricing.default || { in: 0, out: 0 };
+
+  const inputCost = (prompt * rates.in) / 1_000_000;
+  const outputCost = (completion * rates.out) / 1_000_000;
+
+  return Number((inputCost + outputCost).toFixed(6));
+}
+
+/**
+ * Human-readable formatted cost in USD
+ */
+export function formatCostUsd(amount, isLocal = false) {
+  const num = Number(amount || 0);
+  if (isLocal || num === 0) {
+    return '$0.00 (Local)';
+  }
+  if (num < 0.0001) {
+    return '<$0.0001';
+  }
+  return `$${num.toFixed(4)}`;
+}
+
+/**
+ * Format token count with thousand separators
+ */
+export function formatTokenCount(count) {
+  const num = Number(count || 0);
+  return num.toLocaleString('en-US');
+}
+
+/**
+ * Creates a normalized token usage object
+ */
+export function createTokenUsage({ prompt = 0, completion = 0, reasoning = 0 } = {}) {
+  const p = Math.max(0, Math.round(Number(prompt) || 0));
+  const c = Math.max(0, Math.round(Number(completion) || 0));
+  const r = Math.max(0, Math.round(Number(reasoning) || 0));
+  return {
+    prompt: p,
+    completion: c,
+    reasoning: r,
+    total: p + c
+  };
+}