@customize-agent/knowledge 4.0.6 → 4.0.7

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

@@ -108,6 +108,11 @@ export declare class KnowledgeBaseManager {
         content: Buffer;
         targetRelativePath?: string;
     }>, operationId?: string): Promise<import("./index-state-store.js").KnowledgeIndexJob[]>;
+    stageUploadedFilePaths(files: Array<{
+        fileName: string;
+        sourcePath: string;
+        targetRelativePath?: string;
+    }>, operationId?: string, offset?: number): Promise<import("./index-state-store.js").KnowledgeIndexJob[]>;
     uploadFiles(files: Array<{
         fileName: string;
         content: Buffer;
@@ -173,6 +178,7 @@ export declare class KnowledgeBaseManager {
     private ensureVectorIndexFresh;
     private hasUsableContent;
     private defaultUploadRelativePath;
+    private validateUploadRelativePath;
     private resolveKbRelativePath;
     private normalizeRelativePath;
 }

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

@@ -426,7 +426,7 @@ export class KnowledgeBaseManager {
         return this.incrementalIndex();
     }
     getUploadRelativePath(fileName, targetRelativePath) {
-        return targetRelativePath ? this.normalizeRelativePath(targetRelativePath) : this.defaultUploadRelativePath(fileName);
+        return this.validateUploadRelativePath(targetRelativePath ? this.normalizeRelativePath(targetRelativePath) : this.defaultUploadRelativePath(fileName));
     }
     async uploadFile(fileName, content, targetRelativePath, onProgress, options = {}) {
         return this.uploadFiles([{ fileName, content, targetRelativePath }], onProgress, options);
@@ -448,6 +448,23 @@ export class KnowledgeBaseManager {
         }
         return jobs;
     }
+    async stageUploadedFilePaths(files, operationId = `upload-${Date.now()}`, offset = 0) {
+        this.initialize();
+        const jobs = [];
+        for (let index = 0; index < files.length; index++) {
+            const file = files[index];
+            const relativePath = this.getUploadRelativePath(file.fileName, file.targetRelativePath);
+            const targetPath = this.resolveKbRelativePath(relativePath);
+            fs.mkdirSync(path.dirname(targetPath), { recursive: true });
+            fs.copyFileSync(file.sourcePath, targetPath);
+            const record = this.store.listRecords().find(item => item.relativePath === relativePath);
+            if (record)
+                await this.deleteVectorFile(record.collectionName, relativePath);
+            this.store.deleteRecord(relativePath);
+            jobs.push(this.store.enqueueIndexJob({ id: `${operationId}-${offset + index}`, relativePath, message: '文件已落盘，等待后台解析' }));
+        }
+        return jobs;
+    }
     async uploadFiles(files, onProgress, options = {}) {
         await this.stageUploadedFiles(files);
         return this.incrementalIndex({ onProgress, vectorMode: options.vectorMode });
@@ -922,6 +939,21 @@ ${resultsText}
         const dir = configDirs[classification.category] ?? DEFAULT_CATEGORY_DIRS[classification.category];
         return `${dir}/${path.basename(fileName)}`;
     }
+    validateUploadRelativePath(relativePath) {
+        const normalized = this.normalizeRelativePath(relativePath);
+        if (!normalized || normalized === '.')
+            throw new Error('上传文件路径无效');
+        if (normalized.length > 1000)
+            throw new Error('上传文件路径过长，请缩短文件夹层级或文件名');
+        if (normalized.includes('\0'))
+            throw new Error('上传文件路径包含非法字符');
+        const parts = normalized.split('/');
+        if (parts.some(part => !part || part === '..'))
+            throw new Error('上传文件路径无效');
+        if (parts.some(part => part.length > 255))
+            throw new Error('上传文件名过长，请缩短文件名后重试');
+        return normalized;
+    }
     resolveKbRelativePath(relativePath) {
         const normalized = this.normalizeRelativePath(relativePath);
         const targetPath = path.resolve(this.kbPath, normalized);

package/package.json

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