kernl 0.2.1 → 0.6.1

package/dist/kernl.js

@@ -1,26 +1,52 @@
-import { KernlHooks } from "./lifecycle";
+import { KernlHooks } from "./lifecycle.js";
+import { InMemoryStorage } from "./storage/index.js";
+import { RThreads } from "./api/resources/threads/index.js";
 /**
- * The kernl - manages agent processes, scheduling, and task lifecycle
+ * The kernl - manages agent processes, scheduling, and task lifecycle.
  *
  * Orchestrates agent execution, including guardrails, tool calls, session persistence, and
  * tracing.
  */
 export class Kernl extends KernlHooks {
     agents = new Map();
-    threads = new Map();
+    models = new Map();
+    storage;
+    athreads = new Map(); /* active threads */
+    initPromise = null;
+    // --- public API ---
+    threads; /* Threads resource */
+    constructor(options = {}) {
+        super();
+        this.storage = options.storage?.db ?? new InMemoryStorage();
+        this.storage.bind({ agents: this.agents, models: this.models });
+        this.threads = new RThreads(this.storage.threads);
+    }
     /**
      * Registers a new agent with the kernl instance.
      */
     register(agent) {
         this.agents.set(agent.id, agent);
         agent.bind(this);
+        // (TODO): implement exhaustive model registry in protocol/ package
+        //
+        // auto-populate model registry for storage hydration
+        const key = `${agent.model.provider}/${agent.model.modelId}`;
+        if (!this.models.has(key)) {
+            this.models.set(key, agent.model);
+        }
     }
     /**
      * Spawn a new thread - blocking execution
      */
     async spawn(thread) {
-        this.threads.set(thread.id, thread);
-        return await thread.execute();
+        await this.ensureInitialized();
+        this.athreads.set(thread.tid, thread);
+        try {
+            return await thread.execute();
+        }
+        finally {
+            this.athreads.delete(thread.tid);
+        }
     }
     /**
      * Schedule an existing thread - blocking execution
@@ -28,23 +54,60 @@ export class Kernl extends KernlHooks {
      * NOTE: just blocks for now
      */
     async schedule(thread) {
-        return await thread.execute();
+        await this.ensureInitialized();
+        this.athreads.set(thread.tid, thread);
+        try {
+            return await thread.execute();
+        }
+        finally {
+            this.athreads.delete(thread.tid);
+        }
     }
     /**
-     * (TMP) - probably won't make sense with assync scheduling contexts
+     * (TMP) - won't make sense in async scheduling contexts
      *
      * Spawn a new thread - streaming execution
      */
     async *spawnStream(thread) {
-        this.threads.set(thread.id, thread);
-        yield* thread.stream();
+        await this.ensureInitialized();
+        this.athreads.set(thread.tid, thread);
+        try {
+            yield* thread.stream();
+        }
+        finally {
+            this.athreads.delete(thread.tid);
+        }
     }
     /**
-     * (TMP) - probably won't make sense with assync scheduling contexts
+     * (TMP) - won't make sense with async scheduling contexts
      *
      * Schedule an existing thread - streaming execution
      */
     async *scheduleStream(thread) {
-        yield* thread.stream();
+        await this.ensureInitialized();
+        this.athreads.set(thread.tid, thread);
+        try {
+            yield* thread.stream();
+        }
+        finally {
+            this.athreads.delete(thread.tid);
+        }
+    }
+    /**
+     * Ensure the underlying storage backend has been initialized.
+     *
+     * This is called lazily on first use so that callers do not need to worry
+     * about calling storage.init() themselves. Safe and idempotent to call
+     * multiple times; concurrent calls share the same initialization promise.
+     */
+    async ensureInitialized() {
+        if (!this.initPromise) {
+            this.initPromise = this.storage.init().catch((error) => {
+                // allow a retry if initialization fails.
+                this.initPromise = null;
+                throw error;
+            });
+        }
+        await this.initPromise;
     }
 }

package/dist/lib/env.d.ts

