@customize-agent/knowledge 1.0.1

package/dist/server/dashboard-styles.js

@@ -0,0 +1,152 @@
+export const dashboardStyles = `
+:root {
+  color-scheme: light;
+  --bg:#f5f5f7;
+  --bg-glass:rgba(255,255,255,.72);
+  --panel:rgba(255,255,255,.86);
+  --panel-solid:#ffffff;
+  --line:rgba(60,60,67,.16);
+  --line-strong:rgba(60,60,67,.26);
+  --text:#1d1d1f;
+  --muted:#6e6e73;
+  --accent:#007aff;
+  --accent-2:#5856d6;
+  --danger:#ff3b30;
+  --ok:#34c759;
+  --warn:#ff9500;
+  --shadow:0 18px 45px rgba(0,0,0,.08);
+  --shadow-soft:0 8px 24px rgba(0,0,0,.06);
+}
+* { box-sizing: border-box; }
+html { background:var(--bg); }
+body {
+  margin:0; min-height:100vh; font-family: -apple-system, BlinkMacSystemFont, "SF Pro Display", "SF Pro Text", "Segoe UI", sans-serif;
+  background:
+    radial-gradient(circle at 15% 0%, rgba(0,122,255,.12), transparent 28rem),
+    radial-gradient(circle at 85% 8%, rgba(88,86,214,.10), transparent 26rem),
+    linear-gradient(180deg, #fbfbfd 0%, var(--bg) 100%);
+  color:var(--text);
+}
+header {
+  position: sticky; top:0; z-index:10; backdrop-filter: saturate(180%) blur(22px); -webkit-backdrop-filter: saturate(180%) blur(22px);
+  padding:20px max(32px, calc((100vw - 1480px) / 2 + 32px)); border-bottom:1px solid var(--line); display:flex; align-items:center; justify-content:space-between; gap:18px;
+  background:var(--bg-glass);
+}
+main { padding:28px 32px 40px; max-width:1480px; margin:0 auto; }
+main, aside, main > div { min-width:0; }
+.page { display:none; }
+.page.active { display:block; }
+.topbar { flex-wrap:nowrap; justify-content:flex-start; overflow-x:auto; padding-bottom:2px; scrollbar-width:none; }
+.topbar::-webkit-scrollbar { display:none; }
+.topbar > * { flex:0 0 auto; }
+.tab { background:rgba(255,255,255,.72); border-radius:999px; padding:9px 15px; color:var(--muted); }
+.tab.active { background:#1d1d1f; color:white; border-color:#1d1d1f; }
+section {
+  position:relative; overflow:visible;
+  background:var(--panel); border:1px solid rgba(255,255,255,.7); border-radius:26px; padding:20px; box-shadow:var(--shadow-soft);
+  backdrop-filter: blur(16px); -webkit-backdrop-filter: blur(16px);
+}
+section:has(.custom-select.open), section.select-host-open { z-index:10000; }
+section:hover { box-shadow:var(--shadow); transform:translateY(-1px); transition:.2s ease; }
+h1 { font-size:24px; margin:0; letter-spacing:-.02em; font-weight:700; }
+h2 { font-size:12px; margin:0 0 14px; color:var(--muted); text-transform:uppercase; letter-spacing:.12em; font-weight:700; }
+h3 { font-size:15px; margin:14px 0 8px; }
+.subtitle { margin-top:6px; color:var(--muted); font-size:14px; }
+button, input, select {
+  min-width:0; background:rgba(255,255,255,.88); color:var(--text); border:1px solid var(--line); border-radius:14px; padding:10px 13px; outline:none;
+  font: inherit; box-shadow: inset 0 0 0 1px rgba(255,255,255,.25);
+}
+select {
+  min-width:132px; appearance:none; -webkit-appearance:none; padding-right:34px;
+  background-image:linear-gradient(45deg, transparent 50%, #6e6e73 50%), linear-gradient(135deg, #6e6e73 50%, transparent 50%);
+  background-position:calc(100% - 18px) 50%, calc(100% - 12px) 50%; background-size:6px 6px, 6px 6px; background-repeat:no-repeat;
+}
+button { cursor:pointer; white-space:nowrap; transition:.16s ease; font-weight:600; }
+button:hover { border-color:rgba(0,122,255,.35); transform:translateY(-1px); box-shadow:0 8px 18px rgba(0,122,255,.12); }
+button.primary { background:linear-gradient(180deg, #0a84ff, #007aff); border-color:transparent; color:white; box-shadow:0 10px 22px rgba(0,122,255,.24); }
+input { width:100%; }
+input::placeholder { color:#a1a1a6; }
+input:focus, select:focus { border-color:rgba(0,122,255,.55); box-shadow:0 0 0 4px rgba(0,122,255,.12); }
+.row { display:flex; gap:10px; margin-bottom:10px; align-items:center; }
+.row input { flex:1; }
+.stack { display:grid; gap:10px; }
+.stack.compact { gap:8px; }
+.field-row { display:grid; grid-template-columns:minmax(0, 1fr) auto; gap:8px; align-items:center; }
+.field-row:has(input + input + button) { grid-template-columns:minmax(0, 1fr) minmax(0, 1fr) auto; }
+.helper-title { font-size:15px; font-weight:700; color:var(--text); margin:2px 0 6px; }
+.helper-text { color:var(--muted); font-size:13px; line-height:1.6; margin-bottom:12px; overflow-wrap:anywhere; }
+.upload-card { display:flex; gap:14px; align-items:center; border:1.5px dashed rgba(0,122,255,.32); background:rgba(0,122,255,.06); border-radius:20px; padding:16px; cursor:pointer; margin-bottom:10px; }
+.upload-card:hover { background:rgba(0,122,255,.10); border-color:rgba(0,122,255,.55); }
+button.loading, .upload-card.loading { position:relative; pointer-events:none; opacity:.82; cursor:progress; }
+button.loading::after, .upload-card.loading::after { content:''; width:14px; height:14px; border:2px solid currentColor; border-top-color:transparent; border-radius:50%; display:inline-block; margin-left:8px; vertical-align:-2px; animation:spin .8s linear infinite; }
+.upload-card.loading::after { margin-left:auto; color:#007aff; width:18px; height:18px; }
+button:disabled { cursor:progress; }
+@keyframes spin { to { transform:rotate(360deg); } }
+.upload-card input { display:none; }
+.upload-card b { display:block; font-size:15px; }
+.upload-card span { display:block; color:var(--muted); font-size:12px; line-height:1.45; margin-top:3px; overflow-wrap:anywhere; }
+.upload-icon { width:38px; height:38px; flex:0 0 38px; border-radius:50%; display:grid; place-items:center; background:#007aff; color:white; font-weight:800; font-size:20px; }
+.advanced-block { margin-top:12px; border:1px solid var(--line); border-radius:16px; padding:10px 12px; background:rgba(255,255,255,.55); }
+.advanced-block summary { cursor:pointer; color:var(--muted); font-size:13px; font-weight:600; margin-bottom:8px; }
+.divider { height:1px; background:var(--line); margin:16px 0; }
+.search-bar { display:grid; grid-template-columns: 170px minmax(0, 1fr) auto; gap:10px; align-items:end; }
+.filter-bar { display:grid; grid-template-columns: repeat(2, minmax(180px, 260px)); gap:10px; align-items:end; margin:12px 0 14px; }
+.select-wrap { position:relative; display:grid; gap:5px; color:var(--muted); font-size:12px; font-weight:600; min-width:0; overflow:visible; }
+.select-wrap.compact { min-width:130px; }
+.language-select { width:96px; min-width:96px; }
+.language-select select { min-width:96px; width:96px; cursor:pointer; }
+.select-wrap select { width:100%; }
+.select-native-hidden { position:absolute !important; opacity:0 !important; pointer-events:none !important; width:1px !important; height:1px !important; }
+.custom-select { position:relative; min-width:0; z-index:1; }
+.custom-select.open { z-index:10001; }
+.custom-select-trigger { width:100%; justify-content:space-between; text-align:left; display:flex; align-items:center; gap:10px; background:rgba(255,255,255,.88); color:var(--text); border:1px solid var(--line); border-radius:14px; padding:10px 13px; box-shadow: inset 0 0 0 1px rgba(255,255,255,.25); }
+.custom-select-trigger::after { content:'⌄'; color:var(--muted); font-size:13px; line-height:1; display:inline-flex; align-items:center; justify-content:center; transform:translateY(-1px); flex:0 0 auto; }
+.custom-select.open .custom-select-trigger { border-color:rgba(0,122,255,.55); box-shadow:0 0 0 4px rgba(0,122,255,.12); }
+.custom-select-menu { display:none; position:absolute; top:calc(100% + 6px); left:0; right:0; z-index:10002; max-height:260px; overflow:auto; padding:6px; border:1px solid var(--line); border-radius:16px; background:rgba(255,255,255,.98); box-shadow:0 18px 40px rgba(0,0,0,.16); backdrop-filter:blur(16px); }
+.custom-select.open .custom-select-menu { display:block; }
+.custom-select-option { display:block; width:100%; border:0; box-shadow:none; background:transparent; text-align:left; color:var(--text); border-radius:10px; padding:9px 10px; white-space:normal; }
+.custom-select-option:hover, .custom-select-option.active { background:rgba(0,122,255,.10); color:var(--accent); transform:none; box-shadow:none; }
+.config-row { display:grid; grid-template-columns:90px minmax(0, 1fr); gap:10px; align-items:start; padding:8px 0; border-top:1px solid var(--line); }
+.config-row:first-child { border-top:0; }
+.config-row b, .config-row span { min-width:0; overflow-wrap:anywhere; line-height:1.5; }
+.file-group { border:1px solid var(--line); border-radius:18px; background:rgba(255,255,255,.58); margin-top:10px; overflow:hidden; }
+.file-group summary { cursor:pointer; padding:12px 14px; font-weight:700; display:flex; justify-content:space-between; align-items:center; }
+.file-group summary span { color:var(--muted); font-size:12px; }
+.file-item { display:grid; grid-template-columns:minmax(0, 1fr) auto; gap:12px; padding:12px 14px; border-top:1px solid var(--line); align-items:center; }
+.file-main { min-width:0; }
+.file-actions { display:flex; gap:6px; flex-wrap:wrap; justify-content:flex-end; }
+.file-actions button { padding:7px 9px; font-size:12px; }
+.grid { display:grid; gap:20px; grid-template-columns: repeat(2, minmax(0, 1fr)); }
+.panel-gap { margin-top:20px; }
+.stat-grid { display:grid; grid-template-columns: repeat(3, 1fr); gap:10px; margin:10px 0 12px; }
+.metric { background:linear-gradient(180deg, #fff, #f7f7fa); border:1px solid var(--line); border-radius:18px; padding:14px; }
+.metric b { display:block; font-size:24px; margin-top:4px; letter-spacing:-.03em; }
+.metric span { color:var(--muted); font-size:12px; }
+.stat { display:flex; justify-content:space-between; align-items:flex-start; color:var(--muted); margin:8px 0; gap:12px; }
+.stat b { color:var(--text); }
+.card { border-top:1px solid var(--line); padding:14px 0; }
+.card:first-child { border-top:0; padding-top:0; }
+.path { color:var(--accent); font-size:13px; overflow-wrap:anywhere; word-break:normal; font-weight:600; line-height:1.45; }
+.meta { color:var(--muted); font-size:12px; overflow-wrap:anywhere; word-break:normal; line-height:1.55; min-width:0; }
+.badge { display:inline-flex; align-items:center; border:1px solid var(--line); border-radius:999px; padding:3px 9px; font-size:12px; color:var(--muted); margin:3px 5px 3px 0; background:rgba(255,255,255,.7); }
+.badge.ok { color:#1d7f3a; border-color:rgba(52,199,89,.28); background:rgba(52,199,89,.10); }
+.badge.warn { color:#9a5a00; border-color:rgba(255,149,0,.30); background:rgba(255,149,0,.10); }
+#capabilities { max-height: 460px; overflow:auto; padding-right:4px; }
+#results .card { background:var(--panel-solid); border-radius:20px; border:1px solid var(--line); margin-top:12px; padding:16px; box-shadow:var(--shadow-soft); }
+.status { color:var(--muted); min-height:20px; font-size:12px; margin-top:8px; }
+.status.ok { color:var(--ok); }
+.status.error { color:var(--danger); }
+.toolbar { display:flex; gap:10px; align-items:center; justify-content:flex-end; }
+pre { white-space:pre-wrap; word-break:break-word; color:#3a3a3c; margin:10px 0 0; font-family: ui-monospace, SFMono-Regular, Menlo, Consolas, monospace; font-size:12px; line-height:1.55; }
+.subsection { margin-top:18px; padding-top:16px; border-top:1px solid var(--line); }
+.modal-backdrop { position:fixed; inset:0; z-index:20000; display:grid; place-items:center; background:rgba(15,23,42,.35); backdrop-filter:blur(10px); }
+.modal-backdrop[hidden] { display:none; }
+.modal-card { width:min(520px, calc(100vw - 32px)); background:rgba(255,255,255,.98); border:1px solid var(--line); border-radius:24px; box-shadow:0 28px 70px rgba(0,0,0,.22); padding:22px; }
+.modal-card h3 { margin:0 0 8px; }
+.modal-card p { white-space:pre-wrap; color:var(--muted); margin:0 0 14px; }
+.modal-input { width:100%; box-sizing:border-box; margin:0 0 14px; border:1px solid var(--line); border-radius:14px; padding:11px 12px; font:inherit; }
+.modal-actions { display:flex; justify-content:flex-end; gap:10px; }
+button.danger { background:linear-gradient(135deg,#ff3b30,#ff6b5f); color:white; border-color:rgba(255,59,48,.25); }
+@media (max-width: 1080px) { .grid { grid-template-columns:1fr; } }
+@media (max-width: 680px) { main { padding:16px; } header { padding:16px; align-items:flex-start; flex-direction:column; } .toolbar { width:100%; justify-content:flex-start; flex-wrap:wrap; } .row { flex-direction:column; align-items:stretch; } .search-bar, .filter-bar { grid-template-columns:1fr; } select { width:100%; } .field-row, .field-row:has(input + input + button), .file-item, .config-row { grid-template-columns:1fr; } .file-actions { justify-content:flex-start; } .stat-grid { grid-template-columns:1fr; } .upload-card { align-items:flex-start; } }
+`;

