openhorse 0.1.5

package/dist/memory/vector-store.js

@@ -0,0 +1,259 @@
+"use strict";
+/**
+ * openhorse - Vector Store
+ *
+ * 基于 sqlite-vec 的向量存储层
+ */
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.VectorStore = void 0;
+exports.getVectorStore = getVectorStore;
+exports.resetVectorStore = resetVectorStore;
+const better_sqlite3_1 = __importDefault(require("better-sqlite3"));
+const path_1 = require("path");
+const fs_1 = require("fs");
+const embeddings_1 = require("./embeddings");
+// ============================================================================
+// Vector Store
+// ============================================================================
+class VectorStore {
+    constructor(config) {
+        this.initialized = false;
+        // Determine database path
+        const configHome = process.env.OPENHORSE_CONFIG_HOME || (0, path_1.join)(process.env.HOME || '', '.openhorse');
+        const dbPath = config?.dbPath || (0, path_1.join)(configHome, 'vector.db');
+        // Ensure directory exists
+        const dbDir = (0, path_1.join)(dbPath, '..');
+        if (!(0, fs_1.existsSync)(dbDir)) {
+            (0, fs_1.mkdirSync)(dbDir, { recursive: true });
+        }
+        // Initialize database
+        this.db = new better_sqlite3_1.default(dbPath);
+        // Get embedding service
+        this.embeddingService = (0, embeddings_1.getEmbeddingService)(config?.embeddingConfig);
+        // Initialize tables
+        this.initialize();
+    }
+    /** Initialize database tables */
+    initialize() {
+        // Create memories table (without vector column initially)
+        this.db.exec(`
+      CREATE TABLE IF NOT EXISTS memories (
+        id TEXT PRIMARY KEY,
+        name TEXT NOT NULL,
+        type TEXT NOT NULL,
+        content TEXT NOT NULL,
+        description TEXT,
+        project TEXT,
+        created_at INTEGER NOT NULL,
+        updated_at INTEGER NOT NULL
+      )
+    `);
+        // Create index on type and project
+        this.db.exec(`
+      CREATE INDEX IF NOT EXISTS idx_memories_type ON memories(type)
+    `);
+        this.db.exec(`
+      CREATE INDEX IF NOT EXISTS idx_memories_project ON memories(project)
+    `);
+        // Try to create vector column using sqlite-vec
+        try {
+            const dimension = this.embeddingService.getDimension();
+            this.db.exec(`
+        CREATE VIRTUAL TABLE IF NOT EXISTS vec_memories USING vec0(
+          embedding FLOAT[${dimension}]
+        )
+      `);
+            // Link table for vector -> memory
+            this.db.exec(`
+        CREATE TABLE IF NOT EXISTS memory_vectors (
+          memory_id TEXT PRIMARY KEY,
+          vector_rowid INTEGER,
+          FOREIGN KEY (memory_id) REFERENCES memories(id)
+        )
+      `);
+            this.initialized = true;
+        }
+        catch (err) {
+            // sqlite-vec may not be available - fall back to text search only
+            console.warn(`[VectorStore] sqlite-vec not available: ${err.message}`);
+            this.initialized = false;
+        }
+    }
+    /** Check if vector search is available */
+    isVectorSearchAvailable() {
+        return this.initialized;
+    }
+    /** Insert or update memory with embedding */
+    async upsert(entry, projectPath) {
+        const projectHash = projectPath ? this.hashProject(projectPath) : 'global';
+        // Insert/update memory record
+        const stmt = this.db.prepare(`
+      INSERT OR REPLACE INTO memories (id, name, type, content, description, project, created_at, updated_at)
+      VALUES (?, ?, ?, ?, ?, ?, ?, ?)
+    `);
+        stmt.run(entry.name, entry.name, entry.type, entry.content, entry.description || entry.content.slice(0, 100), projectHash, entry.createdAt, entry.updatedAt);
+        // Generate and store embedding if vector search available
+        if (this.initialized) {
+            try {
+                const vector = await this.embeddingService.embed(entry.content);
+                // Delete old vector if exists
+                this.db.prepare('DELETE FROM memory_vectors WHERE memory_id = ?').run(entry.name);
+                this.db.prepare('DELETE FROM vec_memories WHERE rowid IN (SELECT vector_rowid FROM memory_vectors WHERE memory_id = ?)').run(entry.name);
+                // Insert new vector
+                const vectorStmt = this.db.prepare(`INSERT INTO vec_memories (embedding) VALUES (?)`);
+                const result = vectorStmt.run(JSON.stringify(vector));
+                // Link vector to memory
+                this.db.prepare('INSERT INTO memory_vectors (memory_id, vector_rowid) VALUES (?, ?)').run(entry.name, result.lastInsertRowid);
+            }
+            catch (err) {
+                console.warn(`[VectorStore] Failed to store embedding: ${err.message}`);
+            }
+        }
+    }
+    /** Delete memory */
+    delete(name) {
+        // Delete vector link
+        if (this.initialized) {
+            this.db.prepare('DELETE FROM vec_memories WHERE rowid IN (SELECT vector_rowid FROM memory_vectors WHERE memory_id = ?)').run(name);
+            this.db.prepare('DELETE FROM memory_vectors WHERE memory_id = ?').run(name);
+        }
+        // Delete memory record
+        this.db.prepare('DELETE FROM memories WHERE id = ?').run(name);
+    }
+    /** Search by similarity */
+    async search(query, limit = 10, projectPath) {
+        const projectHash = projectPath ? this.hashProject(projectPath) : undefined;
+        // If vector search available, use semantic search
+        if (this.initialized) {
+            return this.semanticSearch(query, limit, projectHash);
+        }
+        // Otherwise fall back to text search
+        return this.textSearch(query, limit, projectHash);
+    }
+    /** Semantic search using vectors */
+    async semanticSearch(query, limit, projectHash) {
+        try {
+            const queryVector = await this.embeddingService.embed(query);
+            // Use sqlite-vec for similarity search
+            const sql = projectHash
+                ? `
+          SELECT m.id, m.name, m.type, m.content, m.description, m.created_at,
+                 vec_distance_cosine(v.embedding, ?) as distance
+          FROM memories m
+          JOIN memory_vectors mv ON m.id = mv.memory_id
+          JOIN vec_memories v ON mv.vector_rowid = v.rowid
+          WHERE m.project = ?
+          ORDER BY distance ASC
+          LIMIT ?
+        `
+                : `
+          SELECT m.id, m.name, m.type, m.content, m.description, m.created_at,
+                 vec_distance_cosine(v.embedding, ?) as distance
+          FROM memories m
+          JOIN memory_vectors mv ON m.id = mv.memory_id
+          JOIN vec_memories v ON mv.vector_rowid = v.rowid
+          ORDER BY distance ASC
+          LIMIT ?
+        `;
+            const stmt = this.db.prepare(sql);
+            const params = projectHash
+                ? [JSON.stringify(queryVector), projectHash, limit]
+                : [JSON.stringify(queryVector), limit];
+            const rows = stmt.all(...params);
+            return rows.map(row => ({
+                id: row.id,
+                name: row.name,
+                type: row.type,
+                content: row.content,
+                description: row.description,
+                score: 1 - row.distance, // Convert distance to similarity score
+                createdAt: row.created_at,
+            }));
+        }
+        catch (err) {
+            console.warn(`[VectorStore] Semantic search failed: ${err.message}`);
+            return this.textSearch(query, limit, projectHash);
+        }
+    }
+    /** Text search fallback */
+    textSearch(query, limit, projectHash) {
+        const sql = projectHash
+            ? `
+        SELECT id, name, type, content, description, created_at
+        FROM memories
+        WHERE project = ? AND (content LIKE ? OR name LIKE ? OR description LIKE ?)
+        LIMIT ?
+      `
+            : `
+        SELECT id, name, type, content, description, created_at
+        FROM memories
+        WHERE content LIKE ? OR name LIKE ? OR description LIKE ?
+        LIMIT ?
+      `;
+        const searchTerm = `%${query}%`;
+        const stmt = this.db.prepare(sql);
+        const params = projectHash
+            ? [projectHash, searchTerm, searchTerm, searchTerm, limit]
+            : [searchTerm, searchTerm, searchTerm, limit];
+        const rows = stmt.all(...params);
+        return rows.map(row => ({
+            id: row.id,
+            name: row.name,
+            type: row.type,
+            content: row.content,
+            description: row.description,
+            score: 0.5, // Default score for text search
+            createdAt: row.created_at,
+        }));
+    }
+    /** Get all memories for a project */
+    getAll(projectPath) {
+        const projectHash = projectPath ? this.hashProject(projectPath) : undefined;
+        const sql = projectHash
+            ? 'SELECT id, name, type, content, description, created_at FROM memories WHERE project = ?'
+            : 'SELECT id, name, type, content, description, created_at FROM memories';
+        const stmt = this.db.prepare(sql);
+        const params = projectHash ? [projectHash] : [];
+        const rows = stmt.all(...params);
+        return rows.map(row => ({
+            id: row.id,
+            name: row.name,
+            type: row.type,
+            content: row.content,
+            description: row.description,
+            score: 1,
+            createdAt: row.created_at,
+        }));
+    }
+    /** Hash project path */
+    hashProject(projectPath) {
+        // Simple hash for project identification
+        const hash = projectPath.split('/').slice(-2).join('/');
+        return hash;
+    }
+    /** Close database */
+    close() {
+        this.db.close();
+    }
+}
+exports.VectorStore = VectorStore;
+// ============================================================================
+// Factory
+// ============================================================================
+let defaultStore = null;
+function getVectorStore(config) {
+    if (!defaultStore) {
+        defaultStore = new VectorStore(config);
+    }
+    return defaultStore;
+}
+function resetVectorStore() {
+    if (defaultStore) {
+        defaultStore.close();
+        defaultStore = null;
+    }
+}
+//# sourceMappingURL=vector-store.js.map

