@townco/ui 0.1.109 → 0.1.111

package/dist/gui/components/ToolCall.js

@@ -0,0 +1,226 @@
+import { jsxs as _jsxs, jsx as _jsx } from "react/jsx-runtime";
+import JsonView from "@uiw/react-json-view";
+import { AlertCircle, CheckSquare, ChevronDown, ChevronRight, CircleDot, Cloud, Edit, FileText, Globe, Image, Link, Search, Wrench, } from "lucide-react";
+import React, { useState } from "react";
+import { ChatLayout } from "./index.js";
+import { SubAgentDetails } from "./SubAgentDetails.js";
+import { useTheme } from "./ThemeProvider.js";
+/**
+ * Map of icon names to Lucide components
+ */
+const ICON_MAP = {
+    Globe: Globe,
+    Image: Image,
+    Link: Link,
+    Cloud: Cloud,
+    CheckSquare: CheckSquare,
+    Search: Search,
+    FileText: FileText,
+    Edit: Edit,
+    Wrench: Wrench,
+    CircleDot: CircleDot,
+};
+/**
+ * Tool call kind icons (using emoji for simplicity)
+ */
+const _kindIcons = {
+    read: "\u{1F4C4}",
+    edit: "\u{270F}\u{FE0F}",
+    delete: "\u{1F5D1}\u{FE0F}",
+    move: "\u{1F4E6}",
+    search: "\u{1F50D}",
+    execute: "\u{2699}\u{FE0F}",
+    think: "\u{1F4AD}",
+    fetch: "\u{1F310}",
+    switch_mode: "\u{1F501}",
+    other: "\u{1F527}",
+};
+/**
+ * ToolCall component - displays a single tool call with collapsible details
+ */
+export function ToolCall({ toolCall }) {
+    const [isExpanded, setIsExpanded] = useState(false);
+    const [isSubagentExpanded, setIsSubagentExpanded] = useState(false);
+    const { resolvedTheme } = useTheme();
+    // Detect TodoWrite tool and subagent
+    const isTodoWrite = toolCall.title === "todo_write";
+    const isSubagentCall = !!(toolCall.subagentPort && toolCall.subagentSessionId);
+    // Safely access ChatLayout context - will be undefined if not within ChatLayout
+    const layoutContext = React.useContext(ChatLayout.Context);
+    // Click handler: toggle sidepanel for TodoWrite, subagent details for subagents, expand for others
+    const handleHeaderClick = React.useCallback(() => {
+        if (isTodoWrite && layoutContext) {
+            // Toggle sidepanel - close if already open on todo tab, otherwise open
+            if (layoutContext.panelSize !== "hidden" &&
+                layoutContext.activeTab === "todo") {
+                layoutContext.setPanelSize("hidden");
+            }
+            else {
+                layoutContext.setPanelSize("small");
+                layoutContext.setActiveTab("todo");
+            }
+        }
+        else if (isSubagentCall) {
+            // Toggle subagent details
+            setIsSubagentExpanded(!isSubagentExpanded);
+        }
+        else {
+            // Normal expand/collapse
+            setIsExpanded(!isExpanded);
+        }
+    }, [
+        isTodoWrite,
+        layoutContext,
+        isExpanded,
+        isSubagentCall,
+        isSubagentExpanded,
+    ]);
+    // Determine which icon to show
+    const IconComponent = toolCall.icon && ICON_MAP[toolCall.icon]
+        ? ICON_MAP[toolCall.icon]
+        : CircleDot;
+    // Determine display name
+    const displayName = toolCall.prettyName || toolCall.title;
+    // Determine icon color based on status (especially for subagents)
+    const isSubagentRunning = isSubagentCall &&
+        (toolCall.status === "in_progress" || toolCall.status === "pending");
+    const isSubagentFailed = isSubagentCall && toolCall.status === "failed";
+    const iconColorClass = isSubagentCall
+        ? isSubagentFailed
+            ? "text-destructive"
+            : isSubagentRunning
+                ? "text-foreground animate-pulse"
+                : "text-green-500"
+        : "text-muted-foreground";
+    const statusTooltip = isSubagentCall
+        ? isSubagentFailed
+            ? "Sub-agent failed"
+            : isSubagentRunning
+                ? "Sub-agent running"
+                : "Sub-agent completed"
+        : undefined;
+    // Check if there's an error
+    const hasError = toolCall.status === "failed" || !!toolCall.error;
+    // Check if this is a preliminary (pending) tool call without full details yet
+    const isPreliminary = toolCall.status === "pending" &&
+        (!toolCall.rawInput || Object.keys(toolCall.rawInput).length === 0);
+    // JSON View style based on theme
+    const jsonStyle = {
+        fontSize: "11px",
+        backgroundColor: "transparent",
+        fontFamily: "inherit",
+        "--w-rjv-color": resolvedTheme === "dark" ? "#fafafa" : "#09090b",
+        "--w-rjv-key-string": resolvedTheme === "dark" ? "#fafafa" : "#09090b",
+        "--w-rjv-background-color": "transparent",
+        "--w-rjv-line-color": resolvedTheme === "dark" ? "#27272a" : "#e4e4e7",
+        "--w-rjv-arrow-color": resolvedTheme === "dark" ? "#a1a1aa" : "#71717a",
+        "--w-rjv-edit-color": resolvedTheme === "dark" ? "#fafafa" : "#09090b",
+        "--w-rjv-info-color": resolvedTheme === "dark" ? "#a1a1aa" : "#71717a",
+        "--w-rjv-update-color": resolvedTheme === "dark" ? "#fafafa" : "#09090b",
+        "--w-rjv-copied-color": resolvedTheme === "dark" ? "#fafafa" : "#09090b",
+        "--w-rjv-copied-success-color": resolvedTheme === "dark" ? "#22c55e" : "#16a34a",
+        "--w-rjv-curlybraces-color": resolvedTheme === "dark" ? "#a1a1aa" : "#71717a",
+        "--w-rjv-colon-color": resolvedTheme === "dark" ? "#a1a1aa" : "#71717a",
+        "--w-rjv-brackets-color": resolvedTheme === "dark" ? "#a1a1aa" : "#71717a",
+        "--w-rjv-quotes-color": resolvedTheme === "dark" ? "#a1a1aa" : "#71717a",
+        "--w-rjv-quotes-string-color": resolvedTheme === "dark" ? "#a1a1aa" : "#71717a",
+        "--w-rjv-type-string-color": resolvedTheme === "dark" ? "#22c55e" : "#16a34a",
+        "--w-rjv-type-int-color": resolvedTheme === "dark" ? "#f59e0b" : "#d97706",
+        "--w-rjv-type-float-color": resolvedTheme === "dark" ? "#f59e0b" : "#d97706",
+        "--w-rjv-type-bigint-color": resolvedTheme === "dark" ? "#f59e0b" : "#d97706",
+        "--w-rjv-type-boolean-color": resolvedTheme === "dark" ? "#3b82f6" : "#2563eb",
+        "--w-rjv-type-date-color": resolvedTheme === "dark" ? "#ec4899" : "#db2777",
+        "--w-rjv-type-url-color": resolvedTheme === "dark" ? "#3b82f6" : "#2563eb",
+        "--w-rjv-type-null-color": resolvedTheme === "dark" ? "#ef4444" : "#dc2626",
+        "--w-rjv-type-nan-color": resolvedTheme === "dark" ? "#ef4444" : "#dc2626",
+        "--w-rjv-type-undefined-color": resolvedTheme === "dark" ? "#ef4444" : "#dc2626",
+    };
+    // Preliminary tool calls show as simple light gray text without expansion
+    if (isPreliminary) {
+        return (_jsx("div", { className: "flex flex-col my-4", children: _jsxs("span", { className: "text-paragraph-sm text-muted-foreground/50", children: ["Invoking ", displayName] }) }));
+    }
+    return (_jsxs("div", { className: "flex flex-col my-4", children: [_jsxs("button", { type: "button", className: "flex flex-col items-start gap-0.5 cursor-pointer bg-transparent border-none p-0 text-left group w-fit", onClick: handleHeaderClick, "aria-expanded": isTodoWrite ? undefined : isExpanded, children: [_jsxs("div", { className: "flex items-center gap-1.5 text-[11px] font-medium text-muted-foreground", children: [_jsx("div", { className: iconColorClass, title: statusTooltip, children: _jsx(IconComponent, { className: "h-3 w-3" }) }), _jsx("span", { className: "text-paragraph-sm text-muted-foreground", children: displayName }), hasError && _jsx(AlertCircle, { className: "h-3 w-3 text-destructive" }), isTodoWrite ? (_jsx(ChevronRight, { className: "h-3 w-3 text-muted-foreground/70" })) : (_jsx(ChevronDown, { className: `h-3 w-3 text-muted-foreground/70 transition-transform duration-200 ${isExpanded ? "rotate-180" : ""}` }))] }), toolCall.subline && (_jsx("span", { className: "text-paragraph-sm text-muted-foreground/70 pl-4.5", children: toolCall.subline }))] }), !isTodoWrite && isSubagentCall && (_jsx("div", { className: "pl-4.5", children: _jsx(SubAgentDetails, { port: toolCall.subagentPort, sessionId: toolCall.subagentSessionId, parentStatus: toolCall.status, agentName: toolCall.rawInput?.agentName, query: toolCall.rawInput?.query, isExpanded: isSubagentExpanded, onExpandChange: setIsSubagentExpanded }) })), !isTodoWrite && !isSubagentCall && isExpanded && (_jsxs("div", { className: "mt-2 text-sm border border-border rounded-lg bg-card overflow-hidden w-full", children: [toolCall.rawInput &&
+                        Object.keys(toolCall.rawInput).length > 0 &&
+                        !toolCall.subagentPort && (_jsxs("div", { className: "p-3 border-b border-border", children: [_jsx("div", { className: "text-[10px] font-bold text-muted-foreground uppercase tracking-wider mb-1.5 font-sans", children: "Input" }), _jsx("div", { className: "text-[11px] font-mono text-foreground", children: _jsx(JsonView, { value: toolCall.rawInput, collapsed: false, displayDataTypes: false, displayObjectSize: false, enableClipboard: true, style: jsonStyle }) })] })), toolCall.locations && toolCall.locations.length > 0 && (_jsxs("div", { className: "p-3 border-b border-border", children: [_jsx("div", { className: "text-[10px] font-bold text-muted-foreground uppercase tracking-wider mb-1.5 font-sans", children: "Files" }), _jsx("ul", { className: "space-y-1", children: toolCall.locations.map((loc) => (_jsxs("li", { className: "font-mono text-[11px] text-foreground bg-muted px-1.5 py-0.5 rounded w-fit", children: [loc.path, loc.line !== null &&
+                                            loc.line !== undefined &&
+                                            `:${loc.line}`] }, `${loc.path}:${loc.line ?? ""}`))) })] })), (toolCall.content && toolCall.content.length > 0) ||
+                        toolCall.error ? (_jsxs("div", { className: "p-3 border-b border-border last:border-0", children: [_jsx("div", { className: "text-[10px] font-bold text-muted-foreground uppercase tracking-wider mb-1.5 font-sans", children: "Output" }), _jsxs("div", { className: "space-y-2 text-[11px] text-foreground", children: [toolCall.content?.map((block, idx) => {
+                                        // Generate a stable key based on content
+                                        const getBlockKey = () => {
+                                            if (block.type === "diff" && "path" in block) {
+                                                return `diff-${block.path}-${idx}`;
+                                            }
+                                            if (block.type === "terminal" && "terminalId" in block) {
+                                                return `terminal-${block.terminalId}`;
+                                            }
+                                            if (block.type === "text" && "text" in block) {
+                                                return `text-${block.text.substring(0, 20)}-${idx}`;
+                                            }
+                                            if (block.type === "content" && "content" in block) {
+                                                const innerContent = block.content;
+                                                return `content-${innerContent.text?.substring(0, 20)}-${idx}`;
+                                            }
+                                            return `block-${idx}`;
+                                        };
+                                        // Helper to render text content (with JSON parsing if applicable)
+                                        const renderTextContent = (text, key) => {
+                                            // Try to parse as JSON
+                                            try {
+                                                const parsed = JSON.parse(text);
+                                                // If it's an object or array, render with JsonView
+                                                if (typeof parsed === "object" && parsed !== null) {
+                                                    return (_jsx("div", { className: "text-[11px]", children: _jsx(JsonView, { value: parsed, collapsed: false, displayDataTypes: false, displayObjectSize: false, enableClipboard: true, style: jsonStyle }) }, key));
+                                                }
+                                            }
+                                            catch {
+                                                // Not valid JSON, render as plain text
+                                            }
+                                            // Render as plain text
+                                            return (_jsx("pre", { className: "whitespace-pre-wrap font-mono text-[11px] text-foreground overflow-x-auto", children: text }, key));
+                                        };
+                                        // Handle nested content blocks (ACP format)
+                                        if (block.type === "content" && "content" in block) {
+                                            const innerContent = block.content;
+                                            if (innerContent.type === "text" && innerContent.text) {
+                                                return renderTextContent(innerContent.text, getBlockKey());
+                                            }
+                                        }
+                                        // Handle direct text blocks
+                                        if (block.type === "text" && "text" in block) {
+                                            return renderTextContent(block.text, getBlockKey());
+                                        }
+                                        // Handle image blocks
+                                        if (block.type === "image") {
+                                            const alt = block.alt || "Generated image";
+                                            let imageSrc;
+                                            if ("data" in block) {
+                                                // Base64 encoded image
+                                                const mimeType = block.mimeType || "image/png";
+                                                imageSrc = `data:${mimeType};base64,${block.data}`;
+                                            }
+                                            else if ("url" in block) {
+                                                // URL or file path
+                                                imageSrc = block.url;
+                                            }
+                                            else {
+                                                return null;
+                                            }
+                                            return (_jsx("div", { className: "my-2", children: _jsx("img", { src: imageSrc, alt: alt, className: "max-w-full h-auto rounded-md border border-border" }) }, getBlockKey()));
+                                        }
+                                        // Handle diff blocks
+                                        if (block.type === "diff" &&
+                                            "path" in block &&
+                                            "oldText" in block &&
+                                            "newText" in block) {
+                                            return (_jsxs("div", { className: "border border-border rounded bg-card", children: [_jsxs("div", { className: "bg-muted px-2 py-1 text-[10px] font-mono text-muted-foreground border-b border-border", children: [block.path, "line" in block &&
+                                                                block.line !== null &&
+                                                                block.line !== undefined &&
+                                                                `:${block.line}`] }), _jsxs("div", { className: "p-2 font-mono text-[11px]", children: [_jsxs("div", { className: "text-red-500 dark:text-red-400", children: ["- ", block.oldText] }), _jsxs("div", { className: "text-green-500 dark:text-green-400", children: ["+ ", block.newText] })] })] }, getBlockKey()));
+                                        }
+                                        // Handle terminal blocks
+                                        if (block.type === "terminal" && "terminalId" in block) {
+                                            return (_jsxs("div", { className: "bg-neutral-900 text-neutral-100 p-2 rounded text-[11px] font-mono", children: ["Terminal: ", block.terminalId] }, getBlockKey()));
+                                        }
+                                        return null;
+                                    }), toolCall.error && (_jsxs("div", { className: "text-destructive font-mono text-[11px] mt-2", children: ["Error: ", toolCall.error] }))] })] })) : null, toolCall._meta?.truncationWarning && (_jsxs("div", { className: "mx-3 mt-3 mb-0 flex items-center gap-2 rounded-md bg-yellow-50 dark:bg-yellow-950/20 px-3 py-2 text-[11px] text-yellow-800 dark:text-yellow-200 border border-yellow-200 dark:border-yellow-900", children: [_jsx("span", { className: "text-yellow-600 dark:text-yellow-500", children: "\u26A0\uFE0F" }), _jsx("span", { children: toolCall._meta.truncationWarning })] })), (toolCall.tokenUsage || toolCall.startedAt) && (_jsxs("div", { className: "p-2 bg-muted/50 border-t border-border flex flex-wrap gap-4 text-[10px] text-muted-foreground font-sans", children: [toolCall.tokenUsage && (_jsxs("div", { className: "flex gap-3", children: [toolCall.tokenUsage.inputTokens !== undefined && (_jsxs("div", { children: [_jsx("span", { className: "uppercase tracking-wide font-semibold mr-1", children: "Input:" }), toolCall.tokenUsage.inputTokens.toLocaleString()] })), toolCall.tokenUsage.outputTokens !== undefined && (_jsxs("div", { children: [_jsx("span", { className: "uppercase tracking-wide font-semibold mr-1", children: "Output:" }), toolCall.tokenUsage.outputTokens.toLocaleString()] })), toolCall.tokenUsage.totalTokens !== undefined && (_jsxs("div", { children: [_jsx("span", { className: "uppercase tracking-wide font-semibold mr-1", children: "Total:" }), toolCall.tokenUsage.totalTokens.toLocaleString()] }))] })), toolCall.startedAt && (_jsxs("div", { className: "flex gap-3 ml-auto", children: [_jsxs("span", { children: ["Started: ", new Date(toolCall.startedAt).toLocaleTimeString()] }), toolCall.completedAt && (_jsxs("span", { children: ["Completed:", " ", new Date(toolCall.completedAt).toLocaleTimeString(), " (", Math.round((toolCall.completedAt - toolCall.startedAt) / 1000), "s)"] }))] }))] }))] }))] }));
+}