package/dist/types.d.ts

@@ -0,0 +1,82 @@
+export type KBScope = 'project' | 'global' | 'session';
+export type FileCategory = 'document' | 'spreadsheet' | 'image' | 'cad' | 'code' | 'data' | 'web' | 'diagram' | 'archive' | 'other';
+export type ProjectStatus = 'active' | 'idle' | 'error';
+export interface ProjectInfo {
+    projectId: string;
+    projectRoot: string;
+    projectName?: string;
+    kbPath: string;
+    fileCount: number;
+    chunkCount: number;
+    totalSizeBytes: number;
+    lastIndexedAt: number;
+    lastOpenedAt: number;
+    status: ProjectStatus;
+}
+export interface ProjectConfig {
+    projectId: string;
+    projectName?: string;
+    enabled: boolean;
+    includeGlobal: boolean;
+    priorityOverGlobal: boolean;
+    watch: boolean;
+    autoIndex: boolean;
+    kbignore: string[];
+    projectTags: string[];
+    categoryDirs: Partial<Record<FileCategory, string>>;
+    createdAt: number;
+    lastOpenedAt: number;
+}
+export interface ClassifiedFile {
+    absolutePath: string;
+    relativePath: string;
+    category: FileCategory;
+    format: string;
+    fileSize: number;
+    mtime: number;
+    mimeType: string;
+}
+export interface IndexStateRecord {
+    relativePath: string;
+    category: FileCategory;
+    format: string;
+    contentHash: string;
+    fileSize: number;
+    mtime: number;
+    chunkCount: number;
+    collectionName: string;
+    indexedAt: number;
+    lastVerifiedAt: number;
+    status: 'active' | 'outdated' | 'error' | 'deleted';
+    errorMessage?: string;
+    metadataJson?: string;
+}
+export interface DiffResult {
+    newFiles: ClassifiedFile[];
+    modifiedFiles: ClassifiedFile[];
+    deletedFiles: IndexStateRecord[];
+    unchangedCount: number;
+    mtimeOnlyCount: number;
+    skippedFiles: Array<{
+        file: ClassifiedFile;
+        reason: string;
+    }>;
+    hasChanges: boolean;
+    diffTimeMs: number;
+}
+export interface KnowledgeBaseStats {
+    scope: Exclude<KBScope, 'session'>;
+    projectId?: string;
+    fileCount: number;
+    chunkCount: number;
+    totalSizeBytes: number;
+    lastIndexedAt: number;
+}
+export interface CrossProjectDuplicate {
+    contentHash: string;
+    files: Array<{
+        projectId: string;
+        projectRoot: string;
+        relativePath: string;
+    }>;
+}

