@agentium/transport 1.0.0 → 2.0.0

package/dist/index.cjs

@@ -20,12 +20,15 @@ var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: tru
 // src/index.ts
 var index_exports = {};
 __export(index_exports, {
+  InMemoryEventLog: () => InMemoryEventLog,
   MCPManager: () => MCPManager,
+  agentUIStream: () => agentUIStream,
   buildMultiModalInput: () => buildMultiModalInput,
   createA2AServer: () => createA2AServer,
   createAdminRouter: () => createAdminRouter,
   createAgentGateway: () => createAgentGateway,
   createAgentRouter: () => createAgentRouter,
+  createAgentUIStreamResponse: () => createAgentUIStreamResponse,
   createBrowserGateway: () => createBrowserGateway,
   createFileUploadMiddleware: () => createFileUploadMiddleware,
   createGatewayRouter: () => createGatewayRouter,
@@ -33,10 +36,13 @@ __export(index_exports, {
   createRbacMiddleware: () => createRbacMiddleware,
   createVisionGateway: () => createVisionGateway,
   createVoiceGateway: () => createVoiceGateway,
+  defaultEventLog: () => defaultEventLog,
   errorHandler: () => errorHandler,
+  formatSSEEvent: () => formatSSEEvent,
   generateAgentCard: () => generateAgentCard,
   generateMultiAgentCard: () => generateMultiAgentCard,
   generateOpenAPISpec: () => generateOpenAPISpec,
+  pipeAgentUIStreamToResponse: () => pipeAgentUIStreamToResponse,
   requestLogger: () => requestLogger
 });
 module.exports = __toCommonJS(index_exports);
@@ -2035,6 +2041,143 @@ function createAgentRouter(opts) {
   return router;
 }
 
+// src/express/ui-stream.ts
+function ssePayload(obj) {
+  return `data: ${JSON.stringify(obj)}
+
+`;
+}
+function agentUIStream(agent, input, options = {}) {
+  const encoder = new TextEncoder();
+  const messageId = `msg-${Date.now().toString(36)}-${Math.random().toString(36).slice(2, 8)}`;
+  let textPartId = 0;
+  return new ReadableStream({
+    async start(controller) {
+      try {
+        controller.enqueue(encoder.encode(ssePayload({ type: "start", messageId })));
+        let currentTextId = null;
+        for await (const chunk of agent.stream(input, {
+          sessionId: options.sessionId,
+          userId: options.userId,
+          apiKey: options.apiKey,
+          signal: options.signal
+        })) {
+          const t = chunk.type;
+          if (t === "text" || t === "text-delta") {
+            const delta = chunk.delta ?? chunk.text ?? "";
+            if (!delta) continue;
+            if (!currentTextId) {
+              currentTextId = `text-${++textPartId}`;
+              controller.enqueue(encoder.encode(ssePayload({ type: "text-start", id: currentTextId })));
+            }
+            controller.enqueue(encoder.encode(ssePayload({ type: "text-delta", id: currentTextId, textDelta: delta })));
+          } else if (t === "reasoning" || t === "thinking") {
+            const r = chunk.reasoning ?? chunk.thinking ?? chunk.delta ?? chunk.text ?? "";
+            if (r) {
+              controller.enqueue(
+                encoder.encode(ssePayload({ type: "reasoning-delta", id: messageId, reasoningDelta: r }))
+              );
+            }
+          } else if (t === "tool.call" || t === "tool-call" || t === "tool-input-start") {
+            if (currentTextId) {
+              controller.enqueue(encoder.encode(ssePayload({ type: "text-end", id: currentTextId })));
+              currentTextId = null;
+            }
+            controller.enqueue(
+              encoder.encode(
+                ssePayload({
+                  type: "tool-input-start",
+                  toolCallId: chunk.toolCallId,
+                  toolName: chunk.toolName
+                })
+              )
+            );
+            const args = chunk.arguments ?? chunk.args ?? chunk.input;
+            if (args !== void 0) {
+              controller.enqueue(
+                encoder.encode(
+                  ssePayload({
+                    type: "tool-input-available",
+                    toolCallId: chunk.toolCallId,
+                    toolName: chunk.toolName,
+                    input: args
+                  })
+                )
+              );
+            }
+          } else if (t === "tool.result" || t === "tool-result" || t === "tool-output-available") {
+            controller.enqueue(
+              encoder.encode(
+                ssePayload({
+                  type: "tool-output-available",
+                  toolCallId: chunk.toolCallId,
+                  output: chunk.output ?? chunk.result
+                })
+              )
+            );
+          } else if (t === "error") {
+            const errText = typeof chunk.error === "string" ? chunk.error : chunk.error?.message ?? "unknown error";
+            controller.enqueue(encoder.encode(ssePayload({ type: "error", errorText: errText })));
+          } else if (t === "finish" || t === "done" || t === "stream.end") {
+            if (currentTextId) {
+              controller.enqueue(encoder.encode(ssePayload({ type: "text-end", id: currentTextId })));
+              currentTextId = null;
+            }
+            controller.enqueue(
+              encoder.encode(
+                ssePayload({
+                  type: "finish-step",
+                  finishReason: chunk.finishReason ?? "stop",
+                  usage: chunk.usage
+                })
+              )
+            );
+          }
+        }
+        if (currentTextId) {
+          controller.enqueue(encoder.encode(ssePayload({ type: "text-end", id: currentTextId })));
+        }
+        controller.enqueue(encoder.encode(ssePayload({ type: "finish" })));
+        controller.enqueue(encoder.encode("data: [DONE]\n\n"));
+        controller.close();
+      } catch (err) {
+        controller.enqueue(encoder.encode(ssePayload({ type: "error", errorText: err?.message ?? String(err) })));
+        controller.close();
+      }
+    }
+  });
+}
+function createAgentUIStreamResponse(agent, input, options = {}) {
+  const body = agentUIStream(agent, input, options);
+  return new Response(body, {
+    headers: {
+      "Content-Type": "text/event-stream",
+      "Cache-Control": "no-cache",
+      Connection: "keep-alive",
+      "x-vercel-ai-ui-message-stream": "v1"
+    }
+  });
+}
+async function pipeAgentUIStreamToResponse(agent, input, res, options = {}) {
+  res.writeHead(200, {
+    "Content-Type": "text/event-stream",
+    "Cache-Control": "no-cache",
+    Connection: "keep-alive",
+    "x-vercel-ai-ui-message-stream": "v1"
+  });
+  const stream = agentUIStream(agent, input, options);
+  const reader = stream.getReader();
+  try {
+    while (true) {
+      const { done, value } = await reader.read();
+      if (done) break;
+      res.write(Buffer.from(value));
+    }
+  } finally {
+    res.end();
+  }
+}
+
 // src/socketio/browser-gateway.ts
 function createBrowserGateway(opts) {
   const ns = opts.io.of(opts.namespace ?? "/agentium-browser");
@@ -2567,14 +2710,73 @@ function createVoiceGateway(opts) {
     });
   });
 }
