kiku-agent 1.0.0 → 1.0.3

package/server/agent/agentCore.js

@@ -1,782 +1,892 @@
-import { config } from './config.js';
-import { toolsSchema, executeTool, prepareToolArguments } from './tools.js';
-import { LMStudioClient } from './llmClient.js';
-import { AgentSession } from './session.js';
-import { AgentMemory } from './memory.js';
-import { AgentStateMachine, AgentStates } from './agentState.js';
-import { eventBus, Events } from './eventBus.js';
-import { sanitizeQuery, isLikelySafe } from './sanitizer.js';
-import { createAbortError, isAbortError } from './errors.js';
-
-export function detectDominantLanguage(text) {
-  if (!text || typeof text !== 'string') return 'th';
-
-  const thaiMatches = text.match(/[\u0E00-\u0E7F]/g) || [];
-  const thaiCount = thaiMatches.length;
-
-  const chineseMatches = text.match(/[\u4E00-\u9FA5]/g) || [];
-  const chineseCount = chineseMatches.length;
-
-  const englishMatches = text.match(/[a-zA-Z]+/g) || [];
-  const englishCount = englishMatches.join('').length;
-
-  if (thaiCount >= chineseCount && thaiCount >= englishCount && thaiCount > 0) {
-    return 'th';
-  } else if (chineseCount > thaiCount && chineseCount >= englishCount && chineseCount > 0) {
-    return 'zh';
-  } else if (englishCount > 0) {
-    return 'en';
-  }
-
-  return 'th';
-}
-
-function stableToolSignature(toolName, args) {
-  const sortValue = (value) => {
-    if (Array.isArray(value)) return value.map(sortValue);
-    if (value && typeof value === 'object') {
-      return Object.fromEntries(Object.keys(value).sort().map(key => [key, sortValue(value[key])]));
-    }
-    return value;
-  };
-  return `${toolName}:${JSON.stringify(sortValue(args || {}))}`;
-}
-
-export class KikuAgent {
-  constructor(options = {}) {
-    this.client = new LMStudioClient({
-      baseUrl: options.baseUrl || config.lmStudio.baseUrl,
-      model: options.model || config.lmStudio.model
-    });
-    this.maxIterations = options.maxIterations || config.agent.maxIterations || 5;
-    this.slidingWindowSize = options.slidingWindowSize || config.agent.slidingWindowSize || 8;
-    this.session = new AgentSession({ thinkingMode: options.thinkingMode || 'hide' });
-    this.memory = new AgentMemory();
-    this.systemPrompt = this.buildSystemPrompt();
-    this.initHistory();
-    this.restoredMessageCount = 0;
-
-    this.stateMachine = new AgentStateMachine();
-
-    this.stateMachine.on('transition', ({ from, to }) => {
-      eventBus.emit(Events.AGENT_STATE_CHANGE, {
-        state: to,
-        label: this.stateMachine.label,
-        elapsed: this.stateMachine.elapsedMs,
-        queryId: this.stateMachine.queryId,
-      });
-    });
-
-    this._ready = this._init();
-  }
-
-  async _init() {
-    await this.memory.load();
-    await this.loadSession();
-    if (!this.conversationHistory || this.conversationHistory.length === 0) {
-      this.initHistory();
-    }
-  }
-
-  async ready() {
-    return this._ready;
-  }
-
-  initHistory() {
-    this.conversationHistory = [
-      {
-        role: 'system',
-        content: this.systemPrompt
-      }
-    ];
-  }
-
-  clearMemory() {
-    if (this.session.messages.length > 0) {
-      this.memory.archiveSession(this.session.toSnapshot()).catch(() => {});
-      this.memory.consolidateSession(this.session.toSnapshot()).catch(() => {});
-    }
-    this.initHistory();
-    this.session.reset();
-    this.session.clearDisk().catch(() => {});
-  }
-
-  async loadSession() {
-    const loaded = await this.session.loadFromDisk();
-    if (loaded && this.session.messages.length > 0) {
-      this.conversationHistory = [
-        {
-          role: 'system',
-          content: this.systemPrompt
-        }
-      ];
-      
-      for (const msg of this.session.messages) {
-        if (msg.role === 'user') {
-          const textPart = msg.parts.find(p => p.type === 'text');
-          if (textPart) {
-            this.conversationHistory.push({
-              role: 'user',
-              content: textPart.content
-            });
-          }
-        } else if (msg.role === 'assistant') {
-          const textPart = msg.parts.find(p => p.type === 'text');
-          if (textPart) {
-            this.conversationHistory.push({
-              role: 'assistant',
-              content: textPart.content
-            });
-          }
-        }
-      }
-      this.restoredMessageCount = this.session.messages.length;
-    }
-  }
-
-  getEnrichedSystemPrompt() {
-    const memoryContext = this.memory.buildMemoryContext();
-    return this.systemPrompt + memoryContext;
-  }
-
-  async saveSession() {
-    await this.session.saveToDisk();
-    eventBus.emit(Events.SESSION_SAVED, { sessionId: this.session.id });
-  }
-
-  applySlidingWindow() {
-    if (this.conversationHistory.length > this.slidingWindowSize + 1) {
-      const systemMessage = this.conversationHistory[0];
-      const recentMessages = this.conversationHistory.slice(-this.slidingWindowSize);
-      this.conversationHistory = [systemMessage, ...recentMessages];
-    }
-  }
-
-  buildSystemPrompt() {
-    const toolsSummary = toolsSchema.map(t => {
-      const f = t.function;
-      const paramKeys = Object.keys(f.parameters?.properties || {}).join(', ');
-      return `- ${f.name}(${paramKeys}): ${f.description}`;
-    }).join('\n');
-
-    return `${config.agent.systemPersona}
-
-=== AVAILABLE KNOWLEDGE & LIVE TOOLS ===
-${toolsSummary}
-
-=== DYNAMIC MULTILINGUAL & RESPONSE RULES (CRITICAL) ===
-
-0. NO SYSTEM LEAK (ABSOLUTE RULE):
-   - NEVER include "Thinking Process", "Tool Calling", "Action:", "Action Input:", function names, parameter schemas, JSON tool call blocks, or any internal reasoning in your response text.
-   - NEVER include advertisements, sponsor mentions, or promotional content.
-   - Your response must be CLEAN user-facing content ONLY.
-   - Think silently using tools, then present ONLY the final polished answer.
-
-1. LANGUAGE MIRRORING:
-   - THAI ➔ polite Thai (ใช้ ค่ะ/นะคะ เสมอ)
-   - CHINESE ➔ fluent, polite Chinese (简体/繁体中文)
-   - ENGLISH ➔ polished, professional English
-   - MIXED ➔ Match dominant language, keep bilingual titles
-
-2. REAL YOUTUBE & MEDIA LINKS:
-   - Direct Watch: https://www.youtube.com/watch?v=VIDEO_ID
-   - Search: https://www.youtube.com/results?search_query=QUERY
-   - Also Bilibili, QQ Music, Youku when applicable
-   - NEVER use fake placeholder IDs
-
-3. FACT ACCURACY & NO DUPLICATION:
-   - Always use tools to verify facts before answering. Do NOT hallucinate.
-   - Current year is 2026. Prioritize 2026/2025 projects.
-   - Never repeat the same statement. Sort newest first.
-
-4. RESPONSE STYLE:
-   - NEVER start with template greetings — dive DIRECTLY into the answer.
-   - Format with Markdown headers (###), bold (**text**), bullet points (•).
-   - Emoji are optional: use at most one contextually relevant emoji per response and never repeat them across headings or bullets.
-   - Match answer depth to what the user asked.
-
-5. SMART TOOL SELECTION:
-   - Videos/stages ➔ get_music_videos or get_media_links
-   - Songs/OSTs ➔ search_discography or get_media_links
-   - Dramas ➔ search_works or get_work_detail
-   - News/external facts ➔ web_search or web_extract
-   - Deep multi-source research ➔ web_extract (search + auto-fetch)
-   - Read webpage content ➔ fetch_webpage
-   - Fetch API/JSON ➔ fetch_json
-   - Local files ➔ read_local_file
-   - Directory listing ➔ list_directory
-   - Screenshot ➔ screenshot_webpage
-
-6. AGGRESSIVE WEB SEARCH FALLBACK:
-   - If local tools return insufficient results, IMMEDIATELY call web_search or web_extract.
-   - Combine local data WITH web search for complete, accurate answers.
-   - Use web_extract for comprehensive research on any topic.
-
-7. LEGAL & CAREER KNOWLEDGE:
-   - When discussing controversies, be specific and accurate:
-     • Contract dispute: Star48/Siba Media supplementary agreement. JJY claims main contract ended June 2024, alleges forged signatures. Star48 claims until 2033.
-     • Tax allegation: Tip-off report filed by Star48 side; JJY denies and confirms tax compliance.
-     • Portrait/copyright cases: JJY has sued beauty clinics and brands for unauthorized image use (肖像权纠纷).
-   - Use web_search or web_extract for latest verified information on legal matters.`;
-  }
-
-  parseTextToolCalls(content) {
-    if (!content) return [];
-    const calls = [];
-
-    const jsonBlockRegex = /```(?:json)?\s*(\{[\s\S]*?\})\s*```/g;
-    let match;
-    while ((match = jsonBlockRegex.exec(content)) !== null) {
-      try {
-        const parsed = JSON.parse(match[1]);
-        const toolName = parsed.action || parsed.tool || parsed.name || parsed.function;
-        const toolArgs = parsed.action_input || parsed.parameters || parsed.arguments || parsed.args || {};
-        if (toolName && toolsSchema.some(t => t.function.name === toolName)) {
-          calls.push({
-            id: `call_txt_${Date.now()}_${calls.length}`,
-            function: {
-              name: toolName,
-              arguments: typeof toolArgs === 'string' ? toolArgs : JSON.stringify(toolArgs)
-            }
-          });
-        }
-      } catch {
-      }
-    }
-
-    const actionRegex = /Action:\s*([a-zA-Z0-9_]+)\s*\nAction Input:\s*(\{[\s\S]*?\}|[^\n]+)/g;
-    while ((match = actionRegex.exec(content)) !== null) {
-      const toolName = match[1].trim();
-      let rawInput = match[2].trim();
-      if (toolsSchema.some(t => t.function.name === toolName)) {
-        let args = {};
-        try {
-          args = JSON.parse(rawInput);
-        } catch {
-          args = { query: rawInput };
-        }
-        calls.push({
-          id: `call_act_${Date.now()}_${calls.length}`,
-          function: {
-            name: toolName,
-            arguments: JSON.stringify(args)
-          }
-        });
-      }
-    }
-
-    return calls;
-  }
-
-  async ask(userPrompt, callbacks = {}) {
-    await this._ready;
-
-    const { sanitized: cleanPrompt, warnings: sanitizationWarnings } = sanitizeQuery(userPrompt);
-    if (sanitizationWarnings.length > 0) {
-      eventBus.emit(Events.AGENT_QUERY_ERROR, {
-        type: 'sanitization',
-        warnings: sanitizationWarnings,
-        original: userPrompt,
-      });
-    }
-    const promptForLLM = cleanPrompt || userPrompt;
-
-    const topics = this._extractQuickTopics(promptForLLM);
-    for (const topic of topics) {
-      this.memory.trackTopicInterest(topic);
-    }
-    
-    const {
-      onStepStart = () => {},
-      onReasoningStart = () => {},
-      onReasoningDelta = () => {},
-      onReasoningEnd = () => {},
-      onToolStart = () => {},
-      onToolResult = () => {},
-      onTokenStart = () => {},
-      onTokenDelta = () => {},
-      onFinalAnswer = () => {},
-      onError = () => {},
-      signal = null
-    } = callbacks;
-
-    const queryStartTime = Date.now();
-    const queryId = `q_${Date.now()}_${Math.random().toString(36).slice(2, 6)}`;
-    const dominantLang = detectDominantLanguage(promptForLLM);
-
-    this.stateMachine.startQuery(queryId);
-    eventBus.emit(Events.AGENT_QUERY_START, { queryId, prompt: promptForLLM });
-
-    if (!this.conversationHistory || this.conversationHistory.length === 0 || this.conversationHistory[0]?.role !== 'system') {
-      this.conversationHistory = [
-        { role: 'system', content: this.systemPrompt },
-        ...(this.conversationHistory?.filter(m => m.role !== 'system') || [])
-      ];
-    }
-
-    this.session.createUserMessage(promptForLLM);
-    const userMessage = { role: 'user', content: promptForLLM };
-    this.conversationHistory.push(userMessage);
-
-    const memoryContext = this.memory.buildMemoryContext();
-    if (memoryContext && this.conversationHistory[0]?.content && !this.conversationHistory[0].content.includes('LONG-TERM MEMORY CONTEXT')) {
-      this.conversationHistory[0] = {
-        role: 'system',
-        content: this.systemPrompt + memoryContext
-      };
-    }
-
-    let langInstruction;
-    if (dominantLang === 'th') {
-      langInstruction = '[SYSTEM NOTE: User language is Thai. Reply in Thai only.]\n';
-    } else if (dominantLang === 'zh') {
-      langInstruction = '[SYSTEM NOTE: User language is Chinese. Reply in Chinese only.]\n';
-    } else {
-      langInstruction = '[SYSTEM NOTE: User language is English. Reply in English only.]\n';
-    }
-    this.conversationHistory.push({ role: 'system', content: langInstruction });
-
-    this.applySlidingWindow();
-
-    const currentAssistantSessionMsg = this.session.createAssistantMessage();
-
-    let currentIteration = 0;
-    const toolResultCache = new Map();
-
-    while (currentIteration < this.maxIterations) {
-      if (signal?.aborted) {
-        this.stateMachine.reset();
-        throw createAbortError();
-      }
-
-      currentIteration++;
-      onStepStart(currentIteration, this.maxIterations);
-
-      let assistantMessage = null;
-      let isReasoning = false;
-      let isTokenEmitting = false;
-
-      try {
-        assistantMessage = await this.client.chat({
-          messages: this.conversationHistory,
-          tools: toolsSchema,
-          temperature: 0.3,
-          stream: true,
-          signal,
-          onThought: (thoughtChunk) => {
-            if (!isReasoning) {
-              isReasoning = true;
-              this.stateMachine.transition(AgentStates.STREAMING_THOUGHTS);
-              onReasoningStart();
-              eventBus.emit(Events.AGENT_REASONING_START, { queryId });
-            }
-            this.session.addReasoningPart(currentAssistantSessionMsg, thoughtChunk);
-            onReasoningDelta(thoughtChunk);
-            eventBus.emit(Events.AGENT_REASONING_DELTA, { queryId, chunk: thoughtChunk });
-          },
-          onToken: (tokenChunk) => {
-            if (isReasoning) {
-              isReasoning = false;
-              this.session.finalizeReasoningPart(currentAssistantSessionMsg);
-              this.stateMachine.transition(AgentStates.STREAMING_ANSWER);
-              onReasoningEnd();
-              eventBus.emit(Events.AGENT_REASONING_END, { queryId });
-            }
-            if (!isTokenEmitting) {
-              isTokenEmitting = true;
-              if (this.stateMachine.state !== AgentStates.STREAMING_ANSWER) {
-                this.stateMachine.transition(AgentStates.STREAMING_ANSWER);
-              }
-              onTokenStart();
-              eventBus.emit(Events.AGENT_TOKEN_START, { queryId });
-            }
-            this.session.addTextPart(currentAssistantSessionMsg, tokenChunk);
-            onTokenDelta(tokenChunk);
-            eventBus.emit(Events.AGENT_TOKEN_DELTA, { queryId, chunk: tokenChunk });
-          }
-        });
-
-        if (isReasoning) {
-          this.session.finalizeReasoningPart(currentAssistantSessionMsg);
-          if (this.stateMachine.state === AgentStates.STREAMING_THOUGHTS) {
-            this.stateMachine.transition(AgentStates.STREAMING_ANSWER);
-          }
-          onReasoningEnd();
-        }
-      } catch (err) {
-        this.stateMachine.reset();
-        if (!isAbortError(err)) {
-          onError(err);
-          eventBus.emit(Events.AGENT_QUERY_ERROR, { queryId, error: err.message });
-        }
-        throw err;
-      }
-
-      if (!assistantMessage) {
-        this.stateMachine.reset();
-        throw new Error("No response received from LM Studio.");
-      }
-
-      let toolCalls = assistantMessage.tool_calls || [];
-      if (toolCalls.length === 0 && assistantMessage.content) {
-        toolCalls = this.parseTextToolCalls(assistantMessage.content);
-      }
-
-      this.conversationHistory.push(assistantMessage);
-
-      if (!toolCalls || toolCalls.length === 0) {
-        const totalDurationMs = Date.now() - queryStartTime;
-        this.session.stats.totalDurationMs += totalDurationMs;
-
-        const finalAnswer = assistantMessage.content || (
-          dominantLang === 'zh' ? "抱歉，无法处理您的回答。" :
-          dominantLang === 'en' ? "Sorry, unable to process a response." :
-          "ขออภัยค่ะ ไม่สามารถประมวลผลคำตอบได้"
-        );
-        onFinalAnswer(finalAnswer, { durationMs: totalDurationMs, language: dominantLang });
-        this._extractFactsFromFinalAnswer(finalAnswer);
-
-        this.stateMachine.transition(AgentStates.COMPLETE);
-        eventBus.emit(Events.AGENT_FINAL_ANSWER, { queryId, answer: finalAnswer, durationMs: totalDurationMs });
-        this.stateMachine.reset();
-
-        return finalAnswer;
-      }
-
-      for (const [toolCallIndex, tc] of toolCalls.entries()) {
-        if (signal?.aborted) {
-          this.stateMachine.reset();
-          throw createAbortError();
-        }
-
-        const toolName = tc?.function?.name || '';
-        const toolCallId = tc?.id || `call_${queryId}_${currentIteration}_${toolCallIndex}`;
-        const prepared = prepareToolArguments(toolName, tc?.function?.arguments || {});
-        if (!prepared.safe) {
-          const rejectedResult = {
-            status: 'error',
-            message: `Input rejected: ${prepared.error}`,
-            validationWarnings: prepared.warnings,
-          };
-          eventBus.emit(Events.AGENT_TOOL_ERROR, {
-            queryId, toolName, args: tc.function.arguments,
-            error: prepared.error,
-          });
-          onToolStart(toolName, {});
-          onToolResult(toolName, rejectedResult, 0);
-          const toolMessage = {
-            role: 'tool',
-            tool_call_id: toolCallId,
-            name: toolName,
-            content: JSON.stringify(rejectedResult)
-          };
-          this.conversationHistory.push(toolMessage);
-          continue;
-        }
-        if (prepared.warnings.length > 0) {
-          eventBus.emit(Events.AGENT_TOOL_ERROR, {
-            queryId, toolName, warnings: prepared.warnings,
-          });
-        }
-
-        const finalArgs = prepared.args;
-
-        this.stateMachine.transition(AgentStates.FETCHING_TOOL, { toolName });
-
-        const toolPart = this.session.addToolPart(currentAssistantSessionMsg, toolName, finalArgs);
-        onToolStart(toolName, finalArgs);
-        eventBus.emit(Events.AGENT_TOOL_START, { queryId, toolName, args: finalArgs });
-
-        const toolStartTime = Date.now();
-        const signature = stableToolSignature(toolName, finalArgs);
-        const cachedResult = toolResultCache.get(signature);
-        const result = cachedResult
-          ? { ...cachedResult, cache: { hit: true, reason: 'duplicate_call_in_query' } }
-          : await executeTool(toolName, finalArgs);
-        if (!cachedResult) toolResultCache.set(signature, result);
-        const toolDurationMs = Date.now() - toolStartTime;
-
-        this.session.finalizeToolPart(currentAssistantSessionMsg, toolPart.id, result);
-        onToolResult(toolName, result, toolDurationMs);
-        eventBus.emit(Events.AGENT_TOOL_RESULT, { queryId, toolName, result, durationMs: toolDurationMs });
-
-        this._extractFactsFromToolResult(toolName, finalArgs, result);
-
-        const toolMessage = {
-          role: 'tool',
-          tool_call_id: toolCallId,
-          name: toolName,
-          content: JSON.stringify({
-            authoritative: true,
-            result,
-            instruction: result.status === 'not_found'
-              ? 'No matching data was found. Do not invent records; say that the catalog has no match or use an appropriate fallback tool.'
-              : result.status === 'error'
-                ? 'The tool failed. Do not claim that the requested facts were verified.'
-                : 'Use only facts present in this result when answering.'
-          })
-        };
-
-        this.conversationHistory.push(toolMessage);
-      }
-    }
-
-    const langPrompt = dominantLang === 'zh' 
-      ? "请根据上述工具查询结果，直接回答（不要开头问候语），用流畅优雅的中文，先1-2句概述，再列要点。包含具体的 YouTube 真实链接（如 Direct Watch: https://www.youtube.com/watch?v=... หรือ Search: https://www.youtube.com/results?search_query=...）及事实数据。"
-      : dominantLang === 'en'
-      ? "Based on the tool results above, answer directly (no opening greeting or template phrase). Give a 1-2 sentence summary, then concise bullet points in polished English. Include verified YouTube direct and search links (https://www.youtube.com/results?search_query=...)."
-      : "สรุปคำตอบจากผลลัพธ์เครื่องมือข้างต้น โดยไม่ต้องทักทาย ตอบตรงประเด็น 1-2 ประโยค แล้วตามด้วย bullet points สั้นๆ เป็นภาษาไทย พร้อมแสดงลิงก์ YouTube จริง (Direct Watch Link และ https://www.youtube.com/results?search_query=...) และข้อมูลผลงานที่ถูกต้องแม่นยำ";
-
-    this.conversationHistory.push({ role: 'user', content: langPrompt });
-
-    if (this.stateMachine.state !== AgentStates.STREAMING_ANSWER) {
-      this.stateMachine.transition(AgentStates.STREAMING_ANSWER);
-    }
-
-    let isReasoning = false;
-    let synthTokenStarted = false;
-    const finalRes = await this.client.chat({
-      messages: this.conversationHistory,
-      temperature: 0.3,
-      stream: true,
-      signal,
-      onThought: (thoughtChunk) => {
-        if (!isReasoning) {
-          isReasoning = true;
-          onReasoningStart();
-        }
-        this.session.addReasoningPart(currentAssistantSessionMsg, thoughtChunk);
-        onReasoningDelta(thoughtChunk);
-      },
-      onToken: (tokenChunk) => {
-        if (isReasoning) {
-          isReasoning = false;
-          this.session.finalizeReasoningPart(currentAssistantSessionMsg);
-          onReasoningEnd();
-        }
-        if (!synthTokenStarted) {
-          synthTokenStarted = true;
-          onTokenStart();
-        }
-        this.session.addTextPart(currentAssistantSessionMsg, tokenChunk);
-        onTokenDelta(tokenChunk);
-      }
-    });
-
-    const totalDurationMs = Date.now() - queryStartTime;
-    this.session.stats.totalDurationMs += totalDurationMs;
-
-    const defaultDone = dominantLang === 'zh' ? "查询完毕" : dominantLang === 'en' ? "Search complete." : "เสร็จสิ้นการค้นหาข้อมูล";
-    const output = finalRes?.content || defaultDone;
-    onFinalAnswer(output, { durationMs: totalDurationMs, language: dominantLang });
-    
-    this._extractFactsFromFinalAnswer(output);
-
-    this.stateMachine.transition(AgentStates.COMPLETE);
-    eventBus.emit(Events.AGENT_FINAL_ANSWER, { queryId, answer: output, durationMs: totalDurationMs });
-    eventBus.emit(Events.AGENT_QUERY_END, { queryId, durationMs: totalDurationMs });
-    this.stateMachine.reset();
-    
-    this.saveSession().catch(err => console.error('Session save failed:', err));
-    
-    return output;
-  }
-
-  _extractFactsFromToolResult(toolName, toolArgs, result) {
-    if (!result || result.status === 'error' || result.status === 'not_found') return;
-
-    const facts = [];
-
-    switch (toolName) {
-      case 'search_works': {
-        const works = result.works || [];
-        for (const w of works.slice(0, 3)) {
-          if (w.titleCn && w.role) {
-            facts.push({
-              content: `${w.titleCn} (${w.titleEn || w.titleTh || ''}) — Role: ${w.role} (${w.year || '?'})`,
-              category: 'drama',
-              tags: [w.titleEn, w.titleCn, String(w.year)].filter(Boolean)
-            });
-          }
-        }
-        break;
-      }
-
-      case 'get_work_detail': {
-        const w = result.work;
-        if (w && w.titleCn) {
-          facts.push({
-            content: `${w.titleCn} (${w.titleEn || w.titleTh || ''}) — ${w.role || 'Actor'} | ${w.platform || ''} | ${w.year || ''}`,
-            category: 'drama',
-            tags: [w.titleEn, w.titleCn, w.platform, String(w.year)].filter(Boolean)
-          });
-          if (w.synopsis) {
-            facts.push({
-              content: `Synopsis of ${w.titleCn}: ${w.synopsis.slice(0, 150)}`,
-              category: 'drama',
-              tags: [w.titleCn, 'synopsis']
-            });
-          }
-        }
-        break;
-      }
-
-      case 'search_discography': {
-        const songs = result.songs || [];
-        for (const s of songs.slice(0, 3)) {
-          if (s.title && s.year) {
-            facts.push({
-              content: `Song: ${s.title} (${s.year}) — Type: ${s.type || 'Single'}`,
-              category: 'music',
-              tags: [s.titleEn, s.titleCn, String(s.year)].filter(Boolean)
-            });
-          }
-        }
-        break;
-      }
-
-      case 'get_music_videos': {
-        const vids = result.videos || [];
-        for (const v of vids.slice(0, 3)) {
-          if (v.title && v.year) {
-            facts.push({
-              content: `MV/Stage: ${v.title} (${v.year}) — ${v.category || 'MV'}`,
-              category: 'video',
-              tags: [v.title, String(v.year), v.category].filter(Boolean)
-            });
-          }
-        }
-        break;
-      }
-
-      case 'get_brand_endorsements': {
-        const brands = result.brands || [];
-        for (const b of brands.slice(0, 3)) {
-          if (b.brand) {
-            facts.push({
-              content: `Brand: ${b.brand} — ${b.title || 'Ambassador'} (${b.year || ''})${b.isGlobal ? ' [Global]' : ''}`,
-              category: 'brand',
-              tags: [b.brand, String(b.year)].filter(Boolean)
-            });
-          }
-        }
-        break;
-      }
-
-      case 'get_awards': {
-        const awards = result.awards || [];
-        for (const a of awards.slice(0, 3)) {
-          if (a.title && a.year) {
-            facts.push({
-              content: `Award: ${a.title} (${a.year}) — ${a.ceremony || ''}`,
-              category: 'awards',
-              tags: [a.title, String(a.year), a.ceremony].filter(Boolean)
-            });
-          }
-        }
-        break;
-      }
-
-      case 'get_profile_and_fandom': {
-        const p = result.profile;
-        if (p) {
-          if (p.birthDate) facts.push({ content: `Birth date: ${p.birthDate}`, category: 'profile', tags: ['birthday'] });
-          if (p.fandomName) facts.push({ content: `Fandom: ${p.fandomName}`, category: 'profile', tags: ['fandom'] });
-          if (p.slogan) facts.push({ content: `Slogan: ${p.slogan}`, category: 'profile', tags: ['slogan'] });
-        }
-        break;
-      }
-
-      case 'get_career_timeline': {
-        const milestones = result.milestones || [];
-        for (const m of milestones.slice(0, 2)) {
-          if (m.title && m.period) {
-            facts.push({
-              content: `Career milestone: ${m.title} (${m.period})`,
-              category: 'timeline',
-              tags: [m.title, m.period].filter(Boolean)
-            });
-          }
-        }
-        break;
-      }
-
-      case 'web_search': {
-        const items = result.results || [];
-        for (const item of items.slice(0, 2)) {
-          if (item.title && item.snippet) {
-            facts.push({
-              content: `Web: ${item.title.slice(0, 100)} — ${item.snippet.slice(0, 150)}`,
-              category: 'news',
-              tags: ['web-search']
-            });
-          }
-        }
-        break;
-      }
-    }
-
-    for (const fact of facts) {
-      const isDuplicate = this.memory.facts.some(
-        existing => existing.content === fact.content
-      );
-      if (!isDuplicate) {
-        this.memory.addFact(fact.content, {
-          category: fact.category,
-          source: `tool:${toolName}`,
-          tags: fact.tags
-        });
-        eventBus.emit(Events.MEMORY_FACT_ADDED, { fact: fact.content, category: fact.category });
-      }
-    }
-  }
-
-  _extractFactsFromFinalAnswer(answer) {
-    if (!answer || answer.length < 20) return;
-
-    const ytRegex = /https?:\/\/www\.youtube\.com\/watch\?v=([a-zA-Z0-9_-]+)/g;
-    let match;
-    const ytIds = new Set();
-    while ((match = ytRegex.exec(answer)) !== null) {
-      ytIds.add(match[1]);
-    }
-    if (ytIds.size > 0) {
-      this.memory.addFact(
-        `YouTube video IDs discussed: ${[...ytIds].join(', ')}`,
-        { category: 'links', source: 'final-answer', tags: ['youtube'] }
-      );
-    }
-  }
-
-  _extractQuickTopics(text) {
-    const topicMap = {
-      'drama': /drama|series|ละคร|剧|电视|tv|film|movie/i,
-      'music': /song|music|mv|single|ost|เพลง|歌|album/i,
-      'brand': /brand|ambassador|endorsement|แบรนด์|品牌/i,
-      'awards': /award|honor|ceremony|รางวัล|奖/i,
-      'profile': /profile|bio|birthday|ประวัติ|资料/i,
-      'news': /news|latest|recent|update|ข่าว|最新/i,
-      'video': /video|stage|performance|live|คลิป|视频|舞台/i,
-      'social': /weibo|douyin|instagram|social|微博/i,
-      'gallery': /photo|picture|gallery|fashion|ภาพ|照片/i
-    };
-    const found = [];
-    for (const [topic, regex] of Object.entries(topicMap)) {
-      if (regex.test(text)) found.push(topic);
-    }
-    return found;
-  }
-
-  async endSession() {
-    await this._ready;
-    const snapshot = this.session.toSnapshot();
-    if (snapshot.messages.length > 0) {
-      await this.memory.archiveSession(snapshot);
-      await this.memory.consolidateSession(snapshot);
-      eventBus.emit(Events.SESSION_ARCHIVED, { sessionId: snapshot.id });
-    }
-    await this.session.clearDisk();
-    this.session.reset();
-    this.initHistory();
-  }
-}
+import { config } from './config.js';
+import { toolsSchema, executeTool, prepareToolArguments } from './tools.js';
+import { LLMClient, LMStudioClient } from './llmClient.js';
+import { AgentSession } from './session.js';
+import { AgentMemory } from './memory.js';
+import { AgentStateMachine, AgentStates } from './agentState.js';
+import { eventBus, Events } from './eventBus.js';
+import { sanitizeQuery, isLikelySafe } from './sanitizer.js';
+import { createAbortError, isAbortError } from './errors.js';
+import { ResearchSubagent } from './subagent.js';
+import { ResearchOrchestrator } from './research.js';
+import { McpClientManager } from './mcpClient.js';
+import { DynamicToolRouter } from './toolRouter.js';
+import { contextCompactor } from './contextCompactor.js';
+
+export function detectDominantLanguage(text) {
+  if (!text || typeof text !== 'string') return 'th';
+
+  const thaiMatches = text.match(/[\u0E00-\u0E7F]/g) || [];
+  const thaiCount = thaiMatches.length;
+
+  const chineseMatches = text.match(/[\u4E00-\u9FA5]/g) || [];
+  const chineseCount = chineseMatches.length;
+
+  const englishMatches = text.match(/[a-zA-Z]+/g) || [];
+  const englishCount = englishMatches.join('').length;
+
+  if (thaiCount >= chineseCount && thaiCount >= englishCount && thaiCount > 0) {
+    return 'th';
+  } else if (chineseCount > thaiCount && chineseCount >= englishCount && chineseCount > 0) {
+    return 'zh';
+  } else if (englishCount > 0) {
+    return 'en';
+  }
+
+  return 'th';
+}
+
+function stableToolSignature(toolName, args) {
+  const sortValue = (value) => {
+    if (Array.isArray(value)) return value.map(sortValue);
+    if (value && typeof value === 'object') {
+      return Object.fromEntries(Object.keys(value).sort().map(key => [key, sortValue(value[key])]));
+    }
+    return value;
+  };
+  return `${toolName}:${JSON.stringify(sortValue(args || {}))}`;
+}
+
+export class KikuAgent {
+  constructor(options = {}) {
+    this.client = options.client || new LLMClient({
+      provider: options.provider || config.activeProvider,
+      baseUrl: options.baseUrl,
+      apiKey: options.apiKey,
+      model: options.model
+    });
+    this.maxIterations = options.maxIterations || config.agent.maxIterations || 5;
+    this.slidingWindowSize = options.slidingWindowSize || config.agent.slidingWindowSize || 8;
+    this.session = new AgentSession({ thinkingMode: options.thinkingMode || 'hide' });
+    this.memory = new AgentMemory();
+    this.mcpManager = options.mcpManager || new McpClientManager();
+    this.router = new DynamicToolRouter(this.getAllToolsSchema());
+    this.research = new ResearchOrchestrator({ client: this.client });
+    this.systemPrompt = this.buildSystemPrompt();
+    this.initHistory();
+    this.restoredMessageCount = 0;
+
+    this.stateMachine = new AgentStateMachine();
+
+    this.stateMachine.on('transition', ({ from, to }) => {
+      eventBus.emit(Events.AGENT_STATE_CHANGE, {
+        state: to,
+        label: this.stateMachine.label,
+        elapsed: this.stateMachine.elapsedMs,
+        queryId: this.stateMachine.queryId,
+      });
+    });
+
+    this._ready = this._init();
+  }
+
+  async _init() {
+    await this.memory.load();
+    await this.loadSession();
+    if (this.mcpManager && typeof this.mcpManager.loadPersistedServers === 'function') {
+      try {
+        await this.mcpManager.loadPersistedServers();
+        this.router.setTools(this.getAllToolsSchema());
+        this.systemPrompt = this.buildSystemPrompt();
+      } catch {}
+    }
+    if (!this.conversationHistory || this.conversationHistory.length === 0) {
+      this.initHistory();
+    }
+  }
+
+  async ready() {
+    return this._ready;
+  }
+
+  async runSubagent(task, options = {}) {
+    await this._ready;
+    const subagent = new ResearchSubagent({
+      client: this.client,
+      maxIterations: options.maxIterations || 3,
+    });
+    return subagent.run(task, {
+      signal: options.signal,
+      onProgress: options.onProgress,
+      skillContext: this.memory.buildUserSkillContext(),
+    });
+  }
+
+  async runResearch(question, options = {}) {
+    await this._ready;
+    return this.research.run(question, {
+      ...options,
+      skillContext: this.memory.buildUserSkillContext(),
+    });
+  }
+
+  getResearchStatus() {
+    return this.research.getStatus();
+  }
+
+  cancelResearch() {
+    return this.research.cancel();
+  }
+
+  initHistory() {
+    this.conversationHistory = [
+      {
+        role: 'system',
+        content: this.systemPrompt
+      }
+    ];
+  }
+
+  clearMemory() {
+    if (this.session.messages.length > 0) {
+      this.memory.archiveSession(this.session.toSnapshot()).catch(() => {});
+      this.memory.consolidateSession(this.session.toSnapshot()).catch(() => {});
+    }
+    this.initHistory();
+    this.session.reset();
+    this.session.clearDisk().catch(() => {});
+  }
+
+  async loadSession() {
+    const loaded = await this.session.loadFromDisk();
+    if (loaded && this.session.messages.length > 0) {
+      this.conversationHistory = [
+        {
+          role: 'system',
+          content: this.systemPrompt
+        }
+      ];
+      
+      for (const msg of this.session.messages) {
+        if (msg.role === 'user') {
+          const textPart = msg.parts.find(p => p.type === 'text');
+          if (textPart) {
+            this.conversationHistory.push({
+              role: 'user',
+              content: textPart.content
+            });
+          }
+        } else if (msg.role === 'assistant') {
+          const textPart = msg.parts.find(p => p.type === 'text');
+          if (textPart) {
+            this.conversationHistory.push({
+              role: 'assistant',
+              content: textPart.content
+            });
+          }
+        }
+      }
+      this.restoredMessageCount = this.session.messages.length;
+    }
+  }
+
+  getEnrichedSystemPrompt() {
+    const memoryContext = this.memory.buildMemoryContext();
+    return this.systemPrompt + memoryContext;
+  }
+
+  async saveSession() {
+    await this.session.saveToDisk();
+    eventBus.emit(Events.SESSION_SAVED, { sessionId: this.session.id });
+  }
+
+  applySlidingWindow() {
+    if (this.conversationHistory.length > this.slidingWindowSize + 1) {
+      this.conversationHistory = contextCompactor.compact(this.conversationHistory, {
+        maxTokens: 6000,
+        keepRecentMessages: this.slidingWindowSize,
+      });
+    }
+  }
+
+  getAllToolsSchema() {
+    const mcpTools = this.mcpManager ? this.mcpManager.getToolsSchema() : [];
+    return [...toolsSchema, ...mcpTools];
+  }
+
+  async connectMcp(name, serverConfig) {
+    const result = await this.mcpManager.connect(name, serverConfig);
+    this.router.setTools(this.getAllToolsSchema());
+    this.systemPrompt = this.buildSystemPrompt();
+    if (this.conversationHistory && this.conversationHistory[0]?.role === 'system') {
+      this.conversationHistory[0].content = this.getEnrichedSystemPrompt();
+    }
+    return result;
+  }
+
+  async disconnectMcp(name) {
+    const result = await this.mcpManager.disconnect(name);
+    this.router.setTools(this.getAllToolsSchema());
+    this.systemPrompt = this.buildSystemPrompt();
+    if (this.conversationHistory && this.conversationHistory[0]?.role === 'system') {
+      this.conversationHistory[0].content = this.getEnrichedSystemPrompt();
+    }
+    return result;
+  }
+
+  buildSystemPrompt() {
+    const allTools = this.getAllToolsSchema();
+    const toolsSummary = allTools.map(t => {
+      const f = t.function;
+      const paramKeys = Object.keys(f.parameters?.properties || {}).join(', ');
+      return `- ${f.name}(${paramKeys}): ${f.description}`;
+    }).join('\n');
+
+    return `${config.agent.systemPersona}
+
+=== AVAILABLE KNOWLEDGE & LIVE TOOLS ===
+${toolsSummary}
+
+=== DYNAMIC MULTILINGUAL & RESPONSE RULES (CRITICAL) ===
+
+0. NO SYSTEM LEAK (ABSOLUTE RULE):
+   - NEVER include "Thinking Process", "Tool Calling", "Action:", "Action Input:", function names, parameter schemas, JSON tool call blocks, or any internal reasoning in your response text.
+   - NEVER include advertisements, sponsor mentions, or promotional content.
+   - Your response must be CLEAN user-facing content ONLY.
+   - Think silently using tools, then present ONLY the final polished answer.
+
+1. LANGUAGE MIRRORING:
+   - THAI ➔ polite Thai (ใช้ ค่ะ/นะคะ เสมอ)
+   - CHINESE ➔ fluent, polite Chinese (简体/繁体中文)
+   - ENGLISH ➔ polished, professional English
+   - MIXED ➔ Match dominant language, keep bilingual titles
+
+2. REAL YOUTUBE & MEDIA LINKS:
+   - Direct Watch: https://www.youtube.com/watch?v=VIDEO_ID
+   - Search: https://www.youtube.com/results?search_query=QUERY
+   - Also Bilibili, QQ Music, Youku when applicable
+   - NEVER use fake placeholder IDs
+
+3. FACT ACCURACY & NO DUPLICATION:
+   - Always use tools to verify facts before answering. Do NOT hallucinate.
+   - Current year is 2026. Prioritize 2026/2025 projects.
+   - Never repeat the same statement. Sort newest first.
+
+4. RESPONSE STYLE:
+   - NEVER start with template greetings — dive DIRECTLY into the answer.
+   - Format with Markdown headers (###), bold (**text**), bullet points (•).
+   - Emoji are optional: use at most one contextually relevant emoji per response and never repeat them across headings or bullets.
+   - Match answer depth to what the user asked.
+
+5. SMART TOOL SELECTION:
+   - Videos/stages ➔ get_music_videos or get_media_links
+   - Songs/OSTs ➔ search_discography or get_media_links
+   - Dramas ➔ search_works or get_work_detail
+   - News/external facts ➔ web_search or web_extract
+   - Deep multi-source research ➔ web_extract (search + auto-fetch)
+   - Read webpage content ➔ fetch_webpage
+   - Fetch API/JSON ➔ fetch_json
+   - Local files ➔ read_local_file
+   - Directory listing ➔ list_directory
+   - Screenshot ➔ screenshot_webpage
+
+6. AGGRESSIVE WEB SEARCH FALLBACK:
+   - If local tools return insufficient results, IMMEDIATELY call web_search or web_extract.
+   - Combine local data WITH web search for complete, accurate answers.
+   - Use web_extract for comprehensive research on any topic.
+
+7. LEGAL & CAREER KNOWLEDGE:
+   - When discussing controversies, be specific and accurate:
+     • Contract dispute: Star48/Siba Media supplementary agreement. JJY claims main contract ended June 2024, alleges forged signatures. Star48 claims until 2033.
+     • Tax allegation: Tip-off report filed by Star48 side; JJY denies and confirms tax compliance.
+     • Portrait/copyright cases: JJY has sued beauty clinics and brands for unauthorized image use (肖像权纠纷).
+   - Use web_search or web_extract for latest verified information on legal matters.
+
+8. DELIBERATE REASONING & AUTONOMOUS TOOL SELECTION (CRITICAL):
+   - Think carefully about the complexity, goal, and nature of the query before deciding which tool to use:
+   • Direct Specific Query ➔ Use direct catalog tools (search_works, search_discography, get_profile_and_fandom, get_awards, get_brand_endorsements, etc.).
+   • Isolated In-Depth Subtask ➔ Call run_subagent with a specific task parameter to delegate an isolated iterative investigation with evidence and tool call tracing.
+   • Complex Multi-Faceted Research ➔ Call run_deep_research with question when the user asks an open-ended, complex query that requires parallel multi-worker investigation across catalog, live web sources, and independent verifier with cited sources.
+   • Rumors, Cast Leaks & Fact Verification ➔ Call verify_claim with claim when the user asks whether a rumor, statement, or unconfirmed news is true (returns VERIFIED, UNVERIFIED, CONFLICTING, or FALSE).
+   • Teaching / Updating Response Preferences ➔ Call manage_skill with action 'learn' (with instruction), 'list', 'remove' (with skillId), 'clear', 'enable', or 'disable' when the user instructs how they want Kiku to answer in future conversations.`;
+  }
+
+  parseTextToolCalls(content) {
+    if (!content) return [];
+    const calls = [];
+    const allTools = this.getAllToolsSchema();
+
+    const jsonBlockRegex = /```(?:json)?\s*(\{[\s\S]*?\})\s*```/g;
+    let match;
+    while ((match = jsonBlockRegex.exec(content)) !== null) {
+      try {
+        const parsed = JSON.parse(match[1]);
+        const toolName = parsed.action || parsed.tool || parsed.name || parsed.function;
+        const toolArgs = parsed.action_input || parsed.parameters || parsed.arguments || parsed.args || {};
+        if (toolName && allTools.some(t => t.function.name === toolName)) {
+          calls.push({
+            id: `call_txt_${Date.now()}_${calls.length}`,
+            function: {
+              name: toolName,
+              arguments: typeof toolArgs === 'string' ? toolArgs : JSON.stringify(toolArgs)
+            }
+          });
+        }
+      } catch {
+      }
+    }
+
+    const actionRegex = /Action:\s*([a-zA-Z0-9_]+)\s*\nAction Input:\s*(\{[\s\S]*?\}|[^\n]+)/g;
+    while ((match = actionRegex.exec(content)) !== null) {
+      const toolName = match[1].trim();
+      let rawInput = match[2].trim();
+      if (allTools.some(t => t.function.name === toolName)) {
+        let args = {};
+        try {
+          args = JSON.parse(rawInput);
+        } catch {
+          args = { query: rawInput };
+        }
+        calls.push({
+          id: `call_act_${Date.now()}_${calls.length}`,
+          function: {
+            name: toolName,
+            arguments: JSON.stringify(args)
+          }
+        });
+      }
+    }
+
+    return calls;
+  }
+
+  async ask(userPrompt, callbacks = {}) {
+    await this._ready;
+
+    const { sanitized: cleanPrompt, warnings: sanitizationWarnings } = sanitizeQuery(userPrompt);
+    if (sanitizationWarnings.length > 0) {
+      eventBus.emit(Events.AGENT_QUERY_ERROR, {
+        type: 'sanitization',
+        warnings: sanitizationWarnings,
+        original: userPrompt,
+      });
+    }
+    const promptForLLM = cleanPrompt || userPrompt;
+
+    const topics = this._extractQuickTopics(promptForLLM);
+    for (const topic of topics) {
+      this.memory.trackTopicInterest(topic);
+    }
+    
+    const {
+      onStepStart = () => {},
+      onReasoningStart = () => {},
+      onReasoningDelta = () => {},
+      onReasoningEnd = () => {},
+      onToolStart = () => {},
+      onToolResult = () => {},
+      onTokenStart = () => {},
+      onTokenDelta = () => {},
+      onFinalAnswer = () => {},
+      onError = () => {},
+      signal = null
+    } = callbacks;
+
+    const queryStartTime = Date.now();
+    const queryId = `q_${Date.now()}_${Math.random().toString(36).slice(2, 6)}`;
+    const dominantLang = detectDominantLanguage(promptForLLM);
+
+    this.stateMachine.startQuery(queryId);
+    eventBus.emit(Events.AGENT_QUERY_START, { queryId, prompt: promptForLLM });
+
+    if (!this.conversationHistory || this.conversationHistory.length === 0 || this.conversationHistory[0]?.role !== 'system') {
+      this.conversationHistory = [
+        { role: 'system', content: this.systemPrompt },
+        ...(this.conversationHistory?.filter(m => m.role !== 'system') || [])
+      ];
+    }
+
+    this.session.createUserMessage(promptForLLM);
+    const userMessage = { role: 'user', content: promptForLLM };
+    this.conversationHistory.push(userMessage);
+
+    const memoryContext = this.memory.buildMemoryContext(promptForLLM);
+    this.conversationHistory[0] = {
+      role: 'system',
+      content: this.systemPrompt + memoryContext
+    };
+
+    let langInstruction;
+    if (dominantLang === 'th') {
+      langInstruction = '[SYSTEM NOTE: User language is Thai. Reply in Thai only.]\n';
+    } else if (dominantLang === 'zh') {
+      langInstruction = '[SYSTEM NOTE: User language is Chinese. Reply in Chinese only.]\n';
+    } else {
+      langInstruction = '[SYSTEM NOTE: User language is English. Reply in English only.]\n';
+    }
+
+    const directMentions = this.router.extractAtMentions(promptForLLM);
+    if (directMentions.length > 0) {
+      langInstruction += `[SYSTEM DIRECTIVE: User explicitly invoked @${directMentions.join(', @')}. You MUST prioritize calling the referenced tool(s) to fetch accurate data.]\n`;
+    }
+    this.conversationHistory.push({ role: 'system', content: langInstruction });
+
+    this.applySlidingWindow();
+
+    const currentAssistantSessionMsg = this.session.createAssistantMessage();
+
+    let currentIteration = 0;
+    const toolResultCache = new Map();
+    const activeTools = this.router.routeToolsForQuery(promptForLLM, {
+      showAll: callbacks.showAllTools || false,
+      includeMcp: true,
+    });
+
+    while (currentIteration < this.maxIterations) {
+      if (signal?.aborted) {
+        this.stateMachine.reset();
+        throw createAbortError();
+      }
+
+      currentIteration++;
+      onStepStart(currentIteration, this.maxIterations);
+
+      let assistantMessage = null;
+      let isReasoning = false;
+      let isTokenEmitting = false;
+
+      try {
+        assistantMessage = await this.client.chat({
+          messages: this.conversationHistory,
+          tools: activeTools,
+          temperature: 0.3,
+          stream: true,
+          signal,
+          onThought: (thoughtChunk) => {
+            if (!isReasoning) {
+              isReasoning = true;
+              this.stateMachine.transition(AgentStates.STREAMING_THOUGHTS);
+              onReasoningStart();
+              eventBus.emit(Events.AGENT_REASONING_START, { queryId });
+            }
+            this.session.addReasoningPart(currentAssistantSessionMsg, thoughtChunk);
+            onReasoningDelta(thoughtChunk);
+            eventBus.emit(Events.AGENT_REASONING_DELTA, { queryId, chunk: thoughtChunk });
+          },
+          onToken: (tokenChunk) => {
+            if (isReasoning) {
+              isReasoning = false;
+              this.session.finalizeReasoningPart(currentAssistantSessionMsg);
+              this.stateMachine.transition(AgentStates.STREAMING_ANSWER);
+              onReasoningEnd();
+              eventBus.emit(Events.AGENT_REASONING_END, { queryId });
+            }
+            if (!isTokenEmitting) {
+              isTokenEmitting = true;
+              if (this.stateMachine.state !== AgentStates.STREAMING_ANSWER) {
+                this.stateMachine.transition(AgentStates.STREAMING_ANSWER);
+              }
+              onTokenStart();
+              eventBus.emit(Events.AGENT_TOKEN_START, { queryId });
+            }
+            this.session.addTextPart(currentAssistantSessionMsg, tokenChunk);
+            onTokenDelta(tokenChunk);
+            eventBus.emit(Events.AGENT_TOKEN_DELTA, { queryId, chunk: tokenChunk });
+          }
+        });
+
+        if (isReasoning) {
+          this.session.finalizeReasoningPart(currentAssistantSessionMsg);
+          if (this.stateMachine.state === AgentStates.STREAMING_THOUGHTS) {
+            this.stateMachine.transition(AgentStates.STREAMING_ANSWER);
+          }
+          onReasoningEnd();
+        }
+      } catch (err) {
+        this.stateMachine.reset();
+        if (!isAbortError(err)) {
+          onError(err);
+          eventBus.emit(Events.AGENT_QUERY_ERROR, { queryId, error: err.message });
+        }
+        throw err;
+      }
+
+      if (!assistantMessage) {
+        this.stateMachine.reset();
+        throw new Error(`No response received from ${this.client?.providerName || 'AI provider'}.`);
+      }
+
+      let toolCalls = assistantMessage.tool_calls || [];
+      if (toolCalls.length === 0 && assistantMessage.content) {
+        toolCalls = this.parseTextToolCalls(assistantMessage.content);
+      }
+
+      this.conversationHistory.push(assistantMessage);
+
+      if (!toolCalls || toolCalls.length === 0) {
+        const totalDurationMs = Date.now() - queryStartTime;
+        this.session.stats.totalDurationMs += totalDurationMs;
+
+        const finalAnswer = assistantMessage.content || (
+          dominantLang === 'zh' ? "抱歉，无法处理您的回答。" :
+          dominantLang === 'en' ? "Sorry, unable to process a response." :
+          "ขออภัยค่ะ ไม่สามารถประมวลผลคำตอบได้"
+        );
+        onFinalAnswer(finalAnswer, { durationMs: totalDurationMs, language: dominantLang });
+        this._extractFactsFromFinalAnswer(finalAnswer);
+
+        this.stateMachine.transition(AgentStates.COMPLETE);
+        eventBus.emit(Events.AGENT_FINAL_ANSWER, { queryId, answer: finalAnswer, durationMs: totalDurationMs });
+        this.stateMachine.reset();
+
+        return finalAnswer;
+      }
+
+      for (const [toolCallIndex, tc] of toolCalls.entries()) {
+        if (signal?.aborted) {
+          this.stateMachine.reset();
+          throw createAbortError();
+        }
+
+        const toolName = tc?.function?.name || '';
+        const toolCallId = tc?.id || `call_${queryId}_${currentIteration}_${toolCallIndex}`;
+        const isMcp = Boolean(this.mcpManager?.isMcpTool(toolName));
+
+        let prepared;
+        if (isMcp) {
+          let parsedArgs = tc?.function?.arguments || {};
+          if (typeof parsedArgs === 'string') {
+            try { parsedArgs = JSON.parse(parsedArgs); } catch { parsedArgs = {}; }
+          }
+          prepared = { safe: true, args: parsedArgs, warnings: [] };
+        } else {
+          prepared = prepareToolArguments(toolName, tc?.function?.arguments || {});
+        }
+
+        if (!prepared.safe) {
+          const rejectedResult = {
+            status: 'error',
+            message: `Input rejected: ${prepared.error}`,
+            validationWarnings: prepared.warnings,
+          };
+          eventBus.emit(Events.AGENT_TOOL_ERROR, {
+            queryId, toolName, args: tc.function.arguments,
+            error: prepared.error,
+          });
+          onToolStart(toolName, {});
+          onToolResult(toolName, rejectedResult, 0);
+          const toolMessage = {
+            role: 'tool',
+            tool_call_id: toolCallId,
+            name: toolName,
+            content: JSON.stringify(rejectedResult)
+          };
+          this.conversationHistory.push(toolMessage);
+          continue;
+        }
+        if (prepared.warnings.length > 0) {
+          eventBus.emit(Events.AGENT_TOOL_ERROR, {
+            queryId, toolName, warnings: prepared.warnings,
+          });
+        }
+
+        const finalArgs = prepared.args;
+
+        this.stateMachine.transition(AgentStates.FETCHING_TOOL, { toolName });
+
+        const toolPart = this.session.addToolPart(currentAssistantSessionMsg, toolName, finalArgs);
+        onToolStart(toolName, finalArgs);
+        eventBus.emit(Events.AGENT_TOOL_START, { queryId, toolName, args: finalArgs });
+
+        const toolStartTime = Date.now();
+        const signature = stableToolSignature(toolName, finalArgs);
+        const cachedResult = toolResultCache.get(signature);
+
+        let result;
+        if (cachedResult) {
+          result = { ...cachedResult, cache: { hit: true, reason: 'duplicate_call_in_query' } };
+        } else if (isMcp) {
+          result = await this.mcpManager.callTool(toolName, finalArgs);
+        } else {
+          result = await executeTool(toolName, finalArgs, { agent: this, memory: this.memory, client: this.client, signal });
+        }
+
+        if (!cachedResult) toolResultCache.set(signature, result);
+        const toolDurationMs = Date.now() - toolStartTime;
+
+        this.session.finalizeToolPart(currentAssistantSessionMsg, toolPart.id, result);
+        onToolResult(toolName, result, toolDurationMs);
+        eventBus.emit(Events.AGENT_TOOL_RESULT, { queryId, toolName, result, durationMs: toolDurationMs });
+
+        this._extractFactsFromToolResult(toolName, finalArgs, result);
+
+        const toolMessage = {
+          role: 'tool',
+          tool_call_id: toolCallId,
+          name: toolName,
+          content: JSON.stringify({
+            authoritative: true,
+            result,
+            instruction: result.status === 'not_found'
+              ? 'No matching data was found. Do not invent records; say that the catalog has no match or use an appropriate fallback tool.'
+              : result.status === 'error'
+                ? 'The tool failed. Do not claim that the requested facts were verified.'
+                : 'Use only facts present in this result when answering.'
+          })
+        };
+
+        this.conversationHistory.push(toolMessage);
+      }
+    }
+
+    const langPrompt = dominantLang === 'zh' 
+      ? "请根据上述工具查询结果，直接回答（不要开头问候语），用流畅优雅的中文，先1-2句概述，再列要点。包含具体的 YouTube 真实链接（如 Direct Watch: https://www.youtube.com/watch?v=... หรือ Search: https://www.youtube.com/results?search_query=...）及事实数据。"
+      : dominantLang === 'en'
+      ? "Based on the tool results above, answer directly (no opening greeting or template phrase). Give a 1-2 sentence summary, then concise bullet points in polished English. Include verified YouTube direct and search links (https://www.youtube.com/results?search_query=...)."
+      : "สรุปคำตอบจากผลลัพธ์เครื่องมือข้างต้น โดยไม่ต้องทักทาย ตอบตรงประเด็น 1-2 ประโยค แล้วตามด้วย bullet points สั้นๆ เป็นภาษาไทย พร้อมแสดงลิงก์ YouTube จริง (Direct Watch Link และ https://www.youtube.com/results?search_query=...) และข้อมูลผลงานที่ถูกต้องแม่นยำ";
+
+    this.conversationHistory.push({ role: 'user', content: langPrompt });
+
+    if (this.stateMachine.state !== AgentStates.STREAMING_ANSWER) {
+      this.stateMachine.transition(AgentStates.STREAMING_ANSWER);
+    }
+
+    let isReasoning = false;
+    let synthTokenStarted = false;
+    const finalRes = await this.client.chat({
+      messages: this.conversationHistory,
+      temperature: 0.3,
+      stream: true,
+      signal,
+      onThought: (thoughtChunk) => {
+        if (!isReasoning) {
+          isReasoning = true;
+          onReasoningStart();
+        }
+        this.session.addReasoningPart(currentAssistantSessionMsg, thoughtChunk);
+        onReasoningDelta(thoughtChunk);
+      },
+      onToken: (tokenChunk) => {
+        if (isReasoning) {
+          isReasoning = false;
+          this.session.finalizeReasoningPart(currentAssistantSessionMsg);
+          onReasoningEnd();
+        }
+        if (!synthTokenStarted) {
+          synthTokenStarted = true;
+          onTokenStart();
+        }
+        this.session.addTextPart(currentAssistantSessionMsg, tokenChunk);
+        onTokenDelta(tokenChunk);
+      }
+    });
+
+    const totalDurationMs = Date.now() - queryStartTime;
+    this.session.stats.totalDurationMs += totalDurationMs;
+
+    const defaultDone = dominantLang === 'zh' ? "查询完毕" : dominantLang === 'en' ? "Search complete." : "เสร็จสิ้นการค้นหาข้อมูล";
+    const output = finalRes?.content || defaultDone;
+    onFinalAnswer(output, { durationMs: totalDurationMs, language: dominantLang });
+    
+    this._extractFactsFromFinalAnswer(output);
+
+    this.stateMachine.transition(AgentStates.COMPLETE);
+    eventBus.emit(Events.AGENT_FINAL_ANSWER, { queryId, answer: output, durationMs: totalDurationMs });
+    eventBus.emit(Events.AGENT_QUERY_END, { queryId, durationMs: totalDurationMs });
+    this.stateMachine.reset();
+    
+    this.saveSession().catch(err => console.error('Session save failed:', err));
+    
+    return output;
+  }
+
+  _extractFactsFromToolResult(toolName, toolArgs, result) {
+    if (!result || result.status === 'error' || result.status === 'not_found') return;
+
+    const facts = [];
+
+    switch (toolName) {
+      case 'search_works': {
+        const works = result.works || [];
+        for (const w of works.slice(0, 3)) {
+          if (w.titleCn && w.role) {
+            facts.push({
+              content: `${w.titleCn} (${w.titleEn || w.titleTh || ''}) — Role: ${w.role} (${w.year || '?'})`,
+              category: 'drama',
+              tags: [w.titleEn, w.titleCn, String(w.year)].filter(Boolean)
+            });
+          }
+        }
+        break;
+      }
+
+      case 'get_work_detail': {
+        const w = result.work;
+        if (w && w.titleCn) {
+          facts.push({
+            content: `${w.titleCn} (${w.titleEn || w.titleTh || ''}) — ${w.role || 'Actor'} | ${w.platform || ''} | ${w.year || ''}`,
+            category: 'drama',
+            tags: [w.titleEn, w.titleCn, w.platform, String(w.year)].filter(Boolean)
+          });
+          if (w.synopsis) {
+            facts.push({
+              content: `Synopsis of ${w.titleCn}: ${w.synopsis.slice(0, 150)}`,
+              category: 'drama',
+              tags: [w.titleCn, 'synopsis']
+            });
+          }
+        }
+        break;
+      }
+
+      case 'search_discography': {
+        const songs = result.songs || [];
+        for (const s of songs.slice(0, 3)) {
+          if (s.title && s.year) {
+            facts.push({
+              content: `Song: ${s.title} (${s.year}) — Type: ${s.type || 'Single'}`,
+              category: 'music',
+              tags: [s.titleEn, s.titleCn, String(s.year)].filter(Boolean)
+            });
+          }
+        }
+        break;
+      }
+
+      case 'get_music_videos': {
+        const vids = result.videos || [];
+        for (const v of vids.slice(0, 3)) {
+          if (v.title && v.year) {
+            facts.push({
+              content: `MV/Stage: ${v.title} (${v.year}) — ${v.category || 'MV'}`,
+              category: 'video',
+              tags: [v.title, String(v.year), v.category].filter(Boolean)
+            });
+          }
+        }
+        break;
+      }
+
+      case 'get_brand_endorsements': {
+        const brands = result.brands || [];
+        for (const b of brands.slice(0, 3)) {
+          if (b.brand) {
+            facts.push({
+              content: `Brand: ${b.brand} — ${b.title || 'Ambassador'} (${b.year || ''})${b.isGlobal ? ' [Global]' : ''}`,
+              category: 'brand',
+              tags: [b.brand, String(b.year)].filter(Boolean)
+            });
+          }
+        }
+        break;
+      }
+
+      case 'get_awards': {
+        const awards = result.awards || [];
+        for (const a of awards.slice(0, 3)) {
+          if (a.title && a.year) {
+            facts.push({
+              content: `Award: ${a.title} (${a.year}) — ${a.ceremony || ''}`,
+              category: 'awards',
+              tags: [a.title, String(a.year), a.ceremony].filter(Boolean)
+            });
+          }
+        }
+        break;
+      }
+
+      case 'get_profile_and_fandom': {
+        const p = result.profile;
+        if (p) {
+          if (p.birthDate) facts.push({ content: `Birth date: ${p.birthDate}`, category: 'profile', tags: ['birthday'] });
+          if (p.fandomName) facts.push({ content: `Fandom: ${p.fandomName}`, category: 'profile', tags: ['fandom'] });
+          if (p.slogan) facts.push({ content: `Slogan: ${p.slogan}`, category: 'profile', tags: ['slogan'] });
+        }
+        break;
+      }
+
+      case 'get_career_timeline': {
+        const milestones = result.milestones || [];
+        for (const m of milestones.slice(0, 2)) {
+          if (m.title && m.period) {
+            facts.push({
+              content: `Career milestone: ${m.title} (${m.period})`,
+              category: 'timeline',
+              tags: [m.title, m.period].filter(Boolean)
+            });
+          }
+        }
+        break;
+      }
+
+      case 'web_search': {
+        const items = result.results || [];
+        for (const item of items.slice(0, 2)) {
+          if (item.title && item.snippet) {
+            facts.push({
+              content: `Web: ${item.title.slice(0, 100)} — ${item.snippet.slice(0, 150)}`,
+              category: 'news',
+              tags: ['web-search']
+            });
+          }
+        }
+        break;
+      }
+    }
+
+    for (const fact of facts) {
+      const isDuplicate = this.memory.facts.some(
+        existing => existing.content === fact.content
+      );
+      if (!isDuplicate) {
+        this.memory.addFact(fact.content, {
+          category: fact.category,
+          source: `tool:${toolName}`,
+          tags: fact.tags
+        });
+        eventBus.emit(Events.MEMORY_FACT_ADDED, { fact: fact.content, category: fact.category });
+      }
+    }
+  }
+
+  _extractFactsFromFinalAnswer(answer) {
+    if (!answer || answer.length < 20) return;
+
+    const ytRegex = /https?:\/\/www\.youtube\.com\/watch\?v=([a-zA-Z0-9_-]+)/g;
+    let match;
+    const ytIds = new Set();
+    while ((match = ytRegex.exec(answer)) !== null) {
+      ytIds.add(match[1]);
+    }
+    if (ytIds.size > 0) {
+      this.memory.addFact(
+        `YouTube video IDs discussed: ${[...ytIds].join(', ')}`,
+        { category: 'links', source: 'final-answer', tags: ['youtube'] }
+      );
+    }
+  }
+
+  _extractQuickTopics(text) {
+    const topicMap = {
+      'drama': /drama|series|ละคร|剧|电视|tv|film|movie/i,
+      'music': /song|music|mv|single|ost|เพลง|歌|album/i,
+      'brand': /brand|ambassador|endorsement|แบรนด์|品牌/i,
+      'awards': /award|honor|ceremony|รางวัล|奖/i,
+      'profile': /profile|bio|birthday|ประวัติ|资料/i,
+      'news': /news|latest|recent|update|ข่าว|最新/i,
+      'video': /video|stage|performance|live|คลิป|视频|舞台/i,
+      'social': /weibo|douyin|instagram|social|微博/i,
+      'gallery': /photo|picture|gallery|fashion|ภาพ|照片/i
+    };
+    const found = [];
+    for (const [topic, regex] of Object.entries(topicMap)) {
+      if (regex.test(text)) found.push(topic);
+    }
+    return found;
+  }
+
+  async endSession() {
+    await this._ready;
+    const snapshot = this.session.toSnapshot();
+    if (snapshot.messages.length > 0) {
+      const archived = await this.memory.archiveSession(snapshot);
+      if (archived) {
+        await this.memory.consolidateSession(snapshot);
+        eventBus.emit(Events.SESSION_ARCHIVED, { sessionId: snapshot.id });
+      }
+    }
+    await this.session.clearDisk();
+    this.session.reset();
+    this.initHistory();
+  }
+}