package/dist/types.js

@@ -0,0 +1 @@
+export {};

package/dist/vector/chroma-store.d.ts

@@ -0,0 +1,39 @@
+import type { CollectionClient, VectorCollectionInfo, VectorDocument, VectorSearchQuery, VectorSearchResult, VectorStoreInterface } from './types.js';
+interface ChromaQueryResponse {
+    ids?: string[][];
+    documents?: string[][];
+    metadatas?: Array<Array<Record<string, unknown>>>;
+    distances?: number[][];
+}
+export interface ChromaClientOptions {
+    baseUrl?: string;
+    tenant?: string;
+    database?: string;
+}
+export declare class ChromaHttpClient implements CollectionClient {
+    readonly baseUrl: string;
+    readonly tenant: string;
+    readonly database: string;
+    constructor(options?: ChromaClientOptions);
+    heartbeat(): Promise<boolean>;
+    getOrCreateCollection(name: string, metadata?: Record<string, unknown>): Promise<VectorCollectionInfo>;
+    listCollections(): Promise<VectorCollectionInfo[]>;
+    deleteCollection(name: string): Promise<void>;
+    upsert(collectionName: string, documents: VectorDocument[]): Promise<void>;
+    deleteWhere(collectionName: string, where: Record<string, string | number | boolean>): Promise<void>;
+    query(collectionName: string, query: VectorSearchQuery): Promise<ChromaQueryResponse>;
+    private collectionsPath;
+    private request;
+    private toCollectionInfo;
+}
+export declare class ChromaVectorStore implements VectorStoreInterface {
+    private readonly client;
+    readonly collectionName: string;
+    constructor(client: ChromaHttpClient, collectionName: string);
+    ensureCollection(metadata?: Record<string, unknown>): Promise<void>;
+    upsert(documents: VectorDocument[]): Promise<void>;
+    deleteByFilePath(filePath: string): Promise<void>;
+    search(query: VectorSearchQuery): Promise<VectorSearchResult[]>;
+    private toMetadata;
+}
+export {};

