@timbal-ai/timbal-react 0.2.2 → 0.4.0

package/dist/index.esm.js

@@ -1,5 +1,13 @@
 // src/runtime/provider.tsx
-import { useCallback, useEffect, useRef, useState } from "react";
+import {
+  createContext,
+  useCallback,
+  useContext,
+  useEffect,
+  useMemo,
+  useRef,
+  useState
+} from "react";
 import {
   useExternalStoreRuntime,
   AssistantRuntimeProvider
@@ -87,13 +95,506 @@ var fetchCurrentUser = async () => {
   }
 };
 
+// src/artifacts/types.ts
+function isArtifact(value) {
+  return typeof value === "object" && value !== null && !Array.isArray(value) && typeof value.type === "string";
+}
+
+// src/runtime/reducer.ts
+function createReducerState() {
+  return { parts: [], toolIndexById: /* @__PURE__ */ new Map() };
+}
+function reduceSseEvent(state, event) {
+  switch (event.type) {
+    case "DELTA":
+      return reduceDelta(state, event.item);
+    case "OUTPUT": {
+      const path = event.path;
+      const isNested = typeof path === "string" && path.includes(".");
+      if (!isNested) {
+        const errorMessage = readErrorMessage(event);
+        if (errorMessage) {
+          state.parts.push({ type: "text", text: `**Error:** ${errorMessage}` });
+          return true;
+        }
+      }
+      if (isNested) {
+        return reduceNestedOutput(
+          state,
+          path,
+          event.output
+        );
+      }
+      return reduceOutput(
+        state,
+        event.output
+      );
+    }
+    default:
+      return false;
+  }
+}
+function reduceDelta(state, item) {
+  if (!item) return false;
+  if (item.type === "text_delta" && typeof item.text_delta === "string") {
+    lastTextPart(state).text += item.text_delta;
+    return true;
+  }
+  if (item.type === "thinking_delta" && typeof item.thinking_delta === "string") {
+    lastThinkingPart(state).text += item.thinking_delta;
+    return true;
+  }
+  if (item.type === "tool_use") {
+    const toolCallId = item.id || `tool-${crypto.randomUUID()}`;
+    const inputStr = stringifyInput(item.input);
+    const part = {
+      type: "tool-call",
+      toolCallId,
+      toolName: item.name || "unknown",
+      argsText: inputStr,
+      status: "running"
+    };
+    state.parts.push(part);
+    state.toolIndexById.set(toolCallId, state.parts.length - 1);
+    return true;
+  }
+  if (item.type === "tool_use_delta") {
+    const idx = state.toolIndexById.get(item.id);
+    if (idx !== void 0 && typeof item.input_delta === "string") {
+      state.parts[idx].argsText += item.input_delta;
+      return true;
+    }
+  }
+  return false;
+}
+function reduceNestedOutput(state, path, output) {
+  if (!output || typeof output !== "object") return false;
+  if (!Array.isArray(output.content) && isArtifact(output)) {
+    const toolName = toolNameFromPath(path);
+    if (toolName && attachToolResult(state, { toolName, result: output })) {
+      return true;
+    }
+  }
+  return reduceOutput(state, output, { toolResultsOnly: true, allowOrphan: false });
+}
+function reduceOutput(state, output, options) {
+  if (!output) return false;
+  if (typeof output === "string") {
+    if (state.parts.length === 0) {
+      state.parts.push({ type: "text", text: output });
+      return true;
+    }
+    return false;
+  }
+  if (Array.isArray(output.content)) {
+    let changed = false;
+    const blocks = output.content;
+    for (const block of blocks) {
+      if (block.type === "tool_use") {
+        if (!options?.toolResultsOnly && recordToolUse(state, block)) {
+          changed = true;
+        }
+      } else if (block.type === "tool_result") {
+        if (recordToolResult(state, block, options)) changed = true;
+      } else if (!options?.toolResultsOnly) {
+        if (block.type === "text" && typeof block.text === "string" && !lastTextPart(state).text) {
+          lastTextPart(state).text = block.text;
+          changed = true;
+        } else if (block.type === "thinking" && typeof block.thinking === "string" && !lastThinkingPart(state).text) {
+          lastThinkingPart(state).text = block.thinking;
+          changed = true;
+        }
+      }
+    }
+    return changed;
+  }
+  if (state.parts.length === 0) {
+    state.parts.push({ type: "text", text: JSON.stringify(output) });
+    return true;
+  }
+  return false;
+}
+function recordToolUse(state, block) {
+  const id = block.id || `tool-${crypto.randomUUID()}`;
+  if (state.toolIndexById.has(id)) return false;
+  const inputStr = stringifyInput(block.input);
+  const part = {
+    type: "tool-call",
+    toolCallId: id,
+    toolName: block.name || "unknown",
+    argsText: inputStr,
+    status: "running"
+  };
+  state.parts.push(part);
+  state.toolIndexById.set(id, state.parts.length - 1);
+  return true;
+}
+function recordToolResult(state, block, options) {
+  const allowOrphan = options?.allowOrphan !== false;
+  const id = block.id || block.tool_use_id || "";
+  const { result, resultText } = parseToolResultContent(block.content);
+  const toolName = block.name || void 0;
+  if (id) {
+    const idx = state.toolIndexById.get(id);
+    if (idx !== void 0) {
+      const part2 = state.parts[idx];
+      part2.result = result;
+      if (resultText) part2.resultText = resultText;
+      part2.status = "complete";
+      return true;
+    }
+    if (!allowOrphan) return false;
+  }
+  if (!id && toolName && attachToolResult(state, {
+    toolName,
+    result,
+    resultText
+  })) {
+    return true;
+  }
+  if (!id || !allowOrphan) return false;
+  const part = {
+    type: "tool-call",
+    toolCallId: id,
+    toolName: toolName || "unknown",
+    argsText: "",
+    result,
+    resultText,
+    status: "complete"
+  };
+  state.parts.push(part);
+  state.toolIndexById.set(id, state.parts.length - 1);
+  return true;
+}
+function toolNameFromPath(path) {
+  const segment = path.split(".").pop();
+  return segment && segment !== "agent" && segment !== "llm" ? segment : null;
+}
+function attachToolResult(state, {
+  toolCallId,
+  toolName,
+  result,
+  resultText
+}) {
+  if (toolCallId) {
+    const idx = state.toolIndexById.get(toolCallId);
+    if (idx !== void 0) {
+      const part = state.parts[idx];
+      part.result = result;
+      if (resultText) part.resultText = resultText;
+      part.status = "complete";
+      return true;
+    }
+  }
+  if (toolName) {
+    for (let i = state.parts.length - 1; i >= 0; i--) {
+      const part = state.parts[i];
+      if (part.type === "tool-call" && part.toolName === toolName && part.result === void 0) {
+        part.result = result;
+        if (resultText) part.resultText = resultText;
+        part.status = "complete";
+        return true;
+      }
+    }
+  }
+  return false;
+}
+function parseToolResultContent(content) {
+  if (typeof content === "string") {
+    return { result: content, resultText: content };
+  }
+  if (!Array.isArray(content)) {
+    return { result: content };
+  }
+  const textChunks = [];
+  for (const item of content) {
+    if (typeof item === "string") {
+      textChunks.push(item);
+    } else if (item && typeof item === "object") {
+      const obj = item;
+      if (obj.type === "text" && typeof obj.text === "string") {
+        textChunks.push(obj.text);
+      } else if (obj.type === "thinking" && typeof obj.thinking === "string") {
+        textChunks.push(obj.thinking);
+      }
+    }
+  }
+  return {
+    result: content,
+    resultText: textChunks.length > 0 ? textChunks.join("\n") : void 0
+  };
+}
+function readErrorMessage(event) {
+  const status = event.status;
+  const isErrorStatus = status?.code === "error";
+  const error = event.error;
+  let type = null;
+  let message = null;
+  if (isErrorStatus && typeof status?.message === "string" && status.message.length > 0) {
+    message = status.message;
+  }
+  if (!message && typeof error === "string" && error.length > 0) {
+    message = error;
+  } else if (error && typeof error === "object") {
+    const obj = error;
+    if (typeof obj.type === "string" && obj.type.length > 0) {
+      type = obj.type;
+    }
+    if (!message && typeof obj.message === "string" && obj.message.length > 0) {
+      message = obj.message;
+    }
+  }
+  if (!message && !isErrorStatus) return null;
+  if (!message) return "The agent failed to generate a response.";
+  const compact = compactError(message);
+  return type ? `${type}: ${compact}` : compact;
+}
+var ERROR_MAX_CHARS = 480;
+function compactError(message) {
+  const trimmed = message.split(/\n\s*Traceback \(most recent call last\):/u)[0].trim();
+  if (trimmed.length <= ERROR_MAX_CHARS) return trimmed;
+  return `${trimmed.slice(0, ERROR_MAX_CHARS).trimEnd()}\u2026`;
+}
+function stringifyInput(input) {
+  if (input === void 0 || input === null) return "{}";
+  return typeof input === "string" ? input : JSON.stringify(input);
+}
+function lastTextPart(state) {
+  const last = state.parts[state.parts.length - 1];
+  if (last?.type === "text") return last;
+  const next = { type: "text", text: "" };
+  state.parts.push(next);
+  return next;
+}
+function lastThinkingPart(state) {
+  const last = state.parts[state.parts.length - 1];
+  if (last?.type === "thinking") return last;
+  const next = { type: "thinking", text: "" };
+  state.parts.push(next);
+  return next;
+}
+
+// src/runtime/attachments.ts
+async function extractAttachment(attachment) {
+  const file = attachment.file;
+  let src = null;
+  let contentType;
+  let name = attachment.name ?? file?.name;
+  const content = attachment.content;
+  if (content) {
+    for (const block of content) {
+      if (block.type === "image" && typeof block.image === "string") {
+        src = block.image;
+        if (typeof block.mimeType === "string") {
+          contentType = block.mimeType;
+        }
+        break;
+      }
+      if (block.type === "file" && typeof block.data === "string") {
+        src = block.data;
+        if (typeof block.mimeType === "string") contentType = block.mimeType;
+        break;
+      }
+    }
+  }
+  if (src === null && file) {
+    src = await fileToDataUrl(file);
+    if (!contentType) contentType = file.type || void 0;
+    if (!name) name = file.name;
+  }
+  if (!src) return null;
+  if (!contentType) contentType = mimeFromDataUrl(src);
+  const rawType = String(attachment.type ?? "file");
+  const type = rawType === "image" || rawType === "document" ? rawType : "file";
+  return {
+    id: attachment.id ?? crypto.randomUUID(),
+    type,
+    ...name !== void 0 ? { name } : {},
+    ...contentType !== void 0 ? { contentType } : {},
+    dataUrl: src
+  };
+}
+function fileToDataUrl(file) {
+  return new Promise((resolve, reject) => {
+    const reader = new FileReader();
+    reader.onload = () => resolve(reader.result);
+    reader.onerror = () => reject(reader.error);
+    reader.readAsDataURL(file);
+  });
+}
+function mimeFromDataUrl(dataUrl) {
+  const match = /^data:([^;,]+)[;,]/.exec(dataUrl);
+  return match?.[1];
+}
+function buildPromptBody({
+  input,
+  attachments,
+  parentId
+}) {
+  const context = { parent_id: parentId };
+  const files = attachments ?? [];
+  if (files.length === 0) {
+    return { prompt: input, context };
+  }
+  const parts = [];
+  if (input) parts.push({ type: "text", text: input });
+  for (const attachment of files) {
+    parts.push({ type: "file", file: attachment.dataUrl });
+  }
+  return { prompt: parts, context };
+}
+
+// src/runtime/upload-adapter.ts
+var DEFAULT_UPLOAD_ACCEPT = "image/*,application/pdf,text/*,.md,.json,.csv,.tsv,.xlsx,.docx";
+function createDefaultAttachmentAdapter({
+  baseUrl = "",
+  uploadUrl,
+  fetch: fetchFn = authFetch,
+  accept = DEFAULT_UPLOAD_ACCEPT
+} = {}) {
+  const base = baseUrl.replace(/\/$/, "");
+  const resolvedUploadUrl = uploadUrl ?? `${base}/files/upload`;
+  return {
+    accept,
+    async add({ file }) {
+      const isImage = file.type.startsWith("image/");
+      const pending = {
+        id: crypto.randomUUID(),
+        type: isImage ? "image" : "file",
+        name: file.name,
+        contentType: file.type || "application/octet-stream",
+        file,
+        status: { type: "requires-action", reason: "composer-send" }
+      };
+      return pending;
+    },
+    async send(attachment) {
+      const fd = new FormData();
+      fd.append("file", attachment.file);
+      const res = await fetchFn(resolvedUploadUrl, { method: "POST", body: fd });
+      if (!res.ok) {
+        const detail = await res.text().catch(() => "");
+        throw new Error(
+          `Attachment upload failed (${res.status}): ${detail || res.statusText}`
+        );
+      }
+      const remoteUrl = await readUploadedUrl(res);
+      const mime = attachment.contentType ?? "application/octet-stream";
+      const filename = attachment.name;
+      const complete = {
+        ...attachment,
+        status: { type: "complete" },
+        content: mime.startsWith("image/") ? [{ type: "image", image: remoteUrl, filename }] : [{ type: "file", data: remoteUrl, mimeType: mime, filename }]
+      };
+      return complete;
+    },
+    async remove() {
+    }
+  };
+}
+var createUploadAttachmentAdapter = createDefaultAttachmentAdapter;
+async function readUploadedUrl(res) {
+  const contentType = res.headers.get("content-type") ?? "";
+  if (contentType.includes("application/json")) {
+    const data = await res.json();
+    const candidate = data.url ?? data.signed_url ?? data.id;
+    if (typeof candidate === "string" && candidate.length > 0) {
+      return candidate;
+    }
+    throw new Error(
+      "Attachment upload response did not include a `url`, `signed_url`, or `id` field."
+    );
+  }
+  const text = (await res.text()).trim();
+  if (!text) {
+    throw new Error("Attachment upload response was empty.");
+  }
+  return text;
+}
+
+// src/runtime/resolve-attachments.ts
+function isAttachmentAdapter(value) {
+  return typeof value === "object" && value !== null && "accept" in value && typeof value.add === "function" && typeof value.send === "function" && typeof value.remove === "function";
+}
+function resolveAttachmentAdapter(attachments, options = {}) {
+  const baseUrl = options.baseUrl ?? "/api";
+  const legacyUploadUrl = options.uploadUrl;
+  const legacyAccept = options.accept;
+  if (attachments === null) return void 0;
+  const legacyEnables = legacyUploadUrl !== void 0 || legacyAccept !== void 0;
+  if (attachments === void 0) {
+    if (!legacyEnables) return void 0;
+    return createDefaultAttachmentAdapter({
+      baseUrl,
+      fetch: options.fetch,
+      uploadUrl: legacyUploadUrl,
+      accept: legacyAccept
+    });
+  }
+  if (attachments === true) {
+    return createDefaultAttachmentAdapter({
+      baseUrl,
+      fetch: options.fetch,
+      uploadUrl: legacyUploadUrl,
+      accept: legacyAccept
+    });
+  }
+  if (isAttachmentAdapter(attachments)) return attachments;
+  const config = attachments;
+  return createDefaultAttachmentAdapter({
+    baseUrl,
+    fetch: options.fetch,
+    uploadUrl: config.uploadUrl ?? legacyUploadUrl,
+    accept: config.accept ?? legacyAccept
+  });
+}
+
 // src/runtime/provider.tsx
 import { jsx } from "react/jsx-runtime";
