sweet-search 0.0.1 → 2.3.0

package/crates/wasm-router/pkg/query_router_wasm.d.ts

@@ -0,0 +1,36 @@
+/* tslint:disable */
+/* eslint-disable */
+
+/**
+ * Route labels
+ */
+export enum RouteMode {
+  Lexical = 0,
+  Semantic = 1,
+  Hybrid = 2,
+}
+
+export class RouteResult {
+  private constructor();
+  free(): void;
+  [Symbol.dispose](): void;
+  readonly mode_str: string;
+  mode: RouteMode;
+  confidence: number;
+  rejected: boolean;
+}
+
+/**
+ * Export features for debugging
+ */
+export function extract_features_js(query: string): Float32Array;
+
+/**
+ * Initialize panic hook for better error messages
+ */
+export function init(): void;
+
+/**
+ * Main entry point - route a query string
+ */
+export function route_query(query: string): RouteResult;

package/crates/wasm-router/pkg/query_router_wasm.js

@@ -0,0 +1,271 @@
+
+let imports = {};
+imports['__wbindgen_placeholder__'] = module.exports;
+
+function getArrayF32FromWasm0(ptr, len) {
+    ptr = ptr >>> 0;
+    return getFloat32ArrayMemory0().subarray(ptr / 4, ptr / 4 + len);
+}
+
+let cachedDataViewMemory0 = null;
+function getDataViewMemory0() {
+    if (cachedDataViewMemory0 === null || cachedDataViewMemory0.buffer.detached === true || (cachedDataViewMemory0.buffer.detached === undefined && cachedDataViewMemory0.buffer !== wasm.memory.buffer)) {
+        cachedDataViewMemory0 = new DataView(wasm.memory.buffer);
+    }
+    return cachedDataViewMemory0;
+}
+
+let cachedFloat32ArrayMemory0 = null;
+function getFloat32ArrayMemory0() {
+    if (cachedFloat32ArrayMemory0 === null || cachedFloat32ArrayMemory0.byteLength === 0) {
+        cachedFloat32ArrayMemory0 = new Float32Array(wasm.memory.buffer);
+    }
+    return cachedFloat32ArrayMemory0;
+}
+
+function getStringFromWasm0(ptr, len) {
+    ptr = ptr >>> 0;
+    return decodeText(ptr, len);
+}
+
+let cachedUint8ArrayMemory0 = null;
+function getUint8ArrayMemory0() {
+    if (cachedUint8ArrayMemory0 === null || cachedUint8ArrayMemory0.byteLength === 0) {
+        cachedUint8ArrayMemory0 = new Uint8Array(wasm.memory.buffer);
+    }
+    return cachedUint8ArrayMemory0;
+}
+
+function passStringToWasm0(arg, malloc, realloc) {
+    if (realloc === undefined) {
+        const buf = cachedTextEncoder.encode(arg);
+        const ptr = malloc(buf.length, 1) >>> 0;
+        getUint8ArrayMemory0().subarray(ptr, ptr + buf.length).set(buf);
+        WASM_VECTOR_LEN = buf.length;
+        return ptr;
+    }
+
+    let len = arg.length;
+    let ptr = malloc(len, 1) >>> 0;
+
+    const mem = getUint8ArrayMemory0();
+
+    let offset = 0;
+
+    for (; offset < len; offset++) {
+        const code = arg.charCodeAt(offset);
+        if (code > 0x7F) break;
+        mem[ptr + offset] = code;
+    }
+    if (offset !== len) {
+        if (offset !== 0) {
+            arg = arg.slice(offset);
+        }
+        ptr = realloc(ptr, len, len = offset + arg.length * 3, 1) >>> 0;
+        const view = getUint8ArrayMemory0().subarray(ptr + offset, ptr + len);
+        const ret = cachedTextEncoder.encodeInto(arg, view);
+
+        offset += ret.written;
+        ptr = realloc(ptr, len, offset, 1) >>> 0;
+    }
+
+    WASM_VECTOR_LEN = offset;
+    return ptr;
+}
+
+let cachedTextDecoder = new TextDecoder('utf-8', { ignoreBOM: true, fatal: true });
+cachedTextDecoder.decode();
+function decodeText(ptr, len) {
+    return cachedTextDecoder.decode(getUint8ArrayMemory0().subarray(ptr, ptr + len));
+}
+
+const cachedTextEncoder = new TextEncoder();
+
+if (!('encodeInto' in cachedTextEncoder)) {
+    cachedTextEncoder.encodeInto = function (arg, view) {
+        const buf = cachedTextEncoder.encode(arg);
+        view.set(buf);
+        return {
+            read: arg.length,
+            written: buf.length
+        };
+    }
+}
+
+let WASM_VECTOR_LEN = 0;
+
+const RouteResultFinalization = (typeof FinalizationRegistry === 'undefined')
+    ? { register: () => {}, unregister: () => {} }
+    : new FinalizationRegistry(ptr => wasm.__wbg_routeresult_free(ptr >>> 0, 1));
+
+/**
+ * Route labels
+ * @enum {0 | 1 | 2}
+ */
+const RouteMode = Object.freeze({
+    Lexical: 0, "0": "Lexical",
+    Semantic: 1, "1": "Semantic",
+    Hybrid: 2, "2": "Hybrid",
+});
+exports.RouteMode = RouteMode;
+
+/**
+ * Routing result
+ */
+class RouteResult {
+    static __wrap(ptr) {
+        ptr = ptr >>> 0;
+        const obj = Object.create(RouteResult.prototype);
+        obj.__wbg_ptr = ptr;
+        RouteResultFinalization.register(obj, obj.__wbg_ptr, obj);
+        return obj;
+    }
+    __destroy_into_raw() {
+        const ptr = this.__wbg_ptr;
+        this.__wbg_ptr = 0;
+        RouteResultFinalization.unregister(this);
+        return ptr;
+    }
+    free() {
+        const ptr = this.__destroy_into_raw();
+        wasm.__wbg_routeresult_free(ptr, 0);
+    }
+    /**
+     * @returns {string}
+     */
+    get mode_str() {
+        let deferred1_0;
+        let deferred1_1;
+        try {
+            const ret = wasm.routeresult_mode_str(this.__wbg_ptr);
+            deferred1_0 = ret[0];
+            deferred1_1 = ret[1];
+            return getStringFromWasm0(ret[0], ret[1]);
+        } finally {
+            wasm.__wbindgen_free(deferred1_0, deferred1_1, 1);
+        }
+    }
+    /**
+     * @returns {RouteMode}
+     */
+    get mode() {
+        const ret = wasm.__wbg_get_routeresult_mode(this.__wbg_ptr);
+        return ret;
+    }
+    /**
+     * @param {RouteMode} arg0
+     */
+    set mode(arg0) {
+        wasm.__wbg_set_routeresult_mode(this.__wbg_ptr, arg0);
+    }
+    /**
+     * @returns {number}
+     */
+    get confidence() {
+        const ret = wasm.__wbg_get_routeresult_confidence(this.__wbg_ptr);
+        return ret;
+    }
+    /**
+     * @param {number} arg0
+     */
+    set confidence(arg0) {
+        wasm.__wbg_set_routeresult_confidence(this.__wbg_ptr, arg0);
+    }
+    /**
+     * @returns {boolean}
+     */
+    get rejected() {
+        const ret = wasm.__wbg_get_routeresult_rejected(this.__wbg_ptr);
+        return ret !== 0;
+    }
+    /**
+     * @param {boolean} arg0
+     */
+    set rejected(arg0) {
+        wasm.__wbg_set_routeresult_rejected(this.__wbg_ptr, arg0);
+    }
+}
+if (Symbol.dispose) RouteResult.prototype[Symbol.dispose] = RouteResult.prototype.free;
+exports.RouteResult = RouteResult;
+
+/**
+ * Export features for debugging
+ * @param {string} query
+ * @returns {Float32Array}
+ */
+function extract_features_js(query) {
+    const ptr0 = passStringToWasm0(query, wasm.__wbindgen_malloc, wasm.__wbindgen_realloc);
+    const len0 = WASM_VECTOR_LEN;
+    const ret = wasm.extract_features_js(ptr0, len0);
+    var v2 = getArrayF32FromWasm0(ret[0], ret[1]).slice();
+    wasm.__wbindgen_free(ret[0], ret[1] * 4, 4);
+    return v2;
+}
+exports.extract_features_js = extract_features_js;
+
+/**
+ * Initialize panic hook for better error messages
+ */
+function init() {
+    wasm.init();
+}
+exports.init = init;
+
+/**
+ * Main entry point - route a query string
+ * @param {string} query
+ * @returns {RouteResult}
+ */
+function route_query(query) {
+    const ptr0 = passStringToWasm0(query, wasm.__wbindgen_malloc, wasm.__wbindgen_realloc);
+    const len0 = WASM_VECTOR_LEN;
+    const ret = wasm.route_query(ptr0, len0);
+    return RouteResult.__wrap(ret);
+}
+exports.route_query = route_query;
+
+exports.__wbg___wbindgen_throw_dd24417ed36fc46e = function(arg0, arg1) {
+    throw new Error(getStringFromWasm0(arg0, arg1));
+};
+
+exports.__wbg_error_7534b8e9a36f1ab4 = function(arg0, arg1) {
+    let deferred0_0;
+    let deferred0_1;
+    try {
+        deferred0_0 = arg0;
+        deferred0_1 = arg1;
+        console.error(getStringFromWasm0(arg0, arg1));
+    } finally {
+        wasm.__wbindgen_free(deferred0_0, deferred0_1, 1);
+    }
+};
+
+exports.__wbg_new_8a6f238a6ece86ea = function() {
+    const ret = new Error();
+    return ret;
+};
+
+exports.__wbg_stack_0ed75d68575b0f3c = function(arg0, arg1) {
+    const ret = arg1.stack;
+    const ptr1 = passStringToWasm0(ret, wasm.__wbindgen_malloc, wasm.__wbindgen_realloc);
+    const len1 = WASM_VECTOR_LEN;
+    getDataViewMemory0().setInt32(arg0 + 4 * 1, len1, true);
+    getDataViewMemory0().setInt32(arg0 + 4 * 0, ptr1, true);
+};
+
+exports.__wbindgen_init_externref_table = function() {
+    const table = wasm.__wbindgen_externrefs;
+    const offset = table.grow(4);
+    table.set(0, undefined);
+    table.set(offset + 0, undefined);
+    table.set(offset + 1, null);
+    table.set(offset + 2, true);
+    table.set(offset + 3, false);
+};
+
+const wasmPath = `${__dirname}/query_router_wasm_bg.wasm`;
+const wasmBytes = require('fs').readFileSync(wasmPath);
+const wasmModule = new WebAssembly.Module(wasmBytes);
+const wasm = exports.__wasm = new WebAssembly.Instance(wasmModule, imports).exports;
+
+wasm.__wbindgen_start();

