@truefoundry/trueforge-ui 0.1.0 → 0.2.1

package/dist/index.js

@@ -6,7 +6,6 @@ import {
   AssistantTextContainer,
   ComposerAttachmentPickerContainer,
   ComposerAttachmentsContainer,
-  ComposerBusyProvider,
   ComposerContainer,
   HistoryLoaderContainer,
   McpAuthContainer,
@@ -22,11 +21,18 @@ import {
   UserEditComposerContainer,
   UserMessageContainer,
   computeAgentStepsSplit,
-  notifyComposerBusyFailure,
   threadHasPendingMcpAuth,
-  useComposerBusyState,
   useComposerPauseView
-} from "./chunk-H7DR64W3.js";
+} from "./chunk-UACLSDNL.js";
+import {
+  ServerInitLoader,
+  TrueForgeUIShell,
+  TrueFoundryChatProvider
+} from "./chunk-EKVN6QSO.js";
+import {
+  ComposerBusyProvider,
+  useComposerBusyState
+} from "./chunk-FXCYLEFB.js";
 import {
   useAui,
   useAuiState
@@ -101,7 +107,6 @@ import {
   UserMessageEdit,
   WelcomeScreen,
   defaultSlots,
-  modelPatchWithReasoningEffort,
   preloadMarkdownOpenUI,
   useBrandName,
   useMCPAuth,
@@ -109,41 +114,41 @@ import {
   useThemeMode,
   useToaster,
   useToasterOptional
-} from "./chunk-QGNLP2EE.js";
+} from "./chunk-UU7C5DQ3.js";
 import {
-  DEFAULT_AGENT_CONFIG,
   DraftCatalogProvider,
   Icon,
+  getErrorMessage,
+  registerIcons,
+  useDraftCatalog
+} from "./chunk-IFLE7OMM.js";
+import {
+  DEFAULT_AGENT_CONFIG,
   ServerProvider,
   ShellModeProvider,
-  Spinner,
-  getErrorMessage,
   libraryAgentId,
-  registerIcons,
   shellIsMutable,
   useCatalogServer,
-  useDraftCatalog,
   useOptionalCatalogServer,
   useOptionalServer,
   useOptionalShellMode,
   useServer,
   useServerCapabilities,
   useShellMode
-} from "./chunk-XAZ5HJXV.js";
+} from "./chunk-WQULF5AT.js";
 import {
   OpenUiFenceBlock
-} from "./chunk-HT5HKQXE.js";
+} from "./chunk-JRRCAKEP.js";
 import {
   PRESETS,
   ThemeProvider,
-  cn,
   resolvePresetTokens,
   useBrand,
   useContentClassNames,
   useOptionalContentClassNames,
   useTheme,
   useThemeIcons
-} from "./chunk-DTF7ZMUJ.js";
+} from "./chunk-72LC2M3R.js";
 
 // src/icons/agent-2.svg
 import * as React from "react";
@@ -304,422 +309,30 @@ function ToolGroupContainer({ children, group }) {
   );
 }
 
