@cc-monitoring/proxy-core 0.3.0 → 0.3.1

package/dist/utils/knowledge-store.js

@@ -123,11 +123,23 @@ export function createKnowledgeStore(root) {
         }
         await fs.writeFile(topicPath, JSON.stringify(topicIdx, null, 2), 'utf8');
     }
-    async function searchChunks(query) {
-        if (!query.trim())
-            return [];
-        const hits = [];
+    /** 匹配词集合：整体 query + whitespace 分割的长度 >1 的 token，去重 */
+    function buildTerms(query) {
         const q = query.toLowerCase();
+        const terms = new Set([q]);
+        for (const tok of q.split(/\s+/)) {
+            if (tok.length > 1)
+                terms.add(tok);
+        }
+        return [...terms];
+    }
+    /** 扫描所有 topic/chunk，返回打分候选（searchChunks / queryKnowledge 共用） */
+    async function scanChunks(query) {
+        const q = query.toLowerCase();
+        if (!q.trim())
+            return [];
+        const terms = buildTerms(query);
+        const candidates = [];
         // topic index 本函数作用域缓存，避免每个 chunk 重复读
         const topicIndexCache = new Map();
         let entries;
@@ -147,7 +159,7 @@ export function createKnowledgeStore(root) {
             const topicIdx = topicIndexCache.get(topic) ?? null;
             const topicTags = topicIdx && Array.isArray(topicIdx.tags) ? topicIdx.tags : [];
             const chunkRefs = topicIdx && Array.isArray(topicIdx.chunks) ? topicIdx.chunks : [];
-            const topicTagHit = topicTags.some(t => t.toLowerCase().includes(q));
+            const topicTagHit = (term) => topicTags.some(t => t.toLowerCase().includes(term));
             const chunksDir = path.join(root, topic, 'chunks');
             let files;
             try {
@@ -161,18 +173,54 @@ export function createKnowledgeStore(root) {
                 if (!CHUNK_RE.test(chunkId))
                     continue;
                 const text = await fs.readFile(path.join(chunksDir, file), 'utf8');
+                const textLc = text.toLowerCase();
                 const ref = chunkRefs.find(c => c.id === chunkId);
+                const refTitle = (ref?.title ?? '').toLowerCase();
+                const refSummary = (ref?.summary ?? '').toLowerCase();
                 const refTags = ref && Array.isArray(ref.tags) ? ref.tags : [];
-                const hit = text.toLowerCase().includes(q) ||
-                    (ref?.title ?? '').toLowerCase().includes(q) ||
-                    (ref?.summary ?? '').toLowerCase().includes(q) ||
-                    topicTagHit ||
-                    refTags.some(t => t.toLowerCase().includes(q));
-                if (hit) {
-                    hits.push({ topic, chunkId, title: parseChunkTitle(text) });
+                let score = 0;
+                for (const term of terms) {
+                    // 每个字段每词只计一次分，body 不按出现次数刷分
+                    if (refTitle.includes(term))
+                        score += 3;
+                    if (refTags.some(t => t.toLowerCase().includes(term)))
+                        score += 2;
+                    if (topicTagHit(term))
+                        score += 2;
+                    if (refSummary.includes(term))
+                        score += 1;
+                    if (textLc.includes(term))
+                        score += 1;
                 }
+                if (score === 0)
+                    continue;
+                const wholeQueryHit = textLc.includes(q) ||
+                    refTitle.includes(q) ||
+                    refSummary.includes(q) ||
+                    topicTagHit(q) ||
+                    refTags.some(t => t.toLowerCase().includes(q));
+                candidates.push({ topic, chunkId, title: parseChunkTitle(text), score, wholeQueryHit });
             }
         }
+        return candidates;
+    }
+    async function searchChunks(query) {
+        if (!query.trim())
+            return [];
+        return (await scanChunks(query))
+            .filter(c => c.wholeQueryHit)
+            .map(({ topic, chunkId, title }) => ({ topic, chunkId, title }));
+    }
+    async function queryKnowledge(query, limit = 3) {
+        const capped = Math.min(Math.max(limit, 1), 5);
+        const candidates = (await scanChunks(query))
+            .sort((a, b) => b.score - a.score)
+            .slice(0, capped);
+        const hits = [];
+        for (const c of candidates) {
+            const body = await readChunk(c.topic, c.chunkId);
+            hits.push({ topic: c.topic, chunkId: c.chunkId, title: c.title, body });
+        }
         return hits;
     }
     async function readChunk(topic, chunkId) {
@@ -187,7 +235,7 @@ export function createKnowledgeStore(root) {
         }
         return fs.readFile(resolved, 'utf8');
     }
-    return { listTopics, getTopic, searchChunks, readChunk, upsertTopic, upsertChunkRef };
+    return { listTopics, getTopic, searchChunks, queryKnowledge, readChunk, upsertTopic, upsertChunkRef };
 }
 // 默认单例：绑生产 docsDir
 export const knowledgeStore = createKnowledgeStore(docsDir);

package/dist/utils/request-writer.d.ts

@@ -42,6 +42,7 @@ export interface RequestSummary {
     baseUrl?: string;
     model?: string;
     machineId?: string;
+    userAgent?: string;
     usage?: TokenUsage;
 }
 export declare function extractTokenUsage(events: any[] | undefined): TokenUsage | undefined;

package/dist/utils/request-writer.js

@@ -60,17 +60,17 @@ export async function saveRequestDB(sessionId, requestId, data) {
     const stmt = db.prepare(`
     INSERT OR REPLACE INTO requests (
       request_id, session_id, timestamp, method, path, status_code, response_time,
-      api_key, proxy_api_key, base_url, model, machine_id,
+      api_key, proxy_api_key, base_url, model, machine_id, user_agent,
       input_tokens, output_tokens, cache_read_tokens, cache_creation_tokens,
       request_headers, request_body, response_headers, response_events,
       completed_at, closed_at, error
     ) VALUES (
-      ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?
+      ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?
     )
   `);
     stmt.run(requestId, sessionId, data.timestamp, data.request.method, data.request.path, data.response?.statusCode ?? 0, data.response?.completedAt && data.timestamp
         ? Math.round(new Date(data.response.completedAt).getTime() - new Date(data.timestamp).getTime())
-        : 0, data.apiKey || null, data.proxyApiKey || null, data.baseUrl || null, data.model || null, data.machineId || null, usage?.input_tokens || null, usage?.output_tokens || null, usage?.cache_read_input_tokens || null, usage?.cache_creation_input_tokens || null, JSON.stringify(data.request.headers), data.request.body ? JSON.stringify(data.request.body) : null, data.response?.headers ? JSON.stringify(data.response.headers) : null, null, // response_events now stored in separate table
+        : 0, data.apiKey || null, data.proxyApiKey || null, data.baseUrl || null, data.model || null, data.machineId || null, data.request.headers['user-agent'] || null, usage?.input_tokens || null, usage?.output_tokens || null, usage?.cache_read_input_tokens || null, usage?.cache_creation_input_tokens || null, JSON.stringify(data.request.headers), data.request.body ? JSON.stringify(data.request.body) : null, data.response?.headers ? JSON.stringify(data.response.headers) : null, null, // response_events now stored in separate table
     data.response?.completedAt || null, data.response?.closedAt || null, data.response?.error || null);
 }
 // Update token columns for a request (streaming: usage extracted from tailBuffer after save)
@@ -164,7 +164,7 @@ export async function listRequestDB(limit = 1000, before) {
     const rows = db.prepare(`
     SELECT
       request_id, session_id, timestamp, method, path, status_code, response_time,
-      api_key, proxy_api_key, base_url, model, machine_id,
+      api_key, proxy_api_key, base_url, model, machine_id, user_agent,
       input_tokens, output_tokens, cache_read_tokens, cache_creation_tokens
     FROM requests
     ${whereClause}
@@ -188,6 +188,7 @@ export async function listRequestDB(limit = 1000, before) {
             baseUrl: row.base_url,
             model: row.model,
             machineId: row.machine_id,
+            userAgent: row.user_agent,
             usage: row.input_tokens || row.output_tokens ? {
                 input_tokens: row.input_tokens,
                 output_tokens: row.output_tokens,

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@cc-monitoring/proxy-core",
-  "version": "0.3.0",
+  "version": "0.3.1",
   "description": "Claude Code 代理工具 + Token 消耗监控面板",
   "keywords": [
     "claude-code-proxy",