package/crates/wasm-router/pkg/query_router_wasm_bg.wasm.d.ts

@@ -0,0 +1,19 @@
+/* tslint:disable */
+/* eslint-disable */
+export const memory: WebAssembly.Memory;
+export const __wbg_get_routeresult_confidence: (a: number) => number;
+export const __wbg_get_routeresult_mode: (a: number) => number;
+export const __wbg_get_routeresult_rejected: (a: number) => number;
+export const __wbg_routeresult_free: (a: number, b: number) => void;
+export const __wbg_set_routeresult_confidence: (a: number, b: number) => void;
+export const __wbg_set_routeresult_mode: (a: number, b: number) => void;
+export const __wbg_set_routeresult_rejected: (a: number, b: number) => void;
+export const extract_features_js: (a: number, b: number) => [number, number];
+export const route_query: (a: number, b: number) => number;
+export const routeresult_mode_str: (a: number) => [number, number];
+export const init: () => void;
+export const __wbindgen_free: (a: number, b: number, c: number) => void;
+export const __wbindgen_malloc: (a: number, b: number) => number;
+export const __wbindgen_realloc: (a: number, b: number, c: number, d: number) => number;
+export const __wbindgen_externrefs: WebAssembly.Table;
+export const __wbindgen_start: () => void;