package/dist/vector/chroma-store.js

@@ -0,0 +1,131 @@
+export class ChromaHttpClient {
+    baseUrl;
+    tenant;
+    database;
+    constructor(options = {}) {
+        this.baseUrl = options.baseUrl ?? 'http://localhost:8000';
+        this.tenant = options.tenant ?? 'default_tenant';
+        this.database = options.database ?? 'default_database';
+    }
+    async heartbeat() {
+        try {
+            await this.request('/api/v1/heartbeat');
+            return true;
+        }
+        catch {
+            return false;
+        }
+    }
+    async getOrCreateCollection(name, metadata = {}) {
+        const response = await this.request(this.collectionsPath(), {
+            method: 'POST',
+            body: JSON.stringify({ name, metadata, get_or_create: true }),
+        });
+        return this.toCollectionInfo(response);
+    }
+    async listCollections() {
+        const response = await this.request(this.collectionsPath());
+        return response.map(collection => this.toCollectionInfo(collection));
+    }
+    async deleteCollection(name) {
+        await this.request(`${this.collectionsPath()}/${encodeURIComponent(name)}`, { method: 'DELETE' });
+    }
+    async upsert(collectionName, documents) {
+        if (documents.length === 0)
+            return;
+        await this.getOrCreateCollection(collectionName);
+        await this.request(`${this.collectionsPath()}/${encodeURIComponent(collectionName)}/upsert`, {
+            method: 'POST',
+            body: JSON.stringify({
+                ids: documents.map(document => document.id),
+                embeddings: documents.map(document => document.embedding),
+                documents: documents.map(document => document.content),
+                metadatas: documents.map(document => document.metadata),
+            }),
+        });
+    }
+    async deleteWhere(collectionName, where) {
+        await this.request(`${this.collectionsPath()}/${encodeURIComponent(collectionName)}/delete`, {
+            method: 'POST',
+            body: JSON.stringify({ where }),
+        });
+    }
+    async query(collectionName, query) {
+        return this.request(`${this.collectionsPath()}/${encodeURIComponent(collectionName)}/query`, {
+            method: 'POST',
+            body: JSON.stringify({
+                query_embeddings: [query.queryEmbedding],
+                n_results: query.topK,
+                where: query.where,
+                include: ['documents', 'metadatas', 'distances'],
+            }),
+        });
+    }
+    collectionsPath() {
+        return `/api/v1/tenants/${encodeURIComponent(this.tenant)}/databases/${encodeURIComponent(this.database)}/collections`;
+    }
+    async request(path, init = {}) {
+        const response = await fetch(`${this.baseUrl}${path}`, {
+            ...init,
+            headers: {
+                'content-type': 'application/json',
+                ...init.headers,
+            },
+        });
+        if (!response.ok) {
+            throw new Error(`ChromaDB request failed: ${response.status} ${response.statusText}`);
+        }
+        if (response.status === 204)
+            return undefined;
+        return await response.json();
+    }
+    toCollectionInfo(collection) {
+        return {
+            id: collection.id,
+            name: collection.name,
+            metadata: collection.metadata,
+        };
+    }
+}
+export class ChromaVectorStore {
+    client;
+    collectionName;
+    constructor(client, collectionName) {
+        this.client = client;
+        this.collectionName = collectionName;
+    }
+    async ensureCollection(metadata) {
+        await this.client.getOrCreateCollection(this.collectionName, metadata);
+    }
+    async upsert(documents) {
+        await this.client.upsert(this.collectionName, documents);
+    }
+    async deleteByFilePath(filePath) {
+        await this.client.deleteWhere(this.collectionName, { file_path: filePath });
+    }
+    async search(query) {
+        const response = await this.client.query(this.collectionName, query);
+        const ids = response.ids?.[0] ?? [];
+        const documents = response.documents?.[0] ?? [];
+        const metadatas = response.metadatas?.[0] ?? [];
+        const distances = response.distances?.[0] ?? [];
+        return ids.map((id, index) => ({
+            collection: this.collectionName,
+            score: 1 - Number(distances[index] ?? 1),
+            document: {
+                id,
+                content: documents[index] ?? '',
+                metadata: this.toMetadata(metadatas[index] ?? {}),
+            },
+        }));
+    }
+    toMetadata(metadata) {
+        const result = {};
+        for (const [key, value] of Object.entries(metadata)) {
+            if (typeof value === 'string' || typeof value === 'number' || typeof value === 'boolean' || value === null) {
+                result[key] = value;
+            }
+        }
+        return result;
+    }
+}