@@ -7,11 +7,11 @@ import { z } from "zod";
  */
 declare const envSchema: z.ZodObject<{
     LOG_LEVEL: z.ZodDefault<z.ZodEnum<{
-        error: "error";
         trace: "trace";
         debug: "debug";
         info: "info";
         warn: "warn";
+        error: "error";
         fatal: "fatal";
     }>>;
     KERNL_LOG_MODEL_DATA: z.ZodPipe<z.ZodOptional<z.ZodEnum<{
@@ -27,11 +27,11 @@ declare const envSchema: z.ZodObject<{
  * Validated and type-safe environment variables
  *
  * @example
- * import { env } from '../env';
+ * import { env } from '@/env';
  * console.log(env.LOG_LEVEL);
  */
 export declare const env: {
-    LOG_LEVEL: "error" | "trace" | "debug" | "info" | "warn" | "fatal";
+    LOG_LEVEL: "trace" | "debug" | "info" | "warn" | "error" | "fatal";
     KERNL_LOG_MODEL_DATA: boolean;
     KERNL_LOG_TOOL_DATA: boolean;
 };

package/dist/lib/env.js

@@ -23,7 +23,7 @@ const envSchema = z.object({
  * Validated and type-safe environment variables
  *
  * @example
- * import { env } from '../env';
+ * import { env } from '@/env';
  * console.log(env.LOG_LEVEL);
  */
 export const env = envSchema.parse(process.env);

package/dist/lib/error.d.ts

@@ -1,7 +1,7 @@
-import { InputGuardrailResult, OutputGuardrailMetadata, OutputGuardrailResult } from "../guardrail";
+import { InputGuardrailResult, OutputGuardrailMetadata, OutputGuardrailResult } from "../guardrail.js";
 type SerializedThread = any;
-import { AgentResponseType } from "../types/agent";
-import { TextResponse } from "../types/thread";
+import { AgentResponseType } from "../types/agent.js";
+import { TextResponse } from "../types/thread.js";
 /**
  * Abstract base class for all `kernl` errors
  * Requires subclasses to implement JSON serialization

package/dist/lib/logger.js

@@ -1,5 +1,5 @@
 import pino from "pino";
-import { env } from "./env";
+import { env } from "./env.js";
 /**
  * By default we don't log LLM inputs/outputs, to prevent exposing sensitive data.
  * Set KERNL_LOG_MODEL_DATA=true to enable.

package/dist/lifecycle.d.ts

@@ -1,10 +1,10 @@
 import { EventEmitter } from "node:events";
-import { Agent } from "./agent";
-import { Context, UnknownContext } from "./context";
-import { Tool } from "./tool";
+import { Agent } from "./agent.js";
+import { Context, UnknownContext } from "./context.js";
+import { Tool } from "./tool/index.js";
 import type { ToolCall } from "@kernl-sdk/protocol";
-import { AgentResponseType } from "./types/agent";
-import { TextResponse } from "./types/thread";
+import { AgentResponseType } from "./types/agent.js";
+import { TextResponse } from "./types/thread.js";
 export type EventEmitterEvents = Record<string, any[]>;
 /**
  * Generic typed event emitter that wraps Node's EventEmitter with type safety

package/dist/mcp/__tests__/base.test.js

@@ -1,6 +1,6 @@
 import { describe, it, expect, vi } from "vitest";
-import { BaseMCPServer } from "../base";
-import { logger } from "../../lib/logger";
+import { BaseMCPServer } from "../base.js";
+import { logger } from "../../lib/logger.js";
 // Create a minimal concrete implementation for testing
 class TestMCPServer extends BaseMCPServer {
     id;

package/dist/mcp/__tests__/fixtures/utils.d.ts

@@ -1,4 +1,4 @@
-import { MCPServerStdio } from "../../stdio";
+import { MCPServerStdio } from "../../stdio.js";
 /**
  * Helper to run a test with an MCP server, ensuring cleanup.
  */

package/dist/mcp/__tests__/fixtures/utils.js

@@ -1,4 +1,4 @@
-import { MCPServerStdio } from "../../stdio";
+import { MCPServerStdio } from "../../stdio.js";
 /**
  * Helper to run a test with an MCP server, ensuring cleanup.
  */

package/dist/mcp/__tests__/integration.test.js

@@ -1,14 +1,14 @@
 import { describe, it, expect, beforeEach, afterEach } from "vitest";
 import path from "path";
-import { MCPServerStdio } from "../stdio";
-import { MCPToolkit } from "../../tool/toolkit";
-import { FunctionToolkit } from "../../tool/toolkit";
-import { Agent } from "../../agent";
-import { Context } from "../../context";
-import { tool } from "../../tool";
+import { MCPServerStdio } from "../stdio.js";
+import { MCPToolkit } from "../../tool/toolkit.js";
+import { FunctionToolkit } from "../../tool/toolkit.js";
+import { Agent } from "../../agent.js";
+import { Context } from "../../context.js";
+import { tool } from "../../tool/index.js";
 import { z } from "zod";
-import { createMCPToolStaticFilter } from "../utils";
-import { createMockModel } from "../../thread/__tests__/fixtures/mock-model";
+import { createMCPToolStaticFilter } from "../utils.js";
+import { createMockModel } from "../../thread/__tests__/fixtures/mock-model.js";
 const TEST_SERVER = path.join(__dirname, "fixtures", "server.ts");
 // Mock model for tests that only need toolkit functionality
 const mockModel = createMockModel(async () => ({
@@ -139,7 +139,7 @@ describe("MCP Integration Tests", () => {
                 model: mockModel,
                 toolkits: [toolkit],
             });
-            const context = new Context({});
+            const context = new Context("test-namespace", {});
             const tools = await toolkit.list(context);
             // Should have 3 math tools only
             expect(tools.length).toBe(3);
@@ -170,7 +170,7 @@ describe("MCP Integration Tests", () => {
                 model: mockModel,
                 toolkits: [toolkit],
             });
-            const context = new Context({});
+            const context = new Context("test-namespace", {});
             const tools = await toolkit.list(context);
             // Should have only add and divide (multiply blocked by toolkit filter)
             expect(tools.length).toBe(2);
@@ -202,7 +202,7 @@ describe("MCP Integration Tests", () => {
                 model: mockModel,
                 toolkits: [toolkit],
             });
-            const context = new Context({ userId: "test-user" });
+            const context = new Context("test-namespace", { userId: "test-user" });
             await toolkit.list(context);
             // Verify context was passed correctly
             expect(receivedContext).toBeDefined();
@@ -230,7 +230,7 @@ describe("MCP Integration Tests", () => {
                 model: mockModel,
                 toolkits: [mcpToolkit],
             });
-            const context = new Context({});
+            const context = new Context("test-namespace", {});
             const tools = await agent.tools(context);
             expect(tools.length).toBe(6);
             expect(tools.find((t) => t.id === "add")).toBeDefined();
@@ -255,7 +255,7 @@ describe("MCP Integration Tests", () => {
                 toolkits: [mcpToolkit],
             });
             // Populate toolkit cache
-            const context = new Context({});
+            const context = new Context("test-namespace", {});
             await agent.tools(context);
             // Now get specific tool
             const addTool = agent.tool("add");
@@ -280,7 +280,7 @@ describe("MCP Integration Tests", () => {
                 model: mockModel,
                 toolkits: [mcpToolkit],
             });
-            const context = new Context({});
+            const context = new Context("test-namespace", {});
             await agent.tools(context);
             const addTool = agent.tool("add");
             expect(addTool).toBeDefined();
@@ -325,7 +325,7 @@ describe("MCP Integration Tests", () => {
                 model: mockModel,
                 toolkits: [mcpToolkit, functionToolkit],
             });
-            const context = new Context({});
+            const context = new Context("test-namespace", {});
             const tools = await agent.tools(context);
             // Should have 7 tools (6 MCP + 1 local)
             expect(tools.length).toBe(7);
@@ -367,7 +367,7 @@ describe("MCP Integration Tests", () => {
                 model: mockModel,
                 toolkits: [mcpToolkit, functionToolkit],
             });
-            const context = new Context({});
+            const context = new Context("test-namespace", {});
             // Should throw error about duplicate tool IDs
             await expect(agent.tools(context)).rejects.toThrow(/Duplicate tool IDs found/);
             await mcpToolkit.destroy();

package/dist/mcp/__tests__/stdio.test.js

@@ -1,6 +1,6 @@
 import { describe, it, expect } from "vitest";
-import { MCPServerStdio } from "../stdio";
-import { withMCPServer, createMCPServer } from "./fixtures/utils";
+import { MCPServerStdio } from "../stdio.js";
+import { withMCPServer, createMCPServer } from "./fixtures/utils.js";
 import path from "path";
 const TEST_SERVER = path.join(__dirname, "fixtures", "server.ts");
 describe("MCPServerStdio", () => {

package/dist/mcp/__tests__/utils.test.js

@@ -1,6 +1,6 @@
 import { describe, it, expect, vi } from "vitest";
-import { mcpToFunctionTool, createMCPToolStaticFilter } from "../utils";
-import { Context } from "../../context";
+import { mcpToFunctionTool, createMCPToolStaticFilter } from "../utils.js";
+import { Context } from "../../context.js";
 describe("mcpToFunctionTool", () => {
     // Create a mock MCP server
     const createMockServer = () => ({
@@ -79,7 +79,7 @@ describe("mcpToFunctionTool", () => {
             },
         };
         const functionTool = mcpToFunctionTool(server, mcpTool);
-        const ctx = new Context({});
+        const ctx = new Context("test-namespace", {});
         const input = { arg1: "value1" };
         await functionTool.invoke(ctx, JSON.stringify(input));
         expect(server.callTool).toHaveBeenCalledWith("test_tool", input);
@@ -98,7 +98,7 @@ describe("mcpToFunctionTool", () => {
             },
         };
         const functionTool = mcpToFunctionTool(server, mcpTool);
-        const ctx = new Context({});
+        const ctx = new Context("test-namespace", {});
         const toolResult = await functionTool.invoke(ctx, JSON.stringify({}));
         expect(toolResult.state).toBe("completed");
         expect(toolResult.result).toEqual({ type: "text", text: "single result" });
@@ -119,7 +119,7 @@ describe("mcpToFunctionTool", () => {
             },
         };
         const functionTool = mcpToFunctionTool(server, mcpTool);
-        const ctx = new Context({});
+        const ctx = new Context("test-namespace", {});
         const toolResult = await functionTool.invoke(ctx, JSON.stringify({}));
         expect(toolResult.state).toBe("completed");
         expect(toolResult.result).toEqual(multipleItems);
@@ -141,7 +141,7 @@ describe("mcpToFunctionTool", () => {
         };
         const functionTool1 = mcpToFunctionTool(server, tool1);
         const functionTool2 = mcpToFunctionTool(server, tool2);
-        const ctx = new Context({});
+        const ctx = new Context("test-namespace", {});
         await functionTool1.invoke(ctx, JSON.stringify({}));
         expect(server.callTool).toHaveBeenCalledWith("tool1", {});
         await functionTool2.invoke(ctx, JSON.stringify({}));
@@ -163,7 +163,7 @@ describe("mcpToFunctionTool", () => {
         };
         const functionTool = mcpToFunctionTool(server, mcpTool);
         // Invoke takes (context, params as JSON string) in that order
-        const ctx = new Context({});
+        const ctx = new Context("test-namespace", {});
         const params = { foo: "bar" };
         await functionTool.invoke(ctx, JSON.stringify(params));
         // Verify the tool was called with the params (not the context)
@@ -173,7 +173,7 @@ describe("mcpToFunctionTool", () => {
 describe("createMCPToolStaticFilter", () => {
     // Helper to create mock filter context
     const createMockFilterContext = () => ({
-        context: new Context({}),
+        context: new Context("test-namespace", {}),
         agent: {},
         serverId: "test-server",
     });

package/dist/mcp/base.d.ts

@@ -1,5 +1,5 @@
-import { Logger } from "../lib/logger";
-import { MCPTool, MCPToolFilter, CallToolResultContent } from "./types";
+import { Logger } from "../lib/logger.js";
+import { MCPTool, MCPToolFilter, CallToolResultContent } from "./types.js";
 export declare const DEFAULT_STDIO_MCP_CLIENT_LOGGER_NAME = "kernl:stdio-mcp-client";
 export declare const DEFAULT_SSE_MCP_CLIENT_LOGGER_NAME = "kernl:sse-mcp-client";
 export declare const DEFAULT_STREAMABLE_HTTP_MCP_CLIENT_LOGGER_NAME = "kernl:streamable-http-mcp-client";

package/dist/mcp/http.d.ts

@@ -1,6 +1,6 @@
 import { Client } from "@modelcontextprotocol/sdk/client/index.js";
-import { BaseMCPServer } from "./base";
-import type { MCPTool, CallToolResultContent, MCPServerStreamableHttpOptions, InitializeResult } from "./types";
+import { BaseMCPServer } from "./base.js";
+import type { MCPTool, CallToolResultContent, MCPServerStreamableHttpOptions, InitializeResult } from "./types.js";
 /**
  * MCP server client that communicates over streamable HTTP protocol.
  */
@@ -9,7 +9,7 @@ export declare class MCPServerStreamableHttp extends BaseMCPServer {
     protected session: Client | null;
     protected timeout: number;
     protected serverInitializeResult: InitializeResult | null;
-    params: MCPServerStreamableHttpOptions;
+    options: MCPServerStreamableHttpOptions;
     private transport;
     constructor(options: MCPServerStreamableHttpOptions);
     /**

package/dist/mcp/http.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"http.d.ts","sourceRoot":"","sources":["../../src/mcp/http.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,EAAE,MAAM,2CAA2C,CAAC;AAUnE,OAAO,EACL,aAAa,EAEd,MAAM,QAAQ,CAAC;AAChB,OAAO,KAAK,EACV,OAAO,EACP,qBAAqB,EACrB,8BAA8B,EAC9B,gBAAgB,EACjB,MAAM,SAAS,CAAC;AAEjB;;GAEG;AACH,qBAAa,uBAAwB,SAAQ,aAAa;IACxD,QAAQ,CAAC,EAAE,EAAE,MAAM,CAAC;IACpB,SAAS,CAAC,OAAO,EAAE,MAAM,GAAG,IAAI,CAAQ;IACxC,SAAS,CAAC,OAAO,EAAE,MAAM,CAAC;IAC1B,SAAS,CAAC,sBAAsB,EAAE,gBAAgB,GAAG,IAAI,CAAQ;IAEjE,MAAM,EAAE,8BAA8B,CAAC;IACvC,OAAO,CAAC,SAAS,CAAa;gBAElB,OAAO,EAAE,8BAA8B;IAcnD;;OAEG;IACG,OAAO,IAAI,OAAO,CAAC,IAAI,CAAC;IAgC9B;;OAEG;IACG,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC;IAW5B;;;;OAIG;cACa,UAAU,IAAI,OAAO,CAAC,OAAO,EAAE,CAAC;IAYhD;;OAEG;IACG,QAAQ,CACZ,QAAQ,EAAE,MAAM,EAChB,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,IAAI,GACnC,OAAO,CAAC,qBAAqB,CAAC;CA2BlC"}
+{"version":3,"file":"http.d.ts","sourceRoot":"","sources":["../../src/mcp/http.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,EAAE,MAAM,2CAA2C,CAAC;AAUnE,OAAO,EACL,aAAa,EAEd,MAAM,QAAQ,CAAC;AAChB,OAAO,KAAK,EACV,OAAO,EACP,qBAAqB,EACrB,8BAA8B,EAC9B,gBAAgB,EACjB,MAAM,SAAS,CAAC;AAEjB;;GAEG;AACH,qBAAa,uBAAwB,SAAQ,aAAa;IACxD,QAAQ,CAAC,EAAE,EAAE,MAAM,CAAC;IACpB,SAAS,CAAC,OAAO,EAAE,MAAM,GAAG,IAAI,CAAQ;IACxC,SAAS,CAAC,OAAO,EAAE,MAAM,CAAC;IAC1B,SAAS,CAAC,sBAAsB,EAAE,gBAAgB,GAAG,IAAI,CAAQ;IAEjE,OAAO,EAAE,8BAA8B,CAAC;IACxC,OAAO,CAAC,SAAS,CAAa;gBAElB,OAAO,EAAE,8BAA8B;IAcnD;;OAEG;IACG,OAAO,IAAI,OAAO,CAAC,IAAI,CAAC;IAgC9B;;OAEG;IACG,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC;IAW5B;;;;OAIG;cACa,UAAU,IAAI,OAAO,CAAC,OAAO,EAAE,CAAC;IAYhD;;OAEG;IACG,QAAQ,CACZ,QAAQ,EAAE,MAAM,EAChB,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,IAAI,GACnC,OAAO,CAAC,qBAAqB,CAAC;CA2BlC"}

package/dist/mcp/http.js

@@ -2,8 +2,8 @@ import { Client } from "@modelcontextprotocol/sdk/client/index.js";
 import { StreamableHTTPClientTransport } from "@modelcontextprotocol/sdk/client/streamableHttp.js";
 import { DEFAULT_REQUEST_TIMEOUT_MSEC } from "@modelcontextprotocol/sdk/shared/protocol.js";
 import { ListToolsResultSchema, CallToolResultSchema, } from "@modelcontextprotocol/sdk/types.js";
-import { getLogger } from "../lib/logger";
-import { BaseMCPServer, DEFAULT_STREAMABLE_HTTP_MCP_CLIENT_LOGGER_NAME, } from "./base";
+import { getLogger } from "../lib/logger.js";
+import { BaseMCPServer, DEFAULT_STREAMABLE_HTTP_MCP_CLIENT_LOGGER_NAME, } from "./base.js";
 /**
  * MCP server client that communicates over streamable HTTP protocol.
  */
@@ -12,7 +12,7 @@ export class MCPServerStreamableHttp extends BaseMCPServer {
     session = null;
     timeout;
     serverInitializeResult = null;
-    params;
+    options;
     transport = null;
     constructor(options) {
         super({
@@ -21,8 +21,8 @@ export class MCPServerStreamableHttp extends BaseMCPServer {
             logger: options.logger ??
                 getLogger(DEFAULT_STREAMABLE_HTTP_MCP_CLIENT_LOGGER_NAME),
         });
-        this.params = options;
-        this.id = options.id || `streamable-http: ${this.params.url}`;
+        this.options = options;
+        this.id = options.id || `streamable-http: ${this.options.url}`;
         this.timeout = options.timeout ?? DEFAULT_REQUEST_TIMEOUT_MSEC;
     }
     /**
@@ -30,12 +30,12 @@ export class MCPServerStreamableHttp extends BaseMCPServer {
      */
     async connect() {
         try {
-            this.transport = new StreamableHTTPClientTransport(new URL(this.params.url), {
-                authProvider: this.params.authProvider,
-                requestInit: this.params.requestInit,
-                fetch: this.params.fetch,
-                reconnectionOptions: this.params.reconnectionOptions,
-                sessionId: this.params.sessionId,
+            this.transport = new StreamableHTTPClientTransport(new URL(this.options.url), {
+                authProvider: this.options.authProvider,
+                requestInit: this.options.requestInit,
+                fetch: this.options.fetch,
+                reconnectionOptions: this.options.reconnectionOptions,
+                sessionId: this.options.sessionId,
             });
             this.session = new Client({
                 name: this.id,

package/dist/mcp/sse.d.ts

@@ -1,6 +1,6 @@
 import { Client } from "@modelcontextprotocol/sdk/client/index.js";
-import { BaseMCPServer } from "./base";
-import type { MCPTool, CallToolResultContent, MCPServerSSEOptions, InitializeResult } from "./types";
+import { BaseMCPServer } from "./base.js";
+import type { MCPTool, CallToolResultContent, MCPServerSSEOptions, InitializeResult } from "./types.js";
 /**
  * MCP server client that communicates over Server-Sent Events (SSE).
  */
@@ -9,7 +9,7 @@ export declare class MCPServerSSE extends BaseMCPServer {
     protected session: Client | null;
     protected timeout: number;
     protected serverInitializeResult: InitializeResult | null;
-    params: MCPServerSSEOptions;
+    options: MCPServerSSEOptions;
     private transport;
     constructor(options: MCPServerSSEOptions);
     /**

package/dist/mcp/sse.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"sse.d.ts","sourceRoot":"","sources":["../../src/mcp/sse.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,EAAE,MAAM,2CAA2C,CAAC;AAUnE,OAAO,EAAE,aAAa,EAAsC,MAAM,QAAQ,CAAC;AAC3E,OAAO,KAAK,EACV,OAAO,EACP,qBAAqB,EACrB,mBAAmB,EACnB,gBAAgB,EACjB,MAAM,SAAS,CAAC;AAEjB;;GAEG;AACH,qBAAa,YAAa,SAAQ,aAAa;IAC7C,QAAQ,CAAC,EAAE,EAAE,MAAM,CAAC;IACpB,SAAS,CAAC,OAAO,EAAE,MAAM,GAAG,IAAI,CAAQ;IACxC,SAAS,CAAC,OAAO,EAAE,MAAM,CAAC;IAC1B,SAAS,CAAC,sBAAsB,EAAE,gBAAgB,GAAG,IAAI,CAAQ;IAEjE,MAAM,EAAE,mBAAmB,CAAC;IAC5B,OAAO,CAAC,SAAS,CAAa;gBAElB,OAAO,EAAE,mBAAmB;IAYxC;;OAEG;IACG,OAAO,IAAI,OAAO,CAAC,IAAI,CAAC;IA2B9B;;;;OAIG;cACa,UAAU,IAAI,OAAO,CAAC,OAAO,EAAE,CAAC;IAYhD;;OAEG;IACG,QAAQ,CACZ,QAAQ,EAAE,MAAM,EAChB,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,IAAI,GACnC,OAAO,CAAC,qBAAqB,CAAC;IA4BjC;;OAEG;IACG,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC;CAU7B"}
+{"version":3,"file":"sse.d.ts","sourceRoot":"","sources":["../../src/mcp/sse.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,EAAE,MAAM,2CAA2C,CAAC;AAUnE,OAAO,EAAE,aAAa,EAAsC,MAAM,QAAQ,CAAC;AAC3E,OAAO,KAAK,EACV,OAAO,EACP,qBAAqB,EACrB,mBAAmB,EACnB,gBAAgB,EACjB,MAAM,SAAS,CAAC;AAEjB;;GAEG;AACH,qBAAa,YAAa,SAAQ,aAAa;IAC7C,QAAQ,CAAC,EAAE,EAAE,MAAM,CAAC;IACpB,SAAS,CAAC,OAAO,EAAE,MAAM,GAAG,IAAI,CAAQ;IACxC,SAAS,CAAC,OAAO,EAAE,MAAM,CAAC;IAC1B,SAAS,CAAC,sBAAsB,EAAE,gBAAgB,GAAG,IAAI,CAAQ;IAEjE,OAAO,EAAE,mBAAmB,CAAC;IAC7B,OAAO,CAAC,SAAS,CAAa;gBAElB,OAAO,EAAE,mBAAmB;IAYxC;;OAEG;IACG,OAAO,IAAI,OAAO,CAAC,IAAI,CAAC;IA2B9B;;;;OAIG;cACa,UAAU,IAAI,OAAO,CAAC,OAAO,EAAE,CAAC;IAYhD;;OAEG;IACG,QAAQ,CACZ,QAAQ,EAAE,MAAM,EAChB,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,IAAI,GACnC,OAAO,CAAC,qBAAqB,CAAC;IA4BjC;;OAEG;IACG,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC;CAU7B"}

package/dist/mcp/sse.js

@@ -2,8 +2,8 @@ import { Client } from "@modelcontextprotocol/sdk/client/index.js";
 import { SSEClientTransport } from "@modelcontextprotocol/sdk/client/sse.js";
 import { DEFAULT_REQUEST_TIMEOUT_MSEC } from "@modelcontextprotocol/sdk/shared/protocol.js";
 import { ListToolsResultSchema, CallToolResultSchema, } from "@modelcontextprotocol/sdk/types.js";
-import { getLogger } from "../lib/logger";
-import { BaseMCPServer, DEFAULT_SSE_MCP_CLIENT_LOGGER_NAME } from "./base";
+import { getLogger } from "../lib/logger.js";
+import { BaseMCPServer, DEFAULT_SSE_MCP_CLIENT_LOGGER_NAME } from "./base.js";
 /**
  * MCP server client that communicates over Server-Sent Events (SSE).
  */
@@ -12,7 +12,7 @@ export class MCPServerSSE extends BaseMCPServer {
     session = null;
     timeout;
     serverInitializeResult = null;
-    params;
+    options;
     transport = null;
     constructor(options) {
         super({
@@ -20,8 +20,8 @@ export class MCPServerSSE extends BaseMCPServer {
             toolFilter: options.toolFilter,
             logger: options.logger ?? getLogger(DEFAULT_SSE_MCP_CLIENT_LOGGER_NAME),
         });
-        this.params = options;
-        this.id = options.id || `sse: ${this.params.url}`;
+        this.options = options;
+        this.id = options.id || `sse: ${this.options.url}`;
         this.timeout = options.timeout ?? DEFAULT_REQUEST_TIMEOUT_MSEC;
     }
     /**
@@ -29,10 +29,10 @@ export class MCPServerSSE extends BaseMCPServer {
      */
     async connect() {
         try {
-            this.transport = new SSEClientTransport(new URL(this.params.url), {
-                authProvider: this.params.authProvider,
-                requestInit: this.params.requestInit,
-                eventSourceInit: this.params.eventSourceInit,
+            this.transport = new SSEClientTransport(new URL(this.options.url), {
+                authProvider: this.options.authProvider,
+                requestInit: this.options.requestInit,
+                eventSourceInit: this.options.eventSourceInit,
             });
             this.session = new Client({
                 name: this.id,

package/dist/mcp/stdio.d.ts

@@ -1,6 +1,6 @@
 import { Client } from "@modelcontextprotocol/sdk/client/index.js";
-import { BaseMCPServer } from "./base";
-import type { MCPTool, CallToolResultContent, MCPServerStdioOptions, InitializeResult } from "./types";
+import { BaseMCPServer } from "./base.js";
+import type { MCPTool, CallToolResultContent, MCPServerStdioOptions, InitializeResult } from "./types.js";
 /**
  * MCP server client that communicates over stdio (standard input/output).
  */

package/dist/mcp/stdio.js

@@ -2,8 +2,8 @@ import { Client } from "@modelcontextprotocol/sdk/client/index.js";
 import { StdioClientTransport } from "@modelcontextprotocol/sdk/client/stdio.js";
 import { DEFAULT_REQUEST_TIMEOUT_MSEC } from "@modelcontextprotocol/sdk/shared/protocol.js";
 import { ListToolsResultSchema, CallToolResultSchema, } from "@modelcontextprotocol/sdk/types.js";
-import { getLogger } from "../lib/logger";
-import { BaseMCPServer, DEFAULT_STDIO_MCP_CLIENT_LOGGER_NAME } from "./base";
+import { getLogger } from "../lib/logger.js";
+import { BaseMCPServer, DEFAULT_STDIO_MCP_CLIENT_LOGGER_NAME } from "./base.js";
 /**
  * MCP server client that communicates over stdio (standard input/output).
  */

package/dist/mcp/types.d.ts

@@ -1,8 +1,8 @@
 import { z } from "zod";
 import type { Tool } from "@modelcontextprotocol/sdk/types.js";
-import { Agent } from "../agent";
-import { Context, UnknownContext } from "../context";
-import { Logger } from "../lib/logger";
+import { Agent } from "../agent.js";
+import { Context, UnknownContext } from "../context.js";
+import { Logger } from "../lib/logger.js";
 /**
  * MCP tool type from the SDK.
  * We use the SDK's type directly to avoid compatibility issues.

package/dist/mcp/utils.d.ts

@@ -1,11 +1,11 @@
 import { z } from "zod";
-import { UnknownContext } from "../context";
-import { MCPServer } from "./base";
-import type { MCPTool, MCPToolFilter } from "./types";
+import { UnknownContext } from "../context.js";
+import { MCPServer } from "./base.js";
+import type { MCPTool, MCPToolFilter } from "./types.js";
 /**
  * Converts an MCP tool definition into a function tool usable by the SDK.
  */
-export declare function mcpToFunctionTool(server: MCPServer, mcpTool: MCPTool): import("../tool").FunctionTool<unknown, z.ZodRecord<z.ZodString, z.ZodAny> | undefined, {
+export declare function mcpToFunctionTool(server: MCPServer, mcpTool: MCPTool): import("../tool/index.js").FunctionTool<unknown, z.ZodObject<{}, z.core.$strip>, {
     type: string;
     text: string;
 } | {

package/dist/mcp/utils.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"utils.d.ts","sourceRoot":"","sources":["../../src/mcp/utils.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAExB,OAAO,EAAE,cAAc,EAAE,MAAM,WAAW,CAAC;AAG3C,OAAO,EAAE,SAAS,EAAE,MAAM,QAAQ,CAAC;AACnC,OAAO,KAAK,EAAE,OAAO,EAAE,aAAa,EAAE,MAAM,SAAS,CAAC;AAEtD;;GAEG;AACH,wBAAgB,iBAAiB,CAAC,MAAM,EAAE,SAAS,EAAE,OAAO,EAAE,OAAO;;;;;;KAmBpE;AAMD;;;GAGG;AACH,wBAAgB,yBAAyB,CAAC,QAAQ,GAAG,cAAc,EAAE,OAAO,CAAC,EAAE;IAC7E,OAAO,CAAC,EAAE,MAAM,EAAE,CAAC;IACnB,OAAO,CAAC,EAAE,MAAM,EAAE,CAAC;CACpB,GAAG,aAAa,CAAC,QAAQ,CAAC,GAAG,SAAS,CAkBtC"}
+{"version":3,"file":"utils.d.ts","sourceRoot":"","sources":["../../src/mcp/utils.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAExB,OAAO,EAAE,cAAc,EAAE,MAAM,WAAW,CAAC;AAG3C,OAAO,EAAE,SAAS,EAAE,MAAM,QAAQ,CAAC;AACnC,OAAO,KAAK,EAAE,OAAO,EAAE,aAAa,EAAE,MAAM,SAAS,CAAC;AAEtD;;GAEG;AACH,wBAAgB,iBAAiB,CAAC,MAAM,EAAE,SAAS,EAAE,OAAO,EAAE,OAAO;;;;;;KAoBpE;AAMD;;;GAGG;AACH,wBAAgB,yBAAyB,CAAC,QAAQ,GAAG,cAAc,EAAE,OAAO,CAAC,EAAE;IAC7E,OAAO,CAAC,EAAE,MAAM,EAAE,CAAC;IACnB,OAAO,CAAC,EAAE,MAAM,EAAE,CAAC;CACpB,GAAG,aAAa,CAAC,QAAQ,CAAC,GAAG,SAAS,CAkBtC"}

package/dist/mcp/utils.js

@@ -1,5 +1,5 @@
 import { z } from "zod";
-import { tool } from "../tool";
+import { tool } from "../tool/index.js";
 /**
  * Converts an MCP tool definition into a function tool usable by the SDK.
  */
@@ -8,11 +8,10 @@ export function mcpToFunctionTool(server, mcpTool) {
         const content = await server.callTool(mcpTool.name, input);
         return content.length === 1 ? content[0] : content;
     }
-    // MCP tools accept an object with any properties based on their inputSchema
-    // We use z.record(z.any()) to represent this flexible schema
-    const parameters = mcpTool.inputSchema
-        ? z.record(z.string(), z.any())
-        : undefined;
+    const hasProperties = mcpTool.inputSchema &&
+        Object.keys(mcpTool.inputSchema.properties || {}).length > 0;
+    // If tool has properties, use passthrough to accept any object, else empty object (matches AI SDK)
+    const parameters = hasProperties ? z.object({}).passthrough() : z.object({});
     return tool({
         id: mcpTool.name,
         name: mcpTool.name,

package/dist/storage/__tests__/in-memory.test.d.ts

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=in-memory.test.d.ts.map

package/dist/storage/__tests__/in-memory.test.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"in-memory.test.d.ts","sourceRoot":"","sources":["../../../src/storage/__tests__/in-memory.test.ts"],"names":[],"mappings":""}