@danielzfliu/memory 1.0.2 → 2.1.0

package/dist/config.js

@@ -0,0 +1,47 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.DEFAULT_MEMORY_CONFIG = void 0;
+exports.resolveConfig = resolveConfig;
+exports.DEFAULT_MEMORY_CONFIG = {
+    chromaUrl: "http://localhost:8000",
+    ollamaUrl: "http://localhost:11434",
+    embeddingModel: "nomic-embed-text-v2-moe:latest",
+    generationModel: "gemma3:latest",
+    collectionName: "pieces",
+};
+const ENV_CONFIG_KEYS = {
+    chromaUrl: ["MEMORY_CHROMA_URL", "CHROMA_URL"],
+    ollamaUrl: ["MEMORY_OLLAMA_URL", "OLLAMA_URL"],
+    embeddingModel: ["MEMORY_EMBEDDING_MODEL", "EMBEDDING_MODEL"],
+    generationModel: ["MEMORY_GENERATION_MODEL", "GENERATION_MODEL"],
+    collectionName: ["MEMORY_COLLECTION_NAME", "COLLECTION_NAME"],
+};
+function resolveEnvOverride(keys) {
+    for (const key of keys) {
+        const value = process.env[key]?.trim();
+        if (value) {
+            return value;
+        }
+    }
+    return undefined;
+}
+function resolveConfig(config = {}) {
+    return {
+        chromaUrl: config.chromaUrl ??
+            resolveEnvOverride(ENV_CONFIG_KEYS.chromaUrl) ??
+            exports.DEFAULT_MEMORY_CONFIG.chromaUrl,
+        ollamaUrl: config.ollamaUrl ??
+            resolveEnvOverride(ENV_CONFIG_KEYS.ollamaUrl) ??
+            exports.DEFAULT_MEMORY_CONFIG.ollamaUrl,
+        embeddingModel: config.embeddingModel ??
+            resolveEnvOverride(ENV_CONFIG_KEYS.embeddingModel) ??
+            exports.DEFAULT_MEMORY_CONFIG.embeddingModel,
+        generationModel: config.generationModel ??
+            resolveEnvOverride(ENV_CONFIG_KEYS.generationModel) ??
+            exports.DEFAULT_MEMORY_CONFIG.generationModel,
+        collectionName: config.collectionName ??
+            resolveEnvOverride(ENV_CONFIG_KEYS.collectionName) ??
+            exports.DEFAULT_MEMORY_CONFIG.collectionName,
+    };
+}
+//# sourceMappingURL=config.js.map

package/dist/config.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"config.js","sourceRoot":"","sources":["../src/config.ts"],"names":[],"mappings":";;;AA6BA,sCAuBC;AAlDY,QAAA,qBAAqB,GAA2B;IACzD,SAAS,EAAE,uBAAuB;IAClC,SAAS,EAAE,wBAAwB;IACnC,cAAc,EAAE,gCAAgC;IAChD,eAAe,EAAE,eAAe;IAChC,cAAc,EAAE,QAAQ;CAC3B,CAAC;AAEF,MAAM,eAAe,GAAmD;IACpE,SAAS,EAAE,CAAC,mBAAmB,EAAE,YAAY,CAAC;IAC9C,SAAS,EAAE,CAAC,mBAAmB,EAAE,YAAY,CAAC;IAC9C,cAAc,EAAE,CAAC,wBAAwB,EAAE,iBAAiB,CAAC;IAC7D,eAAe,EAAE,CAAC,yBAAyB,EAAE,kBAAkB,CAAC;IAChE,cAAc,EAAE,CAAC,wBAAwB,EAAE,iBAAiB,CAAC;CAChE,CAAC;AAEF,SAAS,kBAAkB,CAAC,IAAc;IACtC,KAAK,MAAM,GAAG,IAAI,IAAI,EAAE,CAAC;QACrB,MAAM,KAAK,GAAG,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,IAAI,EAAE,CAAC;QACvC,IAAI,KAAK,EAAE,CAAC;YACR,OAAO,KAAK,CAAC;QACjB,CAAC;IACL,CAAC;IAED,OAAO,SAAS,CAAC;AACrB,CAAC;AAED,SAAgB,aAAa,CAAC,SAAuB,EAAE;IACnD,OAAO;QACH,SAAS,EACL,MAAM,CAAC,SAAS;YAChB,kBAAkB,CAAC,eAAe,CAAC,SAAS,CAAC;YAC7C,6BAAqB,CAAC,SAAS;QACnC,SAAS,EACL,MAAM,CAAC,SAAS;YAChB,kBAAkB,CAAC,eAAe,CAAC,SAAS,CAAC;YAC7C,6BAAqB,CAAC,SAAS;QACnC,cAAc,EACV,MAAM,CAAC,cAAc;YACrB,kBAAkB,CAAC,eAAe,CAAC,cAAc,CAAC;YAClD,6BAAqB,CAAC,cAAc;QACxC,eAAe,EACX,MAAM,CAAC,eAAe;YACtB,kBAAkB,CAAC,eAAe,CAAC,eAAe,CAAC;YACnD,6BAAqB,CAAC,eAAe;QACzC,cAAc,EACV,MAAM,CAAC,cAAc;YACrB,kBAAkB,CAAC,eAAe,CAAC,cAAc,CAAC;YAClD,6BAAqB,CAAC,cAAc;KAC3C,CAAC;AACN,CAAC"}

package/dist/http-main.d.ts

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=http-main.d.ts.map

package/dist/http-main.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"http-main.d.ts","sourceRoot":"","sources":["../src/http-main.ts"],"names":[],"mappings":""}

package/dist/http-main.js

@@ -0,0 +1,9 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+const server_1 = require("./server");
+const PORT = process.env.PORT ?? 3000;
+const app = (0, server_1.createServer)();
+app.listen(PORT, () => {
+    console.log(`Memory RAG server running on http://localhost:${PORT}`);
+});
+//# sourceMappingURL=http-main.js.map

package/dist/http-main.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"http-main.js","sourceRoot":"","sources":["../src/http-main.ts"],"names":[],"mappings":";;AAAA,qCAAwC;AAExC,MAAM,IAAI,GAAG,OAAO,CAAC,GAAG,CAAC,IAAI,IAAI,IAAI,CAAC;AACtC,MAAM,GAAG,GAAG,IAAA,qBAAY,GAAE,CAAC;AAC3B,GAAG,CAAC,MAAM,CAAC,IAAI,EAAE,GAAG,EAAE;IAClB,OAAO,CAAC,GAAG,CAAC,iDAAiD,IAAI,EAAE,CAAC,CAAC;AACzE,CAAC,CAAC,CAAC"}

