@customize-agent/knowledge 4.0.28 → 4.0.29

package/dist/core/knowledge-base-manager.d.ts

@@ -72,6 +72,7 @@ export declare class KnowledgeBaseManager {
         filters?: SearchFilters;
         collections?: string[];
         weights?: RetrievalWeights;
+        generationMode?: boolean;
     }): Promise<FederatedResult>;
     semanticSearch(query: string, options?: {
         limit?: number;

package/dist/core/knowledge-base-manager.js

@@ -339,13 +339,16 @@ export class KnowledgeBaseManager {
         const limit = options.limit ?? 10;
         const start = Date.now();
         const weights = this.retrievalWeights(options.weights);
-        const rewrittenQueries = await this.rewriteQueries(query);
+        const rewrittenQueries = options.generationMode ? [query.trim()].filter(Boolean) : await this.rewriteQueries(query);
         const rankedLists = [];
+        const keywordMultiplier = options.generationMode ? 2 : 3;
+        const vectorMultiplier = options.generationMode ? 3 : 6;
+        const vectorQueryLimit = options.generationMode ? 1 : 3;
         for (const [queryIndex, rewritten] of rewrittenQueries.entries()) {
-            rankedLists.push({ source: 'keyword', items: this.keywordSearchItems(rewritten, limit * 3), queryIndex });
-            if (queryIndex < 3) {
+            rankedLists.push({ source: 'keyword', items: this.keywordSearchItems(rewritten, limit * keywordMultiplier), queryIndex });
+            if (queryIndex < vectorQueryLimit) {
                 try {
-                    rankedLists.push({ source: 'vector', items: (await this.semanticSearch(rewritten, { ...options, limit: limit * 6 })).results.slice(0, limit * 3), queryIndex });
+                    rankedLists.push({ source: 'vector', items: (await this.semanticSearch(rewritten, { ...options, limit: limit * vectorMultiplier })).results.slice(0, limit * keywordMultiplier), queryIndex });
                 }
                 catch { /* 向量搜索在混合搜索中是可选的 */ }
             }

package/dist/core/multi-project-manager.d.ts

@@ -26,6 +26,8 @@ export declare class MultiProjectManager {
         limit?: number;
         scope?: SearchScope;
         weights?: RetrievalWeights;
+        generationMode?: boolean;
+        skipFreshCheck?: boolean;
     }): Promise<FederatedResult>;
     semanticSearch(projectRoot: string, query: string, options?: {
         limit?: number;

package/dist/core/multi-project-manager.js

@@ -64,18 +64,19 @@ export class MultiProjectManager {
         const limit = options.limit ?? 10;
         const scope = options.scope ?? 'project';
         const project = await this.getProject(projectRoot);
-        await this.ensureFreshForSearch(projectRoot, project);
+        if (!options.skipFreshCheck)
+            await this.ensureFreshForSearch(projectRoot, project);
         if (scope === 'project')
-            return project.hybridSearch(query, { limit, weights: options.weights });
+            return project.hybridSearch(query, { limit, weights: options.weights, generationMode: options.generationMode });
         const projectResults = scope === 'all'
-            ? await project.hybridSearch(query, { limit, weights: options.weights })
+            ? await project.hybridSearch(query, { limit, weights: options.weights, generationMode: options.generationMode })
             : { results: [], scopesSearched: [], queryTimeMs: 0 };
         if (scope === 'global') {
             const global = await this.getGlobalKB();
-            return global.hybridSearch(query, { limit, weights: options.weights });
+            return global.hybridSearch(query, { limit, weights: options.weights, generationMode: options.generationMode });
         }
         const global = await this.getGlobalKB();
-        const globalResults = await global.hybridSearch(query, { limit, weights: options.weights });
+        const globalResults = await global.hybridSearch(query, { limit, weights: options.weights, generationMode: options.generationMode });
         const merged = new FederationSearch().merge([...projectResults.results, ...globalResults.results], limit, 'all');
         return {
             ...merged,

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@customize-agent/knowledge",
-  "version": "4.0.28",
+  "version": "4.0.29",
   "description": "Local knowledge base infrastructure for customize-agent",
   "type": "module",
   "main": "./dist/index.js",