bonecode 1.1.0 → 1.2.1

package/dist/extensions/llm_provider.js

@@ -0,0 +1,338 @@
+"use strict";
+/**
+ * BoneCode Extension: LLM Provider
+ *
+ * Implements the LLM provider extension points declared in agent.bone.
+ * Bridges BoneCode's agent domain to the Vercel AI SDK, supporting
+ * Anthropic, OpenAI, Google, Ollama, and any @ai-sdk/* provider.
+ *
+ * This file is preserved across BoneScript recompilation.
+ * Sentinel comments mark the preserved regions.
+ */
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
+    Object.defineProperty(o, "default", { enumerable: true, value: v });
+}) : function(o, v) {
+    o["default"] = v;
+});
+var __importStar = (this && this.__importStar) || function (mod) {
+    if (mod && mod.__esModule) return mod;
+    var result = {};
+    if (mod != null) for (var k in mod) if (k !== "default" && Object.prototype.hasOwnProperty.call(mod, k)) __createBinding(result, mod, k);
+    __setModuleDefault(result, mod);
+    return result;
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.cancel_llm_request = exports.send_prompt_to_llm = void 0;
+const anthropic_1 = require("@ai-sdk/anthropic");
+const openai_1 = require("@ai-sdk/openai");
+const google_1 = require("@ai-sdk/google");
+const ai_1 = require("ai");
+const db_1 = require("../bone/output/session/src/db");
+const logger_1 = require("../bone/output/session/src/logger");
+const metrics_1 = require("../bone/output/session/src/metrics");
+const events_1 = require("../bone/output/session/src/events");
+// <bonescript:ext:send_prompt_to_llm:begin>
+async function send_prompt_to_llm(session_id, message_id) {
+    // Load agent config and message history from DB
+    const agentRow = await db_1.pool.query(`SELECT * FROM agent_instances WHERE session_id = $1 AND state = 'running' ORDER BY created_at DESC LIMIT 1`, [session_id]);
+    if (!agentRow.rows[0]) {
+        throw new Error(`No running agent for session ${session_id}`);
+    }
+    const agent = agentRow.rows[0];
+    // Load message history for context
+    const messagesResult = await db_1.pool.query(`SELECT m.*, json_agg(p.* ORDER BY p.order_index) as parts
+     FROM messages m
+     LEFT JOIN parts p ON p.message_id = m.id
+     WHERE m.session_id = $1
+     GROUP BY m.id
+     ORDER BY m.created_at ASC`, [session_id]);
+    // Build the AI SDK message array
+    const aiMessages = buildAIMessages(messagesResult.rows);
+    // Load tool definitions (built-ins + MCP)
+    const tools = await loadToolDefinitions(session_id);
+    // Get the language model
+    const model = getLanguageModel(agent.provider_id, agent.model_id, agent.config);
+    // Inject RAG context into system prompt
+    const lastUserMessage = messagesResult.rows
+        .filter((m) => m.role === "user")
+        .slice(-1)[0];
+    const userText = lastUserMessage?.parts
+        ?.filter((p) => p.part_type === "text")
+        .map((p) => p.data?.text || "")
+        .join(" ") || "";
+    const systemWithRAG = await buildSystemPromptWithRAG(agent.system_prompt, session_id, userText);
+    // Stream the response
+    const result = await (0, ai_1.streamText)({
+        model,
+        messages: aiMessages,
+        tools,
+        system: systemWithRAG || undefined,
+        temperature: agent.temperature ?? 0.7,
+        maxTokens: agent.max_tokens ?? 8192,
+        onChunk: async ({ chunk }) => {
+            // Broadcast each chunk to the WebSocket channel for live streaming
+            await events_1.eventBus.publish("PartUpdated", { session_id, message_id, chunk }, "LLMProvider");
+        },
+    });
+    // Collect the full response
+    const text = await result.text;
+    const toolCalls = await result.toolCalls;
+    const usage = await result.usage;
+    const finishReason = await result.finishReason;
+    // Some reasoning models (Qwen3) put the final answer in content
+    // and thinking in a separate field. The AI SDK normalizes this.
+    const finalText = text || "";
+    (0, metrics_1.counter)("llm.request.completed", {
+        provider: agent.provider_id,
+        model: agent.model_id,
+        finish_reason: finishReason,
+    });
+    return {
+        text: finalText,
+        tool_calls: toolCalls,
+        usage: {
+            input_tokens: usage.promptTokens,
+            output_tokens: usage.completionTokens,
+        },
+        finish_reason: finishReason,
+    };
+}
+exports.send_prompt_to_llm = send_prompt_to_llm;
+// <bonescript:ext:send_prompt_to_llm:end>
+// <bonescript:ext:cancel_llm_request:begin>
+async function cancel_llm_request(session_id) {
+    // In-flight requests are tracked by session_id in a WeakMap
+    const controller = activeRequests.get(session_id);
+    if (controller) {
+        controller.abort();
+        activeRequests.delete(session_id);
+        logger_1.logger.info("llm_request_cancelled", { event: "cancel", metadata: { session_id } });
+    }
+}
+exports.cancel_llm_request = cancel_llm_request;
+// <bonescript:ext:cancel_llm_request:end>
+// ─── Provider Factory ─────────────────────────────────────────────────────────
+function getLanguageModel(providerId, modelId, config) {
+    const apiKey = config.api_key
+        || process.env[`${providerId.toUpperCase()}_API_KEY`]
+        || process.env.OPENAI_API_KEY
+        || "not-needed";
+    const baseUrl = config.base_url
+        || process.env[`${providerId.toUpperCase()}_BASE_URL`]
+        || process.env.OPENAI_BASE_URL;
+    switch (providerId.toLowerCase()) {
+        case "anthropic": {
+            const provider = (0, anthropic_1.createAnthropic)({ apiKey, baseURL: baseUrl });
+            return provider(modelId);
+        }
+        case "openai":
+        case "openai_compatible": {
+            const provider = (0, openai_1.createOpenAI)({ apiKey, baseURL: baseUrl });
+            return provider(modelId);
+        }
+        case "google": {
+            const provider = (0, google_1.createGoogleGenerativeAI)({ apiKey });
+            return provider(modelId);
+        }
+        case "ollama": {
+            const provider = (0, openai_1.createOpenAI)({
+                apiKey: "ollama",
+                baseURL: baseUrl || "http://localhost:11434/v1",
+            });
+            return provider(modelId);
+        }
+        case "openrouter": {
+            const provider = (0, openai_1.createOpenAI)({
+                apiKey,
+                baseURL: "https://openrouter.ai/api/v1",
+            });
+            return provider(modelId);
+        }
+        default: {
+            // Fallback: treat as OpenAI-compatible
+            const provider = (0, openai_1.createOpenAI)({ apiKey, baseURL: baseUrl });
+            return provider(modelId);
+        }
+    }
+}
+// ─── Message Builder ──────────────────────────────────────────────────────────
+function buildAIMessages(dbMessages) {
+    return dbMessages.map((msg) => {
+        const parts = msg.parts || [];
+        if (msg.role === "user") {
+            return {
+                role: "user",
+                content: parts
+                    .filter((p) => p.part_type === "text")
+                    .map((p) => ({ type: "text", text: p.data.text || "" })),
+            };
+        }
+        if (msg.role === "assistant") {
+            const content = [];
+            for (const part of parts) {
+                if (part.part_type === "text") {
+                    content.push({ type: "text", text: part.data.text || "" });
+                }
+                else if (part.part_type === "tool_invocation") {
+                    content.push({
+                        type: "tool-call",
+                        toolCallId: part.data.tool_call_id,
+                        toolName: part.data.tool_name,
+                        args: part.data.args,
+                    });
+                }
+            }
+            return { role: "assistant", content };
+        }
+        if (msg.role === "tool") {
+            return {
+                role: "tool",
+                content: parts
+                    .filter((p) => p.part_type === "tool_result")
+                    .map((p) => ({
+                    type: "tool-result",
+                    toolCallId: p.data.tool_call_id,
+                    toolName: p.data.tool_name,
+                    result: p.data.result,
+                })),
+            };
+        }
+        return { role: msg.role, content: "" };
+    });
+}
+// ─── RAG Context Injection ────────────────────────────────────────────────────
+/**
+ * Retrieves relevant code context from the knowledge base and prepends it
+ * to the system prompt so the LLM has codebase awareness.
+ */
+async function buildSystemPromptWithRAG(basePrompt, session_id, userMessage) {
+    try {
+        // Find the knowledge base for this session's project
+        const kbResult = await db_1.pool.query(`SELECT kb.id, kb.embedding_model FROM knowledge_bases kb
+       JOIN sessions s ON s.project_id = kb.project_id
+       WHERE s.id = $1 AND kb.state = 'ready'
+       LIMIT 1`, [session_id]);
+        if (!kbResult.rows[0]) {
+            return basePrompt || "";
+        }
+        const kb = kbResult.rows[0];
+        // Import lazily to avoid circular deps
+        const { retrieve_context_for_agent } = await Promise.resolve().then(() => __importStar(require("./rag_search")));
+        const context = await retrieve_context_for_agent(kb.id, userMessage, 8);
+        if (!context || context.startsWith("No relevant")) {
+            return basePrompt || "";
+        }
+        const ragSection = `\n\n## Relevant Codebase Context\n\nThe following code snippets from the project are relevant to the current task:\n\n${context}\n\n---\n`;
+        return (basePrompt || "") + ragSection;
+    }
+    catch {
+        // RAG failure is non-fatal — continue without context
+        return basePrompt || "";
+    }
+}
+// ─── Tool Loader ──────────────────────────────────────────────────────────────
+async function loadToolDefinitions(session_id) {
+    const tools = {};
+    // Built-in tools
+    const builtins = ["read_file", "write_file", "edit_file", "run_command", "search_files", "list_directory", "glob"];
+    for (const name of builtins) {
+        tools[name] = {
+            description: getToolDescription(name),
+            parameters: getToolParameters(name),
+        };
+    }
+    // MCP tools
+    try {
+        const { getMCPToolsForAISDK } = await Promise.resolve().then(() => __importStar(require("./mcp_bridge")));
+        const mcpTools = await getMCPToolsForAISDK();
+        Object.assign(tools, mcpTools);
+    }
+    catch {
+        // MCP unavailable — continue with built-ins only
+    }
+    return tools;
+}
+function getToolDescription(name) {
+    const descriptions = {
+        read_file: "Read the contents of a file. Optionally specify start_line and end_line.",
+        write_file: "Write content to a file, creating it if it doesn't exist.",
+        edit_file: "Replace a specific string in a file with new content.",
+        run_command: "Execute a shell command in the workspace directory.",
+        search_files: "Search for a pattern across files using ripgrep.",
+        list_directory: "List the contents of a directory.",
+        glob: "Find files matching a glob pattern.",
+    };
+    return descriptions[name] || name;
+}
+function getToolParameters(name) {
+    const schemas = {
+        read_file: {
+            type: "object",
+            properties: {
+                path: { type: "string", description: "File path relative to workspace root" },
+                start_line: { type: "number", description: "First line to read (1-indexed)" },
+                end_line: { type: "number", description: "Last line to read (inclusive)" },
+            },
+            required: ["path"],
+        },
+        write_file: {
+            type: "object",
+            properties: {
+                path: { type: "string" },
+                content: { type: "string" },
+            },
+            required: ["path", "content"],
+        },
+        edit_file: {
+            type: "object",
+            properties: {
+                path: { type: "string" },
+                old_str: { type: "string", description: "Exact string to replace" },
+                new_str: { type: "string", description: "Replacement string" },
+            },
+            required: ["path", "old_str", "new_str"],
+        },
+        run_command: {
+            type: "object",
+            properties: {
+                command: { type: "string" },
+                cwd: { type: "string", description: "Working directory (optional)" },
+                timeout_ms: { type: "number", description: "Timeout in milliseconds (default 30000)" },
+            },
+            required: ["command"],
+        },
+        search_files: {
+            type: "object",
+            properties: {
+                pattern: { type: "string", description: "Regex pattern to search for" },
+                glob: { type: "string", description: "File glob pattern (e.g. **/*.ts)" },
+            },
+            required: ["pattern"],
+        },
+        list_directory: {
+            type: "object",
+            properties: { path: { type: "string" } },
+            required: ["path"],
+        },
+        glob: {
+            type: "object",
+            properties: { pattern: { type: "string" } },
+            required: ["pattern"],
+        },
+    };
+    return schemas[name] || { type: "object", properties: {} };
+}
+// ─── Active Request Tracking ──────────────────────────────────────────────────
+const activeRequests = new Map();
+//# sourceMappingURL=llm_provider.js.map

package/dist/extensions/llm_provider.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"llm_provider.js","sourceRoot":"","sources":["../../extensions/llm_provider.ts"],"names":[],"mappings":";AAAA;;;;;;;;;GASG;;;;;;;;;;;;;;;;;;;;;;;;;;AAEH,iDAAoD;AACpD,2CAA8C;AAC9C,2CAA0D;AAC1D,2BAA+C;AAC/C,sDAAqD;AACrD,8DAA2D;AAC3D,gEAA6D;AAC7D,8DAA6D;AAwB7D,4CAA4C;AACrC,KAAK,UAAU,kBAAkB,CACtC,UAAkB,EAClB,UAAkB;IAElB,gDAAgD;IAChD,MAAM,QAAQ,GAAG,MAAM,SAAI,CAAC,KAAK,CAC/B,4GAA4G,EAC5G,CAAC,UAAU,CAAC,CACb,CAAC;IACF,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC;QACtB,MAAM,IAAI,KAAK,CAAC,gCAAgC,UAAU,EAAE,CAAC,CAAC;IAChE,CAAC;IACD,MAAM,KAAK,GAAG,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAE/B,mCAAmC;IACnC,MAAM,cAAc,GAAG,MAAM,SAAI,CAAC,KAAK,CACrC;;;;;+BAK2B,EAC3B,CAAC,UAAU,CAAC,CACb,CAAC;IAEF,iCAAiC;IACjC,MAAM,UAAU,GAAG,eAAe,CAAC,cAAc,CAAC,IAAI,CAAC,CAAC;IAExD,0CAA0C;IAC1C,MAAM,KAAK,GAAG,MAAM,mBAAmB,CAAC,UAAU,CAAC,CAAC;IAEpD,yBAAyB;IACzB,MAAM,KAAK,GAAG,gBAAgB,CAAC,KAAK,CAAC,WAAW,EAAE,KAAK,CAAC,QAAQ,EAAE,KAAK,CAAC,MAAM,CAAC,CAAC;IAEhF,wCAAwC;IACxC,MAAM,eAAe,GAAG,cAAc,CAAC,IAAI;SACxC,MAAM,CAAC,CAAC,CAAM,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,MAAM,CAAC;SACrC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;IAChB,MAAM,QAAQ,GAAG,eAAe,EAAE,KAAK;QACrC,EAAE,MAAM,CAAC,CAAC,CAAM,EAAE,EAAE,CAAC,CAAC,CAAC,SAAS,KAAK,MAAM,CAAC;SAC3C,GAAG,CAAC,CAAC,CAAM,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,EAAE,IAAI,IAAI,EAAE,CAAC;SACnC,IAAI,CAAC,GAAG,CAAC,IAAI,EAAE,CAAC;IACnB,MAAM,aAAa,GAAG,MAAM,wBAAwB,CAAC,KAAK,CAAC,aAAa,EAAE,UAAU,EAAE,QAAQ,CAAC,CAAC;IAEhG,sBAAsB;IACtB,MAAM,MAAM,GAAG,MAAM,IAAA,eAAU,EAAC;QAC9B,KAAK;QACL,QAAQ,EAAE,UAAU;QACpB,KAAK;QACL,MAAM,EAAE,aAAa,IAAI,SAAS;QAClC,WAAW,EAAE,KAAK,CAAC,WAAW,IAAI,GAAG;QACrC,SAAS,EAAE,KAAK,CAAC,UAAU,IAAI,IAAI;QACnC,OAAO,EAAE,KAAK,EAAE,EAAE,KAAK,EAAE,EAAE,EAAE;YAC3B,mEAAmE;YACnE,MAAM,iBAAQ,CAAC,OAAO,CACpB,aAAa,EACb,EAAE,UAAU,EAAE,UAAU,EAAE,KAAK,EAAE,EACjC,aAAa,CACd,CAAC;QACJ,CAAC;KACF,CAAC,CAAC;IAEH,4BAA4B;IAC5B,MAAM,IAAI,GAAG,MAAM,MAAM,CAAC,IAAI,CAAC;IAC/B,MAAM,SAAS,GAAG,MAAM,MAAM,CAAC,SAAS,CAAC;IACzC,MAAM,KAAK,GAAG,MAAM,MAAM,CAAC,KAAK,CAAC;IACjC,MAAM,YAAY,GAAG,MAAM,MAAM,CAAC,YAAY,CAAC;IAE/C,gEAAgE;IAChE,gEAAgE;IAChE,MAAM,SAAS,GAAG,IAAI,IAAI,EAAE,CAAC;IAE7B,IAAA,iBAAO,EAAC,uBAAuB,EAAE;QAC/B,QAAQ,EAAE,KAAK,CAAC,WAAW;QAC3B,KAAK,EAAE,KAAK,CAAC,QAAQ;QACrB,aAAa,EAAE,YAAY;KAC5B,CAAC,CAAC;IAEH,OAAO;QACL,IAAI,EAAE,SAAS;QACf,UAAU,EAAE,SAAS;QACrB,KAAK,EAAE;YACL,YAAY,EAAE,KAAK,CAAC,YAAY;YAChC,aAAa,EAAE,KAAK,CAAC,gBAAgB;SACtC;QACD,aAAa,EAAE,YAAY;KAC5B,CAAC;AACJ,CAAC;AAvFD,gDAuFC;AACD,0CAA0C;AAE1C,4CAA4C;AACrC,KAAK,UAAU,kBAAkB,CAAC,UAAkB;IACzD,4DAA4D;IAC5D,MAAM,UAAU,GAAG,cAAc,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;IAClD,IAAI,UAAU,EAAE,CAAC;QACf,UAAU,CAAC,KAAK,EAAE,CAAC;QACnB,cAAc,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC;QAClC,eAAM,CAAC,IAAI,CAAC,uBAAuB,EAAE,EAAE,KAAK,EAAE,QAAQ,EAAE,QAAQ,EAAE,EAAE,UAAU,EAAE,EAAE,CAAC,CAAC;IACtF,CAAC;AACH,CAAC;AARD,gDAQC;AACD,0CAA0C;AAE1C,iFAAiF;AAEjF,SAAS,gBAAgB,CACvB,UAAkB,EAClB,OAAe,EACf,MAA+B;IAE/B,MAAM,MAAM,GAAI,MAAM,CAAC,OAAkB;WACpC,OAAO,CAAC,GAAG,CAAC,GAAG,UAAU,CAAC,WAAW,EAAE,UAAU,CAAC;WAClD,OAAO,CAAC,GAAG,CAAC,cAAc;WAC1B,YAAY,CAAC;IAClB,MAAM,OAAO,GAAI,MAAM,CAAC,QAAmB;WACtC,OAAO,CAAC,GAAG,CAAC,GAAG,UAAU,CAAC,WAAW,EAAE,WAAW,CAAC;WACnD,OAAO,CAAC,GAAG,CAAC,eAAe,CAAC;IAEjC,QAAQ,UAAU,CAAC,WAAW,EAAE,EAAE,CAAC;QACjC,KAAK,WAAW,CAAC,CAAC,CAAC;YACjB,MAAM,QAAQ,GAAG,IAAA,2BAAe,EAAC,EAAE,MAAM,EAAE,OAAO,EAAE,OAAO,EAAE,CAAC,CAAC;YAC/D,OAAO,QAAQ,CAAC,OAAO,CAAC,CAAC;QAC3B,CAAC;QACD,KAAK,QAAQ,CAAC;QACd,KAAK,mBAAmB,CAAC,CAAC,CAAC;YACzB,MAAM,QAAQ,GAAG,IAAA,qBAAY,EAAC,EAAE,MAAM,EAAE,OAAO,EAAE,OAAO,EAAE,CAAC,CAAC;YAC5D,OAAO,QAAQ,CAAC,OAAO,CAAC,CAAC;QAC3B,CAAC;QACD,KAAK,QAAQ,CAAC,CAAC,CAAC;YACd,MAAM,QAAQ,GAAG,IAAA,iCAAwB,EAAC,EAAE,MAAM,EAAE,CAAC,CAAC;YACtD,OAAO,QAAQ,CAAC,OAAO,CAAC,CAAC;QAC3B,CAAC;QACD,KAAK,QAAQ,CAAC,CAAC,CAAC;YACd,MAAM,QAAQ,GAAG,IAAA,qBAAY,EAAC;gBAC5B,MAAM,EAAE,QAAQ;gBAChB,OAAO,EAAE,OAAO,IAAI,2BAA2B;aAChD,CAAC,CAAC;YACH,OAAO,QAAQ,CAAC,OAAO,CAAC,CAAC;QAC3B,CAAC;QACD,KAAK,YAAY,CAAC,CAAC,CAAC;YAClB,MAAM,QAAQ,GAAG,IAAA,qBAAY,EAAC;gBAC5B,MAAM;gBACN,OAAO,EAAE,8BAA8B;aACxC,CAAC,CAAC;YACH,OAAO,QAAQ,CAAC,OAAO,CAAC,CAAC;QAC3B,CAAC;QACD,OAAO,CAAC,CAAC,CAAC;YACR,uCAAuC;YACvC,MAAM,QAAQ,GAAG,IAAA,qBAAY,EAAC,EAAE,MAAM,EAAE,OAAO,EAAE,OAAO,EAAE,CAAC,CAAC;YAC5D,OAAO,QAAQ,CAAC,OAAO,CAAC,CAAC;QAC3B,CAAC;IACH,CAAC;AACH,CAAC;AAED,iFAAiF;AAEjF,SAAS,eAAe,CAAC,UAAiB;IACxC,OAAO,UAAU,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE;QAC5B,MAAM,KAAK,GAAG,GAAG,CAAC,KAAK,IAAI,EAAE,CAAC;QAC9B,IAAI,GAAG,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;YACxB,OAAO;gBACL,IAAI,EAAE,MAAe;gBACrB,OAAO,EAAE,KAAK;qBACX,MAAM,CAAC,CAAC,CAAM,EAAE,EAAE,CAAC,CAAC,CAAC,SAAS,KAAK,MAAM,CAAC;qBAC1C,GAAG,CAAC,CAAC,CAAM,EAAE,EAAE,CAAC,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,IAAI,EAAE,EAAE,CAAC,CAAC;aAChE,CAAC;QACJ,CAAC;QACD,IAAI,GAAG,CAAC,IAAI,KAAK,WAAW,EAAE,CAAC;YAC7B,MAAM,OAAO,GAAU,EAAE,CAAC;YAC1B,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;gBACzB,IAAI,IAAI,CAAC,SAAS,KAAK,MAAM,EAAE,CAAC;oBAC9B,OAAO,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,CAAC,IAAI,IAAI,EAAE,EAAE,CAAC,CAAC;gBAC7D,CAAC;qBAAM,IAAI,IAAI,CAAC,SAAS,KAAK,iBAAiB,EAAE,CAAC;oBAChD,OAAO,CAAC,IAAI,CAAC;wBACX,IAAI,EAAE,WAAW;wBACjB,UAAU,EAAE,IAAI,CAAC,IAAI,CAAC,YAAY;wBAClC,QAAQ,EAAE,IAAI,CAAC,IAAI,CAAC,SAAS;wBAC7B,IAAI,EAAE,IAAI,CAAC,IAAI,CAAC,IAAI;qBACrB,CAAC,CAAC;gBACL,CAAC;YACH,CAAC;YACD,OAAO,EAAE,IAAI,EAAE,WAAoB,EAAE,OAAO,EAAE,CAAC;QACjD,CAAC;QACD,IAAI,GAAG,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;YACxB,OAAO;gBACL,IAAI,EAAE,MAAe;gBACrB,OAAO,EAAE,KAAK;qBACX,MAAM,CAAC,CAAC,CAAM,EAAE,EAAE,CAAC,CAAC,CAAC,SAAS,KAAK,aAAa,CAAC;qBACjD,GAAG,CAAC,CAAC,CAAM,EAAE,EAAE,CAAC,CAAC;oBAChB,IAAI,EAAE,aAAa;oBACnB,UAAU,EAAE,CAAC,CAAC,IAAI,CAAC,YAAY;oBAC/B,QAAQ,EAAE,CAAC,CAAC,IAAI,CAAC,SAAS;oBAC1B,MAAM,EAAE,CAAC,CAAC,IAAI,CAAC,MAAM;iBACtB,CAAC,CAAC;aACN,CAAC;QACJ,CAAC;QACD,OAAO,EAAE,IAAI,EAAE,GAAG,CAAC,IAAI,EAAE,OAAO,EAAE,EAAE,EAAE,CAAC;IACzC,CAAC,CAAC,CAAC;AACL,CAAC;AAED,iFAAiF;AAEjF;;;GAGG;AACH,KAAK,UAAU,wBAAwB,CACrC,UAA8B,EAC9B,UAAkB,EAClB,WAAmB;IAEnB,IAAI,CAAC;QACH,qDAAqD;QACrD,MAAM,QAAQ,GAAG,MAAM,SAAI,CAAC,KAAK,CAC/B;;;eAGS,EACT,CAAC,UAAU,CAAC,CACb,CAAC;QAEF,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC;YACtB,OAAO,UAAU,IAAI,EAAE,CAAC;QAC1B,CAAC;QAED,MAAM,EAAE,GAAG,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAE5B,uCAAuC;QACvC,MAAM,EAAE,0BAA0B,EAAE,GAAG,wDAAa,cAAc,GAAC,CAAC;QACpE,MAAM,OAAO,GAAG,MAAM,0BAA0B,CAAC,EAAE,CAAC,EAAE,EAAE,WAAW,EAAE,CAAC,CAAC,CAAC;QAExE,IAAI,CAAC,OAAO,IAAI,OAAO,CAAC,UAAU,CAAC,aAAa,CAAC,EAAE,CAAC;YAClD,OAAO,UAAU,IAAI,EAAE,CAAC;QAC1B,CAAC;QAED,MAAM,UAAU,GAAG,yHAAyH,OAAO,WAAW,CAAC;QAC/J,OAAO,CAAC,UAAU,IAAI,EAAE,CAAC,GAAG,UAAU,CAAC;IACzC,CAAC;IAAC,MAAM,CAAC;QACP,sDAAsD;QACtD,OAAO,UAAU,IAAI,EAAE,CAAC;IAC1B,CAAC;AACH,CAAC;AAED,iFAAiF;AAEjF,KAAK,UAAU,mBAAmB,CAAC,UAAkB;IACnD,MAAM,KAAK,GAAwB,EAAE,CAAC;IAEtC,iBAAiB;IACjB,MAAM,QAAQ,GAAG,CAAC,WAAW,EAAE,YAAY,EAAE,WAAW,EAAE,aAAa,EAAE,cAAc,EAAE,gBAAgB,EAAE,MAAM,CAAC,CAAC;IACnH,KAAK,MAAM,IAAI,IAAI,QAAQ,EAAE,CAAC;QAC5B,KAAK,CAAC,IAAI,CAAC,GAAG;YACZ,WAAW,EAAE,kBAAkB,CAAC,IAAI,CAAC;YACrC,UAAU,EAAE,iBAAiB,CAAC,IAAI,CAAC;SACpC,CAAC;IACJ,CAAC;IAED,YAAY;IACZ,IAAI,CAAC;QACH,MAAM,EAAE,mBAAmB,EAAE,GAAG,wDAAa,cAAc,GAAC,CAAC;QAC7D,MAAM,QAAQ,GAAG,MAAM,mBAAmB,EAAE,CAAC;QAC7C,MAAM,CAAC,MAAM,CAAC,KAAK,EAAE,QAAQ,CAAC,CAAC;IACjC,CAAC;IAAC,MAAM,CAAC;QACP,iDAAiD;IACnD,CAAC;IAED,OAAO,KAAK,CAAC;AACf,CAAC;AAED,SAAS,kBAAkB,CAAC,IAAY;IACtC,MAAM,YAAY,GAA2B;QAC3C,SAAS,EAAE,0EAA0E;QACrF,UAAU,EAAE,2DAA2D;QACvE,SAAS,EAAE,uDAAuD;QAClE,WAAW,EAAE,qDAAqD;QAClE,YAAY,EAAE,kDAAkD;QAChE,cAAc,EAAE,mCAAmC;QACnD,IAAI,EAAE,qCAAqC;KAC5C,CAAC;IACF,OAAO,YAAY,CAAC,IAAI,CAAC,IAAI,IAAI,CAAC;AACpC,CAAC;AAED,SAAS,iBAAiB,CAAC,IAAY;IACrC,MAAM,OAAO,GAA4C;QACvD,SAAS,EAAE;YACT,IAAI,EAAE,QAAQ;YACd,UAAU,EAAE;gBACV,IAAI,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,WAAW,EAAE,sCAAsC,EAAE;gBAC7E,UAAU,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,WAAW,EAAE,gCAAgC,EAAE;gBAC7E,QAAQ,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,WAAW,EAAE,+BAA+B,EAAE;aAC3E;YACD,QAAQ,EAAE,CAAC,MAAM,CAAC;SACnB;QACD,UAAU,EAAE;YACV,IAAI,EAAE,QAAQ;YACd,UAAU,EAAE;gBACV,IAAI,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE;gBACxB,OAAO,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE;aAC5B;YACD,QAAQ,EAAE,CAAC,MAAM,EAAE,SAAS,CAAC;SAC9B;QACD,SAAS,EAAE;YACT,IAAI,EAAE,QAAQ;YACd,UAAU,EAAE;gBACV,IAAI,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE;gBACxB,OAAO,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,WAAW,EAAE,yBAAyB,EAAE;gBACnE,OAAO,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,WAAW,EAAE,oBAAoB,EAAE;aAC/D;YACD,QAAQ,EAAE,CAAC,MAAM,EAAE,SAAS,EAAE,SAAS,CAAC;SACzC;QACD,WAAW,EAAE;YACX,IAAI,EAAE,QAAQ;YACd,UAAU,EAAE;gBACV,OAAO,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE;gBAC3B,GAAG,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,WAAW,EAAE,8BAA8B,EAAE;gBACpE,UAAU,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,WAAW,EAAE,yCAAyC,EAAE;aACvF;YACD,QAAQ,EAAE,CAAC,SAAS,CAAC;SACtB;QACD,YAAY,EAAE;YACZ,IAAI,EAAE,QAAQ;YACd,UAAU,EAAE;gBACV,OAAO,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,WAAW,EAAE,6BAA6B,EAAE;gBACvE,IAAI,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,WAAW,EAAE,kCAAkC,EAAE;aAC1E;YACD,QAAQ,EAAE,CAAC,SAAS,CAAC;SACtB;QACD,cAAc,EAAE;YACd,IAAI,EAAE,QAAQ;YACd,UAAU,EAAE,EAAE,IAAI,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,EAAE;YACxC,QAAQ,EAAE,CAAC,MAAM,CAAC;SACnB;QACD,IAAI,EAAE;YACJ,IAAI,EAAE,QAAQ;YACd,UAAU,EAAE,EAAE,OAAO,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,EAAE;YAC3C,QAAQ,EAAE,CAAC,SAAS,CAAC;SACtB;KACF,CAAC;IACF,OAAO,OAAO,CAAC,IAAI,CAAC,IAAI,EAAE,IAAI,EAAE,QAAQ,EAAE,UAAU,EAAE,EAAE,EAAE,CAAC;AAC7D,CAAC;AAED,iFAAiF;AAEjF,MAAM,cAAc,GAAG,IAAI,GAAG,EAA2B,CAAC"}

