@fraziym/axoniz 0.1.0

package/dist/integrations/mcp.js

@@ -0,0 +1,307 @@
+/**
+ * Generic MCP (Model Context Protocol) Client
+ * ============================================
+ * Speaks JSON-RPC 2.0 over HTTP to any MCP server.
+ * Used to connect axoniz to any MCP-compatible tool server
+ * (MemPalace, Goose extensions, browser, filesystem, etc.)
+ *
+ * Port of axoniz/integrations/mcp.py
+ *
+ * Port note: `@modelcontextprotocol/sdk` is NOT present in this workspace's
+ * node_modules, and this port must not add dependencies — so, exactly like the
+ * Python original, the transport is hand-rolled JSON-RPC 2.0 over HTTP (global
+ * `fetch` replaces `urllib.request`). The public surface is unchanged; if the
+ * SDK is installed later it can back `_rpc()` without touching callers.
+ * `urllib.error.URLError` → `MCPConnectionError` (any transport or HTTP-status
+ * failure), `RuntimeError("MCP error ...")` → a plain `Error`.
+ */
+export class MCPConnectionError extends Error {
+    constructor(message) {
+        super(message);
+        this.name = "MCPConnectionError";
+    }
+}
+function reasonOf(e) {
+    if (e instanceof Error) {
+        const cause = e.cause;
+        if (cause instanceof Error && cause.message)
+            return `${e.message} (${cause.message})`;
+        return e.message;
+    }
+    return String(e);
+}
+export class MCPClient {
+    static _id = 0;
+    base_url;
+    timeout;
+    _initialized = false;
+    _server_info = {};
+    _tools_cache = null;
+    constructor(baseUrl, timeout = 30) {
+        this.base_url = baseUrl.replace(/\/+$/, "");
+        this.timeout = timeout;
+    }
+    _next_id() {
+        MCPClient._id += 1;
+        return MCPClient._id;
+    }
+    async _rpc(method, params = null) {
+        const payload = {
+            jsonrpc: "2.0",
+            id: this._next_id(),
+            method,
+            params: params ?? {},
+        };
+        const endpoint = `${this.base_url}/mcp`;
+        let res;
+        try {
+            res = await fetch(endpoint, {
+                method: "POST",
+                headers: { "Content-Type": "application/json" },
+                body: JSON.stringify(payload),
+                signal: AbortSignal.timeout(Math.max(1, this.timeout) * 1000),
+            });
+        }
+        catch (e) {
+            throw new MCPConnectionError(`MCP server unreachable at ${this.base_url}: ${reasonOf(e)}`);
+        }
+        if (!res.ok) {
+            throw new MCPConnectionError(`MCP server unreachable at ${this.base_url}: HTTP ${res.status} ${res.statusText}`);
+        }
+        let resp;
+        try {
+            resp = JSON.parse(await res.text());
+        }
+        catch (e) {
+            throw new MCPConnectionError(`MCP server unreachable at ${this.base_url}: ${reasonOf(e)}`);
+        }
+        const err = resp["error"];
+        if (err) {
+            throw new Error(`MCP error ${String(err["code"])}: ${String(err["message"])}`);
+        }
+        return (resp["result"] ?? {});
+    }
+    /** Perform the MCP handshake. Call once before using tools. */
+    async initialize() {
+        if (this._initialized) {
+            return this._server_info;
+        }
+        const result = await this._rpc("initialize", {
+            protocolVersion: "2024-11-05",
+            capabilities: { tools: {}, resources: {}, prompts: {} },
+            clientInfo: { name: "axoniz", version: "1.0.0" },
+        });
+        this._server_info = result;
+        this._initialized = true;
+        return result;
+    }
+    async is_available() {
+        try {
+            await this.initialize();
+            return true;
+        }
+        catch {
+            return false;
+        }
+    }
+    /** Return the list of available tools from the MCP server. */
+    async list_tools(forceRefresh = false) {
+        if (this._tools_cache !== null && !forceRefresh) {
+            return this._tools_cache;
+        }
+        if (!this._initialized) {
+            await this.initialize();
+        }
+        const result = await this._rpc("tools/list");
+        const tools = (result["tools"] ?? []);
+        this._tools_cache = tools;
+        return tools;
+    }
+    /** Call an MCP tool and return its text output. */
+    async call_tool(name, args = null) {
+        if (!this._initialized) {
+            await this.initialize();
+        }
+        const result = await this._rpc("tools/call", {
+            name,
+            arguments: args ?? {},
+        });
+        // Extract text content from the MCP ContentBlock list
+        const content = result["content"];
+        if (Array.isArray(content)) {
+            const parts = [];
+            for (const block of content) {
+                if (block && typeof block === "object") {
+                    const b = block;
+                    if (b["type"] === "text") {
+                        parts.push(String(b["text"] ?? ""));
+                    }
+                    else if (b["type"] === "image") {
+                        parts.push(`[image: ${String(b["url"] ?? "")}]`);
+                    }
+                }
+            }
+            return parts.join("\n").trim();
+        }
+        return JSON.stringify(result);
+    }
+    async list_resources() {
+        if (!this._initialized) {
+            await this.initialize();
+        }
+        const result = await this._rpc("resources/list");
+        return (result["resources"] ?? []);
+    }
+    async read_resource(uri) {
+        if (!this._initialized) {
+            await this.initialize();
+        }
+        const result = await this._rpc("resources/read", { uri });
+        const contents = (result["contents"] ?? []);
+        if (contents.length > 0) {
+            return String(contents[0]["text"] ?? JSON.stringify(contents[0]));
+        }
+        return JSON.stringify(result);
+    }
+    async list_prompts() {
+        if (!this._initialized) {
+            await this.initialize();
+        }
+        const result = await this._rpc("prompts/list");
+        return (result["prompts"] ?? []);
+    }
+    /**
+     * Dynamically build an axoniz tool map from discovered MCP tools.
+     * Each MCP tool becomes an async callable.
+     */
+    async as_axoniz_tool_map() {
+        const tools = await this.list_tools();
+        const toolMap = {};
+        for (const t of tools) {
+            const name = String(t["name"] ?? "");
+            if (!name)
+                continue;
+            toolMap[name] = (...args) => {
+                const kwargs = (args[0] ?? {});
+                return this.call_tool(name, kwargs);
+            };
+        }
+        return toolMap;
+    }
+    /** Convert MCP tool schemas to OpenAI function-calling format. */
+    async as_axoniz_schemas() {
+        const tools = await this.list_tools();
+        return tools.map((t) => ({
+            type: "function",
+            function: {
+                name: String(t["name"] ?? ""),
+                description: String(t["description"] ?? ""),
+                parameters: t["inputSchema"] ?? {
+                    type: "object",
+                    properties: {},
+                    required: [],
+                },
+            },
+        }));
+    }
+    async server_info() {
+        if (!this._initialized) {
+            await this.initialize();
+        }
+        return this._server_info;
+    }
+    /* camelCase aliases */
+    asAxonizToolMap() {
+        return this.as_axoniz_tool_map();
+    }
+    asAxonizSchemas() {
+        return this.as_axoniz_schemas();
+    }
+    serverInfo() {
+        return this.server_info();
+    }
+}
+/**
+ * Manages multiple MCP server connections and merges their tools into a single
+ * unified tool map + schema list for axoniz.
+ */
+export class MCPRegistry {
+    _clients = new Map();
+    add(name, baseUrl, timeout = 30) {
+        this._clients.set(name, new MCPClient(baseUrl, timeout));
+        return this;
+    }
+    remove(name) {
+        this._clients.delete(name);
+    }
+    async available_servers() {
+        const names = [];
+        for (const [name, client] of this._clients) {
+            if (await client.is_available())
+                names.push(name);
+        }
+        return names;
+    }
+    async merged_tool_map() {
+        const merged = {};
+        for (const client of this._clients.values()) {
+            try {
+                if (await client.is_available()) {
+                    Object.assign(merged, await client.as_axoniz_tool_map());
+                }
+            }
+            catch {
+                /* skip unreachable servers */
+            }
+        }
+        return merged;
+    }
+    async merged_schemas() {
+        const schemas = [];
+        for (const client of this._clients.values()) {
+            try {
+                if (await client.is_available()) {
+                    schemas.push(...(await client.as_axoniz_schemas()));
+                }
+            }
+            catch {
+                /* skip unreachable servers */
+            }
+        }
+        return schemas;
+    }
+    /** Call a tool on whichever server has it. */
+    async call(toolName, args = null) {
+        for (const client of this._clients.values()) {
+            try {
+                if (!(await client.is_available()))
+                    continue;
+                const tools = await client.list_tools();
+                if (tools.some((t) => t["name"] === toolName)) {
+                    return await client.call_tool(toolName, args);
+                }
+            }
+            catch {
+                continue;
+            }
+        }
+        return `[ERROR] Tool '${toolName}' not found in any connected MCP server`;
+    }
+    /* camelCase aliases */
+    availableServers() {
+        return this.available_servers();
+    }
+    mergedToolMap() {
+        return this.merged_tool_map();
+    }
+    mergedSchemas() {
+        return this.merged_schemas();
+    }
+}
+/* Module-level shared registry */
+const _registry = new MCPRegistry();
+export function get_registry() {
+    return _registry;
+}
+export const getRegistry = get_registry;
+//# sourceMappingURL=mcp.js.map