+
+// src/sse-event-log.ts
+var InMemoryEventLog = class {
+  buffers = /* @__PURE__ */ new Map();
+  nextIds = /* @__PURE__ */ new Map();
+  finalizeTimers = /* @__PURE__ */ new Map();
+  maxEvents;
+  ttlMs;
+  constructor(config = {}) {
+    this.maxEvents = config.maxEventsPerRun ?? 1024;
+    this.ttlMs = config.ttlMs ?? 5 * 6e4;
+  }
+  record(runId, event) {
+    const buf = this.buffers.get(runId) ?? [];
+    const nextId = (this.nextIds.get(runId) ?? 0) + 1;
+    const full = { id: nextId, recordedAt: Date.now(), ...event };
+    buf.push(full);
+    if (buf.length > this.maxEvents) buf.splice(0, buf.length - this.maxEvents);
+    this.buffers.set(runId, buf);
+    this.nextIds.set(runId, nextId);
+    return full;
+  }
+  since(runId, afterId) {
+    const buf = this.buffers.get(runId) ?? [];
+    return buf.filter((e) => e.id > afterId);
+  }
+  all(runId) {
+    return this.buffers.get(runId) ?? [];
+  }
+  finalize(runId) {
+    const existing = this.finalizeTimers.get(runId);
+    if (existing) clearTimeout(existing);
+    const t = setTimeout(() => this.drop(runId), this.ttlMs);
+    t.unref?.();
+    this.finalizeTimers.set(runId, t);
+  }
+  drop(runId) {
+    this.buffers.delete(runId);
+    this.nextIds.delete(runId);
+    const t = this.finalizeTimers.get(runId);
+    if (t) clearTimeout(t);
+    this.finalizeTimers.delete(runId);
+  }
+};
+function formatSSEEvent(ev) {
+  let out = `id: ${ev.id}
+`;
+  if (ev.event) out += `event: ${ev.event}
+`;
+  const data = typeof ev.payload === "string" ? ev.payload : JSON.stringify(ev.payload);
+  out += `data: ${data}
+
+`;
+  return out;
+}
+var defaultEventLog = new InMemoryEventLog();
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
+  InMemoryEventLog,
   MCPManager,
