kiku-agent 1.0.0 → 1.0.3

package/server/agent/theme.js

@@ -1,124 +1,125 @@
-import fs from 'node:fs';
-import path from 'node:path';
-
-const THEME_STATE_DIR = path.resolve(process.cwd(), '.kiku_sessions');
-const THEME_STATE_FILE = path.join(THEME_STATE_DIR, 'theme.json');
-
-export const THEMES = Object.freeze({
-  sakura: Object.freeze({
-    label: 'Sakura',
-    description: 'Pink accents with neutral gray surfaces',
-    colors: Object.freeze({
-      primary: '255;110;180', secondary: '255;165;195', highlight: '255;215;80',
-      warning: '245;158;11', info: '56;210;255', success: '50;235;130',
-      tertiary: '180;140;255', muted: '130;140;160', subdued: '80;85;95',
-      text: '255;241;247', error: '255;82;115', panel: '45;48;54', panelHeader: '72;52;65',
-    }),
-  }),
-  midnight: Object.freeze({
-    label: 'Midnight',
-    description: 'Deep blue surfaces with cool accents',
-    colors: Object.freeze({
-      primary: '96;165;250', secondary: '125;211;252', highlight: '250;204;21',
-      warning: '251;191;36', info: '34;211;238', success: '74;222;128',
-      tertiary: '167;139;250', muted: '148;163;184', subdued: '71;85;105',
-      text: '219;234;254', error: '251;113;133', panel: '15;23;42', panelHeader: '30;41;69',
-    }),
-  }),
-  ocean: Object.freeze({
-    label: 'Ocean',
-    description: 'Cyan and blue accents on marine-gray surfaces',
-    colors: Object.freeze({
-      primary: '34;211;238', secondary: '103;232;249', highlight: '253;224;71',
-      warning: '251;191;36', info: '56;189;248', success: '45;212;191',
-      tertiary: '96;165;250', muted: '148;163;184', subdued: '71;85;105',
-      text: '207;250;254', error: '251;113;133', panel: '8;47;59', panelHeader: '14;74;89',
-    }),
-  }),
-  forest: Object.freeze({
-    label: 'Forest',
-    description: 'Green and teal accents on natural dark surfaces',
-    colors: Object.freeze({
-      primary: '74;222;128', secondary: '110;231;183', highlight: '250;204;21',
-      warning: '251;191;36', info: '45;212;191', success: '134;239;172',
-      tertiary: '163;230;53', muted: '148;163;153', subdued: '71;85;79',
-      text: '220;252;231', error: '252;129;129', panel: '20;45;34', panelHeader: '30;64;46',
-    }),
-  }),
-  sunset: Object.freeze({
-    label: 'Sunset',
-    description: 'Warm orange accents on charcoal-brown surfaces',
-    colors: Object.freeze({
-      primary: '251;146;60', secondary: '253;186;116', highlight: '250;204;21',
-      warning: '251;191;36', info: '56;189;248', success: '74;222;128',
-      tertiary: '244;114;182', muted: '168;151;143', subdued: '87;72;64',
-      text: '255;237;213', error: '253;106;106', panel: '61;35;28', panelHeader: '87;48;38',
-    }),
-  }),
-  monochrome: Object.freeze({
-    label: 'Monochrome',
-    description: 'Low-distraction grayscale presentation',
-    colors: Object.freeze({
-      primary: '229;231;235', secondary: '209;213;219', highlight: '255;255;255',
-      warning: '209;213;219', info: '229;231;235', success: '229;231;235',
-      tertiary: '209;213;219', muted: '156;163;175', subdued: '107;114;128',
-      text: '245;245;245', error: '212;212;212', panel: '31;31;31', panelHeader: '64;64;64',
-    }),
-  }),
-});
-
-export const THEME_NAMES = Object.freeze(Object.keys(THEMES));
-
-function loadSavedTheme() {
-  const envTheme = process.env.KIKU_THEME?.toLowerCase();
-  if (envTheme && THEMES[envTheme]) return envTheme;
-  try {
-    const saved = JSON.parse(fs.readFileSync(THEME_STATE_FILE, 'utf8'));
-    if (typeof saved.theme === 'string' && THEMES[saved.theme]) return saved.theme;
-  } catch {}
-  return 'sakura';
-}
-
-let activeThemeName = loadSavedTheme();
-
-export function getThemeName() {
-  return activeThemeName;
-}
-
-export function getTheme(name = activeThemeName) {
-  return THEMES[name] || THEMES.sakura;
-}
-
-export function getThemePalette(name = activeThemeName) {
-  return getTheme(name).colors;
-}
-
-export function setTheme(name, { persist = true } = {}) {
-  const normalized = String(name || '').toLowerCase();
-  if (!THEMES[normalized]) return null;
-  activeThemeName = normalized;
-  process.env.KIKU_THEME = normalized;
-
-  if (persist) {
-    fs.mkdirSync(THEME_STATE_DIR, { recursive: true });
-    fs.writeFileSync(THEME_STATE_FILE, JSON.stringify({ theme: normalized }, null, 2) + '\n', 'utf8');
-  }
-  return getTheme(normalized);
-}
-
-export function ansiForeground(rgb) {
-  return `\x1b[38;2;${rgb}m`;
-}
-
-export function ansiBackground(rgb) {
-  return `\x1b[48;2;${rgb}m`;
-}
-
-export function inkColor(rgb) {
-  return `rgb(${rgb.replaceAll(';', ',')})`;
-}
-
-export function getInkPalette(name = activeThemeName) {
-  const palette = getThemePalette(name);
-  return Object.fromEntries(Object.entries(palette).map(([key, value]) => [key, inkColor(value)]));
-}
+import fs from 'node:fs';
+import path from 'node:path';
+import { KIKU_DATA_DIR } from './appPaths.js';
+
+const THEME_STATE_DIR = KIKU_DATA_DIR;
+const THEME_STATE_FILE = path.join(THEME_STATE_DIR, 'theme.json');
+
+export const THEMES = Object.freeze({
+  sakura: Object.freeze({
+    label: 'Sakura',
+    description: 'Pink accents with neutral gray surfaces',
+    colors: Object.freeze({
+      primary: '255;110;180', secondary: '255;165;195', highlight: '255;215;80',
+      warning: '245;158;11', info: '56;210;255', success: '50;235;130',
+      tertiary: '180;140;255', muted: '130;140;160', subdued: '80;85;95',
+      text: '255;241;247', error: '255;82;115', panel: '45;48;54', panelHeader: '72;52;65',
+    }),
+  }),
+  midnight: Object.freeze({
+    label: 'Midnight',
+    description: 'Deep blue surfaces with cool accents',
+    colors: Object.freeze({
+      primary: '96;165;250', secondary: '125;211;252', highlight: '250;204;21',
+      warning: '251;191;36', info: '34;211;238', success: '74;222;128',
+      tertiary: '167;139;250', muted: '148;163;184', subdued: '71;85;105',
+      text: '219;234;254', error: '251;113;133', panel: '15;23;42', panelHeader: '30;41;69',
+    }),
+  }),
+  ocean: Object.freeze({
+    label: 'Ocean',
+    description: 'Cyan and blue accents on marine-gray surfaces',
+    colors: Object.freeze({
+      primary: '34;211;238', secondary: '103;232;249', highlight: '253;224;71',
+      warning: '251;191;36', info: '56;189;248', success: '45;212;191',
+      tertiary: '96;165;250', muted: '148;163;184', subdued: '71;85;105',
+      text: '207;250;254', error: '251;113;133', panel: '8;47;59', panelHeader: '14;74;89',
+    }),
+  }),
+  forest: Object.freeze({
+    label: 'Forest',
+    description: 'Green and teal accents on natural dark surfaces',
+    colors: Object.freeze({
+      primary: '74;222;128', secondary: '110;231;183', highlight: '250;204;21',
+      warning: '251;191;36', info: '45;212;191', success: '134;239;172',
+      tertiary: '163;230;53', muted: '148;163;153', subdued: '71;85;79',
+      text: '220;252;231', error: '252;129;129', panel: '20;45;34', panelHeader: '30;64;46',
+    }),
+  }),
+  sunset: Object.freeze({
+    label: 'Sunset',
+    description: 'Warm orange accents on charcoal-brown surfaces',
+    colors: Object.freeze({
+      primary: '251;146;60', secondary: '253;186;116', highlight: '250;204;21',
+      warning: '251;191;36', info: '56;189;248', success: '74;222;128',
+      tertiary: '244;114;182', muted: '168;151;143', subdued: '87;72;64',
+      text: '255;237;213', error: '253;106;106', panel: '61;35;28', panelHeader: '87;48;38',
+    }),
+  }),
+  monochrome: Object.freeze({
+    label: 'Monochrome',
+    description: 'Low-distraction grayscale presentation',
+    colors: Object.freeze({
+      primary: '229;231;235', secondary: '209;213;219', highlight: '255;255;255',
+      warning: '209;213;219', info: '229;231;235', success: '229;231;235',
+      tertiary: '209;213;219', muted: '156;163;175', subdued: '107;114;128',
+      text: '245;245;245', error: '212;212;212', panel: '31;31;31', panelHeader: '64;64;64',
+    }),
+  }),
+});
+
+export const THEME_NAMES = Object.freeze(Object.keys(THEMES));
+
+function loadSavedTheme() {
+  const envTheme = process.env.KIKU_THEME?.toLowerCase();
+  if (envTheme && THEMES[envTheme]) return envTheme;
+  try {
+    const saved = JSON.parse(fs.readFileSync(THEME_STATE_FILE, 'utf8'));
+    if (typeof saved.theme === 'string' && THEMES[saved.theme]) return saved.theme;
+  } catch {}
+  return 'sakura';
+}
+
+let activeThemeName = loadSavedTheme();
+
+export function getThemeName() {
+  return activeThemeName;
+}
+
+export function getTheme(name = activeThemeName) {
+  return THEMES[name] || THEMES.sakura;
+}
+
+export function getThemePalette(name = activeThemeName) {
+  return getTheme(name).colors;
+}
+
+export function setTheme(name, { persist = true } = {}) {
+  const normalized = String(name || '').toLowerCase();
+  if (!THEMES[normalized]) return null;
+  activeThemeName = normalized;
+  process.env.KIKU_THEME = normalized;
+
+  if (persist) {
+    fs.mkdirSync(THEME_STATE_DIR, { recursive: true });
+    fs.writeFileSync(THEME_STATE_FILE, JSON.stringify({ theme: normalized }, null, 2) + '\n', 'utf8');
+  }
+  return getTheme(normalized);
+}
+
+export function ansiForeground(rgb) {
+  return `\x1b[38;2;${rgb}m`;
+}
+
+export function ansiBackground(rgb) {
+  return `\x1b[48;2;${rgb}m`;
+}
+
+export function inkColor(rgb) {
+  return `rgb(${rgb.replaceAll(';', ',')})`;
+}
+
+export function getInkPalette(name = activeThemeName) {
+  const palette = getThemePalette(name);
+  return Object.fromEntries(Object.entries(palette).map(([key, value]) => [key, inkColor(value)]));
+}