-// src/containers/TrueforgeUI.tsx
-import { lazy, Suspense, useMemo as useMemo2 } from "react";
-
-// src/atoms/draft/DraftSpecPreferenceBridge.tsx
-import { useTrueFoundryAgentSpec, useTrueFoundryUpdateAgentSpec } from "@truefoundry/assistant-ui-runtime";
-import { useEffect as useEffect2 } from "react";
-function mountName(mount) {
-  const name = Reflect.get(mount, "name");
-  return typeof name === "string" ? name : null;
-}
-function filterMounts(mounts, availableNames) {
-  if (mounts === void 0) return void 0;
-  const filtered = mounts.filter((mount) => {
-    const name = mountName(mount);
-    return name !== null && availableNames.has(name);
-  });
-  return filtered.length === mounts.length ? mounts : filtered;
-}
-function reconcileDraftSpecPreferences({
-  agentSpec,
-  models,
-  skills,
-  connectors,
-  skillsEnabled
-}) {
-  const update = {};
-  const selectedModel = agentSpec.model.name.trim();
-  if (!models.some((model) => model.name === selectedModel)) {
-    const fallback = models[0];
-    if (fallback !== void 0) {
-      update.model = modelPatchWithReasoningEffort(
-        fallback.name,
-        agentSpec.model.params,
-        fallback.properties.reasoningEfforts
-      );
-    }
-  }
-  const nextMcpServers = filterMounts(agentSpec.mcpServers, new Set(connectors.map((connector) => connector.name)));
-  if (nextMcpServers !== agentSpec.mcpServers) {
-    update.mcpServers = nextMcpServers;
-  }
-  if (skillsEnabled === false) {
-    if (agentSpec.skills !== void 0 && agentSpec.skills.length > 0) {
-      update.skills = [];
-    }
-  } else {
-    const nextSkills = filterMounts(agentSpec.skills, new Set(skills.map((skill) => skill.name)));
-    if (nextSkills !== agentSpec.skills) {
-      update.skills = nextSkills;
-    }
-  }
-  return update;
-}
-function DraftSpecPreferenceBridge() {
-  const { mode, pendingSessionId, rememberDraftSpec } = useShellMode();
-  const { agentSpec } = useTrueFoundryAgentSpec();
-  const updateAgentSpec = useTrueFoundryUpdateAgentSpec();
-  const capabilities = useServerCapabilities();
-  const { models, skills, connectors, loaded, error, ensureLoaded } = useDraftCatalog();
-  const isPlainDraft = mode.status === "active" && mode.isMutable && mode.agentId == null && pendingSessionId == null;
-  useEffect2(() => {
-    if (isPlainDraft) ensureLoaded();
-  }, [ensureLoaded, isPlainDraft]);
-  useEffect2(() => {
-    if (isPlainDraft && agentSpec != null) {
-      rememberDraftSpec(agentSpec);
-    }
-  }, [agentSpec, isPlainDraft, rememberDraftSpec]);
-  useEffect2(() => {
-    if (!isPlainDraft || agentSpec == null || updateAgentSpec == null || !loaded || error != null) return;
-    const update = reconcileDraftSpecPreferences({
-      agentSpec,
-      models,
-      skills,
-      connectors,
-      skillsEnabled: capabilities?.skill.enabled
-    });
-    if (Object.keys(update).length > 0) {
-      updateAgentSpec(update);
-    }
-  }, [
-    agentSpec,
-    capabilities?.skill.enabled,
-    connectors,
-    error,
-    isPlainDraft,
-    loaded,
-    models,
-    skills,
-    updateAgentSpec
-  ]);
-  return null;
-}
-
-// src/containers/TrueFoundryChatProvider.tsx
-import { AssistantRuntimeProvider } from "@assistant-ui/react";
-import {
-  trueFoundryAttachmentAdapter,
-  useTrueFoundryAgentRuntime
-} from "@truefoundry/assistant-ui-runtime";
-import { useCallback, useMemo } from "react";
+// src/containers/TrueForgeUI.tsx
+import { lazy, Suspense } from "react";
 import { jsx as jsx11 } from "react/jsx-runtime";