package/dist/integrations/mcp.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"mcp.js","sourceRoot":"","sources":["../../src/integrations/mcp.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;;;GAgBG;AAIH,MAAM,OAAO,kBAAmB,SAAQ,KAAK;IAC3C,YAAY,OAAe;QACzB,KAAK,CAAC,OAAO,CAAC,CAAC;QACf,IAAI,CAAC,IAAI,GAAG,oBAAoB,CAAC;IACnC,CAAC;CACF;AAED,SAAS,QAAQ,CAAC,CAAU;IAC1B,IAAI,CAAC,YAAY,KAAK,EAAE,CAAC;QACvB,MAAM,KAAK,GAAI,CAAyB,CAAC,KAAK,CAAC;QAC/C,IAAI,KAAK,YAAY,KAAK,IAAI,KAAK,CAAC,OAAO;YAAE,OAAO,GAAG,CAAC,CAAC,OAAO,KAAK,KAAK,CAAC,OAAO,GAAG,CAAC;QACtF,OAAO,CAAC,CAAC,OAAO,CAAC;IACnB,CAAC;IACD,OAAO,MAAM,CAAC,CAAC,CAAC,CAAC;AACnB,CAAC;AAED,MAAM,OAAO,SAAS;IACV,MAAM,CAAC,GAAG,GAAG,CAAC,CAAC;IAEzB,QAAQ,CAAS;IACjB,OAAO,CAAS;IACN,YAAY,GAAG,KAAK,CAAC;IACrB,YAAY,GAA4B,EAAE,CAAC;IAC3C,YAAY,GAA0C,IAAI,CAAC;IAErE,YAAY,OAAe,EAAE,OAAO,GAAG,EAAE;QACvC,IAAI,CAAC,QAAQ,GAAG,OAAO,CAAC,OAAO,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC;QAC5C,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;IACzB,CAAC;IAED,QAAQ;QACN,SAAS,CAAC,GAAG,IAAI,CAAC,CAAC;QACnB,OAAO,SAAS,CAAC,GAAG,CAAC;IACvB,CAAC;IAED,KAAK,CAAC,IAAI,CAAC,MAAc,EAAE,MAAM,GAAY,IAAI;QAC/C,MAAM,OAAO,GAAG;YACd,OAAO,EAAE,KAAK;YACd,EAAE,EAAE,IAAI,CAAC,QAAQ,EAAE;YACnB,MAAM;YACN,MAAM,EAAE,MAAM,IAAI,EAAE;SACrB,CAAC;QACF,MAAM,QAAQ,GAAG,GAAG,IAAI,CAAC,QAAQ,MAAM,CAAC;QAExC,IAAI,GAAa,CAAC;QAClB,IAAI,CAAC;YACH,GAAG,GAAG,MAAM,KAAK,CAAC,QAAQ,EAAE;gBAC1B,MAAM,EAAE,MAAM;gBACd,OAAO,EAAE,EAAE,cAAc,EAAE,kBAAkB,EAAE;gBAC/C,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC;gBAC7B,MAAM,EAAE,WAAW,CAAC,OAAO,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,OAAO,CAAC,GAAG,IAAI,CAAC;aAC9D,CAAC,CAAC;QACL,CAAC;QAAC,OAAO,CAAC,EAAE,CAAC;YACX,MAAM,IAAI,kBAAkB,CAAC,6BAA6B,IAAI,CAAC,QAAQ,KAAK,QAAQ,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;QAC7F,CAAC;QAED,IAAI,CAAC,GAAG,CAAC,EAAE,EAAE,CAAC;YACZ,MAAM,IAAI,kBAAkB,CAC1B,6BAA6B,IAAI,CAAC,QAAQ,UAAU,GAAG,CAAC,MAAM,IAAI,GAAG,CAAC,UAAU,EAAE,CACnF,CAAC;QACJ,CAAC;QAED,IAAI,IAA6B,CAAC;QAClC,IAAI,CAAC;YACH,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,IAAI,EAAE,CAA4B,CAAC;QACjE,CAAC;QAAC,OAAO,CAAC,EAAE,CAAC;YACX,MAAM,IAAI,kBAAkB,CAAC,6BAA6B,IAAI,CAAC,QAAQ,KAAK,QAAQ,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;QAC7F,CAAC;QAED,MAAM,GAAG,GAAG,IAAI,CAAC,OAAO,CAAwC,CAAC;QACjE,IAAI,GAAG,EAAE,CAAC;YACR,MAAM,IAAI,KAAK,CAAC,aAAa,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,KAAK,MAAM,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC,EAAE,CAAC,CAAC;QACjF,CAAC;QACD,OAAO,CAAC,IAAI,CAAC,QAAQ,CAAC,IAAI,EAAE,CAA4B,CAAC;IAC3D,CAAC;IAED,+DAA+D;IAC/D,KAAK,CAAC,UAAU;QACd,IAAI,IAAI,CAAC,YAAY,EAAE,CAAC;YACtB,OAAO,IAAI,CAAC,YAAY,CAAC;QAC3B,CAAC;QACD,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC,YAAY,EAAE;YAC3C,eAAe,EAAE,YAAY;YAC7B,YAAY,EAAE,EAAE,KAAK,EAAE,EAAE,EAAE,SAAS,EAAE,EAAE,EAAE,OAAO,EAAE,EAAE,EAAE;YACvD,UAAU,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,OAAO,EAAE,OAAO,EAAE;SACjD,CAAC,CAAC;QACH,IAAI,CAAC,YAAY,GAAG,MAAM,CAAC;QAC3B,IAAI,CAAC,YAAY,GAAG,IAAI,CAAC;QACzB,OAAO,MAAM,CAAC;IAChB,CAAC;IAED,KAAK,CAAC,YAAY;QAChB,IAAI,CAAC;YACH,MAAM,IAAI,CAAC,UAAU,EAAE,CAAC;YACxB,OAAO,IAAI,CAAC;QACd,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,KAAK,CAAC;QACf,CAAC;IACH,CAAC;IAED,8DAA8D;IAC9D,KAAK,CAAC,UAAU,CAAC,YAAY,GAAG,KAAK;QACnC,IAAI,IAAI,CAAC,YAAY,KAAK,IAAI,IAAI,CAAC,YAAY,EAAE,CAAC;YAChD,OAAO,IAAI,CAAC,YAAY,CAAC;QAC3B,CAAC;QACD,IAAI,CAAC,IAAI,CAAC,YAAY,EAAE,CAAC;YACvB,MAAM,IAAI,CAAC,UAAU,EAAE,CAAC;QAC1B,CAAC;QACD,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;QAC7C,MAAM,KAAK,GAAG,CAAC,MAAM,CAAC,OAAO,CAAC,IAAI,EAAE,CAAmC,CAAC;QACxE,IAAI,CAAC,YAAY,GAAG,KAAK,CAAC;QAC1B,OAAO,KAAK,CAAC;IACf,CAAC;IAED,mDAAmD;IACnD,KAAK,CAAC,SAAS,CAAC,IAAY,EAAE,IAAI,GAAmC,IAAI;QACvE,IAAI,CAAC,IAAI,CAAC,YAAY,EAAE,CAAC;YACvB,MAAM,IAAI,CAAC,UAAU,EAAE,CAAC;QAC1B,CAAC;QACD,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC,YAAY,EAAE;YAC3C,IAAI;YACJ,SAAS,EAAE,IAAI,IAAI,EAAE;SACtB,CAAC,CAAC;QACH,sDAAsD;QACtD,MAAM,OAAO,GAAG,MAAM,CAAC,SAAS,CAAC,CAAC;QAClC,IAAI,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC;YAC3B,MAAM,KAAK,GAAa,EAAE,CAAC;YAC3B,KAAK,MAAM,KAAK,IAAI,OAAO,EAAE,CAAC;gBAC5B,IAAI,KAAK,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;oBACvC,MAAM,CAAC,GAAG,KAAgC,CAAC;oBAC3C,IAAI,CAAC,CAAC,MAAM,CAAC,KAAK,MAAM,EAAE,CAAC;wBACzB,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,MAAM,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC;oBACtC,CAAC;yBAAM,IAAI,CAAC,CAAC,MAAM,CAAC,KAAK,OAAO,EAAE,CAAC;wBACjC,KAAK,CAAC,IAAI,CAAC,WAAW,MAAM,CAAC,CAAC,CAAC,KAAK,CAAC,IAAI,EAAE,CAAC,GAAG,CAAC,CAAC;oBACnD,CAAC;gBACH,CAAC;YACH,CAAC;YACD,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,CAAC;QACjC,CAAC;QACD,OAAO,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;IAChC,CAAC;IAED,KAAK,CAAC,cAAc;QAClB,IAAI,CAAC,IAAI,CAAC,YAAY,EAAE,CAAC;YACvB,MAAM,IAAI,CAAC,UAAU,EAAE,CAAC;QAC1B,CAAC;QACD,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC;QACjD,OAAO,CAAC,MAAM,CAAC,WAAW,CAAC,IAAI,EAAE,CAAmC,CAAC;IACvE,CAAC;IAED,KAAK,CAAC,aAAa,CAAC,GAAW;QAC7B,IAAI,CAAC,IAAI,CAAC,YAAY,EAAE,CAAC;YACvB,MAAM,IAAI,CAAC,UAAU,EAAE,CAAC;QAC1B,CAAC;QACD,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC,gBAAgB,EAAE,EAAE,GAAG,EAAE,CAAC,CAAC;QAC1D,MAAM,QAAQ,GAAG,CAAC,MAAM,CAAC,UAAU,CAAC,IAAI,EAAE,CAAmC,CAAC;QAC9E,IAAI,QAAQ,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACxB,OAAO,MAAM,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,IAAI,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;QACpE,CAAC;QACD,OAAO,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;IAChC,CAAC;IAED,KAAK,CAAC,YAAY;QAChB,IAAI,CAAC,IAAI,CAAC,YAAY,EAAE,CAAC;YACvB,MAAM,IAAI,CAAC,UAAU,EAAE,CAAC;QAC1B,CAAC;QACD,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;QAC/C,OAAO,CAAC,MAAM,CAAC,SAAS,CAAC,IAAI,EAAE,CAAmC,CAAC;IACrE,CAAC;IAED;;;OAGG;IACH,KAAK,CAAC,kBAAkB;QACtB,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,UAAU,EAAE,CAAC;QACtC,MAAM,OAAO,GAA8B,EAAE,CAAC;QAC9C,KAAK,MAAM,CAAC,IAAI,KAAK,EAAE,CAAC;YACtB,MAAM,IAAI,GAAG,MAAM,CAAC,CAAC,CAAC,MAAM,CAAC,IAAI,EAAE,CAAC,CAAC;YACrC,IAAI,CAAC,IAAI;gBAAE,SAAS;YACpB,OAAO,CAAC,IAAI,CAAC,GAAG,CAAC,GAAG,IAAW,EAAE,EAAE;gBACjC,MAAM,MAAM,GAAG,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,EAAE,CAA4B,CAAC;gBAC1D,OAAO,IAAI,CAAC,SAAS,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;YACtC,CAAC,CAAC;QACJ,CAAC;QACD,OAAO,OAAO,CAAC;IACjB,CAAC;IAED,kEAAkE;IAClE,KAAK,CAAC,iBAAiB;QACrB,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,UAAU,EAAE,CAAC;QACtC,OAAO,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;YACvB,IAAI,EAAE,UAAU;YAChB,QAAQ,EAAE;gBACR,IAAI,EAAE,MAAM,CAAC,CAAC,CAAC,MAAM,CAAC,IAAI,EAAE,CAAC;gBAC7B,WAAW,EAAE,MAAM,CAAC,CAAC,CAAC,aAAa,CAAC,IAAI,EAAE,CAAC;gBAC3C,UAAU,EACP,CAAC,CAAC,aAAa,CAAyC,IAAI;oBAC3D,IAAI,EAAE,QAAQ;oBACd,UAAU,EAAE,EAAE;oBACd,QAAQ,EAAE,EAAE;iBACb;aACJ;SACF,CAAC,CAAC,CAAC;IACN,CAAC;IAED,KAAK,CAAC,WAAW;QACf,IAAI,CAAC,IAAI,CAAC,YAAY,EAAE,CAAC;YACvB,MAAM,IAAI,CAAC,UAAU,EAAE,CAAC;QAC1B,CAAC;QACD,OAAO,IAAI,CAAC,YAAY,CAAC;IAC3B,CAAC;IAED,uBAAuB;IACvB,eAAe;QACb,OAAO,IAAI,CAAC,kBAAkB,EAAE,CAAC;IACnC,CAAC;IACD,eAAe;QACb,OAAO,IAAI,CAAC,iBAAiB,EAAE,CAAC;IAClC,CAAC;IACD,UAAU;QACR,OAAO,IAAI,CAAC,WAAW,EAAE,CAAC;IAC5B,CAAC;CACF;AAED;;;GAGG;AACH,MAAM,OAAO,WAAW;IACZ,QAAQ,GAAG,IAAI,GAAG,EAAqB,CAAC;IAElD,GAAG,CAAC,IAAY,EAAE,OAAe,EAAE,OAAO,GAAG,EAAE;QAC7C,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,IAAI,EAAE,IAAI,SAAS,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC,CAAC;QACzD,OAAO,IAAI,CAAC;IACd,CAAC;IAED,MAAM,CAAC,IAAY;QACjB,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;IAC7B,CAAC;IAED,KAAK,CAAC,iBAAiB;QACrB,MAAM,KAAK,GAAa,EAAE,CAAC;QAC3B,KAAK,MAAM,CAAC,IAAI,EAAE,MAAM,CAAC,IAAI,IAAI,CAAC,QAAQ,EAAE,CAAC;YAC3C,IAAI,MAAM,MAAM,CAAC,YAAY,EAAE;gBAAE,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACpD,CAAC;QACD,OAAO,KAAK,CAAC;IACf,CAAC;IAED,KAAK,CAAC,eAAe;QACnB,MAAM,MAAM,GAA8B,EAAE,CAAC;QAC7C,KAAK,MAAM,MAAM,IAAI,IAAI,CAAC,QAAQ,CAAC,MAAM,EAAE,EAAE,CAAC;YAC5C,IAAI,CAAC;gBACH,IAAI,MAAM,MAAM,CAAC,YAAY,EAAE,EAAE,CAAC;oBAChC,MAAM,CAAC,MAAM,CAAC,MAAM,EAAE,MAAM,MAAM,CAAC,kBAAkB,EAAE,CAAC,CAAC;gBAC3D,CAAC;YACH,CAAC;YAAC,MAAM,CAAC;gBACP,8BAA8B;YAChC,CAAC;QACH,CAAC;QACD,OAAO,MAAM,CAAC;IAChB,CAAC;IAED,KAAK,CAAC,cAAc;QAClB,MAAM,OAAO,GAAmC,EAAE,CAAC;QACnD,KAAK,MAAM,MAAM,IAAI,IAAI,CAAC,QAAQ,CAAC,MAAM,EAAE,EAAE,CAAC;YAC5C,IAAI,CAAC;gBACH,IAAI,MAAM,MAAM,CAAC,YAAY,EAAE,EAAE,CAAC;oBAChC,OAAO,CAAC,IAAI,CAAC,GAAG,CAAC,MAAM,MAAM,CAAC,iBAAiB,EAAE,CAAC,CAAC,CAAC;gBACtD,CAAC;YACH,CAAC;YAAC,MAAM,CAAC;gBACP,8BAA8B;YAChC,CAAC;QACH,CAAC;QACD,OAAO,OAAO,CAAC;IACjB,CAAC;IAED,8CAA8C;IAC9C,KAAK,CAAC,IAAI,CAAC,QAAgB,EAAE,IAAI,GAAmC,IAAI;QACtE,KAAK,MAAM,MAAM,IAAI,IAAI,CAAC,QAAQ,CAAC,MAAM,EAAE,EAAE,CAAC;YAC5C,IAAI,CAAC;gBACH,IAAI,CAAC,CAAC,MAAM,MAAM,CAAC,YAAY,EAAE,CAAC;oBAAE,SAAS;gBAC7C,MAAM,KAAK,GAAG,MAAM,MAAM,CAAC,UAAU,EAAE,CAAC;gBACxC,IAAI,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,QAAQ,CAAC,EAAE,CAAC;oBAC9C,OAAO,MAAM,MAAM,CAAC,SAAS,CAAC,QAAQ,EAAE,IAAI,CAAC,CAAC;gBAChD,CAAC;YACH,CAAC;YAAC,MAAM,CAAC;gBACP,SAAS;YACX,CAAC;QACH,CAAC;QACD,OAAO,iBAAiB,QAAQ,yCAAyC,CAAC;IAC5E,CAAC;IAED,uBAAuB;IACvB,gBAAgB;QACd,OAAO,IAAI,CAAC,iBAAiB,EAAE,CAAC;IAClC,CAAC;IACD,aAAa;QACX,OAAO,IAAI,CAAC,eAAe,EAAE,CAAC;IAChC,CAAC;IACD,aAAa;QACX,OAAO,IAAI,CAAC,cAAc,EAAE,CAAC;IAC/B,CAAC;CACF;AAED,kCAAkC;AAClC,MAAM,SAAS,GAAG,IAAI,WAAW,EAAE,CAAC;AAEpC,MAAM,UAAU,YAAY;IAC1B,OAAO,SAAS,CAAC;AACnB,CAAC;AAED,MAAM,CAAC,MAAM,WAAW,GAAG,YAAY,CAAC"}