package/dist/hybrid.d.ts

@@ -0,0 +1,8 @@
+export declare function tokenize(text: string): string[];
+export declare function keywordScore(queryTokens: string[], docText: string): number;
+export interface RankedItem<T> {
+    item: T;
+    score: number;
+}
+export declare function reciprocalRankFusion<T>(rankings: RankedItem<T>[][], idFn: (item: T) => string, k?: number): RankedItem<T>[];
+//# sourceMappingURL=hybrid.d.ts.map

package/dist/hybrid.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"hybrid.d.ts","sourceRoot":"","sources":["../src/hybrid.ts"],"names":[],"mappings":"AAUA,wBAAgB,QAAQ,CAAC,IAAI,EAAE,MAAM,GAAG,MAAM,EAAE,CAK/C;AAED,wBAAgB,YAAY,CAAC,WAAW,EAAE,MAAM,EAAE,EAAE,OAAO,EAAE,MAAM,GAAG,MAAM,CAQ3E;AAED,MAAM,WAAW,UAAU,CAAC,CAAC;IACzB,IAAI,EAAE,CAAC,CAAC;IACR,KAAK,EAAE,MAAM,CAAC;CACjB;AAED,wBAAgB,oBAAoB,CAAC,CAAC,EAClC,QAAQ,EAAE,UAAU,CAAC,CAAC,CAAC,EAAE,EAAE,EAC3B,IAAI,EAAE,CAAC,IAAI,EAAE,CAAC,KAAK,MAAM,EACzB,CAAC,SAAK,GACP,UAAU,CAAC,CAAC,CAAC,EAAE,CAejB"}

package/dist/hybrid.js

@@ -0,0 +1,46 @@
+"use strict";
+// ── Hybrid search helpers ────────────────────────────────────────────────────
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.tokenize = tokenize;
+exports.keywordScore = keywordScore;
+exports.reciprocalRankFusion = reciprocalRankFusion;
+const STOP_WORDS = new Set([
+    "a", "an", "the", "is", "are", "was", "were", "be", "been", "being",
+    "have", "has", "had", "do", "does", "did", "will", "would", "shall",
+    "should", "may", "might", "must", "can", "could", "of", "in", "to",
+    "for", "with", "on", "at", "from", "by", "as", "into", "about",
+    "it", "its", "this", "that", "and", "or", "but", "not", "no", "so",
+]);
+function tokenize(text) {
+    return text
+        .toLowerCase()
+        .split(/[^a-z0-9]+/)
+        .filter((t) => t.length > 1 && !STOP_WORDS.has(t));
+}
+function keywordScore(queryTokens, docText) {
+    if (queryTokens.length === 0)
+        return 0;
+    const docLower = docText.toLowerCase();
+    let matches = 0;
+    for (const token of queryTokens) {
+        if (docLower.includes(token))
+            matches++;
+    }
+    return matches / queryTokens.length;
+}
+function reciprocalRankFusion(rankings, idFn, k = 60) {
+    const scores = new Map();
+    const items = new Map();
+    for (const ranking of rankings) {
+        for (let rank = 0; rank < ranking.length; rank++) {
+            const id = idFn(ranking[rank].item);
+            scores.set(id, (scores.get(id) ?? 0) + 1 / (k + rank + 1));
+            if (!items.has(id))
+                items.set(id, ranking[rank].item);
+        }
+    }
+    return Array.from(scores.entries())
+        .sort((a, b) => b[1] - a[1])
+        .map(([id, score]) => ({ item: items.get(id), score }));
+}
+//# sourceMappingURL=hybrid.js.map

package/dist/hybrid.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"hybrid.js","sourceRoot":"","sources":["../src/hybrid.ts"],"names":[],"mappings":";AAAA,gFAAgF;;AAUhF,4BAKC;AAED,oCAQC;AAOD,oDAmBC;AAjDD,MAAM,UAAU,GAAG,IAAI,GAAG,CAAC;IACvB,GAAG,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,KAAK,EAAE,KAAK,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO;IACnE,MAAM,EAAE,KAAK,EAAE,KAAK,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,OAAO,EAAE,OAAO;IACnE,QAAQ,EAAE,KAAK,EAAE,OAAO,EAAE,MAAM,EAAE,KAAK,EAAE,OAAO,EAAE,IAAI,EAAE,IAAI,EAAE,IAAI;IAClE,KAAK,EAAE,MAAM,EAAE,IAAI,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO;IAC9D,IAAI,EAAE,KAAK,EAAE,MAAM,EAAE,MAAM,EAAE,KAAK,EAAE,IAAI,EAAE,KAAK,EAAE,KAAK,EAAE,IAAI,EAAE,IAAI;CACrE,CAAC,CAAC;AAEH,SAAgB,QAAQ,CAAC,IAAY;IACjC,OAAO,IAAI;SACN,WAAW,EAAE;SACb,KAAK,CAAC,YAAY,CAAC;SACnB,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,MAAM,GAAG,CAAC,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;AAC3D,CAAC;AAED,SAAgB,YAAY,CAAC,WAAqB,EAAE,OAAe;IAC/D,IAAI,WAAW,CAAC,MAAM,KAAK,CAAC;QAAE,OAAO,CAAC,CAAC;IACvC,MAAM,QAAQ,GAAG,OAAO,CAAC,WAAW,EAAE,CAAC;IACvC,IAAI,OAAO,GAAG,CAAC,CAAC;IAChB,KAAK,MAAM,KAAK,IAAI,WAAW,EAAE,CAAC;QAC9B,IAAI,QAAQ,CAAC,QAAQ,CAAC,KAAK,CAAC;YAAE,OAAO,EAAE,CAAC;IAC5C,CAAC;IACD,OAAO,OAAO,GAAG,WAAW,CAAC,MAAM,CAAC;AACxC,CAAC;AAOD,SAAgB,oBAAoB,CAChC,QAA2B,EAC3B,IAAyB,EACzB,CAAC,GAAG,EAAE;IAEN,MAAM,MAAM,GAAG,IAAI,GAAG,EAAkB,CAAC;IACzC,MAAM,KAAK,GAAG,IAAI,GAAG,EAAa,CAAC;IAEnC,KAAK,MAAM,OAAO,IAAI,QAAQ,EAAE,CAAC;QAC7B,KAAK,IAAI,IAAI,GAAG,CAAC,EAAE,IAAI,GAAG,OAAO,CAAC,MAAM,EAAE,IAAI,EAAE,EAAE,CAAC;YAC/C,MAAM,EAAE,GAAG,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,CAAC;YACpC,MAAM,CAAC,GAAG,CAAC,EAAE,EAAE,CAAC,MAAM,CAAC,GAAG,CAAC,EAAE,CAAC,IAAI,CAAC,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC,GAAG,IAAI,GAAG,CAAC,CAAC,CAAC,CAAC;YAC3D,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,EAAE,CAAC;gBAAE,KAAK,CAAC,GAAG,CAAC,EAAE,EAAE,OAAO,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,CAAC;QAC1D,CAAC;IACL,CAAC;IAED,OAAO,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC;SAC9B,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;SAC3B,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,KAAK,CAAC,EAAE,EAAE,CAAC,CAAC,EAAE,IAAI,EAAE,KAAK,CAAC,GAAG,CAAC,EAAE,CAAE,EAAE,KAAK,EAAE,CAAC,CAAC,CAAC;AACjE,CAAC"}

