wtt-connect 0.2.60 → 0.2.61

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "wtt-connect",
-  "version": "0.2.60",
+  "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);
@@ -218,6 +235,8 @@ export class Runner {
         discussionRouting ? '' : null,
         agentSoul,
         agentSoul ? '' : null,
+        knowledgeContext,
+        knowledgeContext ? '' : null,
         renderCompactSummaryBlock(compactSummary),
         compactSummary ? '' : null,
         'User message:',
@@ -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)}`,
@@ -1423,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, '')
@@ -1498,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);
@@ -1509,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 });
   }