package/dist/integrations/mempalace.d.ts

@@ -0,0 +1,18 @@
+/**
+ * axoniz/integrations/mempalace.ts
+ * =================================
+ * Legacy shim — MemPalaceBridge and MemPalaceMemory are now part of
+ * axoniz.integrations.unified_memory.UnifiedMemory.
+ *
+ * Port of axoniz/integrations/mempalace.py
+ *
+ * Kept so existing code that imports MemPalaceBridge / MemPalaceMemory /
+ * get_bridge() keeps working. `MemPalaceBridge` IS `PalaceLayer` and
+ * `MemPalaceMemory` IS `UnifiedMemory`, exactly as in Python.
+ */
+import { PalaceLayer, UnifiedMemory } from "./unified_memory.js";
+export { PalaceLayer as MemPalaceBridge, UnifiedMemory as MemPalaceMemory };
+export { PalaceLayer, UnifiedMemory };
+export declare function get_bridge(_baseUrl?: string | null): PalaceLayer;
+export declare const getBridge: typeof get_bridge;
+//# sourceMappingURL=mempalace.d.ts.map

package/dist/integrations/mempalace.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"mempalace.d.ts","sourceRoot":"","sources":["../../src/integrations/mempalace.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;GAWG;AACH,OAAO,EAAE,WAAW,EAAE,aAAa,EAAE,MAAM,qBAAqB,CAAC;AAEjE,OAAO,EAAE,WAAW,IAAI,eAAe,EAAE,aAAa,IAAI,eAAe,EAAE,CAAC;AAC5E,OAAO,EAAE,WAAW,EAAE,aAAa,EAAE,CAAC;AAKtC,wBAAgB,UAAU,CAAC,QAAQ,CAAC,EAAE,MAAM,GAAG,IAAI,GAAG,WAAW,CAKhE;AAED,eAAO,MAAM,SAAS,mBAAa,CAAC"}

