@axhub/acp 0.1.0

package/dist/components/assistant-ui/addons/host.mjs

@@ -0,0 +1,85 @@
+"use client";
+import { jsx as _jsx, jsxs as _jsxs } from "react/jsx-runtime";
+import { createContext, useContext, useEffect, useMemo, useRef, useState, } from "react";
+import { TooltipIconButton } from "../tooltip-icon-button.mjs";
+import { Sheet, SheetContent, SheetDescription, SheetHeader, SheetTitle, } from "../../ui/sheet.mjs";
+import { cn } from "../../../lib/utils.mjs";
+import { getAcpPlanKey, useLatestAcpPlan } from "./plan-data.mjs";
+import { getThreadAddons } from "./registry.mjs";
+const ThreadAddonContext = createContext(null);
+function useThreadAddonContext() {
+    const context = useContext(ThreadAddonContext);
+    if (!context) {
+        throw new Error("useThreadAddonContext must be used within ThreadAddonProvider.");
+    }
+    return context;
+}
+function useIsXlViewport() {
+    const [matches, setMatches] = useState(false);
+    useEffect(() => {
+        const media = window.matchMedia("(min-width: 1280px)");
+        const updateMatches = () => setMatches(media.matches);
+        updateMatches();
+        media.addEventListener("change", updateMatches);
+        return () => media.removeEventListener("change", updateMatches);
+    }, []);
+    return matches;
+}
+export function ThreadAddonProvider({ children }) {
+    var _a, _b, _c, _d;
+    const addons = useMemo(() => getThreadAddons(), []);
+    const defaultAddonId = (_b = (_a = addons[0]) === null || _a === void 0 ? void 0 : _a.id) !== null && _b !== void 0 ? _b : null;
+    const [activeAddonId, setActiveAddonId] = useState(defaultAddonId);
+    const [isOpen, setIsOpen] = useState(false);
+    const isDesktop = useIsXlViewport();
+    const plan = useLatestAcpPlan();
+    const planKey = getAcpPlanKey(plan);
+    const lastAutoOpenedPlanKeyRef = useRef("");
+    useEffect(() => {
+        if (!isDesktop || !planKey)
+            return;
+        if (lastAutoOpenedPlanKeyRef.current === planKey)
+            return;
+        lastAutoOpenedPlanKeyRef.current = planKey;
+        setActiveAddonId("plan");
+        setIsOpen(true);
+    }, [isDesktop, planKey]);
+    const activeAddon = (_d = (_c = addons.find((addon) => addon.id === activeAddonId)) !== null && _c !== void 0 ? _c : addons[0]) !== null && _d !== void 0 ? _d : null;
+    const value = useMemo(() => {
+        var _a;
+        return ({
+            addons,
+            activeAddon,
+            activeAddonId: (_a = activeAddon === null || activeAddon === void 0 ? void 0 : activeAddon.id) !== null && _a !== void 0 ? _a : null,
+            isDesktop,
+            isOpen,
+            closeAddon: () => setIsOpen(false),
+            toggleAddon: (addonId) => {
+                setActiveAddonId(addonId);
+                setIsOpen((currentOpen) => {
+                    if (currentOpen && (activeAddon === null || activeAddon === void 0 ? void 0 : activeAddon.id) === addonId)
+                        return false;
+                    return true;
+                });
+            },
+        });
+    }, [activeAddon, addons, isDesktop, isOpen]);
+    return (_jsx(ThreadAddonContext.Provider, { value: value, children: children }));
+}
+export function ThreadAddonToolbar() {
+    const { addons, activeAddonId, isOpen, toggleAddon } = useThreadAddonContext();
+    if (addons.length === 0)
+        return null;
+    return (_jsx("div", { "data-slot": "thread-addon-toolbar", className: "ml-auto flex items-center gap-1", children: addons.map((addon) => {
+            const Icon = addon.Icon;
+            const active = isOpen && activeAddonId === addon.id;
+            return (_jsx(TooltipIconButton, { tooltip: addon.label, "aria-pressed": active, className: cn("size-8 rounded-lg bg-background/80 text-muted-foreground backdrop-blur-sm [&_svg]:size-4", active && "bg-muted text-foreground"), onClick: () => toggleAddon(addon.id), children: _jsx(Icon, { className: "size-4" }) }, addon.id));
+        }) }));
+}
+export function ThreadAddonHost({ children }) {
+    var _a;
+    const { activeAddon, closeAddon, isDesktop, isOpen } = useThreadAddonContext();
+    const Panel = activeAddon === null || activeAddon === void 0 ? void 0 : activeAddon.Panel;
+    const sheetOpen = Boolean(Panel && isOpen && !isDesktop);
+    return (_jsxs("div", { "data-slot": "thread-addon-host", className: "flex h-full min-h-0 w-full", children: [_jsx("div", { className: "min-w-0 flex-1 overflow-hidden", children: children }), Panel && isDesktop && isOpen ? (_jsx("aside", { "data-slot": "thread-addon-desktop-panel", className: "hidden w-[22rem] shrink-0 bg-background px-3 pt-20 pb-4 xl:block", children: _jsx(Panel, { location: "desktop" }) })) : null, _jsx(Sheet, { open: sheetOpen, onOpenChange: (open) => !open && closeAddon(), children: _jsxs(SheetContent, { className: "w-[88vw] gap-0 p-0 sm:max-w-md", children: [_jsxs(SheetHeader, { className: "sr-only", children: [_jsx(SheetTitle, { children: (_a = activeAddon === null || activeAddon === void 0 ? void 0 : activeAddon.label) !== null && _a !== void 0 ? _a : "扩展面板" }), _jsx(SheetDescription, { children: "\u663E\u793A\u5F53\u524D\u5BF9\u8BDD\u7684\u6269\u5C55\u9762\u677F\u3002" })] }), _jsx("div", { className: "min-h-0 flex-1 overflow-y-auto p-4 pt-20", children: Panel ? _jsx(Panel, { location: "mobile" }) : null })] }) })] }));
+}

