wtt-connect 0.2.59 → 0.2.61

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "wtt-connect",
-  "version": "0.2.59",
+  "version": "0.2.61",
   "private": false,
   "description": "WTT-native connector daemon for Codex, Claude Code, Cursor, Gemini, ACP, and other coding agent surfaces.",
   "type": "module",

package/src/runner.js

@@ -77,6 +77,18 @@ export class Runner {
     return buildRuntimeInfo(this.config, this.store.getRuntime());
   }
 
+  async buildKnowledgeContext(metadata, query) {
+    const meta = parseMetadata(metadata);
+    if (!meta || meta.kb_mode !== true) return '';
+    const kbTaskId = String(meta.kb_task_id || meta.kbTaskId || '').trim();
+    if (!kbTaskId) return '';
+    const result = await this.api.searchKnowledgeContext(kbTaskId, query, { limit: 8 });
+    return renderKnowledgeContextBlock(result?.results || [], {
+      taskId: kbTaskId,
+      scope: meta.kb_scope || meta.kbScope || 'personal',
+    });
+  }
+
   async onEvent(msg) {
     if (msg.type === 'shell_run') {
       await this.handleShellRun(msg);
@@ -182,6 +194,11 @@ export class Runner {
       await this.wtt.typing(topicId, 'start', { statusText: `${adapterDisplayName(adapter.name)} 正在执行${modelConfig.model ? ` · ${modelConfig.model}` : ''}`, statusKind: 'running', adapter: adapter.name, model: modelConfig.model || undefined, ttlMs: 30000 });
       const agentProfile = await this.getAgentProfile();
       const agentSoul = renderAgentSoulContext(m.metadata, agentProfile?.role_template);
+      const knowledgeContext = await this.buildKnowledgeContext(m.metadata, [
+        content,
+        staged.promptBlock,
+        transcripts.map((item) => item.text).join('\n'),
+      ].filter(Boolean).join('\n\n'));
       const discussionRouting = renderDiscussionRoutingInstruction(m, this.config);
       const currentDisplayName = effectiveAgentDisplayName(this.config, agentProfile);
       const isSlashPassthrough = isAgentSlashCommand(m, content);
@@ -198,7 +215,7 @@ export class Runner {
         if (localSlash !== null) {
           const reply = stripHiddenContextLeak(localSlash || '(empty response)') || '(empty response)';
           await chatStream.finish(reply);
-          await this.wtt.publish(topicId, reply, 'CHAT_REPLY', { stream_id: chatStream.id });
+          await this.wtt.publish(topicId, reply, 'CHAT_REPLY', chatStream.finalMetadata());
           log('info', 'slash command replied', { topicId, adapter: adapter.name, chars: reply.length });
           return;
         }
@@ -218,6 +235,8 @@ export class Runner {
         discussionRouting ? '' : null,
         agentSoul,
         agentSoul ? '' : null,
+        knowledgeContext,
+        knowledgeContext ? '' : null,
         renderCompactSummaryBlock(compactSummary),
         compactSummary ? '' : null,
         'User message:',
@@ -241,12 +260,12 @@ export class Runner {
       const reply = prepared.text || '(empty response)';
       await chatStream.finish(reply);
       await this.maybeAttachSpeech(topicId, reply, `chat-${topicId}`);
-      await this.wtt.publish(topicId, reply, 'CHAT_REPLY', { stream_id: chatStream.id });
+      await this.wtt.publish(topicId, reply, 'CHAT_REPLY', chatStream.finalMetadata());
       await this.publishGeneratedFileArtifacts(topicId, prepared.refs, { source: 'chat', sourceId: topicId, adapter: adapter.name });
       log('info', 'chat replied', { topicId, chars: reply.length });
     } catch (err) {
       await chatStream?.fail(err?.message || String(err));
-      await this.wtt.publish(topicId, `执行失败：${err.message}`, 'CHAT_REPLY', chatStream?.id ? { stream_id: chatStream.id } : null);
+      await this.wtt.publish(topicId, `执行失败：${err.message}`, 'CHAT_REPLY', chatStream?.finalMetadata?.() || null);
       log('error', 'chat failed', { topicId, error: err.message });
     } finally {
       if (runtimeSelection) this.recordRuntimeSelection(runtimeSelection.adapter, runtimeSelection.modelConfig, 'idle');
@@ -267,7 +286,25 @@ export class Runner {
     this.recordRuntimeSelection(adapter.name, modelConfig, 'running');
     const agentProfile = await this.getAgentProfile();
     if (topicId) await this.wtt.typing(topicId, 'start', { statusText: `${adapterDisplayName(adapter.name)} 正在执行任务${modelConfig.model ? ` · ${modelConfig.model}` : ''}`, statusKind: 'running', adapter: adapter.name, model: modelConfig.model || undefined, ttlMs: 30000 });
-    const prompt = buildTaskPrompt(task, this.config, staged, transcripts, agentProfile);
+    const taskMetadata = task.metadata || task.msg_metadata || event?.message?.metadata || event?.metadata;
+    const taskQuery = [
+      task.title,
+      task.description,
+      task.content,
+      task.task_request,
+      event?.message?.content,
+      staged.promptBlock,
+      transcripts.map((item) => item.text).join('\n'),
+    ].filter(Boolean).join('\n\n');
+    const knowledgeContext = await this.buildKnowledgeContext(taskMetadata, taskQuery);
+    const prompt = buildTaskPrompt(
+      { ...task, metadata: taskMetadata || task.metadata },
+      this.config,
+      staged,
+      transcripts,
+      agentProfile,
+      knowledgeContext,
+    );
     try {
       const output = await this.runAdapter(adapter, prompt, {
         sessionKey: `wtt:task:${taskId}:${adapter.name}:${sessionModelKey(modelConfig)}`,
@@ -610,6 +647,14 @@ export class Runner {
 
     return {
       id: streamId,
+      finalMetadata() {
+        return {
+          stream_id: streamId,
+          streamId,
+          adapter: adapterName,
+          model: modelConfig.model || modelConfig.model_id || modelConfig.modelId || '',
+        };
+      },
       async start() {
         await send('start');
       },
@@ -1415,6 +1460,31 @@ function renderAgentSoulContext(metadata, profileRole = null) {
   ].join('\n');
 }
 
+function renderKnowledgeContextBlock(results = [], options = {}) {
+  const rows = Array.isArray(results) ? results : [];
+  if (!rows.length) {
+    return [
+      'WTT personal knowledge base mode is enabled, but no indexed chunk matched this request.',
+      'If the user expects a recently uploaded file to be used, say the knowledge base may still be indexing and ask for the specific file or detail needed.',
+    ].join('\n');
+  }
+  const lines = [
+    'WTT personal knowledge base context. Use these user-provided references when relevant, cite filenames naturally, and do not claim you read files that are not represented here.',
+    `knowledge_scope: ${options.scope || 'personal'}`,
+    `knowledge_task_id: ${options.taskId || 'unknown'}`,
+  ];
+  rows.slice(0, 8).forEach((row, index) => {
+    const filename = String(row.filename || row.title || row.source_id || `source-${index + 1}`).slice(0, 180);
+    const chunk = row.chunk_index !== undefined ? `#${row.chunk_index}` : `#${index + 1}`;
+    const content = truncateForTranscript(String(row.content || row.snippet || ''), 2400);
+    if (!content.trim()) return;
+    lines.push('');
+    lines.push(`[KB ${index + 1}] ${filename} ${chunk}`);
+    lines.push(content);
+  });
+  return truncateForTranscript(lines.join('\n'), 18000);
+}
+
 function stripHiddenContextLeak(text) {
   return String(text || '')
     .replace(/\[Agent Role Template\][\s\S]*?\[\/Agent Role Template\]\s*/gi, '')
@@ -1490,7 +1560,7 @@ function taskFromEvent(taskId, event) {
   return { id: taskId, title: m.title || `WTT task ${taskId}`, description: m.content || '', topic_id: m.topic_id };
 }
 
-function buildTaskPrompt(task, config, staged = { promptBlock: '' }, transcripts = [], agentProfile = null) {
+function buildTaskPrompt(task, config, staged = { promptBlock: '' }, transcripts = [], agentProfile = null, knowledgeContext = '') {
   const title = task.title || task.name || `Task ${task.id}`;
   const description = task.description || task.content || task.task_request || '';
   const agentSoul = renderAgentSoulContext(task.metadata || task.msg_metadata, agentProfile?.role_template);
@@ -1501,6 +1571,8 @@ function buildTaskPrompt(task, config, staged = { promptBlock: '' }, transcripts
     '',
     agentSoul,
     agentSoul ? '' : null,
+    knowledgeContext,
+    knowledgeContext ? '' : null,
     'Description:',
     description,
     staged.promptBlock,

package/src/wtt-api.js

@@ -92,6 +92,20 @@ export class WTTApi {
     }
   }
 
+  async searchKnowledgeContext(kbTaskId, query, { limit = 8 } = {}) {
+    if (!kbTaskId || !String(query || '').trim()) return { results: [] };
+    const params = new URLSearchParams({ kb_task_id: String(kbTaskId) });
+    try {
+      return await this.request('POST', `/kb/context/search?${params.toString()}`, {
+        headers: this.agentHeaders(),
+        json: { query: String(query || '').slice(0, 4000), limit },
+      });
+    } catch (err) {
+      log('warn', 'kb_context_search failed', { kbTaskId, error: err.message });
+      return { results: [] };
+    }
+  }
+
   async runTask(taskId, payload = {}) {
     return this.request('POST', `/tasks/${taskId}/run`, { json: payload });
   }