package/dist/integrations/mempalace.js

@@ -0,0 +1,25 @@
+/**
+ * axoniz/integrations/mempalace.ts
+ * =================================
+ * Legacy shim — MemPalaceBridge and MemPalaceMemory are now part of
+ * axoniz.integrations.unified_memory.UnifiedMemory.
+ *
+ * Port of axoniz/integrations/mempalace.py
+ *
+ * Kept so existing code that imports MemPalaceBridge / MemPalaceMemory /
+ * get_bridge() keeps working. `MemPalaceBridge` IS `PalaceLayer` and
+ * `MemPalaceMemory` IS `UnifiedMemory`, exactly as in Python.
+ */
+import { PalaceLayer, UnifiedMemory } from "./unified_memory.js";
+export { PalaceLayer as MemPalaceBridge, UnifiedMemory as MemPalaceMemory };
+export { PalaceLayer, UnifiedMemory };
+/** Backwards-compat singleton (Python's `get_bridge()` ignored its argument). */
+let _bridge = null;
+export function get_bridge(_baseUrl) {
+    if (_bridge === null) {
+        _bridge = new PalaceLayer();
+    }
+    return _bridge;
+}
+export const getBridge = get_bridge;
+//# sourceMappingURL=mempalace.js.map

package/dist/integrations/mempalace.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"mempalace.js","sourceRoot":"","sources":["../../src/integrations/mempalace.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;GAWG;AACH,OAAO,EAAE,WAAW,EAAE,aAAa,EAAE,MAAM,qBAAqB,CAAC;AAEjE,OAAO,EAAE,WAAW,IAAI,eAAe,EAAE,aAAa,IAAI,eAAe,EAAE,CAAC;AAC5E,OAAO,EAAE,WAAW,EAAE,aAAa,EAAE,CAAC;AAEtC,iFAAiF;AACjF,IAAI,OAAO,GAAuB,IAAI,CAAC;AAEvC,MAAM,UAAU,UAAU,CAAC,QAAwB;IACjD,IAAI,OAAO,KAAK,IAAI,EAAE,CAAC;QACrB,OAAO,GAAG,IAAI,WAAW,EAAE,CAAC;IAC9B,CAAC;IACD,OAAO,OAAO,CAAC;AACjB,CAAC;AAED,MAAM,CAAC,MAAM,SAAS,GAAG,UAAU,CAAC"}