package/dist/memory/vector-store.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"vector-store.js","sourceRoot":"","sources":["../../src/memory/vector-store.ts"],"names":[],"mappings":";AAAA;;;;GAIG;;;;;;AAoTH,wCAKC;AAED,4CAKC;AA9TD,oEAAsC;AACtC,+BAA4B;AAC5B,2BAA2C;AAC3C,6CAAyE;AAsBzE,+EAA+E;AAC/E,eAAe;AACf,+EAA+E;AAE/E,MAAa,WAAW;IAKtB,YAAY,MAA0B;QAF9B,gBAAW,GAAY,KAAK,CAAC;QAGnC,0BAA0B;QAC1B,MAAM,UAAU,GAAG,OAAO,CAAC,GAAG,CAAC,qBAAqB,IAAI,IAAA,WAAI,EAAC,OAAO,CAAC,GAAG,CAAC,IAAI,IAAI,EAAE,EAAE,YAAY,CAAC,CAAC;QACnG,MAAM,MAAM,GAAG,MAAM,EAAE,MAAM,IAAI,IAAA,WAAI,EAAC,UAAU,EAAE,WAAW,CAAC,CAAC;QAE/D,0BAA0B;QAC1B,MAAM,KAAK,GAAG,IAAA,WAAI,EAAC,MAAM,EAAE,IAAI,CAAC,CAAC;QACjC,IAAI,CAAC,IAAA,eAAU,EAAC,KAAK,CAAC,EAAE,CAAC;YACvB,IAAA,cAAS,EAAC,KAAK,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;QACxC,CAAC;QAED,sBAAsB;QACtB,IAAI,CAAC,EAAE,GAAG,IAAI,wBAAQ,CAAC,MAAM,CAAC,CAAC;QAE/B,wBAAwB;QACxB,IAAI,CAAC,gBAAgB,GAAG,IAAA,gCAAmB,EAAC,MAAM,EAAE,eAAe,CAAC,CAAC;QAErE,oBAAoB;QACpB,IAAI,CAAC,UAAU,EAAE,CAAC;IACpB,CAAC;IAED,iCAAiC;IACzB,UAAU;QAChB,0DAA0D;QAC1D,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC;;;;;;;;;;;KAWZ,CAAC,CAAC;QAEH,mCAAmC;QACnC,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC;;KAEZ,CAAC,CAAC;QACH,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC;;KAEZ,CAAC,CAAC;QAEH,+CAA+C;QAC/C,IAAI,CAAC;YACH,MAAM,SAAS,GAAG,IAAI,CAAC,gBAAgB,CAAC,YAAY,EAAE,CAAC;YACvD,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC;;4BAES,SAAS;;OAE9B,CAAC,CAAC;YAEH,kCAAkC;YAClC,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC;;;;;;OAMZ,CAAC,CAAC;YAEH,IAAI,CAAC,WAAW,GAAG,IAAI,CAAC;QAC1B,CAAC;QAAC,OAAO,GAAQ,EAAE,CAAC;YAClB,kEAAkE;YAClE,OAAO,CAAC,IAAI,CAAC,2CAA2C,GAAG,CAAC,OAAO,EAAE,CAAC,CAAC;YACvE,IAAI,CAAC,WAAW,GAAG,KAAK,CAAC;QAC3B,CAAC;IACH,CAAC;IAED,0CAA0C;IAC1C,uBAAuB;QACrB,OAAO,IAAI,CAAC,WAAW,CAAC;IAC1B,CAAC;IAED,6CAA6C;IAC7C,KAAK,CAAC,MAAM,CAAC,KAAkB,EAAE,WAAoB;QACnD,MAAM,WAAW,GAAG,WAAW,CAAC,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC;QAE3E,8BAA8B;QAC9B,MAAM,IAAI,GAAG,IAAI,CAAC,EAAE,CAAC,OAAO,CAAC;;;KAG5B,CAAC,CAAC;QAEH,IAAI,CAAC,GAAG,CACN,KAAK,CAAC,IAAI,EACV,KAAK,CAAC,IAAI,EACV,KAAK,CAAC,IAAI,EACV,KAAK,CAAC,OAAO,EACb,KAAK,CAAC,WAAW,IAAI,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC,EAChD,WAAW,EACX,KAAK,CAAC,SAAS,EACf,KAAK,CAAC,SAAS,CAChB,CAAC;QAEF,0DAA0D;QAC1D,IAAI,IAAI,CAAC,WAAW,EAAE,CAAC;YACrB,IAAI,CAAC;gBACH,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,gBAAgB,CAAC,KAAK,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;gBAEhE,8BAA8B;gBAC9B,IAAI,CAAC,EAAE,CAAC,OAAO,CAAC,gDAAgD,CAAC,CAAC,GAAG,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;gBAClF,IAAI,CAAC,EAAE,CAAC,OAAO,CAAC,uGAAuG,CAAC,CAAC,GAAG,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;gBAEzI,oBAAoB;gBACpB,MAAM,UAAU,GAAG,IAAI,CAAC,EAAE,CAAC,OAAO,CAAC,iDAAiD,CAAC,CAAC;gBACtF,MAAM,MAAM,GAAG,UAAU,CAAC,GAAG,CAAC,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC,CAAC;gBAEtD,wBAAwB;gBACxB,IAAI,CAAC,EAAE,CAAC,OAAO,CAAC,oEAAoE,CAAC,CAAC,GAAG,CACvF,KAAK,CAAC,IAAI,EACV,MAAM,CAAC,eAAe,CACvB,CAAC;YACJ,CAAC;YAAC,OAAO,GAAQ,EAAE,CAAC;gBAClB,OAAO,CAAC,IAAI,CAAC,4CAA4C,GAAG,CAAC,OAAO,EAAE,CAAC,CAAC;YAC1E,CAAC;QACH,CAAC;IACH,CAAC;IAED,oBAAoB;IACpB,MAAM,CAAC,IAAY;QACjB,qBAAqB;QACrB,IAAI,IAAI,CAAC,WAAW,EAAE,CAAC;YACrB,IAAI,CAAC,EAAE,CAAC,OAAO,CAAC,uGAAuG,CAAC,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;YACnI,IAAI,CAAC,EAAE,CAAC,OAAO,CAAC,gDAAgD,CAAC,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;QAC9E,CAAC;QAED,uBAAuB;QACvB,IAAI,CAAC,EAAE,CAAC,OAAO,CAAC,mCAAmC,CAAC,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;IACjE,CAAC;IAED,2BAA2B;IAC3B,KAAK,CAAC,MAAM,CAAC,KAAa,EAAE,QAAgB,EAAE,EAAE,WAAoB;QAClE,MAAM,WAAW,GAAG,WAAW,CAAC,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;QAE5E,kDAAkD;QAClD,IAAI,IAAI,CAAC,WAAW,EAAE,CAAC;YACrB,OAAO,IAAI,CAAC,cAAc,CAAC,KAAK,EAAE,KAAK,EAAE,WAAW,CAAC,CAAC;QACxD,CAAC;QAED,qCAAqC;QACrC,OAAO,IAAI,CAAC,UAAU,CAAC,KAAK,EAAE,KAAK,EAAE,WAAW,CAAC,CAAC;IACpD,CAAC;IAED,oCAAoC;IAC5B,KAAK,CAAC,cAAc,CAAC,KAAa,EAAE,KAAa,EAAE,WAAoB;QAC7E,IAAI,CAAC;YACH,MAAM,WAAW,GAAG,MAAM,IAAI,CAAC,gBAAgB,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;YAE7D,uCAAuC;YACvC,MAAM,GAAG,GAAG,WAAW;gBACrB,CAAC,CAAC;;;;;;;;;SASD;gBACD,CAAC,CAAC;;;;;;;;SAQD,CAAC;YAEJ,MAAM,IAAI,GAAG,IAAI,CAAC,EAAE,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;YAClC,MAAM,MAAM,GAAG,WAAW;gBACxB,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,WAAW,CAAC,EAAE,WAAW,EAAE,KAAK,CAAC;gBACnD,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,WAAW,CAAC,EAAE,KAAK,CAAC,CAAC;YAEzC,MAAM,IAAI,GAAG,IAAI,CAAC,GAAG,CAAC,GAAG,MAAM,CAAU,CAAC;YAE1C,OAAO,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;gBACtB,EAAE,EAAE,GAAG,CAAC,EAAE;gBACV,IAAI,EAAE,GAAG,CAAC,IAAI;gBACd,IAAI,EAAE,GAAG,CAAC,IAAkB;gBAC5B,OAAO,EAAE,GAAG,CAAC,OAAO;gBACpB,WAAW,EAAE,GAAG,CAAC,WAAW;gBAC5B,KAAK,EAAE,CAAC,GAAG,GAAG,CAAC,QAAQ,EAAE,uCAAuC;gBAChE,SAAS,EAAE,GAAG,CAAC,UAAU;aAC1B,CAAC,CAAC,CAAC;QACN,CAAC;QAAC,OAAO,GAAQ,EAAE,CAAC;YAClB,OAAO,CAAC,IAAI,CAAC,yCAAyC,GAAG,CAAC,OAAO,EAAE,CAAC,CAAC;YACrE,OAAO,IAAI,CAAC,UAAU,CAAC,KAAK,EAAE,KAAK,EAAE,WAAW,CAAC,CAAC;QACpD,CAAC;IACH,CAAC;IAED,2BAA2B;IACnB,UAAU,CAAC,KAAa,EAAE,KAAa,EAAE,WAAoB;QACnE,MAAM,GAAG,GAAG,WAAW;YACrB,CAAC,CAAC;;;;;OAKD;YACD,CAAC,CAAC;;;;;OAKD,CAAC;QAEJ,MAAM,UAAU,GAAG,IAAI,KAAK,GAAG,CAAC;QAChC,MAAM,IAAI,GAAG,IAAI,CAAC,EAAE,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;QAClC,MAAM,MAAM,GAAG,WAAW;YACxB,CAAC,CAAC,CAAC,WAAW,EAAE,UAAU,EAAE,UAAU,EAAE,UAAU,EAAE,KAAK,CAAC;YAC1D,CAAC,CAAC,CAAC,UAAU,EAAE,UAAU,EAAE,UAAU,EAAE,KAAK,CAAC,CAAC;QAEhD,MAAM,IAAI,GAAG,IAAI,CAAC,GAAG,CAAC,GAAG,MAAM,CAAU,CAAC;QAE1C,OAAO,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;YACtB,EAAE,EAAE,GAAG,CAAC,EAAE;YACV,IAAI,EAAE,GAAG,CAAC,IAAI;YACd,IAAI,EAAE,GAAG,CAAC,IAAkB;YAC5B,OAAO,EAAE,GAAG,CAAC,OAAO;YACpB,WAAW,EAAE,GAAG,CAAC,WAAW;YAC5B,KAAK,EAAE,GAAG,EAAE,gCAAgC;YAC5C,SAAS,EAAE,GAAG,CAAC,UAAU;SAC1B,CAAC,CAAC,CAAC;IACN,CAAC;IAED,qCAAqC;IACrC,MAAM,CAAC,WAAoB;QACzB,MAAM,WAAW,GAAG,WAAW,CAAC,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;QAE5E,MAAM,GAAG,GAAG,WAAW;YACrB,CAAC,CAAC,yFAAyF;YAC3F,CAAC,CAAC,uEAAuE,CAAC;QAE5E,MAAM,IAAI,GAAG,IAAI,CAAC,EAAE,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;QAClC,MAAM,MAAM,GAAG,WAAW,CAAC,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;QAChD,MAAM,IAAI,GAAG,IAAI,CAAC,GAAG,CAAC,GAAG,MAAM,CAAU,CAAC;QAE1C,OAAO,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;YACtB,EAAE,EAAE,GAAG,CAAC,EAAE;YACV,IAAI,EAAE,GAAG,CAAC,IAAI;YACd,IAAI,EAAE,GAAG,CAAC,IAAkB;YAC5B,OAAO,EAAE,GAAG,CAAC,OAAO;YACpB,WAAW,EAAE,GAAG,CAAC,WAAW;YAC5B,KAAK,EAAE,CAAC;YACR,SAAS,EAAE,GAAG,CAAC,UAAU;SAC1B,CAAC,CAAC,CAAC;IACN,CAAC;IAED,wBAAwB;IAChB,WAAW,CAAC,WAAmB;QACrC,yCAAyC;QACzC,MAAM,IAAI,GAAG,WAAW,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QACxD,OAAO,IAAI,CAAC;IACd,CAAC;IAED,qBAAqB;IACrB,KAAK;QACH,IAAI,CAAC,EAAE,CAAC,KAAK,EAAE,CAAC;IAClB,CAAC;CACF;AA7QD,kCA6QC;AAED,+EAA+E;AAC/E,UAAU;AACV,+EAA+E;AAE/E,IAAI,YAAY,GAAuB,IAAI,CAAC;AAE5C,SAAgB,cAAc,CAAC,MAA0B;IACvD,IAAI,CAAC,YAAY,EAAE,CAAC;QAClB,YAAY,GAAG,IAAI,WAAW,CAAC,MAAM,CAAC,CAAC;IACzC,CAAC;IACD,OAAO,YAAY,CAAC;AACtB,CAAC;AAED,SAAgB,gBAAgB;IAC9B,IAAI,YAAY,EAAE,CAAC;QACjB,YAAY,CAAC,KAAK,EAAE,CAAC;QACrB,YAAY,GAAG,IAAI,CAAC;IACtB,CAAC;AACH,CAAC"}