package/dist/vector/collection-manager.d.ts

@@ -0,0 +1,16 @@
+import type { FileCategory } from '../types.js';
+import type { CollectionClient } from './types.js';
+export declare function projectCollectionName(projectId: string, category: FileCategory): string;
+export declare function globalCollectionName(category: FileCategory): string;
+export declare class CollectionManager {
+    private readonly client?;
+    constructor(client?: CollectionClient | undefined);
+    ensureProjectCollections(projectId: string): Promise<void>;
+    ensureGlobalCollections(): Promise<void>;
+    deleteProjectCollections(projectId: string): Promise<number>;
+    listProjectsFromClient(): Promise<string[]>;
+    getProjectCollections(projectId: string): string[];
+    getGlobalCollections(): string[];
+    getCollectionName(scope: 'project' | 'global', category: FileCategory, projectId?: string): string;
+    listProjectsFromCollectionNames(collectionNames: string[]): string[];
+}

package/dist/vector/collection-manager.js

@@ -0,0 +1,77 @@
+import { ALL_CATEGORIES, COLLECTION_CATEGORY_NAMES } from '../constants.js';
+export function projectCollectionName(projectId, category) {
+    return `proj_${projectId}_kb_${COLLECTION_CATEGORY_NAMES[category]}`;
+}
+export function globalCollectionName(category) {
+    return `global_kb_${COLLECTION_CATEGORY_NAMES[category]}`;
+}
+export class CollectionManager {
+    client;
+    constructor(client) {
+        this.client = client;
+    }
+    async ensureProjectCollections(projectId) {
+        if (!this.client)
+            return;
+        for (const category of ALL_CATEGORIES) {
+            await this.client.getOrCreateCollection(projectCollectionName(projectId, category), {
+                'hnsw:space': 'cosine',
+                project_id: projectId,
+                category,
+            });
+        }
+    }
+    async ensureGlobalCollections() {
+        if (!this.client)
+            return;
+        for (const category of ALL_CATEGORIES) {
+            await this.client.getOrCreateCollection(globalCollectionName(category), {
+                'hnsw:space': 'cosine',
+                scope: 'global',
+                category,
+            });
+        }
+    }
+    async deleteProjectCollections(projectId) {
+        if (!this.client)
+            return 0;
+        const prefix = `proj_${projectId}_`;
+        const collections = await this.client.listCollections();
+        let deleted = 0;
+        for (const collection of collections) {
+            if (collection.name.startsWith(prefix)) {
+                await this.client.deleteCollection(collection.name);
+                deleted += 1;
+            }
+        }
+        return deleted;
+    }
+    async listProjectsFromClient() {
+        if (!this.client)
+            return [];
+        const collections = await this.client.listCollections();
+        return this.listProjectsFromCollectionNames(collections.map(collection => collection.name));
+    }
+    getProjectCollections(projectId) {
+        return ALL_CATEGORIES.map(category => projectCollectionName(projectId, category));
+    }
+    getGlobalCollections() {
+        return ALL_CATEGORIES.map(category => globalCollectionName(category));
+    }
+    getCollectionName(scope, category, projectId) {
+        if (scope === 'global')
+            return globalCollectionName(category);
+        if (!projectId)
+            throw new Error('project scope requires projectId');
+        return projectCollectionName(projectId, category);
+    }
+    listProjectsFromCollectionNames(collectionNames) {
+        const projectIds = new Set();
+        for (const name of collectionNames) {
+            const match = name.match(/^proj_([a-f0-9]{12})_kb_/);
+            if (match?.[1])
+                projectIds.add(match[1]);
+        }
+        return [...projectIds];
+    }
+}