-function ChatRuntimeScope({
-  server,
-  agent,
-  agentName,
-  listSessionsAgentId,
-  initialSessionId,
-  adapters,
-  onError,
-  children
-}) {
-  const { showError } = useToaster();
-  const reportError = onError ?? showError;
-  const resolvedOnError = useCallback(
-    (error) => {
-      notifyComposerBusyFailure();
-      reportError(error);
-    },
-    [reportError]
-  );
-  const runtime = useTrueFoundryAgentRuntime({
-    server,
-    agent,
-    agentName,
-    listSessionsAgentId,
-    initialSessionId,
-    onError: resolvedOnError,
-    adapters: {
-      ...adapters,
-      attachments: adapters?.attachments ?? trueFoundryAttachmentAdapter
-    }
-  });
-  return /* @__PURE__ */ jsx11(AssistantRuntimeProvider, { runtime, children });
-}
-function TrueFoundryChatProvider(props) {
-  const { server, initialSessionId, adapters, onError, children, agent, agentName, listSessionsAgentId } = props;
-  const stableServer = useMemo(() => server, [server]);
-  return /* @__PURE__ */ jsx11(ToasterProvider, { children: /* @__PURE__ */ jsx11(
-    ChatRuntimeScope,
-    {
-      server: stableServer,
-      agent,
-      agentName,
-      listSessionsAgentId,
-      initialSessionId,
-      adapters,
-      onError,
-      children
-    }
-  ) });
-}
-
-// src/containers/useResolvedServer.ts
-import { createTrueFoundryAgentUIServer } from "@truefoundry/assistant-ui-runtime/plugins/truefoundry-agent-server-adapter";
-import { useEffect as useEffect3, useState as useState3 } from "react";
-var DEFAULT_CAPABILITIES = {
-  data: {
-    sandbox: { enabled: true },
-    skill: { enabled: true },
-    settings: { enabled: true }
-  }
-};
-function isBuiltInConfig(config) {
-  return typeof config === "object" && config !== null && "type" in config && (config.type === "truefoundry" || config.type === "trueforge");
-}
-function hasGetCapabilities(server) {
-  return typeof Reflect.get(server, "getCapabilities") === "function";
-}
-function toAgentUIServer(server, catalog) {
-  const getCapabilities = hasGetCapabilities(server) ? () => server.getCapabilities() : async () => DEFAULT_CAPABILITIES;
-  const withCapabilities = { ...server, getCapabilities };
-  const withCatalog = catalog != null ? { ...withCapabilities, catalog } : withCapabilities;
-  if (!isAgentUIServer(withCatalog)) {
-    throw new Error("TrueforgeUI: runtime adapter returned an incomplete AgentUIServer");
-  }
-  return withCatalog;
-}
-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";
-}
-function useResolvedServer(config, onError) {
-  const builtIn = isBuiltInConfig(config) ? config : null;
-  const directServer = isBuiltInConfig(config) ? null : config;
-  const type = builtIn?.type;
-  const apiKey = builtIn?.type === "truefoundry" ? builtIn.apiKey : "";
-  const controlPlaneURL = builtIn?.type === "truefoundry" ? builtIn.controlPlaneURL : "";
-  const gatewayPlaneURL = builtIn?.type === "truefoundry" ? builtIn.gatewayPlaneURL ?? "" : "";
-  const trueforgeBaseUrl = builtIn?.type === "trueforge" ? builtIn.baseUrl ?? "" : "";
-  const trueforgeToken = builtIn?.type === "trueforge" ? builtIn.token ?? "" : "";
-  const trueforgeFetch = builtIn?.type === "trueforge" ? builtIn.fetch : void 0;
-  const catalog = builtIn?.catalog;
-  const [state, setState] = useState3(() => {
-    if (directServer) {
-      return { status: "ready", server: directServer, error: null };
-    }
-    return { status: "loading", server: null, error: null };
-  });
-  useEffect3(() => {
-    if (directServer) {
-      setState({ status: "ready", server: directServer, error: null });
-      return;
-    }
-    let cancelled = false;
-    setState({ status: "loading", server: null, error: null });
-    const resolve = async () => {
-      if (type === "trueforge") {
-        const { createTrueForgeAgentUIServer } = await import("./plugins/trueforge-agent-server-adapter/index.js");
-        return createTrueForgeAgentUIServer({
-          ...trueforgeBaseUrl ? { baseUrl: trueforgeBaseUrl } : {},
-          ...trueforgeToken ? { token: trueforgeToken } : {},
-          ...trueforgeFetch !== void 0 ? { fetch: trueforgeFetch } : {},
-          ...catalog != null ? { catalog } : {}
-        });
-      }
-      const runtimeServer = await createTrueFoundryAgentUIServer({
-        apiKey,
-        cpURL: controlPlaneURL,
-        ...gatewayPlaneURL ? { gatewayURL: gatewayPlaneURL } : {}
-      });
-      return toAgentUIServer(runtimeServer, catalog);
-    };
-    void resolve().then((server) => {
-      if (cancelled) return;
-      try {
-        setState({
-          status: "ready",
-          server: type === "trueforge" ? toAgentUIServer(server, void 0) : server,
-          error: null
-        });
-      } catch (error) {
-        onError?.(error);
-        setState({ status: "error", server: null, error });
-      }
-    }).catch((error) => {
-      if (cancelled) return;
-      onError?.(error);
-      setState({ status: "error", server: null, error });
-    });
-    return () => {
-      cancelled = true;
-    };
-  }, [
-    type,
-    directServer,
-    apiKey,
-    controlPlaneURL,
-    gatewayPlaneURL,
-    trueforgeBaseUrl,
-    trueforgeToken,
-    trueforgeFetch,
-    catalog,
-    onError
-  ]);
-  return state;
-}
-
-// src/containers/TrueforgeUI.tsx
-import { jsx as jsx12, jsxs as jsxs6 } from "react/jsx-runtime";
-var SidebarLayout = lazy(() => import("./SidebarLayout-F3XFXW7B.js").then((m) => ({ default: m.SidebarLayout })));
-var DrawerLayout = lazy(() => import("./DrawerLayout-ZVWCKEIS.js").then((m) => ({ default: m.DrawerLayout })));
-var DockLayout = lazy(() => import("./DockLayout-KEMFRWIR.js").then((m) => ({ default: m.DockLayout })));
-var WidgetLayout = lazy(() => import("./WidgetLayout-SZMABKIR.js").then((m) => ({ default: m.WidgetLayout })));
-function LayoutFallback({ className }) {
-  return /* @__PURE__ */ jsx12(
-    "div",
-    {
-      className,
-      style: {
-        height: "100%",
-        minHeight: "12rem",
-        background: "linear-gradient(90deg, color-mix(in oklab, var(--secondary-bg) 55%, transparent) 25%, color-mix(in oklab, var(--secondary-bg) 25%, transparent) 50%, color-mix(in oklab, var(--secondary-bg) 55%, transparent) 75%)",
-        backgroundSize: "200% 100%",
-        animation: "tfy-aui-pulse 1.2s ease-in-out infinite"
-      },
-      "aria-hidden": true
-    }
-  );
-}
-function ServerInitLoader({ className }) {
-  return /* @__PURE__ */ jsxs6(
-    "div",
-    {
-      role: "status",
-      "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__ */ jsx12(Spinner, { size: 28, className: "text-text-primary" }),
-        /* @__PURE__ */ jsx12("span", { className: "sr-only", children: "Loading" })
-      ]
-    }
-  );
-}
-function ServerInitError({ error, className }) {
-  const message = getErrorMessage(error, "Failed to initialize server");
-  return /* @__PURE__ */ jsx12(
-    "div",
-    {
-      role: "alert",
-      className: cn(
-        "flex h-full min-h-48 items-center justify-center bg-primary-bg px-6 text-center text-sm text-failure-bg",
-        className
-      ),
-      children: message
-    }
-  );
-}
-function isBuiltInLayout(layout) {
-  return layout === "sidebar" || layout === "drawer" || layout === "dock" || layout === "widget";
-}
-function BuiltInLayoutView({ layout, className }) {
-  switch (layout) {
-    case "sidebar":
-      return /* @__PURE__ */ jsx12(SidebarLayout, { className });
-    case "drawer":
-      return /* @__PURE__ */ jsx12(DrawerLayout, { className });
-    case "dock":
-      return /* @__PURE__ */ jsx12(DockLayout, { className });
-    case "widget":
-      return /* @__PURE__ */ jsx12(WidgetLayout, { className });
-  }
-}
-function LayoutChildren({ layout, className }) {
-  if (isBuiltInLayout(layout)) {
-    return /* @__PURE__ */ jsx12(Suspense, { fallback: /* @__PURE__ */ jsx12(LayoutFallback, { className }), children: /* @__PURE__ */ jsx12(BuiltInLayoutView, { layout, className }) });
-  }
-  const HostLayout = layout;
-  return /* @__PURE__ */ jsx12(HostLayout, { className });
-}
-function ChatProviderFromShell({
-  children,
-  initialSessionId: hostInitialSessionId,
-  ...providerRest
-}) {
-  const { mode, runtimeKey, listSessionsAgentId, pendingSessionId } = useShellMode();
-  const draftDefaultAgentSpec = useMemo2(() => {
-    if (mode.status === "active" && mode.isMutable) {
-      return mode.agentSpec ?? { model: { name: "openai-main/gpt-4.1" } };
-    }
-    return { model: { name: "openai-main/gpt-4.1" } };
-  }, [runtimeKey]);
-  const agent = useMemo2(() => {
-    if (mode.status === "idle") {
-      return {
-        mode: "draft",
-        defaultAgentSpec: { model: { name: "openai-main/gpt-4.1" } }
-      };
-    }
-    if (!mode.isMutable) {
-      const agentName = mode.agentName ?? mode.agentId ?? pendingSessionId;
-      if (agentName == null) {
-        return {
-          mode: "draft",
-          defaultAgentSpec: { model: { name: "openai-main/gpt-4.1" } }
-        };
-      }
-      return { mode: "named", agentName };
-    }
-    return {
-      mode: "draft",
-      defaultAgentSpec: draftDefaultAgentSpec
-    };
-  }, [mode, draftDefaultAgentSpec, pendingSessionId]);
-  return /* @__PURE__ */ jsx12(
-    TrueFoundryChatProvider,
-    {
-      ...providerRest,
-      agent,
-      listSessionsAgentId,
-      initialSessionId: pendingSessionId ?? hostInitialSessionId,
-      children: /* @__PURE__ */ jsxs6(DraftCatalogProvider, { children: [
-        /* @__PURE__ */ jsx12(DraftSpecPreferenceBridge, {}),
-        children
-      ] })
-    },
-    runtimeKey
-  );
-}
-function TrueforgeUI(props) {
+var TrueForgeUIWithRouter = lazy(
+  () => import("./TrueForgeUIWithRouter-IAQQTD5V.js").then((m) => ({ default: m.TrueForgeUIWithRouter }))
+);
+function TrueForgeUI(props) {
   const isMcpAuthScreen = typeof window !== "undefined" && new URLSearchParams(window.location.search).get("screenType") === "mcp-auth";
-  return isMcpAuthScreen ? /* @__PURE__ */ jsx12(SlotsProvider, { overrides: props.overrides, theme: props.theme, children: /* @__PURE__ */ jsx12(PostMcpOauthScreen_default, {}) }) : /* @__PURE__ */ jsx12(TrueforgeUIShell, { ...props });
-}
-function TrueforgeUIShell(props) {
-  const {
-    layout = "sidebar",
-    overrides,
-    theme,
-    className,
-    agentConfig = DEFAULT_AGENT_CONFIG,
-    initialSettingsOpen = false,
-    server: serverConfig,
-    onError,
-    ...providerRest
-  } = props;
-  const resolved = useResolvedServer(serverConfig, onError);
-  if (resolved.status === "loading") {
-    return /* @__PURE__ */ jsx12(SlotsProvider, { overrides, theme, children: /* @__PURE__ */ jsx12(ServerInitLoader, { className }) });
+  if (isMcpAuthScreen) {
+    return /* @__PURE__ */ jsx11(SlotsProvider, { overrides: props.overrides, theme: props.theme, children: /* @__PURE__ */ jsx11(PostMcpOauthScreen_default, {}) });
   }
-  if (resolved.status === "error") {
-    return /* @__PURE__ */ jsx12(SlotsProvider, { overrides, theme, children: /* @__PURE__ */ jsx12(ServerInitError, { error: resolved.error, className }) });
+  const { withRouter, ...rest } = props;
+  if (withRouter) {
+    return /* @__PURE__ */ jsx11(Suspense, { fallback: /* @__PURE__ */ jsx11(ServerInitLoader, { className: props.className }), children: /* @__PURE__ */ jsx11(TrueForgeUIWithRouter, { ...rest }) });
   }
-  const server = resolved.server;
-  const layoutTree = /* @__PURE__ */ jsx12(LayoutChildren, { layout, className });
-  return /* @__PURE__ */ jsx12(SlotsProvider, { overrides, theme, children: /* @__PURE__ */ jsx12(ServerProvider, { server, children: /* @__PURE__ */ jsx12(ShellModeProvider, { agentConfig, initialSettingsOpen, children: /* @__PURE__ */ jsx12(ChatProviderFromShell, { server, onError, ...providerRest, children: layoutTree }) }) }) });
+  return /* @__PURE__ */ jsx11(TrueForgeUIShell, { ...rest });
 }
 
 // src/index.ts
 import {
   mergeAgentSpec,
-  trueFoundryAttachmentAdapter as trueFoundryAttachmentAdapter2,
-  useTrueFoundryAgentRuntime as useTrueFoundryAgentRuntime2,
-  useTrueFoundryAgentSpec as useTrueFoundryAgentSpec2,
+  trueFoundryAttachmentAdapter,
+  useTrueFoundryAgentRuntime,
+  useTrueFoundryAgentSpec,
   useTrueFoundryCancel,
   useTrueFoundryDownloadSandboxFile,
   useTrueFoundryHistoryPagination,
@@ -727,7 +340,7 @@ import {
   useTrueFoundryRespondToToolApproval,
   useTrueFoundryToolResponses,
   useTrueFoundryTurnId,
-  useTrueFoundryUpdateAgentSpec as useTrueFoundryUpdateAgentSpec2
+  useTrueFoundryUpdateAgentSpec
 } from "@truefoundry/assistant-ui-runtime";
 export {
   AgentStepsCard,
@@ -819,8 +432,8 @@ export {
   ToolCallContentBlockContainer,
   ToolGroupCard,
   ToolGroupContainer,
+  TrueForgeUI,
   TrueFoundryChatProvider,
-  TrueforgeUI,
   USER_MESSAGE_ATTACHMENT_PREVIEW_REM,
   UserEditComposerContainer,
   UserMessageActionBar,
@@ -838,7 +451,7 @@ export {
   resolvePresetTokens,
   shellIsMutable,
   threadHasPendingMcpAuth,
-  trueFoundryAttachmentAdapter2 as trueFoundryAttachmentAdapter,
+  trueFoundryAttachmentAdapter,
   useAui,
   useAuiState,
   useBrand,
@@ -862,8 +475,8 @@ export {
   useThemeMode,
   useToaster,
   useToasterOptional,
-  useTrueFoundryAgentRuntime2 as useTrueFoundryAgentRuntime,
-  useTrueFoundryAgentSpec2 as useTrueFoundryAgentSpec,
+  useTrueFoundryAgentRuntime,
+  useTrueFoundryAgentSpec,
   useTrueFoundryCancel,
   useTrueFoundryDownloadSandboxFile,
   useTrueFoundryHistoryPagination,
@@ -871,6 +484,6 @@ export {
   useTrueFoundryRespondToToolApproval,
   useTrueFoundryToolResponses,
   useTrueFoundryTurnId,
-  useTrueFoundryUpdateAgentSpec2 as useTrueFoundryUpdateAgentSpec
+  useTrueFoundryUpdateAgentSpec
 };
 //# sourceMappingURL=index.js.map