package/dist/extensions/mcp_bridge.d.ts

@@ -0,0 +1,44 @@
+/**
+ * BoneCode Extension: MCP Bridge
+ *
+ * Manages Model Context Protocol server connections and bridges them
+ * into the BoneCode tool execution system.
+ *
+ * Supports:
+ *   - stdio transport (spawns a child process)
+ *   - HTTP/SSE transport (connects to a running server)
+ *
+ * MCP servers are configured in .bonecode/config.json under "mcpServers".
+ */
+export interface MCPTool {
+    name: string;
+    description: string;
+    inputSchema: Record<string, unknown>;
+    serverName: string;
+}
+export interface MCPCallResult {
+    content: Array<{
+        type: string;
+        text?: string;
+        data?: string;
+        mimeType?: string;
+    }>;
+    isError?: boolean;
+}
+declare class MCPClientManager {
+    private clients;
+    private tools;
+    private initialized;
+    initialize(): Promise<void>;
+    private connectServer;
+    getAllTools(): MCPTool[];
+    callTool(toolName: string, input: Record<string, unknown>): Promise<MCPCallResult>;
+    disconnect(): Promise<void>;
+}
+export declare const mcpManager: MCPClientManager;
+/**
+ * Returns MCP tools formatted as AI SDK tool definitions.
+ * Call this when building the tools object for streamText().
+ */
+export declare function getMCPToolsForAISDK(): Promise<Record<string, any>>;
+export {};