package/dist/index.d.ts

@@ -1,6 +1,8 @@
 export { PieceStore } from "./store";
 export { RagPipeline } from "./rag";
 export { EmbeddingClient } from "./embeddings";
+export { MemoryMcpServer } from "./mcp";
 export { createServer } from "./server";
-export { Piece, MemoryConfig, DEFAULT_MEMORY_CONFIG, QueryOptions, QueryResult, RagResult, } from "./types";
+export { Piece, MemoryConfig, QueryOptions, QueryResult, RagResult, } from "./types";
+export { DEFAULT_MEMORY_CONFIG } from "./config";
 //# sourceMappingURL=index.d.ts.map

package/dist/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,MAAM,SAAS,CAAC;AACrC,OAAO,EAAE,WAAW,EAAE,MAAM,OAAO,CAAC;AACpC,OAAO,EAAE,eAAe,EAAE,MAAM,cAAc,CAAC;AAC/C,OAAO,EAAE,YAAY,EAAE,MAAM,UAAU,CAAC;AACxC,OAAO,EACH,KAAK,EACL,YAAY,EACZ,qBAAqB,EACrB,YAAY,EACZ,WAAW,EACX,SAAS,GACZ,MAAM,SAAS,CAAC"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,MAAM,SAAS,CAAC;AACrC,OAAO,EAAE,WAAW,EAAE,MAAM,OAAO,CAAC;AACpC,OAAO,EAAE,eAAe,EAAE,MAAM,cAAc,CAAC;AAC/C,OAAO,EAAE,eAAe,EAAE,MAAM,OAAO,CAAC;AACxC,OAAO,EAAE,YAAY,EAAE,MAAM,UAAU,CAAC;AACxC,OAAO,EACH,KAAK,EACL,YAAY,EACZ,YAAY,EACZ,WAAW,EACX,SAAS,GACZ,MAAM,SAAS,CAAC;AACjB,OAAO,EAAE,qBAAqB,EAAE,MAAM,UAAU,CAAC"}

package/dist/index.js

@@ -1,14 +1,16 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.DEFAULT_MEMORY_CONFIG = exports.createServer = exports.EmbeddingClient = exports.RagPipeline = exports.PieceStore = void 0;
+exports.DEFAULT_MEMORY_CONFIG = exports.createServer = exports.MemoryMcpServer = exports.EmbeddingClient = exports.RagPipeline = exports.PieceStore = void 0;
 var store_1 = require("./store");
 Object.defineProperty(exports, "PieceStore", { enumerable: true, get: function () { return store_1.PieceStore; } });
 var rag_1 = require("./rag");
 Object.defineProperty(exports, "RagPipeline", { enumerable: true, get: function () { return rag_1.RagPipeline; } });
 var embeddings_1 = require("./embeddings");
 Object.defineProperty(exports, "EmbeddingClient", { enumerable: true, get: function () { return embeddings_1.EmbeddingClient; } });
+var mcp_1 = require("./mcp");
+Object.defineProperty(exports, "MemoryMcpServer", { enumerable: true, get: function () { return mcp_1.MemoryMcpServer; } });
 var server_1 = require("./server");
 Object.defineProperty(exports, "createServer", { enumerable: true, get: function () { return server_1.createServer; } });
-var types_1 = require("./types");
-Object.defineProperty(exports, "DEFAULT_MEMORY_CONFIG", { enumerable: true, get: function () { return types_1.DEFAULT_MEMORY_CONFIG; } });
+var config_1 = require("./config");
+Object.defineProperty(exports, "DEFAULT_MEMORY_CONFIG", { enumerable: true, get: function () { return config_1.DEFAULT_MEMORY_CONFIG; } });
 //# sourceMappingURL=index.js.map

package/dist/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":";;;AAAA,iCAAqC;AAA5B,mGAAA,UAAU,OAAA;AACnB,6BAAoC;AAA3B,kGAAA,WAAW,OAAA;AACpB,2CAA+C;AAAtC,6GAAA,eAAe,OAAA;AACxB,mCAAwC;AAA/B,sGAAA,YAAY,OAAA;AACrB,iCAOiB;AAJb,8GAAA,qBAAqB,OAAA"}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":";;;AAAA,iCAAqC;AAA5B,mGAAA,UAAU,OAAA;AACnB,6BAAoC;AAA3B,kGAAA,WAAW,OAAA;AACpB,2CAA+C;AAAtC,6GAAA,eAAe,OAAA;AACxB,6BAAwC;AAA/B,sGAAA,eAAe,OAAA;AACxB,mCAAwC;AAA/B,sGAAA,YAAY,OAAA;AAQrB,mCAAiD;AAAxC,+GAAA,qBAAqB,OAAA"}

package/dist/main.d.ts

@@ -1,2 +1,3 @@
+#!/usr/bin/env node
 export {};
 //# sourceMappingURL=main.d.ts.map

package/dist/main.js