package/dist/gui/components/ToolCallGroup.d.ts

@@ -0,0 +1,8 @@
+import type { ToolCall as ToolCallType } from "../../core/schemas/tool-call.js";
+export interface ToolCallGroupProps {
+    toolCalls: ToolCallType[];
+}
+/**
+ * ToolCallGroup component - displays a group of parallel tool calls with collapsible details
+ */
+export declare function ToolCallGroup({ toolCalls }: ToolCallGroupProps): import("react/jsx-runtime").JSX.Element;

package/dist/gui/components/ToolCallGroup.js

@@ -0,0 +1,29 @@
+import { jsx as _jsx, jsxs as _jsxs } from "react/jsx-runtime";
+import { ChevronDown, ListVideo } from "lucide-react";
+import React, { useState } from "react";
+import { ToolCall } from "./ToolCall.js";
+/**
+ * ToolCallGroup component - displays a group of parallel tool calls with collapsible details
+ */
+export function ToolCallGroup({ toolCalls }) {
+    const [isExpanded, setIsExpanded] = useState(false);
+    // Calculate group status based on individual tool call statuses
+    const getGroupStatus = () => {
+        const statuses = toolCalls.map((tc) => tc.status);
+        if (statuses.some((s) => s === "failed"))
+            return "failed";
+        if (statuses.some((s) => s === "in_progress"))
+            return "in_progress";
+        if (statuses.every((s) => s === "completed"))
+            return "completed";
+        return "pending";
+    };
+    const groupStatus = getGroupStatus();
+    // Generate summary of tool names
+    const toolNames = toolCalls.map((tc) => tc.prettyName || tc.title);
+    const uniqueNames = [...new Set(toolNames)];
+    const summary = uniqueNames.length <= 2
+        ? uniqueNames.join(", ")
+        : `${uniqueNames.slice(0, 2).join(", ")} +${uniqueNames.length - 2} more`;
+    return (_jsxs("div", { className: "flex flex-col my-4", children: [_jsxs("button", { type: "button", className: "flex flex-col items-start gap-0.5 cursor-pointer bg-transparent border-none p-0 text-left group w-fit", onClick: () => setIsExpanded(!isExpanded), "aria-expanded": isExpanded, children: [_jsxs("div", { className: "flex items-center gap-1.5 text-[11px] font-medium text-muted-foreground", children: [_jsx("div", { className: "text-muted-foreground", children: _jsx(ListVideo, { className: "h-3 w-3" }) }), _jsx("span", { className: "text-paragraph-sm text-muted-foreground", children: "Parallel operation" }), _jsx("span", { className: "text-[10px] bg-muted px-1.5 py-0.5 rounded text-muted-foreground/70", children: toolCalls.length }), _jsx(ChevronDown, { className: `h-3 w-3 text-muted-foreground/70 transition-transform duration-200 ${isExpanded ? "rotate-180" : ""}` })] }), !isExpanded && (_jsx("span", { className: "text-paragraph-sm text-muted-foreground/70 pl-4.5", children: summary }))] }), isExpanded && (_jsx("div", { className: "mt-1", children: toolCalls.map((toolCall) => (_jsxs("div", { className: "flex items-start", children: [_jsx("div", { className: "w-2.5 h-4 border-l-2 border-b-2 border-border rounded-bl-[6px] mt-1 mr-0.5 shrink-0" }), _jsx("div", { className: "flex-1 -mt-2", children: _jsx(ToolCall, { toolCall: toolCall }) })] }, toolCall.id))) }))] }));
+}