+  agentUIStream,
   buildMultiModalInput,
   createA2AServer,
   createAdminRouter,
   createAgentGateway,
   createAgentRouter,
+  createAgentUIStreamResponse,
   createBrowserGateway,
   createFileUploadMiddleware,
   createGatewayRouter,
@@ -2582,9 +2784,12 @@ function createVoiceGateway(opts) {
   createRbacMiddleware,
   createVisionGateway,
   createVoiceGateway,
+  defaultEventLog,
   errorHandler,
+  formatSSEEvent,
   generateAgentCard,
   generateMultiAgentCard,
   generateOpenAPISpec,
+  pipeAgentUIStreamToResponse,
   requestLogger
 });

package/dist/index.d.cts

@@ -280,6 +280,82 @@ interface OpenAPISpec {
 }
 declare function generateOpenAPISpec(routerOpts: RouterOptions, swaggerOpts?: SwaggerOptions): OpenAPISpec;
 
+/**
+ * Vercel AI SDK UI Message Stream adapter.
+ *
+ * Converts the chunks emitted by `agent.stream(...)` into the line-delimited
+ * JSON protocol consumed by Vercel's `useChat` / `createAgentUIStreamResponse`.
+ *
+ * Spec: https://ai-sdk.dev/docs/ai-sdk-ui/stream-protocol
+ */
+interface AgentUIStreamOptions {
+    /** sessionId forwarded to `agent.run` / `agent.stream`. */
+    sessionId?: string;
+    /** userId forwarded to `agent.run` / `agent.stream`. */
+    userId?: string;
+    /** Optional per-request API key override. */
+    apiKey?: string;
+    /** Optional abort signal to cancel the run. */
+    signal?: AbortSignal;
+}
+interface AgentLikeChunk {
+    type?: string;
+    text?: string;
+    delta?: string;
+    toolCallId?: string;
+    toolName?: string;
+    arguments?: unknown;
+    args?: unknown;
+    input?: unknown;
+    output?: unknown;
+    result?: unknown;
+    finishReason?: string;
+    usage?: {
+        promptTokens?: number;
+        completionTokens?: number;
+        totalTokens?: number;
+    };
+    error?: string | {
+        message?: string;
+    };
+    reasoning?: string;
+    thinking?: string;
+}
+interface AgentLike {
+    stream(input: string, opts?: {
+        sessionId?: string;
+        userId?: string;
+        apiKey?: string;
+        signal?: AbortSignal;
+    }): AsyncIterable<AgentLikeChunk>;
+}
+/**
+ * Render an `agent.stream()` source as a Vercel-compatible UI message stream.
+ *
+ * The returned `ReadableStream<Uint8Array>` can be wrapped in a `Response`
+ * (web / edge / fetch) or piped to a Node `ServerResponse`.
+ */
+declare function agentUIStream(agent: Agent | AgentLike, input: string, options?: AgentUIStreamOptions): ReadableStream<Uint8Array>;
+/**
+ * Create a `Response` whose body streams Vercel UI messages. Use as the return
+ * value of a Web / Edge / Next.js Route Handler.
+ *
+ * @example
+ * ```ts
+ * import { createAgentUIStreamResponse } from "@agentium/transport";
+ *
+ * export async function POST(req: Request) {
+ *   const { input, sessionId } = await req.json();
+ *   return createAgentUIStreamResponse(agent, input, { sessionId });
+ * }
+ * ```
+ */
+declare function createAgentUIStreamResponse(agent: Agent | AgentLike, input: string, options?: AgentUIStreamOptions): Response;
+/**
+ * Pipe a Vercel UI message stream into a Node `ServerResponse` (Express style).
+ */
+declare function pipeAgentUIStreamToResponse(agent: Agent | AgentLike, input: string, res: any, options?: AgentUIStreamOptions): Promise<void>;
+
 /**
  * Minimal interface for a BrowserAgent — avoids a hard dependency on @agentium/browser.
  * Any object that matches this shape (e.g. a real BrowserAgent) works.
@@ -390,4 +466,58 @@ interface VoiceGatewayOptions {
 }
 declare function createVoiceGateway(opts: VoiceGatewayOptions): void;
 
-export { type A2AServerOptions, type AdminRouterOptions, type BrowserGatewayOptions, type FileUploadOptions, type GatewayConfig, type GatewayOptions, type JwtConfig, MCPManager, type MCPServerEntry, type MCPServerSummary, type RbacConfig, type RemoteEndpoint, type RouterOptions, type SwaggerOptions, type VisionGatewayOptions, type VoiceGatewayOptions, buildMultiModalInput, createA2AServer, createAdminRouter, createAgentGateway, createAgentRouter, createBrowserGateway, createFileUploadMiddleware, createGatewayRouter, createJwtMiddleware, createRbacMiddleware, createVisionGateway, createVoiceGateway, errorHandler, generateAgentCard, generateMultiAgentCard, generateOpenAPISpec, requestLogger };
+/**
+ * Per-run in-memory ring buffer of SSE events. Each event gets a monotonically
+ * increasing numeric `id` so clients can pass `Last-Event-ID` on reconnect and
+ * replay missed events.
+ *
+ * Storage is intentionally in-process; for multi-instance deployments, plug a
+ * Redis-backed implementation behind the same interface.
+ */
+interface SSEEvent {
+    /** Monotonic numeric ID. */
+    id: number;
+    /** Optional `event:` line value. Default omitted. */
+    event?: string;
+    /** Payload (will be JSON-stringified into the `data:` field). */
+    payload: unknown;
+    /** Wall-clock time the event was recorded. */
+    recordedAt: number;
+}
+interface SSEEventLog {
+    /** Record a new event and return its assigned id. */
+    record(runId: string, event: Omit<SSEEvent, "id" | "recordedAt">): SSEEvent;
+    /** Return events for a run with id greater than `afterId`. */
+    since(runId: string, afterId: number): SSEEvent[];
+    /** All events for a run. */
+    all(runId: string): SSEEvent[];
+    /** Mark a run completed (frees buffers after `ttlMs`). */
+    finalize(runId: string): void;
+    /** Remove a run's buffer immediately. */
+    drop(runId: string): void;
+}
+interface InMemoryEventLogConfig {
+    /** Max events kept per run. Oldest evicted when exceeded. Default: 1024. */
+    maxEventsPerRun?: number;
+    /** Milliseconds after finalize before a run's buffer is dropped. Default: 300_000 (5min). */
+    ttlMs?: number;
+}
+declare class InMemoryEventLog implements SSEEventLog {
+    private buffers;
+    private nextIds;
+    private finalizeTimers;
+    private maxEvents;
+    private ttlMs;
+    constructor(config?: InMemoryEventLogConfig);
+    record(runId: string, event: Omit<SSEEvent, "id" | "recordedAt">): SSEEvent;
+    since(runId: string, afterId: number): SSEEvent[];
+    all(runId: string): SSEEvent[];
+    finalize(runId: string): void;
+    drop(runId: string): void;
+}
+/** Format an `SSEEvent` for the wire. */
+declare function formatSSEEvent(ev: SSEEvent): string;
+/** Process-wide default log so multiple endpoints can share one buffer. */
+declare const defaultEventLog: InMemoryEventLog;
+
+export { type A2AServerOptions, type AdminRouterOptions, type AgentUIStreamOptions, type BrowserGatewayOptions, type FileUploadOptions, type GatewayConfig, type GatewayOptions, InMemoryEventLog, type InMemoryEventLogConfig, type JwtConfig, MCPManager, type MCPServerEntry, type MCPServerSummary, type RbacConfig, type RemoteEndpoint, type RouterOptions, type SSEEvent, type SSEEventLog, type SwaggerOptions, type VisionGatewayOptions, type VoiceGatewayOptions, agentUIStream, buildMultiModalInput, createA2AServer, createAdminRouter, createAgentGateway, createAgentRouter, createAgentUIStreamResponse, createBrowserGateway, createFileUploadMiddleware, createGatewayRouter, createJwtMiddleware, createRbacMiddleware, createVisionGateway, createVoiceGateway, defaultEventLog, errorHandler, formatSSEEvent, generateAgentCard, generateMultiAgentCard, generateOpenAPISpec, pipeAgentUIStreamToResponse, requestLogger };