@@ -1,9 +1,10 @@
+#!/usr/bin/env node
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-const server_1 = require("./server");
-const PORT = process.env.PORT ?? 3000;
-const app = (0, server_1.createServer)();
-app.listen(PORT, () => {
-    console.log(`Memory RAG server running on http://localhost:${PORT}`);
+const mcp_1 = require("./mcp");
+const server = new mcp_1.MemoryMcpServer();
+server.start().catch((err) => {
+    console.error("Failed to start MCP server:", err);
+    process.exit(1);
 });
 //# sourceMappingURL=main.js.map

package/dist/main.js.map

@@ -1 +1 @@
-{"version":3,"file":"main.js","sourceRoot":"","sources":["../src/main.ts"],"names":[],"mappings":";;AAAA,qCAAwC;AAExC,MAAM,IAAI,GAAG,OAAO,CAAC,GAAG,CAAC,IAAI,IAAI,IAAI,CAAC;AACtC,MAAM,GAAG,GAAG,IAAA,qBAAY,GAAE,CAAC;AAC3B,GAAG,CAAC,MAAM,CAAC,IAAI,EAAE,GAAG,EAAE;IAClB,OAAO,CAAC,GAAG,CAAC,iDAAiD,IAAI,EAAE,CAAC,CAAC;AACzE,CAAC,CAAC,CAAC"}
+{"version":3,"file":"main.js","sourceRoot":"","sources":["../src/main.ts"],"names":[],"mappings":";;;AAEA,+BAAwC;AAExC,MAAM,MAAM,GAAG,IAAI,qBAAe,EAAE,CAAC;AACrC,MAAM,CAAC,KAAK,EAAE,CAAC,KAAK,CAAC,CAAC,GAAG,EAAE,EAAE;IACzB,OAAO,CAAC,KAAK,CAAC,6BAA6B,EAAE,GAAG,CAAC,CAAC;IAClD,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;AACpB,CAAC,CAAC,CAAC"}

package/dist/mcp.d.ts

@@ -0,0 +1,20 @@
+import { Transport } from "@modelcontextprotocol/sdk/shared/transport.js";
+import { MemoryConfig } from "./types";
+export declare class MemoryMcpServer {
+    private readonly mcpServer;
+    private readonly store;
+    private readonly rag;
+    private initPromise;
+    constructor(config?: MemoryConfig);
+    /**
+     * Start the server using the given transport (defaults to stdio).
+     */
+    start(transport?: Transport): Promise<void>;
+    /**
+     * Shut down the server and its transport.
+     */
+    close(): Promise<void>;
+    private ensureStoreInitialized;
+    private registerTools;
+}
+//# sourceMappingURL=mcp.d.ts.map

package/dist/mcp.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"mcp.d.ts","sourceRoot":"","sources":["../src/mcp.ts"],"names":[],"mappings":"AAGA,OAAO,EAAE,SAAS,EAAE,MAAM,+CAA+C,CAAC;AAI1E,OAAO,EAAE,YAAY,EAAiC,MAAM,SAAS,CAAC;AAuBtE,qBAAa,eAAe;IACxB,OAAO,CAAC,QAAQ,CAAC,SAAS,CAAY;IACtC,OAAO,CAAC,QAAQ,CAAC,KAAK,CAAa;IACnC,OAAO,CAAC,QAAQ,CAAC,GAAG,CAAc;IAClC,OAAO,CAAC,WAAW,CAA8B;gBAErC,MAAM,GAAE,YAAiB;IAiBrC;;OAEG;IACG,KAAK,CAAC,SAAS,CAAC,EAAE,SAAS,GAAG,OAAO,CAAC,IAAI,CAAC;IAKjD;;OAEG;IACG,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC;YAMd,sBAAsB;IAcpC,OAAO,CAAC,aAAa;CAwIxB"}

package/dist/mcp.js

@@ -0,0 +1,170 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.MemoryMcpServer = void 0;
+const mcp_js_1 = require("@modelcontextprotocol/sdk/server/mcp.js");
+const stdio_js_1 = require("@modelcontextprotocol/sdk/server/stdio.js");
+const zod_1 = require("zod");
+const store_1 = require("./store");
+const rag_1 = require("./rag");
+const config_1 = require("./config");
+const package_json_1 = __importDefault(require("../package.json"));
+// ──────────────────────────────────────────────────────────────────────────────
+// Helpers
+// ──────────────────────────────────────────────────────────────────────────────
+function toJsonText(payload) {
+    return JSON.stringify(payload, null, 2);
+}
+function toolResult(payload, isError = false) {
+    return {
+        content: [{ type: "text", text: toJsonText(payload) }],
+        ...(isError ? { isError: true } : {}),
+    };
+}
+// ──────────────────────────────────────────────────────────────────────────────
+// MCP Server
+// ──────────────────────────────────────────────────────────────────────────────
+class MemoryMcpServer {
+    mcpServer;
+    store;
+    rag;
+    initPromise = null;
+    constructor(config = {}) {
+        const resolved = (0, config_1.resolveConfig)(config);
+        this.store = new store_1.PieceStore(resolved);
+        this.rag = new rag_1.RagPipeline(this.store, resolved.ollamaUrl, resolved.generationModel);
+        this.mcpServer = new mcp_js_1.McpServer({
+            name: "memory",
+            version: package_json_1.default.version,
+        });
+        this.registerTools();
+    }
+    /**
+     * Start the server using the given transport (defaults to stdio).
+     */
+    async start(transport) {
+        const t = transport ?? new stdio_js_1.StdioServerTransport();
+        await this.mcpServer.connect(t);
+    }
+    /**
+     * Shut down the server and its transport.
+     */
+    async close() {
+        await this.mcpServer.close();
+    }
+    // ── Store lifecycle ─────────────────────────────────────────────────────
+    async ensureStoreInitialized() {
+        if (!this.initPromise) {
+            this.initPromise = this.store.init();
+        }
+        try {
+            await this.initPromise;
+        }
+        catch (err) {
+            this.initPromise = null;
+            throw err;
+        }
+    }
+    // ── Tool registration ───────────────────────────────────────────────────
+    registerTools() {
+        this.mcpServer.registerTool("add_piece", {
+            description: "Add a tagged text piece to the memory store.",
+            inputSchema: zod_1.z.object({
+                content: zod_1.z.string().describe("Piece content"),
+                title: zod_1.z.string().optional().describe("Optional title for display and retrieval context"),
+                tags: zod_1.z.array(zod_1.z.string()).optional().describe("Optional tags for filtering and retrieval"),
+                collection: zod_1.z.string().optional().describe("Optional collection name (defaults to the configured collection)"),
+            }),
+        }, async ({ content, title, tags, collection }) => {
+            await this.ensureStoreInitialized();
+            const piece = await this.store.addPiece(content, tags ?? [], title, collection);
+            return toolResult(piece);
+        });
+        this.mcpServer.registerTool("get_piece", {
+            description: "Get a piece by ID.",
+            inputSchema: zod_1.z.object({
+                id: zod_1.z.string().describe("Piece ID"),
+                collection: zod_1.z.string().optional().describe("Optional collection name (defaults to the configured collection)"),
+            }),
+        }, async ({ id, collection }) => {
+            await this.ensureStoreInitialized();
+            const piece = await this.store.getPiece(id, collection);
+            return toolResult({ found: !!piece, piece });
+        });
+        this.mcpServer.registerTool("update_piece", {
+            description: "Update a piece's content and/or tags.",
+            inputSchema: zod_1.z.object({
+                id: zod_1.z.string().describe("Piece ID"),
+                content: zod_1.z.string().optional().describe("New content (optional)"),
+                title: zod_1.z.union([zod_1.z.string(), zod_1.z.null()]).optional().describe("New title, or null to clear it (optional)"),
+                tags: zod_1.z.array(zod_1.z.string()).optional().describe("New tags (optional)"),
+                collection: zod_1.z.string().optional().describe("Optional collection name (defaults to the configured collection)"),
+            }),
+        }, async ({ id, content, title, tags, collection }) => {
+            await this.ensureStoreInitialized();
+            const piece = await this.store.updatePiece(id, content, tags, title, collection);
+            return toolResult({ found: !!piece, piece });
+        });
+        this.mcpServer.registerTool("delete_piece", {
+            description: "Delete a piece by ID.",
+            inputSchema: zod_1.z.object({
+                id: zod_1.z.string().describe("Piece ID"),
+                collection: zod_1.z.string().optional().describe("Optional collection name (defaults to the configured collection)"),
+            }),
+        }, async ({ id, collection }) => {
+            await this.ensureStoreInitialized();
+            await this.store.deletePiece(id, collection);
+            return toolResult({ deleted: true, id });
+        });
+        this.mcpServer.registerTool("query_pieces", {
+            description: "Run semantic search over pieces. Enable hybrid search to combine vector similarity with keyword matching via Reciprocal Rank Fusion.",
+            inputSchema: zod_1.z.object({
+                query: zod_1.z.string().describe("Semantic query text"),
+                tags: zod_1.z.array(zod_1.z.string()).optional().describe("Optional tag filter"),
+                topK: zod_1.z.number().int().positive().optional().describe("Maximum number of results (default: 10)"),
+                useHybridSearch: zod_1.z.boolean().optional().describe("Combine vector and keyword search via RRF (default: false)"),
+                collection: zod_1.z.string().optional().describe("Optional collection name (defaults to the configured collection)"),
+            }),
+        }, async ({ query, tags, topK, useHybridSearch, collection }) => {
+            await this.ensureStoreInitialized();
+            const results = await this.store.queryPieces(query, { tags, topK, useHybridSearch }, collection);
+            return toolResult(results);
+        });
+        this.mcpServer.registerTool("rag_query", {
+            description: "Run full RAG: retrieve relevant pieces and generate an answer.",
+            inputSchema: zod_1.z.object({
+                query: zod_1.z.string().describe("User question"),
+                tags: zod_1.z.array(zod_1.z.string()).optional().describe("Optional tag filter"),
+                topK: zod_1.z.number().int().positive().optional().describe("Maximum number of retrieved sources"),
+                useHybridSearch: zod_1.z.boolean().optional().describe("Combine vector and keyword search via RRF (default: false)"),
+                collection: zod_1.z.string().optional().describe("Optional collection name (defaults to the configured collection)"),
+            }),
+        }, async ({ query, tags, topK, useHybridSearch, collection }) => {
+            await this.ensureStoreInitialized();
+            const result = await this.rag.query(query, { tags, topK, useHybridSearch }, collection);
+            return toolResult(result);
+        });
+        this.mcpServer.registerTool("list_collections", {
+            description: "List all collection names in the memory store.",
+            inputSchema: zod_1.z.object({}),
+        }, async () => {
+            await this.ensureStoreInitialized();
+            const collections = await this.store.listCollections();
+            return toolResult({ collections });
+        });
+        this.mcpServer.registerTool("delete_collection", {
+            description: "Delete an entire collection and all its pieces from the memory store.",
+            inputSchema: zod_1.z.object({
+                collection: zod_1.z.string().describe("Name of the collection to delete"),
+            }),
+        }, async ({ collection }) => {
+            await this.ensureStoreInitialized();
+            await this.store.deleteCollection(collection);
+            return toolResult({ deleted: true, collection });
+        });
+    }
+}
+exports.MemoryMcpServer = MemoryMcpServer;
+//# sourceMappingURL=mcp.js.map

package/dist/mcp.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"mcp.js","sourceRoot":"","sources":["../src/mcp.ts"],"names":[],"mappings":";;;;;;AAAA,oEAAoE;AACpE,wEAAiF;AAGjF,6BAAwB;AACxB,mCAAqC;AACrC,+BAAoC;AAEpC,qCAAyC;AACzC,mEAA0C;AAE1C,iFAAiF;AACjF,UAAU;AACV,iFAAiF;AAEjF,SAAS,UAAU,CAAC,OAAgB;IAChC,OAAO,IAAI,CAAC,SAAS,CAAC,OAAO,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC;AAC5C,CAAC;AAED,SAAS,UAAU,CAAC,OAAgB,EAAE,OAAO,GAAG,KAAK;IACjD,OAAO;QACH,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,UAAU,CAAC,OAAO,CAAC,EAAE,CAAC;QACtD,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;KACxC,CAAC;AACN,CAAC;AAED,iFAAiF;AACjF,aAAa;AACb,iFAAiF;AAEjF,MAAa,eAAe;IACP,SAAS,CAAY;IACrB,KAAK,CAAa;IAClB,GAAG,CAAc;IAC1B,WAAW,GAAyB,IAAI,CAAC;IAEjD,YAAY,SAAuB,EAAE;QACjC,MAAM,QAAQ,GAAG,IAAA,sBAAa,EAAC,MAAM,CAAC,CAAC;QACvC,IAAI,CAAC,KAAK,GAAG,IAAI,kBAAU,CAAC,QAAQ,CAAC,CAAC;QACtC,IAAI,CAAC,GAAG,GAAG,IAAI,iBAAW,CACtB,IAAI,CAAC,KAAK,EACV,QAAQ,CAAC,SAAS,EAClB,QAAQ,CAAC,eAAe,CAC3B,CAAC;QAEF,IAAI,CAAC,SAAS,GAAG,IAAI,kBAAS,CAAC;YAC3B,IAAI,EAAE,QAAQ;YACd,OAAO,EAAE,sBAAW,CAAC,OAAO;SAC/B,CAAC,CAAC;QAEH,IAAI,CAAC,aAAa,EAAE,CAAC;IACzB,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,KAAK,CAAC,SAAqB;QAC7B,MAAM,CAAC,GAAG,SAAS,IAAI,IAAI,+BAAoB,EAAE,CAAC;QAClD,MAAM,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC;IACpC,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,KAAK;QACP,MAAM,IAAI,CAAC,SAAS,CAAC,KAAK,EAAE,CAAC;IACjC,CAAC;IAED,2EAA2E;IAEnE,KAAK,CAAC,sBAAsB;QAChC,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC;YACpB,IAAI,CAAC,WAAW,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,EAAE,CAAC;QACzC,CAAC;QACD,IAAI,CAAC;YACD,MAAM,IAAI,CAAC,WAAW,CAAC;QAC3B,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACX,IAAI,CAAC,WAAW,GAAG,IAAI,CAAC;YACxB,MAAM,GAAG,CAAC;QACd,CAAC;IACL,CAAC;IAED,2EAA2E;IAEnE,aAAa;QACjB,IAAI,CAAC,SAAS,CAAC,YAAY,CACvB,WAAW,EACX;YACI,WAAW,EAAE,8CAA8C;YAC3D,WAAW,EAAE,OAAC,CAAC,MAAM,CAAC;gBAClB,OAAO,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,eAAe,CAAC;gBAC7C,KAAK,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,kDAAkD,CAAC;gBACzF,IAAI,EAAE,OAAC,CAAC,KAAK,CAAC,OAAC,CAAC,MAAM,EAAE,CAAC,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,2CAA2C,CAAC;gBAC1F,UAAU,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,kEAAkE,CAAC;aACjH,CAAC;SACL,EACD,KAAK,EAAE,EAAE,OAAO,EAAE,KAAK,EAAE,IAAI,EAAE,UAAU,EAAE,EAA2B,EAAE;YACpE,MAAM,IAAI,CAAC,sBAAsB,EAAE,CAAC;YACpC,MAAM,KAAK,GAAU,MAAM,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,OAAO,EAAE,IAAI,IAAI,EAAE,EAAE,KAAK,EAAE,UAAU,CAAC,CAAC;YACvF,OAAO,UAAU,CAAC,KAAK,CAAC,CAAC;QAC7B,CAAC,CACJ,CAAC;QAEF,IAAI,CAAC,SAAS,CAAC,YAAY,CACvB,WAAW,EACX;YACI,WAAW,EAAE,oBAAoB;YACjC,WAAW,EAAE,OAAC,CAAC,MAAM,CAAC;gBAClB,EAAE,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,UAAU,CAAC;gBACnC,UAAU,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,kEAAkE,CAAC;aACjH,CAAC;SACL,EACD,KAAK,EAAE,EAAE,EAAE,EAAE,UAAU,EAAE,EAA2B,EAAE;YAClD,MAAM,IAAI,CAAC,sBAAsB,EAAE,CAAC;YACpC,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,EAAE,EAAE,UAAU,CAAC,CAAC;YACxD,OAAO,UAAU,CAAC,EAAE,KAAK,EAAE,CAAC,CAAC,KAAK,EAAE,KAAK,EAAE,CAAC,CAAC;QACjD,CAAC,CACJ,CAAC;QAEF,IAAI,CAAC,SAAS,CAAC,YAAY,CACvB,cAAc,EACd;YACI,WAAW,EAAE,uCAAuC;YACpD,WAAW,EAAE,OAAC,CAAC,MAAM,CAAC;gBAClB,EAAE,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,UAAU,CAAC;gBACnC,OAAO,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,wBAAwB,CAAC;gBACjE,KAAK,EAAE,OAAC,CAAC,KAAK,CAAC,CAAC,OAAC,CAAC,MAAM,EAAE,EAAE,OAAC,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,2CAA2C,CAAC;gBACvG,IAAI,EAAE,OAAC,CAAC,KAAK,CAAC,OAAC,CAAC,MAAM,EAAE,CAAC,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,qBAAqB,CAAC;gBACpE,UAAU,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,kEAAkE,CAAC;aACjH,CAAC;SACL,EACD,KAAK,EAAE,EAAE,EAAE,EAAE,OAAO,EAAE,KAAK,EAAE,IAAI,EAAE,UAAU,EAAE,EAA2B,EAAE;YACxE,MAAM,IAAI,CAAC,sBAAsB,EAAE,CAAC;YACpC,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,WAAW,CAAC,EAAE,EAAE,OAAO,EAAE,IAAI,EAAE,KAAK,EAAE,UAAU,CAAC,CAAC;YACjF,OAAO,UAAU,CAAC,EAAE,KAAK,EAAE,CAAC,CAAC,KAAK,EAAE,KAAK,EAAE,CAAC,CAAC;QACjD,CAAC,CACJ,CAAC;QAEF,IAAI,CAAC,SAAS,CAAC,YAAY,CACvB,cAAc,EACd;YACI,WAAW,EAAE,uBAAuB;YACpC,WAAW,EAAE,OAAC,CAAC,MAAM,CAAC;gBAClB,EAAE,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,UAAU,CAAC;gBACnC,UAAU,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,kEAAkE,CAAC;aACjH,CAAC;SACL,EACD,KAAK,EAAE,EAAE,EAAE,EAAE,UAAU,EAAE,EAA2B,EAAE;YAClD,MAAM,IAAI,CAAC,sBAAsB,EAAE,CAAC;YACpC,MAAM,IAAI,CAAC,KAAK,CAAC,WAAW,CAAC,EAAE,EAAE,UAAU,CAAC,CAAC;YAC7C,OAAO,UAAU,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE,EAAE,EAAE,CAAC,CAAC;QAC7C,CAAC,CACJ,CAAC;QAEF,IAAI,CAAC,SAAS,CAAC,YAAY,CACvB,cAAc,EACd;YACI,WAAW,EAAE,sIAAsI;YACnJ,WAAW,EAAE,OAAC,CAAC,MAAM,CAAC;gBAClB,KAAK,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,qBAAqB,CAAC;gBACjD,IAAI,EAAE,OAAC,CAAC,KAAK,CAAC,OAAC,CAAC,MAAM,EAAE,CAAC,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,qBAAqB,CAAC;gBACpE,IAAI,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,GAAG,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,yCAAyC,CAAC;gBAChG,eAAe,EAAE,OAAC,CAAC,OAAO,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,4DAA4D,CAAC;gBAC9G,UAAU,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,kEAAkE,CAAC;aACjH,CAAC;SACL,EACD,KAAK,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,IAAI,EAAE,eAAe,EAAE,UAAU,EAAE,EAA2B,EAAE;YAClF,MAAM,IAAI,CAAC,sBAAsB,EAAE,CAAC;YACpC,MAAM,OAAO,GAAkB,MAAM,IAAI,CAAC,KAAK,CAAC,WAAW,CAAC,KAAK,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,eAAe,EAAE,EAAE,UAAU,CAAC,CAAC;YAChH,OAAO,UAAU,CAAC,OAAO,CAAC,CAAC;QAC/B,CAAC,CACJ,CAAC;QAEF,IAAI,CAAC,SAAS,CAAC,YAAY,CACvB,WAAW,EACX;YACI,WAAW,EAAE,gEAAgE;YAC7E,WAAW,EAAE,OAAC,CAAC,MAAM,CAAC;gBAClB,KAAK,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,eAAe,CAAC;gBAC3C,IAAI,EAAE,OAAC,CAAC,KAAK,CAAC,OAAC,CAAC,MAAM,EAAE,CAAC,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,qBAAqB,CAAC;gBACpE,IAAI,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,GAAG,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,qCAAqC,CAAC;gBAC5F,eAAe,EAAE,OAAC,CAAC,OAAO,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,4DAA4D,CAAC;gBAC9G,UAAU,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,kEAAkE,CAAC;aACjH,CAAC;SACL,EACD,KAAK,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,IAAI,EAAE,eAAe,EAAE,UAAU,EAAE,EAA2B,EAAE;YAClF,MAAM,IAAI,CAAC,sBAAsB,EAAE,CAAC;YACpC,MAAM,MAAM,GAAc,MAAM,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,KAAK,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,eAAe,EAAE,EAAE,UAAU,CAAC,CAAC;YACnG,OAAO,UAAU,CAAC,MAAM,CAAC,CAAC;QAC9B,CAAC,CACJ,CAAC;QAEF,IAAI,CAAC,SAAS,CAAC,YAAY,CACvB,kBAAkB,EAClB;YACI,WAAW,EAAE,gDAAgD;YAC7D,WAAW,EAAE,OAAC,CAAC,MAAM,CAAC,EAAE,CAAC;SAC5B,EACD,KAAK,IAA6B,EAAE;YAChC,MAAM,IAAI,CAAC,sBAAsB,EAAE,CAAC;YACpC,MAAM,WAAW,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,eAAe,EAAE,CAAC;YACvD,OAAO,UAAU,CAAC,EAAE,WAAW,EAAE,CAAC,CAAC;QACvC,CAAC,CACJ,CAAC;QAEF,IAAI,CAAC,SAAS,CAAC,YAAY,CACvB,mBAAmB,EACnB;YACI,WAAW,EAAE,uEAAuE;YACpF,WAAW,EAAE,OAAC,CAAC,MAAM,CAAC;gBAClB,UAAU,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,kCAAkC,CAAC;aACtE,CAAC;SACL,EACD,KAAK,EAAE,EAAE,UAAU,EAAE,EAA2B,EAAE;YAC9C,MAAM,IAAI,CAAC,sBAAsB,EAAE,CAAC;YACpC,MAAM,IAAI,CAAC,KAAK,CAAC,gBAAgB,CAAC,UAAU,CAAC,CAAC;YAC9C,OAAO,UAAU,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE,UAAU,EAAE,CAAC,CAAC;QACrD,CAAC,CACJ,CAAC;IACN,CAAC;CACJ;AA9LD,0CA8LC"}