package/dist/extensions/mcp_bridge.js

@@ -0,0 +1,151 @@
+"use strict";
+/**
+ * BoneCode Extension: MCP Bridge
+ *
+ * Manages Model Context Protocol server connections and bridges them
+ * into the BoneCode tool execution system.
+ *
+ * Supports:
+ *   - stdio transport (spawns a child process)
+ *   - HTTP/SSE transport (connects to a running server)
+ *
+ * MCP servers are configured in .bonecode/config.json under "mcpServers".
+ */
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.getMCPToolsForAISDK = exports.mcpManager = void 0;
+const index_js_1 = require("@modelcontextprotocol/sdk/client/index.js");
+const stdio_js_1 = require("@modelcontextprotocol/sdk/client/stdio.js");
+const sse_js_1 = require("@modelcontextprotocol/sdk/client/sse.js");
+const logger_1 = require("../bone/output/session/src/logger");
+const metrics_1 = require("../bone/output/session/src/metrics");
+const config_1 = require("../src/config");
+// ─── MCP Client Manager ───────────────────────────────────────────────────────
+class MCPClientManager {
+    constructor() {
+        this.clients = new Map();
+        this.tools = new Map(); // serverName → tools
+        this.initialized = false;
+    }
+    async initialize() {
+        if (this.initialized)
+            return;
+        const { mcpServers } = (0, config_1.getConfig)();
+        for (const serverConfig of mcpServers) {
+            try {
+                await this.connectServer(serverConfig);
+            }
+            catch (e) {
+                logger_1.logger.error("mcp_connect_failed", {
+                    event: "mcp_init",
+                    metadata: { server: serverConfig.name, error: e.message },
+                });
+            }
+        }
+        this.initialized = true;
+    }
+    async connectServer(config) {
+        const client = new index_js_1.Client({ name: "bonecode", version: "1.0.0" }, { capabilities: { tools: {} } });
+        let transport;
+        if (config.transport === "stdio") {
+            transport = new stdio_js_1.StdioClientTransport({
+                command: config.command,
+                args: config.args || [],
+                env: { ...process.env, ...(config.env || {}) },
+            });
+        }
+        else {
+            transport = new sse_js_1.SSEClientTransport(new URL(config.url));
+        }
+        await client.connect(transport);
+        // Discover tools
+        const toolsResult = await client.listTools();
+        const mcpTools = toolsResult.tools.map(t => ({
+            name: `${config.name}__${t.name}`,
+            description: t.description || "",
+            inputSchema: t.inputSchema,
+            serverName: config.name,
+        }));
+        this.clients.set(config.name, client);
+        this.tools.set(config.name, mcpTools);
+        logger_1.logger.info("mcp_connected", {
+            event: "mcp_connect",
+            metadata: { server: config.name, tools: mcpTools.length },
+        });
+    }
+    getAllTools() {
+        const all = [];
+        for (const tools of this.tools.values()) {
+            all.push(...tools);
+        }
+        return all;
+    }
+    async callTool(toolName, input) {
+        // Tool names are prefixed: "serverName__toolName"
+        const separatorIdx = toolName.indexOf("__");
+        if (separatorIdx === -1) {
+            throw new Error(`Invalid MCP tool name format: ${toolName}`);
+        }
+        const serverName = toolName.slice(0, separatorIdx);
+        const actualToolName = toolName.slice(separatorIdx + 2);
+        const client = this.clients.get(serverName);
+        if (!client) {
+            throw new Error(`MCP server not connected: ${serverName}`);
+        }
+        const startMs = Date.now();
+        try {
+            const result = await client.callTool({ name: actualToolName, arguments: input });
+            (0, metrics_1.counter)("mcp.tool.called", { server: serverName, tool: actualToolName, success: "true" });
+            logger_1.logger.info("mcp_tool_called", {
+                event: "mcp_call",
+                metadata: { server: serverName, tool: actualToolName, duration_ms: Date.now() - startMs },
+            });
+            return result;
+        }
+        catch (e) {
+            (0, metrics_1.counter)("mcp.tool.called", { server: serverName, tool: actualToolName, success: "false" });
+            throw e;
+        }
+    }
+    async disconnect() {
+        for (const [name, client] of this.clients) {
+            try {
+                await client.close();
+            }
+            catch (e) {
+                logger_1.logger.error("mcp_disconnect_error", {
+                    event: "mcp_disconnect",
+                    metadata: { server: name, error: e.message },
+                });
+            }
+        }
+        this.clients.clear();
+        this.tools.clear();
+        this.initialized = false;
+    }
+}
+exports.mcpManager = new MCPClientManager();
+// ─── Tool Schema Builder (for AI SDK) ─────────────────────────────────────────
+/**
+ * Returns MCP tools formatted as AI SDK tool definitions.
+ * Call this when building the tools object for streamText().
+ */
+async function getMCPToolsForAISDK() {
+    await exports.mcpManager.initialize();
+    const tools = {};
+    for (const mcpTool of exports.mcpManager.getAllTools()) {
+        tools[mcpTool.name] = {
+            description: mcpTool.description,
+            parameters: mcpTool.inputSchema,
+            execute: async (input) => {
+                const result = await exports.mcpManager.callTool(mcpTool.name, input);
+                // Convert MCP content array to a string for the AI SDK
+                return result.content
+                    .map(c => c.text || (c.data ? `[binary: ${c.mimeType}]` : ""))
+                    .join("\n");
+            },
+        };
+    }
+    return tools;
+}
+exports.getMCPToolsForAISDK = getMCPToolsForAISDK;
+//# sourceMappingURL=mcp_bridge.js.map