package/dist/screens/REPL.d.ts

@@ -0,0 +1,21 @@
+/**
+ * OpenHorse - Simple REPL Screen
+ *
+ * A simplified REPL interface using Ink components.
+ */
+import React from 'react';
+import { render } from '../ink';
+interface Message {
+    role: 'user' | 'assistant';
+    content: string;
+}
+interface REPLProps {
+    initialMessages?: Message[];
+}
+/**
+ * REPL - Main REPL component
+ */
+declare function REPL({ initialMessages }: REPLProps): React.JSX.Element;
+export { REPL, render };
+export default REPL;
+//# sourceMappingURL=REPL.d.ts.map

package/dist/screens/REPL.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"REPL.d.ts","sourceRoot":"","sources":["../../src/screens/REPL.tsx"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,KAAgC,MAAM,OAAO,CAAC;AACrD,OAAO,EAA+B,MAAM,EAAE,MAAM,QAAQ,CAAC;AAM7D,UAAU,OAAO;IACf,IAAI,EAAE,MAAM,GAAG,WAAW,CAAC;IAC3B,OAAO,EAAE,MAAM,CAAC;CACjB;AAED,UAAU,SAAS;IACjB,eAAe,CAAC,EAAE,OAAO,EAAE,CAAC;CAC7B;AA6DD;;GAEG;AACH,iBAAS,IAAI,CAAC,EAAE,eAAoB,EAAE,EAAE,SAAS,qBAyEhD;AAMD,OAAO,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC;AACxB,eAAe,IAAI,CAAC"}

package/dist/screens/REPL.js

@@ -0,0 +1,139 @@
+"use strict";
+/**
+ * OpenHorse - Simple REPL Screen
+ *
+ * A simplified REPL interface using Ink components.
+ */
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
+    Object.defineProperty(o, "default", { enumerable: true, value: v });
+}) : function(o, v) {
+    o["default"] = v;
+});
+var __importStar = (this && this.__importStar) || (function () {
+    var ownKeys = function(o) {
+        ownKeys = Object.getOwnPropertyNames || function (o) {
+            var ar = [];
+            for (var k in o) if (Object.prototype.hasOwnProperty.call(o, k)) ar[ar.length] = k;
+            return ar;
+        };
+        return ownKeys(o);
+    };
+    return function (mod) {
+        if (mod && mod.__esModule) return mod;
+        var result = {};
+        if (mod != null) for (var k = ownKeys(mod), i = 0; i < k.length; i++) if (k[i] !== "default") __createBinding(result, mod, k[i]);
+        __setModuleDefault(result, mod);
+        return result;
+    };
+})();
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.render = void 0;
+exports.REPL = REPL;
+const react_1 = __importStar(require("react"));
+const ink_1 = require("../ink");
+Object.defineProperty(exports, "render", { enumerable: true, get: function () { return ink_1.render; } });
+// ============================================================================
+// Components
+// ============================================================================
+/**
+ * Spinner - Loading indicator
+ */
+function Spinner({ text = 'Thinking' }) {
+    const [frame, setFrame] = (0, react_1.useState)(0);
+    const frames = ['⠋', '⠙', '⠹', '⠸', '⠼', '⠴', '⠦', '⠧', '⠇', '⠏'];
+    react_1.default.useEffect(() => {
+        const timer = setInterval(() => {
+            setFrame((f) => (f + 1) % frames.length);
+        }, 80);
+        return () => clearInterval(timer);
+    }, []);
+    return (react_1.default.createElement(ink_1.Text, { color: "cyan" },
+        frames[frame],
+        " ",
+        text));
+}
+/**
+ * MessageList - Display conversation history
+ */
+function MessageList({ messages }) {
+    return (react_1.default.createElement(ink_1.Box, { flexDirection: "column" }, messages.map((msg, i) => (react_1.default.createElement(ink_1.Box, { flexDirection: "row", key: i },
+        react_1.default.createElement(ink_1.Text, { bold: true, color: msg.role === 'user' ? 'green' : 'blue' }, msg.role === 'user' ? '> ' : '< '),
+        react_1.default.createElement(ink_1.Text, { dim: msg.role === 'assistant' }, msg.content))))));
+}
+/**
+ * InputLine - Show current input
+ */
+function InputLine({ value, placeholder = 'Type your message...' }) {
+    return (react_1.default.createElement(ink_1.Box, { flexDirection: "row" },
+        react_1.default.createElement(ink_1.Text, { color: "yellow", bold: true }, '> '),
+        react_1.default.createElement(ink_1.Text, null, value || placeholder)));
+}
+/**
+ * REPL - Main REPL component
+ */
+function REPL({ initialMessages = [] }) {
+    const [messages, setMessages] = (0, react_1.useState)(initialMessages);
+    const [input, setInput] = (0, react_1.useState)('');
+    const [isStreaming, setIsStreaming] = (0, react_1.useState)(false);
+    const { exit } = (0, ink_1.useApp)();
+    // Handle keyboard input
+    (0, ink_1.useInput)((char, key) => {
+        if (key.ctrl && key.name === 'c') {
+            exit();
+            return;
+        }
+        if (isStreaming)
+            return;
+        // Enter key submits
+        if (char === '\r' || char === '\n') {
+            if (input.trim()) {
+                handleSubmit(input);
+            }
+            return;
+        }
+        // Backspace
+        if (char === '\x7f' || key.name === 'backspace') {
+            setInput((prev) => prev.slice(0, -1));
+            return;
+        }
+        // Regular character
+        if (char.length === 1 && char.charCodeAt(0) >= 32) {
+            setInput((prev) => prev + char);
+        }
+    });
+    // Handle message submission
+    const handleSubmit = (0, react_1.useCallback)((text) => {
+        const userMessage = { role: 'user', content: text };
+        setMessages((prev) => [...prev, userMessage]);
+        setInput('');
+        setIsStreaming(true);
+        // Simulate response (in real implementation, call LLM)
+        setTimeout(() => {
+            const assistantMessage = {
+                role: 'assistant',
+                content: 'This is a placeholder response. LLM integration pending.',
+            };
+            setMessages((prev) => [...prev, assistantMessage]);
+            setIsStreaming(false);
+        }, 1000);
+    }, []);
+    return (react_1.default.createElement(ink_1.Box, { flexDirection: "column" },
+        react_1.default.createElement(ink_1.Text, { bold: true, color: "magenta" }, "OpenHorse REPL"),
+        react_1.default.createElement(ink_1.Text, { dim: true }, "Press Enter to send, Ctrl+C to exit"),
+        react_1.default.createElement(MessageList, { messages: messages }),
+        isStreaming && react_1.default.createElement(Spinner, null),
+        react_1.default.createElement(InputLine, { value: input })));
+}
+exports.default = REPL;
+//# sourceMappingURL=REPL.js.map