package/dist/components/assistant-ui/addons/image-generation-records-section.d.ts

@@ -0,0 +1,14 @@
+import { type FC } from "react";
+import { type ImageGenerationRecord } from "../../../tools/image-generation/shared";
+type ImageGenerationRecordsState = {
+    records: ImageGenerationRecord[];
+    loading: boolean;
+    error: string | null;
+    showSection: boolean;
+    loadRecords: () => Promise<void>;
+};
+export declare function useImageGenerationRecords(): ImageGenerationRecordsState;
+export declare const ImageGenerationRecordsSection: FC<{
+    compact?: boolean;
+} & ImageGenerationRecordsState>;
+export {};

package/dist/components/assistant-ui/addons/image-generation-records-section.mjs

@@ -0,0 +1,116 @@
+"use client";
+import { jsx as _jsx, jsxs as _jsxs } from "react/jsx-runtime";
+import { useAuiState } from "@assistant-ui/react";
+import { ImageIcon, RefreshCwIcon } from "lucide-react";
+import { useCallback, useEffect, useMemo, useRef, useState, } from "react";
+import { TooltipIconButton } from "../tooltip-icon-button.mjs";
+import { useAcpUiRuntimeContext } from "../../../lib/acp2aisdk/client-context.mjs";
+import { acpApiClient } from "../../../lib/api/client.mjs";
+import { cn } from "../../../lib/utils.mjs";
+import { ACP_PROVIDER_AGENT_DYNAMIC_TOOL_NAME, IMAGE_GENERATION_MCP_NAME, IMAGE_GENERATION_MCP_TOOL_TITLE, IMAGE_GENERATION_TOOL_NAME, isRecord, parseAcpDynamicToolArgs, parseImageGenerationOutput, } from "../../../tools/image-generation/shared.mjs";
+import { ImageGenerationRecordCard } from "../../../tools/image-generation/ui.mjs";
+const IMAGE_RECORD_LIMIT = 20;
+const IMAGE_RECORD_IDLE_POLL_MS = 12000;
+const IMAGE_RECORD_RUNNING_POLL_MS = 1500;
+function isMcpImageGenerationToolArgs(value) {
+    return (isRecord(value) &&
+        value.server === IMAGE_GENERATION_MCP_NAME &&
+        value.tool === IMAGE_GENERATION_TOOL_NAME);
+}
+function isImageGenerationToolCall(part) {
+    if (!isRecord(part) || part.type !== "tool-call")
+        return false;
+    if (part.toolName === IMAGE_GENERATION_MCP_TOOL_TITLE)
+        return true;
+    const dynamicArgs = parseAcpDynamicToolArgs(part.args);
+    return (part.toolName === ACP_PROVIDER_AGENT_DYNAMIC_TOOL_NAME &&
+        ((dynamicArgs === null || dynamicArgs === void 0 ? void 0 : dynamicArgs.toolName) === IMAGE_GENERATION_TOOL_NAME ||
+            (dynamicArgs === null || dynamicArgs === void 0 ? void 0 : dynamicArgs.toolName) === IMAGE_GENERATION_MCP_TOOL_TITLE ||
+            isMcpImageGenerationToolArgs(dynamicArgs === null || dynamicArgs === void 0 ? void 0 : dynamicArgs.args)));
+}
+function getImageGenerationRecordsKey(messages) {
+    const completed = new Set();
+    let running = 0;
+    for (const message of messages) {
+        if (!isRecord(message) || !Array.isArray(message.content))
+            continue;
+        for (const part of message.content) {
+            if (!isImageGenerationToolCall(part))
+                continue;
+            const result = parseImageGenerationOutput(part.result);
+            if (result === null || result === void 0 ? void 0 : result.recordId) {
+                completed.add(result.recordId);
+            }
+            else {
+                running += 1;
+            }
+        }
+    }
+    if (completed.size === 0 && running === 0)
+        return "";
+    return JSON.stringify({
+        completed: [...completed].sort(),
+        running,
+    });
+}
+export function useImageGenerationRecords() {
+    const { workspacePath } = useAcpUiRuntimeContext();
+    const isRunning = useAuiState((state) => state.thread.isRunning);
+    const messages = useAuiState((state) => state.thread.messages);
+    const threadId = useAuiState((state) => { var _a; return (_a = state.threadListItem.remoteId) !== null && _a !== void 0 ? _a : state.threads.mainThreadId; });
+    const recordsKey = useMemo(() => getImageGenerationRecordsKey([...messages]), [messages]);
+    const [records, setRecords] = useState([]);
+    const [state, setState] = useState("idle");
+    const [error, setError] = useState(null);
+    const loadRequestIdRef = useRef(0);
+    const loadRecords = useCallback(async () => {
+        const loadRequestId = loadRequestIdRef.current + 1;
+        loadRequestIdRef.current = loadRequestId;
+        if (!threadId) {
+            setRecords([]);
+            return;
+        }
+        setState((current) => (current === "idle" ? "loading" : current));
+        try {
+            const response = await acpApiClient.listImageGenerationRecords(workspacePath, threadId, IMAGE_RECORD_LIMIT);
+            if (loadRequestId !== loadRequestIdRef.current)
+                return;
+            setRecords(response.records);
+            setError(null);
+            setState("idle");
+        }
+        catch (loadError) {
+            if (loadRequestId !== loadRequestIdRef.current)
+                return;
+            setError(loadError instanceof Error ? loadError.message : String(loadError));
+            setState("error");
+        }
+    }, [threadId, workspacePath]);
+    useEffect(() => {
+        void loadRecords();
+    }, [loadRecords]);
+    useEffect(() => {
+        if (recordsKey)
+            void loadRecords();
+    }, [loadRecords, recordsKey]);
+    useEffect(() => {
+        const interval = window.setInterval(() => {
+            void loadRecords();
+        }, isRunning ? IMAGE_RECORD_RUNNING_POLL_MS : IMAGE_RECORD_IDLE_POLL_MS);
+        return () => window.clearInterval(interval);
+    }, [isRunning, loadRecords]);
+    const loading = state === "loading";
+    const showSection = records.length > 0;
+    return {
+        records,
+        loading,
+        error: state === "error" ? error : null,
+        showSection,
+        loadRecords,
+    };
+}
+export const ImageGenerationRecordsSection = ({ compact, records, loading, error, showSection, loadRecords }) => {
+    if (!showSection)
+        return null;
+    return (_jsxs("section", { "data-slot": "thread-image-generation-records", className: "flex min-h-0 flex-col gap-2", children: [_jsxs("div", { className: "flex items-center justify-between gap-2", children: [_jsxs("div", { className: "flex min-w-0 items-center gap-2", children: [_jsx(ImageIcon, { className: "size-4 shrink-0 text-muted-foreground" }), _jsx("h2", { className: "truncate font-medium text-muted-foreground text-sm", children: "\u56FE\u7247\u8BB0\u5F55" })] }), _jsx(TooltipIconButton, { tooltip: "\u5237\u65B0\u56FE\u7247\u8BB0\u5F55", size: "icon-xs", className: "size-7", onClick: () => void loadRecords(), children: _jsx(RefreshCwIcon, { className: cn("size-3.5", loading && "animate-spin") }) })] }), _jsx("div", { className: cn("min-h-0 overflow-y-auto pr-1", compact ? "max-h-72" : "max-h-[min(42vh,24rem)]"), children: records.length > 0 ? (_jsx("div", { className: "flex flex-col gap-2", children: records.map((record) => (_jsx(ImageGenerationRecordCard, { record: record, compact: true }, record.id))) })) : (_jsx("div", { className: "rounded-md border border-dashed px-3 py-4 text-muted-foreground text-sm", children: loading ? "正在加载图片记录" : "暂无图片记录" })) }), error ? (_jsx("p", { className: "rounded-md border border-destructive/30 bg-destructive/10 px-3 py-2 text-destructive text-xs", children: error })) : null] }));
+};

