@tylertech/forge-ai 0.10.0 → 0.11.0

package/dist/ai-chatbot/ai-chatbot.mjs

@@ -1,11 +1,8 @@
-import { unsafeCSS, LitElement, html, nothing } from "lit";
+import { unsafeCSS, html, nothing } from "lit";
 import { property, customElement } from "lit/decorators.js";
 import { createRef, ref } from "lit/directives/ref.js";
 import { when } from "lit/directives/when.js";
-import { SubscriptionManager } from "./event-emitter.mjs";
-import { FileUploadManager } from "./file-upload-manager.mjs";
-import { MessageStateController } from "./message-state-controller.mjs";
-import { generateId, downloadFile } from "./utils.mjs";
+import { AiChatbotBase } from "./ai-chatbot-base.mjs";
 import "../ai-attachment/ai-attachment.mjs";
 import "../ai-chat-header/ai-chat-header.mjs";
 import "../ai-chat-interface/ai-chat-interface.mjs";
@@ -32,168 +29,32 @@ var __decorateClass = (decorators, target, key, kind) => {
 var __accessCheck = (obj, member, msg) => member.has(obj) || __typeError("Cannot " + msg);
 var __privateGet = (obj, member, getter) => (__accessCheck(obj, member, "read from private field"), getter ? getter.call(obj) : member.get(obj));
 var __privateAdd = (obj, member, value) => member.has(obj) ? __typeError("Cannot add the same private member more than once") : member instanceof WeakSet ? member.add(obj) : member.set(obj, value);
-var __privateSet = (obj, member, value, setter) => (__accessCheck(obj, member, "write to private field"), member.set(obj, value), value);
 var __privateMethod = (obj, member, method) => (__accessCheck(obj, member, "access private method"), method);
-var __privateWrapper = (obj, member, setter, getter) => ({
-  set _(value) {
-    __privateSet(obj, member, value);
-  },
-  get _() {
-    return __privateGet(obj, member, getter);
-  }
-});
-var _chatInterfaceRef, _messageThreadRef, _promptRef, _headerRef, _messageStateController, _fileUploadManager, _toolsMap, _adapterSubscriptions, _executingToolHandlers, _AiChatbotComponent_instances, slashCommands_get, messageItems_get, isStreaming_get, isUploading_get, hasMessages_get, tools_get, setupAdapter_fn, handleMessageStart_fn, handleMessageDelta_fn, handleMessageEnd_fn, handleRunFinished_fn, tryCompleteResponse_fn, handleToolCallStart_fn, handleToolCallArgs_fn, handleToolCallEnd_fn, createToolResponse_fn, handleToolCall_fn, executeToolHandler_fn, handleError_fn, handleRunAborted_fn, handleStateChange_fn, handleToolCallResult_fn, sendToolResult_fn, sendUserMessage_fn, handleSend_fn, handleStop_fn, handleCancel_fn, handleCopy_fn, handleUserCopy_fn, handleUserResend_fn, handleUserEdit_fn, handleResend_fn, handleFeedback_fn, handleThumbsUp_fn, handleThumbsDown_fn, processFileUpload_fn, handleFileSelect_fn, createFileUploadCallbacks_fn, handleFileError_fn, handleAttachmentRemove_fn, handleSuggestionSelect_fn, handleVoiceInputResult_fn, handleHeaderExpand_fn, handleHeaderMinimize_fn, handleHeaderClear_fn, handleHeaderInfo_fn, handleAgentChange_fn, handleDebugToggle_fn, handleSlashCommand_fn, formatToolCallForExport_fn, handleExport_fn, sessionFilesTemplate_get, promptSlot_get, messageThread_get, dispatchEvent_fn, dispatchMessageEvent_fn, emitStateChange_fn;
+var _chatInterfaceRef, _headerRef, _AiChatbotComponent_instances, handleHeaderExpand_fn, handleHeaderMinimize_fn, handleHeaderClear_fn, handleHeaderInfo_fn, sessionFilesTemplate_get, promptSlot_get, messageThread_get;
 const AiChatbotComponentTagName = "forge-ai-chatbot";
-let AiChatbotComponent = class extends LitElement {
+let AiChatbotComponent = class extends AiChatbotBase {
   constructor() {
     super(...arguments);
     __privateAdd(this, _AiChatbotComponent_instances);
-    this.fileUpload = "off";
-    this.voiceInput = "on";
-    this.debugCommand = "on";
-    this.placeholder = "Ask a question...";
     this.showExpandButton = false;
     this.showMinimizeButton = false;
     this.expanded = false;
     this.minimizeIcon = "default";
-    this.enableReactions = false;
-    this.titleText = "AI Assistant";
-    this.headingLevel = 2;
-    this.debugMode = false;
-    this.disclaimerText = "AI can make mistakes. Always verify responses.";
-    this.agents = [];
     __privateAdd(this, _chatInterfaceRef, createRef());
-    __privateAdd(this, _messageThreadRef, createRef());
-    __privateAdd(this, _promptRef, createRef());
+    this._messageThreadRef = createRef();
+    this._promptRef = createRef();
     __privateAdd(this, _headerRef, createRef());
-    __privateAdd(this, _messageStateController);
-    __privateAdd(this, _fileUploadManager);
-    __privateAdd(this, _toolsMap);
-    __privateAdd(this, _adapterSubscriptions);
-    __privateAdd(this, _executingToolHandlers, 0);
   }
-  connectedCallback() {
-    super.connectedCallback();
-    __privateSet(this, _messageStateController, new MessageStateController(this, {
-      tools: __privateGet(this, _AiChatbotComponent_instances, tools_get),
-      onThreadSettled: () => __privateMethod(this, _AiChatbotComponent_instances, emitStateChange_fn).call(this)
-    }));
-    __privateSet(this, _fileUploadManager, new FileUploadManager({
-      onError: (error) => {
-        __privateMethod(this, _AiChatbotComponent_instances, dispatchEvent_fn).call(this, {
-          type: "forge-ai-chatbot-error",
-          detail: { error }
-        });
-      },
-      onStateChange: () => {
-        this.requestUpdate();
-      }
-    }));
-    if (this.adapter) {
-      void __privateMethod(this, _AiChatbotComponent_instances, setupAdapter_fn).call(this);
+  _handleInfo() {
+    const header = __privateGet(this, _headerRef).value;
+    if (header?.showAgentInfo) {
+      header.showAgentInfo();
     }
+    super._handleInfo();
   }
-  disconnectedCallback() {
-    super.disconnectedCallback();
-    __privateGet(this, _adapterSubscriptions)?.unsubscribe();
-    this.adapter?.disconnect();
-  }
-  willUpdate(changedProperties) {
-    if (changedProperties.has("adapter") && this.adapter) {
-      __privateSet(this, _toolsMap, void 0);
-      __privateMethod(this, _AiChatbotComponent_instances, setupAdapter_fn).call(this);
-    }
-  }
-  /**
-   * Clears all messages from the chat history.
-   */
-  clearMessages() {
-    __privateGet(this, _messageStateController).clearMessages();
-    __privateGet(this, _fileUploadManager).clear();
-  }
-  /**
-   * Gets the current message history.
-   * @returns Array of chat messages
-   */
-  getMessages() {
-    return __privateGet(this, _messageStateController).getMessages();
-  }
-  /**
-   * Sets the message history. Useful for restoring conversation state.
-   * @param messages - Array of chat messages to set
-   */
-  setMessages(messages) {
-    __privateGet(this, _messageStateController).setMessages(messages);
-  }
-  /**
-   * Gets the currently selected agent.
-   * @returns The selected agent or undefined if none selected
-   */
-  getSelectedAgent() {
-    return this.agents.find((a) => a.id === this.selectedAgentId);
-  }
-  /**
-   * Programmatically sends a message as the user.
-   * @param content - The message content to send
-   * @param files - Optional file objects to attach
-   */
   async sendMessage(content, files) {
-    if (!this.adapter) {
-      console.warn("No adapter configured.");
-      return;
-    }
-    __privateGet(this, _promptRef).value?.closeSlashMenu();
-    if (files) {
-      const timestamp = Date.now();
-      for (const file of files) {
-        __privateMethod(this, _AiChatbotComponent_instances, processFileUpload_fn).call(this, file, timestamp);
-      }
-    }
-    await __privateMethod(this, _AiChatbotComponent_instances, sendUserMessage_fn).call(this, {
-      content,
-      attachments: __privateGet(this, _fileUploadManager).pendingAttachments
-    });
-  }
-  /**
-   * Aborts the current streaming response.
-   */
-  abort() {
-    __privateMethod(this, _AiChatbotComponent_instances, handleStop_fn).call(this);
-  }
-  /**
-   * Scrolls the chat interface to the bottom.
-   */
-  async scrollToBottom({ behavior } = {}) {
-    await __privateGet(this, _messageThreadRef).value?.updateComplete;
-    await new Promise(requestAnimationFrame);
-    __privateGet(this, _messageThreadRef).value?.scrollToBottom({ behavior });
-  }
-  /**
-   * Gets the complete serializable thread state including threadId and messages.
-   * @returns ThreadState object containing threadId, messages, and timestamp
-   */
-  getThreadState() {
-    return {
-      threadId: this.adapter?.threadId,
-      messages: this.getMessages(),
-      timestamp: Date.now(),
-      selectedAgentId: this.selectedAgentId
-    };
-  }
-  /**
-   * Restores thread state from a serialized ThreadState object.
-   * @param state - ThreadState object to restore
-   */
-  async setThreadState(state) {
-    this.setMessages(state.messages);
-    if (state.threadId && this.adapter) {
-      this.adapter.threadId = state.threadId;
-    }
-    this.selectedAgentId = state.selectedAgentId;
-    await this.updateComplete;
-    const userMessages = state.messages.filter((msg) => msg.role === "user").map((msg) => msg.content);
-    __privateGet(this, _promptRef).value?.setHistory(userMessages);
-    this.scrollToBottom({ behavior: "instant" });
+    this._promptRef.value?.closeSlashMenu();
+    await super.sendMessage(content, files);
   }
   render() {
     return html`
@@ -202,16 +63,16 @@ let AiChatbotComponent = class extends LitElement {
         role="region"
         aria-label="AI chatbot"
         aria-live="polite"
-        aria-busy=${__privateGet(this, _AiChatbotComponent_instances, isStreaming_get)}>
+        aria-busy=${this._isStreaming}>
         <forge-ai-chat-header
           ${ref(__privateGet(this, _headerRef))}
           slot="header"
           ?show-expand-button=${this.showExpandButton}
           ?show-minimize-button=${this.showMinimizeButton}
           ?expanded=${this.expanded}
-          ?disable-agent-selector=${__privateGet(this, _AiChatbotComponent_instances, isStreaming_get)}
-          export-option=${__privateGet(this, _AiChatbotComponent_instances, hasMessages_get) ? "enabled" : "off"}
-          clear-option=${__privateGet(this, _AiChatbotComponent_instances, hasMessages_get) ? "enabled" : "off"}
+          ?disable-agent-selector=${this._isStreaming}
+          export-option=${this._hasMessages ? "enabled" : "off"}
+          clear-option=${this._hasMessages ? "enabled" : "off"}
           .minimizeIcon=${this.minimizeIcon}
           .agentInfo=${this.agentInfo}
           .headingLevel=${this.headingLevel}
@@ -221,9 +82,9 @@ let AiChatbotComponent = class extends LitElement {
           @forge-ai-chat-header-expand=${__privateMethod(this, _AiChatbotComponent_instances, handleHeaderExpand_fn)}
           @forge-ai-chat-header-minimize=${__privateMethod(this, _AiChatbotComponent_instances, handleHeaderMinimize_fn)}
           @forge-ai-chat-header-clear=${__privateMethod(this, _AiChatbotComponent_instances, handleHeaderClear_fn)}
-          @forge-ai-chat-header-export=${__privateMethod(this, _AiChatbotComponent_instances, handleExport_fn)}
+          @forge-ai-chat-header-export=${this._handleExport}
           @forge-ai-chat-header-info=${__privateMethod(this, _AiChatbotComponent_instances, handleHeaderInfo_fn)}
-          @forge-ai-chat-header-agent-change=${__privateMethod(this, _AiChatbotComponent_instances, handleAgentChange_fn)}>
+          @forge-ai-chat-header-agent-change=${this._handleAgentChange}>
           <slot name="icon" slot="icon">
             <forge-ai-icon></forge-ai-icon>
           </slot>
@@ -235,600 +96,43 @@ let AiChatbotComponent = class extends LitElement {
   }
 };
 _chatInterfaceRef = /* @__PURE__ */ new WeakMap();
-_messageThreadRef = /* @__PURE__ */ new WeakMap();
-_promptRef = /* @__PURE__ */ new WeakMap();
 _headerRef = /* @__PURE__ */ new WeakMap();
-_messageStateController = /* @__PURE__ */ new WeakMap();
-_fileUploadManager = /* @__PURE__ */ new WeakMap();
-_toolsMap = /* @__PURE__ */ new WeakMap();
-_adapterSubscriptions = /* @__PURE__ */ new WeakMap();
-_executingToolHandlers = /* @__PURE__ */ new WeakMap();
 _AiChatbotComponent_instances = /* @__PURE__ */ new WeakSet();
-slashCommands_get = function() {
-  const commands = [];
-  if (__privateGet(this, _AiChatbotComponent_instances, hasMessages_get)) {
-    commands.push({ id: "clear", name: "Clear", group: "Conversation" });
-    commands.push({ id: "export", name: "Export", group: "Conversation" });
-  }
-  commands.push({ id: "info", name: "Info", group: "Help" });
-  if (this.debugCommand === "on") {
-    commands.push({
-      id: "debug",
-      name: `${this.debugMode ? "Disable debug mode" : "Enable debug mode"}`,
-      group: "Help"
-    });
-  }
-  return commands;
-};
-messageItems_get = function() {
-  return __privateGet(this, _messageStateController)?.messageItems ?? [];
-};
-isStreaming_get = function() {
-  return (this.adapter?.isRunning ?? false) || __privateGet(this, _executingToolHandlers) > 0;
-};
-isUploading_get = function() {
-  return __privateGet(this, _fileUploadManager)?.isUploading ?? false;
-};
-hasMessages_get = function() {
-  return __privateGet(this, _AiChatbotComponent_instances, messageItems_get).length > 0;
-};
-tools_get = function() {
-  if (!__privateGet(this, _toolsMap)) {
-    __privateSet(this, _toolsMap, new Map(this.adapter?.getTools().map((t) => [t.name, t]) ?? []));
-  }
-  return __privateGet(this, _toolsMap);
-};
-setupAdapter_fn = async function() {
-  if (!this.adapter || this.adapter.isConnecting) {
-    return;
-  }
-  await this.adapter.connect();
-  __privateGet(this, _adapterSubscriptions)?.unsubscribe();
-  __privateSet(this, _adapterSubscriptions, new SubscriptionManager());
-  __privateGet(this, _adapterSubscriptions).add(
-    this.adapter.onMessageStart(__privateMethod(this, _AiChatbotComponent_instances, handleMessageStart_fn).bind(this)),
-    this.adapter.onMessageDelta(__privateMethod(this, _AiChatbotComponent_instances, handleMessageDelta_fn).bind(this)),
-    this.adapter.onMessageEnd(__privateMethod(this, _AiChatbotComponent_instances, handleMessageEnd_fn).bind(this)),
-    this.adapter.onToolCallStart(__privateMethod(this, _AiChatbotComponent_instances, handleToolCallStart_fn).bind(this)),
-    this.adapter.onToolCallArgs(__privateMethod(this, _AiChatbotComponent_instances, handleToolCallArgs_fn).bind(this)),
-    this.adapter.onToolCallEnd(__privateMethod(this, _AiChatbotComponent_instances, handleToolCallEnd_fn).bind(this)),
-    this.adapter.onToolCall(__privateMethod(this, _AiChatbotComponent_instances, handleToolCall_fn).bind(this)),
-    this.adapter.onToolCallResult(__privateMethod(this, _AiChatbotComponent_instances, handleToolCallResult_fn).bind(this)),
-    this.adapter.onRunFinished(__privateMethod(this, _AiChatbotComponent_instances, handleRunFinished_fn).bind(this)),
-    this.adapter.onRunAborted(__privateMethod(this, _AiChatbotComponent_instances, handleRunAborted_fn).bind(this)),
-    this.adapter.onError(__privateMethod(this, _AiChatbotComponent_instances, handleError_fn).bind(this)),
-    this.adapter.onStateChange(__privateMethod(this, _AiChatbotComponent_instances, handleStateChange_fn).bind(this))
-  );
-  __privateSet(this, _toolsMap, void 0);
-  __privateGet(this, _messageStateController)?.updateConfig({ tools: __privateGet(this, _AiChatbotComponent_instances, tools_get) });
-  __privateMethod(this, _AiChatbotComponent_instances, dispatchEvent_fn).call(this, { type: "forge-ai-chatbot-connected" });
-};
-handleMessageStart_fn = function(event) {
-  __privateGet(this, _messageStateController).addTextToResponse(event.messageId, "", event);
-};
-handleMessageDelta_fn = function(event) {
-  __privateGet(this, _messageStateController).appendTextDelta(event.messageId, event.delta, event);
-  this.scrollToBottom();
-};
-handleMessageEnd_fn = function(event) {
-  __privateGet(this, _messageStateController).markTextComplete(event.messageId, event);
-};
-handleRunFinished_fn = function() {
-  __privateMethod(this, _AiChatbotComponent_instances, tryCompleteResponse_fn).call(this);
-};
-tryCompleteResponse_fn = function() {
-  if (__privateGet(this, _executingToolHandlers) > 0 || this.adapter?.isRunning) {
-    return;
-  }
-  __privateGet(this, _messageStateController).tryFinalizeResponse();
-  const messages = this.getMessages();
-  const lastMessage = messages[messages.length - 1];
-  if (lastMessage?.role === "assistant" && lastMessage.status === "complete") {
-    __privateMethod(this, _AiChatbotComponent_instances, dispatchMessageEvent_fn).call(this, "forge-ai-chatbot-message-received", lastMessage);
-  }
-};
-handleToolCallStart_fn = function(event) {
-  const toolCall = {
-    id: event.id,
-    messageId: event.messageId,
-    name: event.name,
-    args: {},
-    argsBuffer: "",
-    status: "parsing",
-    type: __privateGet(this, _AiChatbotComponent_instances, tools_get).has(event.name) ? "client" : "agent"
-  };
-  __privateGet(this, _messageStateController).addToolCallToResponse(toolCall, event);
-};
-handleToolCallArgs_fn = function(event) {
-  const updates = {
-    argsBuffer: event.argsBuffer,
-    args: event.partialArgs ?? {}
-  };
-  const rawEvent = { eventType: "tool-call-args", event };
-  __privateGet(this, _messageStateController).updateToolCallInResponse(event.id, updates, rawEvent);
-  this.scrollToBottom();
-};
-handleToolCallEnd_fn = function(event) {
-  const updates = {
-    args: event.args,
-    argsBuffer: void 0,
-    status: "executing"
-  };
-  const rawEvent = { eventType: "tool-call-end", event };
-  __privateGet(this, _messageStateController).updateToolCallInResponse(event.id, updates, rawEvent);
-};
-createToolResponse_fn = function(toolName, handlerReturn) {
-  if (typeof handlerReturn === "string") {
-    return handlerReturn;
-  }
-  if (handlerReturn && typeof handlerReturn === "object") {
-    return handlerReturn;
-  }
-  return `Tool '${toolName}' executed successfully`;
-};
-handleToolCall_fn = async function(event) {
-  let toolCall = __privateGet(this, _messageStateController).getToolCall(event.id);
-  if (toolCall) {
-    __privateGet(this, _messageStateController).updateToolCallInResponse(event.id, {
-      args: event.args,
-      status: "executing"
-    });
-  } else {
-    toolCall = {
-      id: event.id,
-      messageId: event.messageId,
-      name: event.name,
-      args: event.args,
-      status: "executing",
-      type: __privateGet(this, _AiChatbotComponent_instances, tools_get).has(event.name) ? "client" : "agent"
-    };
-    __privateGet(this, _messageStateController).addToolCallToResponse(toolCall);
-  }
-  if (toolCall?.type === "client") {
-    __privateMethod(this, _AiChatbotComponent_instances, dispatchEvent_fn).call(this, {
-      type: "forge-ai-chatbot-tool-call",
-      detail: {
-        toolCallId: event.id,
-        toolName: event.name,
-        arguments: event.args
-      }
-    });
-    const toolDef = __privateGet(this, _AiChatbotComponent_instances, tools_get).get(event.name);
-    if (toolDef?.handler) {
-      await Promise.resolve(__privateMethod(this, _AiChatbotComponent_instances, executeToolHandler_fn).call(this, event.id, event.name, toolDef.handler, event.args));
-    } else {
-      __privateMethod(this, _AiChatbotComponent_instances, sendToolResult_fn).call(this, event.id, __privateMethod(this, _AiChatbotComponent_instances, createToolResponse_fn).call(this, event.name));
-    }
-  } else {
-    __privateGet(this, _messageStateController).updateToolCallInResponse(event.id, { status: "complete" });
-  }
-};
-executeToolHandler_fn = async function(toolCallId, toolName, handler, args) {
-  __privateWrapper(this, _executingToolHandlers)._++;
-  this.requestUpdate();
-  try {
-    const context = {
-      args,
-      toolCallId,
-      toolName,
-      signal: void 0
-    };
-    const handlerReturn = await handler(context);
-    await __privateMethod(this, _AiChatbotComponent_instances, sendToolResult_fn).call(this, toolCallId, __privateMethod(this, _AiChatbotComponent_instances, createToolResponse_fn).call(this, toolName, handlerReturn));
-  } catch (error) {
-    console.error(`Tool handler error [${toolName}]:`, error);
-    __privateGet(this, _messageStateController).updateToolCallInResponse(toolCallId, {
-      status: "error",
-      result: { error: error.message }
-    });
-    const errorMessage = {
-      id: generateId(),
-      role: "assistant",
-      content: "An unexpected error occurred.",
-      timestamp: Date.now(),
-      status: "error"
-    };
-    __privateGet(this, _messageStateController).addMessage(errorMessage);
-    __privateMethod(this, _AiChatbotComponent_instances, dispatchEvent_fn).call(this, { type: "forge-ai-chatbot-error", detail: { error: error.message } });
-  } finally {
-    __privateWrapper(this, _executingToolHandlers)._--;
-    __privateMethod(this, _AiChatbotComponent_instances, tryCompleteResponse_fn).call(this);
-    this.requestUpdate();
-  }
-};
-handleError_fn = function(event) {
-  const errorMessage = {
-    id: generateId(),
-    role: "assistant",
-    content: event.message,
-    timestamp: Date.now(),
-    status: "error"
-  };
-  __privateGet(this, _messageStateController).addMessage(errorMessage);
-  __privateMethod(this, _AiChatbotComponent_instances, dispatchMessageEvent_fn).call(this, "forge-ai-chatbot-message-received", errorMessage);
-  __privateMethod(this, _AiChatbotComponent_instances, dispatchEvent_fn).call(this, { type: "forge-ai-chatbot-error", detail: { error: event.message } });
-};
-handleRunAborted_fn = function() {
-  const abortMessage = {
-    id: generateId(),
-    role: "system",
-    content: "Run cancelled",
-    timestamp: Date.now(),
-    status: "complete",
-    clientOnly: true
-  };
-  __privateGet(this, _messageStateController).addMessage(abortMessage);
-  __privateMethod(this, _AiChatbotComponent_instances, dispatchMessageEvent_fn).call(this, "forge-ai-chatbot-message-received", abortMessage);
-};
-handleStateChange_fn = function(_state) {
-  this.requestUpdate();
-};
-handleToolCallResult_fn = function(event) {
-  __privateGet(this, _messageStateController).completeToolCallInResponse(event.toolCallId, event.result, event);
-};
-sendToolResult_fn = async function(toolCallId, result) {
-  const toolCall = __privateGet(this, _messageStateController).getToolCall(toolCallId);
-  if (!toolCall || !this.adapter) {
-    return;
-  }
-  __privateGet(this, _messageStateController).completeToolCallInResponse(toolCallId, result);
-  this.adapter.sendToolResult(toolCallId, result, this.getMessages());
-};
-sendUserMessage_fn = async function(config) {
-  if (!config.content.trim() || !this.adapter || __privateGet(this, _AiChatbotComponent_instances, isStreaming_get)) {
-    if (!this.adapter) {
-      console.warn("No adapter configured.");
-    }
-    return;
-  }
-  if (!__privateGet(this, _fileUploadManager).canSend()) {
-    return;
-  }
-  const userMessage = {
-    id: generateId(),
-    role: "user",
-    content: config.content,
-    timestamp: config.timestamp ?? Date.now(),
-    status: "pending"
-  };
-  __privateGet(this, _messageStateController).addMessage(userMessage);
-  try {
-    this.adapter.sendMessage(this.getMessages());
-    __privateGet(this, _messageStateController).updateMessageStatus(userMessage.id, "complete");
-  } catch (error) {
-    __privateGet(this, _messageStateController).updateMessageStatus(userMessage.id, "error");
-    const errorMessage = error instanceof Error ? error.message : "Failed to send message";
-    __privateMethod(this, _AiChatbotComponent_instances, dispatchEvent_fn).call(this, { type: "forge-ai-chatbot-error", detail: { error: errorMessage } });
-  } finally {
-    __privateMethod(this, _AiChatbotComponent_instances, dispatchMessageEvent_fn).call(this, "forge-ai-chatbot-message-sent", userMessage.id);
-  }
-};
-handleSend_fn = async function(evt) {
-  const pendingAttachments = [...__privateGet(this, _fileUploadManager).pendingAttachments];
-  await __privateMethod(this, _AiChatbotComponent_instances, sendUserMessage_fn).call(this, {
-    content: evt.detail.value,
-    timestamp: evt.detail.date.getTime(),
-    attachments: pendingAttachments
-  });
-};
-handleStop_fn = function() {
-  if (this.adapter?.isRunning) {
-    this.adapter.abort();
-  }
-  this.requestUpdate();
-};
-handleCancel_fn = function() {
-  __privateMethod(this, _AiChatbotComponent_instances, handleStop_fn).call(this);
-};
-handleCopy_fn = async function(evt) {
-  const responseId = evt.detail.messageId;
-  const responseItem = __privateGet(this, _AiChatbotComponent_instances, messageItems_get).find((item) => item.type === "assistant" && item.data.id === responseId);
-  if (!responseItem || responseItem.type !== "assistant") {
-    return;
-  }
-  const textContent = responseItem.data.children.filter(
-    (c) => c.type === "text"
-  ).map((c) => c.content).join("\n\n");
-  try {
-    await navigator.clipboard.writeText(textContent);
-  } catch {
-  }
-};
-handleUserCopy_fn = async function(evt) {
-  const messageId = evt.detail.messageId;
-  const message = __privateGet(this, _messageStateController).getMessage(messageId);
-  if (!message || message.role !== "user") {
-    return;
-  }
-  try {
-    await navigator.clipboard.writeText(message.content);
-  } catch {
-  }
-};
-handleUserResend_fn = function(evt) {
-  if (!this.adapter) {
-    return;
-  }
-  const messageId = evt.detail.messageId;
-  const messageIndex = __privateGet(this, _AiChatbotComponent_instances, messageItems_get).findIndex((item) => item.type === "message" && item.data.id === messageId);
-  if (messageIndex === -1) {
-    return;
-  }
-  const responseIndex = messageIndex + 1;
-  if (responseIndex < __privateGet(this, _AiChatbotComponent_instances, messageItems_get).length) {
-    __privateGet(this, _messageStateController).removeMessageItemsFrom(responseIndex);
-  }
-  this.adapter.sendMessage(this.getMessages());
-};
-handleUserEdit_fn = function(evt) {
-  if (!this.adapter) {
-    return;
-  }
-  const { messageId, content } = evt.detail;
-  const messageIndex = __privateGet(this, _AiChatbotComponent_instances, messageItems_get).findIndex((item) => item.type === "message" && item.data.id === messageId);
-  if (messageIndex === -1) {
-    return;
-  }
-  __privateGet(this, _messageStateController).updateMessageContent(messageId, content);
-  const responseIndex = messageIndex + 1;
-  if (responseIndex < __privateGet(this, _AiChatbotComponent_instances, messageItems_get).length) {
-    __privateGet(this, _messageStateController).removeMessageItemsFrom(responseIndex);
-  }
-  this.adapter.sendMessage(this.getMessages());
-};
-handleResend_fn = function(evt) {
-  if (!this.adapter) {
-    return;
-  }
-  const responseId = evt.detail.messageId;
-  const responseIndex = __privateGet(this, _AiChatbotComponent_instances, messageItems_get).findIndex(
-    (item) => item.type === "assistant" && item.data.id === responseId
-  );
-  if (responseIndex === -1) {
-    return;
-  }
-  __privateGet(this, _messageStateController).removeMessageItemsFrom(responseIndex);
-  this.adapter.sendMessage(this.getMessages());
-};
-handleFeedback_fn = function(evt, type) {
-  __privateGet(this, _messageStateController).setResponseFeedback(evt.detail.messageId, {
-    type,
-    reason: evt.detail.feedback
-  });
-  __privateMethod(this, _AiChatbotComponent_instances, dispatchEvent_fn).call(this, {
-    type: "forge-ai-chatbot-response-feedback",
-    detail: {
-      messageId: evt.detail.messageId,
-      type,
-      feedback: evt.detail.feedback
-    }
-  });
-};
-handleThumbsUp_fn = function(evt) {
-  __privateMethod(this, _AiChatbotComponent_instances, handleFeedback_fn).call(this, evt, "positive");
-};
-handleThumbsDown_fn = function(evt) {
-  __privateMethod(this, _AiChatbotComponent_instances, handleFeedback_fn).call(this, evt, "negative");
-};
-processFileUpload_fn = function(file, timestamp) {
-  const fileId = generateId();
-  __privateGet(this, _fileUploadManager).addAttachment(fileId, {
-    filename: file.name,
-    size: file.size,
-    mimeType: file.type,
-    timestamp
-  });
-  const callbacks = __privateMethod(this, _AiChatbotComponent_instances, createFileUploadCallbacks_fn).call(this, fileId);
-  this.adapter?.emitFileUpload(file, callbacks);
-  __privateMethod(this, _AiChatbotComponent_instances, dispatchEvent_fn).call(this, {
-    type: "forge-ai-chatbot-file-select",
-    detail: {
-      fileId,
-      file,
-      filename: file.name,
-      size: file.size,
-      mimeType: file.type,
-      timestamp,
-      ...callbacks
-    }
-  });
-};
-handleFileSelect_fn = function(evt) {
-  const { file, timestamp } = evt.detail;
-  __privateMethod(this, _AiChatbotComponent_instances, processFileUpload_fn).call(this, file, timestamp);
-};
-createFileUploadCallbacks_fn = function(fileId) {
-  return {
-    updateProgress: (progress) => {
-      __privateGet(this, _fileUploadManager).updateProgress(fileId, progress);
-    },
-    markComplete: (uploadedFile) => {
-      __privateGet(this, _fileUploadManager).markComplete(fileId, uploadedFile);
-    },
-    markError: (error) => {
-      __privateGet(this, _fileUploadManager).markError(fileId, error);
-    },
-    onAbort: (callback) => {
-      __privateGet(this, _fileUploadManager).registerOnAbort(fileId, callback);
-    }
-  };
-};
-handleFileError_fn = function(evt) {
-  const errorMessage = {
-    id: generateId(),
-    role: "assistant",
-    content: evt.detail.message,
-    timestamp: Date.now(),
-    status: "error"
-  };
-  __privateGet(this, _messageStateController).addMessage(errorMessage);
-};
-handleAttachmentRemove_fn = function(evt) {
-  const { filename } = evt.detail;
-  const pendingAttachment = __privateGet(this, _fileUploadManager).pendingAttachments.find((a) => a.filename === filename);
-  if (pendingAttachment) {
-    __privateGet(this, _fileUploadManager).abort(pendingAttachment.id);
-    return;
-  }
-  const completedAttachment = __privateGet(this, _fileUploadManager).completedAttachments.find((a) => a.filename === filename);
-  if (completedAttachment && completedAttachment.fileId) {
-    __privateGet(this, _fileUploadManager).removeCompletedAttachment(completedAttachment.id);
-    this.requestUpdate();
-    this.adapter?.emitFileRemove(completedAttachment.fileId, {
-      onSuccess: () => {
-      },
-      onError: (error) => {
-        console.error(`Failed to remove file ${filename}:`, error);
-      }
-    });
-  }
-};
-handleSuggestionSelect_fn = async function(evt) {
-  __privateGet(this, _promptRef).value?.addToHistory(evt.detail.text);
-  await this.sendMessage(evt.detail.text);
-  __privateGet(this, _promptRef).value?.focus();
-};
-handleVoiceInputResult_fn = function(evt) {
-  const { transcript } = evt.detail;
-  if (transcript && __privateGet(this, _promptRef).value) {
-    __privateGet(this, _promptRef).value.value = transcript;
-  }
-};
 handleHeaderExpand_fn = function() {
-  __privateMethod(this, _AiChatbotComponent_instances, dispatchEvent_fn).call(this, { type: "forge-ai-chatbot-expand" });
+  this._dispatchHostEvent({ type: "forge-ai-chatbot-expand" });
 };
 handleHeaderMinimize_fn = function() {
-  __privateMethod(this, _AiChatbotComponent_instances, dispatchEvent_fn).call(this, { type: "forge-ai-chatbot-minimize" });
+  this._dispatchHostEvent({ type: "forge-ai-chatbot-minimize" });
 };
 handleHeaderClear_fn = function() {
-  const event = __privateMethod(this, _AiChatbotComponent_instances, dispatchEvent_fn).call(this, { type: "forge-ai-chatbot-clear", cancelable: true });
-  if (!event.defaultPrevented) {
-    this.clearMessages();
-  }
+  this.clearMessages();
 };
 handleHeaderInfo_fn = function() {
-  const header = __privateGet(this, _headerRef).value;
-  if (header?.showAgentInfo) {
-    header.showAgentInfo();
-  }
-  __privateMethod(this, _AiChatbotComponent_instances, dispatchEvent_fn).call(this, { type: "forge-ai-chatbot-info" });
-};
-handleAgentChange_fn = function(event) {
-  const { agent, previousAgentId } = event.detail;
-  const changeEvt = __privateMethod(this, _AiChatbotComponent_instances, dispatchEvent_fn).call(this, {
-    type: "forge-ai-chatbot-agent-change",
-    detail: { agent, previousAgentId }
-  });
-  if (!changeEvt.defaultPrevented) {
-    this.selectedAgentId = agent?.id;
-    if (this.adapter) {
-      this.adapter.threadId = generateId();
-    }
-    if (__privateGet(this, _AiChatbotComponent_instances, hasMessages_get)) {
-      const agentName = agent?.name ?? this.titleText;
-      const systemMessage = {
-        id: generateId(),
-        role: "system",
-        content: `Switched to ${agentName}`,
-        timestamp: Date.now(),
-        status: "complete",
-        clientOnly: true
-      };
-      __privateGet(this, _messageStateController).addMessage(systemMessage);
-    }
-  }
-};
-handleDebugToggle_fn = function() {
-  this.debugMode = !this.debugMode;
-};
-handleSlashCommand_fn = function(evt) {
-  const commandId = evt.detail.commandId;
-  switch (commandId) {
-    case "clear":
-      __privateMethod(this, _AiChatbotComponent_instances, handleHeaderClear_fn).call(this);
-      break;
-    case "export":
-      __privateMethod(this, _AiChatbotComponent_instances, handleExport_fn).call(this);
-      break;
-    case "info":
-      __privateMethod(this, _AiChatbotComponent_instances, handleHeaderInfo_fn).call(this);
-      break;
-    case "debug":
-      __privateMethod(this, _AiChatbotComponent_instances, handleDebugToggle_fn).call(this);
-      break;
-  }
-};
-formatToolCallForExport_fn = function(toolCall) {
-  const lines = [`  Tool: ${toolCall.name}`, `  Args: ${JSON.stringify(toolCall.args)}`];
-  if (toolCall.status === "error") {
-    const errorMsg = typeof toolCall.result === "object" && toolCall.result !== null && "error" in toolCall.result ? toolCall.result.error : "Unknown error";
-    lines.push(`  Error: ${errorMsg}`);
-  } else if (toolCall.result !== void 0) {
-    lines.push(`  Result: ${JSON.stringify(toolCall.result)}`);
-  }
-  return lines.join("\n");
-};
-handleExport_fn = function() {
-  const messages = this.getMessages();
-  if (messages.length === 0) {
-    return;
-  }
-  const chatText = messages.map((message) => {
-    const timestamp = new Date(message.timestamp).toLocaleString();
-    const role = message.role === "user" ? "You" : "Assistant";
-    let output = `[${timestamp}] ${role}:
-${message.content}
-`;
-    if (message.toolCalls?.length) {
-      output += "\n" + message.toolCalls.map((tc) => __privateMethod(this, _AiChatbotComponent_instances, formatToolCallForExport_fn).call(this, tc)).join("\n\n") + "\n";
-    }
-    return output;
-  }).join("\n");
-  const filename = `chat-history-${(/* @__PURE__ */ new Date()).toISOString().slice(0, 19).replace(/:/g, "-")}.txt`;
-  downloadFile(chatText, filename, "text/plain");
+  this._handleInfo();
 };
 sessionFilesTemplate_get = function() {
-  const completed = __privateGet(this, _fileUploadManager).completedAttachments;
-  const uploading = __privateGet(this, _fileUploadManager).pendingAttachments;
-  const allFiles = [...uploading, ...completed];
-  if (allFiles.length === 0) {
+  const content = this._sessionFilesTemplate;
+  if (content === nothing) {
     return nothing;
   }
-  return html`
-      <div class="session-files" slot="attachments">
-        <div class="session-files-header">Session Files (${allFiles.length})</div>
-        <div class="session-files-list">
-          ${allFiles.map(
-    (attachment) => html`
-              <forge-ai-attachment
-                .filename=${attachment.filename}
-                .size=${attachment.size}
-                ?uploading=${attachment.uploading ?? false}
-                removable
-                @forge-ai-attachment-remove=${__privateMethod(this, _AiChatbotComponent_instances, handleAttachmentRemove_fn)}>
-              </forge-ai-attachment>
-            `
-  )}
-        </div>
-      </div>
-    `;
+  return html`<div class="session-files" slot="attachments">${content}</div>`;
 };
 promptSlot_get = function() {
-  const isUploading = __privateGet(this, _AiChatbotComponent_instances, isUploading_get);
+  const isUploading = this._isUploading;
   return html`
       <forge-ai-prompt
-        ${ref(__privateGet(this, _promptRef))}
+        ${ref(this._promptRef)}
         slot="prompt"
         .placeholder=${this.placeholder}
-        .running=${__privateGet(this, _AiChatbotComponent_instances, isStreaming_get) || isUploading}
-        .slashCommands=${__privateGet(this, _AiChatbotComponent_instances, slashCommands_get)}
+        .running=${this._isStreaming || isUploading}
+        .slashCommands=${this._slashCommands}
         ?disabled=${isUploading}
         ?debug-mode=${this.debugMode}
-        @forge-ai-prompt-send=${__privateMethod(this, _AiChatbotComponent_instances, handleSend_fn)}
-        @forge-ai-prompt-stop=${__privateMethod(this, _AiChatbotComponent_instances, handleStop_fn)}
-        @forge-ai-prompt-cancel=${__privateMethod(this, _AiChatbotComponent_instances, handleCancel_fn)}
-        @forge-ai-prompt-debug-toggle=${__privateMethod(this, _AiChatbotComponent_instances, handleDebugToggle_fn)}
-        @forge-ai-prompt-command=${__privateMethod(this, _AiChatbotComponent_instances, handleSlashCommand_fn)}>
+        @forge-ai-prompt-send=${this._handleSend}
+        @forge-ai-prompt-stop=${this._handleStop}
+        @forge-ai-prompt-cancel=${this._handleCancel}
+        @forge-ai-prompt-debug-toggle=${this._handleDebugToggle}
+        @forge-ai-prompt-command=${this._handleSlashCommand}>
         ${when(
     this.fileUpload === "on",
     () => html`
@@ -837,16 +141,16 @@ promptSlot_get = function() {
               variant="icon-button"
               multiple
               ?disabled=${isUploading}
-              .selectedFiles=${__privateGet(this, _fileUploadManager).pendingAttachments.map((a) => a.filename)}
-              @forge-ai-file-picker-change=${__privateMethod(this, _AiChatbotComponent_instances, handleFileSelect_fn)}
-              @forge-ai-file-picker-error=${__privateMethod(this, _AiChatbotComponent_instances, handleFileError_fn)}>
+              .selectedFiles=${this._coreController?.pendingAttachments.map((a) => a.filename) ?? []}
+              @forge-ai-file-picker-change=${this._handleFileSelect}
+              @forge-ai-file-picker-error=${this._handleFileError}>
             </forge-ai-file-picker>
           `
   )}
         ${when(
     this.voiceInput === "on",
     () => html`
-            <forge-ai-voice-input slot="actions-end" @forge-ai-voice-input-result=${__privateMethod(this, _AiChatbotComponent_instances, handleVoiceInputResult_fn)}>
+            <forge-ai-voice-input slot="actions-end" @forge-ai-voice-input-result=${this._handleVoiceInputResult}>
             </forge-ai-voice-input>
           `
   )}
@@ -856,69 +160,32 @@ promptSlot_get = function() {
 messageThread_get = function() {
   return html`
       <forge-ai-message-thread
-        ${ref(__privateGet(this, _messageThreadRef))}
-        .messageItems=${__privateGet(this, _AiChatbotComponent_instances, messageItems_get)}
-        .tools=${__privateGet(this, _AiChatbotComponent_instances, tools_get)}
+        ${ref(this._messageThreadRef)}
+        .messageItems=${this._messageItems}
+        .tools=${this._tools}
         ?enable-reactions=${this.enableReactions}
-        ?show-thinking=${__privateGet(this, _AiChatbotComponent_instances, isStreaming_get)}
+        ?show-thinking=${this._isStreaming}
         ?debug-mode=${this.debugMode}
-        @forge-ai-message-thread-copy=${__privateMethod(this, _AiChatbotComponent_instances, handleCopy_fn)}
-        @forge-ai-message-thread-resend=${__privateMethod(this, _AiChatbotComponent_instances, handleResend_fn)}
-        @forge-ai-message-thread-thumbs-up=${__privateMethod(this, _AiChatbotComponent_instances, handleThumbsUp_fn)}
-        @forge-ai-message-thread-thumbs-down=${__privateMethod(this, _AiChatbotComponent_instances, handleThumbsDown_fn)}
-        @forge-ai-message-thread-user-copy=${__privateMethod(this, _AiChatbotComponent_instances, handleUserCopy_fn)}
-        @forge-ai-message-thread-user-resend=${__privateMethod(this, _AiChatbotComponent_instances, handleUserResend_fn)}
-        @forge-ai-message-thread-user-edit=${__privateMethod(this, _AiChatbotComponent_instances, handleUserEdit_fn)}>
+        @forge-ai-message-thread-copy=${this._handleCopy}
+        @forge-ai-message-thread-resend=${this._handleResend}
+        @forge-ai-message-thread-thumbs-up=${this._handleThumbsUp}
+        @forge-ai-message-thread-thumbs-down=${this._handleThumbsDown}
+        @forge-ai-message-thread-user-copy=${this._handleUserCopy}
+        @forge-ai-message-thread-user-resend=${this._handleUserResend}
+        @forge-ai-message-thread-user-edit=${this._handleUserEdit}>
         <slot name="empty-state-heading" slot="empty-state-heading"></slot>
         <slot name="empty-state-message" slot="empty-state-message"></slot>
         <div class="suggestions-container" slot="empty-state-actions">
           <forge-ai-suggestions
             variant="block"
             .suggestions=${this.suggestions ?? []}
-            @forge-ai-suggestions-select=${__privateMethod(this, _AiChatbotComponent_instances, handleSuggestionSelect_fn)}>
+            @forge-ai-suggestions-select=${this._handleSuggestionSelect}>
           </forge-ai-suggestions>
         </div>
       </forge-ai-message-thread>
     `;
 };
-dispatchEvent_fn = function(config) {
-  const event = new CustomEvent(config.type, {
-    detail: config.detail,
-    bubbles: true,
-    composed: true,
-    cancelable: config.cancelable ?? false
-  });
-  this.dispatchEvent(event);
-  return event;
-};
-dispatchMessageEvent_fn = function(type, messageOrId) {
-  const message = typeof messageOrId === "string" ? __privateGet(this, _messageStateController).getMessage(messageOrId) : messageOrId;
-  if (message) {
-    __privateMethod(this, _AiChatbotComponent_instances, dispatchEvent_fn).call(this, { type, detail: { message } });
-  }
-};
-emitStateChange_fn = function() {
-  __privateMethod(this, _AiChatbotComponent_instances, dispatchEvent_fn).call(this, { type: "forge-ai-chatbot-thread-state-change" });
-};
 AiChatbotComponent.styles = unsafeCSS(styles);
-__decorateClass([
-  property({ attribute: false })
-], AiChatbotComponent.prototype, "adapter", 2);
-__decorateClass([
-  property({ attribute: "file-upload" })
-], AiChatbotComponent.prototype, "fileUpload", 2);
-__decorateClass([
-  property({ attribute: "voice-input" })
-], AiChatbotComponent.prototype, "voiceInput", 2);
-__decorateClass([
-  property({ attribute: "debug-command" })
-], AiChatbotComponent.prototype, "debugCommand", 2);
-__decorateClass([
-  property()
-], AiChatbotComponent.prototype, "placeholder", 2);
-__decorateClass([
-  property({ attribute: false })
-], AiChatbotComponent.prototype, "suggestions", 2);
 __decorateClass([
   property({ type: Boolean, attribute: "show-expand-button" })
 ], AiChatbotComponent.prototype, "showExpandButton", 2);
@@ -931,30 +198,6 @@ __decorateClass([
 __decorateClass([
   property({ attribute: "minimize-icon" })
 ], AiChatbotComponent.prototype, "minimizeIcon", 2);
-__decorateClass([
-  property({ type: Boolean, attribute: "enable-reactions" })
-], AiChatbotComponent.prototype, "enableReactions", 2);
-__decorateClass([
-  property({ type: Object, attribute: false })
-], AiChatbotComponent.prototype, "agentInfo", 2);
-__decorateClass([
-  property({ attribute: "title-text" })
-], AiChatbotComponent.prototype, "titleText", 2);
-__decorateClass([
-  property({ attribute: "heading-level", type: Number })
-], AiChatbotComponent.prototype, "headingLevel", 2);
-__decorateClass([
-  property({ type: Boolean, attribute: "debug-mode" })
-], AiChatbotComponent.prototype, "debugMode", 2);
-__decorateClass([
-  property({ attribute: "disclaimer-text" })
-], AiChatbotComponent.prototype, "disclaimerText", 2);
-__decorateClass([
-  property({ attribute: false })
-], AiChatbotComponent.prototype, "agents", 2);
-__decorateClass([
-  property({ attribute: "selected-agent-id" })
-], AiChatbotComponent.prototype, "selectedAgentId", 2);
 AiChatbotComponent = __decorateClass([
   customElement(AiChatbotComponentTagName)
 ], AiChatbotComponent);