package/dist/screens/REPL.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"REPL.js","sourceRoot":"","sources":["../../src/screens/REPL.tsx"],"names":[],"mappings":";AAAA;;;;GAIG;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AA+JM,oBAAI;AA7Jb,+CAAqD;AACrD,gCAA6D;AA4J9C,uFA5JuB,YAAM,OA4JvB;AA7IrB,+EAA+E;AAC/E,aAAa;AACb,+EAA+E;AAE/E;;GAEG;AACH,SAAS,OAAO,CAAC,EAAE,IAAI,GAAG,UAAU,EAAqB;IACvD,MAAM,CAAC,KAAK,EAAE,QAAQ,CAAC,GAAG,IAAA,gBAAQ,EAAC,CAAC,CAAC,CAAC;IACtC,MAAM,MAAM,GAAG,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,CAAC,CAAC;IAElE,eAAK,CAAC,SAAS,CAAC,GAAG,EAAE;QACnB,MAAM,KAAK,GAAG,WAAW,CAAC,GAAG,EAAE;YAC7B,QAAQ,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,GAAG,MAAM,CAAC,MAAM,CAAC,CAAC;QAC3C,CAAC,EAAE,EAAE,CAAC,CAAC;QACP,OAAO,GAAG,EAAE,CAAC,aAAa,CAAC,KAAK,CAAC,CAAC;IACpC,CAAC,EAAE,EAAE,CAAC,CAAC;IAEP,OAAO,CACL,8BAAC,UAAI,IAAC,KAAK,EAAC,MAAM;QACf,MAAM,CAAC,KAAK,CAAC;;QAAG,IAAI,CAChB,CACR,CAAC;AACJ,CAAC;AAED;;GAEG;AACH,SAAS,WAAW,CAAC,EAAE,QAAQ,EAA2B;IACxD,OAAO,CACL,8BAAC,SAAG,IAAC,aAAa,EAAC,QAAQ,IACxB,QAAQ,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,CACxB,8BAAC,SAAG,IAAC,aAAa,EAAC,KAAK,EAAC,GAAG,EAAE,CAAC;QAC7B,8BAAC,UAAI,IAAC,IAAI,QAAC,KAAK,EAAE,GAAG,CAAC,IAAI,KAAK,MAAM,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,IACrD,GAAG,CAAC,IAAI,KAAK,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAC7B;QACP,8BAAC,UAAI,IAAC,GAAG,EAAE,GAAG,CAAC,IAAI,KAAK,WAAW,IAChC,GAAG,CAAC,OAAO,CACP,CACH,CACP,CAAC,CACE,CACP,CAAC;AACJ,CAAC;AAED;;GAEG;AACH,SAAS,SAAS,CAAC,EAAE,KAAK,EAAE,WAAW,GAAG,sBAAsB,EAA2C;IACzG,OAAO,CACL,8BAAC,SAAG,IAAC,aAAa,EAAC,KAAK;QACtB,8BAAC,UAAI,IAAC,KAAK,EAAC,QAAQ,EAAC,IAAI,UAAE,IAAI,CAAQ;QACvC,8BAAC,UAAI,QACF,KAAK,IAAI,WAAW,CAChB,CACH,CACP,CAAC;AACJ,CAAC;AAED;;GAEG;AACH,SAAS,IAAI,CAAC,EAAE,eAAe,GAAG,EAAE,EAAa;IAC/C,MAAM,CAAC,QAAQ,EAAE,WAAW,CAAC,GAAG,IAAA,gBAAQ,EAAY,eAAe,CAAC,CAAC;IACrE,MAAM,CAAC,KAAK,EAAE,QAAQ,CAAC,GAAG,IAAA,gBAAQ,EAAC,EAAE,CAAC,CAAC;IACvC,MAAM,CAAC,WAAW,EAAE,cAAc,CAAC,GAAG,IAAA,gBAAQ,EAAC,KAAK,CAAC,CAAC;IACtD,MAAM,EAAE,IAAI,EAAE,GAAG,IAAA,YAAM,GAAE,CAAC;IAE1B,wBAAwB;IACxB,IAAA,cAAQ,EAAC,CAAC,IAAI,EAAE,GAAG,EAAE,EAAE;QACrB,IAAI,GAAG,CAAC,IAAI,IAAI,GAAG,CAAC,IAAI,KAAK,GAAG,EAAE,CAAC;YACjC,IAAI,EAAE,CAAC;YACP,OAAO;QACT,CAAC;QAED,IAAI,WAAW;YAAE,OAAO;QAExB,oBAAoB;QACpB,IAAI,IAAI,KAAK,IAAI,IAAI,IAAI,KAAK,IAAI,EAAE,CAAC;YACnC,IAAI,KAAK,CAAC,IAAI,EAAE,EAAE,CAAC;gBACjB,YAAY,CAAC,KAAK,CAAC,CAAC;YACtB,CAAC;YACD,OAAO;QACT,CAAC;QAED,YAAY;QACZ,IAAI,IAAI,KAAK,MAAM,IAAI,GAAG,CAAC,IAAI,KAAK,WAAW,EAAE,CAAC;YAChD,QAAQ,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;YACtC,OAAO;QACT,CAAC;QAED,oBAAoB;QACpB,IAAI,IAAI,CAAC,MAAM,KAAK,CAAC,IAAI,IAAI,CAAC,UAAU,CAAC,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC;YAClD,QAAQ,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,GAAG,IAAI,CAAC,CAAC;QAClC,CAAC;IACH,CAAC,CAAC,CAAC;IAEH,4BAA4B;IAC5B,MAAM,YAAY,GAAG,IAAA,mBAAW,EAAC,CAAC,IAAY,EAAE,EAAE;QAChD,MAAM,WAAW,GAAY,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC;QAC7D,WAAW,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,GAAG,IAAI,EAAE,WAAW,CAAC,CAAC,CAAC;QAC9C,QAAQ,CAAC,EAAE,CAAC,CAAC;QACb,cAAc,CAAC,IAAI,CAAC,CAAC;QAErB,uDAAuD;QACvD,UAAU,CAAC,GAAG,EAAE;YACd,MAAM,gBAAgB,GAAY;gBAChC,IAAI,EAAE,WAAW;gBACjB,OAAO,EAAE,0DAA0D;aACpE,CAAC;YACF,WAAW,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,GAAG,IAAI,EAAE,gBAAgB,CAAC,CAAC,CAAC;YACnD,cAAc,CAAC,KAAK,CAAC,CAAC;QACxB,CAAC,EAAE,IAAI,CAAC,CAAC;IACX,CAAC,EAAE,EAAE,CAAC,CAAC;IAEP,OAAO,CACL,8BAAC,SAAG,IAAC,aAAa,EAAC,QAAQ;QAEzB,8BAAC,UAAI,IAAC,IAAI,QAAC,KAAK,EAAC,SAAS,qBAEnB;QACP,8BAAC,UAAI,IAAC,GAAG,gDAEF;QAGP,8BAAC,WAAW,IAAC,QAAQ,EAAE,QAAQ,GAAI;QAGlC,WAAW,IAAI,8BAAC,OAAO,OAAG;QAG3B,8BAAC,SAAS,IAAC,KAAK,EAAE,KAAK,GAAI,CACvB,CACP,CAAC;AACJ,CAAC;AAOD,kBAAe,IAAI,CAAC"}

