@aomi-labs/react 0.0.0 → 0.2.0

package/dist/index.js

@@ -18,369 +18,640 @@ var __spreadValues = (a, b) => {
 };
 var __spreadProps = (a, b) => __defProps(a, __getOwnPropDescs(b));
 
-// src/api/client.ts
-function toQueryString(payload) {
-  const params = new URLSearchParams();
-  for (const [key, value] of Object.entries(payload)) {
-    if (value === void 0 || value === null) continue;
-    params.set(key, String(value));
-  }
-  const qs = params.toString();
-  return qs ? `?${qs}` : "";
-}
-async function postState(backendUrl, path, payload) {
-  const query = toQueryString(payload);
-  const url = `${backendUrl}${path}${query}`;
-  console.log("\u{1F535} [postState] URL:", url);
-  console.log("\u{1F535} [postState] Payload:", payload);
-  const response = await fetch(url, {
-    method: "POST"
-  });
-  console.log("\u{1F535} [postState] Response status:", response.status);
-  if (!response.ok) {
-    console.error("\u{1F534} [postState] Error:", response.status, response.statusText);
-    throw new Error(`HTTP ${response.status}: ${response.statusText}`);
-  }
-  const data = await response.json();
-  console.log("\u{1F7E2} [postState] Success:", data);
-  return data;
+// src/backend/sse.ts
+function extractSseData(rawEvent) {
+  const dataLines = rawEvent.split("\n").filter((line) => line.startsWith("data:")).map((line) => line.slice(5).trimStart());
+  if (!dataLines.length) return null;
+  return dataLines.join("\n");
 }
-var BackendApi = class {
-  constructor(backendUrl) {
-    this.backendUrl = backendUrl;
-    this.connectionStatus = false;
-    this.eventSource = null;
-    this.updatesEventSources = /* @__PURE__ */ new Map();
-  }
-  async fetchState(sessionId, options) {
-    console.log("\u{1F535} [fetchState] Called with sessionId:", sessionId);
-    const url = `${this.backendUrl}/api/state?session_id=${encodeURIComponent(sessionId)}`;
-    console.log("\u{1F535} [fetchState] URL:", url);
-    const response = await fetch(url, { signal: options == null ? void 0 : options.signal });
-    console.log("\u{1F535} [fetchState] Response status:", response.status, response.statusText);
-    if (!response.ok) {
-      console.error("\u{1F534} [fetchState] Error:", response.status, response.statusText);
-      throw new Error(`HTTP ${response.status}: ${response.statusText}`);
-    }
-    const data = await response.json();
-    console.log("\u{1F7E2} [fetchState] Success:", data);
-    return data;
-  }
-  async postChatMessage(sessionId, message, publicKey) {
-    console.log("\u{1F535} [postChatMessage] Called with sessionId:", sessionId, "message:", message);
-    const result = await postState(this.backendUrl, "/api/chat", {
-      message,
-      session_id: sessionId,
-      public_key: publicKey
-    });
-    console.log("\u{1F7E2} [postChatMessage] Success:", result);
-    return result;
-  }
-  async postSystemMessage(sessionId, message) {
-    console.log("\u{1F535} [postSystemMessage] Called with sessionId:", sessionId, "message:", message);
-    const result = await postState(this.backendUrl, "/api/system", {
-      message,
-      session_id: sessionId
-    });
-    console.log("\u{1F7E2} [postSystemMessage] Success:", result);
-    return result;
-  }
-  async postInterrupt(sessionId) {
-    console.log("\u{1F535} [postInterrupt] Called with sessionId:", sessionId);
-    const result = await postState(this.backendUrl, "/api/interrupt", {
-      session_id: sessionId
-    });
-    console.log("\u{1F7E2} [postInterrupt] Success:", result);
-    return result;
-  }
-  disconnectSSE() {
-    if (this.eventSource) {
-      this.eventSource.close();
-      this.eventSource = null;
+async function readSseStream(stream, signal, onMessage) {
+  const reader = stream.getReader();
+  const decoder = new TextDecoder();
+  let buffer = "";
+  try {
+    while (!signal.aborted) {
+      const { value, done } = await reader.read();
+      if (done) break;
+      buffer += decoder.decode(value, { stream: true });
+      buffer = buffer.replace(/\r/g, "");
+      let separatorIndex = buffer.indexOf("\n\n");
+      while (separatorIndex >= 0) {
+        const rawEvent = buffer.slice(0, separatorIndex);
+        buffer = buffer.slice(separatorIndex + 2);
+        const data = extractSseData(rawEvent);
+        if (data) {
+          onMessage(data);
+        }
+        separatorIndex = buffer.indexOf("\n\n");
+      }
     }
-    this.setConnectionStatus(false);
-  }
-  setConnectionStatus(on) {
-    this.connectionStatus = on;
+  } finally {
+    reader.releaseLock();
   }
-  async connectSSE(sessionId, publicKey) {
-    this.disconnectSSE();
-    try {
-      const url = new URL(`${this.backendUrl}/api/chat/stream`);
-      url.searchParams.set("session_id", sessionId);
-      if (publicKey) {
-        url.searchParams.set("public_key", publicKey);
+}
+function createSseSubscriber({
+  backendUrl,
+  getHeaders,
+  shouldLog = process.env.NODE_ENV !== "production"
+}) {
+  const subscriptions = /* @__PURE__ */ new Map();
+  const subscribe2 = (sessionId, onUpdate, onError) => {
+    const existing = subscriptions.get(sessionId);
+    const listener = { onUpdate, onError };
+    if (existing) {
+      existing.listeners.add(listener);
+      if (shouldLog) {
+        console.debug("[aomi][sse] listener added", {
+          sessionId,
+          listeners: existing.listeners.size
+        });
       }
-      this.eventSource = new EventSource(url.toString());
-      this.eventSource.onopen = () => {
-        console.log("\u{1F310} SSE connection opened to:", url.toString());
-        this.setConnectionStatus(true);
-      };
-      this.eventSource.onmessage = (event) => {
-        try {
-          JSON.parse(event.data);
-        } catch (error) {
-          console.error("Failed to parse SSE data:", error);
+      return () => {
+        existing.listeners.delete(listener);
+        if (shouldLog) {
+          console.debug("[aomi][sse] listener removed", {
+            sessionId,
+            listeners: existing.listeners.size
+          });
+        }
+        if (existing.listeners.size === 0) {
+          existing.stop("unsubscribe");
+          if (subscriptions.get(sessionId) === existing) {
+            subscriptions.delete(sessionId);
+          }
         }
       };
-      this.eventSource.onerror = (error) => {
-        console.error("SSE connection error:", error);
-      };
-    } catch (error) {
-      console.error("Failed to establish SSE connection:", error);
-      this.handleConnectionError(sessionId, publicKey);
-    }
-  }
-  handleConnectionError(sessionId, publicKey) {
-    this.setConnectionStatus(false);
-    let attempt = 0;
-    const total = 3;
-    if (attempt < total) {
-      attempt++;
-      console.log(`Attempting to reconnect (${attempt}/${total})...`);
-      setTimeout(() => {
-        this.connectSSE(sessionId, publicKey);
-      }, 100);
-    } else {
-      console.error("Max reconnection attempts reached");
-      this.setConnectionStatus(false);
-    }
-  }
-  subscribeToUpdatesInternal(sessionId, onUpdate, onError, logLabel) {
-    const updatesUrl = new URL("/api/updates", this.backendUrl);
-    updatesUrl.searchParams.set("session_id", sessionId);
-    const updatesUrlString = updatesUrl.toString();
-    const existing = this.updatesEventSources.get(sessionId);
-    if (existing) {
-      existing.cleanup();
-      this.updatesEventSources.delete(sessionId);
     }
     const subscription = {
-      eventSource: null,
+      abortController: null,
       retries: 0,
       retryTimer: null,
       stopped: false,
-      cleanup: () => {
+      listeners: /* @__PURE__ */ new Set([listener]),
+      stop: (reason) => {
+        var _a;
         subscription.stopped = true;
         if (subscription.retryTimer) {
           clearTimeout(subscription.retryTimer);
           subscription.retryTimer = null;
         }
-        if (subscription.eventSource) {
-          subscription.eventSource.close();
-          subscription.eventSource = null;
+        (_a = subscription.abortController) == null ? void 0 : _a.abort();
+        subscription.abortController = null;
+        if (shouldLog) {
+          console.debug("[aomi][sse] stop", {
+            sessionId,
+            reason,
+            retries: subscription.retries
+          });
         }
       }
     };
     const scheduleRetry = () => {
+      if (subscription.stopped) return;
       subscription.retries += 1;
       const delayMs = Math.min(500 * 2 ** (subscription.retries - 1), 1e4);
-      console.warn(
-        `\u{1F501} [${logLabel}] retrying in ${delayMs}ms (attempt ${subscription.retries})`,
-        { sessionId }
-      );
+      if (shouldLog) {
+        console.debug("[aomi][sse] retry scheduled", {
+          sessionId,
+          delayMs,
+          retries: subscription.retries
+        });
+      }
       subscription.retryTimer = setTimeout(() => {
-        open();
+        void open();
       }, delayMs);
     };
-    const open = () => {
+    const open = async () => {
+      var _a;
       if (subscription.stopped) return;
       if (subscription.retryTimer) {
         clearTimeout(subscription.retryTimer);
         subscription.retryTimer = null;
       }
-      if (subscription.eventSource) {
-        subscription.eventSource.close();
-      }
-      const updatesEventSource = new EventSource(updatesUrlString);
-      subscription.eventSource = updatesEventSource;
-      console.log(`\u{1F514} [updates] subscribed`, updatesUrlString);
-      updatesEventSource.onopen = () => {
-        subscription.retries = 0;
-        console.log("\u{1F514} [updates] open", updatesUrlString);
-      };
-      updatesEventSource.onmessage = (event) => {
-        try {
-          console.log("\u{1F514} [updates] message", { url: updatesUrlString, data: event.data });
-          const parsed = JSON.parse(event.data);
-          onUpdate(parsed);
-        } catch (error) {
-          console.error("Failed to parse system update SSE:", error);
-          onError == null ? void 0 : onError(error);
+      const controller = new AbortController();
+      subscription.abortController = controller;
+      const openedAt = Date.now();
+      try {
+        const response = await fetch(`${backendUrl}/api/updates`, {
+          headers: getHeaders(sessionId),
+          signal: controller.signal
+        });
+        if (!response.ok) {
+          throw new Error(
+            `SSE HTTP ${response.status}: ${response.statusText}`
+          );
         }
-      };
-      updatesEventSource.onerror = (error) => {
-        console.error("System updates SSE error:", {
-          url: updatesUrlString,
-          readyState: updatesEventSource.readyState,
-          error
+        if (!response.body) {
+          throw new Error("SSE response missing body");
+        }
+        subscription.retries = 0;
+        await readSseStream(response.body, controller.signal, (data) => {
+          var _a2, _b;
+          let parsed;
+          try {
+            parsed = JSON.parse(data);
+          } catch (error) {
+            for (const item of subscription.listeners) {
+              (_a2 = item.onError) == null ? void 0 : _a2.call(item, error);
+            }
+            return;
+          }
+          for (const item of subscription.listeners) {
+            try {
+              item.onUpdate(parsed);
+            } catch (error) {
+              (_b = item.onError) == null ? void 0 : _b.call(item, error);
+            }
+          }
         });
-        onError == null ? void 0 : onError(error);
-        if (subscription.stopped) return;
-        updatesEventSource.close();
+        if (shouldLog) {
+          console.debug("[aomi][sse] stream ended", {
+            sessionId,
+            aborted: controller.signal.aborted,
+            stopped: subscription.stopped,
+            durationMs: Date.now() - openedAt
+          });
+        }
+      } catch (error) {
+        if (!controller.signal.aborted && !subscription.stopped) {
+          for (const item of subscription.listeners) {
+            (_a = item.onError) == null ? void 0 : _a.call(item, error);
+          }
+        }
+      }
+      if (!subscription.stopped) {
         scheduleRetry();
-      };
+      }
     };
-    this.updatesEventSources.set(sessionId, subscription);
-    open();
+    subscriptions.set(sessionId, subscription);
+    void open();
     return () => {
-      const current = this.updatesEventSources.get(sessionId);
-      if (current === subscription) {
-        current.cleanup();
-        this.updatesEventSources.delete(sessionId);
-      } else {
-        subscription.cleanup();
+      subscription.listeners.delete(listener);
+      if (shouldLog) {
+        console.debug("[aomi][sse] listener removed", {
+          sessionId,
+          listeners: subscription.listeners.size
+        });
+      }
+      if (subscription.listeners.size === 0) {
+        subscription.stop("unsubscribe");
+        if (subscriptions.get(sessionId) === subscription) {
+          subscriptions.delete(sessionId);
+        }
       }
     };
+  };
+  return { subscribe: subscribe2 };
+}
+
+// src/backend/client.ts
+var SESSION_ID_HEADER = "X-Session-Id";
+function toQueryString(payload) {
+  const params = new URLSearchParams();
+  for (const [key, value] of Object.entries(payload)) {
+    if (value === void 0 || value === null) continue;
+    params.set(key, String(value));
   }
-  subscribeToUpdates(sessionId, onUpdate, onError) {
-    return this.subscribeToUpdatesInternal(
-      sessionId,
-      onUpdate,
-      onError,
-      "subscribeToUpdates"
+  const qs = params.toString();
+  return qs ? `?${qs}` : "";
+}
+function withSessionHeader(sessionId, init) {
+  const headers = new Headers(init);
+  headers.set(SESSION_ID_HEADER, sessionId);
+  return headers;
+}
+async function postState(backendUrl, path, payload, sessionId) {
+  const query = toQueryString(payload);
+  const url = `${backendUrl}${path}${query}`;
+  const response = await fetch(url, {
+    method: "POST",
+    headers: withSessionHeader(sessionId)
+  });
+  if (!response.ok) {
+    throw new Error(`HTTP ${response.status}: ${response.statusText}`);
+  }
+  return await response.json();
+}
+var BackendApi = class {
+  constructor(backendUrl) {
+    this.backendUrl = backendUrl;
+    this.sseSubscriber = createSseSubscriber({
+      backendUrl,
+      getHeaders: (sessionId) => withSessionHeader(sessionId, { Accept: "text/event-stream" })
+    });
+  }
+  async fetchState(sessionId, userState) {
+    const url = new URL("/api/state", this.backendUrl);
+    if (userState) {
+      url.searchParams.set("user_state", JSON.stringify(userState));
+    }
+    const response = await fetch(url.toString(), {
+      headers: withSessionHeader(sessionId)
+    });
+    if (!response.ok) {
+      throw new Error(`HTTP ${response.status}: ${response.statusText}`);
+    }
+    return await response.json();
+  }
+  async postChatMessage(sessionId, message, publicKey) {
+    return postState(
+      this.backendUrl,
+      "/api/chat",
+      {
+        message,
+        public_key: publicKey
+      },
+      sessionId
+    );
+  }
+  async postSystemMessage(sessionId, message) {
+    return postState(
+      this.backendUrl,
+      "/api/system",
+      {
+        message
+      },
+      sessionId
+    );
+  }
+  async postInterrupt(sessionId) {
+    return postState(
+      this.backendUrl,
+      "/api/interrupt",
+      {},
+      sessionId
     );
   }
+  /**
+   * Subscribe to SSE updates for a session.
+   * Uses fetch streaming and reconnects on disconnects.
+   * Returns an unsubscribe function.
+   */
+  subscribeSSE(sessionId, onUpdate, onError) {
+    return this.sseSubscriber.subscribe(sessionId, onUpdate, onError);
+  }
   async fetchThreads(publicKey) {
-    console.log("\u{1F535} [fetchThreads] Called with publicKey:", publicKey);
     const url = `${this.backendUrl}/api/sessions?public_key=${encodeURIComponent(publicKey)}`;
-    console.log("\u{1F535} [fetchThreads] URL:", url);
     const response = await fetch(url);
-    console.log("\u{1F535} [fetchThreads] Response status:", response.status);
     if (!response.ok) {
-      console.error("\u{1F534} [fetchThreads] Error:", response.status);
       throw new Error(`Failed to fetch threads: HTTP ${response.status}`);
     }
-    const data = await response.json();
-    console.log("\u{1F7E2} [fetchThreads] Success:", data);
-    return data;
+    return await response.json();
   }
-  async createThread(publicKey, title) {
-    console.log("\u{1F535} [createThread] Called with publicKey:", publicKey, "title:", title);
-    const body = {};
-    if (publicKey) {
-      body.public_key = publicKey;
-    }
-    if (title) {
-      body.title = title;
+  async fetchThread(sessionId) {
+    const url = `${this.backendUrl}/api/sessions/${encodeURIComponent(sessionId)}`;
+    const response = await fetch(url, {
+      headers: withSessionHeader(sessionId)
+    });
+    if (!response.ok) {
+      throw new Error(`HTTP ${response.status}: ${response.statusText}`);
     }
-    console.log("\u{1F535} [createThread] Request body:", body);
+    return await response.json();
+  }
+  async createThread(threadId, publicKey) {
+    const body = {};
+    if (publicKey) body.public_key = publicKey;
     const url = `${this.backendUrl}/api/sessions`;
-    console.log("\u{1F535} [createThread] URL:", url);
     const response = await fetch(url, {
       method: "POST",
-      headers: { "Content-Type": "application/json" },
+      headers: withSessionHeader(threadId, {
+        "Content-Type": "application/json"
+      }),
       body: JSON.stringify(body)
     });
-    console.log("\u{1F535} [createThread] Response status:", response.status);
     if (!response.ok) {
-      console.error("\u{1F534} [createThread] Error:", response.status);
       throw new Error(`Failed to create thread: HTTP ${response.status}`);
     }
-    const data = await response.json();
-    console.log("\u{1F7E2} [createThread] Success:", data);
-    return data;
+    return await response.json();
   }
   async archiveThread(sessionId) {
-    console.log("\u{1F535} [archiveThread] Called with sessionId:", sessionId);
     const url = `${this.backendUrl}/api/sessions/${encodeURIComponent(sessionId)}/archive`;
-    console.log("\u{1F535} [archiveThread] URL:", url);
-    const response = await fetch(url, { method: "POST" });
-    console.log("\u{1F535} [archiveThread] Response status:", response.status);
+    const response = await fetch(url, {
+      method: "POST",
+      headers: withSessionHeader(sessionId)
+    });
     if (!response.ok) {
-      console.error("\u{1F534} [archiveThread] Error:", response.status);
       throw new Error(`Failed to archive thread: HTTP ${response.status}`);
     }
-    console.log("\u{1F7E2} [archiveThread] Success");
   }
   async unarchiveThread(sessionId) {
-    console.log("\u{1F535} [unarchiveThread] Called with sessionId:", sessionId);
     const url = `${this.backendUrl}/api/sessions/${encodeURIComponent(sessionId)}/unarchive`;
-    console.log("\u{1F535} [unarchiveThread] URL:", url);
-    const response = await fetch(url, { method: "POST" });
-    console.log("\u{1F535} [unarchiveThread] Response status:", response.status);
+    const response = await fetch(url, {
+      method: "POST",
+      headers: withSessionHeader(sessionId)
+    });
     if (!response.ok) {
-      console.error("\u{1F534} [unarchiveThread] Error:", response.status);
       throw new Error(`Failed to unarchive thread: HTTP ${response.status}`);
     }
-    console.log("\u{1F7E2} [unarchiveThread] Success");
   }
   async deleteThread(sessionId) {
-    console.log("\u{1F535} [deleteThread] Called with sessionId:", sessionId);
     const url = `${this.backendUrl}/api/sessions/${encodeURIComponent(sessionId)}`;
-    console.log("\u{1F535} [deleteThread] URL:", url);
-    const response = await fetch(url, { method: "DELETE" });
-    console.log("\u{1F535} [deleteThread] Response status:", response.status);
+    const response = await fetch(url, {
+      method: "DELETE",
+      headers: withSessionHeader(sessionId)
+    });
     if (!response.ok) {
-      console.error("\u{1F534} [deleteThread] Error:", response.status);
       throw new Error(`Failed to delete thread: HTTP ${response.status}`);
     }
-    console.log("\u{1F7E2} [deleteThread] Success");
   }
   async renameThread(sessionId, newTitle) {
-    console.log("\u{1F535} [renameThread] Called with sessionId:", sessionId, "newTitle:", newTitle);
     const url = `${this.backendUrl}/api/sessions/${encodeURIComponent(sessionId)}`;
-    console.log("\u{1F535} [renameThread] URL:", url);
     const response = await fetch(url, {
       method: "PATCH",
-      headers: { "Content-Type": "application/json" },
+      headers: withSessionHeader(sessionId, {
+        "Content-Type": "application/json"
+      }),
       body: JSON.stringify({ title: newTitle })
     });
-    console.log("\u{1F535} [renameThread] Response status:", response.status);
     if (!response.ok) {
-      console.error("\u{1F534} [renameThread] Error:", response.status);
       throw new Error(`Failed to rename thread: HTTP ${response.status}`);
     }
-    console.log("\u{1F7E2} [renameThread] Success");
   }
-  async fetchEventsAfter(sessionId, afterId = 0, limit = 100) {
+  async getSystemEvents(sessionId, count) {
     const url = new URL("/api/events", this.backendUrl);
-    url.searchParams.set("session_id", sessionId);
-    if (afterId > 0) url.searchParams.set("after_id", String(afterId));
-    if (limit) url.searchParams.set("limit", String(limit));
-    console.log("\u{1F535} [fetchEventsAfter] URL:", url.toString());
-    const response = await fetch(url.toString());
+    if (count !== void 0) {
+      url.searchParams.set("count", String(count));
+    }
+    const response = await fetch(url.toString(), {
+      headers: withSessionHeader(sessionId)
+    });
     if (!response.ok) {
-      throw new Error(`Failed to fetch events: HTTP ${response.status}`);
+      if (response.status === 404) return [];
+      throw new Error(`Failed to get system events: HTTP ${response.status}`);
     }
     return await response.json();
   }
-  subscribeToUpdatesWithNotification(sessionId, onUpdate, onError) {
-    return this.subscribeToUpdatesInternal(
-      sessionId,
-      onUpdate,
-      onError,
-      "subscribeToUpdatesWithNotification"
-    );
-  }
+  // fetchEventsAfter removed: /api/events only supports count now
 };
 
 // src/runtime/aomi-runtime.tsx
-import { useCallback as useCallback3, useEffect as useEffect3, useMemo as useMemo2, useRef as useRef4, useState as useState3 } from "react";
+import { useMemo as useMemo3 } from "react";
+
+// src/contexts/event-context.tsx
 import {
-  AssistantRuntimeProvider,
-  useExternalStoreRuntime
-} from "@assistant-ui/react";
+  createContext,
+  useCallback,
+  useContext,
+  useEffect,
+  useRef,
+  useState
+} from "react";
+
+// src/backend/types.ts
+function isInlineCall(event) {
+  return "InlineCall" in event;
+}
+function isSystemNotice(event) {
+  return "SystemNotice" in event;
+}
+function isSystemError(event) {
+  return "SystemError" in event;
+}
+function isAsyncCallback(event) {
+  return "AsyncCallback" in event;
+}
+
+// src/state/event-buffer.ts
+function createEventBuffer() {
+  return {
+    inboundQueue: [],
+    outboundQueue: [],
+    sseStatus: "disconnected",
+    lastEventId: null,
+    subscribers: /* @__PURE__ */ new Map()
+  };
+}
+function enqueueInbound(state, event) {
+  state.inboundQueue.push(__spreadProps(__spreadValues({}, event), {
+    status: "pending",
+    timestamp: Date.now()
+  }));
+}
+function subscribe(state, type, callback) {
+  if (!state.subscribers.has(type)) {
+    state.subscribers.set(type, /* @__PURE__ */ new Set());
+  }
+  state.subscribers.get(type).add(callback);
+  return () => {
+    var _a;
+    (_a = state.subscribers.get(type)) == null ? void 0 : _a.delete(callback);
+  };
+}
+function dispatch(state, event) {
+  const typeSubscribers = state.subscribers.get(event.type);
+  if (typeSubscribers) {
+    for (const callback of typeSubscribers) {
+      callback(event);
+    }
+  }
+  const allSubscribers = state.subscribers.get("*");
+  if (allSubscribers) {
+    for (const callback of allSubscribers) {
+      callback(event);
+    }
+  }
+}
+function setSSEStatus(state, status) {
+  state.sseStatus = status;
+}
 
-// src/runtime/hooks.ts
-import { createContext, useContext } from "react";
-var RuntimeActionsContext = createContext(void 0);
-var RuntimeActionsProvider = RuntimeActionsContext.Provider;
-function useRuntimeActions() {
-  const context = useContext(RuntimeActionsContext);
+// src/contexts/event-context.tsx
+import { jsx } from "react/jsx-runtime";
+var EventContextState = createContext(null);
+function useEventContext() {
+  const context = useContext(EventContextState);
   if (!context) {
-    throw new Error("useRuntimeActions must be used within AomiRuntimeProvider");
+    throw new Error(
+      "useEventContext must be used within EventContextProvider. Wrap your app with <EventContextProvider>...</EventContextProvider>"
+    );
   }
   return context;
 }
+function EventContextProvider({
+  children,
+  backendApi,
+  sessionId
+}) {
+  const bufferRef = useRef(null);
+  if (!bufferRef.current) {
+    bufferRef.current = createEventBuffer();
+  }
+  const buffer = bufferRef.current;
+  const [sseStatus, setSseStatus] = useState("disconnected");
+  useEffect(() => {
+    setSSEStatus(buffer, "connecting");
+    setSseStatus("connecting");
+    const unsubscribe = backendApi.subscribeSSE(
+      sessionId,
+      (event) => {
+        enqueueInbound(buffer, {
+          type: event.type,
+          sessionId: event.session_id,
+          payload: event
+        });
+        const inboundEvent = {
+          type: event.type,
+          sessionId: event.session_id,
+          payload: event,
+          status: "fetched",
+          timestamp: Date.now()
+        };
+        dispatch(buffer, inboundEvent);
+      },
+      (error) => {
+        console.error("SSE error:", error);
+        setSSEStatus(buffer, "disconnected");
+        setSseStatus("disconnected");
+      }
+    );
+    setSSEStatus(buffer, "connected");
+    setSseStatus("connected");
+    return () => {
+      unsubscribe();
+      setSSEStatus(buffer, "disconnected");
+      setSseStatus("disconnected");
+    };
+  }, [backendApi, sessionId, buffer]);
+  const subscribeCallback = useCallback(
+    (type, callback) => {
+      return subscribe(buffer, type, callback);
+    },
+    [buffer]
+  );
+  const sendOutbound = useCallback(
+    async (event) => {
+      try {
+        const message = JSON.stringify({
+          type: event.type,
+          payload: event.payload
+        });
+        await backendApi.postSystemMessage(event.sessionId, message);
+      } catch (error) {
+        console.error("Failed to send outbound event:", error);
+      }
+    },
+    [backendApi]
+  );
+  const dispatchSystemEvents = useCallback(
+    (sessionId2, events) => {
+      var _a;
+      for (const event of events) {
+        let eventType;
+        let payload;
+        if (isInlineCall(event)) {
+          eventType = event.InlineCall.type;
+          payload = (_a = event.InlineCall.payload) != null ? _a : event.InlineCall;
+        } else if (isSystemNotice(event)) {
+          eventType = "system_notice";
+          payload = { message: event.SystemNotice };
+        } else if (isSystemError(event)) {
+          eventType = "system_error";
+          payload = { message: event.SystemError };
+        } else if (isAsyncCallback(event)) {
+          eventType = "async_callback";
+          payload = event.AsyncCallback;
+        } else {
+          console.warn("Unknown system event type:", event);
+          continue;
+        }
+        const inboundEvent = {
+          type: eventType,
+          sessionId: sessionId2,
+          payload,
+          status: "fetched",
+          timestamp: Date.now()
+        };
+        enqueueInbound(buffer, {
+          type: eventType,
+          sessionId: sessionId2,
+          payload
+        });
+        dispatch(buffer, inboundEvent);
+      }
+    },
+    [buffer]
+  );
+  const contextValue = {
+    subscribe: subscribeCallback,
+    sendOutboundSystem: sendOutbound,
+    dispatchInboundSystem: dispatchSystemEvents,
+    sseStatus
+  };
+  return /* @__PURE__ */ jsx(EventContextState.Provider, { value: contextValue, children });
+}
 
-// src/runtime/orchestrator.ts
-import { useCallback, useEffect, useRef as useRef2, useState } from "react";
+// src/contexts/notification-context.tsx
+import {
+  createContext as createContext2,
+  useCallback as useCallback2,
+  useContext as useContext2,
+  useState as useState2
+} from "react";
+import { jsx as jsx2 } from "react/jsx-runtime";
+var NotificationContext = createContext2(null);
+function useNotification() {
+  const context = useContext2(NotificationContext);
+  if (!context) {
+    throw new Error(
+      "useNotification must be used within NotificationContextProvider"
+    );
+  }
+  return context;
+}
+var notificationIdCounter = 0;
+function generateId() {
+  return `notif-${Date.now()}-${++notificationIdCounter}`;
+}
+function NotificationContextProvider({
+  children
+}) {
+  const [notifications, setNotifications] = useState2([]);
+  const showNotification = useCallback2((params) => {
+    const id = generateId();
+    const notification = __spreadProps(__spreadValues({}, params), {
+      id,
+      timestamp: Date.now()
+    });
+    setNotifications((prev) => [notification, ...prev]);
+    return id;
+  }, []);
+  const dismissNotification = useCallback2((id) => {
+    setNotifications((prev) => prev.filter((n) => n.id !== id));
+  }, []);
+  const clearAll = useCallback2(() => {
+    setNotifications([]);
+  }, []);
+  const value = {
+    notifications,
+    showNotification,
+    dismissNotification,
+    clearAll
+  };
+  return /* @__PURE__ */ jsx2(NotificationContext.Provider, { value, children });
+}
 
-// src/state/thread-context.tsx
-import { createContext as createContext2, useContext as useContext2, useMemo, useRef, useSyncExternalStore } from "react";
+// src/contexts/thread-context.tsx
+import {
+  createContext as createContext3,
+  useContext as useContext3,
+  useMemo,
+  useRef as useRef2,
+  useSyncExternalStore
+} from "react";
 
 // src/state/thread-store.ts
+var shouldLogThreadUpdates = process.env.NODE_ENV !== "production";
+var logThreadMetadataChange = (source, threadId, prev, next) => {
+  if (!shouldLogThreadUpdates) return;
+  if (!prev && !next) return;
+  if (!prev || !next) {
+    console.debug(`[aomi][thread:${source}]`, { threadId, prev, next });
+    return;
+  }
+  if (prev.title !== next.title || prev.status !== next.status || prev.lastActiveAt !== next.lastActiveAt) {
+    console.debug(`[aomi][thread:${source}]`, { threadId, prev, next });
+  }
+};
 var ThreadStore = class {
   constructor(options) {
     this.listeners = /* @__PURE__ */ new Set();
@@ -407,10 +678,24 @@ var ThreadStore = class {
       this.updateState({ threads: new Map(nextThreads) });
     };
     this.setThreadMetadata = (updater) => {
-      const nextMetadata = this.resolveStateAction(updater, this.state.threadMetadata);
-      this.updateState({ threadMetadata: new Map(nextMetadata) });
-    };
-    this.setThreadMessages = (threadId, messages) => {
+      const prevMetadata = this.state.threadMetadata;
+      const nextMetadata = this.resolveStateAction(updater, prevMetadata);
+      for (const [threadId, next] of nextMetadata.entries()) {
+        logThreadMetadataChange(
+          "setThreadMetadata",
+          threadId,
+          prevMetadata.get(threadId),
+          next
+        );
+      }
+      for (const [threadId, prev] of prevMetadata.entries()) {
+        if (!nextMetadata.has(threadId)) {
+          logThreadMetadataChange("setThreadMetadata", threadId, prev, void 0);
+        }
+      }
+      this.updateState({ threadMetadata: new Map(nextMetadata) });
+    };
+    this.setThreadMessages = (threadId, messages) => {
       this.ensureThreadExists(threadId);
       const nextThreads = new Map(this.state.threads);
       nextThreads.set(threadId, messages);
@@ -428,8 +713,10 @@ var ThreadStore = class {
       if (!existing) {
         return;
       }
+      const next = __spreadValues(__spreadValues({}, existing), updates);
       const nextMetadata = new Map(this.state.threadMetadata);
-      nextMetadata.set(threadId, __spreadValues(__spreadValues({}, existing), updates));
+      nextMetadata.set(threadId, next);
+      logThreadMetadataChange("updateThreadMetadata", threadId, existing, next);
       this.updateState({ threadMetadata: nextMetadata });
     };
     var _a;
@@ -487,9 +774,9 @@ var ThreadStore = class {
       setCurrentThreadId: this.setCurrentThreadId,
       threadViewKey: this.state.threadViewKey,
       bumpThreadViewKey: this.bumpThreadViewKey,
-      threads: this.state.threads,
+      allThreads: this.state.threads,
       setThreads: this.setThreads,
-      threadMetadata: this.state.threadMetadata,
+      allThreadsMetadata: this.state.threadMetadata,
       setThreadMetadata: this.setThreadMetadata,
       threadCnt: this.state.threadCnt,
       setThreadCnt: this.setThreadCnt,
@@ -501,11 +788,11 @@ var ThreadStore = class {
   }
 };
 
-// src/state/thread-context.tsx
-import { jsx } from "react/jsx-runtime";
-var ThreadContextState = createContext2(null);
+// src/contexts/thread-context.tsx
+import { jsx as jsx3 } from "react/jsx-runtime";
+var ThreadContextState = createContext3(null);
 function useThreadContext() {
-  const context = useContext2(ThreadContextState);
+  const context = useContext3(ThreadContextState);
   if (!context) {
     throw new Error(
       "useThreadContext must be used within ThreadContextProvider. Wrap your app with <ThreadContextProvider>...</ThreadContextProvider>"
@@ -517,24 +804,133 @@ function ThreadContextProvider({
   children,
   initialThreadId
 }) {
-  const storeRef = useRef(null);
+  const storeRef = useRef2(null);
   if (!storeRef.current) {
     storeRef.current = new ThreadStore({ initialThreadId });
   }
   const store = storeRef.current;
-  const value = useSyncExternalStore(store.subscribe, store.getSnapshot);
-  return /* @__PURE__ */ jsx(ThreadContextState.Provider, { value, children });
+  const value = useSyncExternalStore(
+    store.subscribe,
+    store.getSnapshot,
+    store.getSnapshot
+  );
+  return /* @__PURE__ */ jsx3(ThreadContextState.Provider, { value, children });
 }
 function useCurrentThreadMessages() {
   const { currentThreadId, getThreadMessages } = useThreadContext();
-  return useMemo(() => getThreadMessages(currentThreadId), [currentThreadId, getThreadMessages]);
+  return useMemo(
+    () => getThreadMessages(currentThreadId),
+    [currentThreadId, getThreadMessages]
+  );
 }
 function useCurrentThreadMetadata() {
   const { currentThreadId, getThreadMetadata } = useThreadContext();
-  return useMemo(() => getThreadMetadata(currentThreadId), [currentThreadId, getThreadMetadata]);
+  return useMemo(
+    () => getThreadMetadata(currentThreadId),
+    [currentThreadId, getThreadMetadata]
+  );
+}
+
+// src/contexts/user-context.tsx
+import {
+  createContext as createContext4,
+  useCallback as useCallback3,
+  useContext as useContext4,
+  useRef as useRef3,
+  useState as useState3
+} from "react";
+import { jsx as jsx4 } from "react/jsx-runtime";
+var UserContext = createContext4(void 0);
+function useUser() {
+  const context = useContext4(UserContext);
+  if (!context) {
+    throw new Error("useUser must be used within UserContextProvider");
+  }
+  return {
+    user: context.user,
+    setUser: context.setUser,
+    getUserState: context.getUserState,
+    onUserStateChange: context.onUserStateChange
+  };
+}
+function UserContextProvider({ children }) {
+  const [user, setUserState] = useState3({
+    isConnected: false,
+    address: void 0,
+    chainId: void 0,
+    ensName: void 0
+  });
+  const userRef = useRef3(user);
+  userRef.current = user;
+  const StateChangeCallbacks = useRef3(
+    /* @__PURE__ */ new Set()
+  );
+  const setUser = useCallback3((data) => {
+    setUserState((prev) => {
+      const next = __spreadValues(__spreadValues({}, prev), data);
+      StateChangeCallbacks.current.forEach((callback) => {
+        callback(next);
+      });
+      return next;
+    });
+  }, []);
+  const getUserState = useCallback3(() => userRef.current, []);
+  const onUserStateChange = useCallback3(
+    (callback) => {
+      StateChangeCallbacks.current.add(callback);
+      return () => {
+        StateChangeCallbacks.current.delete(callback);
+      };
+    },
+    []
+  );
+  return /* @__PURE__ */ jsx4(
+    UserContext.Provider,
+    {
+      value: {
+        user,
+        setUser,
+        getUserState,
+        onUserStateChange
+      },
+      children
+    }
+  );
 }
 
-// src/utils/conversion.ts
+// src/runtime/core.tsx
+import { useCallback as useCallback5, useEffect as useEffect2, useMemo as useMemo2, useRef as useRef5 } from "react";
+import {
+  AssistantRuntimeProvider,
+  useExternalStoreRuntime
+} from "@assistant-ui/react";
+
+// src/runtime/orchestrator.ts
+import { useCallback as useCallback4, useRef as useRef4, useState as useState4 } from "react";
+
+// src/runtime/utils.ts
+import { clsx } from "clsx";
+import { twMerge } from "tailwind-merge";
+function cn(...inputs) {
+  return twMerge(clsx(inputs));
+}
+var parseTimestamp = (value) => {
+  if (value === void 0 || value === null) return 0;
+  if (typeof value === "number") {
+    return Number.isFinite(value) ? value < 1e12 ? value * 1e3 : value : 0;
+  }
+  const numeric = Number(value);
+  if (!Number.isNaN(numeric)) {
+    return numeric < 1e12 ? numeric * 1e3 : numeric;
+  }
+  const ts = Date.parse(value);
+  return Number.isNaN(ts) ? 0 : ts;
+};
+var isPlaceholderTitle = (title) => {
+  var _a;
+  const normalized = (_a = title == null ? void 0 : title.trim()) != null ? _a : "";
+  return !normalized || normalized.startsWith("#[");
+};
 function toInboundMessage(msg) {
   var _a;
   if (msg.sender === "system") return null;
@@ -543,7 +939,7 @@ function toInboundMessage(msg) {
   if (msg.content) {
     content.push({ type: "text", text: msg.content });
   }
-  const [topic, toolContent] = (_a = parseToolStream(msg.tool_stream)) != null ? _a : [];
+  const [topic, toolContent] = (_a = parseToolPayload(msg)) != null ? _a : [];
   if (topic && toolContent) {
     content.push({
       type: "tool-call",
@@ -565,84 +961,61 @@ function toInboundMessage(msg) {
   }, msg.timestamp && { createdAt: new Date(msg.timestamp) });
   return threadMessage;
 }
-function toInboundSystem(msg) {
-  var _a;
-  const [topic] = (_a = parseToolStream(msg.tool_stream)) != null ? _a : [];
-  const messageText = topic || msg.content || "";
-  const timestamp = parseTimestamp(msg.timestamp);
-  if (!messageText.trim()) return null;
-  return __spreadValues({
-    role: "system",
-    content: [{ type: "text", text: messageText }]
-  }, timestamp && { createdAt: timestamp });
-}
-function parseTimestamp(timestamp) {
-  if (!timestamp) return void 0;
-  const parsed = new Date(timestamp);
-  return Number.isNaN(parsed.valueOf()) ? void 0 : parsed;
+function parseToolPayload(msg) {
+  return parseToolResult(msg.tool_result);
 }
-function parseToolStream(toolStream) {
-  if (!toolStream) return null;
-  if (Array.isArray(toolStream) && toolStream.length === 2) {
-    const [topic, content] = toolStream;
-    return [String(topic), content];
-  }
-  if (typeof toolStream === "object") {
-    const topic = toolStream.topic;
-    const content = toolStream.content;
-    return topic ? [String(topic), String(content)] : null;
+function parseToolResult(toolResult) {
+  if (!toolResult) return null;
+  if (Array.isArray(toolResult) && toolResult.length === 2) {
+    const [topic, content] = toolResult;
+    return [String(topic), String(content != null ? content : "")];
   }
   return null;
 }
-
-// src/runtime/utils.ts
-var isTempThreadId = (id) => id.startsWith("temp-");
-var parseTimestamp2 = (value) => {
-  if (value === void 0 || value === null) return 0;
-  if (typeof value === "number") {
-    return Number.isFinite(value) ? value < 1e12 ? value * 1e3 : value : 0;
-  }
-  const numeric = Number(value);
-  if (!Number.isNaN(numeric)) {
-    return numeric < 1e12 ? numeric * 1e3 : numeric;
+var getNetworkName = (chainId) => {
+  if (chainId === void 0) return "";
+  const id = typeof chainId === "string" ? Number(chainId) : chainId;
+  switch (id) {
+    case 1:
+      return "ethereum";
+    case 137:
+      return "polygon";
+    case 42161:
+      return "arbitrum";
+    case 8453:
+      return "base";
+    case 10:
+      return "optimism";
+    case 11155111:
+      return "sepolia";
+    case 1337:
+    case 31337:
+      return "testnet";
+    case 59140:
+      return "linea-sepolia";
+    case 59144:
+      return "linea";
+    default:
+      return "testnet";
   }
-  const ts = Date.parse(value);
-  return Number.isNaN(ts) ? 0 : ts;
-};
-var isPlaceholderTitle = (title) => {
-  var _a;
-  const normalized = (_a = title == null ? void 0 : title.trim()) != null ? _a : "";
-  return !normalized || normalized.startsWith("#[");
 };
+var formatAddress = (addr) => addr ? `${addr.slice(0, 6)}...${addr.slice(-4)}` : "Connect Wallet";
 
-// src/runtime/backend-state.ts
-function createBakendState() {
+// src/state/backend-state.ts
+function createBackendState() {
   return {
-    tempToBackendId: /* @__PURE__ */ new Map(),
     skipInitialFetch: /* @__PURE__ */ new Set(),
     pendingChat: /* @__PURE__ */ new Map(),
-    pendingSystem: /* @__PURE__ */ new Map(),
     runningThreads: /* @__PURE__ */ new Set(),
     creatingThreadId: null,
     createThreadPromise: null
   };
 }
 function resolveThreadId(state, threadId) {
-  var _a;
-  return (_a = state.tempToBackendId.get(threadId)) != null ? _a : threadId;
+  return threadId;
 }
 function isThreadReady(state, threadId) {
-  if (!isTempThreadId(threadId)) return true;
-  return state.tempToBackendId.has(threadId);
-}
-function setBackendMapping(state, tempId, backendId) {
-  state.tempToBackendId.set(tempId, backendId);
-}
-function findTempIdForBackendId(state, backendId) {
-  for (const [tempId, id] of state.tempToBackendId.entries()) {
-    if (id === backendId) return tempId;
-  }
-  return void 0;
+  return state.creatingThreadId !== threadId;
 }
 function markSkipInitialFetch(state, threadId) {
   state.skipInitialFetch.add(threadId);
@@ -678,17 +1051,6 @@ function hasPendingChat(state, threadId) {
   var _a, _b;
   return ((_b = (_a = state.pendingChat.get(threadId)) == null ? void 0 : _a.length) != null ? _b : 0) > 0;
 }
-function enqueuePendingSystem(state, threadId, text) {
-  var _a;
-  const existing = (_a = state.pendingSystem.get(threadId)) != null ? _a : [];
-  state.pendingSystem.set(threadId, [...existing, text]);
-}
-function dequeuePendingSystem(state, threadId) {
-  var _a;
-  const pending = (_a = state.pendingSystem.get(threadId)) != null ? _a : [];
-  state.pendingSystem.delete(threadId);
-  return pending;
-}
 
 // src/runtime/message-controller.ts
 var MessageController = class {
@@ -703,13 +1065,6 @@ var MessageController = class {
     }
     const threadMessages = [];
     for (const msg of msgs) {
-      if (msg.sender === "system") {
-        const systemMessage = toInboundSystem(msg);
-        if (systemMessage) {
-          threadMessages.push(systemMessage);
-        }
-        continue;
-      }
       const threadMessage = toInboundMessage(msg);
       if (threadMessage) {
         threadMessages.push(threadMessage);
@@ -718,9 +1073,13 @@ var MessageController = class {
     this.getThreadContextApi().setThreadMessages(threadId, threadMessages);
   }
   async outbound(message, threadId) {
-    var _a, _b;
+    var _a, _b, _c;
     const backendState = this.config.backendStateRef.current;
-    const text = message.content.filter((part) => part.type === "text").map((part) => part.text).join("\n");
+    const text = message.content.filter(
+      (part) => part.type === "text"
+    ).map(
+      (part) => part.text
+    ).join("\n");
     if (!text) return;
     const threadState = this.getThreadContextApi();
     const existingMessages = threadState.getThreadMessages(threadId);
@@ -730,7 +1089,9 @@ var MessageController = class {
       createdAt: /* @__PURE__ */ new Date()
     };
     threadState.setThreadMessages(threadId, [...existingMessages, userMessage]);
-    threadState.updateThreadMetadata(threadId, { lastActiveAt: (/* @__PURE__ */ new Date()).toISOString() });
+    threadState.updateThreadMetadata(threadId, {
+      lastActiveAt: (/* @__PURE__ */ new Date()).toISOString()
+    });
     if (!isThreadReady(backendState, threadId)) {
       this.markRunning(threadId, true);
       enqueuePendingChat(backendState, threadId, text);
@@ -740,62 +1101,30 @@ var MessageController = class {
     const publicKey = (_b = (_a = this.config).getPublicKey) == null ? void 0 : _b.call(_a);
     try {
       this.markRunning(threadId, true);
-      if (publicKey) {
-        await this.config.backendApiRef.current.postChatMessage(
-          backendThreadId,
-          text,
-          publicKey
-        );
-      } else {
-        await this.config.backendApiRef.current.postChatMessage(backendThreadId, text);
+      const response = publicKey ? await this.config.backendApiRef.current.postChatMessage(
+        backendThreadId,
+        text,
+        publicKey
+      ) : await this.config.backendApiRef.current.postChatMessage(
+        backendThreadId,
+        text
+      );
+      if (response == null ? void 0 : response.messages) {
+        this.inbound(threadId, response.messages);
       }
-      await this.flushPendingSystem(threadId);
-      this.config.polling.start(threadId);
-    } catch (error) {
-      console.error("Failed to send message:", error);
-      this.markRunning(threadId, false);
-    }
-  }
-  async outboundSystem(threadId, text) {
-    const backendState = this.config.backendStateRef.current;
-    if (!isThreadReady(backendState, threadId)) return;
-    const threadMessages = this.getThreadContextApi().getThreadMessages(threadId);
-    const hasUserMessages = threadMessages.some((msg) => msg.role === "user");
-    if (!hasUserMessages) {
-      enqueuePendingSystem(backendState, threadId, text);
-      return;
-    }
-    await this.outboundSystemInner(threadId, text);
-  }
-  async outboundSystemInner(threadId, text) {
-    const backendState = this.config.backendStateRef.current;
-    const threadState = this.getThreadContextApi();
-    const backendThreadId = resolveThreadId(backendState, threadId);
-    this.markRunning(threadId, true);
-    try {
-      const response = await this.config.backendApiRef.current.postSystemMessage(backendThreadId, text);
-      if (response.res) {
-        const systemMessage = toInboundSystem(response.res);
-        if (systemMessage) {
-          const updatedMessages = [...threadState.getThreadMessages(threadId), systemMessage];
-          threadState.setThreadMessages(threadId, updatedMessages);
-        }
+      if (((_c = response == null ? void 0 : response.system_events) == null ? void 0 : _c.length) && this.config.onSyncEvents) {
+        this.config.onSyncEvents(backendThreadId, response.system_events);
+      }
+      if (response == null ? void 0 : response.is_processing) {
+        this.config.polling.start(threadId);
+      } else if (!this.config.polling.isPolling(threadId)) {
+        this.markRunning(threadId, false);
       }
-      await this.flushPendingSystem(threadId);
-      this.config.polling.start(threadId);
     } catch (error) {
-      console.error("Failed to send system message:", error);
+      console.error("Failed to send message:", error);
       this.markRunning(threadId, false);
     }
   }
-  async flushPendingSystem(threadId) {
-    const backendState = this.config.backendStateRef.current;
-    const pending = dequeuePendingSystem(backendState, threadId);
-    if (!pending.length) return;
-    for (const pendingMessage of pending) {
-      await this.outboundSystemInner(threadId, pendingMessage);
-    }
-  }
   async flushPendingChat(threadId) {
     var _a, _b;
     const backendState = this.config.backendStateRef.current;
@@ -812,7 +1141,10 @@ var MessageController = class {
             publicKey
           );
         } else {
-          await this.config.backendApiRef.current.postChatMessage(backendThreadId, text);
+          await this.config.backendApiRef.current.postChatMessage(
+            backendThreadId,
+            text
+          );
         }
       } catch (error) {
         console.error("Failed to send queued message:", error);
@@ -821,12 +1153,19 @@ var MessageController = class {
     this.config.polling.start(threadId);
   }
   async cancel(threadId) {
+    var _a;
     const backendState = this.config.backendStateRef.current;
     if (!isThreadReady(backendState, threadId)) return;
     this.config.polling.stop(threadId);
     const backendThreadId = resolveThreadId(backendState, threadId);
     try {
-      await this.config.backendApiRef.current.postInterrupt(backendThreadId);
+      const response = await this.config.backendApiRef.current.postInterrupt(backendThreadId);
+      if (response == null ? void 0 : response.messages) {
+        this.inbound(threadId, response.messages);
+      }
+      if (((_a = response == null ? void 0 : response.system_events) == null ? void 0 : _a.length) && this.config.onSyncEvents) {
+        this.config.onSyncEvents(backendThreadId, response.system_events);
+      }
       this.markRunning(threadId, false);
     } catch (error) {
       console.error("Failed to cancel:", error);
@@ -852,29 +1191,28 @@ var PollingController = class {
     this.intervals = /* @__PURE__ */ new Map();
     var _a;
     this.intervalMs = (_a = config.intervalMs) != null ? _a : 500;
-    this.handleSystemEvents = config.handleSystemEvents;
-  }
-  setSystemEventsHandler(handler) {
-    this.handleSystemEvents = handler;
   }
   start(threadId) {
+    var _a, _b;
     const backendState = this.config.backendStateRef.current;
     if (!isThreadReady(backendState, threadId)) return;
-    if (this.intervals.has(threadId)) {
-      return;
-    }
+    if (this.intervals.has(threadId)) return;
     const backendThreadId = resolveThreadId(backendState, threadId);
     setThreadRunning(backendState, threadId, true);
     const tick = async () => {
-      if (!this.intervals.has(threadId)) {
-        return;
-      }
+      var _a2, _b2;
+      if (!this.intervals.has(threadId)) return;
       try {
-        console.log("\u{1F535} [PollingController] Fetching state for threadId:", threadId);
-        const state = await this.config.backendApiRef.current.fetchState(backendThreadId);
-        if (!this.intervals.has(threadId)) {
-          return;
-        }
+        console.log(
+          "[PollingController] Fetching state for threadId:",
+          threadId
+        );
+        const userState = (_b2 = (_a2 = this.config).getUserState) == null ? void 0 : _b2.call(_a2);
+        const state = await this.config.backendApiRef.current.fetchState(
+          backendThreadId,
+          userState
+        );
+        if (!this.intervals.has(threadId)) return;
         this.handleState(threadId, state);
       } catch (error) {
         console.error("Polling error:", error);
@@ -883,9 +1221,7 @@ var PollingController = class {
     };
     const intervalId = setInterval(tick, this.intervalMs);
     this.intervals.set(threadId, intervalId);
-    if (this.config.onStart) {
-      this.config.onStart(threadId);
-    }
+    (_b = (_a = this.config).onStart) == null ? void 0 : _b.call(_a, threadId);
   }
   stop(threadId) {
     var _a, _b;
@@ -906,14 +1242,11 @@ var PollingController = class {
     }
   }
   handleState(threadId, state) {
-    if (state.session_exists === false) {
-      this.stop(threadId);
-      return;
-    }
-    const backendState = this.config.backendStateRef.current;
-    const backendThreadId = resolveThreadId(backendState, threadId);
-    if (this.handleSystemEvents && state.system_events) {
-      this.handleSystemEvents(backendThreadId, threadId, state.system_events);
+    var _a;
+    if (((_a = state.system_events) == null ? void 0 : _a.length) && this.config.onSyncEvents) {
+      const backendState = this.config.backendStateRef.current;
+      const sessionId = resolveThreadId(backendState, threadId);
+      this.config.onSyncEvents(sessionId, state.system_events);
     }
     this.config.applyMessages(threadId, state.messages);
     if (!state.is_processing) {
@@ -923,17 +1256,17 @@ var PollingController = class {
 };
 
 // src/runtime/orchestrator.ts
-function useRuntimeOrchestrator(backendUrl, options) {
+function useRuntimeOrchestrator(backendApi, options) {
   const threadContext = useThreadContext();
-  const threadContextRef = useRef2(threadContext);
+  const threadContextRef = useRef4(threadContext);
   threadContextRef.current = threadContext;
-  const backendApiRef = useRef2(new BackendApi(backendUrl));
-  const backendStateRef = useRef2(createBakendState());
-  const [isRunning, setIsRunning] = useState(false);
-  const messageControllerRef = useRef2(null);
-  const pollingRef = useRef2(null);
-  const systemEventsHandlerRef = useRef2(null);
-  const inFlightRef = useRef2(/* @__PURE__ */ new Map());
+  const backendApiRef = useRef4(backendApi);
+  backendApiRef.current = backendApi;
+  const backendStateRef = useRef4(createBackendState());
+  const [isRunning, setIsRunning] = useState4(false);
+  const messageControllerRef = useRef4(null);
+  const pollingRef = useRef4(null);
+  const pendingFetches = useRef4(/* @__PURE__ */ new Set());
   if (!pollingRef.current) {
     pollingRef.current = new PollingController({
       backendApiRef,
@@ -942,6 +1275,8 @@ function useRuntimeOrchestrator(backendUrl, options) {
         var _a;
         (_a = messageControllerRef.current) == null ? void 0 : _a.inbound(threadId, msgs);
       },
+      onSyncEvents: options == null ? void 0 : options.onSyncEvents,
+      getUserState: options == null ? void 0 : options.getUserState,
       onStart: (threadId) => {
         if (threadContextRef.current.currentThreadId === threadId) {
           setIsRunning(true);
@@ -961,80 +1296,56 @@ function useRuntimeOrchestrator(backendUrl, options) {
       threadContextRef,
       polling: pollingRef.current,
       setGlobalIsRunning: setIsRunning,
-      getPublicKey: options == null ? void 0 : options.getPublicKey
+      getPublicKey: options == null ? void 0 : options.getPublicKey,
+      onSyncEvents: options == null ? void 0 : options.onSyncEvents
     });
   }
-  const setSystemEventsHandler = useCallback((handler) => {
-    var _a;
-    systemEventsHandlerRef.current = handler;
-    (_a = pollingRef.current) == null ? void 0 : _a.setSystemEventsHandler(handler != null ? handler : void 0);
-  }, []);
-  useEffect(() => {
-    return () => {
-      for (const controller of inFlightRef.current.values()) {
-        controller.abort();
-      }
-      inFlightRef.current.clear();
-    };
-  }, []);
-  const ensureInitialState = useCallback(
-    async (threadId) => {
-      var _a, _b;
-      const backendState = backendStateRef.current;
-      if (shouldSkipInitialFetch(backendState, threadId)) {
-        clearSkipInitialFetch(backendState, threadId);
-        if (threadContextRef.current.currentThreadId === threadId) {
-          setIsRunning(false);
-        }
-        return;
+  const ensureInitialState = useCallback4(async (threadId) => {
+    var _a, _b, _c, _d;
+    const backendState = backendStateRef.current;
+    if (shouldSkipInitialFetch(backendState, threadId)) {
+      clearSkipInitialFetch(backendState, threadId);
+      if (threadContextRef.current.currentThreadId === threadId) {
+        setIsRunning(false);
       }
-      if (!isThreadReady(backendState, threadId)) {
-        if (threadContextRef.current.currentThreadId === threadId) {
-          setIsRunning(false);
-        }
-        return;
+      return;
+    }
+    if (!isThreadReady(backendState, threadId)) {
+      if (threadContextRef.current.currentThreadId === threadId) {
+        setIsRunning(false);
       }
-      if (inFlightRef.current.has(threadId)) {
-        return;
+      return;
+    }
+    if (pendingFetches.current.has(threadId)) return;
+    const backendThreadId = resolveThreadId(backendState, threadId);
+    pendingFetches.current.add(threadId);
+    try {
+      const userState = (_a = options == null ? void 0 : options.getUserState) == null ? void 0 : _a.call(options);
+      const state = await backendApiRef.current.fetchState(
+        backendThreadId,
+        userState
+      );
+      (_b = messageControllerRef.current) == null ? void 0 : _b.inbound(threadId, state.messages);
+      if (((_c = state.system_events) == null ? void 0 : _c.length) && (options == null ? void 0 : options.onSyncEvents)) {
+        options.onSyncEvents(backendThreadId, state.system_events);
       }
-      const backendThreadId = resolveThreadId(backendState, threadId);
-      const controller = new AbortController();
-      inFlightRef.current.set(threadId, controller);
-      try {
-        console.log("\u{1F535} [Orchestrator] Fetching initial state for threadId:", threadId);
-        const state = await backendApiRef.current.fetchState(backendThreadId, {
-          signal: controller.signal
-        });
-        (_a = messageControllerRef.current) == null ? void 0 : _a.inbound(threadId, state.messages);
-        if (systemEventsHandlerRef.current && state.system_events) {
-          systemEventsHandlerRef.current(backendThreadId, threadId, state.system_events);
-        }
+      if (threadContextRef.current.currentThreadId === threadId) {
         if (state.is_processing) {
-          if (threadContextRef.current.currentThreadId === threadId) {
-            setIsRunning(true);
-          }
-          (_b = pollingRef.current) == null ? void 0 : _b.start(threadId);
+          setIsRunning(true);
+          (_d = pollingRef.current) == null ? void 0 : _d.start(threadId);
         } else {
-          if (threadContextRef.current.currentThreadId === threadId) {
-            setIsRunning(false);
-          }
-        }
-      } catch (error) {
-        if (controller.signal.aborted) {
-          return;
-        }
-        console.error("Failed to fetch initial state:", error);
-        if (threadContextRef.current.currentThreadId === threadId) {
           setIsRunning(false);
         }
-      } finally {
-        if (inFlightRef.current.get(threadId) === controller) {
-          inFlightRef.current.delete(threadId);
-        }
       }
-    },
-    [backendApiRef, backendStateRef, pollingRef, messageControllerRef, setIsRunning]
-  );
+    } catch (error) {
+      console.error("Failed to fetch initial state:", error);
+      if (threadContextRef.current.currentThreadId === threadId) {
+        setIsRunning(false);
+      }
+    } finally {
+      pendingFetches.current.delete(threadId);
+    }
+  }, []);
   return {
     backendStateRef,
     polling: pollingRef.current,
@@ -1042,595 +1353,275 @@ function useRuntimeOrchestrator(backendUrl, options) {
     isRunning,
     setIsRunning,
     ensureInitialState,
-    setSystemEventsHandler,
     backendApiRef
   };
 }
 
-// src/lib/notification-context.tsx
-import {
-  createContext as createContext3,
-  useCallback as useCallback2,
-  useContext as useContext3,
-  useState as useState2
-} from "react";
-import { jsx as jsx2 } from "react/jsx-runtime";
-var NotificationContext = createContext3(
-  void 0
-);
-function useNotification() {
-  const context = useContext3(NotificationContext);
-  if (!context) {
-    throw new Error("useNotification must be used within NotificationProvider");
-  }
-  return context;
-}
-function NotificationProvider({ children }) {
-  const [notifications, setNotifications] = useState2([]);
-  const showNotification = useCallback2(
-    (notification) => {
-      var _a, _b;
-      const id = `notification-${Date.now()}-${Math.random().toString(36).substr(2, 9)}`;
-      const newNotification = __spreadProps(__spreadValues({}, notification), {
-        id,
-        duration: (_a = notification.duration) != null ? _a : 5e3
-      });
-      setNotifications((prev) => [newNotification, ...prev]);
-      const duration = (_b = newNotification.duration) != null ? _b : 5e3;
-      if (duration > 0) {
-        setTimeout(() => {
-          setNotifications((prev) => prev.filter((n) => n.id !== id));
-        }, duration);
-      }
-    },
-    []
+// src/runtime/threadlist-adapter.ts
+var sortByLastActiveDesc = ([, metaA], [, metaB]) => {
+  const tsA = parseTimestamp(metaA.lastActiveAt);
+  const tsB = parseTimestamp(metaB.lastActiveAt);
+  return tsB - tsA;
+};
+function buildThreadLists(threadMetadata) {
+  const entries = Array.from(threadMetadata.entries()).filter(
+    ([, meta]) => !isPlaceholderTitle(meta.title)
   );
-  const dismissNotification = useCallback2((id) => {
-    setNotifications((prev) => prev.filter((n) => n.id !== id));
-  }, []);
-  return /* @__PURE__ */ jsx2(
-    NotificationContext.Provider,
-    {
-      value: { showNotification, notifications, dismissNotification },
-      children
-    }
+  const regularThreads = entries.filter(([, meta]) => meta.status !== "archived").sort(sortByLastActiveDesc).map(
+    ([id, meta]) => ({
+      id,
+      title: meta.title || "New Chat",
+      status: "regular"
+    })
   );
+  const archivedThreads = entries.filter(([, meta]) => meta.status === "archived").sort(sortByLastActiveDesc).map(
+    ([id, meta]) => ({
+      id,
+      title: meta.title || "New Chat",
+      status: "archived"
+    })
+  );
+  return { regularThreads, archivedThreads };
 }
-
-// src/utils/wallet.ts
-import { useEffect as useEffect2, useRef as useRef3 } from "react";
-var getNetworkName = (chainId) => {
-  if (chainId === void 0) return "";
-  const id = typeof chainId === "string" ? Number(chainId) : chainId;
-  switch (id) {
-    case 1:
-      return "ethereum";
-    case 137:
-      return "polygon";
-    case 42161:
-      return "arbitrum";
-    case 8453:
-      return "base";
-    case 10:
-      return "optimism";
-    case 11155111:
-      return "sepolia";
-    case 1337:
-    case 31337:
-      return "testnet";
-    case 59140:
-      return "linea-sepolia";
-    case 59144:
-      return "linea";
-    default:
-      return "testnet";
-  }
-};
-var formatAddress = (addr) => addr ? `${addr.slice(0, 6)}...${addr.slice(-4)}` : "Connect Wallet";
-function normalizeWalletError(error) {
-  var _a, _b, _c, _d, _e, _f, _g;
-  const e = error;
-  const cause = (_a = e == null ? void 0 : e.cause) != null ? _a : null;
-  const code = (_b = typeof (e == null ? void 0 : e.code) === "number" ? e.code : void 0) != null ? _b : typeof (cause == null ? void 0 : cause.code) === "number" ? cause.code : void 0;
-  const name = (_c = typeof (e == null ? void 0 : e.name) === "string" ? e.name : void 0) != null ? _c : typeof (cause == null ? void 0 : cause.name) === "string" ? cause.name : void 0;
-  const msg = (_g = (_f = (_e = (_d = typeof (e == null ? void 0 : e.shortMessage) === "string" ? e.shortMessage : void 0) != null ? _d : typeof (cause == null ? void 0 : cause.shortMessage) === "string" ? cause.shortMessage : void 0) != null ? _e : typeof (e == null ? void 0 : e.message) === "string" ? e.message : void 0) != null ? _f : typeof (cause == null ? void 0 : cause.message) === "string" ? cause.message : void 0) != null ? _g : "Unknown wallet error";
-  const rejected = code === 4001 || name === "UserRejectedRequestError" || name === "RejectedRequestError" || /user rejected|rejected the request|denied|request rejected|canceled|cancelled/i.test(
-    msg
-  );
-  return { rejected, message: msg };
-}
-function toHexQuantity(value) {
-  const trimmed = value.trim();
-  const asBigInt = BigInt(trimmed);
-  return `0x${asBigInt.toString(16)}`;
-}
-async function pickInjectedProvider(publicKey) {
-  const ethereum = globalThis.ethereum;
-  if (!(ethereum == null ? void 0 : ethereum.request)) return void 0;
-  const candidates = Array.isArray(ethereum.providers) ? ethereum.providers.filter(
-    (p) => !!(p == null ? void 0 : p.request)
-  ) : [ethereum];
-  const target = publicKey == null ? void 0 : publicKey.toLowerCase();
-  if (target) {
-    for (const candidate of candidates) {
-      try {
-        const accounts = await candidate.request({
-          method: "eth_accounts"
-        });
-        const list = Array.isArray(accounts) ? accounts.map((a) => String(a).toLowerCase()) : [];
-        if (list.includes(target)) return candidate;
-      } catch (e) {
-      }
-    }
-  }
-  return candidates[0];
-}
-function WalletSystemMessageEmitter({
-  wallet
+function buildThreadListAdapter({
+  backendStateRef,
+  backendApiRef,
+  threadContext,
+  currentThreadIdRef,
+  polling,
+  userAddress,
+  setIsRunning
 }) {
-  const { sendSystemMessage } = useRuntimeActions();
-  const lastWalletRef = useRef3({ isConnected: false });
-  useEffect2(() => {
-    const prev = lastWalletRef.current;
-    const { address, chainId, isConnected } = wallet;
-    const normalizedAddress = address == null ? void 0 : address.toLowerCase();
-    if (isConnected && normalizedAddress && chainId && (!prev.isConnected || prev.address !== normalizedAddress)) {
-      const networkName = getNetworkName(chainId);
-      const message = `User connected wallet with address ${normalizedAddress} on ${networkName} network (Chain ID: ${chainId}). Ready to help with transactions.`;
-      console.log(message);
-      void sendSystemMessage(message);
-      lastWalletRef.current = {
-        isConnected: true,
-        address: normalizedAddress,
-        chainId
-      };
-      return;
-    }
-    if (!isConnected && prev.isConnected) {
-      void sendSystemMessage("Wallet disconnected by user.");
-      console.log("Wallet disconnected by user.");
-      lastWalletRef.current = { isConnected: false };
-      return;
+  const backendState = backendStateRef.current;
+  const { regularThreads, archivedThreads } = buildThreadLists(
+    threadContext.allThreadsMetadata
+  );
+  const preparePendingThread = (threadId) => {
+    const previousPendingId = backendState.creatingThreadId;
+    if (previousPendingId && previousPendingId !== threadId) {
+      threadContext.setThreadMetadata((prev) => {
+        const next = new Map(prev);
+        next.delete(previousPendingId);
+        return next;
+      });
+      threadContext.setThreads((prev) => {
+        const next = new Map(prev);
+        next.delete(previousPendingId);
+        return next;
+      });
+      backendState.pendingChat.delete(previousPendingId);
+      backendState.skipInitialFetch.delete(previousPendingId);
     }
-    if (isConnected && normalizedAddress && chainId && prev.isConnected && prev.address === normalizedAddress && prev.chainId !== chainId) {
-      const networkName = getNetworkName(chainId);
-      const message = `User switched wallet to ${networkName} network (Chain ID: ${chainId}).`;
-      console.log(message);
-      void sendSystemMessage(message);
-      lastWalletRef.current = {
-        isConnected: true,
-        address: normalizedAddress,
-        chainId
-      };
+    backendState.creatingThreadId = threadId;
+    backendState.pendingChat.delete(threadId);
+    threadContext.setThreadMetadata(
+      (prev) => new Map(prev).set(threadId, {
+        title: "New Chat",
+        status: "pending",
+        lastActiveAt: (/* @__PURE__ */ new Date()).toISOString()
+      })
+    );
+    threadContext.setThreadMessages(threadId, []);
+    threadContext.setCurrentThreadId(threadId);
+    setIsRunning(false);
+    threadContext.bumpThreadViewKey();
+  };
+  const findPendingThreadId = () => {
+    if (backendState.creatingThreadId) return backendState.creatingThreadId;
+    for (const [id, meta] of threadContext.allThreadsMetadata.entries()) {
+      if (meta.status === "pending") return id;
     }
-  }, [wallet, sendSystemMessage]);
-  return null;
-}
-
-// src/runtime/wallet-handler.ts
-var WalletHandler = class {
-  constructor(config) {
-    this.config = config;
-    this.handledRequests = /* @__PURE__ */ new Set();
-    this.queue = [];
-    this.inFlight = false;
-  }
-  handleRequest(sessionId, threadId, request) {
-    if (this.config.getCurrentThreadId() !== threadId) return;
-    const description = request.description || request.topic || "Wallet transaction requested";
-    this.config.showNotification({
-      type: "notice",
-      iconType: "wallet",
-      title: "Transaction Request",
-      message: description
-    });
-    this.enqueue(sessionId, threadId, request);
-    void this.drain();
-  }
-  enqueue(sessionId, threadId, request) {
-    var _a;
-    const key = `${sessionId}:${(_a = request.timestamp) != null ? _a : JSON.stringify(request)}`;
-    if (this.handledRequests.has(key)) return;
-    this.handledRequests.add(key);
-    this.queue.push({ sessionId, threadId, request });
-  }
-  async drain() {
-    var _a, _b, _c;
-    if (this.inFlight) return;
-    const next = this.queue.shift();
-    if (!next) return;
-    this.inFlight = true;
-    try {
-      if (this.config.onWalletTxRequest) {
-        const txHash2 = await this.config.onWalletTxRequest(next.request, {
-          sessionId: next.sessionId,
-          threadId: next.threadId,
-          publicKey: this.config.publicKey
-        });
-        this.config.showNotification({
-          type: "success",
-          iconType: "transaction",
-          title: "Transaction Sent",
-          message: `Hash: ${txHash2}`
-        });
-        await this.config.backendApiRef.current.postSystemMessage(
-          next.sessionId,
-          `Transaction sent: ${txHash2}`
-        );
-        await this.refreshThreadIfCurrent(next.sessionId, next.threadId);
+    return null;
+  };
+  return {
+    threadId: threadContext.currentThreadId,
+    threads: regularThreads,
+    archivedThreads,
+    onSwitchToNewThread: async () => {
+      var _a;
+      const pendingId = findPendingThreadId();
+      if (pendingId) {
+        preparePendingThread(pendingId);
         return;
       }
-      const activeProvider = await pickInjectedProvider(this.config.publicKey);
-      if (!(activeProvider == null ? void 0 : activeProvider.request)) {
-        this.config.showNotification({
-          type: "error",
-          iconType: "wallet",
-          title: "Wallet Not Found",
-          message: "No wallet provider found (window.ethereum missing)."
-        });
-        await this.config.backendApiRef.current.postSystemMessage(
-          next.sessionId,
-          "No wallet provider found (window.ethereum missing)."
+      if (backendState.createThreadPromise) {
+        preparePendingThread(
+          (_a = backendState.creatingThreadId) != null ? _a : crypto.randomUUID()
         );
         return;
       }
-      const accounts = await activeProvider.request({
-        method: "eth_accounts"
-      });
-      const addresses = Array.isArray(accounts) ? accounts.map(String) : [];
-      const from = this.config.publicKey || addresses[0];
-      if (!from) {
-        await activeProvider.request({ method: "eth_requestAccounts" });
-      }
-      const fromAddress = this.config.publicKey || await activeProvider.request({ method: "eth_accounts" });
-      const resolvedFrom = this.config.publicKey || (Array.isArray(fromAddress) ? String((_a = fromAddress[0]) != null ? _a : "") : "");
-      if (!resolvedFrom) {
-        this.config.showNotification({
-          type: "error",
-          iconType: "wallet",
-          title: "Wallet Not Connected",
-          message: "Please connect a wallet to sign the requested transaction."
+      const threadId = crypto.randomUUID();
+      preparePendingThread(threadId);
+      const createPromise = backendApiRef.current.createThread(threadId, userAddress).then(async (newThread) => {
+        var _a2;
+        const uiThreadId = (_a2 = backendState.creatingThreadId) != null ? _a2 : threadId;
+        const backendId = newThread.session_id;
+        if (uiThreadId !== backendId) {
+          console.warn("[aomi][thread] backend id mismatch", {
+            uiThreadId,
+            backendId
+          });
+        }
+        markSkipInitialFetch(backendState, uiThreadId);
+        threadContext.setThreadMetadata((prev) => {
+          var _a3, _b;
+          const next = new Map(prev);
+          const existing = next.get(uiThreadId);
+          const nextStatus = (existing == null ? void 0 : existing.status) === "archived" ? "archived" : "regular";
+          next.set(uiThreadId, {
+            title: (_a3 = existing == null ? void 0 : existing.title) != null ? _a3 : "New Chat",
+            status: nextStatus,
+            lastActiveAt: (_b = existing == null ? void 0 : existing.lastActiveAt) != null ? _b : (/* @__PURE__ */ new Date()).toISOString()
+          });
+          return next;
         });
-        await this.config.backendApiRef.current.postSystemMessage(
-          next.sessionId,
-          "Wallet is not connected; please connect a wallet to sign the requested transaction."
-        );
-        return;
-      }
-      const gas = (_c = (_b = next.request.gas) != null ? _b : next.request.gas_limit) != null ? _c : void 0;
-      let valueHex;
-      let gasHex;
+        if (backendState.creatingThreadId === uiThreadId) {
+          backendState.creatingThreadId = null;
+        }
+        const pendingMessages = backendState.pendingChat.get(uiThreadId);
+        if (pendingMessages == null ? void 0 : pendingMessages.length) {
+          backendState.pendingChat.delete(uiThreadId);
+          for (const text of pendingMessages) {
+            try {
+              await backendApiRef.current.postChatMessage(backendId, text);
+            } catch (error) {
+              console.error("Failed to send queued message:", error);
+            }
+          }
+          if (currentThreadIdRef.current === uiThreadId) {
+            polling == null ? void 0 : polling.start(uiThreadId);
+          }
+        }
+      }).catch((error) => {
+        var _a2;
+        console.error("Failed to create new thread:", error);
+        const failedId = (_a2 = backendState.creatingThreadId) != null ? _a2 : threadId;
+        threadContext.setThreadMetadata((prev) => {
+          const next = new Map(prev);
+          next.delete(failedId);
+          return next;
+        });
+        threadContext.setThreads((prev) => {
+          const next = new Map(prev);
+          next.delete(failedId);
+          return next;
+        });
+        if (backendState.creatingThreadId === failedId) {
+          backendState.creatingThreadId = null;
+        }
+      }).finally(() => {
+        backendState.createThreadPromise = null;
+      });
+      backendState.createThreadPromise = createPromise;
+    },
+    onSwitchToThread: (threadId) => {
+      threadContext.setCurrentThreadId(threadId);
+    },
+    onRename: async (threadId, newTitle) => {
+      var _a, _b;
+      const previousTitle = (_b = (_a = threadContext.getThreadMetadata(threadId)) == null ? void 0 : _a.title) != null ? _b : "";
+      const normalizedTitle = isPlaceholderTitle(newTitle) ? "" : newTitle;
+      threadContext.updateThreadMetadata(threadId, {
+        title: normalizedTitle
+      });
       try {
-        valueHex = toHexQuantity(next.request.value);
-        if (gas) gasHex = toHexQuantity(gas);
+        await backendApiRef.current.renameThread(threadId, newTitle);
       } catch (error) {
-        this.config.showNotification({
-          type: "error",
-          iconType: "transaction",
-          title: "Invalid Transaction",
-          message: error.message
+        console.error("Failed to rename thread:", error);
+        threadContext.updateThreadMetadata(threadId, {
+          title: previousTitle
         });
-        await this.config.backendApiRef.current.postSystemMessage(
-          next.sessionId,
-          `Invalid wallet transaction request payload: ${error.message}`
-        );
-        return;
       }
-      const txParams = __spreadValues({
-        from: resolvedFrom,
-        to: next.request.to,
-        value: valueHex,
-        data: next.request.data
-      }, gasHex ? { gas: gasHex } : {});
-      const txHash = await activeProvider.request({
-        method: "eth_sendTransaction",
-        params: [txParams]
-      });
-      this.config.showNotification({
-        type: "success",
-        title: "Transaction sent",
-        message: `Transaction hash: ${txHash}`
-      });
-      await this.config.backendApiRef.current.postSystemMessage(
-        next.sessionId,
-        `Transaction sent: ${txHash}`
-      );
-      await this.refreshThreadIfCurrent(next.sessionId, next.threadId);
-    } catch (error) {
-      const normalized = normalizeWalletError(error);
-      const final = normalized.rejected ? "Transaction rejected by user." : `Transaction failed: ${normalized.message}`;
-      this.config.showNotification({
-        type: normalized.rejected ? "notice" : "error",
-        iconType: normalized.rejected ? "transaction" : "error",
-        title: normalized.rejected ? "Transaction Rejected" : "Transaction Failed",
-        message: normalized.rejected ? "Transaction was rejected by user." : normalized.message
-      });
+    },
+    onArchive: async (threadId) => {
+      threadContext.updateThreadMetadata(threadId, { status: "archived" });
       try {
-        await this.config.backendApiRef.current.postSystemMessage(
-          next.sessionId,
-          final
-        );
-        await this.refreshThreadIfCurrent(next.sessionId, next.threadId);
-      } catch (postError) {
-        console.error("Failed to report wallet tx result to backend:", postError);
+        await backendApiRef.current.archiveThread(threadId);
+      } catch (error) {
+        console.error("Failed to archive thread:", error);
+        threadContext.updateThreadMetadata(threadId, { status: "regular" });
       }
-    } finally {
-      this.inFlight = false;
-      void this.drain();
-    }
-  }
-  async refreshThreadIfCurrent(sessionId, threadId) {
-    if (this.config.getCurrentThreadId() !== threadId) return;
-    try {
-      const state = await this.config.backendApiRef.current.fetchState(sessionId);
-      this.config.applySessionMessagesToThread(threadId, state.messages);
-    } catch (refreshError) {
-      console.error("Failed to refresh state after wallet tx:", refreshError);
-    }
-  }
-};
-
-// src/runtime/event-controller.ts
-var EventController = class {
-  constructor(config) {
-    this.config = config;
-    this.lastEventIdBySession = /* @__PURE__ */ new Map();
-    this.eventsInFlight = /* @__PURE__ */ new Set();
-    this.updateSubscriptions = /* @__PURE__ */ new Map();
-    this.subscribableSessionId = null;
-  }
-  setSubscribableSessionId(sessionId) {
-    console.log("\u{1F514} [updates] setSubscribableSessionId", { sessionId });
-    this.subscribableSessionId = sessionId;
-    this.syncSubscriptions();
-  }
-  syncSubscriptions() {
-    console.log("\u{1F514} [updates] syncSubscriptions", {
-      subscribableSessionId: this.subscribableSessionId,
-      active: Array.from(this.updateSubscriptions.keys())
-    });
-    const nextSessions = /* @__PURE__ */ new Set();
-    if (this.subscribableSessionId) {
-      nextSessions.add(this.subscribableSessionId);
-    }
-    for (const sessionId of this.updateSubscriptions.keys()) {
-      if (!nextSessions.has(sessionId)) {
-        this.removeSubscription(sessionId);
+    },
+    onUnarchive: async (threadId) => {
+      threadContext.updateThreadMetadata(threadId, { status: "regular" });
+      try {
+        await backendApiRef.current.unarchiveThread(threadId);
+      } catch (error) {
+        console.error("Failed to unarchive thread:", error);
+        threadContext.updateThreadMetadata(threadId, { status: "archived" });
       }
-    }
-    for (const sessionId of nextSessions) {
-      this.ensureSubscription(sessionId);
-    }
-  }
-  ensureSubscription(sessionId) {
-    if (this.updateSubscriptions.has(sessionId)) return;
-    console.log("\u{1F514} [updates] ensureSubscription", { sessionId });
-    const unsubscribe = this.config.backendApiRef.current.subscribeToUpdates(
-      sessionId,
-      (update) => {
-        console.log("\u{1F514} [updates] event_available", {
-          sessionId: update.session_id,
-          eventId: update.event_id,
-          eventType: update.event_type
+    },
+    onDelete: async (threadId) => {
+      try {
+        await backendApiRef.current.deleteThread(threadId);
+        threadContext.setThreadMetadata((prev) => {
+          const next = new Map(prev);
+          next.delete(threadId);
+          return next;
         });
-        if (update.type !== "event_available") return;
-        void this.drainEvents(update.session_id);
-      },
-      (error) => {
-        console.error("Failed to handle system update SSE:", error);
-      }
-    );
-    this.updateSubscriptions.set(sessionId, unsubscribe);
-  }
-  removeSubscription(sessionId) {
-    const unsubscribe = this.updateSubscriptions.get(sessionId);
-    if (!unsubscribe) return;
-    unsubscribe();
-    this.updateSubscriptions.delete(sessionId);
-  }
-  async drainEvents(sessionId) {
-    var _a;
-    if (this.eventsInFlight.has(sessionId)) {
-      console.log("\u23F3 [events] drain already in flight", { sessionId });
-      return;
-    }
-    this.eventsInFlight.add(sessionId);
-    try {
-      let afterId = (_a = this.lastEventIdBySession.get(sessionId)) != null ? _a : 0;
-      console.log("\u{1F4E5} [events] start drain", { sessionId, afterId });
-      for (; ; ) {
-        const requestAfterId = afterId;
-        const events = await this.config.backendApiRef.current.fetchEventsAfter(
-          sessionId,
-          requestAfterId,
-          200
-        );
-        console.log("\u{1F4E5} [events] fetched batch", {
-          sessionId,
-          afterId: requestAfterId,
-          count: events.length
+        threadContext.setThreads((prev) => {
+          const next = new Map(prev);
+          next.delete(threadId);
+          return next;
         });
-        if (!events.length) break;
-        for (const event of events) {
-          const eventId = typeof event.event_id === "number" ? event.event_id : Number(event.event_id);
-          if (Number.isFinite(eventId)) afterId = Math.max(afterId, eventId);
-          if (event.type === "title_changed" && typeof event.new_title === "string") {
-            console.log("\u{1F3F7}\uFE0F [events] title_changed", {
-              sessionId,
-              eventId,
-              newTitle: event.new_title
-            });
-            this.applyTitleChanged(sessionId, event.new_title);
-          }
-          if (event.type === "wallet_tx_request") {
-            const payload = event.payload;
-            if (payload && typeof payload === "object") {
-              const req = payload;
-              if (typeof req.to === "string" && typeof req.value === "string" && typeof req.data === "string") {
-                const threadId = findTempIdForBackendId(
-                  this.config.backendStateRef.current,
-                  sessionId
-                ) || sessionId;
-                this.config.handleWalletTxRequest(
-                  sessionId,
-                  threadId,
-                  req
-                );
-              }
-            }
-          }
+        backendState.pendingChat.delete(threadId);
+        backendState.skipInitialFetch.delete(threadId);
+        backendState.runningThreads.delete(threadId);
+        if (backendState.creatingThreadId === threadId) {
+          backendState.creatingThreadId = null;
         }
-        if (events.length < 200) break;
-      }
-      console.log("\u{1F4E5} [events] drain complete", { sessionId, afterId });
-      this.lastEventIdBySession.set(sessionId, afterId);
-    } catch (error) {
-      console.error("Failed to fetch async events:", error);
-    } finally {
-      this.eventsInFlight.delete(sessionId);
-    }
-  }
-  applyTitleChanged(sessionId, newTitle) {
-    const backendState = this.config.backendStateRef.current;
-    const tempId = findTempIdForBackendId(backendState, sessionId);
-    const threadIdToUpdate = tempId || sessionId;
-    console.log("\u{1F3F7}\uFE0F [title] applying", {
-      sessionId,
-      threadId: threadIdToUpdate,
-      newTitle
-    });
-    this.config.setThreadMetadata((prev) => {
-      var _a;
-      const next = new Map(prev);
-      const existing = next.get(threadIdToUpdate);
-      const normalizedTitle = isPlaceholderTitle(newTitle) ? "" : newTitle;
-      const nextStatus = (existing == null ? void 0 : existing.status) === "archived" ? "archived" : "regular";
-      next.set(threadIdToUpdate, {
-        title: normalizedTitle,
-        status: nextStatus,
-        lastActiveAt: (_a = existing == null ? void 0 : existing.lastActiveAt) != null ? _a : (/* @__PURE__ */ new Date()).toISOString()
-      });
-      return next;
-    });
-    if (!isPlaceholderTitle(newTitle) && backendState.creatingThreadId === threadIdToUpdate) {
-      backendState.creatingThreadId = null;
-    }
-  }
-  handleBackendSystemEvents(sessionId, threadId, rawEvents) {
-    if (!(rawEvents == null ? void 0 : rawEvents.length)) return;
-    for (const raw of rawEvents) {
-      const parsed = this.parseBackendSystemEvent(raw);
-      if (!parsed) continue;
-      if ("InlineDisplay" in parsed) {
-        const payload = parsed.InlineDisplay;
-        if (!payload || typeof payload !== "object") continue;
-        const type = payload.type;
-        if (type !== "wallet_tx_request") continue;
-        const requestValue = payload.payload;
-        if (!requestValue || typeof requestValue !== "object") continue;
-        const req = requestValue;
-        if (typeof req.to !== "string" || typeof req.value !== "string" || typeof req.data !== "string") {
-          continue;
+        if (threadContext.currentThreadId === threadId) {
+          const firstRegularThread = Array.from(
+            threadContext.allThreadsMetadata.entries()
+          ).find(([id, meta]) => meta.status === "regular" && id !== threadId);
+          if (firstRegularThread) {
+            threadContext.setCurrentThreadId(firstRegularThread[0]);
+          } else {
+            const defaultId = "default-session";
+            threadContext.setThreadMetadata(
+              (prev) => new Map(prev).set(defaultId, {
+                title: "New Chat",
+                status: "regular",
+                lastActiveAt: (/* @__PURE__ */ new Date()).toISOString()
+              })
+            );
+            threadContext.setThreadMessages(defaultId, []);
+            threadContext.setCurrentThreadId(defaultId);
+          }
         }
-        this.config.handleWalletTxRequest(
-          sessionId,
-          threadId,
-          req
-        );
-      }
-      if ("SystemError" in parsed) {
-        this.config.showNotification({
-          type: "error",
-          iconType: "error",
-          title: "Error",
-          message: parsed.SystemError
-        });
-      }
-      if ("SystemNotice" in parsed) {
-        this.config.showNotification({
-          type: "notice",
-          iconType: "notice",
-          title: "Notice",
-          message: parsed.SystemNotice
-        });
+      } catch (error) {
+        console.error("Failed to delete thread:", error);
+        throw error;
       }
     }
-  }
-  parseBackendSystemEvent(value) {
-    if (!value || typeof value !== "object") return null;
-    const entries = Object.entries(value);
-    if (entries.length !== 1) return null;
-    const [key, payload] = entries[0];
-    switch (key) {
-      case "InlineDisplay":
-        return { InlineDisplay: payload };
-      case "SystemNotice":
-        return {
-          SystemNotice: typeof payload === "string" ? payload : String(payload)
-        };
-      case "SystemError":
-        return {
-          SystemError: typeof payload === "string" ? payload : String(payload)
-        };
-      case "AsyncUpdate":
-        return { AsyncUpdate: payload };
-      default:
-        return null;
-    }
-  }
-  cleanup() {
-    for (const unsubscribe of this.updateSubscriptions.values()) {
-      unsubscribe();
-    }
-    this.updateSubscriptions.clear();
-  }
-};
+  };
+}
 
-// src/runtime/aomi-runtime.tsx
-import { jsx as jsx3 } from "react/jsx-runtime";
-var sortByLastActiveDesc = ([, metaA], [, metaB]) => {
-  const tsA = parseTimestamp2(metaA.lastActiveAt);
-  const tsB = parseTimestamp2(metaB.lastActiveAt);
-  return tsB - tsA;
-};
-function buildThreadLists(threadMetadata) {
-  const entries = Array.from(threadMetadata.entries()).filter(
-    ([, meta]) => !isPlaceholderTitle(meta.title)
-  );
-  const regularThreads = entries.filter(([, meta]) => meta.status === "regular").sort(sortByLastActiveDesc).map(([id, meta]) => ({
-    id,
-    title: meta.title || "New Chat",
-    status: "regular"
-  }));
-  const archivedThreads = entries.filter(([, meta]) => meta.status === "archived").sort(sortByLastActiveDesc).map(([id, meta]) => ({
-    id,
-    title: meta.title || "New Chat",
-    status: "archived"
-  }));
-  return { regularThreads, archivedThreads };
+// src/interface.tsx
+import { createContext as createContext5, useContext as useContext5 } from "react";
+var AomiRuntimeContext = createContext5(null);
+var AomiRuntimeApiProvider = AomiRuntimeContext.Provider;
+function useAomiRuntime() {
+  const context = useContext5(AomiRuntimeContext);
+  if (!context) {
+    throw new Error(
+      "useAomiRuntime must be used within AomiRuntimeProvider. Wrap your app with <AomiRuntimeProvider>...</AomiRuntimeProvider>"
+    );
+  }
+  return context;
 }
-function AomiRuntimeProvider({
+
+// src/runtime/core.tsx
+import { jsx as jsx5 } from "react/jsx-runtime";
+function AomiRuntimeCore({
   children,
-  backendUrl = "http://localhost:8080",
-  publicKey,
-  onWalletTxRequest
+  backendApi
 }) {
   const threadContext = useThreadContext();
-  const threadContextRef = useRef4(threadContext);
-  threadContextRef.current = threadContext;
-  const currentThreadIdRef = useRef4(threadContext.currentThreadId);
-  useEffect3(() => {
-    currentThreadIdRef.current = threadContext.currentThreadId;
-  }, [threadContext.currentThreadId]);
-  const publicKeyRef = useRef4(publicKey);
-  useEffect3(() => {
-    publicKeyRef.current = publicKey;
-  }, [publicKey]);
-  const getPublicKey = useCallback3(() => publicKeyRef.current, []);
-  const lastSubscribedThreadRef = useRef4(null);
-  const { showNotification } = useNotification();
-  const eventControllerRef = useRef4(null);
-  const walletHandlerRef = useRef4(null);
+  const eventContext = useEventContext();
+  const notificationContext = useNotification();
+  const { dispatchInboundSystem: dispatchSystemEvents } = eventContext;
+  const { user, onUserStateChange, getUserState } = useUser();
   const {
     backendStateRef,
     polling,
@@ -1638,72 +1629,66 @@ function AomiRuntimeProvider({
     isRunning,
     setIsRunning,
     ensureInitialState,
-    setSystemEventsHandler,
     backendApiRef
-  } = useRuntimeOrchestrator(backendUrl, { getPublicKey });
-  if (!walletHandlerRef.current) {
-    walletHandlerRef.current = new WalletHandler({
-      backendApiRef,
-      onWalletTxRequest,
-      publicKey,
-      showNotification,
-      applySessionMessagesToThread: (threadId, msgs) => {
-        messageController.inbound(threadId, msgs);
-      },
-      getCurrentThreadId: () => currentThreadIdRef.current
-    });
-  }
-  if (!eventControllerRef.current) {
-    eventControllerRef.current = new EventController({
-      backendApiRef,
-      backendStateRef,
-      showNotification,
-      handleWalletTxRequest: (sessionId, threadId, request) => {
-        var _a;
-        (_a = walletHandlerRef.current) == null ? void 0 : _a.handleRequest(sessionId, threadId, request);
-      },
-      setThreadMetadata: threadContext.setThreadMetadata
+  } = useRuntimeOrchestrator(backendApi, {
+    onSyncEvents: dispatchSystemEvents,
+    getPublicKey: () => getUserState().address,
+    getUserState
+  });
+  useEffect2(() => {
+    const unsubscribe = onUserStateChange(async (newUser) => {
+      const sessionId = threadContext.currentThreadId;
+      const message = JSON.stringify({
+        type: "wallet:state_changed",
+        payload: {
+          address: newUser.address,
+          chainId: newUser.chainId,
+          isConnected: newUser.isConnected,
+          ensName: newUser.ensName
+        }
+      });
+      await backendApiRef.current.postSystemMessage(sessionId, message);
     });
-  }
-  const handleSystemEvents = useCallback3(
-    (sessionId, threadId, events) => {
-      var _a;
-      (_a = eventControllerRef.current) == null ? void 0 : _a.handleBackendSystemEvents(sessionId, threadId, events);
-    },
-    []
-  );
-  useEffect3(() => {
-    setSystemEventsHandler(handleSystemEvents);
-    return () => {
-      setSystemEventsHandler(null);
-    };
-  }, [handleSystemEvents, setSystemEventsHandler]);
-  const [updateSubscriptionsTick, setUpdateSubscriptionsTick] = useState3(0);
-  const bumpUpdateSubscriptions = useCallback3(() => {
-    setUpdateSubscriptionsTick((prev) => prev + 1);
-  }, []);
-  useEffect3(() => {
+    return unsubscribe;
+  }, [onUserStateChange, backendApiRef, threadContext.currentThreadId]);
+  const threadContextRef = useRef5(threadContext);
+  threadContextRef.current = threadContext;
+  const currentThreadIdRef = useRef5(threadContext.currentThreadId);
+  useEffect2(() => {
+    currentThreadIdRef.current = threadContext.currentThreadId;
+  }, [threadContext.currentThreadId]);
+  useEffect2(() => {
     void ensureInitialState(threadContext.currentThreadId);
   }, [ensureInitialState, threadContext.currentThreadId]);
-  useEffect3(() => {
+  useEffect2(() => {
     const threadId = threadContext.currentThreadId;
-    const isCurrentlyRunning = isThreadRunning(backendStateRef.current, threadId);
-    setIsRunning(isCurrentlyRunning);
-  }, [threadContext.currentThreadId, setIsRunning]);
-  const currentMessages = threadContext.getThreadMessages(threadContext.currentThreadId);
-  useEffect3(() => {
-    if (!publicKey) return;
+    setIsRunning(isThreadRunning(backendStateRef.current, threadId));
+  }, [backendStateRef, setIsRunning, threadContext.currentThreadId]);
+  const currentMessages = threadContext.getThreadMessages(
+    threadContext.currentThreadId
+  );
+  const resolvedSessionId = useMemo2(
+    () => resolveThreadId(backendStateRef.current, threadContext.currentThreadId),
+    [
+      backendStateRef,
+      threadContext.currentThreadId,
+      threadContext.allThreadsMetadata
+    ]
+  );
+  useEffect2(() => {
+    const userAddress = user.address;
+    if (!userAddress) return;
     const fetchThreadList = async () => {
       var _a, _b;
       try {
-        const threadList = await backendApiRef.current.fetchThreads(publicKey);
+        const threadList = await backendApiRef.current.fetchThreads(userAddress);
         const currentContext = threadContextRef.current;
-        const newMetadata = new Map(currentContext.threadMetadata);
+        const newMetadata = new Map(currentContext.allThreadsMetadata);
         let maxChatNum = currentContext.threadCnt;
         for (const thread of threadList) {
           const rawTitle = (_a = thread.title) != null ? _a : "";
           const title = isPlaceholderTitle(rawTitle) ? "" : rawTitle;
-          const lastActive = thread.last_active_at || thread.updated_at || thread.created_at || ((_b = newMetadata.get(thread.session_id)) == null ? void 0 : _b.lastActiveAt) || (/* @__PURE__ */ new Date()).toISOString();
+          const lastActive = ((_b = newMetadata.get(thread.session_id)) == null ? void 0 : _b.lastActiveAt) || (/* @__PURE__ */ new Date()).toISOString();
           newMetadata.set(thread.session_id, {
             title,
             status: thread.is_archived ? "archived" : "regular",
@@ -1726,247 +1711,127 @@ function AomiRuntimeProvider({
       }
     };
     void fetchThreadList();
-  }, [publicKey]);
-  const threadListAdapter = useMemo2(() => {
+  }, [user.address, backendApiRef]);
+  const threadListAdapter = useMemo2(
+    () => buildThreadListAdapter({
+      backendStateRef,
+      backendApiRef,
+      threadContext,
+      currentThreadIdRef,
+      polling,
+      userAddress: user.address,
+      setIsRunning
+    }),
+    [
+      backendApiRef,
+      polling,
+      user.address,
+      backendStateRef,
+      setIsRunning,
+      threadContext,
+      threadContext.currentThreadId,
+      threadContext.allThreadsMetadata
+    ]
+  );
+  useEffect2(() => {
     const backendState = backendStateRef.current;
-    const { regularThreads, archivedThreads } = buildThreadLists(threadContext.threadMetadata);
-    const preparePendingThread = (threadId) => {
-      const previousPendingId = backendState.creatingThreadId;
-      if (previousPendingId && previousPendingId !== threadId) {
-        threadContext.setThreadMetadata((prev) => {
-          const next = new Map(prev);
-          next.delete(previousPendingId);
-          return next;
-        });
-        threadContext.setThreads((prev) => {
-          const next = new Map(prev);
-          next.delete(previousPendingId);
-          return next;
-        });
-        backendState.pendingChat.delete(previousPendingId);
-        backendState.pendingSystem.delete(previousPendingId);
-        backendState.tempToBackendId.delete(previousPendingId);
-        backendState.skipInitialFetch.delete(previousPendingId);
-      }
-      backendState.creatingThreadId = threadId;
-      backendState.pendingChat.delete(threadId);
-      backendState.pendingSystem.delete(threadId);
-      threadContext.setThreadMetadata(
-        (prev) => new Map(prev).set(threadId, {
-          title: "New Chat",
-          status: "pending",
-          lastActiveAt: (/* @__PURE__ */ new Date()).toISOString()
-        })
-      );
-      threadContext.setThreadMessages(threadId, []);
-      threadContext.setCurrentThreadId(threadId);
-      setIsRunning(false);
-      threadContext.bumpThreadViewKey();
-    };
-    const findPendingThreadId = () => {
-      if (backendState.creatingThreadId) return backendState.creatingThreadId;
-      for (const [id, meta] of threadContext.threadMetadata.entries()) {
-        if (meta.status === "pending") return id;
-      }
-      return null;
-    };
-    return {
-      threadId: threadContext.currentThreadId,
-      threads: regularThreads,
-      archivedThreads,
-      onSwitchToNewThread: async () => {
-        var _a;
-        const previousThreadId = currentThreadIdRef.current;
-        polling.stopAll();
-        if (isRunning && isThreadReady(backendState, previousThreadId)) {
-          const backendId = resolveThreadId(backendState, previousThreadId);
-          void backendApiRef.current.postInterrupt(backendId);
-        }
-        const pendingId = findPendingThreadId();
-        if (pendingId) {
-          preparePendingThread(pendingId);
-          return;
-        }
-        if (backendState.createThreadPromise) {
-          preparePendingThread((_a = backendState.creatingThreadId) != null ? _a : `temp-${crypto.randomUUID()}`);
-          return;
-        }
-        const tempId = `temp-${crypto.randomUUID()}`;
-        preparePendingThread(tempId);
-        const createPromise = backendApiRef.current.createThread(publicKey, void 0).then(async (newThread) => {
-          var _a2;
-          const uiThreadId = (_a2 = backendState.creatingThreadId) != null ? _a2 : tempId;
-          const backendId = newThread.session_id;
-          setBackendMapping(backendState, uiThreadId, backendId);
-          markSkipInitialFetch(backendState, uiThreadId);
-          bumpUpdateSubscriptions();
-          const backendTitle = newThread.title;
-          if (backendTitle && !isPlaceholderTitle(backendTitle)) {
-            threadContext.setThreadMetadata((prev) => {
-              var _a3;
-              const next = new Map(prev);
-              const existing = next.get(uiThreadId);
-              const nextStatus = (existing == null ? void 0 : existing.status) === "archived" ? "archived" : "regular";
-              next.set(uiThreadId, {
-                title: backendTitle,
-                status: nextStatus,
-                lastActiveAt: (_a3 = existing == null ? void 0 : existing.lastActiveAt) != null ? _a3 : (/* @__PURE__ */ new Date()).toISOString()
-              });
-              return next;
+    const currentSessionId = threadContext.currentThreadId;
+    if (process.env.NODE_ENV !== "production") {
+      console.debug("[aomi][sse] subscribe", {
+        currentSessionId,
+        resolvedSessionId,
+        hasMapping: currentSessionId !== resolvedSessionId
+      });
+    }
+    const unsubscribe = backendApiRef.current.subscribeSSE(
+      resolvedSessionId,
+      (event) => {
+        const eventType = event.type;
+        const sessionId = event.session_id;
+        if (eventType === "title_changed") {
+          const newTitle = event.new_title;
+          const targetThreadId = resolveThreadId(backendState, sessionId);
+          const normalizedTitle = isPlaceholderTitle(newTitle) ? "" : newTitle;
+          if (process.env.NODE_ENV !== "production") {
+            console.debug("[aomi][sse] title_changed", {
+              sessionId,
+              newTitle,
+              normalizedTitle,
+              currentThreadId: threadContextRef.current.currentThreadId,
+              targetThreadId,
+              hasMapping: sessionId !== targetThreadId,
+              creatingThreadId: backendState.creatingThreadId
             });
           }
-          if (backendState.creatingThreadId === uiThreadId) {
-            backendState.creatingThreadId = null;
-          }
-          const pendingMessages = backendState.pendingChat.get(uiThreadId);
-          if (pendingMessages == null ? void 0 : pendingMessages.length) {
-            backendState.pendingChat.delete(uiThreadId);
-            for (const text of pendingMessages) {
-              try {
-                const activePublicKey = publicKeyRef.current;
-                if (activePublicKey) {
-                  await backendApiRef.current.postChatMessage(
-                    backendId,
-                    text,
-                    activePublicKey
-                  );
-                } else {
-                  await backendApiRef.current.postChatMessage(backendId, text);
-                }
-              } catch (error) {
-                console.error("Failed to send queued message:", error);
-              }
-            }
-            if (currentThreadIdRef.current === uiThreadId) {
-              polling == null ? void 0 : polling.start(uiThreadId);
-            }
-          }
-        }).catch((error) => {
-          var _a2;
-          console.error("Failed to create new thread:", error);
-          const failedId = (_a2 = backendState.creatingThreadId) != null ? _a2 : tempId;
-          threadContext.setThreadMetadata((prev) => {
-            const next = new Map(prev);
-            next.delete(failedId);
-            return next;
-          });
-          threadContext.setThreads((prev) => {
-            const next = new Map(prev);
-            next.delete(failedId);
-            return next;
-          });
-          if (backendState.creatingThreadId === failedId) {
-            backendState.creatingThreadId = null;
-          }
-        }).finally(() => {
-          backendState.createThreadPromise = null;
-        });
-        backendState.createThreadPromise = createPromise;
-      },
-      onSwitchToThread: (threadId) => {
-        const previousThreadId = currentThreadIdRef.current;
-        polling.stopAll();
-        if (isRunning && isThreadReady(backendState, previousThreadId)) {
-          const backendId = resolveThreadId(backendState, previousThreadId);
-          void backendApiRef.current.postInterrupt(backendId);
-        }
-        threadContext.setCurrentThreadId(threadId);
-      },
-      onRename: async (threadId, newTitle) => {
-        var _a, _b;
-        const previousTitle = (_b = (_a = threadContext.getThreadMetadata(threadId)) == null ? void 0 : _a.title) != null ? _b : "";
-        const normalizedTitle = isPlaceholderTitle(newTitle) ? "" : newTitle;
-        threadContext.updateThreadMetadata(threadId, {
-          title: normalizedTitle
-        });
-        try {
-          await backendApiRef.current.renameThread(threadId, newTitle);
-        } catch (error) {
-          console.error("Failed to rename thread:", error);
-          threadContext.updateThreadMetadata(threadId, { title: previousTitle });
-        }
-      },
-      onArchive: async (threadId) => {
-        threadContext.updateThreadMetadata(threadId, { status: "archived" });
-        try {
-          await backendApiRef.current.archiveThread(threadId);
-        } catch (error) {
-          console.error("Failed to archive thread:", error);
-          threadContext.updateThreadMetadata(threadId, { status: "regular" });
-        }
-      },
-      onUnarchive: async (threadId) => {
-        threadContext.updateThreadMetadata(threadId, { status: "regular" });
-        try {
-          await backendApiRef.current.unarchiveThread(threadId);
-        } catch (error) {
-          console.error("Failed to unarchive thread:", error);
-          threadContext.updateThreadMetadata(threadId, { status: "archived" });
-        }
-      },
-      onDelete: async (threadId) => {
-        try {
-          await backendApiRef.current.deleteThread(threadId);
-          threadContext.setThreadMetadata((prev) => {
-            const next = new Map(prev);
-            next.delete(threadId);
-            return next;
-          });
-          threadContext.setThreads((prev) => {
+          threadContextRef.current.setThreadMetadata((prev) => {
+            var _a;
             const next = new Map(prev);
-            next.delete(threadId);
+            const existing = next.get(targetThreadId);
+            const nextStatus = (existing == null ? void 0 : existing.status) === "archived" ? "archived" : "regular";
+            next.set(targetThreadId, {
+              title: normalizedTitle,
+              status: nextStatus,
+              lastActiveAt: (_a = existing == null ? void 0 : existing.lastActiveAt) != null ? _a : (/* @__PURE__ */ new Date()).toISOString()
+            });
             return next;
           });
-          backendState.pendingChat.delete(threadId);
-          backendState.pendingSystem.delete(threadId);
-          backendState.tempToBackendId.delete(threadId);
-          backendState.skipInitialFetch.delete(threadId);
-          backendState.runningThreads.delete(threadId);
-          if (backendState.creatingThreadId === threadId) {
+          if (!isPlaceholderTitle(newTitle) && backendState.creatingThreadId === targetThreadId) {
             backendState.creatingThreadId = null;
           }
-          if (threadContext.currentThreadId === threadId) {
-            const firstRegularThread = Array.from(threadContext.threadMetadata.entries()).find(
-              ([id, meta]) => meta.status === "regular" && id !== threadId
-            );
-            if (firstRegularThread) {
-              threadContext.setCurrentThreadId(firstRegularThread[0]);
-            } else {
-              const defaultId = "default-session";
-              threadContext.setThreadMetadata(
-                (prev) => new Map(prev).set(defaultId, {
-                  title: "New Chat",
-                  status: "regular",
-                  lastActiveAt: (/* @__PURE__ */ new Date()).toISOString()
-                })
-              );
-              threadContext.setThreadMessages(defaultId, []);
-              threadContext.setCurrentThreadId(defaultId);
-            }
-          }
-        } catch (error) {
-          console.error("Failed to delete thread:", error);
-          throw error;
         }
       }
+    );
+    return () => {
+      unsubscribe == null ? void 0 : unsubscribe();
     };
   }, [
     backendApiRef,
-    polling,
-    publicKey,
     backendStateRef,
-    setIsRunning,
-    threadContext,
     threadContext.currentThreadId,
-    threadContext.threadMetadata,
-    bumpUpdateSubscriptions
+    resolvedSessionId
   ]);
-  useEffect3(() => {
+  useEffect2(() => {
     const threadId = threadContext.currentThreadId;
-    if (!isTempThreadId(threadId)) return;
     if (!isThreadReady(backendStateRef.current, threadId)) return;
     void messageController.flushPendingChat(threadId);
   }, [messageController, backendStateRef, threadContext.currentThreadId]);
+  useEffect2(() => {
+    const showToolNotification = (eventType) => (event) => {
+      const payload = event.payload;
+      const toolName = typeof (payload == null ? void 0 : payload.tool_name) === "string" ? payload.tool_name : void 0;
+      const title = toolName ? `${eventType === "tool_update" ? "Tool update" : "Tool complete"}: ${toolName}` : eventType === "tool_update" ? "Tool update" : "Tool complete";
+      const message = typeof (payload == null ? void 0 : payload.message) === "string" ? payload.message : typeof (payload == null ? void 0 : payload.result) === "string" ? payload.result : void 0;
+      notificationContext.showNotification({
+        type: "notice",
+        title,
+        message
+      });
+    };
+    const unsubscribeUpdate = eventContext.subscribe(
+      "tool_update",
+      showToolNotification("tool_update")
+    );
+    const unsubscribeComplete = eventContext.subscribe(
+      "tool_complete",
+      showToolNotification("tool_complete")
+    );
+    return () => {
+      unsubscribeUpdate();
+      unsubscribeComplete();
+    };
+  }, [eventContext, notificationContext]);
+  useEffect2(() => {
+    const unsubscribe = eventContext.subscribe("system_notice", (event) => {
+      const payload = event.payload;
+      const message = payload == null ? void 0 : payload.message;
+      notificationContext.showNotification({
+        type: "notice",
+        title: "System notice",
+        message
+      });
+    });
+    return unsubscribe;
+  }, [eventContext, notificationContext]);
   const runtime = useExternalStoreRuntime({
     messages: currentMessages,
     setMessages: (msgs) => threadContext.setThreadMessages(threadContext.currentThreadId, [...msgs]),
@@ -1976,82 +1841,283 @@ function AomiRuntimeProvider({
     convertMessage: (msg) => msg,
     adapters: { threadList: threadListAdapter }
   });
-  useEffect3(() => {
-    const threadId = threadContext.currentThreadId;
-    if (isTempThreadId(threadId)) return;
-    const hasUserMessages = currentMessages.some((msg) => msg.role === "user");
-    if (hasUserMessages) {
-      void messageController.flushPendingSystem(threadId);
-    }
-  }, [currentMessages, messageController, threadContext.currentThreadId]);
-  useEffect3(() => {
+  useEffect2(() => {
     return () => {
-      var _a;
       polling.stopAll();
-      (_a = eventControllerRef.current) == null ? void 0 : _a.cleanup();
     };
   }, [polling]);
-  useEffect3(() => {
-    var _a, _b, _c;
-    const backendState = backendStateRef.current;
-    const threadId = threadContext.currentThreadId;
-    const isReady = isThreadReady(backendState, threadId);
-    if (!isReady) {
-      lastSubscribedThreadRef.current = null;
-      (_a = eventControllerRef.current) == null ? void 0 : _a.setSubscribableSessionId(null);
-      return;
-    }
-    const isCurrentlyRunning = isThreadRunning(backendState, threadId);
-    if (isCurrentlyRunning) {
-      const sessionId = resolveThreadId(backendState, threadId);
-      (_b = eventControllerRef.current) == null ? void 0 : _b.setSubscribableSessionId(sessionId);
-      lastSubscribedThreadRef.current = threadId;
-      return;
-    }
-    if (lastSubscribedThreadRef.current !== threadId) {
-      (_c = eventControllerRef.current) == null ? void 0 : _c.setSubscribableSessionId(null);
-    }
-  }, [backendStateRef, threadContext.currentThreadId, updateSubscriptionsTick, isRunning]);
-  return /* @__PURE__ */ jsx3(
-    RuntimeActionsProvider,
+  const userContext = useUser();
+  const sendMessage = useCallback5(
+    async (text) => {
+      const appendMessage = {
+        role: "user",
+        content: [{ type: "text", text }]
+      };
+      await messageController.outbound(
+        appendMessage,
+        threadContext.currentThreadId
+      );
+    },
+    [messageController, threadContext.currentThreadId]
+  );
+  const cancelGeneration = useCallback5(() => {
+    messageController.cancel(threadContext.currentThreadId);
+  }, [messageController, threadContext.currentThreadId]);
+  const getMessages = useCallback5(
+    (threadId) => {
+      const id = threadId != null ? threadId : threadContext.currentThreadId;
+      return threadContext.getThreadMessages(id);
+    },
+    [threadContext]
+  );
+  const createThread = useCallback5(async () => {
+    await threadListAdapter.onSwitchToNewThread();
+    return threadContextRef.current.currentThreadId;
+  }, [threadListAdapter]);
+  const deleteThread = useCallback5(
+    async (threadId) => {
+      await threadListAdapter.onDelete(threadId);
+    },
+    [threadListAdapter]
+  );
+  const renameThread = useCallback5(
+    async (threadId, title) => {
+      await threadListAdapter.onRename(threadId, title);
+    },
+    [threadListAdapter]
+  );
+  const archiveThread = useCallback5(
+    async (threadId) => {
+      await threadListAdapter.onArchive(threadId);
+    },
+    [threadListAdapter]
+  );
+  const selectThread = useCallback5(
+    (threadId) => {
+      if (threadContext.allThreadsMetadata.has(threadId)) {
+        threadListAdapter.onSwitchToThread(threadId);
+      } else {
+        void threadListAdapter.onSwitchToNewThread();
+      }
+    },
+    [threadContext.allThreadsMetadata, threadListAdapter]
+  );
+  const aomiRuntimeApi = useMemo2(
+    () => ({
+      // User API
+      user: userContext.user,
+      getUserState: userContext.getUserState,
+      setUser: userContext.setUser,
+      onUserStateChange: userContext.onUserStateChange,
+      // Thread API
+      currentThreadId: threadContext.currentThreadId,
+      threadViewKey: threadContext.threadViewKey,
+      threadMetadata: threadContext.allThreadsMetadata,
+      getThreadMetadata: threadContext.getThreadMetadata,
+      createThread,
+      deleteThread,
+      renameThread,
+      archiveThread,
+      selectThread,
+      // Chat API
+      isRunning,
+      getMessages,
+      sendMessage,
+      cancelGeneration,
+      // Notification API
+      notifications: notificationContext.notifications,
+      showNotification: notificationContext.showNotification,
+      dismissNotification: notificationContext.dismissNotification,
+      clearAllNotifications: notificationContext.clearAll,
+      // Event API
+      subscribe: eventContext.subscribe,
+      sendSystemCommand: eventContext.sendOutboundSystem,
+      sseStatus: eventContext.sseStatus
+    }),
+    [
+      userContext,
+      threadContext.currentThreadId,
+      threadContext.threadViewKey,
+      threadContext.allThreadsMetadata,
+      threadContext.getThreadMetadata,
+      createThread,
+      deleteThread,
+      renameThread,
+      archiveThread,
+      selectThread,
+      isRunning,
+      getMessages,
+      sendMessage,
+      cancelGeneration,
+      notificationContext,
+      eventContext
+    ]
+  );
+  return /* @__PURE__ */ jsx5(AomiRuntimeApiProvider, { value: aomiRuntimeApi, children: /* @__PURE__ */ jsx5(AssistantRuntimeProvider, { runtime, children }) });
+}
+
+// src/runtime/aomi-runtime.tsx
+import { jsx as jsx6 } from "react/jsx-runtime";
+function AomiRuntimeProvider({
+  children,
+  backendUrl = "http://localhost:8080"
+}) {
+  const backendApi = useMemo3(() => new BackendApi(backendUrl), [backendUrl]);
+  return /* @__PURE__ */ jsx6(ThreadContextProvider, { children: /* @__PURE__ */ jsx6(NotificationContextProvider, { children: /* @__PURE__ */ jsx6(UserContextProvider, { children: /* @__PURE__ */ jsx6(AomiRuntimeInner, { backendApi, children }) }) }) });
+}
+function AomiRuntimeInner({
+  children,
+  backendApi
+}) {
+  const threadContext = useThreadContext();
+  return /* @__PURE__ */ jsx6(
+    EventContextProvider,
     {
-      value: {
-        sendSystemMessage: (message) => messageController.outboundSystem(threadContext.currentThreadId, message)
-      },
-      children: /* @__PURE__ */ jsx3(AssistantRuntimeProvider, { runtime, children })
+      backendApi,
+      sessionId: threadContext.currentThreadId,
+      children: /* @__PURE__ */ jsx6(AomiRuntimeCore, { backendApi, children })
     }
   );
 }
-function AomiRuntimeProviderWithNotifications(props) {
-  return /* @__PURE__ */ jsx3(NotificationProvider, { children: /* @__PURE__ */ jsx3(AomiRuntimeProvider, __spreadValues({}, props)) });
+
+// src/handlers/wallet-handler.ts
+import { useCallback as useCallback6, useEffect as useEffect3, useState as useState5 } from "react";
+function useWalletHandler({
+  sessionId,
+  onTxRequest
+}) {
+  const { subscribe: subscribe2, sendOutboundSystem: sendOutbound } = useEventContext();
+  const { setUser, getUserState } = useUser();
+  const [pendingTxRequests, setPendingTxRequests] = useState5(
+    []
+  );
+  useEffect3(() => {
+    const unsubscribe = subscribe2(
+      "wallet_tx_request",
+      (event) => {
+        const request = event.payload;
+        setPendingTxRequests((prev) => [...prev, request]);
+        onTxRequest == null ? void 0 : onTxRequest(request);
+      }
+    );
+    return unsubscribe;
+  }, [subscribe2, onTxRequest]);
+  useEffect3(() => {
+    const unsubscribe = subscribe2(
+      "user_state_request",
+      (event) => {
+        sendOutbound({
+          type: "user_state_response",
+          sessionId,
+          payload: getUserState()
+        });
+      }
+    );
+    return unsubscribe;
+  }, [subscribe2, onTxRequest]);
+  const sendTxComplete = useCallback6(
+    (tx) => {
+      sendOutbound({
+        type: "wallet:tx_complete",
+        sessionId,
+        payload: tx
+      });
+    },
+    [sendOutbound, sessionId]
+  );
+  const sendConnectionChange = useCallback6(
+    (status, address, chainId) => {
+      if (status === "connected") {
+        setUser({
+          isConnected: true,
+          address,
+          chainId
+        });
+      } else {
+        setUser({
+          isConnected: false,
+          address: void 0,
+          chainId: void 0
+        });
+      }
+      sendOutbound({
+        type: status === "connected" ? "wallet:connected" : "wallet:disconnected",
+        sessionId,
+        payload: { status, address }
+      });
+    },
+    [setUser, sendOutbound, sessionId]
+  );
+  const clearTxRequest = useCallback6((index) => {
+    setPendingTxRequests((prev) => prev.filter((_, i) => i !== index));
+  }, []);
+  return {
+    sendTxComplete,
+    sendConnectionChange,
+    pendingTxRequests,
+    clearTxRequest
+  };
 }
 
-// src/lib/utils.ts
-import { clsx } from "clsx";
-import { twMerge } from "tailwind-merge";
-function cn(...inputs) {
-  return twMerge(clsx(inputs));
+// src/handlers/notification-handler.ts
+import { useCallback as useCallback7, useEffect as useEffect4, useState as useState6 } from "react";
+var notificationIdCounter2 = 0;
+function generateNotificationId() {
+  return `notif-${Date.now()}-${++notificationIdCounter2}`;
+}
+function useNotificationHandler({
+  onNotification
+} = {}) {
+  const { subscribe: subscribe2 } = useEventContext();
+  const [notifications, setNotifications] = useState6([]);
+  useEffect4(() => {
+    const unsubscribe = subscribe2("notification", (event) => {
+      var _a, _b;
+      const payload = event.payload;
+      const notification = {
+        id: generateNotificationId(),
+        type: (_a = payload.type) != null ? _a : "notification",
+        title: (_b = payload.title) != null ? _b : "Notification",
+        body: payload.body,
+        handled: false,
+        timestamp: event.timestamp,
+        sessionId: event.sessionId
+      };
+      setNotifications((prev) => [notification, ...prev]);
+      onNotification == null ? void 0 : onNotification(notification);
+    });
+    return unsubscribe;
+  }, [subscribe2, onNotification]);
+  const unhandledCount = notifications.filter((n) => !n.handled).length;
+  const markHandled = useCallback7((id) => {
+    setNotifications(
+      (prev) => prev.map((n) => n.id === id ? __spreadProps(__spreadValues({}, n), { handled: true }) : n)
+    );
+  }, []);
+  return {
+    notifications,
+    unhandledCount,
+    markDone: markHandled
+  };
 }
 export {
   AomiRuntimeProvider,
-  AomiRuntimeProviderWithNotifications,
   BackendApi,
-  NotificationProvider,
-  RuntimeActionsProvider,
+  EventContextProvider,
+  NotificationContextProvider,
   ThreadContextProvider,
-  WalletSystemMessageEmitter,
+  UserContextProvider,
   cn,
-  toInboundSystem as constructSystemMessage,
-  toInboundMessage as constructThreadMessage,
   formatAddress,
   getNetworkName,
-  normalizeWalletError,
-  pickInjectedProvider,
-  toHexQuantity,
+  useAomiRuntime,
   useCurrentThreadMessages,
   useCurrentThreadMetadata,
+  useEventContext,
   useNotification,
-  useRuntimeActions,
-  useThreadContext
+  useNotificationHandler,
+  useThreadContext,
+  useUser,
+  useWalletHandler
 };
 //# sourceMappingURL=index.js.map