package/dist/rag.d.ts

@@ -5,6 +5,6 @@ export declare class RagPipeline {
     private readonly ollama;
     private readonly model;
     constructor(store: PieceStore, ollamaUrl: string, model: string);
-    query(query: string, options?: QueryOptions): Promise<RagResult>;
+    query(query: string, options?: QueryOptions, collection?: string): Promise<RagResult>;
 }
 //# sourceMappingURL=rag.d.ts.map

package/dist/rag.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"rag.d.ts","sourceRoot":"","sources":["../src/rag.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,UAAU,EAAE,MAAM,SAAS,CAAC;AACrC,OAAO,EAAE,YAAY,EAAE,SAAS,EAAE,MAAM,SAAS,CAAC;AAElD,qBAAa,WAAW;IACpB,OAAO,CAAC,QAAQ,CAAC,KAAK,CAAa;IACnC,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAS;IAChC,OAAO,CAAC,QAAQ,CAAC,KAAK,CAAS;gBAEnB,KAAK,EAAE,UAAU,EAAE,SAAS,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM;IAMzD,KAAK,CAAC,KAAK,EAAE,MAAM,EAAE,OAAO,GAAE,YAAiB,GAAG,OAAO,CAAC,SAAS,CAAC;CAuC7E"}
+{"version":3,"file":"rag.d.ts","sourceRoot":"","sources":["../src/rag.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,UAAU,EAAE,MAAM,SAAS,CAAC;AACrC,OAAO,EAAE,YAAY,EAAE,SAAS,EAAE,MAAM,SAAS,CAAC;AAElD,qBAAa,WAAW;IACpB,OAAO,CAAC,QAAQ,CAAC,KAAK,CAAa;IACnC,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAS;IAChC,OAAO,CAAC,QAAQ,CAAC,KAAK,CAAS;gBAEnB,KAAK,EAAE,UAAU,EAAE,SAAS,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM;IAMzD,KAAK,CAAC,KAAK,EAAE,MAAM,EAAE,OAAO,GAAE,YAAiB,EAAE,UAAU,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,SAAS,CAAC;CA+ClG"}

package/dist/rag.js

@@ -11,8 +11,8 @@ class RagPipeline {
         this.ollama = new ollama_1.Ollama({ host: ollamaUrl });
         this.model = model;
     }
-    async query(query, options = {}) {
-        const sources = await this.store.queryPieces(query, options);
+    async query(query, options = {}, collection) {
+        const sources = await this.store.queryPieces(query, options, collection);
         if (sources.length === 0) {
             return {
                 answer: "I don't have enough context to answer this question. " +
@@ -21,12 +21,19 @@ class RagPipeline {
             };
         }
         const contextBlock = sources
-            .map((s, i) => `[${i + 1}] (tags: ${s.piece.tags.join(", ")})\n${s.piece.content}`)
+            .map((s, i) => `[${i + 1}]${s.piece.title ? ` ${s.piece.title}` : ""} (tags: ${s.piece.tags.join(", ")})\n${s.piece.content}`)
             .join("\n\n");
-        const systemPrompt = "You are a helpful assistant. Answer the user's question based on the provided context. " +
-            "If the context does not contain enough information, say so. " +
-            "Cite sources by their number when relevant.";
-        const userPrompt = `Context:\n${contextBlock}\n\nQuestion: ${query}`;
+        const systemPrompt = "You are a retrieval-grounded assistant. Answer the user's question using only the provided context. " +
+            "Do not use outside knowledge, background assumptions, or examples that are not explicitly supported by the context. " +
+            "If the context is incomplete or does not answer the question, say that you do not have enough context instead of guessing. " +
+            "Keep the answer concise. Cite sources for every supported factual claim with source numbers like [1].";
+        const userPrompt = `Context:\n${contextBlock}\n\n` +
+            `Question: ${query}\n\n` +
+            "Instructions:\n" +
+            "- Use only the context above.\n" +
+            "- Do not mention tools, frameworks, companies, or facts that do not appear in the context.\n" +
+            "- If the context is insufficient, explicitly say so.\n" +
+            "- Cite supported statements with source numbers.";
         const response = await this.ollama.chat({
             model: this.model,
             messages: [

package/dist/rag.js.map

@@ -1 +1 @@
-{"version":3,"file":"rag.js","sourceRoot":"","sources":["../src/rag.ts"],"names":[],"mappings":";;;AAAA,mCAAgC;AAIhC,MAAa,WAAW;IACH,KAAK,CAAa;IAClB,MAAM,CAAS;IACf,KAAK,CAAS;IAE/B,YAAY,KAAiB,EAAE,SAAiB,EAAE,KAAa;QAC3D,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC;QACnB,IAAI,CAAC,MAAM,GAAG,IAAI,eAAM,CAAC,EAAE,IAAI,EAAE,SAAS,EAAE,CAAC,CAAC;QAC9C,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC;IACvB,CAAC;IAED,KAAK,CAAC,KAAK,CAAC,KAAa,EAAE,UAAwB,EAAE;QACjD,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,WAAW,CAAC,KAAK,EAAE,OAAO,CAAC,CAAC;QAE7D,IAAI,OAAO,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACvB,OAAO;gBACH,MAAM,EACF,uDAAuD;oBACvD,sDAAsD;gBAC1D,OAAO,EAAE,EAAE;aACd,CAAC;QACN,CAAC;QAED,MAAM,YAAY,GAAG,OAAO;aACvB,GAAG,CACA,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CACL,IAAI,CAAC,GAAG,CAAC,YAAY,CAAC,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,KAAK,CAAC,OAAO,EAAE,CAC1E;aACA,IAAI,CAAC,MAAM,CAAC,CAAC;QAElB,MAAM,YAAY,GACd,yFAAyF;YACzF,8DAA8D;YAC9D,6CAA6C,CAAC;QAElD,MAAM,UAAU,GAAG,aAAa,YAAY,iBAAiB,KAAK,EAAE,CAAC;QAErE,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC;YACpC,KAAK,EAAE,IAAI,CAAC,KAAK;YACjB,QAAQ,EAAE;gBACN,EAAE,IAAI,EAAE,QAAQ,EAAE,OAAO,EAAE,YAAY,EAAE;gBACzC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,UAAU,EAAE;aACxC;SACJ,CAAC,CAAC;QAEH,OAAO;YACH,MAAM,EAAE,QAAQ,CAAC,OAAO,CAAC,OAAO;YAChC,OAAO;SACV,CAAC;IACN,CAAC;CACJ;AAlDD,kCAkDC"}
+{"version":3,"file":"rag.js","sourceRoot":"","sources":["../src/rag.ts"],"names":[],"mappings":";;;AAAA,mCAAgC;AAIhC,MAAa,WAAW;IACH,KAAK,CAAa;IAClB,MAAM,CAAS;IACf,KAAK,CAAS;IAE/B,YAAY,KAAiB,EAAE,SAAiB,EAAE,KAAa;QAC3D,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC;QACnB,IAAI,CAAC,MAAM,GAAG,IAAI,eAAM,CAAC,EAAE,IAAI,EAAE,SAAS,EAAE,CAAC,CAAC;QAC9C,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC;IACvB,CAAC;IAED,KAAK,CAAC,KAAK,CAAC,KAAa,EAAE,UAAwB,EAAE,EAAE,UAAmB;QACtE,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,WAAW,CAAC,KAAK,EAAE,OAAO,EAAE,UAAU,CAAC,CAAC;QAEzE,IAAI,OAAO,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACvB,OAAO;gBACH,MAAM,EACF,uDAAuD;oBACvD,sDAAsD;gBAC1D,OAAO,EAAE,EAAE;aACd,CAAC;QACN,CAAC;QAED,MAAM,YAAY,GAAG,OAAO;aACvB,GAAG,CACA,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CACL,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,KAAK,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC,EAAE,WAAW,CAAC,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,KAAK,CAAC,OAAO,EAAE,CACrH;aACA,IAAI,CAAC,MAAM,CAAC,CAAC;QAElB,MAAM,YAAY,GACd,sGAAsG;YACtG,sHAAsH;YACtH,6HAA6H;YAC7H,uGAAuG,CAAC;QAE5G,MAAM,UAAU,GACZ,aAAa,YAAY,MAAM;YAC/B,aAAa,KAAK,MAAM;YACxB,iBAAiB;YACjB,iCAAiC;YACjC,8FAA8F;YAC9F,wDAAwD;YACxD,kDAAkD,CAAC;QAEvD,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC;YACpC,KAAK,EAAE,IAAI,CAAC,KAAK;YACjB,QAAQ,EAAE;gBACN,EAAE,IAAI,EAAE,QAAQ,EAAE,OAAO,EAAE,YAAY,EAAE;gBACzC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,UAAU,EAAE;aACxC;SACJ,CAAC,CAAC;QAEH,OAAO;YACH,MAAM,EAAE,QAAQ,CAAC,OAAO,CAAC,OAAO;YAChC,OAAO;SACV,CAAC;IACN,CAAC;CACJ;AA1DD,kCA0DC"}

package/dist/server.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"server.d.ts","sourceRoot":"","sources":["../src/server.ts"],"names":[],"mappings":"AAGA,OAAO,EAAE,YAAY,EAAyB,MAAM,SAAS,CAAC;AAE9D,wBAAgB,YAAY,CAAC,MAAM,GAAE,YAAiB,+CAwHrD"}
+{"version":3,"file":"server.d.ts","sourceRoot":"","sources":["../src/server.ts"],"names":[],"mappings":"AAGA,OAAO,EAAE,YAAY,EAAE,MAAM,SAAS,CAAC;AAiBvC,wBAAgB,YAAY,CAAC,MAAM,GAAE,YAAiB,+CA4MrD"}