package/dist/services/agent-runner.d.ts

@@ -0,0 +1,68 @@
+/**
+ * openhorse - Agent 执行器
+ *
+ * 将 LLM 集成到 Agent 执行流程中：
+ *   - 构建系统提示词与任务上下文
+ *   - 通过 LLM 驱动 Agent 执行
+ *   - 支持流式输出、重试、Token 统计
+ */
+import { EventEmitter } from 'eventemitter3';
+import { BaseAgent, Task, TaskResult } from '../core/agent';
+import { LLMService } from './llm';
+/** AgentRunner 配置 */
+export interface AgentRunnerConfig {
+    /** 最大重试次数 */
+    maxRetries?: number;
+    /** 是否在出错时回退到 BaseAgent.execute */
+    fallbackOnError?: boolean;
+    /** 自定义系统提示词模板 */
+    systemPrompt?: string;
+}
+/** AgentRunner 执行结果（含 LLM 信息） */
+export interface AgentRunnerResult extends TaskResult {
+    /** LLM 回复内容 */
+    llmContent?: string;
+    /** Token 用量 */
+    tokenUsage?: {
+        promptTokens: number;
+        completionTokens: number;
+    };
+    /** 使用的模型 */
+    model?: string;
+    /** 重试次数 */
+    retries: number;
+}
+/** AgentRunner 事件 */
+export interface AgentRunnerEvents {
+    'execute-start': (task: Task) => void;
+    'execute-complete': (task: Task, result: AgentRunnerResult) => void;
+    'execute-failed': (task: Task, error: Error) => void;
+    'stream-chunk': (chunk: string) => void;
+}
+export declare class AgentRunner extends EventEmitter {
+    private agent;
+    private llm;
+    private config;
+    constructor(agent: BaseAgent, llm: LLMService, config?: AgentRunnerConfig);
+    /**
+     * 执行任务 — 通过 LLM 驱动 Agent
+     */
+    run(task: Task): Promise<AgentRunnerResult>;
+    /**
+     * 流式执行 — 适合需要实时反馈的场景
+     */
+    runStream(task: Task, onChunk?: (chunk: string) => void): Promise<AgentRunnerResult>;
+    /**
+     * 获取 Agent 引用
+     */
+    getAgent(): BaseAgent;
+    /**
+     * 获取 LLM 引用
+     */
+    getLLM(): LLMService;
+    /** 构建 LLM 对话消息 */
+    private buildMessages;
+    /** 解析 LLM 响应 */
+    private parseResponse;
+}
+//# sourceMappingURL=agent-runner.d.ts.map