package/dist/components/assistant-ui/addons/output-artifacts-section.d.ts

@@ -0,0 +1,9 @@
+import type { FC } from "react";
+import { type AcpOutputArtifact } from "./output-artifacts";
+type OutputArtifactsSectionProps = {
+    artifacts: AcpOutputArtifact[];
+    compact?: boolean;
+};
+export declare function useAcpOutputArtifacts(): AcpOutputArtifact[];
+export declare const OutputArtifactsSection: FC<OutputArtifactsSectionProps>;
+export {};

package/dist/components/assistant-ui/addons/output-artifacts-section.mjs

@@ -0,0 +1,24 @@
+"use client";
+import { jsx as _jsx, jsxs as _jsxs } from "react/jsx-runtime";
+import { useAuiState } from "@assistant-ui/react";
+import { FileCodeIcon, PackageOpenIcon } from "lucide-react";
+import { useMemo } from "react";
+import { cn } from "../../../lib/utils.mjs";
+import { normalizeAcpOutputArtifacts, } from "./output-artifacts.mjs";
+export function useAcpOutputArtifacts() {
+    const messages = useAuiState((state) => state.thread.messages);
+    return useMemo(() => normalizeAcpOutputArtifacts([...messages]), [messages]);
+}
+function getArtifactTypeLabel(artifact) {
+    var _a;
+    return (_a = artifact.status) !== null && _a !== void 0 ? _a : "文件";
+}
+function OutputArtifactRow({ artifact }) {
+    const typeLabel = getArtifactTypeLabel(artifact);
+    return (_jsxs("li", { className: "flex min-w-0 items-start gap-2 rounded-md border border-border/70 bg-muted/15 px-2.5 py-2", children: [_jsx(FileCodeIcon, { className: "mt-0.5 size-3.5 shrink-0 text-muted-foreground" }), _jsx("div", { className: "min-w-0 flex-1", children: _jsxs("div", { className: "flex min-w-0 items-center gap-2", children: [_jsx("p", { className: "truncate font-medium text-sm leading-5", children: artifact.title }), _jsx("span", { className: "shrink-0 rounded-sm border bg-background px-1.5 py-0.5 text-[10px] text-muted-foreground leading-none", children: typeLabel })] }) })] }));
+}
+export const OutputArtifactsSection = ({ artifacts, compact, }) => {
+    if (artifacts.length === 0)
+        return null;
+    return (_jsxs("section", { "data-slot": "thread-output-artifacts", className: "flex min-h-0 flex-col gap-2", children: [_jsxs("div", { className: "flex min-w-0 items-center gap-2", children: [_jsx(PackageOpenIcon, { className: "size-4 shrink-0 text-muted-foreground" }), _jsx("h2", { className: "truncate font-medium text-muted-foreground text-sm", children: "\u4EA7\u7269" })] }), _jsx("ol", { className: cn("flex min-h-0 flex-col gap-2 overflow-y-auto pr-1", compact ? "max-h-56" : "max-h-[min(28vh,16rem)]"), children: artifacts.map((artifact) => (_jsx(OutputArtifactRow, { artifact: artifact }, artifact.id))) })] }));
+};

