@sanqianx/claude-ai-workbench 0.13.2 → 0.13.4

package/packages/server/lib/json-file.js

@@ -1,26 +1,26 @@
-'use strict';
-
-const fs = require('fs');
-const path = require('path');
-const crypto = require('crypto');
-
-function readJson(file, fallback) {
-  try { return JSON.parse(fs.readFileSync(file, 'utf8')); }
-  catch { return fallback; }
-}
-
-function writeJson(file, value, mode = 0o600) {
-  fs.mkdirSync(path.dirname(file), { recursive: true });
-  const temp = `${file}.${process.pid}-${crypto.randomUUID()}.tmp`;
-  try {
-    fs.writeFileSync(temp, `${JSON.stringify(value, null, 2)}\n`, { encoding: 'utf8', mode, flag: 'wx' });
-    // Never unlink the previous record if Windows refuses replacement.
-    // ENOSPC or a locked destination must preserve the last durable history.
-    fs.renameSync(temp, file);
-    try { fs.chmodSync(file, mode); } catch {}
-  } finally {
-    try { fs.unlinkSync(temp); } catch {}
-  }
-}
-
-module.exports = { readJson, writeJson };
+'use strict';
+
+const fs = require('fs');
+const path = require('path');
+const crypto = require('crypto');
+
+function readJson(file, fallback) {
+  try { return JSON.parse(fs.readFileSync(file, 'utf8')); }
+  catch { return fallback; }
+}
+
+function writeJson(file, value, mode = 0o600) {
+  fs.mkdirSync(path.dirname(file), { recursive: true });
+  const temp = `${file}.${process.pid}-${crypto.randomUUID()}.tmp`;
+  try {
+    fs.writeFileSync(temp, `${JSON.stringify(value, null, 2)}\n`, { encoding: 'utf8', mode, flag: 'wx' });
+    // Never unlink the previous record if Windows refuses replacement.
+    // ENOSPC or a locked destination must preserve the last durable history.
+    fs.renameSync(temp, file);
+    try { fs.chmodSync(file, mode); } catch {}
+  } finally {
+    try { fs.unlinkSync(temp); } catch {}
+  }
+}
+
+module.exports = { readJson, writeJson };

package/packages/server/lib/knowledge-context.js

