@o-lang/semantic-doc-search 1.0.24 → 1.0.26

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@o-lang/semantic-doc-search",
-  "version": "1.0.24",
+  "version": "1.0.26",
   "description": "O-lang Semantic Document Search Resolver with hybrid search, embeddings, rerank, and streaming.",
   "main": "src/index.js",
   "type": "commonjs",

package/src/adapters/vectorCapabilities.js

@@ -1,29 +1,40 @@
+/**
+ * Capability & metadata for vector backends.
+ * No hardcoded "default" assumptions.
+ * Resolvers and VectorRouter decide which backend to use at runtime.
+ */
+
 module.exports = {
   pgvector: {
     persistent: true,
     offline: false,
     distance: "cosine",
-    maxDimension: 2000
+    maxDimension: 2000,
+    // capability declaration
+    capabilities: ["vector.insert", "vector.search"]
   },
 
   pinecone: {
     persistent: true,
     offline: false,
     distance: "cosine",
-    maxDimension: 1536
+    maxDimension: 1536,
+    capabilities: ["vector.insert", "vector.search"]
   },
 
   redis: {
     persistent: true,
     offline: false,
     distance: "cosine",
-    maxDimension: 2048
+    maxDimension: 2048,
+    capabilities: ["vector.insert", "vector.search"]
   },
 
   memory: {
     persistent: false,
     offline: true,
     distance: "cosine",
-    maxDimension: 4096
+    maxDimension: 4096,
+    capabilities: ["vector.insert", "vector.search"]
   }
-};
+};

package/src/adapters/vectorRouter.js

@@ -1,7 +1,22 @@
 class VectorRouter {
   static create(config = {}) {
-    const backend = config.backend || "pgvector";
+    // 1. Explicit backend override (advanced users only)
+    if (config.backend) {
+      return VectorRouter._createExplicitBackend(config.backend, config);
+    }
+
+    // 2. Auto-detect pgvector (DEFAULT)
+    if (config.POSTGRES_URL) {
+      const PgVectorAdapter = require("./pgvectorAdapter");
+      return new PgVectorAdapter(config);
+    }
+
+    // 3. Fallback to in-memory (safe default)
+    const InMemoryAdapter = require("./inMemoryAdapter");
+    return new InMemoryAdapter(config);
+  }
 
+  static _createExplicitBackend(backend, config) {
     switch (backend) {
       case "pgvector": {
         const PgVectorAdapter = require("./pgvectorAdapter");

package/src/index.js

@@ -246,27 +246,10 @@ async function performDocQA(query, context) {
 
 /* ---------------- O-LANG RESOLVER ---------------- */
 
-async function docSearchResolver(action, context) {
-  if (!action.startsWith("Ask doc-search")) return;
-
-  // Extract the query string
-  const match = action.match(/"(.*)"|'(.*)'/);
-  const query = match ? match[1] || match[2] : action.replace("Ask doc-search", "").trim();
-
-  // Optional: extract topK and minScore if provided in action, e.g. "Ask doc-search 'Vacation policy' topK=3 minScore=0.8"
-  let topK = 5;
-  let minScore = 0.75;
-
-  const topKMatch = action.match(/topK\s*=\s*(\d+)/i);
-  if (topKMatch) topK = parseInt(topKMatch[1], 10);
+const semanticResolver = require("./resolver"); // ✅ IMPORT NEW FILE
 
-  const minScoreMatch = action.match(/minScore\s*=\s*(0?\.\d+|1(\.0)?)/i);
-  if (minScoreMatch) minScore = parseFloat(minScoreMatch[1]);
-
-  // Pass these into context for hybrid search
-  const searchContext = { ...context, topK, minScore };
-
-  return performDocQA(query, searchContext);
+async function docSearchResolver(action, context) {
+  return semanticResolver(action, context);
 }
 
 docSearchResolver.resolverName = "doc-search";

package/src/resolver.js

@@ -0,0 +1,40 @@
+const VectorRouter = require("./adapters/vectorRouter");
+const { embed } = require("./embeddings/local");
+const { extractQuery } = require("./utils/extractQuery");
+const { formatResults } = require("./utils/formatResults");
+
+/**
+ * Semantic Doc Search Resolver
+ * Invoked by O-Lang runtime when workflow uses:
+ *   Ask doc-search: "..."
+ */
+async function resolver(action, context = {}) {
+  // Only handle doc-search actions
+  if (typeof action !== "string") return;
+  if (!action.toLowerCase().startsWith("ask doc-search")) return;
+
+  // 1️⃣ Extract natural language query
+  const query = extractQuery(action);
+
+  // 2️⃣ Select vector backend (pgvector if available, else fallback)
+  const vectorStore = VectorRouter.create(context);
+
+  if (!vectorStore.supports("vector.search")) {
+    throw new Error("Vector backend does not support vector.search");
+  }
+
+  // 3️⃣ Generate embedding
+  const vector = await embed(query);
+
+  // 4️⃣ Execute similarity search
+  const results = await vectorStore.query({
+    vector,
+    topK: context.topK || 5,
+    minScore: context.minScore || 0
+  });
+
+  // 5️⃣ Normalize output for workflow consumption
+  return formatResults(results, query);
+}
+
+module.exports = resolver;

package/src/utils/extractQuery.js

@@ -0,0 +1,18 @@
+/**
+ * Extracts the query from an O-Lang action.
+ * Example:
+ *   "Ask doc-search: vacation policy"
+ */
+function extractQuery(action) {
+  const match = action.match(/ask doc-search\s*:?\s*(.+)$/i);
+
+  if (!match || !match[1]) {
+    throw new Error("Invalid doc-search action format");
+  }
+
+  return match[1].trim();
+}
+
+module.exports = {
+  extractQuery
+};

package/src/utils/formatResults.js

@@ -0,0 +1,20 @@
+/**
+ * Normalizes vector search results for O-Lang workflows.
+ */
+function formatResults(results = [], query) {
+  return {
+    query,
+    matches: results.map(r => ({
+      id: r.id,
+      content: r.content,
+      source: r.source,
+      score: typeof r.score === "number"
+        ? Number(r.score.toFixed(4))
+        : undefined
+    }))
+  };
+}
+
+module.exports = {
+  formatResults
+};