@truefoundry/trueforge-ui 0.3.0-rc.0 → 0.3.0-rc.10

package/dist/{chunk-WREC7D54.js → chunk-PCI6VMB5.js}

@@ -1,42 +1,96 @@
+import {
+  WidgetVisibilityProvider
+} from "./chunk-DJUFS2BW.js";
 import {
   CustomActionRenderersProvider,
   notifyComposerBusyFailure
 } from "./chunk-BSIPI3KV.js";
 import {
-  useAuiState
-} from "./chunk-M6MHLVSO.js";
+  ResolvedRoutesProvider
+} from "./chunk-Y6S4OVVQ.js";
 import {
+  IS_CREATE_AGENT_METADATA_KEY,
+  isCreateAgentMetadataValue,
+  sessionIsCreateAgent
+} from "./chunk-RDXBMTP5.js";
+import {
+  ActiveSessionPermissionsProvider,
+  AgentConfigInstructionsProvider,
+  CurrentUserProvider,
   SlotsProvider,
   ToasterProvider,
   modelPatchWithReasoningEffort,
+  mountName,
+  useThemeMode,
   useToaster
-} from "./chunk-OSV3HI5M.js";
+} from "./chunk-GG23N2GW.js";
 import {
   DraftCatalogProvider,
-  Spinner,
-  getErrorMessage,
   useDraftCatalog
-} from "./chunk-FJ5RSSMA.js";
+} from "./chunk-3ZW74FHC.js";
 import {
   DEFAULT_AGENT_CONFIG,
   ServerProvider,
   ShellModeProvider,
+  getErrorMessage,
+  isSessionsChromeEnabled,
+  readSessionShareSearch,
+  useOptionalAgentSessionsServer,
+  useOptionalShellMode,
   useServerCapabilities,
   useShellMode,
   withCapabilitiesSandbox
-} from "./chunk-WQULF5AT.js";
+} from "./chunk-2GIZY3JJ.js";
+import {
+  useAui,
+  useAuiState
+} from "./chunk-M6MHLVSO.js";
 import {
   cn
-} from "./chunk-36KOSC2J.js";
+} from "./chunk-JCL3Q7RW.js";
 
 // src/containers/TrueFoundryChatProvider.tsx
-import { AssistantRuntimeProvider } from "@assistant-ui/react";
+import { AssistantRuntimeProvider, useAuiState as useAuiState2 } from "@assistant-ui/react";
 import {
   trueFoundryAttachmentAdapter,
   useTrueFoundryAgentRuntime
 } from "@truefoundry/assistant-ui-runtime";
 import { useCallback, useMemo } from "react";
 import { jsx } from "react/jsx-runtime";