package/dist/integrations/multi_agent.d.ts

@@ -0,0 +1,82 @@
+/** Tasks matching these patterns route to Goose (complex multi-step code tasks). */
+export declare const GOOSE_PATTERNS: string[];
+/** Tasks for Claude (cloud, high reasoning, long-form). */
+export declare const CLAUDE_PATTERNS: string[];
+/** Memory-heavy tasks route to the memory bridge first. */
+export declare const MEMORY_PATTERNS: string[];
+export declare function _matches_any(text: string, patterns: string[]): boolean;
+export declare const matchesAny: typeof _matches_any;
+export type AnyIterable<T> = AsyncIterable<T> | Iterable<T>;
+export interface RouterAgentLike {
+    config: Record<string, unknown>;
+    run(task: string): unknown;
+    chat_stream?(message: string): AnyIterable<string>;
+    chatStream?(message: string): AnyIterable<string>;
+}
+export interface GooseBridgeLike {
+    is_available(): boolean | Promise<boolean>;
+    ask(task: string): unknown;
+    ask_stream?(task: string): AnyIterable<string>;
+    askStream?(task: string): AnyIterable<string>;
+}
+export interface MemoryBridgeLike {
+    is_available(): boolean | Promise<boolean>;
+    search(query: string, options?: {
+        top_k?: number;
+        limit?: number;
+    }): unknown;
+}
+export type AgentSwitchCallback = (agent: string) => void;
+/**
+ * Routes tasks to the best available agent.
+ *
+ * Priority order (configurable):
+ *   1. Check memory patterns → inject recalled context first
+ *   2. If Goose is available + matches goose patterns → delegate to Goose
+ *   3. If Claude is available + matches claude patterns → delegate to Claude
+ *   4. Default → axoniz local agent
+ */
+export declare class MultiAgentRouter {
+    axoniz: RouterAgentLike | null;
+    goose: GooseBridgeLike | null;
+    mempalace: MemoryBridgeLike | null;
+    on_token: ((token: string) => void) | null;
+    on_agent_switch: AgentSwitchCallback | null;
+    protected _custom_agents: Map<string, (task: string) => unknown>;
+    constructor(options?: {
+        axoniz_agent?: RouterAgentLike | null;
+        goose_bridge?: GooseBridgeLike | null;
+        mempalace_bridge?: MemoryBridgeLike | null;
+        on_token?: ((token: string) => void) | null;
+        on_agent_switch?: AgentSwitchCallback | null;
+    });
+    /** Register a custom agent handler: `handler(task) -> string` */
+    register(name: string, handler: (task: string) => unknown): this;
+    /**
+     * Route a task to the best agent and return the result.
+     *
+     * mode: "auto" | "axoniz" | "goose" | "claude"
+     */
+    route(task: string, mode?: string): Promise<string>;
+    /** Stream route — yields `[agent_name, token]` tuples. */
+    route_stream(task: string, mode?: string): AsyncGenerator<[string, string]>;
+    /** Predict which agent would handle this task without running it. */
+    which_agent(task: string): Promise<string>;
+    /** Availability status of all connected agents. */
+    status(): Promise<Record<string, unknown>>;
+    protected _should_use_goose(task: string): Promise<boolean>;
+    protected _should_use_claude(task: string): boolean;
+    protected _memoryPrefix(task: string): Promise<string>;
+    protected _run_goose(task: string): Promise<string>;
+    protected _run_axoniz(task: string): Promise<string>;
+    protected _run_claude(task: string): Promise<string>;
+    protected _stream_goose(task: string): AsyncGenerator<string>;
+    /** Stream axoniz chat (not a full agent run, for simplicity). */
+    protected _stream_axoniz(task: string): AsyncGenerator<string>;
+    protected _stream_claude(task: string): AsyncGenerator<string>;
+    routeStream(task: string, mode?: string): AsyncGenerator<[string, string]>;
+    whichAgent(task: string): Promise<string>;
+}
+export declare function get_router(axonizAgent?: RouterAgentLike | null): Promise<MultiAgentRouter>;
+export declare const getRouter: typeof get_router;
+//# sourceMappingURL=multi_agent.d.ts.map