package/dist/extensions/mcp_bridge.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"mcp_bridge.js","sourceRoot":"","sources":["../../extensions/mcp_bridge.ts"],"names":[],"mappings":";AAAA;;;;;;;;;;;GAWG;;;AAEH,wEAAmE;AACnE,wEAAiF;AACjF,oEAA6E;AAC7E,8DAA2D;AAC3D,gEAA6D;AAC7D,0CAA2D;AAc3D,iFAAiF;AAEjF,MAAM,gBAAgB;IAAtB;QACU,YAAO,GAAG,IAAI,GAAG,EAAkB,CAAC;QACpC,UAAK,GAAG,IAAI,GAAG,EAAqB,CAAC,CAAC,qBAAqB;QAC3D,gBAAW,GAAG,KAAK,CAAC;IA2G9B,CAAC;IAzGC,KAAK,CAAC,UAAU;QACd,IAAI,IAAI,CAAC,WAAW;YAAE,OAAO;QAC7B,MAAM,EAAE,UAAU,EAAE,GAAG,IAAA,kBAAS,GAAE,CAAC;QACnC,KAAK,MAAM,YAAY,IAAI,UAAU,EAAE,CAAC;YACtC,IAAI,CAAC;gBACH,MAAM,IAAI,CAAC,aAAa,CAAC,YAAY,CAAC,CAAC;YACzC,CAAC;YAAC,OAAO,CAAM,EAAE,CAAC;gBAChB,eAAM,CAAC,KAAK,CAAC,oBAAoB,EAAE;oBACjC,KAAK,EAAE,UAAU;oBACjB,QAAQ,EAAE,EAAE,MAAM,EAAE,YAAY,CAAC,IAAI,EAAE,KAAK,EAAE,CAAC,CAAC,OAAO,EAAE;iBAC1D,CAAC,CAAC;YACL,CAAC;QACH,CAAC;QACD,IAAI,CAAC,WAAW,GAAG,IAAI,CAAC;IAC1B,CAAC;IAEO,KAAK,CAAC,aAAa,CAAC,MAAuB;QACjD,MAAM,MAAM,GAAG,IAAI,iBAAM,CACvB,EAAE,IAAI,EAAE,UAAU,EAAE,OAAO,EAAE,OAAO,EAAE,EACtC,EAAE,YAAY,EAAE,EAAE,KAAK,EAAE,EAAE,EAAE,EAAE,CAChC,CAAC;QAEF,IAAI,SAAS,CAAC;QACd,IAAI,MAAM,CAAC,SAAS,KAAK,OAAO,EAAE,CAAC;YACjC,SAAS,GAAG,IAAI,+BAAoB,CAAC;gBACnC,OAAO,EAAE,MAAM,CAAC,OAAO;gBACvB,IAAI,EAAE,MAAM,CAAC,IAAI,IAAI,EAAE;gBACvB,GAAG,EAAE,EAAE,GAAG,OAAO,CAAC,GAAG,EAAE,GAAG,CAAC,MAAM,CAAC,GAAG,IAAI,EAAE,CAAC,EAA4B;aACzE,CAAC,CAAC;QACL,CAAC;aAAM,CAAC;YACN,SAAS,GAAG,IAAI,2BAAkB,CAAC,IAAI,GAAG,CAAC,MAAM,CAAC,GAAI,CAAC,CAAC,CAAC;QAC3D,CAAC;QAED,MAAM,MAAM,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;QAEhC,iBAAiB;QACjB,MAAM,WAAW,GAAG,MAAM,MAAM,CAAC,SAAS,EAAE,CAAC;QAC7C,MAAM,QAAQ,GAAc,WAAW,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;YACtD,IAAI,EAAE,GAAG,MAAM,CAAC,IAAI,KAAK,CAAC,CAAC,IAAI,EAAE;YACjC,WAAW,EAAE,CAAC,CAAC,WAAW,IAAI,EAAE;YAChC,WAAW,EAAE,CAAC,CAAC,WAAsC;YACrD,UAAU,EAAE,MAAM,CAAC,IAAI;SACxB,CAAC,CAAC,CAAC;QAEJ,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;QACtC,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,MAAM,CAAC,IAAI,EAAE,QAAQ,CAAC,CAAC;QAEtC,eAAM,CAAC,IAAI,CAAC,eAAe,EAAE;YAC3B,KAAK,EAAE,aAAa;YACpB,QAAQ,EAAE,EAAE,MAAM,EAAE,MAAM,CAAC,IAAI,EAAE,KAAK,EAAE,QAAQ,CAAC,MAAM,EAAE;SAC1D,CAAC,CAAC;IACL,CAAC;IAED,WAAW;QACT,MAAM,GAAG,GAAc,EAAE,CAAC;QAC1B,KAAK,MAAM,KAAK,IAAI,IAAI,CAAC,KAAK,CAAC,MAAM,EAAE,EAAE,CAAC;YACxC,GAAG,CAAC,IAAI,CAAC,GAAG,KAAK,CAAC,CAAC;QACrB,CAAC;QACD,OAAO,GAAG,CAAC;IACb,CAAC;IAED,KAAK,CAAC,QAAQ,CAAC,QAAgB,EAAE,KAA8B;QAC7D,kDAAkD;QAClD,MAAM,YAAY,GAAG,QAAQ,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;QAC5C,IAAI,YAAY,KAAK,CAAC,CAAC,EAAE,CAAC;YACxB,MAAM,IAAI,KAAK,CAAC,iCAAiC,QAAQ,EAAE,CAAC,CAAC;QAC/D,CAAC;QACD,MAAM,UAAU,GAAG,QAAQ,CAAC,KAAK,CAAC,CAAC,EAAE,YAAY,CAAC,CAAC;QACnD,MAAM,cAAc,GAAG,QAAQ,CAAC,KAAK,CAAC,YAAY,GAAG,CAAC,CAAC,CAAC;QAExD,MAAM,MAAM,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;QAC5C,IAAI,CAAC,MAAM,EAAE,CAAC;YACZ,MAAM,IAAI,KAAK,CAAC,6BAA6B,UAAU,EAAE,CAAC,CAAC;QAC7D,CAAC;QAED,MAAM,OAAO,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAC3B,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,QAAQ,CAAC,EAAE,IAAI,EAAE,cAAc,EAAE,SAAS,EAAE,KAAK,EAAE,CAAC,CAAC;YACjF,IAAA,iBAAO,EAAC,iBAAiB,EAAE,EAAE,MAAM,EAAE,UAAU,EAAE,IAAI,EAAE,cAAc,EAAE,OAAO,EAAE,MAAM,EAAE,CAAC,CAAC;YAC1F,eAAM,CAAC,IAAI,CAAC,iBAAiB,EAAE;gBAC7B,KAAK,EAAE,UAAU;gBACjB,QAAQ,EAAE,EAAE,MAAM,EAAE,UAAU,EAAE,IAAI,EAAE,cAAc,EAAE,WAAW,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,OAAO,EAAE;aAC1F,CAAC,CAAC;YACH,OAAO,MAAuB,CAAC;QACjC,CAAC;QAAC,OAAO,CAAM,EAAE,CAAC;YAChB,IAAA,iBAAO,EAAC,iBAAiB,EAAE,EAAE,MAAM,EAAE,UAAU,EAAE,IAAI,EAAE,cAAc,EAAE,OAAO,EAAE,OAAO,EAAE,CAAC,CAAC;YAC3F,MAAM,CAAC,CAAC;QACV,CAAC;IACH,CAAC;IAED,KAAK,CAAC,UAAU;QACd,KAAK,MAAM,CAAC,IAAI,EAAE,MAAM,CAAC,IAAI,IAAI,CAAC,OAAO,EAAE,CAAC;YAC1C,IAAI,CAAC;gBACH,MAAM,MAAM,CAAC,KAAK,EAAE,CAAC;YACvB,CAAC;YAAC,OAAO,CAAM,EAAE,CAAC;gBAChB,eAAM,CAAC,KAAK,CAAC,sBAAsB,EAAE;oBACnC,KAAK,EAAE,gBAAgB;oBACvB,QAAQ,EAAE,EAAE,MAAM,EAAE,IAAI,EAAE,KAAK,EAAE,CAAC,CAAC,OAAO,EAAE;iBAC7C,CAAC,CAAC;YACL,CAAC;QACH,CAAC;QACD,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC;QACrB,IAAI,CAAC,KAAK,CAAC,KAAK,EAAE,CAAC;QACnB,IAAI,CAAC,WAAW,GAAG,KAAK,CAAC;IAC3B,CAAC;CACF;AAEY,QAAA,UAAU,GAAG,IAAI,gBAAgB,EAAE,CAAC;AAEjD,iFAAiF;AAEjF;;;GAGG;AACI,KAAK,UAAU,mBAAmB;IACvC,MAAM,kBAAU,CAAC,UAAU,EAAE,CAAC;IAC9B,MAAM,KAAK,GAAwB,EAAE,CAAC;IAEtC,KAAK,MAAM,OAAO,IAAI,kBAAU,CAAC,WAAW,EAAE,EAAE,CAAC;QAC/C,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,GAAG;YACpB,WAAW,EAAE,OAAO,CAAC,WAAW;YAChC,UAAU,EAAE,OAAO,CAAC,WAAW;YAC/B,OAAO,EAAE,KAAK,EAAE,KAA8B,EAAE,EAAE;gBAChD,MAAM,MAAM,GAAG,MAAM,kBAAU,CAAC,QAAQ,CAAC,OAAO,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC;gBAC9D,uDAAuD;gBACvD,OAAO,MAAM,CAAC,OAAO;qBAClB,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,YAAY,CAAC,CAAC,QAAQ,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;qBAC7D,IAAI,CAAC,IAAI,CAAC,CAAC;YAChB,CAAC;SACF,CAAC;IACJ,CAAC;IAED,OAAO,KAAK,CAAC;AACf,CAAC;AAnBD,kDAmBC"}