package/dist/index.d.ts

@@ -280,6 +280,82 @@ interface OpenAPISpec {
 }
 declare function generateOpenAPISpec(routerOpts: RouterOptions, swaggerOpts?: SwaggerOptions): OpenAPISpec;
 
+/**
+ * Vercel AI SDK UI Message Stream adapter.
+ *
+ * Converts the chunks emitted by `agent.stream(...)` into the line-delimited
+ * JSON protocol consumed by Vercel's `useChat` / `createAgentUIStreamResponse`.
+ *
+ * Spec: https://ai-sdk.dev/docs/ai-sdk-ui/stream-protocol
+ */
+interface AgentUIStreamOptions {
+    /** sessionId forwarded to `agent.run` / `agent.stream`. */
+    sessionId?: string;
+    /** userId forwarded to `agent.run` / `agent.stream`. */
+    userId?: string;
+    /** Optional per-request API key override. */
+    apiKey?: string;
+    /** Optional abort signal to cancel the run. */
+    signal?: AbortSignal;
+}
+interface AgentLikeChunk {
+    type?: string;
+    text?: string;
+    delta?: string;
+    toolCallId?: string;
+    toolName?: string;
+    arguments?: unknown;
+    args?: unknown;
+    input?: unknown;
+    output?: unknown;
+    result?: unknown;
+    finishReason?: string;
+    usage?: {
+        promptTokens?: number;
+        completionTokens?: number;
+        totalTokens?: number;
+    };
+    error?: string | {
+        message?: string;
+    };
+    reasoning?: string;
+    thinking?: string;
+}
+interface AgentLike {
+    stream(input: string, opts?: {
+        sessionId?: string;
+        userId?: string;
+        apiKey?: string;
+        signal?: AbortSignal;
+    }): AsyncIterable<AgentLikeChunk>;
+}
+/**
+ * Render an `agent.stream()` source as a Vercel-compatible UI message stream.
+ *
+ * The returned `ReadableStream<Uint8Array>` can be wrapped in a `Response`
+ * (web / edge / fetch) or piped to a Node `ServerResponse`.
+ */
+declare function agentUIStream(agent: Agent | AgentLike, input: string, options?: AgentUIStreamOptions): ReadableStream<Uint8Array>;
+/**
+ * Create a `Response` whose body streams Vercel UI messages. Use as the return
+ * value of a Web / Edge / Next.js Route Handler.
+ *
+ * @example
+ * ```ts
+ * import { createAgentUIStreamResponse } from "@agentium/transport";
+ *
+ * export async function POST(req: Request) {
+ *   const { input, sessionId } = await req.json();
+ *   return createAgentUIStreamResponse(agent, input, { sessionId });
+ * }
+ * ```
+ */
+declare function createAgentUIStreamResponse(agent: Agent | AgentLike, input: string, options?: AgentUIStreamOptions): Response;
+/**
+ * Pipe a Vercel UI message stream into a Node `ServerResponse` (Express style).
+ */
+declare function pipeAgentUIStreamToResponse(agent: Agent | AgentLike, input: string, res: any, options?: AgentUIStreamOptions): Promise<void>;
+
 /**
  * Minimal interface for a BrowserAgent — avoids a hard dependency on @agentium/browser.
  * Any object that matches this shape (e.g. a real BrowserAgent) works.
@@ -390,4 +466,58 @@ interface VoiceGatewayOptions {
 }
 declare function createVoiceGateway(opts: VoiceGatewayOptions): void;
 
-export { type A2AServerOptions, type AdminRouterOptions, type BrowserGatewayOptions, type FileUploadOptions, type GatewayConfig, type GatewayOptions, type JwtConfig, MCPManager, type MCPServerEntry, type MCPServerSummary, type RbacConfig, type RemoteEndpoint, type RouterOptions, type SwaggerOptions, type VisionGatewayOptions, type VoiceGatewayOptions, buildMultiModalInput, createA2AServer, createAdminRouter, createAgentGateway, createAgentRouter, createBrowserGateway, createFileUploadMiddleware, createGatewayRouter, createJwtMiddleware, createRbacMiddleware, createVisionGateway, createVoiceGateway, errorHandler, generateAgentCard, generateMultiAgentCard, generateOpenAPISpec, requestLogger };
+/**
+ * Per-run in-memory ring buffer of SSE events. Each event gets a monotonically
+ * increasing numeric `id` so clients can pass `Last-Event-ID` on reconnect and
+ * replay missed events.
+ *
+ * Storage is intentionally in-process; for multi-instance deployments, plug a
+ * Redis-backed implementation behind the same interface.
+ */
+interface SSEEvent {
+    /** Monotonic numeric ID. */
+    id: number;
+    /** Optional `event:` line value. Default omitted. */
+    event?: string;
+    /** Payload (will be JSON-stringified into the `data:` field). */
+    payload: unknown;
+    /** Wall-clock time the event was recorded. */
+    recordedAt: number;
+}
+interface SSEEventLog {
+    /** Record a new event and return its assigned id. */
+    record(runId: string, event: Omit<SSEEvent, "id" | "recordedAt">): SSEEvent;
+    /** Return events for a run with id greater than `afterId`. */
+    since(runId: string, afterId: number): SSEEvent[];
+    /** All events for a run. */
+    all(runId: string): SSEEvent[];
+    /** Mark a run completed (frees buffers after `ttlMs`). */
+    finalize(runId: string): void;
+    /** Remove a run's buffer immediately. */
+    drop(runId: string): void;
+}
+interface InMemoryEventLogConfig {
+    /** Max events kept per run. Oldest evicted when exceeded. Default: 1024. */
+    maxEventsPerRun?: number;
+    /** Milliseconds after finalize before a run's buffer is dropped. Default: 300_000 (5min). */
+    ttlMs?: number;
+}
+declare class InMemoryEventLog implements SSEEventLog {
+    private buffers;
+    private nextIds;
+    private finalizeTimers;
+    private maxEvents;
+    private ttlMs;
+    constructor(config?: InMemoryEventLogConfig);
+    record(runId: string, event: Omit<SSEEvent, "id" | "recordedAt">): SSEEvent;
+    since(runId: string, afterId: number): SSEEvent[];
+    all(runId: string): SSEEvent[];
+    finalize(runId: string): void;
+    drop(runId: string): void;
+}
+/** Format an `SSEEvent` for the wire. */
+declare function formatSSEEvent(ev: SSEEvent): string;
+/** Process-wide default log so multiple endpoints can share one buffer. */
+declare const defaultEventLog: InMemoryEventLog;
+
+export { type A2AServerOptions, type AdminRouterOptions, type AgentUIStreamOptions, type BrowserGatewayOptions, type FileUploadOptions, type GatewayConfig, type GatewayOptions, InMemoryEventLog, type InMemoryEventLogConfig, type JwtConfig, MCPManager, type MCPServerEntry, type MCPServerSummary, type RbacConfig, type RemoteEndpoint, type RouterOptions, type SSEEvent, type SSEEventLog, type SwaggerOptions, type VisionGatewayOptions, type VoiceGatewayOptions, agentUIStream, buildMultiModalInput, createA2AServer, createAdminRouter, createAgentGateway, createAgentRouter, createAgentUIStreamResponse, createBrowserGateway, createFileUploadMiddleware, createGatewayRouter, createJwtMiddleware, createRbacMiddleware, createVisionGateway, createVoiceGateway, defaultEventLog, errorHandler, formatSSEEvent, generateAgentCard, generateMultiAgentCard, generateOpenAPISpec, pipeAgentUIStreamToResponse, requestLogger };