package/mcp/config-gen.js

@@ -0,0 +1,121 @@
+#!/usr/bin/env node
+
+import path from 'node:path';
+import { fileURLToPath } from 'node:url';
+
+const __filename = fileURLToPath(import.meta.url);
+const __dirname = path.dirname(__filename);
+const PROJECT_ROOT = path.resolve(__dirname, '..');
+
+// ---------------------------------------------------------------------------
+// Profiles
+// ---------------------------------------------------------------------------
+
+function devLocal(platform) {
+  if (platform === 'claude') {
+    return {
+      mcpServers: {
+        'sweet-search': {
+          command: 'node',
+          args: ['./mcp/server.js'],
+        },
+      },
+    };
+  }
+
+  // codex
+  return [
+    '[mcp_servers.sweet-search]',
+    'command = "node"',
+    'args = ["./mcp/server.js"]',
+  ].join('\n');
+}
+
+function externalProject(platform, targetPath) {
+  const serverPath = path.resolve(PROJECT_ROOT, 'mcp', 'server.js');
+  const target = targetPath ? path.resolve(targetPath) : '/absolute/path/to/target-project';
+
+  if (platform === 'claude') {
+    return {
+      mcpServers: {
+        'sweet-search': {
+          command: 'node',
+          args: [serverPath, '--project-root', target],
+          env: {
+            SWEET_SEARCH_PROJECT_ROOT: target,
+          },
+        },
+      },
+    };
+  }
+
+  // codex
+  return [
+    '[mcp_servers.sweet-search]',
+    `command = "node"`,
+    `args = ["${serverPath}"]`,
+    `cwd = "${target}"`,
+    `env = { "SWEET_SEARCH_PROJECT_ROOT" = "${target}" }`,
+    '# enabled = true',
+    '# startup_timeout_sec = 15',
+    '# tool_timeout_sec = 120',
+    '# enabled_tools = ["search", "health"]',
+    '# disabled_tools = []',
+  ].join('\n');
+}
+
+function published(platform) {
+  if (platform === 'claude') {
+    return {
+      mcpServers: {
+        'sweet-search': {
+          command: 'npx',
+          args: ['-y', 'sweet-search-mcp'],
+        },
+      },
+    };
+  }
+
+  // codex
+  return [
+    '[mcp_servers.sweet-search]',
+    'command = "npx"',
+    'args = ["-y", "sweet-search-mcp"]',
+    '# env = { "SWEET_SEARCH_PROJECT_ROOT" = "/path/to/project" }',
+    '# enabled = true',
+    '# startup_timeout_sec = 15',
+    '# tool_timeout_sec = 120',
+  ].join('\n');
+}
+
+// ---------------------------------------------------------------------------
+// CLI
+// ---------------------------------------------------------------------------
+
+const args = process.argv.slice(2);
+const platform = args[0] || 'claude';
+const profileIdx = args.indexOf('--profile');
+const profile = profileIdx !== -1 ? args[profileIdx + 1] : 'dev-local';
+const targetIdx = args.indexOf('--target');
+const targetPath = targetIdx !== -1 ? args[targetIdx + 1] : undefined;
+
+if (!['claude', 'codex'].includes(platform)) {
+  console.error(`Usage: node mcp/config-gen.js [claude|codex] [--profile dev-local|external-project|published] [--target <path>]`);
+  process.exit(1);
+}
+
+const profiles = { 'dev-local': devLocal, 'external-project': externalProject, published };
+const generator = profiles[profile];
+
+if (!generator) {
+  console.error(`Unknown profile: ${profile}. Available: ${Object.keys(profiles).join(', ')}`);
+  process.exit(1);
+}
+
+const result = generator(platform, targetPath);
+
+if (typeof result === 'string') {
+  console.log(result);
+} else {
+  console.log(JSON.stringify(result, null, 2));
+}