package/dist/integrations/multi_agent.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"multi_agent.d.ts","sourceRoot":"","sources":["../../src/integrations/multi_agent.ts"],"names":[],"mappings":"AAwBA,oFAAoF;AACpF,eAAO,MAAM,cAAc,EAAE,MAAM,EAelC,CAAC;AAEF,2DAA2D;AAC3D,eAAO,MAAM,eAAe,EAAE,MAAM,EAQnC,CAAC;AAEF,2DAA2D;AAC3D,eAAO,MAAM,eAAe,EAAE,MAAM,EASnC,CAAC;AAEF,wBAAgB,YAAY,CAAC,IAAI,EAAE,MAAM,EAAE,QAAQ,EAAE,MAAM,EAAE,GAAG,OAAO,CAGtE;AAED,eAAO,MAAM,UAAU,qBAAe,CAAC;AAIvC,MAAM,MAAM,WAAW,CAAC,CAAC,IAAI,aAAa,CAAC,CAAC,CAAC,GAAG,QAAQ,CAAC,CAAC,CAAC,CAAC;AAE5D,MAAM,WAAW,eAAe;IAC9B,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;IAChC,GAAG,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC;IAC3B,WAAW,CAAC,CAAC,OAAO,EAAE,MAAM,GAAG,WAAW,CAAC,MAAM,CAAC,CAAC;IACnD,UAAU,CAAC,CAAC,OAAO,EAAE,MAAM,GAAG,WAAW,CAAC,MAAM,CAAC,CAAC;CACnD;AAED,MAAM,WAAW,eAAe;IAC9B,YAAY,IAAI,OAAO,GAAG,OAAO,CAAC,OAAO,CAAC,CAAC;IAC3C,GAAG,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC;IAC3B,UAAU,CAAC,CAAC,IAAI,EAAE,MAAM,GAAG,WAAW,CAAC,MAAM,CAAC,CAAC;IAC/C,SAAS,CAAC,CAAC,IAAI,EAAE,MAAM,GAAG,WAAW,CAAC,MAAM,CAAC,CAAC;CAC/C;AAED,MAAM,WAAW,gBAAgB;IAC/B,YAAY,IAAI,OAAO,GAAG,OAAO,CAAC,OAAO,CAAC,CAAC;IAC3C,MAAM,CAAC,KAAK,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE;QAAE,KAAK,CAAC,EAAE,MAAM,CAAC;QAAC,KAAK,CAAC,EAAE,MAAM,CAAA;KAAE,GAAG,OAAO,CAAC;CAC9E;AAED,MAAM,MAAM,mBAAmB,GAAG,CAAC,KAAK,EAAE,MAAM,KAAK,IAAI,CAAC;AAgC1D;;;;;;;;GAQG;AACH,qBAAa,gBAAgB;IAC3B,MAAM,EAAE,eAAe,GAAG,IAAI,CAAC;IAC/B,KAAK,EAAE,eAAe,GAAG,IAAI,CAAC;IAC9B,SAAS,EAAE,gBAAgB,GAAG,IAAI,CAAC;IACnC,QAAQ,EAAE,CAAC,CAAC,KAAK,EAAE,MAAM,KAAK,IAAI,CAAC,GAAG,IAAI,CAAC;IAC3C,eAAe,EAAE,mBAAmB,GAAG,IAAI,CAAC;IAC5C,SAAS,CAAC,cAAc,qBAA0B,MAAM,KAAK,OAAO,EAAI;IAExE,YAAY,OAAO,GAAE;QACnB,YAAY,CAAC,EAAE,eAAe,GAAG,IAAI,CAAC;QACtC,YAAY,CAAC,EAAE,eAAe,GAAG,IAAI,CAAC;QACtC,gBAAgB,CAAC,EAAE,gBAAgB,GAAG,IAAI,CAAC;QAC3C,QAAQ,CAAC,EAAE,CAAC,CAAC,KAAK,EAAE,MAAM,KAAK,IAAI,CAAC,GAAG,IAAI,CAAC;QAC5C,eAAe,CAAC,EAAE,mBAAmB,GAAG,IAAI,CAAC;KACzC,EAML;IAED,iEAAiE;IACjE,QAAQ,CAAC,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,CAAC,IAAI,EAAE,MAAM,KAAK,OAAO,GAAG,IAAI,CAG/D;IAED;;;;OAIG;IACG,KAAK,CAAC,IAAI,EAAE,MAAM,EAAE,IAAI,SAAS,GAAG,OAAO,CAAC,MAAM,CAAC,CAexD;IAED,0DAA0D;IACnD,YAAY,CAAC,IAAI,EAAE,MAAM,EAAE,IAAI,SAAS,GAAG,cAAc,CAAC,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC,CAejF;IAED,qEAAqE;IAC/D,WAAW,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC,CAI/C;IAED,mDAAmD;IAC7C,MAAM,IAAI,OAAO,CAAC,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC,CAO/C;IAID,UAAgB,iBAAiB,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC,CAKhE;IAED,SAAS,CAAC,kBAAkB,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO,CASlD;IAID,UAAgB,aAAa,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC,CAU3D;IAED,UAAgB,UAAU,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC,CAQxD;IAED,UAAgB,WAAW,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC,CAMzD;IAED,UAAgB,WAAW,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC,CAGzD;IAED,UAAiB,aAAa,CAAC,IAAI,EAAE,MAAM,GAAG,cAAc,CAAC,MAAM,CAAC,CAOnE;IAED,iEAAiE;IACjE,UAAiB,cAAc,CAAC,IAAI,EAAE,MAAM,GAAG,cAAc,CAAC,MAAM,CAAC,CAOpE;IAED,UAAiB,cAAc,CAAC,IAAI,EAAE,MAAM,GAAG,cAAc,CAAC,MAAM,CAAC,CAEpE;IAGD,WAAW,CAAC,IAAI,EAAE,MAAM,EAAE,IAAI,SAAS,GAAG,cAAc,CAAC,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC,CAEzE;IACD,UAAU,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC,CAExC;CACF;AAMD,wBAAsB,UAAU,CAAC,WAAW,CAAC,EAAE,eAAe,GAAG,IAAI,GAAG,OAAO,CAAC,gBAAgB,CAAC,CAehG;AAED,eAAO,MAAM,SAAS,mBAAa,CAAC"}