package/dist/vector/types.d.ts

@@ -0,0 +1,33 @@
+export interface VectorDocument {
+    id: string;
+    content: string;
+    embedding: number[];
+    metadata: Record<string, string | number | boolean | null>;
+}
+export interface VectorSearchQuery {
+    queryEmbedding: number[];
+    topK: number;
+    where?: Record<string, string | number | boolean>;
+}
+export interface VectorSearchResult {
+    document: Omit<VectorDocument, 'embedding'>;
+    score: number;
+    collection: string;
+}
+export interface VectorCollectionInfo {
+    id?: string;
+    name: string;
+    metadata?: Record<string, unknown>;
+}
+export interface VectorStoreInterface {
+    readonly collectionName: string;
+    ensureCollection(metadata?: Record<string, unknown>): Promise<void>;
+    upsert(documents: VectorDocument[]): Promise<void>;
+    deleteByFilePath(filePath: string): Promise<void>;
+    search(query: VectorSearchQuery): Promise<VectorSearchResult[]>;
+}
+export interface CollectionClient {
+    getOrCreateCollection(name: string, metadata?: Record<string, unknown>): Promise<VectorCollectionInfo>;
+    listCollections(): Promise<VectorCollectionInfo[]>;
+    deleteCollection(name: string): Promise<void>;
+}