package/server/agent/toolRouter.js

@@ -0,0 +1,201 @@
+import { toolsSchema } from './tools.js';
+
+export const TOOL_CLUSTERS = Object.freeze({
+  CATALOG: [
+    'global_search',
+    'search_works',
+    'get_work_detail',
+    'get_character_roles',
+    'search_discography',
+    'get_profile_and_fandom',
+    'get_awards',
+    'get_brand_endorsements',
+    'get_career_timeline',
+    'get_social_feed',
+    'get_gallery_photos',
+    'get_quotes',
+  ],
+  MEDIA: [
+    'play_music',
+    'stop_music',
+    'pause_music',
+    'resume_music',
+    'next_track',
+    'prev_track',
+    'add_to_queue',
+    'remove_from_queue',
+    'clear_queue',
+    'get_queue',
+    'search_youtube',
+    'get_current_track',
+    'get_music_status',
+    'get_music_videos',
+    'get_media_links',
+  ],
+  RESEARCH: [
+    'web_search',
+    'web_extract',
+    'fetch_webpage',
+    'fetch_api',
+    'fetch_json',
+    'screenshot_webpage',
+    'verify_claim',
+    'run_subagent',
+    'run_deep_research',
+  ],
+  SYSTEM: [
+    'get_system_stats',
+    'manage_skill',
+    'read_local_file',
+    'list_directory',
+  ]
+});
+
+const DEFAULT_UNIVERSAL_TOOLS = Object.freeze([
+  'global_search',
+  'search_works',
+  'search_discography',
+  'get_profile_and_fandom',
+  'get_media_links',
+  'web_search',
+  'run_subagent',
+]);
+
+const INTENT_PATTERNS = [
+  {
+    type: 'MEDIA',
+    regex: /(เพลง|ร้อง|ฟัง|เอ็มวี|mv|music|song|play|queue|track|audio|sound|listen|stage|dance|concert|ost|bilibili|qqmusic|歌曲|单曲|首发|舞台|演唱)/i,
+    clusters: ['MEDIA'],
+    essentialTools: ['search_discography', 'get_music_videos', 'get_media_links', 'play_music', 'get_queue', 'search_youtube', 'get_current_track', 'get_music_status']
+  },
+  {
+    type: 'WORKS',
+    regex: /(ละคร|ซีรีส์|หนัง|ภาพยนตร์|บทบาท|ตัวละคร|ตอน|ช่อง|ฉาย|drama|series|movie|film|role|character|synopsis|actor|tv|works|broadcast|weibo|影视|剧|新剧|角色|饰演)/i,
+    clusters: ['CATALOG'],
+    essentialTools: ['search_works', 'get_work_detail', 'get_character_roles', 'global_search', 'get_career_timeline']
+  },
+  {
+    type: 'PROFILE_AWARDS_BRANDS',
+    regex: /(ประวัติ|วันเกิด|อายุ|บ้านเกิด|แฟนคลับ|สโลแกน|แบรนด์|พรีเซนเตอร์|โฆษณา|ทูต|รางวัล|เกียรติยศ|นิตยสาร|แฟชั่น|ปก|โซเชียล|ไอจี|profile|bio|birthday|fandom|miju|蜜橘|brand|ambassador|award|honor|endorsement|quote|fashion|magazine|cover|snh48|election|hall of fame|senbatsu|win|social|instagram|weibo|douyin|wechat|account|คำพูด|代言|彩妆|品牌|荣誉|奖项|微博之夜)/i,
+    clusters: ['CATALOG'],
+    essentialTools: ['get_profile_and_fandom', 'get_brand_endorsements', 'get_awards', 'get_quotes', 'get_career_timeline', 'get_gallery_photos', 'get_social_feed']
+  },
+  {
+    type: 'RESEARCH_VERIFY',
+    regex: /(ข่าว|ล่าสุด|จริงไหม|ข่าวลือ|สืบ|ตรวจสอบ|สัญญา|ศาล|star48|siba|news|update|latest|recent|rumor|verify|claim|dispute|legal|lawsuit|contract|truth)/i,
+    clusters: ['RESEARCH'],
+    essentialTools: ['verify_claim', 'web_search', 'web_extract', 'fetch_webpage', 'run_subagent', 'run_deep_research', 'global_search']
+  },
+  {
+    type: 'SYSTEM_SKILL',
+    regex: /(skill|learn|teach|tui|system|stats|status|memory|pref|file|dir|directory|สถิติ|จำนวน|ระบบ|สอน)/i,
+    clusters: ['SYSTEM'],
+    essentialTools: ['manage_skill', 'get_system_stats', 'read_local_file', 'list_directory']
+  }
+];
+
+export class DynamicToolRouter {
+  constructor(allTools = toolsSchema) {
+    this.allTools = Array.isArray(allTools) ? allTools : toolsSchema;
+    this.toolsByName = new Map(this.allTools.map(t => [t.function?.name, t]));
+  }
+
+  setTools(allTools) {
+    this.allTools = Array.isArray(allTools) ? allTools : toolsSchema;
+    this.toolsByName = new Map(this.allTools.map(t => [t.function?.name, t]));
+  }
+
+  detectIntents(prompt = '') {
+    const text = String(prompt || '');
+    const matched = [];
+
+    for (const pattern of INTENT_PATTERNS) {
+      if (pattern.regex.test(text)) {
+        matched.push(pattern);
+      }
+    }
+    return matched;
+  }
+
+  extractAtMentions(prompt = '') {
+    const text = String(prompt || '');
+    const matches = text.match(/@([a-zA-Z0-9_-]+)/g);
+    if (!matches) return [];
+    return matches.map(m => m.slice(1));
+  }
+
+  routeToolsForQuery(prompt = '', options = {}) {
+    if (options.showAll || options.disableRouting) {
+      return [...this.allTools];
+    }
+
+    const atMentions = this.extractAtMentions(prompt);
+    const pinnedTools = [];
+    const selectedNames = new Set();
+
+    for (const mention of atMentions) {
+      const mentionLower = mention.toLowerCase();
+      for (const tool of this.allTools) {
+        const name = tool.function?.name;
+        if (!name) continue;
+        const nameLower = name.toLowerCase();
+        if (
+          nameLower === mentionLower ||
+          nameLower === `mcp_${mentionLower}` ||
+          nameLower.endsWith(`_${mentionLower}`) ||
+          nameLower.startsWith(`mcp_${mentionLower}_`)
+        ) {
+          selectedNames.add(name);
+          if (!pinnedTools.some(t => t.function?.name === name)) {
+            pinnedTools.push(tool);
+          }
+        }
+      }
+    }
+
+    const matchedIntents = this.detectIntents(prompt);
+
+    if (matchedIntents.length === 0) {
+      for (const name of DEFAULT_UNIVERSAL_TOOLS) {
+        selectedNames.add(name);
+      }
+    } else {
+      for (const intent of matchedIntents) {
+        for (const toolName of intent.essentialTools) {
+          selectedNames.add(toolName);
+        }
+      }
+      selectedNames.add('global_search');
+      selectedNames.add('web_search');
+    }
+
+    if (options.includeExtra && Array.isArray(options.includeExtra)) {
+      for (const name of options.includeExtra) {
+        selectedNames.add(name);
+      }
+    }
+
+    const routed = [...pinnedTools];
+    const seenNames = new Set(pinnedTools.map(t => t.function?.name));
+
+    for (const tool of this.allTools) {
+      const name = tool.function?.name;
+      if (name && !seenNames.has(name) && (selectedNames.has(name) || (name.startsWith('mcp_') && options.includeMcp !== false))) {
+        seenNames.add(name);
+        routed.push(tool);
+      }
+    }
+
+    if (routed.length === 0) {
+      return this.allTools.slice(0, 8);
+    }
+
+    if (routed.length > 15 && !options.showAll) {
+      return routed.slice(0, 15);
+    }
+
+    return routed;
+  }
+}
+
+export const dynamicToolRouter = new DynamicToolRouter();