kiku-agent 1.0.4 → 1.0.6

package/server/agent/tools.js

@@ -34,7 +34,66 @@ export function buildYoukuSearchUrl(query) {
   return `https://so.youku.com/search_video/q_${encodeURIComponent(query)}`;
 }
 
-function getCleanDiscography() {
+export class ToolCache {
+  constructor(maxSize = 300, defaultTtlMs = 600_000) {
+    this.maxSize = maxSize;
+    this.defaultTtlMs = defaultTtlMs;
+    this.cache = new Map();
+  }
+
+  get(key) {
+    const entry = this.cache.get(key);
+    if (!entry) return null;
+    if (Date.now() > entry.expiresAt) {
+      this.cache.delete(key);
+      return null;
+    }
+    this.cache.delete(key);
+    this.cache.set(key, entry);
+    return entry.value;
+  }
+
+  set(key, value, ttlMs = this.defaultTtlMs) {
+    if (this.cache.size >= this.maxSize) {
+      const oldestKey = this.cache.keys().next().value;
+      if (oldestKey) this.cache.delete(oldestKey);
+    }
+    this.cache.set(key, {
+      value,
+      expiresAt: Date.now() + ttlMs,
+    });
+  }
+
+  clear() {
+    this.cache.clear();
+  }
+}
+
+export const globalToolCache = new ToolCache();
+
+const STATIC_CATALOG_TOOLS = new Set([
+  'get_profile_and_fandom',
+  'get_quotes',
+  'get_awards',
+  'get_timeline',
+  'get_brand_endorsements',
+  'get_gallery',
+  'get_music_videos',
+  'search_works',
+  'search_discography',
+  'get_work_detail',
+  'get_social_updates',
+  'get_media_links',
+]);
+
+const WEB_CACHE_TOOLS = new Set([
+  'web_search',
+  'web_extract',
+  'fetch_webpage',
+  'fetch_json',
+]);
+
+const CACHED_DISCOGRAPHY = (() => {
   const map = new Map();
   const allTracks = [
     ...(discographyData.latestReleases || []),
@@ -65,17 +124,25 @@ function getCleanDiscography() {
     }
   }
 
-  return Array.from(map.values()).sort((a, b) => (b.year || 0) - (a.year || 0));
+  return Object.freeze(Array.from(map.values()).sort((a, b) => (b.year || 0) - (a.year || 0)));
+})();
+
+function getCleanDiscography() {
+  return CACHED_DISCOGRAPHY;
 }
 
-function getAllMusicVideos() {
+const CACHED_MUSIC_VIDEOS = (() => {
   if (Array.isArray(musicVideoData)) {
-    return musicVideoData;
+    return Object.freeze(musicVideoData);
   }
-  return [
+  return Object.freeze([
     ...(musicVideoData.officialMvs || []).map(m => ({ ...m, category: 'Official MV' })),
     ...(musicVideoData.stagePerformances || []).map(m => ({ ...m, category: 'Stage Performance' }))
-  ];
+  ]);
+})();
+
+function getAllMusicVideos() {
+  return CACHED_MUSIC_VIDEOS;
 }
 
 export const toolsSchema = [
@@ -620,6 +687,20 @@ export const toolsSchema = [
       }
     }
   },
+  {
+    type: "function",
+    function: {
+      name: "suggest_topics",
+      description: "Generate dynamic, intriguing research questions and exploration topics about Ju Jingyi across dramas, music, fashion, milestones, and trivia.",
+      parameters: {
+        type: "object",
+        properties: {
+          topic: { type: "string", description: "Optional topic or category focus (e.g. 'drama', 'music', 'fashion', '2026')." },
+          limit: { type: "number", description: "Number of suggestions to generate (default: 5)." }
+        }
+      }
+    }
+  },
   {
     type: "function",
     function: {
@@ -1483,6 +1564,19 @@ export const toolHandlers = {
     return { status: "success", claim, answer: result.answer, sources: result.sources, reports: result.reports };
   },
 
+  suggest_topics: async ({ topic, limit }, options = {}) => {
+    if (options?.agent?.generateSuggestions) {
+      const suggestions = await options.agent.generateSuggestions({ topic, limit, signal: options.signal });
+      return { status: "success", topic, suggestions };
+    }
+    const { SuggestionEngine } = await import('./suggestions.js');
+    const { LMStudioClient } = await import('./llmClient.js');
+    const client = options?.client || new LMStudioClient();
+    const engine = new SuggestionEngine({ client });
+    const suggestions = await engine.generateSuggestions({ topic, limit, signal: options?.signal });
+    return { status: "success", topic, suggestions };
+  },
+
   manage_skill: async ({ action, instruction, skillId }, options = {}) => {
     let memory = options?.agent?.memory || options?.memory;
     if (!memory) {
@@ -1628,7 +1722,9 @@ export function normalizeToolResult(toolName, result) {
   if (normalized.error && !normalized.message) normalized.message = String(normalized.error);
   if (normalized.status === 'success' && (normalized.count === 0 || normalized.returnedCount === 0)) {
     normalized.status = 'not_found';
-    if (!normalized.message) normalized.message = 'No matching data found.';
+    if (!normalized.message) {
+      normalized.message = 'No matching data found in local catalog. Hint: You can use web_search or web_extract to look up online information.';
+    }
   }
   return normalized;
 }
@@ -1652,6 +1748,15 @@ export async function executeTool(toolName, args = {}, options = {}) {
       };
     }
 
+    const isStatic = STATIC_CATALOG_TOOLS.has(toolName);
+    const isWeb = WEB_CACHE_TOOLS.has(toolName);
+    const cacheKey = (isStatic || isWeb) ? `${toolName}:${JSON.stringify(prepared.args)}` : null;
+
+    if (cacheKey && !options.noCache) {
+      const cached = globalToolCache.get(cacheKey);
+      if (cached) return cached;
+    }
+
     const configuredTimeout = Number(options.timeoutMs ?? process.env.KIKU_TOOL_TIMEOUT_MS ?? 45_000);
     const timeoutMs = Number.isFinite(configuredTimeout) && configuredTimeout > 0 ? configuredTimeout : 45_000;
     let timeoutId;
@@ -1669,6 +1774,12 @@ export async function executeTool(toolName, args = {}, options = {}) {
     if (prepared.warnings.length > 0) {
       normalized.validationWarnings = prepared.warnings;
     }
+
+    if (cacheKey && normalized.status !== 'error') {
+      const ttl = isStatic ? 3600_000 : 900_000;
+      globalToolCache.set(cacheKey, normalized, ttl);
+    }
+
     return normalized;
   } catch (err) {
     return {