@o-lang/semantic-doc-search 1.1.0 → 1.1.2

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@o-lang/semantic-doc-search",
-  "version": "1.1.0",
+  "version": "1.1.2",
   "description": "O-Lang semantic document search resolver with vector embeddings",
   "main": "src/index.js",
   "exports": {

package/src/resolver.js

@@ -36,125 +36,84 @@ function hashText(str) {
 }
 
 // ─────────────────────────────────────────────
-// 🔥 Parse O-Lang Action String
-// Format: Action <resolver> "<actionType>" "<arg1>" "<arg2>" ...
-// ─────────────────────────────────────────────
-function parseActionString(action) {
-  if (!action || typeof action !== "string") return null;
-  
-  // Remove "Action doc-search " prefix if present
-  const cleaned = action.replace(/^Action\s+doc-search\s+/, "").trim();
-  
-  // Extract quoted strings: "vector.search" "query" "./docs"
-  const matches = [...cleaned.matchAll(/"([^"]*)"/g)].map(m => m[1]);
-  
-  if (matches.length < 1) return null;
-  
-  return {
-    actionType: matches[0],           // "vector.search" or "vector.insert"
-    args: matches.slice(1),           // remaining args
-    raw: action
-  };
-}
-
-// ─────────────────────────────────────────────
-// 🔥 MAIN RESOLVER
+// 🔥 MAIN RESOLVER - Simplified: action type = resolver name
+// Format: Action vector.search "arg1" "arg2" ...
+// - 1 arg = ingest: "{doc_root}"
+// - 2+ args = search: "{query}" "{doc_root}"
 // ─────────────────────────────────────────────
 async function resolver(action, context = {}) {
   if (typeof action !== "string") return;
 
-  // Parse the action string
-  const parsed = parseActionString(action);
-  if (!parsed) {
-    console.error("❌ Invalid doc-search action format:", action);
-    throw new Error("Invalid doc-search action format");
-  }
+  // Extract ALL quoted args from action string
+  const args = [...action.matchAll(/"([^"]*)"/g)].map(m => m[1]);
 
-  const { actionType, args } = parsed;
-  
   const vectorStore = VectorRouter.create(context);
   const getEmbedFn = await embedder({ dimension: 384 });
 
-  const doc_root = context.doc_root || args[1] || "./docs";
-  const useCache = !!context.POSTGRES_URL || !!context.REDIS_URL;
-  const cache = useCache ? loadCache() : {};
+  // ✅ Always use cache for local persistence — no Postgres/Redis required
+  const useCache = true;
+  const cache = loadCache();
 
   // =====================================================
-  // ✅ 1. VECTOR INSERT (INGEST)
+  // ✅ INGEST: 1 arg = doc_root
   // =====================================================
-  if (actionType.includes("vector.insert")) {
+  if (args.length === 1) {
     let inserted = 0;
-    const ingestRoot = args[0] || doc_root;
+    const ingestRoot = args[0] || context.doc_root || "./docs";
 
     if (fs.existsSync(ingestRoot)) {
       const files = fs.readdirSync(ingestRoot);
-
       for (const file of files) {
         const fullPath = path.join(ingestRoot, file);
         if (!fs.statSync(fullPath).isFile()) continue;
-
         const content = fs.readFileSync(fullPath, "utf8");
         if (!content) continue;
-
-        const chunkText = require("./utils/chunker").chunkText;
-        const chunks = chunkText(content, 500, 50);
-
+        const chunks = require("./utils/chunker").chunkText(content, 500, 50);
         for (let i = 0; i < chunks.length; i++) {
           const text = sanitizeTextForEmbedding(chunks[i]);
           if (!text) continue;
-
           const hash = hashText(text);
-          if (useCache && cache[hash]) continue;
-
+          if (cache[hash]) continue; // skip already-ingested chunks
           const rawVector = await getEmbedFn(text);
-          const vector = Array.from(rawVector);
-
           await vectorStore.upsert({
             id: `${file}:${i}`,
-            vector,
+            vector: Array.from(rawVector),
             content: text,
             source: `file:${file}`,
           });
-
-          if (useCache) cache[hash] = true;
+          cache[hash] = true;
           inserted++;
         }
       }
     }
 
-    if (useCache) saveCache(cache);
+    saveCache(cache);
     if (vectorStore.close) await vectorStore.close();
-
+    console.log(`[vector.search] ✅ Ingested ${inserted} chunks from ${ingestRoot}`);
     return { inserted, doc_root: ingestRoot };
   }
 
   // =====================================================
-  // ✅ 2. VECTOR SEARCH
+  // ✅ SEARCH: 2+ args = query + doc_root
   // =====================================================
-  if (actionType.includes("vector.search")) {
-    const query = sanitizeTextForEmbedding(args[0] || extractQuery(action) || "");
+  if (args.length >= 2) {
+    const query = sanitizeTextForEmbedding(args[0]);
     if (!query) return { text: "", matches: [] };
-
     const rawQueryVector = await getEmbedFn(query);
-    const queryVector = Array.from(rawQueryVector);
-
-    const results = await vectorStore.query(queryVector, {
-      topK: context.topK || 5,
-    });
-
+    const results = await vectorStore.query(Array.from(rawQueryVector), { topK: context.topK || 5 });
     if (vectorStore.close) await vectorStore.close();
-
     return formatResults(results, query);
   }
 
   // =====================================================
-  // ❌ Unknown action
+  // ❌ Unknown action format
   // =====================================================
-  console.warn(`⚠️  Unknown doc-search action: "${actionType}"`);
+  console.warn(`⚠️  Unknown vector.search action format: "${action}"`);
   return;
 }
 
-resolver.resolverName = "doc-search";
-resolver.version = "1.0.42";
+// ✅ Must match workflow's "Allow resolvers: - vector.search"
+resolver.resolverName = "vector.search";
+resolver.version = "1.0.43";
 
 module.exports = resolver;