package/dist/vector/types.js

@@ -0,0 +1 @@
+export {};

package/dist/vector/vector-indexer.d.ts

@@ -0,0 +1,18 @@
+import type { EmbeddingProvider } from '../embedding/embedding-provider.js';
+import type { StoredChunk } from '../core/index-state-store.js';
+import type { VectorStoreInterface } from './types.js';
+export interface VectorIndexResult {
+    collectionName: string;
+    chunkCount: number;
+    embeddingModel: string;
+    embeddingDimension: number;
+}
+export declare class VectorIndexer {
+    private readonly embeddingProvider;
+    private readonly vectorStores;
+    constructor(embeddingProvider: EmbeddingProvider, vectorStores: Map<string, VectorStoreInterface>);
+    indexChunks(chunks: StoredChunk[]): Promise<VectorIndexResult[]>;
+    deleteFile(collectionName: string, filePath: string): Promise<void>;
+    private groupByCollection;
+    private toVectorDocument;
+}

package/dist/vector/vector-indexer.js

@@ -0,0 +1,61 @@
+export class VectorIndexer {
+    embeddingProvider;
+    vectorStores;
+    constructor(embeddingProvider, vectorStores) {
+        this.embeddingProvider = embeddingProvider;
+        this.vectorStores = vectorStores;
+    }
+    async indexChunks(chunks) {
+        const byCollection = this.groupByCollection(chunks);
+        const results = [];
+        for (const [collectionName, collectionChunks] of byCollection) {
+            const store = this.vectorStores.get(collectionName);
+            if (!store)
+                continue;
+            await store.ensureCollection({
+                embedding_model: this.embeddingProvider.model,
+                embedding_dimension: this.embeddingProvider.dimensions,
+            });
+            const embeddings = await this.embeddingProvider.embedDocuments(collectionChunks.map(chunk => chunk.content));
+            const documents = collectionChunks.map((chunk, index) => this.toVectorDocument(chunk, embeddings[index] ?? []));
+            await store.upsert(documents);
+            results.push({
+                collectionName,
+                chunkCount: documents.length,
+                embeddingModel: this.embeddingProvider.model,
+                embeddingDimension: this.embeddingProvider.dimensions,
+            });
+        }
+        return results;
+    }
+    async deleteFile(collectionName, filePath) {
+        const store = this.vectorStores.get(collectionName);
+        if (!store)
+            return;
+        await store.deleteByFilePath(filePath);
+    }
+    groupByCollection(chunks) {
+        const grouped = new Map();
+        for (const chunk of chunks) {
+            const list = grouped.get(chunk.collectionName) ?? [];
+            list.push(chunk);
+            grouped.set(chunk.collectionName, list);
+        }
+        return grouped;
+    }
+    toVectorDocument(chunk, embedding) {
+        return {
+            id: chunk.id,
+            content: chunk.content,
+            embedding,
+            metadata: {
+                file_path: chunk.relativePath,
+                chunk_index: chunk.chunkIndex,
+                category: chunk.category,
+                format: chunk.format,
+                token_count: chunk.tokenCount,
+                section_title: chunk.sectionTitle ?? null,
+            },
+        };
+    }
+}