package/dist/components/assistant-ui/addons/output-artifacts.d.ts

@@ -0,0 +1,14 @@
+export type AcpOutputArtifactKind = "diff";
+export type AcpOutputArtifact = {
+    id: string;
+    kind: AcpOutputArtifactKind;
+    title: string;
+    detail?: string;
+    status?: "新增" | "修改";
+    path?: string;
+    oldText?: string | null;
+    newText?: string;
+    toolCallId: string;
+    updatedAt: number;
+};
+export declare function normalizeAcpOutputArtifacts(messages: readonly unknown[]): AcpOutputArtifact[];

package/dist/components/assistant-ui/addons/output-artifacts.mjs

@@ -0,0 +1,87 @@
+function isRecord(value) {
+    return Boolean(value && typeof value === "object" && !Array.isArray(value));
+}
+function normalizeString(value) {
+    return typeof value === "string" ? value.trim() : "";
+}
+function normalizeUpdatedAt(value) {
+    const updatedAt = Number(value);
+    return Number.isFinite(updatedAt) ? updatedAt : 0;
+}
+function recordsFromCollection(value) {
+    if (Array.isArray(value)) {
+        return value.filter((item) => isRecord(item));
+    }
+    if (!isRecord(value))
+        return [];
+    return Object.values(value).filter((item) => isRecord(item));
+}
+function getCustomMetadata(message) {
+    if (!isRecord(message))
+        return undefined;
+    const metadata = isRecord(message.metadata) ? message.metadata : undefined;
+    const custom = metadata && isRecord(metadata.custom) ? metadata.custom : undefined;
+    return custom;
+}
+function getRecordToolCallId(record, fallbackToolCallId) {
+    return normalizeString(record.toolCallId) || fallbackToolCallId;
+}
+function normalizeDiffArtifact(record, fallbackToolCallId) {
+    const type = normalizeString(record.type);
+    if (type && type !== "diff")
+        return undefined;
+    const toolCallId = getRecordToolCallId(record, fallbackToolCallId);
+    if (!toolCallId)
+        return undefined;
+    const path = normalizeString(record.path);
+    const status = record.oldText === null || record.oldText === undefined ? "新增" : "修改";
+    const title = path || "未命名文件";
+    return {
+        id: `diff:${toolCallId}:${path || title}`,
+        kind: "diff",
+        title,
+        detail: status,
+        status,
+        path: path || undefined,
+        oldText: typeof record.oldText === "string" || record.oldText === null
+            ? record.oldText
+            : undefined,
+        newText: typeof record.newText === "string" ? record.newText : undefined,
+        toolCallId,
+        updatedAt: normalizeUpdatedAt(record.updatedAt),
+    };
+}
+function collectArtifactsFromToolGroup(group, toolCallId) {
+    return recordsFromCollection(group.diffs).flatMap((record) => {
+        const artifact = normalizeDiffArtifact(record, toolCallId);
+        return artifact ? [artifact] : [];
+    });
+}
+function getArtifactDedupeKey(artifact) {
+    const identity = artifact.path || artifact.title;
+    return `${artifact.kind}:${artifact.toolCallId}:${identity || artifact.title}`;
+}
+export function normalizeAcpOutputArtifacts(messages) {
+    const byDedupeKey = new Map();
+    for (const message of messages) {
+        const custom = getCustomMetadata(message);
+        const acpToolContent = custom === null || custom === void 0 ? void 0 : custom.acpToolContent;
+        if (!isRecord(acpToolContent))
+            continue;
+        const byToolCallId = acpToolContent.byToolCallId;
+        if (!isRecord(byToolCallId))
+            continue;
+        for (const [toolCallId, group] of Object.entries(byToolCallId)) {
+            if (!isRecord(group))
+                continue;
+            for (const artifact of collectArtifactsFromToolGroup(group, toolCallId)) {
+                const dedupeKey = getArtifactDedupeKey(artifact);
+                const previous = byDedupeKey.get(dedupeKey);
+                if (!previous || artifact.updatedAt >= previous.updatedAt) {
+                    byDedupeKey.set(dedupeKey, artifact);
+                }
+            }
+        }
+    }
+    return [...byDedupeKey.values()].sort((left, right) => left.updatedAt - right.updatedAt);
+}