@@ -1,33 +1,33 @@
-'use strict';
-const { readJson, writeJson } = require('./json-file');
-class KnowledgeContext {
-  constructor({ engineUrl, fetchImpl = fetch, contextFile } = {}) { this.engineUrl = engineUrl; this.fetch = fetchImpl; this.contextFile = contextFile; this.contexts = new Map(); }
-  remember(sessionId, value) {
-    const { prompt, ...context } = value;
-    if (this.contextFile) {
-      const data = readJson(this.contextFile, { schema: 'session-knowledge-context/v1', sessions: {} });
-      data.sessions[sessionId] = context; writeJson(this.contextFile, data);
-    } else this.contexts.set(sessionId, context);
-  }
-  saved(sessionId) { return this.contextFile ? readJson(this.contextFile, { sessions: {} }).sessions[sessionId] : this.contexts.get(sessionId); }
-  forget(sessionId) {
-    if (this.contextFile) {
-      const data = readJson(this.contextFile, { schema: 'session-knowledge-context/v1', sessions: {} });
-      if (Object.hasOwn(data.sessions, sessionId)) { delete data.sessions[sessionId]; writeJson(this.contextFile, data); }
-    } else this.contexts.delete(sessionId);
-  }
-  async retrieve(repoPath, query, scope, metadata = {}) {
-    if (!this.engineUrl || !query) return { status: 'disabled', references: [], prompt: '' };
-    try {
-      const target = new URL(this.engineUrl);
-      if (!['http:', 'https:'].includes(target.protocol) || !['localhost', '127.0.0.1', '::1', '[::1]'].includes(target.hostname) || target.username || target.password || target.search || target.hash) throw new Error('KNOWLEDGE_UNAVAILABLE');
-      const response = await this.fetch(new URL('/api/context', target), { method: 'POST', redirect: 'error', headers: { 'Content-Type': 'application/json' }, body: JSON.stringify({ repoPath, query, ...(scope ? { scope } : {}), client: 'workbench', sessionId: metadata.sessionId }), signal: AbortSignal.timeout(10000) });
-      if (response.status === 404) return { status: 'unregistered', references: [], prompt: '' };
-      if (!response.ok) throw new Error('KNOWLEDGE_UNAVAILABLE');
-      const found = await response.json();
-      if (found.ok !== true || !Array.isArray(found.references) || !Array.isArray(found.memories) || typeof found.prompt !== 'string' || found.prompt.length > 8000 || !['ready', 'empty', 'file-fallback'].includes(found.status)) throw new Error('KNOWLEDGE_UNAVAILABLE');
-      return { status: found.status, projectId: found.projectId, scope: found.scope, references: found.references, memoryCount: found.memories.length, memoryStatus: found.memoryStatus, truncated: found.truncated, requestId: found.requestId, retrievedAt: new Date().toISOString(), prompt: found.prompt ? `\n\n${found.prompt}` : '' };
-    } catch { return { status: 'unavailable', references: [], prompt: '\n\n知识库读取未完成，可继续开发；不要将读取失败当作没有相关知识。' }; }
-  }
-}
-module.exports = { KnowledgeContext };
+'use strict';
+const { readJson, writeJson } = require('./json-file');
+class KnowledgeContext {
+  constructor({ engineUrl, fetchImpl = fetch, contextFile } = {}) { this.engineUrl = engineUrl; this.fetch = fetchImpl; this.contextFile = contextFile; this.contexts = new Map(); }
+  remember(sessionId, value) {
+    const { prompt, ...context } = value;
+    if (this.contextFile) {
+      const data = readJson(this.contextFile, { schema: 'session-knowledge-context/v1', sessions: {} });
+      data.sessions[sessionId] = context; writeJson(this.contextFile, data);
+    } else this.contexts.set(sessionId, context);
+  }
+  saved(sessionId) { return this.contextFile ? readJson(this.contextFile, { sessions: {} }).sessions[sessionId] : this.contexts.get(sessionId); }
+  forget(sessionId) {
+    if (this.contextFile) {
+      const data = readJson(this.contextFile, { schema: 'session-knowledge-context/v1', sessions: {} });
+      if (Object.hasOwn(data.sessions, sessionId)) { delete data.sessions[sessionId]; writeJson(this.contextFile, data); }
+    } else this.contexts.delete(sessionId);
+  }
+  async retrieve(repoPath, query, scope, metadata = {}) {
+    if (!this.engineUrl || !query) return { status: 'disabled', references: [], prompt: '' };
+    try {
+      const target = new URL(this.engineUrl);
+      if (!['http:', 'https:'].includes(target.protocol) || !['localhost', '127.0.0.1', '::1', '[::1]'].includes(target.hostname) || target.username || target.password || target.search || target.hash) throw new Error('KNOWLEDGE_UNAVAILABLE');
+      const response = await this.fetch(new URL('/api/context', target), { method: 'POST', redirect: 'error', headers: { 'Content-Type': 'application/json' }, body: JSON.stringify({ repoPath, query, ...(scope ? { scope } : {}), client: 'workbench', sessionId: metadata.sessionId }), signal: AbortSignal.timeout(10000) });
+      if (response.status === 404) return { status: 'unregistered', references: [], prompt: '' };
+      if (!response.ok) throw new Error('KNOWLEDGE_UNAVAILABLE');
+      const found = await response.json();
+      if (found.ok !== true || !Array.isArray(found.references) || !Array.isArray(found.memories) || typeof found.prompt !== 'string' || found.prompt.length > 8000 || !['ready', 'empty', 'file-fallback'].includes(found.status)) throw new Error('KNOWLEDGE_UNAVAILABLE');
+      return { status: found.status, projectId: found.projectId, scope: found.scope, references: found.references, memoryCount: found.memories.length, memoryStatus: found.memoryStatus, truncated: found.truncated, requestId: found.requestId, retrievedAt: new Date().toISOString(), prompt: found.prompt ? `\n\n${found.prompt}` : '' };
+    } catch { return { status: 'unavailable', references: [], prompt: '\n\n知识库读取未完成，可继续开发；不要将读取失败当作没有相关知识。' }; }
+  }
+}
+module.exports = { KnowledgeContext };