package/dist/services/agent-runner.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"agent-runner.d.ts","sourceRoot":"","sources":["../../src/services/agent-runner.ts"],"names":[],"mappings":"AAAA;;;;;;;GAOG;AAEH,OAAO,EAAE,YAAY,EAAE,MAAM,eAAe,CAAC;AAC7C,OAAO,EAAE,SAAS,EAAE,IAAI,EAAE,UAAU,EAAE,MAAM,eAAe,CAAC;AAC5D,OAAO,EAAE,UAAU,EAAW,MAAM,OAAO,CAAC;AAM5C,qBAAqB;AACrB,MAAM,WAAW,iBAAiB;IAChC,aAAa;IACb,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,kCAAkC;IAClC,eAAe,CAAC,EAAE,OAAO,CAAC;IAC1B,iBAAiB;IACjB,YAAY,CAAC,EAAE,MAAM,CAAC;CACvB;AAED,iCAAiC;AACjC,MAAM,WAAW,iBAAkB,SAAQ,UAAU;IACnD,eAAe;IACf,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,eAAe;IACf,UAAU,CAAC,EAAE;QAAE,YAAY,EAAE,MAAM,CAAC;QAAC,gBAAgB,EAAE,MAAM,CAAA;KAAE,CAAC;IAChE,YAAY;IACZ,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,WAAW;IACX,OAAO,EAAE,MAAM,CAAC;CACjB;AAED,qBAAqB;AACrB,MAAM,WAAW,iBAAiB;IAChC,eAAe,EAAE,CAAC,IAAI,EAAE,IAAI,KAAK,IAAI,CAAC;IACtC,kBAAkB,EAAE,CAAC,IAAI,EAAE,IAAI,EAAE,MAAM,EAAE,iBAAiB,KAAK,IAAI,CAAC;IACpE,gBAAgB,EAAE,CAAC,IAAI,EAAE,IAAI,EAAE,KAAK,EAAE,KAAK,KAAK,IAAI,CAAC;IACrD,cAAc,EAAE,CAAC,KAAK,EAAE,MAAM,KAAK,IAAI,CAAC;CACzC;AAwBD,qBAAa,WAAY,SAAQ,YAAY;IAC3C,OAAO,CAAC,KAAK,CAAY;IACzB,OAAO,CAAC,GAAG,CAAa;IACxB,OAAO,CAAC,MAAM,CAA8B;gBAEhC,KAAK,EAAE,SAAS,EAAE,GAAG,EAAE,UAAU,EAAE,MAAM,CAAC,EAAE,iBAAiB;IAWzE;;OAEG;IACG,GAAG,CAAC,IAAI,EAAE,IAAI,GAAG,OAAO,CAAC,iBAAiB,CAAC;IA6DjD;;OAEG;IACG,SAAS,CACb,IAAI,EAAE,IAAI,EACV,OAAO,CAAC,EAAE,CAAC,KAAK,EAAE,MAAM,KAAK,IAAI,GAChC,OAAO,CAAC,iBAAiB,CAAC;IAuC7B;;OAEG;IACH,QAAQ,IAAI,SAAS;IAIrB;;OAEG;IACH,MAAM,IAAI,UAAU;IAMpB,kBAAkB;IAClB,OAAO,CAAC,aAAa;IAuBrB,gBAAgB;IAChB,OAAO,CAAC,aAAa;CAYtB"}