package/package.json

@@ -0,0 +1,45 @@
+{
+  "name": "@customize-agent/knowledge",
+  "version": "1.0.1",
+  "description": "Local knowledge base infrastructure for customize-agent",
+  "type": "module",
+  "main": "./dist/index.js",
+  "types": "./dist/index.d.ts",
+  "license": "MIT",
+  "author": "Pan-jijian",
+  "repository": {
+    "type": "git",
+    "url": "https://github.com/Pan-jijian/customize-agent.git",
+    "directory": "packages/knowledge"
+  },
+  "files": [
+    "dist"
+  ],
+  "publishConfig": {
+    "access": "public"
+  },
+  "keywords": [
+    "ai-agent",
+    "llm",
+    "customize-agent",
+    "knowledge-base"
+  ],
+  "dependencies": {
+    "better-sqlite3": "^12.10.0",
+    "fast-glob": "^3.3.3",
+    "jszip": "^3.10.1",
+    "mammoth": "^1.11.0",
+    "pdf-parse": "^2.4.5",
+    "tesseract.js": "^6.0.1",
+    "xlsx": "^0.18.5"
+  },
+  "devDependencies": {
+    "@types/better-sqlite3": "^7.6.13",
+    "@types/node": "^25.9.3"
+  },
+  "scripts": {
+    "build": "tsc",
+    "typecheck": "tsc --noEmit",
+    "lint": "eslint src/"
+  }
+}