+function ActiveSessionPermissionScope({
+  locallyCreatedSessionIds,
+  children
+}) {
+  const remoteId = useAuiState2((state) => state.threadListItem.remoteId);
+  return /* @__PURE__ */ jsx(
+    ActiveSessionPermissionsProvider,
+    {
+      sessionId: remoteId,
+      assumeManage: remoteId != null && locallyCreatedSessionIds.has(remoteId),
+      children
+    }
+  );
+}
+function runtimeServer({
+  server,
+  locallyCreatedSessionIds
+}) {
+  return {
+    ...server,
+    async createSession(request) {
+      const session = await server.createSession(request);
+      locallyCreatedSessionIds.add(session.id);
+      return session;
+    },
+    async listSessions(request) {
+      const result = await server.listSessions(request);
+      return {
+        ...result,
+        data: result.data.filter((session) => !sessionIsCreateAgent(session))
+      };
+    }
+  };
+}
 function ChatRuntimeScope({
   server,
   agent,
@@ -49,6 +103,12 @@ function ChatRuntimeScope({
 }) {
   const { showError } = useToaster();
   const reportError = onError ?? showError;
+  const localSessionState = useMemo(() => ({ server, ids: /* @__PURE__ */ new Set() }), [server]);
+  const locallyCreatedSessionIds = localSessionState.ids;
+  const historyServer = useMemo(
+    () => runtimeServer({ server, locallyCreatedSessionIds }),
+    [locallyCreatedSessionIds, server]
+  );
   const resolvedOnError = useCallback(
     (error) => {
       notifyComposerBusyFailure();
@@ -57,10 +117,11 @@ function ChatRuntimeScope({
     [reportError]
   );
   const runtime = useTrueFoundryAgentRuntime({
-    server,
+    server: historyServer,
     agent,
     agentName,
     listSessionsAgentId,
+    listSessionsCreatedByMe: true,
     initialSessionId,
     onError: resolvedOnError,
     adapters: {
@@ -68,7 +129,7 @@ function ChatRuntimeScope({
       attachments: adapters?.attachments ?? trueFoundryAttachmentAdapter
     }
   });
-  return /* @__PURE__ */ jsx(AssistantRuntimeProvider, { runtime, children });
+  return /* @__PURE__ */ jsx(AssistantRuntimeProvider, { runtime, children: /* @__PURE__ */ jsx(ActiveSessionPermissionScope, { locallyCreatedSessionIds, children }) });
 }
 function TrueFoundryChatProvider(props) {
   const { server, initialSessionId, adapters, onError, children, agent, agentName, listSessionsAgentId } = props;
@@ -90,13 +151,17 @@ function TrueFoundryChatProvider(props) {
 
 // src/containers/TrueForgeUIShell.tsx
 import { lazy, Suspense, useCallback as useCallback2, useMemo as useMemo2, useState as useState2 } from "react";
+import { ThinkingOrb } from "thinking-orbs";
 
 // src/atoms/draft/DraftSpecPreferenceBridge.tsx
 import { useTrueFoundryAgentSpec, useTrueFoundryUpdateAgentSpec } from "@truefoundry/assistant-ui-runtime";
 import { useEffect } from "react";
-function mountName(mount) {
-  const name = Reflect.get(mount, "name");
-  return typeof name === "string" ? name : null;
+function readDraftSandboxEnabled(spec) {
+  if (spec.config === void 0) return void 0;
+  const sandbox = Reflect.get(spec.config, "sandbox");
+  if (typeof sandbox !== "object" || sandbox === null) return void 0;
+  const enabled = Reflect.get(sandbox, "enabled");
+  return typeof enabled === "boolean" ? enabled : void 0;
 }
 function filterMounts(mounts, availableNames) {
   if (mounts === void 0) return void 0;
@@ -143,8 +208,30 @@ function reconcileDraftSpecPreferences({
 }
 function reconcileDraftSandbox({
   agentSpec,
-  sandboxEnabled
+  sandboxEnabled,
+  kind = "agent"
 }) {
+  if (kind === "chat") {
+    if (sandboxEnabled == null) return {};
+    const current = readDraftSandboxEnabled(agentSpec);
+    if (sandboxEnabled === true && current !== true) {
+      return {
+        config: {
+          ...agentSpec.config,
+          sandbox: { ...agentSpec.config?.sandbox, enabled: true }
+        }
+      };
+    }
+    if (sandboxEnabled === false && current === true) {
+      return {
+        config: {
+          ...agentSpec.config,
+          sandbox: { ...agentSpec.config?.sandbox, enabled: false }
+        }
+      };
+    }
+    return {};
+  }
   const nextSpec = withCapabilitiesSandbox(agentSpec, sandboxEnabled);
   return nextSpec === agentSpec ? {} : { config: nextSpec.config };
 }
@@ -156,21 +243,22 @@ function DraftSpecPreferenceBridge() {
   const sandboxEnabled = capabilities?.sandbox.enabled;
   const { models, skills, connectors, loaded, error, ensureLoaded } = useDraftCatalog();
   const isPlainDraft = mode.status === "active" && mode.isMutable && mode.agentId == null && pendingSessionId == null;
+  const preferenceKind = mode.status === "active" && mode.isMutable && mode.isCreateAgent ? "agent" : "chat";
   useEffect(() => {
     if (isPlainDraft) ensureLoaded();
   }, [ensureLoaded, isPlainDraft]);
   useEffect(() => {
     if (isPlainDraft && agentSpec != null) {
-      rememberDraftSpec(agentSpec);
+      rememberDraftSpec(agentSpec, preferenceKind);
     }
-  }, [agentSpec, isPlainDraft, rememberDraftSpec]);
+  }, [agentSpec, isPlainDraft, preferenceKind, rememberDraftSpec]);
   useEffect(() => {
     if (!isPlainDraft || agentSpec == null || updateAgentSpec == null) return;
-    const update = reconcileDraftSandbox({ agentSpec, sandboxEnabled });
+    const update = reconcileDraftSandbox({ agentSpec, sandboxEnabled, kind: preferenceKind });
     if (Object.keys(update).length > 0) {
       updateAgentSpec(update);
     }
-  }, [agentSpec, isPlainDraft, sandboxEnabled, updateAgentSpec]);
+  }, [agentSpec, isPlainDraft, preferenceKind, sandboxEnabled, updateAgentSpec]);
   useEffect(() => {
     if (!isPlainDraft || agentSpec == null || updateAgentSpec == null || !loaded || error != null) return;
     const update = reconcileDraftSpecPreferences({
@@ -197,19 +285,100 @@ function DraftSpecPreferenceBridge() {
   return null;
 }
 
-// src/routing/RemoteIdRouteBridge.tsx
+// src/routing/HistorySessionSwitchBridge.tsx
 import { useEffect as useEffect2 } from "react";
+function HistorySessionSwitchBridge() {
+  const shell = useOptionalShellMode();
+  const aui = useAui();
+  const pendingSessionId = shell?.pendingSessionId;
+  const pendingSessionEpoch = shell?.pendingSessionEpoch ?? 0;
+  useEffect2(() => {
+    if (pendingSessionId == null) return;
+    void Promise.resolve(aui.threads().switchToThread(pendingSessionId)).catch(() => void 0);
+  }, [aui, pendingSessionId, pendingSessionEpoch]);
+  return null;
+}
+
+// src/routing/LibrarySessionShareBoot.tsx
+import { useEffect as useEffect3 } from "react";
+function LibrarySessionShareBoot() {
+  const { openLibraryAgent, setSessionsOpen } = useShellMode();
+  const sessions = useOptionalAgentSessionsServer();
+  const sessionsEnabled = isSessionsChromeEnabled({ sessions });
+  useEffect3(() => {
+    if (!sessionsEnabled) return;
+    const share = readSessionShareSearch(window.location.search);
+    if (share.view === "sessions") {
+      setSessionsOpen(true);
+      return;
+    }
+    if (share.agentId == null) return;
+    openLibraryAgent(share.agentId);
+  }, [openLibraryAgent, sessionsEnabled, setSessionsOpen]);
+  return null;
+}
+
+// src/routing/RemoteIdRouteBridge.tsx
+import { useEffect as useEffect4 } from "react";
 function RemoteIdRouteBridge({ onRemoteIdChange }) {
   const remoteId = useAuiState((s) => s.threadListItem.remoteId);
-  useEffect2(() => {
+  useEffect4(() => {
     onRemoteIdChange(remoteId ?? void 0);
   }, [remoteId, onRemoteIdChange]);
   return null;
 }
 
+// src/server/sessionListCache.ts
+var SESSION_LIST_TTL_MS = 3e4;
+function createSessionListCache() {
+  return /* @__PURE__ */ new Map();
+}
+function cacheKey(request) {
+  const rest = { ...request };
+  delete rest.startTimestamp;
+  return JSON.stringify(rest, Object.keys(rest).sort());
+}
+function withSessionListCache({
+  server,
+  cache
+}) {
+  const { deleteSession } = server;
+  return {
+    ...server,
+    listSessions(request) {
+      const key = cacheKey(request);
+      const now = Date.now();
+      const entry = cache.get(key);
+      if (entry !== void 0 && now - entry.at < SESSION_LIST_TTL_MS) return entry.page;
+      const page = server.listSessions(request);
+      cache.set(key, { at: now, page });
+      void page.catch(() => {
+        if (cache.get(key)?.page === page) cache.delete(key);
+      });
+      return page;
+    },
+    async createSession(request) {
+      const result = await server.createSession(request);
+      cache.clear();
+      return result;
+    },
+    async updateSession(request) {
+      const result = await server.updateSession(request);
+      cache.clear();
+      return result;
+    },
+    ...deleteSession != null ? {
+      deleteSession: async (request) => {
+        await deleteSession.call(server, request);
+        cache.clear();
+      }
+    } : {}
+  };
+}
+
 // src/containers/useResolvedServer.ts
 import { createTrueFoundryAgentUIServer } from "@truefoundry/assistant-ui-runtime/plugins/truefoundry-agent-server-adapter";
-import { useEffect as useEffect3, useState } from "react";
+import { useEffect as useEffect5, useState } from "react";
 var DEFAULT_CAPABILITIES = {
   data: {
     sandbox: { enabled: true },
@@ -223,14 +392,18 @@ function isBuiltInConfig(config) {
 function hasGetCapabilities(server) {
   return typeof Reflect.get(server, "getCapabilities") === "function";
 }
-function toAgentUIServer(server, catalog) {
+function toAgentUIServer(server, catalog, permissions) {
   const getCapabilities = hasGetCapabilities(server) ? () => server.getCapabilities() : async () => DEFAULT_CAPABILITIES;
   const withCapabilities = { ...server, getCapabilities };
-  const withCatalog = catalog != null ? { ...withCapabilities, catalog } : withCapabilities;
-  if (!isAgentUIServer(withCatalog)) {
+  const withOptionalPorts = {
+    ...withCapabilities,
+    ...catalog == null ? {} : { catalog },
+    ...permissions == null ? {} : { permissions }
+  };
+  if (!isAgentUIServer(withOptionalPorts)) {
     throw new Error("TrueForgeUI: runtime adapter returned an incomplete AgentUIServer");
   }
-  return withCatalog;
+  return withOptionalPorts;
 }
 function isAgentUIServer(value) {
   return typeof Reflect.get(value, "createSession") === "function" && typeof Reflect.get(value, "listSessions") === "function" && typeof Reflect.get(value, "getSession") === "function" && typeof Reflect.get(value, "updateSession") === "function" && typeof Reflect.get(value, "createTurn") === "function" && typeof Reflect.get(value, "cancelSession") === "function" && typeof Reflect.get(value, "listTurns") === "function" && typeof Reflect.get(value, "getTurn") === "function" && typeof Reflect.get(value, "listEvents") === "function" && typeof Reflect.get(value, "getCapabilities") === "function" && typeof Reflect.get(value, "getModels") === "function" && typeof Reflect.get(value, "getSkills") === "function" && typeof Reflect.get(value, "getMcp") === "function" && typeof Reflect.get(value, "searchAgents") === "function" && typeof Reflect.get(value, "saveAgent") === "function";
@@ -246,13 +419,14 @@ function useResolvedServer(config, onError) {
   const trueforgeToken = builtIn?.type === "trueforge" ? builtIn.token ?? "" : "";
   const trueforgeFetch = builtIn?.type === "trueforge" ? builtIn.fetch : void 0;
   const catalog = builtIn?.catalog;
+  const permissions = builtIn?.permissions;
   const [state, setState] = useState(() => {
     if (directServer) {
       return { status: "ready", server: directServer, error: null };
     }
     return { status: "loading", server: null, error: null };
   });
-  useEffect3(() => {
+  useEffect5(() => {
     if (directServer) {
       setState({ status: "ready", server: directServer, error: null });
       return;
@@ -266,22 +440,23 @@ function useResolvedServer(config, onError) {
           ...trueforgeBaseUrl ? { baseUrl: trueforgeBaseUrl } : {},
           ...trueforgeToken ? { token: trueforgeToken } : {},
           ...trueforgeFetch !== void 0 ? { fetch: trueforgeFetch } : {},
-          ...catalog != null ? { catalog } : {}
+          ...catalog != null ? { catalog } : {},
+          ...permissions != null ? { permissions } : {}
         });
       }
-      const runtimeServer = await createTrueFoundryAgentUIServer({
+      const runtimeServer2 = await createTrueFoundryAgentUIServer({
         apiKey,
         cpURL: controlPlaneURL,
         ...gatewayPlaneURL ? { gatewayURL: gatewayPlaneURL } : {}
       });
-      return toAgentUIServer(runtimeServer, catalog);
+      return toAgentUIServer(runtimeServer2, catalog, permissions);
     };
     void resolve().then((server) => {
       if (cancelled) return;
       try {
         setState({
           status: "ready",
-          server: type === "trueforge" ? toAgentUIServer(server, void 0) : server,
+          server: type === "trueforge" ? toAgentUIServer(server, void 0, void 0) : server,
           error: null
         });
       } catch (error) {
@@ -306,6 +481,7 @@ function useResolvedServer(config, onError) {
     trueforgeToken,
     trueforgeFetch,
     catalog,
+    permissions,
     onError
   ]);
   return state;
@@ -313,11 +489,11 @@ function useResolvedServer(config, onError) {
 
 // src/containers/TrueForgeUIShell.tsx
 import { jsx as jsx2, jsxs } from "react/jsx-runtime";
-var SidebarLayout = lazy(() => import("./SidebarLayout-3WZ3CVON.js").then((m) => ({ default: m.SidebarLayout })));
-var DrawerLayout = lazy(() => import("./DrawerLayout-Q6UK4VMI.js").then((m) => ({ default: m.DrawerLayout })));
-var DockLayout = lazy(() => import("./DockLayout-LBNQJZNF.js").then((m) => ({ default: m.DockLayout })));
-var WidgetLayout = lazy(() => import("./WidgetLayout-3QZ357FL.js").then((m) => ({ default: m.WidgetLayout })));
-var ShellRouteSync = lazy(() => import("./ShellRouteSync-T4WLY57Q.js").then((m) => ({ default: m.ShellRouteSync })));
+var SidebarLayout = lazy(() => import("./SidebarLayout-KYA76FQ7.js").then((m) => ({ default: m.SidebarLayout })));
+var DrawerLayout = lazy(() => import("./DrawerLayout-HUV7RAWL.js").then((m) => ({ default: m.DrawerLayout })));
+var DockLayout = lazy(() => import("./DockLayout-D2FPQCAL.js").then((m) => ({ default: m.DockLayout })));
+var WidgetLayout = lazy(() => import("./WidgetLayout-W2RGGOQ3.js").then((m) => ({ default: m.WidgetLayout })));
+var ShellRouteSync = lazy(() => import("./ShellRouteSync-SJXFKUAJ.js").then((m) => ({ default: m.ShellRouteSync })));
 function LayoutFallback({ className }) {
   return /* @__PURE__ */ jsx2(
     "div",
@@ -335,20 +511,30 @@ function LayoutFallback({ className }) {
   );
 }
 function ServerInitLoader({ className }) {
-  return /* @__PURE__ */ jsxs(
+  const themeMode = useThemeMode();
+  return /* @__PURE__ */ jsx2(
     "div",
     {
       role: "status",
+      "aria-label": "Loading",
       "aria-live": "polite",
       "aria-busy": "true",
       className: cn(
         "flex h-full min-h-48 w-full flex-1 items-center justify-center bg-primary-bg text-text-secondary",
         className
       ),
-      children: [
-        /* @__PURE__ */ jsx2(Spinner, { size: 28, className: "text-text-primary" }),
-        /* @__PURE__ */ jsx2("span", { className: "sr-only", children: "Loading" })
-      ]
+      children: /* @__PURE__ */ jsx2(
+        ThinkingOrb,
+        {
+          state: "connecting",
+          speed: 1,
+          theme: themeMode,
+          paused: false,
+          "aria-hidden": true,
+          size: 64,
+          style: { width: "4.5rem", height: "4.5rem" }
+        }
+      )
     }
   );
 }
@@ -392,9 +578,37 @@ function ChatProviderFromShell({
   children,
   initialSessionId: hostInitialSessionId,
   onRemoteIdChange,
+  server,
   ...providerRest
 }) {
-  const { mode, runtimeKey, listSessionsAgentId, pendingSessionId } = useShellMode();
+  const { mode, runtimeKey, historyAgentFilter, listSessionsAgentId, pendingSessionId } = useShellMode();
+  const isCreateAgent = mode.status === "active" && mode.isMutable && mode.isCreateAgent;
+  const [sessionListCache] = useState2(createSessionListCache);
+  const serverWithCreateIntent = useMemo2(() => {
+    return {
+      ...server,
+      createSession: (request) => {
+        if (request.agentSpec === void 0) {
+          return server.createSession(request);
+        }
+        const withMetadata = {
+          ...request,
+          metadata: {
+            [IS_CREATE_AGENT_METADATA_KEY]: isCreateAgentMetadataValue(isCreateAgent)
+          }
+        };
+        return server.createSession(withMetadata);
+      }
+    };
+  }, [server, isCreateAgent]);
+  const cachedRuntimeServer = useMemo2(
+    () => withSessionListCache({ server: serverWithCreateIntent, cache: sessionListCache }),
+    [serverWithCreateIntent, sessionListCache]
+  );
+  const runtimeServer2 = useMemo2(() => {
+    if (historyAgentFilter == null || historyAgentFilter.agentId != null) return cachedRuntimeServer;
+    return { ...cachedRuntimeServer, listSessions: async () => ({ data: [] }) };
+  }, [cachedRuntimeServer, historyAgentFilter]);
   const draftDefaultAgentSpec = useMemo2(() => {
     if (mode.status === "active" && mode.isMutable) {
       return mode.agentSpec ?? { model: { name: "openai-main/gpt-4.1" } };
@@ -423,21 +637,23 @@ function ChatProviderFromShell({
       defaultAgentSpec: draftDefaultAgentSpec
     };
   }, [mode, draftDefaultAgentSpec, pendingSessionId]);
-  return /* @__PURE__ */ jsx2(
+  return /* @__PURE__ */ jsx2(DraftCatalogProvider, { children: /* @__PURE__ */ jsx2(
     TrueFoundryChatProvider,
     {
       ...providerRest,
+      server: runtimeServer2,
       agent,
       listSessionsAgentId,
       initialSessionId: pendingSessionId ?? hostInitialSessionId,
-      children: /* @__PURE__ */ jsxs(DraftCatalogProvider, { children: [
+      children: /* @__PURE__ */ jsxs(AgentConfigInstructionsProvider, { children: [
         /* @__PURE__ */ jsx2(DraftSpecPreferenceBridge, {}),
+        /* @__PURE__ */ jsx2(HistorySessionSwitchBridge, {}),
         onRemoteIdChange != null ? /* @__PURE__ */ jsx2(RemoteIdRouteBridge, { onRemoteIdChange }) : null,
         children
       ] })
     },
     runtimeKey
-  );
+  ) });
 }
 function TrueForgeUIShell(props) {
   const {
@@ -450,6 +666,7 @@ function TrueForgeUIShell(props) {
     server: serverConfig,
     onError,
     customActionRenderers,
+    currentUser,
     resolvedRoutes,
     routes: _routes,
     ...providerRest
@@ -465,7 +682,8 @@ function TrueForgeUIShell(props) {
   }
   const server = resolved.server;
   const layoutTree = /* @__PURE__ */ jsx2(LayoutChildren, { layout, className });
-  return /* @__PURE__ */ jsx2(SlotsProvider, { overrides, theme, children: /* @__PURE__ */ jsx2(CustomActionRenderersProvider, { renderers: customActionRenderers, children: /* @__PURE__ */ jsx2(ServerProvider, { server, children: /* @__PURE__ */ jsxs(ShellModeProvider, { agentConfig, initialSettingsOpen, children: [
+  const shellTree = /* @__PURE__ */ jsxs(ShellModeProvider, { agentConfig, initialSettingsOpen, children: [
+    /* @__PURE__ */ jsx2(LibrarySessionShareBoot, {}),
     resolvedRoutes != null ? /* @__PURE__ */ jsx2(Suspense, { fallback: null, children: /* @__PURE__ */ jsx2(
       ShellRouteSync,
       {
@@ -484,7 +702,9 @@ function TrueForgeUIShell(props) {
         children: layoutTree
       }
     )
-  ] }) }) }) });
+  ] });
+  const visibilityTree = layout === "widget" ? /* @__PURE__ */ jsx2(WidgetVisibilityProvider, { children: shellTree }) : shellTree;
+  return /* @__PURE__ */ jsx2(SlotsProvider, { overrides, theme, children: /* @__PURE__ */ jsx2(CurrentUserProvider, { currentUser, children: /* @__PURE__ */ jsx2(CustomActionRenderersProvider, { renderers: customActionRenderers, children: /* @__PURE__ */ jsx2(ServerProvider, { server, children: resolvedRoutes != null ? /* @__PURE__ */ jsx2(ResolvedRoutesProvider, { routes: resolvedRoutes, children: visibilityTree }) : visibilityTree }) }) }) });
 }
 
 export {
@@ -492,4 +712,4 @@ export {
   ServerInitLoader,
   TrueForgeUIShell
 };
-//# sourceMappingURL=chunk-WREC7D54.js.map
+//# sourceMappingURL=chunk-PCI6VMB5.js.map