-var convertMessage = (message) => ({
-  role: message.role,
-  content: message.content,
-  id: message.id
-});
+function projectAttachment(attachment) {
+  const filename = attachment.name ?? "attachment";
+  const mimeType = attachment.contentType ?? "application/octet-stream";
+  if (attachment.type === "image") {
+    return {
+      id: attachment.id,
+      type: "image",
+      name: filename,
+      contentType: mimeType,
+      status: { type: "complete" },
+      content: [{ type: "image", image: attachment.dataUrl, filename }]
+    };
+  }
+  return {
+    id: attachment.id,
+    type: attachment.type,
+    name: filename,
+    contentType: mimeType,
+    status: { type: "complete" },
+    content: [
+      { type: "file", data: attachment.dataUrl, mimeType, filename }
+    ]
+  };
+}
+var convertMessage = (message) => {
+  const content = message.content.map((part) => {
+    if (part.type === "text") return { type: "text", text: part.text };
+    if (part.type === "thinking") return { type: "reasoning", text: part.text };
+    return {
+      type: "tool-call",
+      toolCallId: part.toolCallId,
+      toolName: part.toolName,
+      argsText: part.argsText,
+      ...part.result !== void 0 ? { result: part.result } : {}
+    };
+  });
+  const attachments = message.attachments && message.attachments.length > 0 ? message.attachments.map(projectAttachment) : void 0;
+  return {
+    role: message.role,
+    content,
+    id: message.id,
+    ...attachments ? { attachments } : {}
+  };
+};
 function findParentId(messages, beforeIndex) {
   const slice = beforeIndex !== void 0 ? messages.slice(0, beforeIndex) : messages;
   for (let i = slice.length - 1; i >= 0; i--) {
@@ -101,18 +602,18 @@ function findParentId(messages, beforeIndex) {
   }
   return null;
 }
-function isTopLevelStart(event) {
-  return event.type === "START" && typeof event.run_id === "string" && typeof event.path === "string" && !event.path.includes(".");
-}
 function getTextFromMessage(message) {
   const part = message.content.find((c) => c.type === "text");
-  return part?.type === "text" ? part.text : null;
+  return part?.type === "text" ? part.text : "";
 }
-function TimbalRuntimeProvider({
+function getAttachmentsFromMessage(message) {
+  return message.attachments?.length ? message.attachments : void 0;
+}
+function useTimbalStream({
   workforceId,
-  children,
   baseUrl = "/api",
-  fetch: fetchFn
+  fetch: fetchFn,
+  debug = false
 }) {
   const [messages, setMessages] = useState([]);
   const [isRunning, setIsRunning] = useState(false);
@@ -122,40 +623,41 @@ function TimbalRuntimeProvider({
   useEffect(() => {
     fetchFnRef.current = fetchFn ?? authFetch;
   }, [fetchFn]);
+  const debugRef = useRef(debug);
+  useEffect(() => {
+    debugRef.current = debug;
+  }, [debug]);
   useEffect(() => {
     messagesRef.current = messages;
   }, [messages]);
   const streamAssistantResponse = useCallback(
-    async (input, userId, assistantId, parentId, signal) => {
-      const parts = [];
-      const toolIndexById = /* @__PURE__ */ new Map();
-      const lastTextPart = () => {
-        const last = parts[parts.length - 1];
-        if (last?.type === "text") return last;
-        const next = { type: "text", text: "" };
-        parts.push(next);
-        return next;
-      };
+    async (input, attachments, userId, assistantId, parentId, signal) => {
+      const state = createReducerState();
       const flush = () => {
         setMessages(
-          (prev) => prev.map((m) => m.id === assistantId ? { ...m, content: [...parts] } : m)
+          (prev) => prev.map(
+            (m) => m.id === assistantId ? { ...m, content: [...state.parts] } : m
+          )
         );
       };
       const stampRunId = (runId) => {
         setMessages(
-          (prev) => prev.map((m) => m.id === userId || m.id === assistantId ? { ...m, runId } : m)
+          (prev) => prev.map(
+            (m) => m.id === userId || m.id === assistantId ? { ...m, runId } : m
+          )
         );
       };
       try {
-        const res = await fetchFnRef.current(`${baseUrl}/workforce/${workforceId}/stream`, {
-          method: "POST",
-          headers: { "Content-Type": "application/json" },
-          body: JSON.stringify({
-            prompt: input,
-            context: { parent_id: parentId }
-          }),
-          signal
-        });
+        const body = buildPromptBody({ input, attachments, parentId });
+        const res = await fetchFnRef.current(
+          `${baseUrl}/workforce/${workforceId}/stream`,
+          {
+            method: "POST",
+            headers: { "Content-Type": "application/json" },
+            body: JSON.stringify(body),
+            signal
+          }
+        );
         if (!res.ok || !res.body) throw new Error(`Request failed: ${res.status}`);
         const reader = res.body.getReader();
         const decoder = new TextDecoder();
@@ -170,84 +672,34 @@ function TimbalRuntimeProvider({
           for (const line of lines) {
             const event = parseSSELine(line);
             if (!event) continue;
-            if (!capturedRunId && isTopLevelStart(event)) {
-              capturedRunId = event.run_id;
-              stampRunId(capturedRunId);
+            if (debugRef.current) {
+              console.debug("[timbal]", event.type, event);
             }
-            switch (event.type) {
-              case "DELTA": {
-                const item = event.item;
-                if (!item) break;
-                if (item.type === "text_delta" && typeof item.text_delta === "string") {
-                  lastTextPart().text += item.text_delta;
-                  flush();
-                } else if (item.type === "tool_use") {
-                  const toolCallId = item.id || `tool-${crypto.randomUUID()}`;
-                  const inputStr = typeof item.input === "string" ? item.input : JSON.stringify(item.input ?? {});
-                  parts.push({
-                    type: "tool-call",
-                    toolCallId,
-                    toolName: item.name || "unknown",
-                    argsText: inputStr
-                  });
-                  toolIndexById.set(toolCallId, parts.length - 1);
-                  flush();
-                } else if (item.type === "tool_use_delta") {
-                  const idx = toolIndexById.get(item.id);
-                  if (idx !== void 0 && typeof item.input_delta === "string") {
-                    parts[idx].argsText += item.input_delta;
-                    flush();
-                  }
-                }
-                break;
-              }
-              case "OUTPUT": {
-                const output = event.output;
-                if (!output) break;
-                if (typeof output === "object" && Array.isArray(output.content)) {
-                  for (const block of output.content) {
-                    if (block.type === "tool_use") {
-                      const id = block.id || `tool-${crypto.randomUUID()}`;
-                      const idx = toolIndexById.get(id);
-                      if (idx !== void 0) {
-                        parts[idx].result = "Tool executed";
-                      } else {
-                        const inputStr = typeof block.input === "string" ? block.input : JSON.stringify(block.input ?? {});
-                        parts.push({
-                          type: "tool-call",
-                          toolCallId: id,
-                          toolName: block.name || "unknown",
-                          argsText: inputStr,
-                          result: "Tool executed"
-                        });
-                        toolIndexById.set(id, parts.length - 1);
-                      }
-                    } else if (block.type === "text" && typeof block.text === "string" && !lastTextPart().text) {
-                      lastTextPart().text = block.text;
-                    }
-                  }
-                  flush();
-                } else if (parts.length === 0) {
-                  const text = typeof output === "string" ? output : JSON.stringify(output);
-                  parts.push({ type: "text", text });
-                  flush();
-                }
-                break;
+            if (!capturedRunId) {
+              const runId = readTopLevelStartRunId(event);
+              if (runId) {
+                capturedRunId = runId;
+                stampRunId(runId);
               }
             }
+            const changed = reduceSseEvent(state, event);
+            if (changed) flush();
           }
         }
         if (buffer.trim()) {
           const event = parseSSELine(buffer);
-          if (event?.type === "OUTPUT" && parts.length === 0 && event.output) {
-            const text = typeof event.output === "string" ? event.output : JSON.stringify(event.output);
-            parts.push({ type: "text", text });
-            flush();
+          if (event) {
+            if (debugRef.current) {
+              console.debug("[timbal]", event.type, event);
+            }
+            if (reduceSseEvent(state, event)) flush();
           }
         }
       } catch (err) {
         if (err.name !== "AbortError") {
-          if (parts.length === 0) parts.push({ type: "text", text: "Something went wrong." });
+          if (state.parts.length === 0) {
+            state.parts.push({ type: "text", text: "Something went wrong." });
+          }
           flush();
         }
       } finally {
@@ -257,44 +709,46 @@ function TimbalRuntimeProvider({
     },
     [workforceId, baseUrl]
   );
-  const onNew = useCallback(
-    async (message) => {
-      const textPart = message.content.find((c) => c.type === "text");
-      if (!textPart || textPart.type !== "text") return;
-      const input = textPart.text;
+  const send = useCallback(
+    async (input, options) => {
       const userId = crypto.randomUUID();
       const assistantId = crypto.randomUUID();
-      let base = messagesRef.current;
-      if (message.parentId !== null) {
-        const parentIdx = base.findIndex((m) => m.id === message.parentId);
-        if (parentIdx >= 0) {
-          base = base.slice(0, parentIdx + 1);
-        }
-      }
-      const parentId = findParentId(base);
+      const base = messagesRef.current;
+      const parentId = options?.parentId !== void 0 ? options.parentId : findParentId(base);
+      const userMessage = {
+        id: userId,
+        role: "user",
+        content: input ? [{ type: "text", text: input }] : [],
+        ...options?.attachments && options.attachments.length > 0 ? { attachments: options.attachments } : {}
+      };
       setMessages([
         ...base,
-        { id: userId, role: "user", content: [{ type: "text", text: input }] }
-      ]);
-      setIsRunning(true);
-      setMessages((prev) => [
-        ...prev,
+        userMessage,
         { id: assistantId, role: "assistant", content: [] }
       ]);
+      setIsRunning(true);
       const controller = new AbortController();
       abortRef.current = controller;
-      await streamAssistantResponse(input, userId, assistantId, parentId, controller.signal);
+      await streamAssistantResponse(
+        input,
+        options?.attachments,
+        userId,
+        assistantId,
+        parentId,
+        controller.signal
+      );
     },
     [streamAssistantResponse]
   );
-  const onReload = useCallback(
+  const reload = useCallback(
     async (messageId) => {
       const current = messagesRef.current;
       const idx = messageId ? current.findIndex((m) => m.id === messageId) : current.length - 2;
       const userMessage = idx >= 0 ? current[idx] : null;
       if (!userMessage || userMessage.role !== "user") return;
       const input = getTextFromMessage(userMessage);
-      if (!input) return;
+      const messageAttachments = getAttachmentsFromMessage(userMessage);
+      if (!input && !messageAttachments?.length) return;
       const assistantId = crypto.randomUUID();
       const parentId = findParentId(current, idx);
       setMessages((prev) => [
@@ -304,25 +758,138 @@ function TimbalRuntimeProvider({
       setIsRunning(true);
       const controller = new AbortController();
       abortRef.current = controller;
-      await streamAssistantResponse(input, userMessage.id, assistantId, parentId, controller.signal);
+      await streamAssistantResponse(
+        input,
+        messageAttachments,
+        userMessage.id,
+        assistantId,
+        parentId,
+        controller.signal
+      );
     },
     [streamAssistantResponse]
   );
-  const onCancel = useCallback(async () => {
+  const cancel = useCallback(() => {
+    abortRef.current?.abort();
+  }, []);
+  const clear = useCallback(() => {
     abortRef.current?.abort();
+    setMessages([]);
   }, []);
+  return useMemo(
+    () => ({ messages, isRunning, send, reload, cancel, clear }),
+    [messages, isRunning, send, reload, cancel, clear]
+  );
+}
+function readTopLevelStartRunId(event) {
+  if (event.type === "START" && typeof event.run_id === "string" && typeof event.path === "string" && !event.path.includes(".")) {
+    return event.run_id;
+  }
+  return null;
+}
+var TimbalStreamContext = createContext(null);
+function useTimbalRuntime() {
+  const ctx = useContext(TimbalStreamContext);
+  if (!ctx) {
+    throw new Error(
+      "useTimbalRuntime must be used inside a <TimbalRuntimeProvider>."
+    );
+  }
+  return ctx;
+}
+function TimbalRuntimeProvider({
+  workforceId,
+  children,
+  baseUrl = "/api",
+  fetch: fetchFn,
+  attachments,
+  attachmentsUploadUrl,
+  attachmentsAccept,
+  debug
+}) {
+  const stream = useTimbalStream({
+    workforceId,
+    baseUrl,
+    fetch: fetchFn,
+    debug
+  });
+  const attachmentAdapter = useMemo(
+    () => resolveAttachmentAdapter(attachments, {
+      baseUrl,
+      fetch: fetchFn,
+      uploadUrl: attachmentsUploadUrl,
+      accept: attachmentsAccept
+    }),
+    [attachments, attachmentsUploadUrl, attachmentsAccept, baseUrl, fetchFn]
+  );
+  const onNew = useCallback(
+    async (message) => {
+      const textPart = message.content.find((c) => c.type === "text");
+      const input = textPart && textPart.type === "text" ? textPart.text : "";
+      const auiAttachments = message.attachments;
+      const attachments2 = auiAttachments ? (await Promise.all(auiAttachments.map(extractAttachment))).filter((a) => a !== null) : [];
+      if (!input && attachments2.length === 0) return;
+      const parentId = message.parentId !== null && message.parentId !== void 0 ? findParentIdFromAuiParent(stream.messages, message.parentId) : void 0;
+      await stream.send(input, {
+        attachments: attachments2.length > 0 ? attachments2 : void 0,
+        ...parentId !== void 0 ? { parentId } : {}
+      });
+    },
+    [stream]
+  );
+  const onReload = useCallback(
+    async (messageId) => {
+      await stream.reload(messageId);
+    },
+    [stream]
+  );
+  const onCancel = useCallback(async () => {
+    stream.cancel();
+  }, [stream]);
   const runtime = useExternalStoreRuntime({
-    isRunning,
-    messages,
+    isRunning: stream.isRunning,
+    messages: stream.messages,
     convertMessage,
     onNew,
     onEdit: onNew,
     onReload,
-    onCancel
+    onCancel,
+    ...attachmentAdapter ? { adapters: { attachments: attachmentAdapter } } : {}
   });
-  return /* @__PURE__ */ jsx(AssistantRuntimeProvider, { runtime, children });
+  return /* @__PURE__ */ jsx(TimbalStreamContext.Provider, { value: stream, children: /* @__PURE__ */ jsx(AssistantRuntimeProvider, { runtime, children }) });
+}
+function findParentIdFromAuiParent(messages, auiParentId) {
+  const idx = messages.findIndex((m) => m.id === auiParentId);
+  if (idx < 0) return null;
+  return findParentId(messages.slice(0, idx + 1));
 }
 
+// src/index.ts
+import { parseSSELine as parseSSELine2 } from "@timbal-ai/timbal-sdk";
+
+// src/components/thread.tsx
+import {
+  ActionBarMorePrimitive,
+  ActionBarPrimitive,
+  AuiIf as AuiIf2,
+  ComposerPrimitive as ComposerPrimitive3,
+  ErrorPrimitive,
+  MessagePartPrimitive,
+  MessagePrimitive as MessagePrimitive2,
+  ThreadPrimitive,
+  useThread
+} from "@assistant-ui/react";
+import {
+  ArrowDownIcon,
+  CheckIcon as CheckIcon3,
+  CopyIcon as CopyIcon2,
+  DownloadIcon,
+  MoreHorizontalIcon,
+  PencilIcon,
+  RefreshCwIcon
+} from "lucide-react";
+import { motion as motion4 } from "motion/react";
+
 // src/components/attachment.tsx
 import { useEffect as useEffect2, useState as useState2 } from "react";
 import { XIcon as XIcon2, PlusIcon, FileText } from "lucide-react";
@@ -535,87 +1102,147 @@ function AvatarFallback({
 }
 
 // src/components/tooltip-icon-button.tsx
-import { forwardRef } from "react";
-import { Slottable } from "@radix-ui/react-slot";
+import { forwardRef as forwardRef2 } from "react";
 
-// src/ui/button.tsx
-import { cva } from "class-variance-authority";
+// src/ui/timbal-v2-button.tsx
+import * as React from "react";
 import { Slot } from "radix-ui";
-import { jsx as jsx5 } from "react/jsx-runtime";
-var buttonVariants = cva(
-  "inline-flex items-center justify-center gap-2 whitespace-nowrap rounded-md text-sm font-medium transition-all disabled:pointer-events-none disabled:opacity-50 [&_svg]:pointer-events-none [&_svg:not([class*='size-'])]:size-4 shrink-0 [&_svg]:shrink-0 outline-none focus-visible:border-ring focus-visible:ring-ring/50 focus-visible:ring-[3px] aria-invalid:ring-destructive/20 dark:aria-invalid:ring-destructive/40 aria-invalid:border-destructive",
-  {
-    variants: {
-      variant: {
-        default: "bg-primary text-primary-foreground hover:bg-primary/90",
-        destructive: "bg-destructive text-white hover:bg-destructive/90 focus-visible:ring-destructive/20 dark:focus-visible:ring-destructive/40 dark:bg-destructive/60",
-        outline: "border bg-background shadow-xs hover:bg-accent hover:text-accent-foreground dark:bg-input/30 dark:border-input dark:hover:bg-input/50",
-        secondary: "bg-secondary text-secondary-foreground hover:bg-secondary/80",
-        ghost: "hover:bg-accent hover:text-accent-foreground dark:hover:bg-accent/50",
-        link: "text-primary underline-offset-4 hover:underline"
-      },
-      size: {
-        default: "h-9 px-4 py-2 has-[>svg]:px-3",
-        xs: "h-6 gap-1 rounded-md px-2 text-xs has-[>svg]:px-1.5 [&_svg:not([class*='size-'])]:size-3",
-        sm: "h-8 rounded-md gap-1.5 px-3 has-[>svg]:px-2.5",
-        lg: "h-10 rounded-md px-6 has-[>svg]:px-4",
-        icon: "size-9",
-        "icon-xs": "size-6 rounded-md [&_svg:not([class*='size-'])]:size-3",
-        "icon-sm": "size-8",
-        "icon-lg": "size-10"
-      }
-    },
-    defaultVariants: {
-      variant: "default",
-      size: "default"
-    }
-  }
-);
-function Button({
-  className,
-  variant = "default",
-  size = "default",
+
+// src/ui/timbal-v2-button-tokens.ts
+var TIMBAL_V2_SIZE_HEIGHT = {
+  xs: "min-h-8 h-8",
+  sm: "min-h-9 h-9",
+  md: "min-h-10 h-10",
+  lg: "min-h-11 h-11"
+};
+var TIMBAL_V2_SIZE_ICON = {
+  xs: "min-h-8 min-w-8 size-8",
+  sm: "min-h-8 min-w-8 size-8",
+  md: "min-h-10 min-w-10 size-10",
+  lg: "min-h-11 min-w-11 size-11"
+};
+var TIMBAL_V2_SIZE_LABEL_PX = {
+  xs: "px-3",
+  sm: "px-4",
+  md: "px-5",
+  lg: "px-6"
+};
+var TIMBAL_V2_FILL = {
+  primary: "bg-gradient-to-b from-neutral-800 to-black group-hover/tbv2:from-neutral-700 group-hover/tbv2:to-neutral-900 group-active/tbv2:from-black group-active/tbv2:to-black dark:from-white dark:to-neutral-200 dark:group-hover/tbv2:from-white dark:group-hover/tbv2:to-neutral-100 dark:group-active/tbv2:from-neutral-200 dark:group-active/tbv2:to-neutral-400",
+  informative: "bg-blue-600 group-active/tbv2:[background-image:linear-gradient(to_top,rgba(0,0,0,0.08),transparent_55%)]",
+  destructive: "bg-gradient-to-b from-white to-neutral-50/75 group-hover/tbv2:from-red-50/90 group-hover/tbv2:to-red-100/70 group-active/tbv2:from-red-100/90 group-active/tbv2:to-red-200/65 dark:from-white/[0.05] dark:to-white/[0.025] dark:group-hover/tbv2:from-red-500/12 dark:group-hover/tbv2:to-red-500/8 dark:group-active/tbv2:from-red-500/20 dark:group-active/tbv2:to-red-500/12",
+  secondary: "bg-gradient-to-b from-white to-neutral-50/70 group-hover/tbv2:from-neutral-50/50 group-hover/tbv2:to-neutral-100/65 group-active/tbv2:from-neutral-100/70 group-active/tbv2:to-neutral-200/65 dark:from-white/[0.05] dark:to-white/[0.025] dark:group-hover/tbv2:from-white/[0.07] dark:group-hover/tbv2:to-white/[0.045] dark:group-active/tbv2:from-white/[0.10] dark:group-active/tbv2:to-white/[0.07]",
+  ghost: "bg-transparent group-hover/tbv2:bg-neutral-100/70 group-active/tbv2:bg-neutral-200/70 dark:group-hover/tbv2:bg-white/10 dark:group-active/tbv2:bg-white/15",
+  link: "bg-transparent"
+};
+var TIMBAL_V2_LABEL = {
+  primary: "text-white dark:text-neutral-900",
+  informative: "text-white",
+  destructive: "text-destructive dark:text-red-400",
+  secondary: "text-foreground",
+  ghost: "text-foreground",
+  link: "text-foreground underline decoration-black/25 underline-offset-2 group-hover/tbv2:decoration-black/45 dark:decoration-white/25 dark:group-hover/tbv2:decoration-white/45"
+};
+var TIMBAL_V2_BORDER = {
+  primary: "",
+  informative: "border border-white/15 dark:border-white/10",
+  destructive: "border border-destructive/45 dark:border-red-500/55",
+  secondary: "border border-neutral-200/80 dark:border-white/[0.08]",
+  ghost: "",
+  link: ""
+};
+var TIMBAL_V2_SHADOW = {
+  primary: "shadow-sm shadow-black/15 dark:shadow-black/40",
+  informative: "shadow-sm shadow-blue-900/20 dark:shadow-black/40",
+  destructive: "shadow-[0_1px_2px_-0.5px_rgba(0,0,0,0.05)] dark:shadow-[0_1px_3px_rgba(0,0,0,0.22)]",
+  secondary: "shadow-[0_1px_2px_-0.5px_rgba(0,0,0,0.05)] dark:shadow-[0_1px_3px_rgba(0,0,0,0.22)]",
+  ghost: "",
+  link: ""
+};
+var TIMBAL_V2_PILL_SURFACE = "bg-gradient-to-b from-white to-neutral-50/70 border border-neutral-200/80 shadow-[0_1px_2px_-0.5px_rgba(0,0,0,0.05)] dark:from-white/[0.05] dark:to-white/[0.025] dark:border-white/[0.08] dark:shadow-[0_1px_3px_rgba(0,0,0,0.22)]";
+var TIMBAL_V2_SECONDARY_CHROME = "bg-gradient-to-b from-white to-neutral-50/70 border border-neutral-200/80 shadow-[0_1px_2px_-0.5px_rgba(0,0,0,0.05)] transition-[background-color,box-shadow,border-color] duration-200 ease-in-out hover:from-neutral-50/40 hover:to-neutral-100/60 active:from-neutral-100/65 active:to-neutral-200/65 dark:from-white/[0.05] dark:to-white/[0.025] dark:border-white/[0.08] dark:shadow-[0_1px_3px_rgba(0,0,0,0.22)] dark:hover:from-white/[0.07] dark:hover:to-white/[0.045] dark:active:from-white/[0.10] dark:active:to-white/[0.07]";
+
+// src/ui/timbal-v2-button.tsx
+import { jsx as jsx5, jsxs as jsxs3 } from "react/jsx-runtime";
+var TimbalV2Button = React.forwardRef(function TimbalV2Button2({
+  variant = "secondary",
+  size = "sm",
+  isIconOnly = false,
+  isLoading = false,
+  fullWidth = false,
   asChild = false,
+  className,
+  disabled,
+  type = "button",
+  children,
   ...props
-}) {
+}, ref) {
+  const isDisabled = disabled || isLoading;
   const Comp = asChild ? Slot.Root : "button";
-  return /* @__PURE__ */ jsx5(
+  const sizeClass = isIconOnly ? TIMBAL_V2_SIZE_ICON[size] : TIMBAL_V2_SIZE_HEIGHT[size];
+  const radiusClass = variant === "link" || variant === "ghost" ? "rounded-md" : "rounded-full";
+  return /* @__PURE__ */ jsxs3(
     Comp,
     {
-      "data-slot": "button",
+      ref,
+      type: asChild ? void 0 : type,
+      disabled: asChild ? void 0 : isDisabled,
+      "aria-disabled": asChild && isDisabled ? true : void 0,
+      "data-slot": "timbal-v2-button",
       "data-variant": variant,
-      "data-size": size,
-      className: cn(buttonVariants({ variant, size, className })),
-      ...props
+      className: cn(
+        "group/tbv2 relative box-border inline-flex flex-col items-stretch overflow-hidden border-0 bg-transparent p-0 text-sm font-normal shadow-none transition duration-200 ease-in-out",
+        "focus-visible:outline-none focus-visible:ring-2 focus-visible:ring-neutral-400/60 focus-visible:ring-offset-1 focus-visible:ring-offset-background",
+        sizeClass,
+        radiusClass,
+        TIMBAL_V2_BORDER[variant],
+        TIMBAL_V2_SHADOW[variant],
+        fullWidth && "w-full",
+        isDisabled && "pointer-events-none opacity-50",
+        className
+      ),
+      ...props,
+      children: [
+        /* @__PURE__ */ jsx5(
+          "span",
+          {
+            "aria-hidden": true,
+            className: cn(
+              "pointer-events-none absolute inset-0 transition duration-200 ease-in-out",
+              TIMBAL_V2_FILL[variant]
+            )
+          }
+        ),
+        /* @__PURE__ */ jsx5(
+          "span",
+          {
+            className: cn(
+              "relative z-10 flex min-h-0 flex-1 items-center justify-center gap-1 leading-tight",
+              !isIconOnly && TIMBAL_V2_SIZE_LABEL_PX[size],
+              TIMBAL_V2_LABEL[variant]
+            ),
+            children: isLoading ? /* @__PURE__ */ jsx5("span", { className: "size-4 animate-spin rounded-full border-2 border-current border-t-transparent" }) : children
+          }
+        )
+      ]
     }
   );
-}
+});
 
 // src/components/tooltip-icon-button.tsx
-import { jsx as jsx6, jsxs as jsxs3 } from "react/jsx-runtime";
-var TooltipIconButton = forwardRef(({ children, tooltip, side = "bottom", className, ...rest }, ref) => {
-  return /* @__PURE__ */ jsxs3(Tooltip, { children: [
-    /* @__PURE__ */ jsx6(TooltipTrigger, { asChild: true, children: /* @__PURE__ */ jsxs3(
-      Button,
-      {
-        variant: "ghost",
-        size: "icon",
-        ...rest,
-        className: cn("aui-button-icon size-6 p-1", className),
-        ref,
-        children: [
-          /* @__PURE__ */ jsx6(Slottable, { children }),
-          /* @__PURE__ */ jsx6("span", { className: "aui-sr-only sr-only", children: tooltip })
-        ]
-      }
-    ) }),
+import { jsx as jsx6, jsxs as jsxs4 } from "react/jsx-runtime";
+var TooltipIconButton = forwardRef2(function TooltipIconButton2({ tooltip, side = "bottom", variant = "secondary", children, ...props }, ref) {
+  return /* @__PURE__ */ jsxs4(Tooltip, { children: [
+    /* @__PURE__ */ jsx6(TooltipTrigger, { asChild: true, children: /* @__PURE__ */ jsxs4(TimbalV2Button, { ref, variant, size: "sm", isIconOnly: true, ...props, children: [
+      children,
+      /* @__PURE__ */ jsx6("span", { className: "sr-only", children: tooltip })
+    ] }) }),
     /* @__PURE__ */ jsx6(TooltipContent, { side, children: tooltip })
   ] });
 });
-TooltipIconButton.displayName = "TooltipIconButton";
 
 // src/components/attachment.tsx
-import { jsx as jsx7, jsxs as jsxs4 } from "react/jsx-runtime";
+import { jsx as jsx7, jsxs as jsxs5 } from "react/jsx-runtime";
 var useFileSrc = (file) => {
   const [src, setSrc] = useState2(void 0);
   useEffect2(() => {
@@ -661,7 +1288,7 @@ var AttachmentPreview = ({ src }) => {
 var AttachmentPreviewDialog = ({ children }) => {
   const src = useAttachmentSrc();
   if (!src) return children;
-  return /* @__PURE__ */ jsxs4(Dialog, { children: [
+  return /* @__PURE__ */ jsxs5(Dialog, { children: [
     /* @__PURE__ */ jsx7(
       DialogTrigger,
       {
@@ -670,7 +1297,7 @@ var AttachmentPreviewDialog = ({ children }) => {
         children
       }
     ),
-    /* @__PURE__ */ jsxs4(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: [
+    /* @__PURE__ */ jsxs5(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: [
       /* @__PURE__ */ jsx7(DialogTitle, { className: "aui-sr-only sr-only", children: "Image Attachment Preview" }),
       /* @__PURE__ */ jsx7("div", { className: "aui-attachment-preview relative mx-auto flex max-h-[80dvh] w-full items-center justify-center overflow-hidden bg-background", children: /* @__PURE__ */ jsx7(AttachmentPreview, { src }) })
     ] })
@@ -679,7 +1306,7 @@ var AttachmentPreviewDialog = ({ children }) => {
 var AttachmentThumb = () => {
   const isImage = useAuiState((s) => s.attachment.type === "image");
   const src = useAttachmentSrc();
-  return /* @__PURE__ */ jsxs4(Avatar, { className: "aui-attachment-tile-avatar h-full w-full rounded-none", children: [
+  return /* @__PURE__ */ jsxs5(Avatar, { className: "aui-attachment-tile-avatar h-full w-full rounded-none", children: [
     /* @__PURE__ */ jsx7(
       AvatarImage,
       {
@@ -708,8 +1335,8 @@ var AttachmentUI = () => {
         throw new Error(`Unknown attachment type: ${type}`);
     }
   });
-  return /* @__PURE__ */ jsxs4(Tooltip, { children: [
-    /* @__PURE__ */ jsxs4(
+  return /* @__PURE__ */ jsxs5(Tooltip, { children: [
+    /* @__PURE__ */ jsxs5(
       AttachmentPrimitive.Root,
       {
         className: cn(
@@ -765,33 +1392,1222 @@ var ComposerAddAttachment = () => {
     {
       tooltip: "Add Attachment",
       side: "bottom",
-      variant: "ghost",
-      size: "icon",
-      className: "aui-composer-add-attachment size-8.5 rounded-full p-1 font-semibold text-xs hover:bg-muted-foreground/15 dark:border-muted-foreground/15 dark:hover:bg-muted-foreground/30",
+      variant: "secondary",
+      className: "aui-composer-add-attachment shrink-0 text-neutral-500 dark:text-muted-foreground",
       "aria-label": "Add Attachment",
-      children: /* @__PURE__ */ jsx7(PlusIcon, { className: "aui-attachment-add-icon size-5 stroke-[1.5px]" })
+      children: /* @__PURE__ */ jsx7(PlusIcon, { className: "aui-attachment-add-icon size-4 stroke-[1.5]" })
+    }
+  ) });
+};
+
+// src/components/markdown-text.tsx
+import "@assistant-ui/react-markdown/styles/dot.css";
+import "katex/dist/katex.min.css";
+import {
+  MarkdownTextPrimitive,
+  unstable_memoizeMarkdownComponents as memoizeMarkdownComponents,
+  useIsMarkdownCodeBlock
+} from "@assistant-ui/react-markdown";
+import remarkGfm from "remark-gfm";
+import remarkMath from "remark-math";
+import rehypeKatex from "rehype-katex";
+import { memo, useState as useState5 } from "react";
+import { CheckIcon as CheckIcon2, CopyIcon } from "lucide-react";
+
+// src/components/syntax-highlighter.tsx
+import { useEffect as useEffect3, useState as useState4 } from "react";
+import { createHighlighterCore } from "shiki/core";
+import { createJavaScriptRegexEngine } from "shiki/engine/javascript";
+
+// src/artifacts/registry.tsx
+import { createContext as createContext3, useContext as useContext3, useMemo as useMemo3 } from "react";
+
+// src/artifacts/chart-artifact.tsx
+import { useMemo as useMemo2 } from "react";
+
+// src/artifacts/artifact-card.tsx
+import { jsx as jsx8, jsxs as jsxs6 } from "react/jsx-runtime";
+var ArtifactCard = ({ title, kind, className, bodyClassName, toolbar, children }) => {
+  const hasHeader = Boolean(title || toolbar);
+  return /* @__PURE__ */ jsxs6(
+    "div",
+    {
+      className: cn(
+        "aui-artifact-root my-3 overflow-hidden rounded-xl border border-border/60 bg-background shadow-sm",
+        className
+      ),
+      "data-artifact-kind": kind,
+      children: [
+        hasHeader && /* @__PURE__ */ jsxs6("div", { className: "aui-artifact-header flex items-center gap-2 border-b border-border/40 bg-muted/30 px-3 py-1.5", children: [
+          title && /* @__PURE__ */ jsx8("span", { className: "aui-artifact-title flex-1 truncate text-xs font-semibold text-foreground/80", children: title }),
+          !title && /* @__PURE__ */ jsx8("span", { className: "flex-1" }),
+          toolbar
+        ] }),
+        /* @__PURE__ */ jsx8("div", { className: cn("aui-artifact-body", bodyClassName), children })
+      ]
+    }
+  );
+};
+
+// src/artifacts/chart-artifact.tsx
+import { Fragment, jsx as jsx9, jsxs as jsxs7 } from "react/jsx-runtime";
+var ChartArtifactView = ({
+  artifact
+}) => {
+  const { type: _t, chartType = "bar", data = [] } = artifact;
+  const xKey = artifact.xKey ?? inferXKey(data);
+  const dataKeys = useMemo2(() => {
+    if (Array.isArray(artifact.dataKey)) return artifact.dataKey;
+    if (typeof artifact.dataKey === "string") return [artifact.dataKey];
+    return inferDataKeys(data, xKey);
+  }, [artifact.dataKey, data, xKey]);
+  return /* @__PURE__ */ jsx9(ArtifactCard, { title: artifact.title, kind: "chart", children: /* @__PURE__ */ jsxs7("div", { className: "aui-artifact-chart p-3", children: [
+    /* @__PURE__ */ jsx9(
+      ChartSvg,
+      {
+        chartType,
+        data,
+        xKey,
+        dataKeys,
+        unit: artifact.unit
+      }
+    ),
+    dataKeys.length > 1 && /* @__PURE__ */ jsx9(Legend, { dataKeys })
+  ] }) });
+};
+var COLORS = [
+  "var(--primary, #6366f1)",
+  "#22c55e",
+  "#f59e0b",
+  "#ef4444",
+  "#06b6d4",
+  "#a855f7"
+];
+var W = 600;
+var H = 240;
+var PAD = { top: 12, right: 16, bottom: 28, left: 36 };
+var ChartSvg = ({ chartType, data, xKey, dataKeys, unit }) => {
+  if (data.length === 0 || dataKeys.length === 0) {
+    return /* @__PURE__ */ jsx9(EmptyState, {});
+  }
+  if (chartType === "pie") {
+    return /* @__PURE__ */ jsx9(PieChart, { data, xKey, dataKey: dataKeys[0] });
+  }
+  const innerW = W - PAD.left - PAD.right;
+  const innerH = H - PAD.top - PAD.bottom;
+  const all = dataKeys.flatMap((k) => data.map((d) => toNum(d[k])));
+  const maxV = Math.max(0, ...all);
+  const minV = Math.min(0, ...all);
+  const range = maxV - minV || 1;
+  const yScale = (v) => PAD.top + innerH - (v - minV) / range * innerH;
+  const xCount = data.length;
+  const xStep = xCount > 1 ? innerW / (xCount - 1) : innerW;
+  const xPos = (i) => chartType === "bar" ? PAD.left + innerW * (i + 0.5) / xCount : PAD.left + i * xStep;
+  const ticks = niceTicks(minV, maxV);
+  return /* @__PURE__ */ jsxs7(
+    "svg",
+    {
+      viewBox: `0 0 ${W} ${H}`,
+      className: "aui-artifact-chart-svg w-full",
+      role: "img",
+      "aria-label": "Chart",
+      children: [
+        ticks.map((t, i) => /* @__PURE__ */ jsxs7("g", { children: [
+          /* @__PURE__ */ jsx9(
+            "line",
+            {
+              x1: PAD.left,
+              x2: W - PAD.right,
+              y1: yScale(t),
+              y2: yScale(t),
+              stroke: "currentColor",
+              strokeOpacity: 0.08
+            }
+          ),
+          /* @__PURE__ */ jsx9(
+            "text",
+            {
+              x: PAD.left - 6,
+              y: yScale(t),
+              textAnchor: "end",
+              dominantBaseline: "middle",
+              className: "fill-muted-foreground text-[10px]",
+              children: formatTick(t, unit)
+            }
+          )
+        ] }, i)),
+        data.map((d, i) => /* @__PURE__ */ jsx9(
+          "text",
+          {
+            x: xPos(i),
+            y: H - PAD.bottom + 14,
+            textAnchor: "middle",
+            className: "fill-muted-foreground text-[10px]",
+            children: String(d[xKey] ?? i)
+          },
+          i
+        )),
+        chartType === "bar" && renderBars({ data, dataKeys, xCount, xPos, yScale, minV, innerW }),
+        chartType === "line" && dataKeys.map((k, ki) => /* @__PURE__ */ jsx9(
+          Polyline,
+          {
+            data,
+            dataKey: k,
+            xPos,
+            yScale,
+            color: COLORS[ki % COLORS.length]
+          },
+          k
+        )),
+        chartType === "area" && dataKeys.map((k, ki) => /* @__PURE__ */ jsx9(
+          Area,
+          {
+            data,
+            dataKey: k,
+            xPos,
+            yScale,
+            baseY: yScale(Math.max(0, minV)),
+            color: COLORS[ki % COLORS.length]
+          },
+          k
+        ))
+      ]
+    }
+  );
+};
+function renderBars(args) {
+  const { data, dataKeys, xCount, xPos, yScale, minV, innerW } = args;
+  const groupWidth = innerW / xCount * 0.7;
+  const barWidth = groupWidth / dataKeys.length;
+  const baseY = yScale(Math.max(0, minV));
+  return dataKeys.flatMap(
+    (k, ki) => data.map((d, i) => {
+      const v = toNum(d[k]);
+      const y = yScale(v);
+      const x = xPos(i) - groupWidth / 2 + ki * barWidth;
+      const top = Math.min(y, baseY);
+      const height = Math.abs(y - baseY);
+      return /* @__PURE__ */ jsx9(
+        "rect",
+        {
+          x,
+          y: top,
+          width: Math.max(1, barWidth - 2),
+          height: Math.max(1, height),
+          rx: 2,
+          fill: COLORS[ki % COLORS.length]
+        },
+        `${k}-${i}`
+      );
+    })
+  );
+}
+var Polyline = ({ data, dataKey, xPos, yScale, color }) => {
+  const points = data.map((d, i) => `${xPos(i)},${yScale(toNum(d[dataKey]))}`).join(" ");
+  return /* @__PURE__ */ jsx9(
+    "polyline",
+    {
+      points,
+      fill: "none",
+      stroke: color,
+      strokeWidth: 2,
+      strokeLinejoin: "round",
+      strokeLinecap: "round"
+    }
+  );
+};
+var Area = ({ data, dataKey, xPos, yScale, baseY, color }) => {
+  if (data.length === 0) return null;
+  const top = data.map((d, i) => `${xPos(i)},${yScale(toNum(d[dataKey]))}`).join(" ");
+  const path = `M ${xPos(0)},${baseY} L ${top} L ${xPos(data.length - 1)},${baseY} Z`;
+  return /* @__PURE__ */ jsxs7(Fragment, { children: [
+    /* @__PURE__ */ jsx9("path", { d: path, fill: color, fillOpacity: 0.18 }),
+    /* @__PURE__ */ jsx9(Polyline, { data, dataKey, xPos, yScale, color })
+  ] });
+};
+var PieChart = ({ data, xKey, dataKey }) => {
+  const cx = W / 2;
+  const cy = H / 2;
+  const r = Math.min(W, H) / 2 - 16;
+  const total = data.reduce((sum, d) => sum + toNum(d[dataKey]), 0) || 1;
+  let acc = 0;
+  return /* @__PURE__ */ jsx9(
+    "svg",
+    {
+      viewBox: `0 0 ${W} ${H}`,
+      className: "aui-artifact-chart-svg w-full",
+      role: "img",
+      "aria-label": "Pie chart",
+      children: data.map((d, i) => {
+        const value = toNum(d[dataKey]);
+        const start = acc / total * Math.PI * 2;
+        acc += value;
+        const end = acc / total * Math.PI * 2;
+        return /* @__PURE__ */ jsx9(
+          PieSlice,
+          {
+            cx,
+            cy,
+            r,
+            start,
+            end,
+            color: COLORS[i % COLORS.length],
+            label: String(d[xKey] ?? i)
+          },
+          i
+        );
+      })
+    }
+  );
+};
+var PieSlice = ({ cx, cy, r, start, end, color, label }) => {
+  const x1 = cx + Math.sin(start) * r;
+  const y1 = cy - Math.cos(start) * r;
+  const x2 = cx + Math.sin(end) * r;
+  const y2 = cy - Math.cos(end) * r;
+  const large = end - start > Math.PI ? 1 : 0;
+  const path = `M ${cx} ${cy} L ${x1} ${y1} A ${r} ${r} 0 ${large} 1 ${x2} ${y2} Z`;
+  const mid = (start + end) / 2;
+  const lx = cx + Math.sin(mid) * (r * 0.65);
+  const ly = cy - Math.cos(mid) * (r * 0.65);
+  return /* @__PURE__ */ jsxs7("g", { children: [
+    /* @__PURE__ */ jsx9("path", { d: path, fill: color, stroke: "var(--background, #fff)", strokeWidth: 1 }),
+    /* @__PURE__ */ jsx9(
+      "text",
+      {
+        x: lx,
+        y: ly,
+        textAnchor: "middle",
+        dominantBaseline: "middle",
+        className: "fill-white text-[10px] font-semibold",
+        children: label
+      }
+    )
+  ] });
+};
+var Legend = ({ dataKeys }) => /* @__PURE__ */ jsx9("div", { className: "aui-artifact-chart-legend mt-2 flex flex-wrap items-center gap-x-3 gap-y-1 text-xs text-muted-foreground", children: dataKeys.map((k, i) => /* @__PURE__ */ jsxs7("span", { className: "inline-flex items-center gap-1.5", children: [
+  /* @__PURE__ */ jsx9(
+    "span",
+    {
+      className: "inline-block size-2 rounded-sm",
+      style: { background: COLORS[i % COLORS.length] },
+      "aria-hidden": true
+    }
+  ),
+  k
+] }, k)) });
+var EmptyState = () => /* @__PURE__ */ jsx9("div", { className: "aui-artifact-chart-empty flex h-32 items-center justify-center text-xs text-muted-foreground", children: "No data" });
+function inferXKey(data) {
+  if (data.length === 0) return "x";
+  for (const k of Object.keys(data[0])) {
+    if (typeof data[0][k] !== "number") return k;
+  }
+  return Object.keys(data[0])[0] ?? "x";
+}
+function inferDataKeys(data, xKey) {
+  if (data.length === 0) return [];
+  return Object.keys(data[0]).filter(
+    (k) => k !== xKey && typeof data[0][k] === "number"
+  );
+}
+function toNum(value) {
+  const n = typeof value === "number" ? value : Number(value);
+  return Number.isFinite(n) ? n : 0;
+}
+function niceTicks(min, max, count = 4) {
+  if (max === min) return [min];
+  const range = max - min;
+  const step = niceStep(range / count);
+  const start = Math.floor(min / step) * step;
+  const out = [];
+  for (let v = start; v <= max + step / 2; v += step) {
+    out.push(round(v));
+  }
+  return out;
+}
+function niceStep(raw) {
+  const exp = Math.floor(Math.log10(Math.abs(raw))) || 0;
+  const base = Math.pow(10, exp);
+  const norm = raw / base;
+  let nice = 1;
+  if (norm >= 5) nice = 5;
+  else if (norm >= 2) nice = 2;
+  return nice * base;
+}
+function round(v) {
+  return Math.round(v * 1e6) / 1e6;
+}
+function formatTick(v, unit) {
+  const s = Math.abs(v) >= 1e3 ? `${(v / 1e3).toFixed(1)}k` : String(round(v));
+  return unit ? `${s}${unit}` : s;
+}
+
+// src/artifacts/question-artifact.tsx
+import { useCallback as useCallback2, useState as useState3 } from "react";
+import { useThreadRuntime } from "@assistant-ui/react";
+import { CheckIcon } from "lucide-react";
+
+// src/ui/chrome.ts
+var STUDIO_TOPBAR_GAP = "0.5rem";
+var STUDIO_TOPBAR_HEIGHT = "3rem";
+var STUDIO_PILL_HEIGHT = "2.5rem";
+var STUDIO_SIDEBAR_GAP = "0.5rem";
+var STUDIO_SIDEBAR_WIDTH = "3rem";
+var STUDIO_INSET_LEFT = `calc(${STUDIO_SIDEBAR_GAP} + ${STUDIO_SIDEBAR_WIDTH})`;
+var studioChromeShellStyle = {
+  "--studio-topbar-gap": STUDIO_TOPBAR_GAP,
+  "--studio-topbar-height": STUDIO_TOPBAR_HEIGHT,
+  "--studio-chrome-pill-height": STUDIO_PILL_HEIGHT,
+  "--studio-inset-top": `calc(${STUDIO_TOPBAR_GAP} + ${STUDIO_TOPBAR_HEIGHT})`,
+  "--studio-sidebar-gap": STUDIO_SIDEBAR_GAP,
+  "--studio-sidebar-width": STUDIO_SIDEBAR_WIDTH,
+  "--studio-inset-left": STUDIO_INSET_LEFT
+};
+var studioTopbarPillHeightClass = "h-[var(--studio-chrome-pill-height)] min-h-[var(--studio-chrome-pill-height)]";
+var studioTopbarIconPillClass = "shrink-0 flex-none size-[var(--studio-chrome-pill-height)] min-h-[var(--studio-chrome-pill-height)] min-w-[var(--studio-chrome-pill-height)]";
+var studioPlaygroundGradientClass = "bg-gradient-to-b from-neutral-200/60 via-neutral-100/30 to-background dark:from-zinc-800 dark:via-zinc-900 dark:to-zinc-950";
+var studioComposeInputShellClass = "flex w-full flex-col rounded-2xl border border-neutral-200/60 bg-background shadow-lg shadow-black/5 outline-none transition-[box-shadow,border-color] focus-within:border-neutral-400/80 focus-within:ring-2 focus-within:ring-foreground/5 focus-within:shadow-xl focus-within:shadow-black/10 dark:border-white/12 dark:bg-zinc-900 dark:shadow-black/20 dark:focus-within:border-white/22 dark:focus-within:ring-0";
+var studioPillSurfaceClass = TIMBAL_V2_PILL_SURFACE;
+var studioSecondaryChromeClass = TIMBAL_V2_SECONDARY_CHROME;
+var studioIntegrationSurfaceSolid = "bg-white bg-gradient-to-b from-white to-neutral-50/70 shadow-[0_1px_2px_-0.5px_rgba(0,0,0,0.05)] dark:bg-zinc-900 dark:from-white/[0.05] dark:to-white/[0.025] dark:shadow-[0_1px_3px_rgba(0,0,0,0.22)]";
+var studioIntegrationBorder = "border border-neutral-200/80 dark:border-white/[0.08]";
+var studioIntegrationCardClass = cn(
+  "rounded-xl",
+  studioIntegrationSurfaceSolid,
+  studioIntegrationBorder
+);
+var studioIntegrationIconTileClass = cn(
+  "flex size-9 shrink-0 items-center justify-center rounded-lg",
+  studioIntegrationSurfaceSolid,
+  studioIntegrationBorder
+);
+var studioListRowButtonClass = cn(
+  "flex w-full cursor-pointer items-center gap-3 rounded-xl px-3 py-2.5 text-left",
+  studioIntegrationCardClass,
+  "transition-[background-color,box-shadow,border-color] duration-200 ease-in-out",
+  "hover:border-neutral-300 dark:hover:border-white/15",
+  "focus-visible:outline-none focus-visible:ring-2 focus-visible:ring-foreground/15 focus-visible:ring-offset-2 focus-visible:ring-offset-background dark:focus-visible:ring-white/20"
+);
+var studioComposerIoWellClass = cn(
+  "rounded-lg",
+  studioIntegrationSurfaceSolid,
+  studioIntegrationBorder
+);
+var studioToolCardShellClass = cn(
+  studioIntegrationCardClass,
+  "my-2 min-h-0 overflow-hidden"
+);
+var studioTimelineRowButtonClass = "group flex w-full min-w-0 cursor-pointer items-center justify-start rounded-md border-0 bg-transparent py-1 text-left shadow-none focus-visible:outline-none focus-visible:ring-2 focus-visible:ring-foreground/15 focus-visible:ring-offset-2";
+var studioTimelineTextClass = "text-xs font-normal leading-snug";
+var studioTimelineActionClass = cn(
+  studioTimelineTextClass,
+  "shrink-0 text-foreground/70 transition-colors duration-150 group-hover:text-foreground/80"
+);
+var studioTimelineShimmerActionClass = cn(
+  studioTimelineTextClass,
+  "shrink-0"
+);
+var studioTimelineDetailClass = cn(
+  studioTimelineTextClass,
+  "min-w-0 truncate text-muted-foreground transition-colors duration-150"
+);
+function studioTimelineChevronClass(expanded) {
+  return cn(
+    "ml-0.5 size-3 min-h-3 min-w-3 shrink-0 transition-all duration-150",
+    expanded ? "rotate-90 text-foreground opacity-60" : "text-muted-foreground opacity-0 group-hover:opacity-70"
+  );
+}
+var studioTimelineBodyPadClass = "flex flex-col gap-2 pt-0.5 pb-0.5";
+var studioArtifactShellClass = cn(
+  studioIntegrationCardClass,
+  "my-2 w-full min-w-0 overflow-hidden"
+);
+var studioQuestionOptionClass = "flex w-full items-center gap-2 rounded-lg border border-transparent px-2 py-1.5 text-left text-sm transition-[background-color,border-color,box-shadow] duration-200 hover:bg-neutral-100/80 dark:hover:bg-white/[0.05] focus-visible:outline-none focus-visible:ring-2 focus-visible:ring-foreground/15 focus-visible:ring-offset-2";
+var studioQuestionOptionSelectedClass = cn(
+  studioQuestionOptionClass,
+  "border-neutral-200/80 bg-neutral-50/90 ring-1 ring-foreground/10 dark:border-white/[0.12] dark:bg-white/[0.06] dark:ring-white/10"
+);
+
+// src/artifacts/question-artifact.tsx
+import { jsx as jsx10, jsxs as jsxs8 } from "react/jsx-runtime";
+function optionKey(option, index) {
+  const id = option.id?.trim();
+  return id ? id : `__option-${index}`;
+}
+var OptionRadio = ({ selected }) => /* @__PURE__ */ jsx10(
+  "span",
+  {
+    className: cn(
+      "flex size-4 shrink-0 items-center justify-center rounded-full border-2 transition-colors",
+      selected ? "border-foreground bg-foreground text-background" : "border-neutral-300 bg-background dark:border-white/20"
+    ),
+    "aria-hidden": true,
+    children: selected ? /* @__PURE__ */ jsx10(CheckIcon, { className: "size-2.5 stroke-[3]" }) : null
+  }
+);
+var QuestionArtifactView = ({
+  artifact
+}) => {
+  const runtime = useThreadRuntime();
+  const [selected, setSelected] = useState3([]);
+  const [submittedIds, setSubmittedIds] = useState3(null);
+  const isMulti = artifact.multi === true;
+  const isDisabled = submittedIds !== null;
+  const send = useCallback2(
+    (keys) => {
+      if (keys.length === 0) return;
+      const labels = artifact.options.map((option, index) => ({ option, key: optionKey(option, index) })).filter(({ key }) => keys.includes(key)).map(({ option }) => option.label);
+      setSubmittedIds(keys);
+      runtime.append({
+        role: "user",
+        content: [{ type: "text", text: labels.join(", ") }]
+      });
+    },
+    [artifact.options, runtime]
+  );
+  const onPick = useCallback2(
+    (key) => {
+      if (isDisabled) return;
+      if (!isMulti) {
+        send([key]);
+        return;
+      }
+      setSelected(
+        (prev) => prev.includes(key) ? prev.filter((id) => id !== key) : [...prev, key]
+      );
+    },
+    [isDisabled, isMulti, send]
+  );
+  const onConfirm = useCallback2(() => {
+    send(selected);
+  }, [selected, send]);
+  return /* @__PURE__ */ jsx10("div", { className: studioArtifactShellClass, "data-artifact-kind": "question", children: /* @__PURE__ */ jsxs8("div", { className: "px-2.5 py-2", children: [
+    artifact.prompt ? /* @__PURE__ */ jsx10("p", { className: "mb-2 text-sm font-normal leading-snug text-foreground", children: artifact.prompt }) : null,
+    /* @__PURE__ */ jsx10("div", { className: "flex flex-col gap-0.5", role: "list", children: artifact.options.map((option, index) => {
+      const key = optionKey(option, index);
+      const isSelected = submittedIds ? submittedIds.includes(key) : isMulti && selected.includes(key);
+      return /* @__PURE__ */ jsxs8(
+        "button",
+        {
+          type: "button",
+          role: "listitem",
+          disabled: isDisabled,
+          onClick: () => onPick(key),
+          className: cn(
+            isSelected ? studioQuestionOptionSelectedClass : studioQuestionOptionClass,
+            isDisabled && (isSelected ? "cursor-default" : "cursor-not-allowed opacity-50")
+          ),
+          children: [
+            /* @__PURE__ */ jsx10(OptionRadio, { selected: isSelected }),
+            /* @__PURE__ */ jsxs8("span", { className: "min-w-0 flex-1 text-left", children: [
+              /* @__PURE__ */ jsx10("span", { className: "block font-normal text-foreground", children: option.label }),
+              option.description ? /* @__PURE__ */ jsx10("span", { className: "mt-0.5 block text-xs text-muted-foreground", children: option.description }) : null
+            ] })
+          ]
+        },
+        key
+      );
+    }) }),
+    isMulti && !submittedIds ? /* @__PURE__ */ jsx10("div", { className: "mt-2 flex justify-end", children: /* @__PURE__ */ jsx10(
+      TimbalV2Button,
+      {
+        type: "button",
+        variant: "primary",
+        size: "sm",
+        disabled: selected.length === 0,
+        onClick: onConfirm,
+        children: "Confirm"
+      }
+    ) }) : null
+  ] }) });
+};
+
+// src/artifacts/html-artifact.tsx
+import { jsx as jsx11 } from "react/jsx-runtime";
+var HtmlArtifactView = ({ artifact }) => {
+  const sandboxed = artifact.sandboxed !== false;
+  const sandbox = sandboxed ? "allow-scripts allow-same-origin allow-forms allow-modals allow-popups allow-pointer-lock" : void 0;
+  const height = artifact.height ?? "320px";
+  return /* @__PURE__ */ jsx11(ArtifactCard, { title: artifact.title, kind: "html", children: /* @__PURE__ */ jsx11(
+    "iframe",
+    {
+      title: artifact.title ?? "HTML artifact",
+      srcDoc: artifact.content,
+      sandbox,
+      className: "aui-artifact-html w-full border-0 bg-background",
+      style: { height }
+    }
+  ) });
+};
+
+// src/artifacts/json-artifact.tsx
+import { jsx as jsx12 } from "react/jsx-runtime";
+var JsonArtifactView = ({
+  artifact
+}) => {
+  const data = "data" in artifact ? artifact.data : artifact;
+  const title = artifact.title;
+  let body;
+  try {
+    body = JSON.stringify(data, null, 2);
+  } catch {
+    body = String(data);
+  }
+  return /* @__PURE__ */ jsx12(ArtifactCard, { title, kind: "json", children: /* @__PURE__ */ jsx12("pre", { className: "aui-artifact-json m-0 max-h-[420px] overflow-auto p-3 font-mono text-[12px] leading-relaxed text-foreground/85", children: body }) });
+};
+
+// src/artifacts/table-artifact.tsx
+import { jsx as jsx13, jsxs as jsxs9 } from "react/jsx-runtime";
+var TableArtifactView = ({ artifact }) => {
+  const rows = artifact.rows ?? [];
+  const columns = artifact.columns ?? deriveColumns(rows);
+  return /* @__PURE__ */ jsx13(ArtifactCard, { title: artifact.title, kind: "table", children: /* @__PURE__ */ jsx13("div", { className: "aui-artifact-table-wrap overflow-x-auto", children: /* @__PURE__ */ jsxs9("table", { className: "aui-artifact-table w-full border-collapse text-sm", children: [
+    /* @__PURE__ */ jsx13("thead", { children: /* @__PURE__ */ jsx13("tr", { className: "border-b border-border/40 bg-muted/20", children: columns.map((col) => /* @__PURE__ */ jsx13(
+      "th",
+      {
+        className: "px-3 py-2 text-left text-xs font-semibold uppercase tracking-wider text-muted-foreground",
+        children: col.label ?? col.key
+      },
+      col.key
+    )) }) }),
+    /* @__PURE__ */ jsx13("tbody", { children: rows.map((row, i) => /* @__PURE__ */ jsx13(
+      "tr",
+      {
+        className: "border-b border-border/30 transition-colors last:border-b-0 hover:bg-muted/20",
+        children: columns.map((col) => /* @__PURE__ */ jsx13(
+          "td",
+          {
+            className: "px-3 py-2 align-top text-foreground/85",
+            children: formatCell(row[col.key])
+          },
+          col.key
+        ))
+      },
+      i
+    )) })
+  ] }) }) });
+};
+function deriveColumns(rows) {
+  const seen = /* @__PURE__ */ new Set();
+  const cols = [];
+  for (const row of rows) {
+    for (const key of Object.keys(row)) {
+      if (!seen.has(key)) {
+        seen.add(key);
+        cols.push({ key });
+      }
+    }
+  }
+  return cols;
+}
+function formatCell(value) {
+  if (value === null || value === void 0) return "";
+  if (typeof value === "object") return JSON.stringify(value);
+  return String(value);
+}
+
+// src/artifacts/ui/ui-artifact.tsx
+import { useReducer } from "react";
+
+// src/artifacts/ui/types.ts
+function isUiBinding(value) {
+  return typeof value === "object" && value !== null && !Array.isArray(value) && typeof value.$bind === "string";
+}
+
+// src/artifacts/ui/state.ts
+function uiStateReducer(state, action) {
+  switch (action.type) {
+    case "set":
+      return setPath(state, action.path, action.value);
+    case "toggle": {
+      const current = getPath(state, action.path);
+      return setPath(state, action.path, !current);
+    }
+    case "replace":
+      return action.state;
+  }
+}
+function getPath(state, path) {
+  if (!path) return void 0;
+  const parts = path.split(".");
+  let cursor = state;
+  for (const part of parts) {
+    if (typeof cursor !== "object" || cursor === null) return void 0;
+    cursor = cursor[part];
+  }
+  return cursor;
+}
+function setPath(state, path, value) {
+  if (!path) return state;
+  const parts = path.split(".");
+  const next = { ...state };
+  let cursor = next;
+  for (let i = 0; i < parts.length - 1; i++) {
+    const key = parts[i];
+    const child = cursor[key];
+    const cloned = typeof child === "object" && child !== null && !Array.isArray(child) ? { ...child } : {};
+    cursor[key] = cloned;
+    cursor = cloned;
+  }
+  cursor[parts[parts.length - 1]] = value;
+  return next;
+}
+function resolveBindable(value, state) {
+  if (isUiBinding(value)) {
+    return getPath(state, value.$bind);
+  }
+  return value;
+}
+
+// src/artifacts/ui/registry.tsx
+import {
+  createContext as createContext2,
+  useContext as useContext2
+} from "react";
+import { jsx as jsx14 } from "react/jsx-runtime";
+var UiStateContext = createContext2({});
+var UiDispatchContext = createContext2(() => {
+});
+var UiStateProvider = ({ state, dispatch, children }) => /* @__PURE__ */ jsx14(UiStateContext.Provider, { value: state, children: /* @__PURE__ */ jsx14(UiDispatchContext.Provider, { value: dispatch, children }) });
+function useUiState() {
+  return useContext2(UiStateContext);
+}
+function useUiDispatch() {
+  return useContext2(UiDispatchContext);
+}
+var UiEventContext = createContext2(
+  null
+);
+var UiEventProvider = ({ onEvent, children }) => /* @__PURE__ */ jsx14(UiEventContext.Provider, { value: onEvent, children });
+function useUiEventEmitter() {
+  return useContext2(UiEventContext);
+}
+var UiCustomNodeRegistryContext = createContext2({});
+var UiCustomNodeRegistryProvider = ({ renderers, children }) => /* @__PURE__ */ jsx14(UiCustomNodeRegistryContext.Provider, { value: renderers, children });
+function useUiCustomNodeRegistry() {
+  return useContext2(UiCustomNodeRegistryContext);
+}
+
+// src/artifacts/ui/nodes.tsx
+import { useCallback as useCallback3 } from "react";
+import { motion } from "motion/react";
+import { useThreadRuntime as useThreadRuntime2 } from "@assistant-ui/react";
+
+// src/ui/button.tsx
+import { cva } from "class-variance-authority";
+import { Slot as Slot2 } from "radix-ui";
+import { jsx as jsx15 } from "react/jsx-runtime";
+var buttonVariants = cva(
+  "inline-flex items-center justify-center gap-2 whitespace-nowrap rounded-md text-sm font-medium transition-all disabled:pointer-events-none disabled:opacity-50 [&_svg]:pointer-events-none [&_svg:not([class*='size-'])]:size-4 shrink-0 [&_svg]:shrink-0 outline-none focus-visible:border-ring focus-visible:ring-ring/50 focus-visible:ring-[3px] aria-invalid:ring-destructive/20 dark:aria-invalid:ring-destructive/40 aria-invalid:border-destructive",
+  {
+    variants: {
+      variant: {
+        default: "bg-primary text-primary-foreground hover:bg-primary/90",
+        destructive: "bg-destructive text-white hover:bg-destructive/90 focus-visible:ring-destructive/20 dark:focus-visible:ring-destructive/40 dark:bg-destructive/60",
+        outline: "border bg-background shadow-xs hover:bg-accent hover:text-accent-foreground dark:bg-input/30 dark:border-input dark:hover:bg-input/50",
+        secondary: "bg-secondary text-secondary-foreground hover:bg-secondary/80",
+        ghost: "hover:bg-accent hover:text-accent-foreground dark:hover:bg-accent/50",
+        link: "text-primary underline-offset-4 hover:underline"
+      },
+      size: {
+        default: "h-9 px-4 py-2 has-[>svg]:px-3",
+        xs: "h-6 gap-1 rounded-md px-2 text-xs has-[>svg]:px-1.5 [&_svg:not([class*='size-'])]:size-3",
+        sm: "h-8 rounded-md gap-1.5 px-3 has-[>svg]:px-2.5",
+        lg: "h-10 rounded-md px-6 has-[>svg]:px-4",
+        icon: "size-9",
+        "icon-xs": "size-6 rounded-md [&_svg:not([class*='size-'])]:size-3",
+        "icon-sm": "size-8",
+        "icon-lg": "size-10"
+      }
+    },
+    defaultVariants: {
+      variant: "default",
+      size: "default"
+    }
+  }
+);
+function Button({
+  className,
+  variant = "default",
+  size = "default",
+  asChild = false,
+  ...props
+}) {
+  const Comp = asChild ? Slot2.Root : "button";
+  return /* @__PURE__ */ jsx15(
+    Comp,
+    {
+      "data-slot": "button",
+      "data-variant": variant,
+      "data-size": size,
+      className: cn(buttonVariants({ variant, size, className })),
+      ...props
+    }
+  );
+}
+
+// src/artifacts/ui/nodes.tsx
+import { jsx as jsx16, jsxs as jsxs10 } from "react/jsx-runtime";
+var UiNodeView = ({ node }) => {
+  switch (node.kind) {
+    case "box":
+      return /* @__PURE__ */ jsx16(BoxNode, { node });
+    case "text":
+      return /* @__PURE__ */ jsx16(TextNode, { node });
+    case "heading":
+      return /* @__PURE__ */ jsx16(HeadingNode, { node });
+    case "badge":
+      return /* @__PURE__ */ jsx16(BadgeNode, { node });
+    case "button":
+      return /* @__PURE__ */ jsx16(ButtonNode, { node });
+    case "toggle":
+      return /* @__PURE__ */ jsx16(ToggleNode, { node });
+    case "slider":
+      return /* @__PURE__ */ jsx16(SliderNode, { node });
+    case "tooltip":
+      return /* @__PURE__ */ jsx16(TooltipNode, { node });
+    case "draggable":
+      return /* @__PURE__ */ jsx16(DraggableNode, { node });
+    case "custom":
+      return /* @__PURE__ */ jsx16(CustomNode, { node });
+    default:
+      return null;
+  }
+};
+function useActionRunner() {
+  const state = useUiState();
+  const dispatch = useUiDispatch();
+  const runtime = useThreadRuntime2();
+  const emit = useUiEventEmitter();
+  return useCallback3(
+    (actions) => {
+      if (!actions) return;
+      const list = Array.isArray(actions) ? actions : [actions];
+      for (const action of list) {
+        switch (action.kind) {
+          case "message": {
+            const text = resolveBindable(action.text, state);
+            if (typeof text === "string" && text.length > 0) {
+              runtime?.append({
+                role: "user",
+                content: [{ type: "text", text }]
+              });
+            }
+            break;
+          }
+          case "set": {
+            const value = resolveBindable(action.value, state);
+            dispatch({ type: "set", path: action.path, value });
+            break;
+          }
+          case "toggle": {
+            dispatch({ type: "toggle", path: action.path });
+            break;
+          }
+          case "emit": {
+            emit?.({ name: action.name, payload: action.payload });
+            break;
+          }
+        }
+      }
+    },
+    [state, dispatch, runtime, emit]
+  );
+}
+var ALIGN_CLS = {
+  start: "items-start",
+  center: "items-center",
+  end: "items-end",
+  stretch: "items-stretch"
+};
+var JUSTIFY_CLS = {
+  start: "justify-start",
+  center: "justify-center",
+  end: "justify-end",
+  between: "justify-between",
+  around: "justify-around"
+};
+var BoxNode = ({ node }) => {
+  const dir = node.direction ?? "col";
+  return /* @__PURE__ */ jsx16(
+    "div",
+    {
+      className: cn(
+        "aui-ui-box flex",
+        dir === "col" ? "flex-col" : "flex-row",
+        node.wrap && "flex-wrap",
+        node.align && ALIGN_CLS[node.align],
+        node.justify && JUSTIFY_CLS[node.justify],
+        node.className
+      ),
+      style: {
+        gap: node.gap !== void 0 ? `${node.gap * 0.25}rem` : void 0,
+        padding: node.padding !== void 0 ? `${node.padding * 0.25}rem` : void 0
+      },
+      children: node.children?.map((child, i) => /* @__PURE__ */ jsx16(UiNodeView, { node: child }, child.id ?? i))
+    }
+  );
+};
+var TEXT_SIZE = {
+  xs: "text-xs",
+  sm: "text-sm",
+  base: "text-base",
+  lg: "text-lg"
+};
+var TEXT_WEIGHT = {
+  normal: "font-normal",
+  medium: "font-medium",
+  semibold: "font-semibold",
+  bold: "font-bold"
+};
+var TextNode = ({ node }) => {
+  const state = useUiState();
+  const value = resolveBindable(node.value, state);
+  return /* @__PURE__ */ jsx16(
+    "span",
+    {
+      className: cn(
+        "aui-ui-text",
+        node.muted && "text-muted-foreground",
+        node.size && TEXT_SIZE[node.size],
+        node.weight && TEXT_WEIGHT[node.weight],
+        node.className
+      ),
+      children: value === void 0 || value === null ? "" : String(value)
+    }
+  );
+};
+var HEADING_CLS = {
+  1: "text-2xl",
+  2: "text-xl",
+  3: "text-lg",
+  4: "text-base"
+};
+var HeadingNode = ({ node }) => {
+  const state = useUiState();
+  const value = String(resolveBindable(node.value, state) ?? "");
+  const level = node.level ?? 2;
+  const cls = cn(
+    "aui-ui-heading font-semibold text-foreground",
+    HEADING_CLS[level],
+    node.className
+  );
+  switch (level) {
+    case 1:
+      return /* @__PURE__ */ jsx16("h1", { className: cls, children: value });
+    case 2:
+      return /* @__PURE__ */ jsx16("h2", { className: cls, children: value });
+    case 3:
+      return /* @__PURE__ */ jsx16("h3", { className: cls, children: value });
+    case 4:
+      return /* @__PURE__ */ jsx16("h4", { className: cls, children: value });
+  }
+};
+var BADGE_TONE = {
+  default: "bg-muted text-foreground",
+  primary: "bg-primary/10 text-primary",
+  success: "bg-emerald-500/10 text-emerald-600 dark:text-emerald-400",
+  warn: "bg-amber-500/10 text-amber-600 dark:text-amber-400",
+  danger: "bg-destructive/10 text-destructive"
+};
+var BadgeNode = ({ node }) => {
+  const state = useUiState();
+  const value = String(resolveBindable(node.value, state) ?? "");
+  return /* @__PURE__ */ jsx16(
+    "span",
+    {
+      className: cn(
+        "aui-ui-badge inline-flex items-center rounded-full px-2 py-0.5 text-xs font-medium",
+        BADGE_TONE[node.tone ?? "default"],
+        node.className
+      ),
+      children: value
+    }
+  );
+};
+var ButtonNode = ({ node }) => {
+  const state = useUiState();
+  const run = useActionRunner();
+  const label = String(resolveBindable(node.label, state) ?? "");
+  const disabled = node.disabled !== void 0 ? Boolean(resolveBindable(node.disabled, state)) : false;
+  return /* @__PURE__ */ jsx16(
+    Button,
+    {
+      variant: node.variant ?? "default",
+      size: node.size ?? "default",
+      disabled,
+      className: node.className,
+      onClick: () => run(node.onClick),
+      children: label
+    }
+  );
+};
+var ToggleNode = ({ node }) => {
+  const state = useUiState();
+  const dispatch = useUiDispatch();
+  const run = useActionRunner();
+  const value = Boolean(getPath(state, node.binding));
+  const label = node.label ? String(resolveBindable(node.label, state) ?? "") : null;
+  const onToggle = () => {
+    dispatch({ type: "toggle", path: node.binding });
+    run(node.onChange);
+  };
+  return /* @__PURE__ */ jsxs10(
+    "label",
+    {
+      className: cn(
+        "aui-ui-toggle inline-flex cursor-pointer items-center gap-2 text-sm select-none",
+        node.className
+      ),
+      children: [
+        /* @__PURE__ */ jsx16(
+          "button",
+          {
+            type: "button",
+            role: "switch",
+            "aria-checked": value,
+            onClick: onToggle,
+            className: cn(
+              "relative inline-flex h-5 w-9 shrink-0 items-center rounded-full border transition-colors",
+              value ? "border-primary bg-primary" : "border-border bg-muted hover:bg-muted/80"
+            ),
+            children: /* @__PURE__ */ jsx16(
+              "span",
+              {
+                className: cn(
+                  "inline-block size-4 transform rounded-full bg-background shadow transition-transform",
+                  value ? "translate-x-4" : "translate-x-0.5"
+                ),
+                "aria-hidden": true
+              }
+            )
+          }
+        ),
+        label && /* @__PURE__ */ jsx16("span", { className: "text-foreground/85", children: label })
+      ]
+    }
+  );
+};
+var SliderNode = ({ node }) => {
+  const state = useUiState();
+  const dispatch = useUiDispatch();
+  const run = useActionRunner();
+  const min = node.min ?? 0;
+  const max = node.max ?? 100;
+  const step = node.step ?? 1;
+  const raw = getPath(state, node.binding);
+  const value = typeof raw === "number" ? raw : min;
+  const showValue = node.showValue ?? true;
+  const label = node.label ? String(resolveBindable(node.label, state) ?? "") : null;
+  const onChange = (e) => {
+    const next = Number(e.target.value);
+    dispatch({ type: "set", path: node.binding, value: next });
+  };
+  return /* @__PURE__ */ jsxs10("div", { className: cn("aui-ui-slider flex flex-col gap-1", node.className), children: [
+    (label || showValue) && /* @__PURE__ */ jsxs10("div", { className: "flex items-center justify-between text-xs text-muted-foreground", children: [
+      label && /* @__PURE__ */ jsx16("span", { children: label }),
+      showValue && /* @__PURE__ */ jsx16("span", { className: "font-mono", children: value })
+    ] }),
+    /* @__PURE__ */ jsx16(
+      "input",
+      {
+        type: "range",
+        min,
+        max,
+        step,
+        value,
+        onChange,
+        onMouseUp: () => run(node.onChange),
+        onKeyUp: (e) => {
+          if (e.key === "Enter" || e.key === " ") run(node.onChange);
+        },
+        onTouchEnd: () => run(node.onChange),
+        className: "aui-ui-slider-input h-1.5 w-full cursor-pointer appearance-none rounded-full bg-muted accent-primary"
+      }
+    )
+  ] });
+};
+var TooltipNode = ({ node }) => {
+  const state = useUiState();
+  const content = String(resolveBindable(node.content, state) ?? "");
+  return /* @__PURE__ */ jsx16(TooltipProvider, { children: /* @__PURE__ */ jsxs10(Tooltip, { children: [
+    /* @__PURE__ */ jsx16(TooltipTrigger, { asChild: true, children: /* @__PURE__ */ jsx16("span", { className: cn("aui-ui-tooltip-trigger inline-flex", node.className), children: /* @__PURE__ */ jsx16(UiNodeView, { node: node.child }) }) }),
+    /* @__PURE__ */ jsx16(TooltipContent, { side: node.side ?? "top", children: content })
+  ] }) });
+};
+var DraggableNode = ({ node }) => {
+  const run = useActionRunner();
+  const snapBack = node.snapBack ?? true;
+  const axis = node.axis ?? "both";
+  const dragProp = axis === "both" ? true : axis;
+  return /* @__PURE__ */ jsx16(
+    motion.div,
+    {
+      drag: dragProp,
+      dragMomentum: false,
+      dragSnapToOrigin: snapBack,
+      whileDrag: { scale: 1.02, cursor: "grabbing" },
+      onDragEnd: () => run(node.onDragEnd),
+      className: cn(
+        "aui-ui-draggable inline-block cursor-grab touch-none",
+        node.className
+      ),
+      children: /* @__PURE__ */ jsx16(UiNodeView, { node: node.child })
     }
-  ) });
+  );
 };
+var CustomNode = ({ node }) => {
+  const state = useUiState();
+  const registry = useUiCustomNodeRegistry();
+  const Renderer = registry[node.name];
+  if (!Renderer) return null;
+  const resolvedProps = resolveProps(node.props ?? {}, state);
+  const children = node.children?.map((child, i) => /* @__PURE__ */ jsx16(UiNodeView, { node: child }, child.id ?? i));
+  return /* @__PURE__ */ jsx16(Renderer, { props: resolvedProps, children });
+};
+function resolveProps(props, state) {
+  const out = {};
+  for (const [k, v] of Object.entries(props)) {
+    out[k] = resolveBindable(v, state);
+  }
+  return out;
+}
 
-// src/components/markdown-text.tsx
-import "@assistant-ui/react-markdown/styles/dot.css";
-import "katex/dist/katex.min.css";
-import {
-  MarkdownTextPrimitive,
-  unstable_memoizeMarkdownComponents as memoizeMarkdownComponents,
-  useIsMarkdownCodeBlock
-} from "@assistant-ui/react-markdown";
-import remarkGfm from "remark-gfm";
-import remarkMath from "remark-math";
-import rehypeKatex from "rehype-katex";
-import { memo, useState as useState4 } from "react";
-import { CheckIcon, CopyIcon } from "lucide-react";
+// src/artifacts/ui/ui-artifact.tsx
+import { jsx as jsx17 } from "react/jsx-runtime";
+var UiArtifactView = ({ artifact }) => {
+  const [state, dispatch] = useReducer(
+    uiStateReducer,
+    artifact.initialState ?? {}
+  );
+  return /* @__PURE__ */ jsx17(ArtifactCard, { title: artifact.title, kind: "ui", children: /* @__PURE__ */ jsx17(UiStateProvider, { state, dispatch, children: /* @__PURE__ */ jsx17("div", { className: "aui-ui-root p-3", children: /* @__PURE__ */ jsx17(UiNodeView, { node: artifact.root }) }) }) });
+};
+
+// src/artifacts/registry.tsx
+import { jsx as jsx18 } from "react/jsx-runtime";
+var defaultArtifactRenderers = {
+  chart: ChartArtifactView,
+  question: QuestionArtifactView,
+  html: HtmlArtifactView,
+  json: JsonArtifactView,
+  table: TableArtifactView,
+  ui: UiArtifactView
+};
+var ArtifactRegistryContext = createContext3(
+  defaultArtifactRenderers
+);
+var ArtifactRegistryProvider = ({ renderers, override, children }) => {
+  const merged = useMemo3(() => {
+    if (!renderers) return defaultArtifactRenderers;
+    if (override) return renderers;
+    return { ...defaultArtifactRenderers, ...renderers };
+  }, [renderers, override]);
+  return /* @__PURE__ */ jsx18(ArtifactRegistryContext.Provider, { value: merged, children });
+};
+function useArtifactRegistry() {
+  return useContext3(ArtifactRegistryContext);
+}
+var ArtifactView = ({ artifact }) => {
+  const registry = useArtifactRegistry();
+  const Renderer = registry[artifact.type] ?? registry.json;
+  if (!Renderer) return null;
+  return /* @__PURE__ */ jsx18(Renderer, { artifact });
+};
+
+// src/artifacts/parse.ts
+var ARTIFACT_FENCE_LANGUAGES = /* @__PURE__ */ new Set([
+  "timbal-artifact",
+  "timbal"
+]);
+function isArtifactFenceLanguage(language) {
+  return typeof language === "string" && ARTIFACT_FENCE_LANGUAGES.has(language);
+}
+function tryParseStructuredText(text) {
+  const trimmed = text.trim();
+  if (!trimmed.startsWith("{") && !trimmed.startsWith("[")) return null;
+  try {
+    return JSON.parse(trimmed);
+  } catch {
+  }
+  try {
+    const asJson = trimmed.replace(/\bTrue\b/g, "true").replace(/\bFalse\b/g, "false").replace(/\bNone\b/g, "null").replace(/'/g, '"');
+    return JSON.parse(asJson);
+  } catch {
+    return null;
+  }
+}
+function parseArtifactFromToolResult(result) {
+  if (result === void 0 || result === null) return null;
+  if (typeof result === "string") {
+    const parsed = tryParseStructuredText(result);
+    if (parsed === null) return null;
+    return parseArtifactFromToolResult(parsed);
+  }
+  if (Array.isArray(result)) {
+    for (const item of result) {
+      if (typeof item === "object" && item !== null && "text" in item) {
+        const text = item.text;
+        if (typeof text === "string") {
+          const fromText = parseArtifactFromToolResult(text);
+          if (fromText) return fromText;
+        }
+      }
+    }
+    return null;
+  }
+  if (typeof result === "object") {
+    const obj = result;
+    if (obj.type === "text" && typeof obj.text === "string") {
+      return parseArtifactFromToolResult(obj.text);
+    }
+    if (obj.type === "thinking" && typeof obj.thinking === "string") {
+      return parseArtifactFromToolResult(obj.thinking);
+    }
+  }
+  return isArtifact(result) ? result : null;
+}
+var FENCE_RE = /```(?:timbal-artifact|timbal)\s*\n([\s\S]*?)\n```/g;
+function findMarkdownArtifacts(markdown) {
+  const matches = [];
+  FENCE_RE.lastIndex = 0;
+  let m;
+  while ((m = FENCE_RE.exec(markdown)) !== null) {
+    const raw = m[0];
+    const body = m[1];
+    try {
+      const parsed = JSON.parse(body);
+      if (isArtifact(parsed)) {
+        matches.push({
+          artifact: parsed,
+          raw,
+          start: m.index,
+          end: m.index + raw.length
+        });
+      }
+    } catch {
+    }
+  }
+  return matches;
+}
+function splitMarkdownByArtifacts(markdown) {
+  const matches = findMarkdownArtifacts(markdown);
+  if (matches.length === 0) return [{ kind: "text", text: markdown }];
+  const segments = [];
+  let cursor = 0;
+  for (const match of matches) {
+    if (match.start > cursor) {
+      segments.push({ kind: "text", text: markdown.slice(cursor, match.start) });
+    }
+    segments.push({ kind: "artifact", artifact: match.artifact });
+    cursor = match.end;
+  }
+  if (cursor < markdown.length) {
+    segments.push({ kind: "text", text: markdown.slice(cursor) });
+  }
+  return segments;
+}
 
 // src/components/syntax-highlighter.tsx
-import { useEffect as useEffect3, useState as useState3 } from "react";
-import { createHighlighterCore } from "shiki/core";
-import { createJavaScriptRegexEngine } from "shiki/engine/javascript";
 import langJavascript from "shiki/langs/javascript.mjs";
 import langTypescript from "shiki/langs/typescript.mjs";
 import langPython from "shiki/langs/python.mjs";
@@ -811,7 +2627,7 @@ import langC from "shiki/langs/c.mjs";
 import langCpp from "shiki/langs/cpp.mjs";
 import themeVitesseDark from "shiki/themes/vitesse-dark.mjs";
 import themeVitesseLight from "shiki/themes/vitesse-light.mjs";
-import { jsx as jsx8 } from "react/jsx-runtime";
+import { jsx as jsx19 } from "react/jsx-runtime";
 var SHIKI_THEME_DARK = "vitesse-dark";
 var SHIKI_THEME_LIGHT = "vitesse-light";
 var highlighterPromise = null;
@@ -849,7 +2665,7 @@ var ShikiSyntaxHighlighter = ({
   language,
   code
 }) => {
-  const [html, setHtml] = useState3(null);
+  const [html, setHtml] = useState4(null);
   useEffect3(() => {
     let cancelled = false;
     (async () => {
@@ -876,8 +2692,17 @@ var ShikiSyntaxHighlighter = ({
       cancelled = true;
     };
   }, [code, language]);
+  if (isArtifactFenceLanguage(language)) {
+    try {
+      const parsed = JSON.parse(code);
+      if (isArtifact(parsed)) {
+        return /* @__PURE__ */ jsx19(ArtifactView, { artifact: parsed });
+      }
+    } catch {
+    }
+  }
   if (html) {
-    return /* @__PURE__ */ jsx8(
+    return /* @__PURE__ */ jsx19(
       "div",
       {
         className: "shiki-wrapper [&>pre]:!m-0 [&>pre]:!rounded-t-none [&>pre]:!rounded-b-lg [&>pre]:!border [&>pre]:!border-t-0 [&>pre]:!border-border/50 [&>pre]:!p-3 [&>pre]:!text-xs [&>pre]:!leading-relaxed [&>pre]:overflow-x-auto",
@@ -885,14 +2710,14 @@ var ShikiSyntaxHighlighter = ({
       }
     );
   }
-  return /* @__PURE__ */ jsx8(Pre, { children: /* @__PURE__ */ jsx8(Code2, { children: code }) });
+  return /* @__PURE__ */ jsx19(Pre, { children: /* @__PURE__ */ jsx19(Code2, { children: code }) });
 };
 var syntax_highlighter_default = ShikiSyntaxHighlighter;
 
 // src/components/markdown-text.tsx
-import { jsx as jsx9, jsxs as jsxs5 } from "react/jsx-runtime";
+import { jsx as jsx20, jsxs as jsxs11 } from "react/jsx-runtime";
 var MarkdownTextImpl = () => {
-  return /* @__PURE__ */ jsx9(
+  return /* @__PURE__ */ jsx20(
     MarkdownTextPrimitive,
     {
       remarkPlugins: [remarkGfm, remarkMath],
@@ -908,24 +2733,25 @@ var MarkdownTextImpl = () => {
 var MarkdownText = memo(MarkdownTextImpl);
 var CodeHeader = ({ language, code }) => {
   const { isCopied, copyToClipboard } = useCopyToClipboard();
+  if (isArtifactFenceLanguage(language)) return null;
   const onCopy = () => {
     if (!code || isCopied) return;
     copyToClipboard(code);
   };
-  return /* @__PURE__ */ jsxs5("div", { className: "aui-code-header flex items-center justify-between rounded-t-lg border border-b-0 border-border/50 bg-zinc-100 px-4 py-2 dark:bg-zinc-800/80", children: [
-    /* @__PURE__ */ jsxs5("span", { className: "flex items-center gap-2 text-xs font-semibold tracking-wide text-muted-foreground/80 uppercase", children: [
-      /* @__PURE__ */ jsx9("span", { className: "inline-block h-2 w-2 rounded-full bg-primary/40" }),
+  return /* @__PURE__ */ jsxs11("div", { className: "aui-code-header flex items-center justify-between rounded-t-lg border border-b-0 border-border/50 bg-zinc-100 px-4 py-2 dark:bg-zinc-800/80", children: [
+    /* @__PURE__ */ jsxs11("span", { className: "flex items-center gap-2 text-xs font-semibold tracking-wide text-muted-foreground/80 uppercase", children: [
+      /* @__PURE__ */ jsx20("span", { className: "inline-block h-2 w-2 rounded-full bg-primary/40" }),
       language
     ] }),
-    /* @__PURE__ */ jsxs5(
+    /* @__PURE__ */ jsxs11(
       TooltipIconButton,
       {
         tooltip: isCopied ? "Copied!" : "Copy",
         onClick: onCopy,
         className: "transition-colors hover:text-foreground",
         children: [
-          !isCopied && /* @__PURE__ */ jsx9(CopyIcon, { className: "h-3.5 w-3.5" }),
-          isCopied && /* @__PURE__ */ jsx9(CheckIcon, { className: "h-3.5 w-3.5 text-emerald-500" })
+          !isCopied && /* @__PURE__ */ jsx20(CopyIcon, { className: "h-3.5 w-3.5" }),
+          isCopied && /* @__PURE__ */ jsx20(CheckIcon2, { className: "h-3.5 w-3.5 text-emerald-500" })
         ]
       }
     )
@@ -934,7 +2760,7 @@ var CodeHeader = ({ language, code }) => {
 var useCopyToClipboard = ({
   copiedDuration = 3e3
 } = {}) => {
-  const [isCopied, setIsCopied] = useState4(false);
+  const [isCopied, setIsCopied] = useState5(false);
   const copyToClipboard = (value) => {
     if (!value) return;
     navigator.clipboard.writeText(value).then(() => {
@@ -945,7 +2771,7 @@ var useCopyToClipboard = ({
   return { isCopied, copyToClipboard };
 };
 var defaultComponents = memoizeMarkdownComponents({
-  h1: ({ className, ...props }) => /* @__PURE__ */ jsx9(
+  h1: ({ className, ...props }) => /* @__PURE__ */ jsx20(
     "h1",
     {
       className: cn(
@@ -955,7 +2781,7 @@ var defaultComponents = memoizeMarkdownComponents({
       ...props
     }
   ),
-  h2: ({ className, ...props }) => /* @__PURE__ */ jsx9(
+  h2: ({ className, ...props }) => /* @__PURE__ */ jsx20(
     "h2",
     {
       className: cn(
@@ -965,7 +2791,7 @@ var defaultComponents = memoizeMarkdownComponents({
       ...props
     }
   ),
-  h3: ({ className, ...props }) => /* @__PURE__ */ jsx9(
+  h3: ({ className, ...props }) => /* @__PURE__ */ jsx20(
     "h3",
     {
       className: cn(
@@ -975,7 +2801,7 @@ var defaultComponents = memoizeMarkdownComponents({
       ...props
     }
   ),
-  h4: ({ className, ...props }) => /* @__PURE__ */ jsx9(
+  h4: ({ className, ...props }) => /* @__PURE__ */ jsx20(
     "h4",
     {
       className: cn(
@@ -985,7 +2811,7 @@ var defaultComponents = memoizeMarkdownComponents({
       ...props
     }
   ),
-  h5: ({ className, ...props }) => /* @__PURE__ */ jsx9(
+  h5: ({ className, ...props }) => /* @__PURE__ */ jsx20(
     "h5",
     {
       className: cn(
@@ -995,7 +2821,7 @@ var defaultComponents = memoizeMarkdownComponents({
       ...props
     }
   ),
-  h6: ({ className, ...props }) => /* @__PURE__ */ jsx9(
+  h6: ({ className, ...props }) => /* @__PURE__ */ jsx20(
     "h6",
     {
       className: cn(
@@ -1005,7 +2831,7 @@ var defaultComponents = memoizeMarkdownComponents({
       ...props
     }
   ),
-  p: ({ className, ...props }) => /* @__PURE__ */ jsx9(
+  p: ({ className, ...props }) => /* @__PURE__ */ jsx20(
     "p",
     {
       className: cn(
@@ -1015,7 +2841,7 @@ var defaultComponents = memoizeMarkdownComponents({
       ...props
     }
   ),
-  a: ({ className, ...props }) => /* @__PURE__ */ jsx9(
+  a: ({ className, ...props }) => /* @__PURE__ */ jsx20(
     "a",
     {
       className: cn(
@@ -1027,7 +2853,7 @@ var defaultComponents = memoizeMarkdownComponents({
       ...props
     }
   ),
-  blockquote: ({ className, ...props }) => /* @__PURE__ */ jsx9(
+  blockquote: ({ className, ...props }) => /* @__PURE__ */ jsx20(
     "blockquote",
     {
       className: cn(
@@ -1037,7 +2863,7 @@ var defaultComponents = memoizeMarkdownComponents({
       ...props
     }
   ),
-  ul: ({ className, ...props }) => /* @__PURE__ */ jsx9(
+  ul: ({ className, ...props }) => /* @__PURE__ */ jsx20(
     "ul",
     {
       className: cn(
@@ -1047,7 +2873,7 @@ var defaultComponents = memoizeMarkdownComponents({
       ...props
     }
   ),
-  ol: ({ className, ...props }) => /* @__PURE__ */ jsx9(
+  ol: ({ className, ...props }) => /* @__PURE__ */ jsx20(
     "ol",
     {
       className: cn(
@@ -1057,7 +2883,7 @@ var defaultComponents = memoizeMarkdownComponents({
       ...props
     }
   ),
-  hr: ({ className, ...props }) => /* @__PURE__ */ jsx9(
+  hr: ({ className, ...props }) => /* @__PURE__ */ jsx20(
     "hr",
     {
       className: cn(
@@ -1067,14 +2893,14 @@ var defaultComponents = memoizeMarkdownComponents({
       ...props
     }
   ),
-  table: ({ className, ...props }) => /* @__PURE__ */ jsx9("div", { className: "my-4 w-full overflow-x-auto rounded-lg border border-border/50", children: /* @__PURE__ */ jsx9(
+  table: ({ className, ...props }) => /* @__PURE__ */ jsx20("div", { className: "my-4 w-full overflow-x-auto rounded-lg border border-border/50", children: /* @__PURE__ */ jsx20(
     "table",
     {
       className: cn("aui-md-table w-full border-collapse text-sm", className),
       ...props
     }
   ) }),
-  th: ({ className, ...props }) => /* @__PURE__ */ jsx9(
+  th: ({ className, ...props }) => /* @__PURE__ */ jsx20(
     "th",
     {
       className: cn(
@@ -1084,7 +2910,7 @@ var defaultComponents = memoizeMarkdownComponents({
       ...props
     }
   ),
-  td: ({ className, ...props }) => /* @__PURE__ */ jsx9(
+  td: ({ className, ...props }) => /* @__PURE__ */ jsx20(
     "td",
     {
       className: cn(
@@ -1094,7 +2920,7 @@ var defaultComponents = memoizeMarkdownComponents({
       ...props
     }
   ),
-  tr: ({ className, ...props }) => /* @__PURE__ */ jsx9(
+  tr: ({ className, ...props }) => /* @__PURE__ */ jsx20(
     "tr",
     {
       className: cn(
@@ -1104,8 +2930,8 @@ var defaultComponents = memoizeMarkdownComponents({
       ...props
     }
   ),
-  li: ({ className, ...props }) => /* @__PURE__ */ jsx9("li", { className: cn("aui-md-li leading-[1.7]", className), ...props }),
-  sup: ({ className, ...props }) => /* @__PURE__ */ jsx9(
+  li: ({ className, ...props }) => /* @__PURE__ */ jsx20("li", { className: cn("aui-md-li leading-[1.7]", className), ...props }),
+  sup: ({ className, ...props }) => /* @__PURE__ */ jsx20(
     "sup",
     {
       className: cn(
@@ -1115,7 +2941,7 @@ var defaultComponents = memoizeMarkdownComponents({
       ...props
     }
   ),
-  pre: ({ className, ...props }) => /* @__PURE__ */ jsx9(
+  pre: ({ className, ...props }) => /* @__PURE__ */ jsx20(
     "pre",
     {
       className: cn(
@@ -1127,7 +2953,7 @@ var defaultComponents = memoizeMarkdownComponents({
   ),
   code: function Code({ className, ...props }) {
     const isCodeBlock = useIsMarkdownCodeBlock();
-    return /* @__PURE__ */ jsx9(
+    return /* @__PURE__ */ jsx20(
       "code",
       {
         className: cn(
@@ -1138,108 +2964,556 @@ var defaultComponents = memoizeMarkdownComponents({
       }
     );
   },
-  strong: ({ className, ...props }) => /* @__PURE__ */ jsx9("strong", { className: cn("font-semibold text-foreground", className), ...props }),
-  em: ({ className, ...props }) => /* @__PURE__ */ jsx9("em", { className: cn("italic", className), ...props }),
+  strong: ({ className, ...props }) => /* @__PURE__ */ jsx20("strong", { className: cn("font-semibold text-foreground", className), ...props }),
+  em: ({ className, ...props }) => /* @__PURE__ */ jsx20("em", { className: cn("italic", className), ...props }),
   CodeHeader
 });
 
 // src/components/tool-fallback.tsx
-import { memo as memo3 } from "react";
-import { WrenchIcon } from "lucide-react";
+import { memo as memo3, useMemo as useMemo5, useState as useState6 } from "react";
+import { ChevronRightIcon } from "lucide-react";
+import {
+  useAuiState as useAuiState2
+} from "@assistant-ui/react";
 
 // src/ui/shimmer.tsx
-import { motion } from "motion/react";
+import { motion as motion2 } from "motion/react";
 import {
   memo as memo2,
-  useMemo
+  useMemo as useMemo4
+} from "react";
+import { jsx as jsx21 } from "react/jsx-runtime";
+var ShimmerComponent = ({
+  children,
+  as: Component = "p",
+  className,
+  duration = 2,
+  spread = 2
+}) => {
+  const MotionComponent = motion2.create(
+    Component
+  );
+  const dynamicSpread = useMemo4(
+    () => (children?.length ?? 0) * spread,
+    [children, spread]
+  );
+  return /* @__PURE__ */ jsx21(
+    MotionComponent,
+    {
+      animate: { backgroundPosition: "0% center" },
+      className: cn(
+        "relative inline-block bg-[length:250%_100%,auto] bg-clip-text text-transparent",
+        "[--bg:linear-gradient(90deg,#0000_calc(50%-var(--spread)),var(--color-background),#0000_calc(50%+var(--spread)))] [background-repeat:no-repeat,padding-box]",
+        className
+      ),
+      initial: { backgroundPosition: "100% center" },
+      style: {
+        "--spread": `${dynamicSpread}px`,
+        backgroundImage: "var(--bg), linear-gradient(var(--color-muted-foreground), var(--color-muted-foreground))"
+      },
+      transition: {
+        repeat: Number.POSITIVE_INFINITY,
+        duration,
+        ease: "linear"
+      },
+      children
+    }
+  );
+};
+var Shimmer = memo2(ShimmerComponent);
+
+// src/components/motion.tsx
+import { AnimatePresence, motion as motion3, useReducedMotion } from "motion/react";
+import { jsx as jsx22 } from "react/jsx-runtime";
+var luxuryEase = [0.16, 1, 0.3, 1];
+var TOOL_ENTER_MS = 0.78;
+var TOOL_EXIT_MS = 0.28;
+function toolPresenceTransition(reduced) {
+  return {
+    enter: {
+      duration: reduced ? 0.35 : TOOL_ENTER_MS,
+      ease: luxuryEase
+    },
+    exit: {
+      duration: reduced ? 0.2 : TOOL_EXIT_MS,
+      ease: [0.4, 0, 1, 1]
+    }
+  };
+}
+function toolMotionState(reduced, entering, variant) {
+  if (reduced) {
+    return entering ? { opacity: 0, y: variant === "executing" ? 8 : 10 } : { opacity: 1, y: 0 };
+  }
+  if (variant === "executing") {
+    return entering ? { opacity: 0, y: 12 } : { opacity: 1, y: 0 };
+  }
+  return entering ? { opacity: 0, y: 14, filter: "blur(10px)" } : { opacity: 1, y: 0, filter: "blur(0px)" };
+}
+function ToolMotion({ children, className, motionKey }) {
+  const reduced = useReducedMotion() ?? false;
+  const { enter, exit } = toolPresenceTransition(reduced);
+  return /* @__PURE__ */ jsx22(
+    motion3.div,
+    {
+      className: cn("aui-tool-motion w-full min-w-0", className),
+      initial: toolMotionState(reduced, true, "settled"),
+      animate: toolMotionState(reduced, false, "settled"),
+      exit: reduced ? { opacity: 0, y: 6, transition: exit } : { opacity: 0, y: 8, filter: "blur(6px)", transition: exit },
+      transition: enter,
+      style: { willChange: "opacity, transform, filter" },
+      children
+    },
+    motionKey
+  );
+}
+function ToolPresence({
+  presenceKey,
+  children,
+  className,
+  variant = "settled"
+}) {
+  const reduced = useReducedMotion() ?? false;
+  const { enter, exit } = toolPresenceTransition(reduced);
+  const enterTransition = variant === "executing" ? { duration: reduced ? 0.3 : 0.52, ease: luxuryEase } : enter;
+  return /* @__PURE__ */ jsx22(AnimatePresence, { mode: "wait", initial: true, children: /* @__PURE__ */ jsx22(
+    motion3.div,
+    {
+      className: cn("aui-tool-presence w-full min-w-0", className),
+      initial: toolMotionState(reduced, true, variant),
+      animate: toolMotionState(reduced, false, variant),
+      exit: reduced ? { opacity: 0, y: 6, transition: exit } : { opacity: 0, y: 8, filter: "blur(6px)", transition: exit },
+      transition: enterTransition,
+      style: {
+        willChange: variant === "executing" ? "opacity, transform" : "opacity, transform, filter"
+      },
+      children
+    },
+    presenceKey
+  ) });
+}
+function ToolBodyPresence({
+  open,
+  children,
+  className
+}) {
+  const reduced = useReducedMotion() ?? false;
+  return /* @__PURE__ */ jsx22(
+    "div",
+    {
+      className: cn(
+        "aui-tool-body grid min-h-0 transition-[grid-template-rows]",
+        open ? reduced ? "duration-200 ease-out" : "duration-[340ms] ease-[cubic-bezier(0.16,1,0.3,1)]" : reduced ? "duration-150 ease-[cubic-bezier(0.4,0,0.2,1)]" : "duration-200 ease-[cubic-bezier(0.4,0,0.2,1)]"
+      ),
+      style: { gridTemplateRows: open ? "1fr" : "0fr" },
+      children: /* @__PURE__ */ jsx22("div", { className: "min-h-0 overflow-hidden", children: /* @__PURE__ */ jsx22(
+        "div",
+        {
+          className: cn(
+            className,
+            "transition-opacity",
+            open ? reduced ? "opacity-100 duration-200 ease-out" : "opacity-100 duration-300 ease-[cubic-bezier(0.16,1,0.3,1)] delay-75" : reduced ? "opacity-0 duration-100 ease-in" : "opacity-0 duration-150 ease-[cubic-bezier(0.4,0,0.2,1)]"
+          ),
+          children
+        }
+      ) })
+    }
+  );
+}
+
+// src/components/tool-fallback.tsx
+import { jsx as jsx23, jsxs as jsxs12 } from "react/jsx-runtime";
+function detectRunning({
+  status,
+  result,
+  streamRunning
+}) {
+  const isError = status?.type === "incomplete" && status.reason !== "cancelled";
+  if (isError) return false;
+  if (status?.type === "running") return true;
+  if (status?.type === "complete") return false;
+  return streamRunning && result === void 0;
+}
+function useToolRunning(props) {
+  const { isRunning: streamRunning } = useTimbalRuntime();
+  const partStatus = useAuiState2((s) => s.part.status);
+  return detectRunning({
+    status: partStatus ?? props.status,
+    result: props.result,
+    streamRunning
+  });
+}
+function formatToolLabel(toolName) {
+  return toolName.replace(/_/g, " ").replace(/([a-z])([A-Z])/g, "$1 $2").toLowerCase();
+}
+function formatToolResult(result) {
+  if (typeof result === "string") return result;
+  try {
+    return JSON.stringify(result, null, 2);
+  } catch {
+    return String(result);
+  }
+}
+var TimelineActionLabel = ({ action, detail, shimmer = false }) => /* @__PURE__ */ jsxs12("span", { className: "inline-flex min-w-0 max-w-full items-baseline gap-1", children: [
+  action ? shimmer ? /* @__PURE__ */ jsx23(
+    Shimmer,
+    {
+      as: "span",
+      className: cn(studioTimelineShimmerActionClass, "aui-tool-shimmer"),
+      duration: 1.8,
+      spread: 2.5,
+      children: action
+    }
+  ) : /* @__PURE__ */ jsx23("span", { className: studioTimelineActionClass, children: action }) : null,
+  detail ? /* @__PURE__ */ jsx23("span", { className: studioTimelineDetailClass, children: detail }) : null
+] });
+var TimelineHoverChevron = ({ expanded }) => /* @__PURE__ */ jsx23(
+  ChevronRightIcon,
+  {
+    className: studioTimelineChevronClass(expanded),
+    "aria-hidden": true
+  }
+);
+var ToolPanel = ({ toolName, argsText, result, isError }) => {
+  const [open, setOpen] = useState6(false);
+  const detail = formatToolLabel(toolName);
+  const formattedArgs = useMemo5(() => {
+    if (!argsText || argsText === "{}") return null;
+    try {
+      return JSON.stringify(JSON.parse(argsText), null, 2);
+    } catch {
+      return argsText;
+    }
+  }, [argsText]);
+  const formattedResult = useMemo5(() => {
+    if (result === void 0 || result === null) return null;
+    return formatToolResult(result);
+  }, [result]);
+  const hasBody = Boolean(formattedArgs || formattedResult);
+  const action = isError ? "Failed" : "Used";
+  if (!hasBody) {
+    return /* @__PURE__ */ jsx23("div", { className: "aui-tool-row w-full min-w-0", children: /* @__PURE__ */ jsx23(TimelineActionLabel, { action, detail }) });
+  }
+  return /* @__PURE__ */ jsxs12("div", { className: "aui-tool-row w-full min-w-0", children: [
+    /* @__PURE__ */ jsx23(
+      "button",
+      {
+        type: "button",
+        onClick: () => setOpen((v) => !v),
+        "aria-expanded": open,
+        "aria-label": `${action} ${detail}`,
+        className: studioTimelineRowButtonClass,
+        children: /* @__PURE__ */ jsxs12(
+          "span",
+          {
+            className: cn(
+              "inline-flex min-w-0 max-w-full items-center gap-0.5",
+              studioTimelineTextClass,
+              "text-foreground"
+            ),
+            children: [
+              /* @__PURE__ */ jsx23(TimelineActionLabel, { action, detail }),
+              /* @__PURE__ */ jsx23(TimelineHoverChevron, { expanded: open })
+            ]
+          }
+        )
+      }
+    ),
+    /* @__PURE__ */ jsxs12(
+      ToolBodyPresence,
+      {
+        open,
+        className: cn(studioTimelineBodyPadClass, "gap-2"),
+        children: [
+          formattedArgs ? /* @__PURE__ */ jsx23(
+            "div",
+            {
+              className: cn(
+                studioComposerIoWellClass,
+                "max-h-48 overflow-auto px-2.5 py-2"
+              ),
+              children: /* @__PURE__ */ jsx23("pre", { className: "whitespace-pre-wrap break-words font-mono text-[11px] font-normal leading-relaxed text-foreground", children: formattedArgs })
+            }
+          ) : null,
+          formattedResult ? /* @__PURE__ */ jsx23(
+            "div",
+            {
+              className: cn(
+                studioComposerIoWellClass,
+                "max-h-48 overflow-auto px-2.5 py-2"
+              ),
+              children: /* @__PURE__ */ jsx23("pre", { className: "whitespace-pre-wrap break-words font-mono text-[11px] font-normal leading-relaxed text-foreground", children: formattedResult })
+            }
+          ) : null
+        ]
+      }
+    )
+  ] });
+};
+var ToolFallbackImpl = ({
+  toolName,
+  argsText,
+  result,
+  status
+}) => {
+  const isRunning = useToolRunning({ status, result });
+  const isError = status?.type === "incomplete" && status.reason !== "cancelled";
+  const presenceKey = isRunning ? "running" : isError ? "error" : "complete";
+  return /* @__PURE__ */ jsx23(
+    ToolPresence,
+    {
+      presenceKey,
+      variant: isRunning ? "executing" : "settled",
+      className: "py-0.5",
+      children: isRunning ? /* @__PURE__ */ jsx23("div", { className: "aui-tool-running", children: /* @__PURE__ */ jsx23(
+        TimelineActionLabel,
+        {
+          action: "Using",
+          detail: formatToolLabel(toolName),
+          shimmer: true
+        }
+      ) }) : /* @__PURE__ */ jsx23(
+        ToolPanel,
+        {
+          toolName,
+          argsText,
+          result,
+          isError
+        }
+      )
+    }
+  );
+};
+var ToolFallback = memo3(
+  ToolFallbackImpl
+);
+ToolFallback.displayName = "ToolFallback";
+
+// src/artifacts/tool-artifact.tsx
+import { jsx as jsx24 } from "react/jsx-runtime";
+var ToolArtifactFallback = (props) => {
+  const registry = useArtifactRegistry();
+  const isRunning = useToolRunning({
+    status: props.status,
+    result: props.result
+  });
+  if (!isRunning) {
+    const artifact = parseArtifactFromToolResult(props.result);
+    if (artifact) {
+      const Renderer = registry[artifact.type];
+      if (Renderer) {
+        return /* @__PURE__ */ jsx24(
+          ToolMotion,
+          {
+            motionKey: `artifact-${artifact.type}`,
+            className: "aui-tool-artifact",
+            children: /* @__PURE__ */ jsx24(Renderer, { artifact })
+          }
+        );
+      }
+    }
+  }
+  return /* @__PURE__ */ jsx24(ToolFallback, { ...props });
+};
+
+// src/components/composer.tsx
+import {
+  AuiIf,
+  ComposerPrimitive as ComposerPrimitive2,
+  useComposerRuntime
+} from "@assistant-ui/react";
+import { ArrowUpIcon, SquareIcon } from "lucide-react";
+import { Fragment as Fragment2, jsx as jsx25, jsxs as jsxs13 } from "react/jsx-runtime";
+var Composer = ({
+  placeholder = "Send a message...",
+  showAttachments = true,
+  toolbar,
+  sendTooltip = "Send message",
+  noAutoFocus,
+  className
+}) => {
+  return /* @__PURE__ */ jsx25(
+    ComposerPrimitive2.Root,
+    {
+      className: cn(
+        "aui-composer-root relative flex w-full flex-col",
+        className
+      ),
+      children: /* @__PURE__ */ jsxs13(
+        ComposerPrimitive2.AttachmentDropzone,
+        {
+          className: cn(
+            studioComposeInputShellClass,
+            "data-[dragging=true]:border-2 data-[dragging=true]:border-dashed data-[dragging=true]:border-primary data-[dragging=true]:bg-accent/50"
+          ),
+          children: [
+            showAttachments && /* @__PURE__ */ jsx25(ComposerAttachments, {}),
+            /* @__PURE__ */ jsx25(ComposerInput, { placeholder, autoFocus: !noAutoFocus }),
+            /* @__PURE__ */ jsx25(
+              ComposerToolbar,
+              {
+                showAttachments,
+                toolbar,
+                sendTooltip
+              }
+            )
+          ]
+        }
+      )
+    }
+  );
+};
+var ComposerInput = ({
+  placeholder,
+  autoFocus
+}) => {
+  const composer = useComposerRuntime();
+  const onKeyDown = (e) => {
+    if (e.key === "Enter" && !e.shiftKey && !e.nativeEvent.isComposing) {
+      e.preventDefault();
+      composer.send();
+    }
+  };
+  const onInput = (e) => {
+    const el = e.currentTarget;
+    el.style.height = "auto";
+    el.style.height = `${Math.min(el.scrollHeight, 240)}px`;
+  };
+  return /* @__PURE__ */ jsx25(
+    ComposerPrimitive2.Input,
+    {
+      placeholder,
+      className: "aui-composer-input max-h-60 min-h-14 w-full resize-none bg-transparent px-3 pt-3 pb-1 text-sm outline-none placeholder:text-neutral-400 focus-visible:ring-0 dark:placeholder:text-neutral-500",
+      rows: 1,
+      autoFocus,
+      "aria-label": "Message input",
+      onKeyDown,
+      onInput
+    }
+  );
+};
+var ComposerToolbar = ({ showAttachments, toolbar, sendTooltip }) => {
+  return /* @__PURE__ */ jsxs13("div", { className: "aui-composer-action-wrapper flex items-center justify-between gap-1 px-2.5 pb-2.5", children: [
+    /* @__PURE__ */ jsxs13("div", { className: "flex items-center gap-1", children: [
+      showAttachments && /* @__PURE__ */ jsx25(ComposerAddAttachment, {}),
+      toolbar
+    ] }),
+    /* @__PURE__ */ jsx25(ComposerSendOrCancel, { sendTooltip })
+  ] });
+};
+var ComposerSendOrCancel = ({ sendTooltip }) => {
+  return /* @__PURE__ */ jsxs13(Fragment2, { children: [
+    /* @__PURE__ */ jsx25(AuiIf, { condition: (s) => !s.thread.isRunning, children: /* @__PURE__ */ jsx25(ComposerPrimitive2.Send, { asChild: true, children: /* @__PURE__ */ jsx25(
+      TooltipIconButton,
+      {
+        tooltip: sendTooltip,
+        variant: "primary",
+        type: "submit",
+        className: "aui-composer-send shrink-0 disabled:opacity-30",
+        "aria-label": "Send message",
+        children: /* @__PURE__ */ jsx25(ArrowUpIcon, { className: "aui-composer-send-icon size-4" })
+      }
+    ) }) }),
+    /* @__PURE__ */ jsx25(AuiIf, { condition: (s) => s.thread.isRunning, children: /* @__PURE__ */ jsx25(ComposerPrimitive2.Cancel, { asChild: true, children: /* @__PURE__ */ jsx25(
+      TooltipIconButton,
+      {
+        tooltip: "Stop generating",
+        variant: "primary",
+        className: "aui-composer-cancel shrink-0",
+        "aria-label": "Stop generating",
+        children: /* @__PURE__ */ jsx25(SquareIcon, { className: "aui-composer-cancel-icon size-3 fill-current" })
+      }
+    ) }) })
+  ] });
+};
+
+// src/components/suggestions.tsx
+import {
+  useEffect as useEffect4,
+  useMemo as useMemo6,
+  useState as useState7
 } from "react";
-import { jsx as jsx10 } from "react/jsx-runtime";
-var ShimmerComponent = ({
-  children,
-  as: Component = "p",
-  className,
-  duration = 2,
-  spread = 2
+import { useThreadRuntime as useThreadRuntime3 } from "@assistant-ui/react";
+import { ArrowUpIcon as ArrowUpIcon2 } from "lucide-react";
+import { jsx as jsx26, jsxs as jsxs14 } from "react/jsx-runtime";
+var Suggestions = ({
+  suggestions,
+  className
 }) => {
-  const MotionComponent = motion.create(
-    Component
-  );
-  const dynamicSpread = useMemo(
-    () => (children?.length ?? 0) * spread,
-    [children, spread]
-  );
-  return /* @__PURE__ */ jsx10(
-    MotionComponent,
+  const items = useResolvedSuggestions(suggestions);
+  if (!items || items.length === 0) return null;
+  return /* @__PURE__ */ jsx26(
+    "div",
     {
-      animate: { backgroundPosition: "0% center" },
       className: cn(
-        "relative inline-block bg-[length:250%_100%,auto] bg-clip-text text-transparent",
-        "[--bg:linear-gradient(90deg,#0000_calc(50%-var(--spread)),var(--color-background),#0000_calc(50%+var(--spread)))] [background-repeat:no-repeat,padding-box]",
+        "aui-thread-suggestions flex w-full flex-col gap-2 pb-2.5",
         className
       ),
-      initial: { backgroundPosition: "100% center" },
-      style: {
-        "--spread": `${dynamicSpread}px`,
-        backgroundImage: "var(--bg), linear-gradient(var(--color-muted-foreground), var(--color-muted-foreground))"
-      },
-      transition: {
-        repeat: Number.POSITIVE_INFINITY,
-        duration,
-        ease: "linear"
-      },
-      children
+      role: "list",
+      "aria-label": "Suggested prompts",
+      children: items.map((suggestion, i) => /* @__PURE__ */ jsx26(SuggestionRow, { suggestion }, (suggestion.prompt ?? suggestion.title) + i))
     }
   );
 };
-var Shimmer = memo2(ShimmerComponent);
-
-// src/components/tool-fallback.tsx
-import { jsx as jsx11, jsxs as jsxs6 } from "react/jsx-runtime";
-var ToolFallbackImpl = ({
-  toolName,
-  status
-}) => {
-  if (status?.type !== "running") return null;
-  return /* @__PURE__ */ jsxs6("div", { className: "flex items-center gap-2 py-1 text-sm text-muted-foreground", children: [
-    /* @__PURE__ */ jsx11(WrenchIcon, { className: "size-4" }),
-    /* @__PURE__ */ jsx11(Shimmer, { as: "span", duration: 1.8, spread: 2.5, children: `Using tool: ${toolName}` })
-  ] });
+var SuggestionRow = ({ suggestion }) => {
+  const runtime = useThreadRuntime3();
+  const onClick = () => {
+    const text = suggestion.prompt ?? suggestion.title;
+    runtime.append({ role: "user", content: [{ type: "text", text }] });
+  };
+  return /* @__PURE__ */ jsxs14(
+    "button",
+    {
+      type: "button",
+      role: "listitem",
+      onClick,
+      className: cn("aui-thread-suggestion", studioListRowButtonClass),
+      children: [
+        /* @__PURE__ */ jsx26("span", { className: "aui-thread-suggestion-icon shrink-0 text-neutral-500 dark:text-muted-foreground", children: suggestion.icon ?? /* @__PURE__ */ jsx26(ArrowUpIcon2, { className: "size-4", strokeWidth: 1.75, "aria-hidden": true }) }),
+        /* @__PURE__ */ jsxs14("span", { className: "aui-thread-suggestion-text min-w-0 flex-1 text-left", children: [
+          /* @__PURE__ */ jsx26("span", { className: "aui-thread-suggestion-text-1 block truncate text-sm font-normal text-foreground dark:text-foreground/95", children: suggestion.title }),
+          suggestion.description && /* @__PURE__ */ jsx26("span", { className: "aui-thread-suggestion-text-2 mt-0.5 block truncate text-xs text-muted-foreground", children: suggestion.description })
+        ] })
+      ]
+    }
+  );
 };
-var ToolFallback = memo3(
-  ToolFallbackImpl
-);
-ToolFallback.displayName = "ToolFallback";
+function useResolvedSuggestions(source) {
+  const [resolved, setResolved] = useState7(
+    () => Array.isArray(source) ? source : void 0
+  );
+  useEffect4(() => {
+    if (!source) {
+      setResolved(void 0);
+      return;
+    }
+    if (Array.isArray(source)) {
+      setResolved(source);
+      return;
+    }
+    let cancelled = false;
+    Promise.resolve().then(() => source()).then((value) => {
+      if (!cancelled) setResolved(value);
+    }).catch(() => {
+      if (!cancelled) setResolved([]);
+    });
+    return () => {
+      cancelled = true;
+    };
+  }, [source]);
+  return useMemo6(() => resolved, [resolved]);
+}
 
 // src/components/thread.tsx
-import {
-  ActionBarMorePrimitive,
-  ActionBarPrimitive,
-  AuiIf,
-  ComposerPrimitive as ComposerPrimitive2,
-  ErrorPrimitive,
-  MessagePrimitive as MessagePrimitive2,
-  ThreadPrimitive,
-  useThreadRuntime
-} from "@assistant-ui/react";
-import {
-  ArrowDownIcon,
-  ArrowUpIcon,
-  CheckIcon as CheckIcon2,
-  CopyIcon as CopyIcon2,
-  DownloadIcon,
-  MoreHorizontalIcon,
-  PencilIcon,
-  RefreshCwIcon,
-  SquareIcon
-} from "lucide-react";
-import { jsx as jsx12, jsxs as jsxs7 } from "react/jsx-runtime";
+import { jsx as jsx27, jsxs as jsxs15 } from "react/jsx-runtime";
 var Thread = ({
   className,
   maxWidth = "44rem",
   welcome,
   suggestions,
   composerPlaceholder = "Send a message...",
-  components
+  components,
+  artifacts,
+  onArtifactEvent
 }) => {
   const WelcomeSlot = components?.Welcome ?? ThreadWelcome;
   const ComposerSlot = components?.Composer ?? Composer;
@@ -1247,202 +3521,217 @@ var Thread = ({
   const AssistantMessageSlot = components?.AssistantMessage ?? AssistantMessage;
   const EditComposerSlot = components?.EditComposer ?? EditComposer;
   const ScrollToBottomSlot = components?.ScrollToBottom ?? ThreadScrollToBottom;
-  return /* @__PURE__ */ jsx12(
-    ThreadPrimitive.Root,
+  const SuggestionsSlot = components?.Suggestions ?? Suggestions;
+  return /* @__PURE__ */ jsx27(
+    ArtifactRegistryProvider,
     {
-      className: cn(
-        "aui-root aui-thread-root @container flex h-full flex-col bg-background",
-        className
-      ),
-      style: { ["--thread-max-width"]: maxWidth },
-      children: /* @__PURE__ */ jsxs7(
-        ThreadPrimitive.Viewport,
+      renderers: artifacts?.renderers,
+      override: artifacts?.override,
+      children: /* @__PURE__ */ jsx27(UiEventProvider, { onEvent: onArtifactEvent ?? (() => {
+      }), children: /* @__PURE__ */ jsx27(
+        ThreadPrimitive.Root,
         {
-          turnAnchor: "bottom",
-          className: "aui-thread-viewport relative flex flex-1 flex-col overflow-x-auto overflow-y-scroll px-4 pt-4",
-          children: [
-            /* @__PURE__ */ jsx12(WelcomeSlot, { config: welcome, suggestions }),
-            /* @__PURE__ */ jsx12(
-              ThreadPrimitive.Messages,
-              {
-                components: {
-                  UserMessage: UserMessageSlot,
-                  EditComposer: EditComposerSlot,
-                  AssistantMessage: AssistantMessageSlot
-                }
-              }
-            ),
-            /* @__PURE__ */ jsxs7(ThreadPrimitive.ViewportFooter, { className: "aui-thread-viewport-footer sticky bottom-0 mx-auto mt-auto flex w-full max-w-(--thread-max-width) flex-col gap-4 overflow-visible rounded-t-3xl bg-background pb-4 md:pb-6", children: [
-              /* @__PURE__ */ jsx12(ScrollToBottomSlot, {}),
-              /* @__PURE__ */ jsx12(ComposerSlot, { placeholder: composerPlaceholder })
-            ] })
-          ]
+          className: cn(
+            "aui-root aui-thread-root @container flex h-full flex-col bg-background",
+            className
+          ),
+          style: { ["--thread-max-width"]: maxWidth },
+          children: /* @__PURE__ */ jsxs15(
+            ThreadPrimitive.Viewport,
+            {
+              turnAnchor: "bottom",
+              className: "aui-thread-viewport relative flex flex-1 flex-col overflow-x-auto overflow-y-scroll px-4 pt-4",
+              children: [
+                /* @__PURE__ */ jsx27(
+                  WelcomeSlot,
+                  {
+                    config: welcome,
+                    suggestions,
+                    Suggestions: SuggestionsSlot
+                  }
+                ),
+                /* @__PURE__ */ jsx27(
+                  ThreadPrimitive.Messages,
+                  {
+                    components: {
+                      UserMessage: UserMessageSlot,
+                      EditComposer: EditComposerSlot,
+                      AssistantMessage: AssistantMessageSlot
+                    }
+                  }
+                ),
+                /* @__PURE__ */ jsxs15(ThreadPrimitive.ViewportFooter, { className: "aui-thread-viewport-footer sticky bottom-0 mx-auto mt-auto flex w-full max-w-(--thread-max-width) flex-col gap-4 overflow-visible rounded-t-3xl bg-background pb-4 md:pb-6", children: [
+                  /* @__PURE__ */ jsx27(ScrollToBottomSlot, {}),
+                  /* @__PURE__ */ jsx27(ComposerSlot, { placeholder: composerPlaceholder })
+                ] })
+              ]
+            }
+          )
         }
-      )
+      ) })
     }
   );
 };
 var ThreadScrollToBottom = () => {
-  return /* @__PURE__ */ jsx12(ThreadPrimitive.ScrollToBottom, { asChild: true, children: /* @__PURE__ */ jsx12(
+  return /* @__PURE__ */ jsx27(ThreadPrimitive.ScrollToBottom, { asChild: true, children: /* @__PURE__ */ jsx27(
     TooltipIconButton,
     {
       tooltip: "Scroll to bottom",
-      variant: "outline",
-      className: "aui-thread-scroll-to-bottom absolute -top-12 z-10 self-center rounded-full p-4 disabled:invisible dark:bg-background dark:hover:bg-accent",
-      children: /* @__PURE__ */ jsx12(ArrowDownIcon, {})
+      variant: "secondary",
+      className: "aui-thread-scroll-to-bottom absolute -top-12 z-10 self-center disabled:invisible",
+      children: /* @__PURE__ */ jsx27(ArrowDownIcon, { className: "size-4" })
     }
   ) });
 };
-var ThreadWelcome = ({ config, suggestions }) => {
-  return /* @__PURE__ */ jsx12(AuiIf, { condition: (s) => s.thread.isEmpty, children: /* @__PURE__ */ jsxs7("div", { className: "aui-thread-welcome-root mx-auto my-auto flex w-full max-w-(--thread-max-width) grow flex-col", children: [
-    /* @__PURE__ */ jsx12("div", { className: "aui-thread-welcome-center flex w-full grow flex-col items-center justify-center", children: /* @__PURE__ */ jsxs7("div", { className: "aui-thread-welcome-message flex size-full flex-col items-center justify-center px-4 text-center", children: [
-      /* @__PURE__ */ jsxs7("div", { className: "fade-in animate-in fill-mode-both relative mb-6 flex size-14 items-center justify-center duration-300", children: [
-        /* @__PURE__ */ jsx12("div", { className: "animate-ai-ring-glow absolute inset-0 rounded-2xl bg-gradient-to-br from-primary/15 to-primary/5 ring-1 ring-primary/15" }),
-        /* @__PURE__ */ jsx12("div", { className: "animate-ai-pulse-ring absolute inset-0" }),
-        /* @__PURE__ */ jsx12(
-          "svg",
-          {
-            xmlns: "http://www.w3.org/2000/svg",
-            viewBox: "0 0 24 24",
-            fill: "none",
-            stroke: "currentColor",
-            strokeWidth: "1.5",
-            strokeLinecap: "round",
-            strokeLinejoin: "round",
-            className: "animate-ai-breathe relative size-7 text-primary/75",
-            children: /* @__PURE__ */ jsx12("path", { d: "M9.937 15.5A2 2 0 0 0 8.5 14.063l-6.135-1.582a.5.5 0 0 1 0-.962L8.5 9.936A2 2 0 0 0 9.937 8.5l1.582-6.135a.5.5 0 0 1 .963 0L14.063 8.5A2 2 0 0 0 15.5 9.937l6.135 1.581a.5.5 0 0 1 0 .964L15.5 14.063a2 2 0 0 0-1.437 1.437l-1.582 6.135a.5.5 0 0 1-.963 0z" })
-          }
-        )
-      ] }),
-      /* @__PURE__ */ jsx12("h1", { className: "aui-thread-welcome-message-inner fade-in slide-in-from-bottom-1 animate-in fill-mode-both font-semibold text-2xl duration-200", children: config?.heading ?? "How can I help you today?" }),
-      /* @__PURE__ */ jsx12("p", { className: "aui-thread-welcome-message-inner fade-in slide-in-from-bottom-1 animate-in fill-mode-both text-muted-foreground mt-2 delay-75 duration-200", children: config?.subheading ?? "Send a message to start a conversation." })
-    ] }) }),
-    suggestions && suggestions.length > 0 && /* @__PURE__ */ jsx12(ThreadSuggestions, { suggestions })
-  ] }) });
-};
-var ThreadSuggestions = ({ suggestions }) => {
-  return /* @__PURE__ */ jsx12("div", { className: "aui-thread-welcome-suggestions grid w-full @md:grid-cols-2 gap-2 pb-4", children: suggestions.map((s, i) => /* @__PURE__ */ jsx12(ThreadSuggestionItem, { title: s.title, description: s.description }, i)) });
+var welcomeStagger = {
+  initial: {},
+  animate: {
+    transition: { staggerChildren: 0.16, delayChildren: 0.12 }
+  }
 };
-var ThreadSuggestionItem = ({ title, description }) => {
-  const runtime = useThreadRuntime();
-  return /* @__PURE__ */ jsx12("div", { className: "aui-thread-welcome-suggestion-display fade-in slide-in-from-bottom-2 animate-in fill-mode-both duration-200", children: /* @__PURE__ */ jsxs7(
-    Button,
-    {
-      variant: "ghost",
-      className: "aui-thread-welcome-suggestion h-auto w-full @md:flex-col flex-wrap items-start justify-start gap-1 rounded-2xl border px-4 py-3 text-left text-sm transition-colors hover:bg-muted",
-      onClick: () => runtime.append({
-        role: "user",
-        content: [{ type: "text", text: title }]
-      }),
-      children: [
-        /* @__PURE__ */ jsx12("span", { className: "aui-thread-welcome-suggestion-text-1 font-medium", children: title }),
-        description && /* @__PURE__ */ jsx12("span", { className: "aui-thread-welcome-suggestion-text-2 text-muted-foreground", children: description })
-      ]
-    }
-  ) });
+var welcomeItem = {
+  initial: { opacity: 0, y: 14 },
+  animate: {
+    opacity: 1,
+    y: 0,
+    transition: { duration: 0.9, ease: luxuryEase }
+  }
 };
-var Composer = ({ placeholder }) => {
-  return /* @__PURE__ */ jsx12(ComposerPrimitive2.Root, { className: "aui-composer-root relative mt-3 flex w-full flex-col", children: /* @__PURE__ */ jsxs7(ComposerPrimitive2.AttachmentDropzone, { className: "aui-composer-attachment-dropzone flex w-full flex-col rounded-2xl border border-input bg-background px-1 pt-2 outline-none transition-shadow has-[textarea:focus-visible]:border-ring has-[textarea:focus-visible]:ring-2 has-[textarea:focus-visible]:ring-ring/20 data-[dragging=true]:border-ring data-[dragging=true]:border-dashed data-[dragging=true]:bg-accent/50", children: [
-    /* @__PURE__ */ jsx12(ComposerAttachments, {}),
-    /* @__PURE__ */ jsx12(
-      ComposerPrimitive2.Input,
-      {
-        placeholder: placeholder ?? "Send a message...",
-        className: "aui-composer-input mb-1 max-h-32 min-h-14 w-full resize-none bg-transparent px-4 pt-2 pb-3 text-sm outline-none placeholder:text-muted-foreground focus-visible:ring-0",
-        rows: 1,
-        autoFocus: true,
-        "aria-label": "Message input"
-      }
-    ),
-    /* @__PURE__ */ jsx12(ComposerAction, {})
-  ] }) });
+var welcomeIcon = {
+  initial: { opacity: 0, y: 10, scale: 0.96 },
+  animate: {
+    opacity: 1,
+    y: 0,
+    scale: 1,
+    transition: { duration: 1.1, ease: luxuryEase }
+  }
 };
-var ComposerAction = () => {
-  return /* @__PURE__ */ jsxs7("div", { className: "aui-composer-action-wrapper relative mx-2 mb-2 flex items-center justify-end", children: [
-    /* @__PURE__ */ jsx12(AuiIf, { condition: (s) => !s.thread.isRunning, children: /* @__PURE__ */ jsx12(ComposerPrimitive2.Send, { asChild: true, children: /* @__PURE__ */ jsx12(
-      TooltipIconButton,
-      {
-        tooltip: "Send message",
-        side: "bottom",
-        type: "submit",
-        variant: "default",
-        size: "icon",
-        className: "aui-composer-send size-8 rounded-full",
-        "aria-label": "Send message",
-        children: /* @__PURE__ */ jsx12(ArrowUpIcon, { className: "aui-composer-send-icon size-4" })
-      }
-    ) }) }),
-    /* @__PURE__ */ jsx12(AuiIf, { condition: (s) => s.thread.isRunning, children: /* @__PURE__ */ jsx12(ComposerPrimitive2.Cancel, { asChild: true, children: /* @__PURE__ */ jsx12(
-      Button,
+var ThreadWelcome = ({
+  config,
+  suggestions,
+  Suggestions: SuggestionsSlot = Suggestions
+}) => {
+  const isEmpty = useThread((s) => s.messages.length === 0);
+  if (!isEmpty) return null;
+  return /* @__PURE__ */ jsxs15("div", { className: "aui-thread-welcome-root mx-auto my-auto flex w-full max-w-(--thread-max-width) grow flex-col", children: [
+    /* @__PURE__ */ jsx27("div", { className: "aui-thread-welcome-center flex w-full grow flex-col items-center justify-center", children: /* @__PURE__ */ jsxs15(
+      motion4.div,
       {
-        type: "button",
-        variant: "default",
-        size: "icon",
-        className: "aui-composer-cancel size-8 rounded-full",
-        "aria-label": "Stop generating",
-        children: /* @__PURE__ */ jsx12(SquareIcon, { className: "aui-composer-cancel-icon size-3 fill-current" })
+        className: "aui-thread-welcome-message flex flex-col items-center justify-center px-4 text-center",
+        variants: welcomeStagger,
+        initial: "initial",
+        animate: "animate",
+        children: [
+          config?.icon && /* @__PURE__ */ jsx27(motion4.div, { variants: welcomeIcon, className: "mb-5", children: config.icon }),
+          /* @__PURE__ */ jsx27(
+            motion4.h1,
+            {
+              variants: welcomeItem,
+              className: "aui-thread-welcome-message-inner font-semibold text-2xl",
+              children: config?.heading ?? "How can I help you today?"
+            }
+          ),
+          /* @__PURE__ */ jsx27(
+            motion4.p,
+            {
+              variants: welcomeItem,
+              className: "aui-thread-welcome-message-inner mt-2 text-muted-foreground",
+              children: config?.subheading ?? "Send a message to start a conversation."
+            }
+          )
+        ]
       }
-    ) }) })
+    ) }),
+    suggestions && /* @__PURE__ */ jsx27("div", { className: "aui-thread-welcome-suggestions mx-auto w-full max-w-(--thread-max-width) px-2", children: /* @__PURE__ */ jsx27(SuggestionsSlot, { suggestions }) })
   ] });
 };
 var MessageError = () => {
-  return /* @__PURE__ */ jsx12(MessagePrimitive2.Error, { children: /* @__PURE__ */ jsx12(ErrorPrimitive.Root, { className: "aui-message-error-root mt-2 rounded-md border border-destructive bg-destructive/10 p-3 text-destructive text-sm dark:bg-destructive/5 dark:text-red-200", children: /* @__PURE__ */ jsx12(ErrorPrimitive.Message, { className: "aui-message-error-message line-clamp-2" }) }) });
+  return /* @__PURE__ */ jsx27(MessagePrimitive2.Error, { children: /* @__PURE__ */ jsx27(ErrorPrimitive.Root, { className: "aui-message-error-root mt-2 rounded-md border border-destructive bg-destructive/10 p-3 text-destructive text-sm dark:bg-destructive/5 dark:text-red-200", children: /* @__PURE__ */ jsx27(ErrorPrimitive.Message, { className: "aui-message-error-message line-clamp-2" }) }) });
 };
 var AssistantMessage = () => {
-  return /* @__PURE__ */ jsxs7(
+  return /* @__PURE__ */ jsxs15(
     MessagePrimitive2.Root,
     {
       className: "aui-assistant-message-root fade-in slide-in-from-bottom-1 relative mx-auto w-full max-w-(--thread-max-width) animate-in py-3 duration-150",
       "data-role": "assistant",
       children: [
-        /* @__PURE__ */ jsxs7("div", { className: "aui-assistant-message-content wrap-break-word px-2 text-foreground leading-relaxed", children: [
-          /* @__PURE__ */ jsx12(
+        /* @__PURE__ */ jsxs15("div", { className: "aui-assistant-message-content wrap-break-word px-2 text-foreground leading-relaxed", children: [
+          /* @__PURE__ */ jsx27(
             MessagePrimitive2.Parts,
             {
               components: {
                 Text: MarkdownText,
-                tools: { Fallback: ToolFallback }
+                // `Override` (not `Fallback`) replaces the default tool renderer
+                // entirely so we never fall back to the assistant-ui boilerplate.
+                tools: { Override: ToolArtifactFallback }
               }
             }
           ),
-          /* @__PURE__ */ jsx12(MessageError, {})
+          /* @__PURE__ */ jsx27(MessageError, {})
         ] }),
-        /* @__PURE__ */ jsx12("div", { className: "aui-assistant-message-footer mt-1 ml-2 flex", children: /* @__PURE__ */ jsx12(AssistantActionBar, {}) })
+        /* @__PURE__ */ jsx27("div", { className: "aui-assistant-message-footer mt-0 ml-1 flex", children: /* @__PURE__ */ jsx27(AssistantActionBar, {}) })
       ]
     }
   );
 };
+var ASSISTANT_ACTION_ICON_CLASS = cn(
+  "size-6 min-h-6 min-w-6 text-muted-foreground/45 hover:text-muted-foreground/80",
+  // The v2 fill span sits inside `group/tbv2 > span:first-child`. We mute it
+  // here so action-bar buttons read as subtle icons rather than full pills.
+  "[&>span:first-child]:bg-transparent",
+  "[&>span:first-child]:group-hover/tbv2:bg-neutral-100/50",
+  "dark:[&>span:first-child]:group-hover/tbv2:bg-white/8"
+);
 var AssistantActionBar = () => {
-  return /* @__PURE__ */ jsxs7(
+  return /* @__PURE__ */ jsxs15(
     ActionBarPrimitive.Root,
     {
       hideWhenRunning: true,
       autohide: "not-last",
-      autohideFloat: "single-branch",
-      className: "aui-assistant-action-bar-root col-start-3 row-start-2 -ml-1 flex gap-1 text-muted-foreground data-floating:absolute data-floating:rounded-md data-floating:border data-floating:bg-background data-floating:p-1 data-floating:shadow-sm",
+      className: "aui-assistant-action-bar-root flex items-center gap-0 bg-transparent px-0 py-0.5 text-muted-foreground/60",
       children: [
-        /* @__PURE__ */ jsx12(ActionBarPrimitive.Copy, { asChild: true, children: /* @__PURE__ */ jsxs7(TooltipIconButton, { tooltip: "Copy", children: [
-          /* @__PURE__ */ jsx12(AuiIf, { condition: (s) => s.message.isCopied, children: /* @__PURE__ */ jsx12(CheckIcon2, {}) }),
-          /* @__PURE__ */ jsx12(AuiIf, { condition: (s) => !s.message.isCopied, children: /* @__PURE__ */ jsx12(CopyIcon2, {}) })
-        ] }) }),
-        /* @__PURE__ */ jsx12(ActionBarPrimitive.Reload, { asChild: true, children: /* @__PURE__ */ jsx12(TooltipIconButton, { tooltip: "Refresh", children: /* @__PURE__ */ jsx12(RefreshCwIcon, {}) }) }),
-        /* @__PURE__ */ jsxs7(ActionBarMorePrimitive.Root, { children: [
-          /* @__PURE__ */ jsx12(ActionBarMorePrimitive.Trigger, { asChild: true, children: /* @__PURE__ */ jsx12(
+        /* @__PURE__ */ jsx27(ActionBarPrimitive.Copy, { asChild: true, children: /* @__PURE__ */ jsxs15(
+          TooltipIconButton,
+          {
+            tooltip: "Copy",
+            variant: "ghost",
+            className: ASSISTANT_ACTION_ICON_CLASS,
+            children: [
+              /* @__PURE__ */ jsx27(AuiIf2, { condition: (s) => s.message.isCopied, children: /* @__PURE__ */ jsx27(CheckIcon3, { className: "size-3" }) }),
+              /* @__PURE__ */ jsx27(AuiIf2, { condition: (s) => !s.message.isCopied, children: /* @__PURE__ */ jsx27(CopyIcon2, { className: "size-3" }) })
+            ]
+          }
+        ) }),
+        /* @__PURE__ */ jsx27(ActionBarPrimitive.Reload, { asChild: true, children: /* @__PURE__ */ jsx27(
+          TooltipIconButton,
+          {
+            tooltip: "Regenerate",
+            variant: "ghost",
+            className: ASSISTANT_ACTION_ICON_CLASS,
+            children: /* @__PURE__ */ jsx27(RefreshCwIcon, { className: "size-3" })
+          }
+        ) }),
+        /* @__PURE__ */ jsxs15(ActionBarMorePrimitive.Root, { children: [
+          /* @__PURE__ */ jsx27(ActionBarMorePrimitive.Trigger, { asChild: true, children: /* @__PURE__ */ jsx27(
             TooltipIconButton,
             {
               tooltip: "More",
-              className: "data-[state=open]:bg-accent",
-              children: /* @__PURE__ */ jsx12(MoreHorizontalIcon, {})
+              variant: "ghost",
+              className: cn(
+                ASSISTANT_ACTION_ICON_CLASS,
+                "data-[state=open]:text-muted-foreground/80"
+              ),
+              children: /* @__PURE__ */ jsx27(MoreHorizontalIcon, { className: "size-3" })
             }
           ) }),
-          /* @__PURE__ */ jsx12(
+          /* @__PURE__ */ jsx27(
             ActionBarMorePrimitive.Content,
             {
               side: "bottom",
               align: "start",
-              className: "aui-action-bar-more-content z-50 min-w-32 overflow-hidden rounded-md border bg-popover p-1 text-popover-foreground shadow-md",
-              children: /* @__PURE__ */ jsx12(ActionBarPrimitive.ExportMarkdown, { asChild: true, children: /* @__PURE__ */ jsxs7(ActionBarMorePrimitive.Item, { className: "aui-action-bar-more-item flex cursor-pointer select-none items-center gap-2 rounded-sm px-2 py-1.5 text-sm outline-none hover:bg-accent hover:text-accent-foreground focus:bg-accent focus:text-accent-foreground", children: [
-                /* @__PURE__ */ jsx12(DownloadIcon, { className: "size-4" }),
+              className: "aui-action-bar-more-content z-50 min-w-36 overflow-hidden rounded-lg border border-neutral-200 bg-white p-1 text-foreground shadow-md dark:border-white/10 dark:bg-zinc-900",
+              children: /* @__PURE__ */ jsx27(ActionBarPrimitive.ExportMarkdown, { asChild: true, children: /* @__PURE__ */ jsxs15(ActionBarMorePrimitive.Item, { className: "aui-action-bar-more-item flex cursor-pointer select-none items-center gap-2 rounded-md px-2 py-1.5 text-sm outline-none hover:bg-neutral-100 focus:bg-neutral-100 dark:hover:bg-zinc-800 dark:focus:bg-zinc-800", children: [
+                /* @__PURE__ */ jsx27(DownloadIcon, { className: "size-4 shrink-0" }),
                 "Export as Markdown"
               ] }) })
             }
@@ -1452,81 +3741,440 @@ var AssistantActionBar = () => {
     }
   );
 };
+var UserMessageText = () => {
+  return /* @__PURE__ */ jsx27("span", { className: "whitespace-pre-wrap", children: /* @__PURE__ */ jsx27(MessagePartPrimitive.Text, { smooth: false }) });
+};
 var UserMessage = () => {
-  return /* @__PURE__ */ jsxs7(
+  return /* @__PURE__ */ jsxs15(
     MessagePrimitive2.Root,
     {
-      className: "aui-user-message-root fade-in slide-in-from-bottom-1 mx-auto grid w-full max-w-(--thread-max-width) animate-in auto-rows-auto grid-cols-[minmax(72px,1fr)_auto] content-start gap-y-2 px-2 py-3 duration-150 [&:where(>*)]:col-start-2",
+      className: "aui-user-message-root mx-auto flex w-full max-w-(--thread-max-width) flex-col items-end gap-2 px-2 py-3",
       "data-role": "user",
       children: [
-        /* @__PURE__ */ jsx12(UserMessageAttachments, {}),
-        /* @__PURE__ */ jsxs7("div", { className: "aui-user-message-content-wrapper relative col-start-2 min-w-0", children: [
-          /* @__PURE__ */ jsx12("div", { className: "aui-user-message-content wrap-break-word rounded-2xl bg-muted px-4 py-2.5 text-foreground", children: /* @__PURE__ */ jsx12(MessagePrimitive2.Parts, {}) }),
-          /* @__PURE__ */ jsx12("div", { className: "aui-user-action-bar-wrapper absolute top-1/2 left-0 -translate-x-full -translate-y-1/2 pr-2", children: /* @__PURE__ */ jsx12(UserActionBar, {}) })
-        ] })
+        /* @__PURE__ */ jsx27(UserMessageAttachments, {}),
+        /* @__PURE__ */ jsxs15(
+          motion4.div,
+          {
+            className: "aui-user-message-content relative inline-block max-w-[80%] rounded-2xl bg-neutral-200 px-4 py-2.5 text-foreground dark:bg-neutral-700",
+            initial: { opacity: 0, y: 8, scale: 0.99 },
+            animate: { opacity: 1, y: 0, scale: 1 },
+            transition: { duration: 0.65, ease: luxuryEase },
+            children: [
+              /* @__PURE__ */ jsx27(MessagePrimitive2.Parts, { components: { Text: UserMessageText } }),
+              /* @__PURE__ */ jsx27("div", { className: "aui-user-action-bar-wrapper absolute top-1/2 left-0 -translate-x-full -translate-y-1/2 pr-2", children: /* @__PURE__ */ jsx27(UserActionBar, {}) })
+            ]
+          }
+        )
       ]
     }
   );
 };
 var UserActionBar = () => {
-  return /* @__PURE__ */ jsx12(
+  return /* @__PURE__ */ jsx27(
     ActionBarPrimitive.Root,
     {
       hideWhenRunning: true,
       autohide: "not-last",
       className: "aui-user-action-bar-root flex flex-col items-end",
-      children: /* @__PURE__ */ jsx12(ActionBarPrimitive.Edit, { asChild: true, children: /* @__PURE__ */ jsx12(TooltipIconButton, { tooltip: "Edit", className: "aui-user-action-edit p-4", children: /* @__PURE__ */ jsx12(PencilIcon, {}) }) })
+      children: /* @__PURE__ */ jsx27(ActionBarPrimitive.Edit, { asChild: true, children: /* @__PURE__ */ jsx27(
+        TooltipIconButton,
+        {
+          tooltip: "Edit",
+          variant: "ghost",
+          className: ASSISTANT_ACTION_ICON_CLASS,
+          children: /* @__PURE__ */ jsx27(PencilIcon, { className: "size-3" })
+        }
+      ) })
     }
   );
 };
 var EditComposer = () => {
-  return /* @__PURE__ */ jsx12(MessagePrimitive2.Root, { className: "aui-edit-composer-wrapper mx-auto flex w-full max-w-(--thread-max-width) flex-col px-2 py-3", children: /* @__PURE__ */ jsxs7(ComposerPrimitive2.Root, { className: "aui-edit-composer-root ml-auto flex w-full max-w-[85%] flex-col rounded-2xl bg-muted", children: [
-    /* @__PURE__ */ jsx12(
-      ComposerPrimitive2.Input,
+  return /* @__PURE__ */ jsx27(MessagePrimitive2.Root, { className: "aui-edit-composer-wrapper mx-auto flex w-full max-w-(--thread-max-width) flex-col px-2 py-3", children: /* @__PURE__ */ jsxs15(ComposerPrimitive3.Root, { className: "aui-edit-composer-root ml-auto flex w-full max-w-[85%] flex-col rounded-2xl bg-muted", children: [
+    /* @__PURE__ */ jsx27(
+      ComposerPrimitive3.Input,
       {
         className: "aui-edit-composer-input min-h-14 w-full resize-none bg-transparent p-4 text-foreground text-sm outline-none",
         autoFocus: true
       }
     ),
-    /* @__PURE__ */ jsxs7("div", { className: "aui-edit-composer-footer mx-3 mb-3 flex items-center gap-2 self-end", children: [
-      /* @__PURE__ */ jsx12(ComposerPrimitive2.Cancel, { asChild: true, children: /* @__PURE__ */ jsx12(Button, { variant: "ghost", size: "sm", children: "Cancel" }) }),
-      /* @__PURE__ */ jsx12(ComposerPrimitive2.Send, { asChild: true, children: /* @__PURE__ */ jsx12(Button, { size: "sm", children: "Update" }) })
+    /* @__PURE__ */ jsxs15("div", { className: "aui-edit-composer-footer mx-3 mb-3 flex items-center gap-2 self-end", children: [
+      /* @__PURE__ */ jsx27(ComposerPrimitive3.Cancel, { asChild: true, children: /* @__PURE__ */ jsx27(TimbalV2Button, { variant: "ghost", size: "sm", children: "Cancel" }) }),
+      /* @__PURE__ */ jsx27(ComposerPrimitive3.Send, { asChild: true, children: /* @__PURE__ */ jsx27(TimbalV2Button, { variant: "primary", size: "sm", children: "Update" }) })
     ] })
   ] }) });
 };
 
 // src/components/chat.tsx
-import { jsx as jsx13 } from "react/jsx-runtime";
+import { jsx as jsx28 } from "react/jsx-runtime";
 function TimbalChat({
   workforceId,
   baseUrl,
   fetch: fetch2,
+  attachments,
+  attachmentsUploadUrl,
+  attachmentsAccept,
+  debug,
   ...threadProps
 }) {
-  return /* @__PURE__ */ jsx13(TimbalRuntimeProvider, { workforceId, baseUrl, fetch: fetch2, children: /* @__PURE__ */ jsx13(Thread, { ...threadProps }) });
+  return /* @__PURE__ */ jsx28(
+    TimbalRuntimeProvider,
+    {
+      workforceId,
+      baseUrl,
+      fetch: fetch2,
+      attachments,
+      attachmentsUploadUrl,
+      attachmentsAccept,
+      debug,
+      children: /* @__PURE__ */ jsx28(Thread, { ...threadProps })
+    }
+  );
+}
+
+// src/artifacts/agent-instructions.ts
+var ARTIFACT_AGENT_INSTRUCTIONS = `
+## Rich artifacts (Timbal chat UI)
+
+When you need charts, tables, choice widgets, or interactive controls, return a **JSON artifact object** instead of plain prose. The chat UI renders these automatically.
+
+### Delivery channels (either works)
+
+1. **Tool result (preferred)** \u2014 return a single JSON object (or a JSON string) from a tool. The object must include a string \`type\` field.
+2. **Inline markdown fence** \u2014 embed the same JSON inside a fenced block:
+
+\`\`\`timbal-artifact
+{"type":"chart","data":[{"month":"Jan","sales":120}]}
+\`\`\`
+
+The alias \`\`\`timbal\`\`\` is also accepted.
+
+### Built-in artifact types
+
+| \`type\` | Use for |
+|---|---|
+| \`chart\` | Bar, line, area, or pie charts. Fields: \`data\`, optional \`chartType\`, \`xKey\`, \`dataKey\`, \`title\`, \`unit\`. |
+| \`table\` | Tabular data. Fields: \`rows\`, optional \`columns\`, \`title\`. |
+| \`question\` | In-thread multiple choice. Fields: \`options: [{ id, label, description? }]\`, optional \`prompt\`, \`multi\`. User replies are sent back as a normal user message. |
+| \`html\` | Custom HTML/CSS/JS in an iframe. Fields: \`content\` (HTML document or fragment), optional \`title\`, \`height\`, \`sandboxed\` (default \`true\`; set \`false\` for unrestricted scripts/CDN). |
+| \`json\` | Fallback structured view. Fields: \`data\`, optional \`title\`. |
+| \`ui\` | **Interactive UI** composed from a fixed node palette (hover, click, drag). See below. |
+
+### When to use \`type: "html"\`
+
+Use \`html\` when the user wants a **rich visual or interactive page** that does not fit the \`ui\` palette \u2014 e.g. styled layouts, SVG graphics, CSS animations, canvas, small games, calculators, or multi-section mockups. Pass a full HTML document or a body fragment in \`content\`.
+
+- Inline \`<style>\`, \`<script>\`, SVG, and canvas are supported.
+- Default \`sandboxed: true\` runs in an isolated iframe with scripts enabled.
+- Set \`sandboxed: false\` only for trusted content that needs external CDN scripts/styles or full DOM freedom.
+- Prefer \`ui\` when controls should send chat messages or host events; prefer \`html\` for self-contained mini-apps and visual demos.
+
+### When to use \`type: "ui"\`
+
+Use a \`ui\` artifact when the user should **hover, click, drag, or adjust controls** in-thread and those actions should integrate with the chat runtime (messages, \`onArtifactEvent\`). For standalone visual/interactive HTML, use \`html\` instead.
+
+Each \`ui\` artifact has:
+
+- \`initialState\` \u2014 optional object seeding local state (per widget instance).
+- \`root\` \u2014 a single node tree (see node kinds below).
+- optional \`title\` \u2014 card heading.
+
+**Bindings:** anywhere a primitive is accepted, you may use \`{ "$bind": "dotted.path" }\` to read from \`initialState\` / local state (e.g. \`{ "$bind": "qty" }\`).
+
+**Actions:** nodes may attach \`onClick\`, \`onChange\`, or \`onDragEnd\` with one action or an array:
+
+| Action | Shape | Effect |
+|---|---|---|
+| User message | \`{ "kind": "message", "text": "..." }\` or \`{ "kind": "message", "text": { "$bind": "path" } }\` | Sends text as the next user message. |
+| Set state | \`{ "kind": "set", "path": "foo", "value": 1 }\` | Writes local widget state. |
+| Toggle boolean | \`{ "kind": "toggle", "path": "enabled" }\` | Flips a boolean at \`path\`. |
+| Host event | \`{ "kind": "emit", "name": "event-name", "payload": { ... } }\` | Bubbles to the host app (\`onArtifactEvent\` on \`<Thread>\`). |
+
+### \`ui\` node palette (\`root.kind\`)
+
+| \`kind\` | Purpose | Key fields |
+|---|---|---|
+| \`box\` | Layout container | \`children\`, \`direction\` (\`row\`/\`col\`), \`gap\`, \`padding\`, \`align\`, \`justify\`, \`wrap\` |
+| \`text\` | Body text | \`value\`, optional \`muted\`, \`size\`, \`weight\` |
+| \`heading\` | Heading | \`value\`, optional \`level\` (1\u20134) |
+| \`badge\` | Pill label | \`value\`, optional \`tone\` (\`default\`, \`primary\`, \`success\`, \`warn\`, \`danger\`) |
+| \`button\` | Clickable button | \`label\`, optional \`variant\`, \`size\`, \`disabled\`, \`onClick\` |
+| \`toggle\` | Boolean switch | \`binding\` (state path), optional \`label\`, \`onChange\` |
+| \`slider\` | Numeric range | \`binding\`, optional \`min\`, \`max\`, \`step\`, \`label\`, \`showValue\`, \`onChange\` |
+| \`tooltip\` | Hover tooltip | \`content\`, \`child\` (single node), optional \`side\` |
+| \`draggable\` | Drag gesture | \`child\`, optional \`axis\` (\`x\`/\`y\`/\`both\`), \`snapBack\`, \`onDragEnd\` |
+| \`custom\` | Host-registered widget | \`name\`, optional \`props\`, \`children\` \u2014 only if the app registered that name |
+
+### Example \`ui\` artifact
+
+\`\`\`json
+{
+  "type": "ui",
+  "title": "Configure plan",
+  "initialState": { "qty": 1, "premium": false },
+  "root": {
+    "kind": "box",
+    "direction": "col",
+    "gap": 3,
+    "children": [
+      { "kind": "heading", "value": "Choose quantity", "level": 3 },
+      {
+        "kind": "tooltip",
+        "content": "Drag to adjust quantity",
+        "child": {
+          "kind": "slider",
+          "binding": "qty",
+          "min": 1,
+          "max": 50,
+          "label": "Quantity",
+          "onChange": { "kind": "emit", "name": "qty-changed" }
+        }
+      },
+      { "kind": "toggle", "binding": "premium", "label": "Premium support" },
+      {
+        "kind": "button",
+        "label": "Confirm",
+        "onClick": { "kind": "message", "text": { "$bind": "qty" } }
+      }
+    ]
+  }
 }
+\`\`\`
+
+### Rules
+
+- Always set \`type\` to a built-in value above unless the app documented a custom type.
+- Prefer \`ui\` over \`html\` when actions must bubble to the host chat (\`message\`, \`emit\`).
+- Prefer \`question\` for simple A/B/C choices; use \`ui\` when you need sliders, toggles, drag, or multi-control layouts.
+- Keep \`data\` arrays reasonably small (charts/tables).
+
+### After calling an artifact tool (critical)
+
+When you call a tool that returns an artifact (\`make_chart\`, \`ask_question\`, \`show_table\`, \`show_html\`, \`make_ui_demo\`, etc.):
+
+1. **Do not** paste, quote, paraphrase as JSON, or fence the tool return value in your assistant message. The chat UI already renders it from the tool result.
+2. **Do not** emit a matching \`\`\`timbal-artifact\`\`\` block for the same payload \u2014 pick **one** channel (tool result only).
+3. Your follow-up text should be **empty**, or at most **one short sentence** (e.g. "Pick an option above." / "Try the controls."). Never include \`type\`, \`options\`, \`data\`, or dict/JSON syntax.
+4. Treat the widget as visible to the user; refer to it as "above" / "the chart" / "the choices" \u2014 never reproduce its contents.
+`.trim();
 
 // src/index.ts
 import {
   ThreadPrimitive as ThreadPrimitive2,
   MessagePrimitive as MessagePrimitive3,
-  ComposerPrimitive as ComposerPrimitive3,
+  MessagePartPrimitive as MessagePartPrimitive2,
+  ComposerPrimitive as ComposerPrimitive4,
   ActionBarPrimitive as ActionBarPrimitive2,
-  useThread,
-  useThreadRuntime as useThreadRuntime2,
+  ActionBarMorePrimitive as ActionBarMorePrimitive2,
+  ErrorPrimitive as ErrorPrimitive2,
+  AuiIf as AuiIf3,
+  AssistantRuntimeProvider as AssistantRuntimeProvider2,
+  useThread as useThread2,
+  useThreadRuntime as useThreadRuntime4,
   useMessageRuntime,
-  useComposerRuntime
+  useComposerRuntime as useComposerRuntime2,
+  useAuiState as useAuiState3
 } from "@assistant-ui/react";
 
+// src/hooks/use-workforces.ts
+import { useEffect as useEffect5, useMemo as useMemo7, useRef as useRef2, useState as useState8 } from "react";
+function useWorkforces(options = {}) {
+  const { baseUrl = "/api", fetch: fetchFn, pickInitial } = options;
+  const [workforces, setWorkforces] = useState8([]);
+  const [selectedId, setSelectedId] = useState8("");
+  const [isLoading, setIsLoading] = useState8(true);
+  const [error, setError] = useState8(null);
+  const fetchFnRef = useRef2(fetchFn ?? authFetch);
+  useEffect5(() => {
+    fetchFnRef.current = fetchFn ?? authFetch;
+  }, [fetchFn]);
+  const pickInitialRef = useRef2(pickInitial);
+  useEffect5(() => {
+    pickInitialRef.current = pickInitial;
+  }, [pickInitial]);
+  const load = useMemo7(() => {
+    return async () => {
+      setIsLoading(true);
+      setError(null);
+      try {
+        const res = await fetchFnRef.current(`${baseUrl}/workforce`);
+        if (!res.ok) throw new Error(`Failed to load workforces (${res.status})`);
+        const data = await res.json();
+        setWorkforces(data);
+        setSelectedId((current) => {
+          if (current && data.some((w) => idOf(w) === current)) return current;
+          const initial = pickInitialRef.current?.(data) ?? data.find((w) => w.type === "agent") ?? data[0];
+          return initial ? idOf(initial) : "";
+        });
+      } catch (err) {
+        setError(err instanceof Error ? err : new Error(String(err)));
+      } finally {
+        setIsLoading(false);
+      }
+    };
+  }, [baseUrl]);
+  useEffect5(() => {
+    load();
+  }, [load]);
+  const selected = useMemo7(
+    () => workforces.find((w) => idOf(w) === selectedId),
+    [workforces, selectedId]
+  );
+  return {
+    workforces,
+    selectedId,
+    setSelectedId,
+    selected,
+    isLoading,
+    error,
+    refresh: load
+  };
+}
+function idOf(item) {
+  return item.id ?? item.uid ?? item.name ?? "";
+}
+
+// src/components/workforce-selector.tsx
+import { ChevronDownIcon } from "lucide-react";
+import { jsx as jsx29, jsxs as jsxs16 } from "react/jsx-runtime";
+var WorkforceSelector = ({
+  workforces,
+  value,
+  onChange,
+  hideWhenSingle = true,
+  className,
+  placeholder = "Select agent"
+}) => {
+  if (workforces.length === 0) return null;
+  if (hideWhenSingle && workforces.length === 1) return null;
+  return /* @__PURE__ */ jsxs16(
+    "div",
+    {
+      className: cn(
+        "aui-workforce-selector relative inline-flex items-center",
+        studioTopbarPillHeightClass,
+        studioSecondaryChromeClass,
+        "rounded-full",
+        className
+      ),
+      children: [
+        /* @__PURE__ */ jsxs16(
+          "select",
+          {
+            className: "aui-workforce-selector-input h-full cursor-pointer appearance-none rounded-full border-none bg-transparent pr-8 pl-3.5 text-sm font-medium text-foreground outline-none focus:outline-none",
+            value,
+            onChange: (e) => onChange(e.target.value),
+            "aria-label": placeholder,
+            children: [
+              !value && /* @__PURE__ */ jsx29("option", { value: "", children: placeholder }),
+              workforces.map((w) => {
+                const id = idOf2(w);
+                return /* @__PURE__ */ jsx29("option", { value: id, children: w.name ?? id }, id);
+              })
+            ]
+          }
+        ),
+        /* @__PURE__ */ jsx29(
+          ChevronDownIcon,
+          {
+            className: "aui-workforce-selector-icon pointer-events-none absolute right-3 size-3.5 text-muted-foreground/70",
+            "aria-hidden": true
+          }
+        )
+      ]
+    }
+  );
+};
+function idOf2(item) {
+  return item.id ?? item.uid ?? item.name ?? "";
+}
+
+// src/components/chat-shell.tsx
+import { jsx as jsx30, jsxs as jsxs17 } from "react/jsx-runtime";
+var TimbalChatShell = ({
+  workforceId,
+  brand,
+  headerActions,
+  hideWorkforceSelector,
+  className,
+  headerClassName,
+  baseUrl,
+  fetch: fetch2,
+  ...chatProps
+}) => {
+  const { workforces, selectedId, setSelectedId } = useWorkforces({
+    baseUrl,
+    fetch: fetch2
+  });
+  const effectiveId = workforceId ?? selectedId;
+  const showSelector = !hideWorkforceSelector && !workforceId && workforces.length > 0;
+  return /* @__PURE__ */ jsxs17(
+    "div",
+    {
+      className: cn(
+        "aui-chat-shell relative flex h-dvh flex-col overflow-hidden bg-background",
+        className
+      ),
+      style: studioChromeShellStyle,
+      children: [
+        /* @__PURE__ */ jsx30(
+          "div",
+          {
+            className: cn(
+              "pointer-events-none absolute inset-0 z-0",
+              studioPlaygroundGradientClass
+            ),
+            "aria-hidden": true
+          }
+        ),
+        /* @__PURE__ */ jsxs17(
+          "header",
+          {
+            className: cn(
+              "aui-chat-shell-header relative z-10 flex shrink-0 items-center justify-between px-4 pt-[var(--studio-topbar-gap)] pb-2",
+              headerClassName
+            ),
+            style: { minHeight: "var(--studio-topbar-height)" },
+            children: [
+              /* @__PURE__ */ jsxs17("div", { className: "flex min-w-0 items-center gap-2", children: [
+                brand,
+                showSelector && /* @__PURE__ */ jsx30(
+                  WorkforceSelector,
+                  {
+                    workforces,
+                    value: selectedId,
+                    onChange: setSelectedId
+                  }
+                )
+              ] }),
+              /* @__PURE__ */ jsx30("div", { className: "flex shrink-0 items-center gap-1", children: headerActions })
+            ]
+          }
+        ),
+        /* @__PURE__ */ jsx30(
+          TimbalChat,
+          {
+            workforceId: effectiveId,
+            baseUrl,
+            fetch: fetch2,
+            className: "relative z-10 min-h-0 flex-1 bg-transparent",
+            ...chatProps
+          },
+          effectiveId
+        )
+      ]
+    }
+  );
+};
+
 // src/auth/provider.tsx
 import {
-  createContext,
-  useCallback as useCallback2,
-  useContext,
-  useEffect as useEffect4,
-  useState as useState5
+  createContext as createContext4,
+  useCallback as useCallback4,
+  useContext as useContext4,
+  useEffect as useEffect6,
+  useState as useState9
 } from "react";
-import { jsx as jsx14 } from "react/jsx-runtime";
+import { jsx as jsx31 } from "react/jsx-runtime";
 function isInsideIframe() {
   try {
     return typeof window !== "undefined" && window.self !== window.top;
@@ -1534,9 +4182,9 @@ function isInsideIframe() {
     return true;
   }
 }
-var SessionContext = createContext(void 0);
+var SessionContext = createContext4(void 0);
 var useSession = () => {
-  const context = useContext(SessionContext);
+  const context = useContext4(SessionContext);
   if (context === void 0) {
     throw new Error("useSession must be used within a SessionProvider");
   }
@@ -1546,10 +4194,10 @@ var SessionProvider = ({
   children,
   enabled = true
 }) => {
-  const [user, setUser] = useState5(null);
-  const [loading, setLoading] = useState5(enabled);
-  const [embedded] = useState5(isInsideIframe);
-  useEffect4(() => {
+  const [user, setUser] = useState9(null);
+  const [loading, setLoading] = useState9(enabled);
+  const [embedded] = useState9(isInsideIframe);
+  useEffect6(() => {
     if (!enabled) {
       setLoading(false);
       return;
@@ -1610,7 +4258,7 @@ var SessionProvider = ({
       messageCleanup?.();
     };
   }, [enabled, embedded]);
-  const logout = useCallback2(() => {
+  const logout = useCallback4(() => {
     clearTokens();
     setUser(null);
     const returnTo = encodeURIComponent(
@@ -1620,7 +4268,7 @@ var SessionProvider = ({
       () => window.location.href = `/api/auth/login?return_to=${returnTo}`
     );
   }, []);
-  return /* @__PURE__ */ jsx14(
+  return /* @__PURE__ */ jsx31(
     SessionContext.Provider,
     {
       value: {
@@ -1637,7 +4285,7 @@ var SessionProvider = ({
 
 // src/auth/guard.tsx
 import { Loader2 } from "lucide-react";
-import { jsx as jsx15 } from "react/jsx-runtime";
+import { jsx as jsx32 } from "react/jsx-runtime";
 var AuthGuard = ({
   children,
   requireAuth = false,
@@ -1648,7 +4296,7 @@ var AuthGuard = ({
     return children;
   }
   if (loading) {
-    return /* @__PURE__ */ jsx15("div", { className: "flex items-center justify-center h-screen", children: /* @__PURE__ */ jsx15(Loader2, { className: "w-8 h-8 animate-spin" }) });
+    return /* @__PURE__ */ jsx32("div", { className: "flex items-center justify-center h-screen", children: /* @__PURE__ */ jsx32(Loader2, { className: "w-8 h-8 animate-spin" }) });
   }
   if (requireAuth && !isAuthenticated && !isEmbedded) {
     const returnTo = encodeURIComponent(
@@ -1660,15 +4308,26 @@ var AuthGuard = ({
   return children;
 };
 export {
+  ARTIFACT_AGENT_INSTRUCTIONS,
+  ARTIFACT_FENCE_LANGUAGES,
+  ActionBarMorePrimitive2 as ActionBarMorePrimitive,
   ActionBarPrimitive2 as ActionBarPrimitive,
+  ArtifactCard,
+  ArtifactRegistryProvider,
+  ArtifactView,
+  AssistantRuntimeProvider2 as AssistantRuntimeProvider,
+  AuiIf3 as AuiIf,
   AuthGuard,
   Avatar,
   AvatarFallback,
   AvatarImage,
   Button,
+  ChartArtifactView,
+  Composer,
   ComposerAddAttachment,
   ComposerAttachments,
-  ComposerPrimitive3 as ComposerPrimitive,
+  ComposerPrimitive4 as ComposerPrimitive,
+  DEFAULT_UPLOAD_ACCEPT,
   Dialog,
   DialogClose,
   DialogContent,
@@ -1676,35 +4335,119 @@ export {
   DialogPortal,
   DialogTitle,
   DialogTrigger,
+  ErrorPrimitive2 as ErrorPrimitive,
+  HtmlArtifactView,
+  JsonArtifactView,
   MarkdownText,
+  MessagePartPrimitive2 as MessagePartPrimitive,
   MessagePrimitive3 as MessagePrimitive,
+  QuestionArtifactView,
+  STUDIO_INSET_LEFT,
+  STUDIO_PILL_HEIGHT,
+  STUDIO_SIDEBAR_GAP,
+  STUDIO_SIDEBAR_WIDTH,
+  STUDIO_TOPBAR_GAP,
+  STUDIO_TOPBAR_HEIGHT,
   SessionProvider,
   Shimmer,
+  Suggestions,
   syntax_highlighter_default as SyntaxHighlighter,
+  TIMBAL_V2_BORDER,
+  TIMBAL_V2_FILL,
+  TIMBAL_V2_LABEL,
+  TIMBAL_V2_PILL_SURFACE,
+  TIMBAL_V2_SECONDARY_CHROME,
+  TIMBAL_V2_SHADOW,
+  TIMBAL_V2_SIZE_HEIGHT,
+  TIMBAL_V2_SIZE_ICON,
+  TIMBAL_V2_SIZE_LABEL_PX,
+  TableArtifactView,
   Thread,
   ThreadPrimitive2 as ThreadPrimitive,
   TimbalChat,
+  TimbalChatShell,
   TimbalRuntimeProvider,
+  TimbalV2Button,
+  ToolArtifactFallback,
+  ToolBodyPresence,
   ToolFallback,
+  ToolMotion,
+  ToolPresence,
   Tooltip,
   TooltipContent,
   TooltipIconButton,
   TooltipProvider,
   TooltipTrigger,
+  UiArtifactView,
+  UiCustomNodeRegistryProvider,
+  UiEventProvider,
+  UiNodeView,
   UserMessageAttachments,
+  WorkforceSelector,
   authFetch,
   buttonVariants,
   clearTokens,
   cn,
+  createDefaultAttachmentAdapter,
+  createUploadAttachmentAdapter,
+  defaultArtifactRenderers,
   fetchCurrentUser,
+  findMarkdownArtifacts,
   getAccessToken,
+  getPath,
   getRefreshToken,
+  isArtifact,
+  isArtifactFenceLanguage,
+  isUiBinding,
+  luxuryEase,
+  parseArtifactFromToolResult,
+  parseSSELine2 as parseSSELine,
   refreshAccessToken,
+  resolveAttachmentAdapter,
+  resolveBindable,
   setAccessToken,
+  setPath,
   setRefreshToken,
-  useComposerRuntime,
+  splitMarkdownByArtifacts,
+  studioArtifactShellClass,
+  studioChromeShellStyle,
+  studioComposeInputShellClass,
+  studioComposerIoWellClass,
+  studioIntegrationBorder,
+  studioIntegrationCardClass,
+  studioIntegrationIconTileClass,
+  studioIntegrationSurfaceSolid,
+  studioListRowButtonClass,
+  studioPillSurfaceClass,
+  studioPlaygroundGradientClass,
+  studioQuestionOptionClass,
+  studioQuestionOptionSelectedClass,
+  studioSecondaryChromeClass,
+  studioTimelineActionClass,
+  studioTimelineBodyPadClass,
+  studioTimelineChevronClass,
+  studioTimelineDetailClass,
+  studioTimelineRowButtonClass,
+  studioTimelineShimmerActionClass,
+  studioTimelineTextClass,
+  studioToolCardShellClass,
+  studioTopbarIconPillClass,
+  studioTopbarPillHeightClass,
+  toolPresenceTransition,
+  useArtifactRegistry,
+  useAuiState3 as useAuiState,
+  useComposerRuntime2 as useComposerRuntime,
   useMessageRuntime,
+  useResolvedSuggestions,
   useSession,
-  useThread,
-  useThreadRuntime2 as useThreadRuntime
+  useThread2 as useThread,
+  useThreadRuntime4 as useThreadRuntime,
+  useTimbalRuntime,
+  useTimbalStream,
+  useToolRunning,
+  useUiCustomNodeRegistry,
+  useUiDispatch,
+  useUiEventEmitter,
+  useUiState,
+  useWorkforces
 };