package/dist/components/assistant-ui/addons/plan-addon.d.ts

@@ -0,0 +1,2 @@
+import type { ThreadAddon } from "./types";
+export declare const planThreadAddon: ThreadAddon;

package/dist/components/assistant-ui/addons/plan-addon.mjs

@@ -0,0 +1,39 @@
+"use client";
+import { jsx as _jsx, jsxs as _jsxs } from "react/jsx-runtime";
+import { CheckCircle2Icon, CircleIcon, ListTodoIcon, ListXIcon, LoaderCircleIcon, } from "lucide-react";
+import { cn } from "../../../lib/utils.mjs";
+import { ImageGenerationRecordsSection, useImageGenerationRecords, } from "./image-generation-records-section.mjs";
+import { OutputArtifactsSection, useAcpOutputArtifacts, } from "./output-artifacts-section.mjs";
+import { useLatestAcpPlan } from "./plan-data.mjs";
+const statusIconMap = {
+    pending: CircleIcon,
+    in_progress: LoaderCircleIcon,
+    completed: CheckCircle2Icon,
+};
+const statusLabelMap = {
+    pending: "待处理",
+    in_progress: "进行中",
+    completed: "已完成",
+};
+const PlanAddonPanel = ({ location }) => {
+    const plan = useLatestAcpPlan();
+    const artifacts = useAcpOutputArtifacts();
+    const imageRecords = useImageGenerationRecords();
+    const compact = location === "mobile";
+    const hasPlan = plan.length > 0;
+    const hasArtifacts = artifacts.length > 0;
+    const isEmpty = !hasPlan && !hasArtifacts && !imageRecords.showSection;
+    return (_jsxs("div", { "data-slot": "thread-plan-addon", className: cn("flex h-full min-h-0 flex-col gap-5", compact && "px-1"), children: [hasPlan ? (_jsxs("section", { className: "flex min-h-0 flex-col gap-3", children: [_jsx("h2", { className: "font-medium text-muted-foreground text-sm", children: "\u4EFB\u52A1\u8FDB\u5EA6" }), _jsx("ol", { className: "flex max-h-[min(56vh,32rem)] flex-col gap-3 overflow-x-hidden overflow-y-auto pr-1", children: plan.map((entry) => {
+                            const Icon = statusIconMap[entry.status];
+                            const isRunning = entry.status === "in_progress";
+                            return (_jsxs("li", { className: "flex min-w-0 items-start gap-3", children: [_jsx("span", { "aria-hidden": "true", className: cn("mt-0.5 flex size-4 shrink-0 items-center justify-center overflow-hidden text-muted-foreground", isRunning && "text-foreground"), children: _jsx(Icon, { className: cn("size-4 shrink-0", entry.status === "completed" &&
+                                                "fill-muted-foreground text-background", isRunning && "animate-spin motion-reduce:animate-none") }) }), _jsxs("div", { className: "min-w-0 flex-1", children: [_jsx("p", { className: "wrap-break-word font-medium text-sm leading-5", children: entry.content }), _jsx("span", { className: "sr-only", children: statusLabelMap[entry.status] })] })] }, `${entry.status}-${entry.priority}-${entry.content}`));
+                        }) })] })) : null, _jsx(OutputArtifactsSection, { artifacts: artifacts, compact: compact }), _jsx(ImageGenerationRecordsSection, Object.assign({ compact: compact }, imageRecords)), isEmpty ? (_jsxs("div", { className: "flex flex-col items-center justify-center gap-2 rounded-lg border border-dashed px-4 py-8 text-center text-muted-foreground text-sm", children: [_jsx(ListXIcon, { className: "size-5" }), _jsx("p", { children: "\u6682\u65E0\u4EFB\u52A1\u3001\u4EA7\u7269\u548C\u5DE5\u5177\u8BB0\u5F55" })] })) : null] }));
+};
+export const planThreadAddon = {
+    id: "plan",
+    label: "任务",
+    Icon: ListTodoIcon,
+    Panel: PlanAddonPanel,
+    order: 10,
+};

