@nxuss/lemma 0.6.0 → 0.7.0

package/dist/esm/cli/lemma-proxy.js

@@ -96,6 +96,15 @@ async function writeJson(file, data) {
 function projectHash(name) {
     return crypto.createHash('sha1').update(name).digest('hex').slice(0, 12);
 }
+function getCollectionName(projectName) {
+    let cleanName = projectName.replace(/[^a-zA-Z0-9._-]/g, '_');
+    cleanName = cleanName.replace(/^[^a-zA-Z0-9]+/, '');
+    cleanName = cleanName.replace(/[^a-zA-Z0-9]+$/, '');
+    if (cleanName.length < 3) {
+        cleanName = 'lemma_proj_' + cleanName;
+    }
+    return `lemma-cache-${cleanName}`;
+}
 function detectProject() {
     const pkgPath = path.join(process.cwd(), 'package.json');
     if (fs.existsSync(pkgPath)) {
@@ -609,7 +618,7 @@ async function semanticGet(provider, prompt, projectName = 'global') {
     let localHit = null;
     try {
         const collection = await chroma.getOrCreateCollection({
-            name: `lemma-cache-${projectName}`,
+            name: getCollectionName(projectName),
             embeddingFunction: dummyEmbeddingFunction,
             metadata: { "hnsw:space": "cosine" }
         });
@@ -648,7 +657,7 @@ async function semanticGet(provider, prompt, projectName = 'global') {
     // ─── Cross-Project Bug Telepathy Fallback ───
     try {
         const collections = await chroma.listCollections();
-        const projectCollections = collections.filter((c) => c.name.startsWith('lemma-cache-') && c.name !== `lemma-cache-${projectName}`);
+        const projectCollections = collections.filter((c) => c.name.startsWith('lemma-cache-') && c.name !== getCollectionName(projectName));
         for (const colInfo of projectCollections) {
             try {
                 const col = await chroma.getOrCreateCollection({
@@ -701,7 +710,7 @@ async function semanticSet(provider, prompt, data, projectName = 'global') {
         return;
     try {
         const collection = await chroma.getOrCreateCollection({
-            name: `lemma-cache-${projectName}`,
+            name: getCollectionName(projectName),
             embeddingFunction: dummyEmbeddingFunction,
             metadata: { "hnsw:space": "cosine" }
         });
@@ -1092,7 +1101,7 @@ class LemmaServer {
                 // Try to query ChromaDB for semantic cache count
                 try {
                     const collection = await chroma.getOrCreateCollection({
-                        name: `lemma-cache-${this.projectName}`,
+                        name: getCollectionName(this.projectName),
                         embeddingFunction: dummyEmbeddingFunction,
                         metadata: { "hnsw:space": "cosine" }
                     });
@@ -1148,28 +1157,46 @@ class LemmaServer {
                     const emb = await getEmbedding(query);
                     if (!emb)
                         throw new Error('Failed to generate embedding');
-                    const collection = await chroma.getOrCreateCollection({
-                        name: `lemma-cache-${this.projectName}`,
-                        embeddingFunction: dummyEmbeddingFunction,
-                        metadata: { "hnsw:space": "cosine" }
-                    });
-                    const results = await collection.query({
-                        queryEmbeddings: [emb],
-                        nResults: 10
-                    });
-                    const formatted = (results.ids[0] || []).map((id, i) => {
-                        const meta = results.metadatas[0][i];
-                        const dists = results.distances;
-                        const dist = (dists && dists[0] && dists[0][i] !== null && dists[0][i] !== undefined) ? dists[0][i] : 1.0;
-                        return {
-                            id,
-                            prompt: meta.prompt,
-                            response: JSON.parse(meta.response),
-                            timestamp: meta.timestamp,
-                            similarity: 1 - dist
-                        };
-                    });
-                    res.json({ results: formatted });
+                    let allResults = [];
+                    // 1. Query all collections starting with 'lemma-cache' for cross-project search
+                    const collections = await chroma.listCollections();
+                    const targetCollections = collections.filter((c) => c.name.startsWith('lemma-cache'));
+                    for (const colInfo of targetCollections) {
+                        try {
+                            const collection = await chroma.getOrCreateCollection({
+                                name: colInfo.name,
+                                embeddingFunction: dummyEmbeddingFunction,
+                                metadata: { "hnsw:space": "cosine" }
+                            });
+                            const results = await collection.query({
+                                queryEmbeddings: [emb],
+                                nResults: 5
+                            });
+                            const originProject = colInfo.name === 'lemma-cache' ? 'global' : colInfo.name.replace('lemma-cache-', '');
+                            (results.ids[0] || []).forEach((id, i) => {
+                                const meta = results.metadatas[0][i];
+                                const dists = results.distances;
+                                const dist = (dists && dists[0] && dists[0][i] !== null && dists[0][i] !== undefined) ? dists[0][i] : 1.0;
+                                let parsedResponse = meta.response;
+                                try {
+                                    parsedResponse = JSON.parse(meta.response);
+                                }
+                                catch { }
+                                allResults.push({
+                                    id,
+                                    prompt: meta.prompt,
+                                    response: parsedResponse,
+                                    timestamp: meta.timestamp,
+                                    similarity: 1 - dist,
+                                    project: originProject
+                                });
+                            });
+                        }
+                        catch { }
+                    }
+                    // 2. Sort all project results by similarity descending
+                    allResults.sort((a, b) => b.similarity - a.similarity);
+                    res.json({ results: allResults.slice(0, 10) });
                 }
                 catch (e) {
                     res.status(500).json({ error: e.message });
@@ -1198,7 +1225,7 @@ class LemmaServer {
                     }
                     // Clear Chroma DB collection
                     try {
-                        await chroma.deleteCollection({ name: `lemma-cache-${this.projectName}` }).catch(() => { });
+                        await chroma.deleteCollection({ name: getCollectionName(this.projectName) }).catch(() => { });
                     }
                     catch (e) {
                         console.error(`[CacheClear] Failed to clear Chroma collection: ${e.message}`);
@@ -1215,7 +1242,7 @@ class LemmaServer {
                     // Selective delete from Chroma DB collection using provider metadata filter
                     try {
                         const collection = await chroma.getOrCreateCollection({
-                            name: `lemma-cache-${this.projectName}`,
+                            name: getCollectionName(this.projectName),
                             embeddingFunction: dummyEmbeddingFunction,
                             metadata: { "hnsw:space": "cosine" }
                         });
@@ -1283,7 +1310,7 @@ class LemmaServer {
                         const emb = await getEmbedding("architectural design guidelines and core stack configuration decisions");
                         if (emb) {
                             const collection = await chroma.getOrCreateCollection({
-                                name: `lemma-cache-${this.projectName}`,
+                                name: getCollectionName(this.projectName),
                                 embeddingFunction: dummyEmbeddingFunction,
                                 metadata: { "hnsw:space": "cosine" }
                             });
@@ -1354,7 +1381,7 @@ class LemmaServer {
                         return res.status(500).json({ error: 'Failed to generate embedding' });
                     }
                     const collection = await chroma.getOrCreateCollection({
-                        name: `lemma-cache-${this.projectName}`,
+                        name: getCollectionName(this.projectName),
                         embeddingFunction: dummyEmbeddingFunction,
                         metadata: { "hnsw:space": "cosine" }
                     });