package/dist/index.js

@@ -925,7 +925,12 @@ function routeMatches(actual, pattern) {
 
 // src/express/router-factory.ts
 import { createRequire as createRequire7 } from "module";
-import { classifyServables, collectToolkitTools, describeToolLibrary, registry as globalRegistry } from "@agentium/core";
+import {
+  classifyServables,
+  collectToolkitTools,
+  describeToolLibrary,
+  registry as globalRegistry
+} from "@agentium/core";
 
 // src/express/swagger.ts
 import { createRequire as createRequire6 } from "module";
@@ -1988,6 +1993,143 @@ function createAgentRouter(opts) {
   return router;
 }
 
+// src/express/ui-stream.ts
+function ssePayload(obj) {
+  return `data: ${JSON.stringify(obj)}
+
+`;
+}
+function agentUIStream(agent, input, options = {}) {
+  const encoder = new TextEncoder();
+  const messageId = `msg-${Date.now().toString(36)}-${Math.random().toString(36).slice(2, 8)}`;
+  let textPartId = 0;
+  return new ReadableStream({
+    async start(controller) {
+      try {
+        controller.enqueue(encoder.encode(ssePayload({ type: "start", messageId })));
+        let currentTextId = null;
+        for await (const chunk of agent.stream(input, {
+          sessionId: options.sessionId,
+          userId: options.userId,
+          apiKey: options.apiKey,
+          signal: options.signal
+        })) {
+          const t = chunk.type;
+          if (t === "text" || t === "text-delta") {
+            const delta = chunk.delta ?? chunk.text ?? "";
+            if (!delta) continue;
+            if (!currentTextId) {
+              currentTextId = `text-${++textPartId}`;
+              controller.enqueue(encoder.encode(ssePayload({ type: "text-start", id: currentTextId })));
+            }
+            controller.enqueue(encoder.encode(ssePayload({ type: "text-delta", id: currentTextId, textDelta: delta })));
+          } else if (t === "reasoning" || t === "thinking") {
+            const r = chunk.reasoning ?? chunk.thinking ?? chunk.delta ?? chunk.text ?? "";
+            if (r) {
+              controller.enqueue(
+                encoder.encode(ssePayload({ type: "reasoning-delta", id: messageId, reasoningDelta: r }))
+              );
+            }
+          } else if (t === "tool.call" || t === "tool-call" || t === "tool-input-start") {
+            if (currentTextId) {
+              controller.enqueue(encoder.encode(ssePayload({ type: "text-end", id: currentTextId })));
+              currentTextId = null;
+            }
+            controller.enqueue(
+              encoder.encode(
+                ssePayload({
+                  type: "tool-input-start",
+                  toolCallId: chunk.toolCallId,
+                  toolName: chunk.toolName
+                })
+              )
+            );
+            const args = chunk.arguments ?? chunk.args ?? chunk.input;
+            if (args !== void 0) {
+              controller.enqueue(
+                encoder.encode(
+                  ssePayload({
+                    type: "tool-input-available",
+                    toolCallId: chunk.toolCallId,
+                    toolName: chunk.toolName,
+                    input: args
+                  })
+                )
+              );
+            }
+          } else if (t === "tool.result" || t === "tool-result" || t === "tool-output-available") {
+            controller.enqueue(
+              encoder.encode(
+                ssePayload({
+                  type: "tool-output-available",
+                  toolCallId: chunk.toolCallId,
+                  output: chunk.output ?? chunk.result
+                })
+              )
+            );
+          } else if (t === "error") {
+            const errText = typeof chunk.error === "string" ? chunk.error : chunk.error?.message ?? "unknown error";
+            controller.enqueue(encoder.encode(ssePayload({ type: "error", errorText: errText })));
+          } else if (t === "finish" || t === "done" || t === "stream.end") {
+            if (currentTextId) {
+              controller.enqueue(encoder.encode(ssePayload({ type: "text-end", id: currentTextId })));
+              currentTextId = null;
+            }
+            controller.enqueue(
+              encoder.encode(
+                ssePayload({
+                  type: "finish-step",
+                  finishReason: chunk.finishReason ?? "stop",
+                  usage: chunk.usage
+                })
+              )
+            );
+          }
+        }
+        if (currentTextId) {
+          controller.enqueue(encoder.encode(ssePayload({ type: "text-end", id: currentTextId })));
+        }
+        controller.enqueue(encoder.encode(ssePayload({ type: "finish" })));
+        controller.enqueue(encoder.encode("data: [DONE]\n\n"));
+        controller.close();
+      } catch (err) {
+        controller.enqueue(encoder.encode(ssePayload({ type: "error", errorText: err?.message ?? String(err) })));
+        controller.close();
+      }
+    }
+  });
+}
+function createAgentUIStreamResponse(agent, input, options = {}) {
+  const body = agentUIStream(agent, input, options);
+  return new Response(body, {
+    headers: {
+      "Content-Type": "text/event-stream",
+      "Cache-Control": "no-cache",
+      Connection: "keep-alive",
+      "x-vercel-ai-ui-message-stream": "v1"
+    }
+  });
+}
+async function pipeAgentUIStreamToResponse(agent, input, res, options = {}) {
+  res.writeHead(200, {
+    "Content-Type": "text/event-stream",
+    "Cache-Control": "no-cache",
+    Connection: "keep-alive",
+    "x-vercel-ai-ui-message-stream": "v1"
+  });
+  const stream = agentUIStream(agent, input, options);
+  const reader = stream.getReader();
+  try {
+    while (true) {
+      const { done, value } = await reader.read();
+      if (done) break;
+      res.write(Buffer.from(value));
+    }
+  } finally {
+    res.end();
+  }
+}
+
 // src/socketio/browser-gateway.ts
 function createBrowserGateway(opts) {
   const ns = opts.io.of(opts.namespace ?? "/agentium-browser");
@@ -2102,7 +2244,12 @@ function createBrowserGateway(opts) {
 }
 
 // src/socketio/gateway.ts
-import { classifyServables as classifyServables2, collectToolkitTools as collectToolkitTools2, describeToolLibrary as describeToolLibrary2, registry as globalRegistry2 } from "@agentium/core";
+import {
+  classifyServables as classifyServables2,
+  collectToolkitTools as collectToolkitTools2,
+  describeToolLibrary as describeToolLibrary2,
+  registry as globalRegistry2
+} from "@agentium/core";
 function createSocketRateLimiter(maxPerMinute = 60) {
   return () => {
     let count = 0;
@@ -2520,13 +2667,72 @@ function createVoiceGateway(opts) {
     });
   });
 }