package/dist/components/assistant-ui/addons/plan-data.d.ts

@@ -0,0 +1,5 @@
+import type { PlanEntry } from "@agentclientprotocol/sdk/dist/schema";
+export type AcpPlanEntry = Pick<PlanEntry, "content" | "priority" | "status">;
+export declare function normalizeAcpPlan(value: unknown): AcpPlanEntry[];
+export declare function useLatestAcpPlan(): AcpPlanEntry[];
+export declare function getAcpPlanKey(plan: readonly AcpPlanEntry[]): string;

package/dist/components/assistant-ui/addons/plan-data.mjs

@@ -0,0 +1,97 @@
+"use client";
+import { useAuiState } from "@assistant-ui/react";
+import { useMemo } from "react";
+const PLAN_PRIORITIES = new Set(["high", "medium", "low"]);
+const PLAN_STATUSES = new Set(["pending", "in_progress", "completed"]);
+function isRecord(value) {
+    return Boolean(value && typeof value === "object" && !Array.isArray(value));
+}
+export function normalizeAcpPlan(value) {
+    if (!Array.isArray(value))
+        return [];
+    return value.flatMap((entry) => {
+        if (!isRecord(entry))
+            return [];
+        const { content, priority, status } = entry;
+        if (typeof content !== "string")
+            return [];
+        if (typeof priority !== "string" || !PLAN_PRIORITIES.has(priority)) {
+            return [];
+        }
+        if (typeof status !== "string" || !PLAN_STATUSES.has(status)) {
+            return [];
+        }
+        return [
+            {
+                content,
+                priority: priority,
+                status: status,
+            },
+        ];
+    });
+}
+function getCustomMetadata(message) {
+    var _a;
+    return (_a = message.metadata) === null || _a === void 0 ? void 0 : _a.custom;
+}
+function normalizePlanId(value) {
+    return typeof value === "string" ? value.trim() : "";
+}
+function getLatestKeyedAcpPlan(custom) {
+    const acpPlans = custom === null || custom === void 0 ? void 0 : custom.acpPlans;
+    if (!isRecord(acpPlans))
+        return [];
+    const byId = isRecord(acpPlans.byId) ? acpPlans.byId : {};
+    const removed = isRecord(acpPlans.removed) ? acpPlans.removed : {};
+    const latestId = normalizePlanId(acpPlans.latestId);
+    const sortedCandidates = Object.entries(byId).sort(([, left], [, right]) => {
+        const leftUpdatedAt = isRecord(left) ? Number(left.updatedAt) : 0;
+        const rightUpdatedAt = isRecord(right) ? Number(right.updatedAt) : 0;
+        return rightUpdatedAt - leftUpdatedAt;
+    });
+    const candidates = latestId
+        ? [
+            [latestId, byId[latestId]],
+            ...sortedCandidates.filter(([id]) => id !== latestId),
+        ]
+        : sortedCandidates;
+    for (const [id, plan] of candidates) {
+        if (removed[id])
+            continue;
+        if (!isRecord(plan) || plan.removed)
+            continue;
+        if (plan.type !== "items")
+            continue;
+        const entries = normalizeAcpPlan(plan.entries);
+        if (entries.length > 0)
+            return entries;
+    }
+    return [];
+}
+export function useLatestAcpPlan() {
+    const messages = useAuiState((state) => state.thread.messages);
+    return useMemo(() => {
+        for (let index = messages.length - 1; index >= 0; index -= 1) {
+            const message = messages[index];
+            if (!message)
+                continue;
+            if (message.role === "user")
+                return [];
+            if (message.role !== "assistant")
+                continue;
+            const custom = getCustomMetadata(message);
+            const keyedPlan = getLatestKeyedAcpPlan(custom);
+            if (keyedPlan.length > 0)
+                return keyedPlan;
+            const plan = normalizeAcpPlan(custom === null || custom === void 0 ? void 0 : custom.acpPlan);
+            if (plan.length > 0)
+                return plan;
+        }
+        return [];
+    }, [messages]);
+}
+export function getAcpPlanKey(plan) {
+    if (plan.length === 0)
+        return "";
+    return JSON.stringify(plan);
+}