package/dist/extensions/rag_search.d.ts

@@ -0,0 +1,38 @@
+/**
+ * BoneCode Extension: RAG Hybrid Search
+ *
+ * Implements the hybrid_search extension point from rag.bone.
+ * Combines:
+ *   1. pgvector cosine similarity search (semantic)
+ *   2. PostgreSQL full-text search (keyword)
+ *   3. Metadata filters (language, file path, symbol type)
+ *   4. Reciprocal Rank Fusion (RRF) to merge results
+ *
+ * This file is preserved across BoneScript recompilation.
+ */
+export interface SearchResult {
+    chunk_id: string;
+    code_file_id: string;
+    file_path: string;
+    content: string;
+    start_line: number;
+    end_line: number;
+    language?: string;
+    symbol_name?: string;
+    symbol_type?: string;
+    score: number;
+    vector_score?: number;
+    text_score?: number;
+}
+export interface SearchFilters {
+    language?: string;
+    file_path_pattern?: string;
+    symbol_type?: string;
+    min_score?: number;
+}
+export declare function hybrid_search(knowledge_base_id: string, query_embedding: number[], query_text: string, filters: SearchFilters, limit: number): Promise<SearchResult[]>;
+/**
+ * High-level context retrieval for agent use.
+ * Embeds the query, runs hybrid search, and formats results for LLM context.
+ */
+export declare function retrieve_context_for_agent(knowledge_base_id: string, query: string, limit?: number, filters?: SearchFilters): Promise<string>;