+
+// src/sse-event-log.ts
+var InMemoryEventLog = class {
+  buffers = /* @__PURE__ */ new Map();
+  nextIds = /* @__PURE__ */ new Map();
+  finalizeTimers = /* @__PURE__ */ new Map();
+  maxEvents;
+  ttlMs;
+  constructor(config = {}) {
+    this.maxEvents = config.maxEventsPerRun ?? 1024;
+    this.ttlMs = config.ttlMs ?? 5 * 6e4;
+  }
+  record(runId, event) {
+    const buf = this.buffers.get(runId) ?? [];
+    const nextId = (this.nextIds.get(runId) ?? 0) + 1;
+    const full = { id: nextId, recordedAt: Date.now(), ...event };
+    buf.push(full);
+    if (buf.length > this.maxEvents) buf.splice(0, buf.length - this.maxEvents);
+    this.buffers.set(runId, buf);
+    this.nextIds.set(runId, nextId);
+    return full;
+  }
+  since(runId, afterId) {
+    const buf = this.buffers.get(runId) ?? [];
+    return buf.filter((e) => e.id > afterId);
+  }
+  all(runId) {
+    return this.buffers.get(runId) ?? [];
+  }
+  finalize(runId) {
+    const existing = this.finalizeTimers.get(runId);
+    if (existing) clearTimeout(existing);
+    const t = setTimeout(() => this.drop(runId), this.ttlMs);
+    t.unref?.();
+    this.finalizeTimers.set(runId, t);
+  }
+  drop(runId) {
+    this.buffers.delete(runId);
+    this.nextIds.delete(runId);
+    const t = this.finalizeTimers.get(runId);
+    if (t) clearTimeout(t);
+    this.finalizeTimers.delete(runId);
+  }
+};
+function formatSSEEvent(ev) {
+  let out = `id: ${ev.id}
+`;
+  if (ev.event) out += `event: ${ev.event}
+`;
+  const data = typeof ev.payload === "string" ? ev.payload : JSON.stringify(ev.payload);
+  out += `data: ${data}
+
+`;
+  return out;
+}
+var defaultEventLog = new InMemoryEventLog();
 export {
+  InMemoryEventLog,
   MCPManager,
+  agentUIStream,
   buildMultiModalInput,
   createA2AServer,
   createAdminRouter,
   createAgentGateway,
   createAgentRouter,
+  createAgentUIStreamResponse,
   createBrowserGateway,
   createFileUploadMiddleware,
   createGatewayRouter,
@@ -2534,9 +2740,12 @@ export {
   createRbacMiddleware,
   createVisionGateway,
   createVoiceGateway,
+  defaultEventLog,
   errorHandler,
+  formatSSEEvent,
   generateAgentCard,
   generateMultiAgentCard,
   generateOpenAPISpec,
+  pipeAgentUIStreamToResponse,
   requestLogger
 };

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@agentium/transport",
-  "version": "1.0.0",
+  "version": "2.0.0",
   "description": "HTTP and WebSocket transport layer for Agentium agents",
   "license": "MIT",
   "repository": {
@@ -42,7 +42,7 @@
     "typescript": "^5.6.0"
   },
   "peerDependencies": {
-    "@agentium/core": "^1.0.0",
+    "@agentium/core": "^2.0.0",
     "@types/express": "^4.0.0 || ^5.0.0",
     "express": "^4.0.0 || ^5.0.0",
     "multer": ">=2.0.0",