package/dist/components/assistant-ui/addons/registry.d.ts

@@ -0,0 +1,2 @@
+import type { ThreadAddon } from "./types";
+export declare function getThreadAddons(): ThreadAddon[];

package/dist/components/assistant-ui/addons/registry.mjs

@@ -0,0 +1,8 @@
+"use client";
+import { planThreadAddon } from "./plan-addon.mjs";
+const THREAD_ADDONS = [
+    planThreadAddon,
+];
+export function getThreadAddons() {
+    return [...THREAD_ADDONS].sort((left, right) => left.order - right.order);
+}

package/dist/components/assistant-ui/addons/types.d.ts

@@ -0,0 +1,13 @@
+import type { LucideIcon } from "lucide-react";
+import type { ComponentType } from "react";
+export type ThreadAddonPanelLocation = "desktop" | "mobile";
+export type ThreadAddonPanelProps = {
+    location: ThreadAddonPanelLocation;
+};
+export type ThreadAddon = {
+    id: string;
+    label: string;
+    Icon: LucideIcon;
+    Panel: ComponentType<ThreadAddonPanelProps>;
+    order: number;
+};

package/dist/components/assistant-ui/addons/types.mjs

@@ -0,0 +1 @@
+"use client";

package/dist/components/assistant-ui/attachment.d.ts

@@ -0,0 +1,7 @@
+import { type FC } from "react";
+export declare const UserMessageAttachments: FC;
+export declare const ComposerAttachments: FC;
+export type ComposerAddAttachmentProps = {
+    label?: string;
+};
+export declare const ComposerAddAttachment: FC<ComposerAddAttachmentProps>;

package/dist/components/assistant-ui/attachment.mjs