package/dist/sdk/schemas/session.d.ts

@@ -637,6 +637,74 @@ export declare const SessionUpdate: z.ZodUnion<readonly [z.ZodObject<{
         rawOutput: z.ZodOptional<z.ZodRecord<z.ZodString, z.ZodUnknown>>;
     }, z.core.$strip>;
     messageId: z.ZodOptional<z.ZodString>;
+}, z.core.$strip>, z.ZodObject<{
+    sessionId: z.ZodString;
+    status: z.ZodOptional<z.ZodEnum<{
+        active: "active";
+        connected: "connected";
+        connecting: "connecting";
+        disconnected: "disconnected";
+        error: "error";
+        idle: "idle";
+        streaming: "streaming";
+    }>>;
+    message: z.ZodOptional<z.ZodObject<{
+        id: z.ZodString;
+        role: z.ZodEnum<{
+            assistant: "assistant";
+            system: "system";
+            tool: "tool";
+            user: "user";
+        }>;
+        content: z.ZodArray<z.ZodDiscriminatedUnion<[z.ZodObject<{
+            type: z.ZodLiteral<"text">;
+            text: z.ZodString;
+        }, z.core.$strip>, z.ZodObject<{
+            type: z.ZodLiteral<"image">;
+            url: z.ZodOptional<z.ZodString>;
+            source: z.ZodOptional<z.ZodObject<{
+                type: z.ZodLiteral<"base64">;
+                media_type: z.ZodEnum<{
+                    "image/gif": "image/gif";
+                    "image/jpeg": "image/jpeg";
+                    "image/png": "image/png";
+                    "image/webp": "image/webp";
+                }>;
+                data: z.ZodString;
+            }, z.core.$strip>>;
+        }, z.core.$strip>, z.ZodObject<{
+            type: z.ZodLiteral<"file">;
+            name: z.ZodString;
+            path: z.ZodOptional<z.ZodString>;
+            url: z.ZodOptional<z.ZodString>;
+            mimeType: z.ZodString;
+            size: z.ZodOptional<z.ZodNumber>;
+        }, z.core.$strip>, z.ZodObject<{
+            type: z.ZodLiteral<"tool_call">;
+            id: z.ZodString;
+            name: z.ZodString;
+            arguments: z.ZodRecord<z.ZodString, z.ZodUnknown>;
+        }, z.core.$strip>, z.ZodObject<{
+            type: z.ZodLiteral<"tool_result">;
+            callId: z.ZodString;
+            result: z.ZodUnknown;
+            error: z.ZodOptional<z.ZodString>;
+        }, z.core.$strip>], "type">>;
+        timestamp: z.ZodISODateTime;
+        metadata: z.ZodOptional<z.ZodRecord<z.ZodString, z.ZodUnknown>>;
+    }, z.core.$strip>>;
+    error: z.ZodOptional<z.ZodString>;
+    _meta: z.ZodOptional<z.ZodRecord<z.ZodString, z.ZodUnknown>>;
+    type: z.ZodLiteral<"sources">;
+    sources: z.ZodArray<z.ZodObject<{
+        id: z.ZodString;
+        url: z.ZodString;
+        title: z.ZodString;
+        snippet: z.ZodOptional<z.ZodString>;
+        favicon: z.ZodOptional<z.ZodString>;
+        toolCallId: z.ZodString;
+        sourceName: z.ZodOptional<z.ZodString>;
+    }, z.core.$strip>>;
 }, z.core.$strip>, z.ZodObject<{
     sessionId: z.ZodString;
     status: z.ZodOptional<z.ZodEnum<{

package/dist/sdk/schemas/session.js

@@ -1,6 +1,6 @@
 import { z } from "zod";
 import { ToolCallSchema, ToolCallUpdateSchema, } from "../../core/schemas/tool-call.js";
-import { Message } from "./message.js";
+import { CitationSource, Message } from "./message.js";
 /**
  * Session status
  */
@@ -83,6 +83,14 @@ const ToolOutputSessionUpdate = BaseSessionUpdate.extend({
     }),
     messageId: z.string().optional(),
 });
+/**
+ * Session update with sources (sessionUpdate: "sources")
+ * Sent during replay to restore persisted citation sources
+ */
+const SourcesSessionUpdate = BaseSessionUpdate.extend({
+    type: z.literal("sources"),
+    sources: z.array(CitationSource),
+});
 /**
  * Generic session update
  */
@@ -96,5 +104,6 @@ export const SessionUpdate = z.union([
     ToolCallSessionUpdate,
     ToolCallUpdateSessionUpdate,
     ToolOutputSessionUpdate,
+    SourcesSessionUpdate,
     GenericSessionUpdate,
 ]);

package/dist/sdk/transports/http.d.ts

@@ -22,6 +22,7 @@ export declare class HttpTransport implements Transport {
     private options;
     private isReceivingMessages;
     private isInReplayMode;
+    private pendingReplayUpdates;
     private agentInfo?;
     constructor(options: HttpTransportOptions);
     connect(): Promise<void>;