@@ -0,0 +1,88 @@
+"use client";
+import { jsx as _jsx, jsxs as _jsxs } from "react/jsx-runtime";
+import { AttachmentPrimitive, ComposerPrimitive, MessagePrimitive, useAui, useAuiState, } from "@assistant-ui/react";
+import { FileText, PlusIcon, XIcon } from "lucide-react";
+import { useEffect, useState } from "react";
+import { useShallow } from "zustand/shallow";
+import { TooltipIconButton } from "./tooltip-icon-button.mjs";
+import { Avatar, AvatarFallback, AvatarImage } from "../ui/avatar.mjs";
+import { Dialog, DialogContent, DialogTitle, DialogTrigger, } from "../ui/dialog.mjs";
+import { Tooltip, TooltipContent, TooltipTrigger, } from "../ui/tooltip.mjs";
+import { cn } from "../../lib/utils.mjs";
+const useFileSrc = (file) => {
+    const [src, setSrc] = useState(undefined);
+    useEffect(() => {
+        if (!file) {
+            setSrc(undefined);
+            return;
+        }
+        const objectUrl = URL.createObjectURL(file);
+        setSrc(objectUrl);
+        return () => {
+            URL.revokeObjectURL(objectUrl);
+        };
+    }, [file]);
+    return src;
+};
+const useAttachmentSrc = () => {
+    var _a;
+    const { file, src } = useAuiState(useShallow((s) => {
+        var _a, _b;
+        if (s.attachment.type !== "image")
+            return {};
+        if (s.attachment.file)
+            return { file: s.attachment.file };
+        const src = (_b = (_a = s.attachment.content) === null || _a === void 0 ? void 0 : _a.filter((c) => c.type === "image")[0]) === null || _b === void 0 ? void 0 : _b.image;
+        if (!src)
+            return {};
+        return { src };
+    }));
+    return (_a = useFileSrc(file)) !== null && _a !== void 0 ? _a : src;
+};
+const AttachmentPreview = ({ src }) => {
+    const [isLoaded, setIsLoaded] = useState(false);
+    return (_jsx("img", { src: src, alt: "\u9644\u4EF6\u9884\u89C8", className: cn("block h-auto max-h-[80vh] w-auto max-w-full object-contain", isLoaded
+            ? "aui-attachment-preview-image-loaded"
+            : "aui-attachment-preview-image-loading invisible"), onLoad: () => setIsLoaded(true) }));
+};
+const AttachmentPreviewDialog = ({ children }) => {
+    const src = useAttachmentSrc();
+    if (!src)
+        return children;
+    return (_jsxs(Dialog, { children: [_jsx(DialogTrigger, { className: "aui-attachment-preview-trigger cursor-pointer transition-colors hover:bg-accent/50", asChild: true, children: children }), _jsxs(DialogContent, { className: "aui-attachment-preview-dialog-content p-2 sm:max-w-3xl [&>button]:rounded-full [&>button]:bg-foreground/60 [&>button]:p-1 [&>button]:opacity-100 [&>button]:ring-0! [&_svg]:text-background [&>button]:hover:[&_svg]:text-destructive", children: [_jsx(DialogTitle, { className: "aui-sr-only sr-only", children: "\u56FE\u7247\u9644\u4EF6\u9884\u89C8" }), _jsx("div", { className: "aui-attachment-preview relative mx-auto flex max-h-[80dvh] w-full items-center justify-center overflow-hidden bg-background", children: _jsx(AttachmentPreview, { src: src }) })] })] }));
+};
+const AttachmentThumb = () => {
+    const src = useAttachmentSrc();
+    return (_jsxs(Avatar, { className: "aui-attachment-tile-avatar h-full w-full rounded-none", children: [_jsx(AvatarImage, { src: src, alt: "\u9644\u4EF6\u9884\u89C8", className: "aui-attachment-tile-image object-cover" }), _jsx(AvatarFallback, { children: _jsx(FileText, { className: "aui-attachment-tile-fallback-icon size-8 text-muted-foreground" }) })] }));
+};
+const AttachmentUI = () => {
+    const aui = useAui();
+    const isComposer = aui.attachment.source !== "message";
+    const isImage = useAuiState((s) => s.attachment.type === "image");
+    const typeLabel = useAuiState((s) => {
+        const type = s.attachment.type;
+        switch (type) {
+            case "image":
+                return "图片";
+            case "document":
+                return "文档";
+            case "file":
+                return "文件";
+            default:
+                return type;
+        }
+    });
+    return (_jsxs(Tooltip, { children: [_jsxs(AttachmentPrimitive.Root, { className: cn("aui-attachment-root relative", isImage && "aui-attachment-root-composer only:*:first:size-24"), children: [_jsx(AttachmentPreviewDialog, { children: _jsx(TooltipTrigger, { asChild: true, children: _jsx("button", { type: "button", className: "aui-attachment-tile size-14 cursor-pointer overflow-hidden rounded-[calc(var(--composer-radius)-var(--composer-padding))] border bg-muted p-0 transition-opacity hover:opacity-75", "aria-label": `${typeLabel}附件`, children: _jsx(AttachmentThumb, {}) }) }) }), isComposer && _jsx(AttachmentRemove, {})] }), _jsx(TooltipContent, { side: "top", children: _jsx(AttachmentPrimitive.Name, {}) })] }));
+};
+const AttachmentRemove = () => {
+    return (_jsx(AttachmentPrimitive.Remove, { asChild: true, children: _jsx(TooltipIconButton, { tooltip: "\u79FB\u9664\u6587\u4EF6", className: "aui-attachment-tile-remove absolute end-1.5 top-1.5 size-3.5 rounded-full bg-white text-muted-foreground opacity-100 shadow-sm hover:bg-white! [&_svg]:text-black hover:[&_svg]:text-destructive", side: "top", children: _jsx(XIcon, { className: "aui-attachment-remove-icon size-3 dark:stroke-[2.5px]" }) }) }));
+};
+export const UserMessageAttachments = () => {
+    return (_jsx("div", { className: "aui-user-message-attachments-end col-span-full col-start-1 row-start-1 flex w-full flex-row justify-end gap-2", children: _jsx(MessagePrimitive.Attachments, { children: () => _jsx(AttachmentUI, {}) }) }));
+};
+export const ComposerAttachments = () => {
+    return (_jsx("div", { className: "aui-composer-attachments flex w-full flex-row items-center gap-2 overflow-x-auto empty:hidden", children: _jsx(ComposerPrimitive.Attachments, { children: () => _jsx(AttachmentUI, {}) }) }));
+};
+export const ComposerAddAttachment = ({ label = "添加附件", }) => {
+    return (_jsx(ComposerPrimitive.AddAttachment, { asChild: true, children: _jsx(TooltipIconButton, { tooltip: label, side: "bottom", variant: "ghost", size: "icon", className: "aui-composer-add-attachment size-8 rounded-full p-1 font-semibold text-xs hover:bg-muted-foreground/15 dark:border-muted-foreground/15 dark:hover:bg-muted-foreground/30", "aria-label": label, children: _jsx(PlusIcon, { className: "aui-attachment-add-icon size-5 stroke-[1.5px]" }) }) }));
+};