@truefoundry/trueforge-ui 0.1.0-rc.0 → 0.1.0-rc.2

package/dist/index.js

@@ -26,7 +26,7 @@ import {
   threadHasPendingMcpAuth,
   useComposerBusyState,
   useComposerPauseView
-} from "./chunk-UXZOL5GO.js";
+} from "./chunk-H7DR64W3.js";
 import {
   useAui,
   useAuiState
@@ -55,7 +55,6 @@ import {
   ComposerRightSection,
   ComposerSendButton,
   ComposerShell,
-  DraftCatalogProvider,
   DraftComposerLeftSection,
   DraftComposerRightSection,
   DraftCompositeSelector,
@@ -102,36 +101,38 @@ import {
   UserMessageEdit,
   WelcomeScreen,
   defaultSlots,
-  getErrorMessage,
+  modelPatchWithReasoningEffort,
   preloadMarkdownOpenUI,
   useBrandName,
-  useDraftCatalog,
   useMCPAuth,
   useSlot,
   useThemeMode,
   useToaster,
   useToasterOptional
-} from "./chunk-XSHXH6EI.js";
+} from "./chunk-QGNLP2EE.js";
 import {
   DEFAULT_AGENT_CONFIG,
+  DraftCatalogProvider,
   Icon,
   ServerProvider,
   ShellModeProvider,
   Spinner,
+  getErrorMessage,
   libraryAgentId,
   registerIcons,
   shellIsMutable,
   useCatalogServer,
+  useDraftCatalog,
   useOptionalCatalogServer,
   useOptionalServer,
   useOptionalShellMode,
   useServer,
   useServerCapabilities,
   useShellMode
-} from "./chunk-UHGSHDBM.js";
+} from "./chunk-XAZ5HJXV.js";
 import {
   OpenUiFenceBlock
-} from "./chunk-42GF723P.js";
+} from "./chunk-HT5HKQXE.js";
 import {
   PRESETS,
   ThemeProvider,
@@ -142,7 +143,7 @@ import {
   useOptionalContentClassNames,
   useTheme,
   useThemeIcons
-} from "./chunk-UBOFEI77.js";
+} from "./chunk-DTF7ZMUJ.js";
 
 // src/icons/agent-2.svg
 import * as React from "react";
@@ -306,6 +307,97 @@ 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 {
@@ -367,7 +459,7 @@ function TrueFoundryChatProvider(props) {
 
 // src/containers/useResolvedServer.ts
 import { createTrueFoundryAgentUIServer } from "@truefoundry/assistant-ui-runtime/plugins/truefoundry-agent-server-adapter";
-import { useEffect as useEffect2, useState as useState3 } from "react";
+import { useEffect as useEffect3, useState as useState3 } from "react";
 var DEFAULT_CAPABILITIES = {
   data: {
     sandbox: { enabled: true },
@@ -410,7 +502,7 @@ function useResolvedServer(config, onError) {
     }
     return { status: "loading", server: null, error: null };
   });
-  useEffect2(() => {
+  useEffect3(() => {
     if (directServer) {
       setState({ status: "ready", server: directServer, error: null });
       return;
@@ -471,10 +563,10 @@ function useResolvedServer(config, onError) {
 
 // src/containers/TrueforgeUI.tsx
 import { jsx as jsx12, jsxs as jsxs6 } from "react/jsx-runtime";
-var SidebarLayout = lazy(() => import("./SidebarLayout-M7HEOZVK.js").then((m) => ({ default: m.SidebarLayout })));
-var DrawerLayout = lazy(() => import("./DrawerLayout-X2UXYTOS.js").then((m) => ({ default: m.DrawerLayout })));
-var DockLayout = lazy(() => import("./DockLayout-WRF7EQLD.js").then((m) => ({ default: m.DockLayout })));
-var WidgetLayout = lazy(() => import("./WidgetLayout-7MDOKLHU.js").then((m) => ({ default: m.WidgetLayout })));
+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",
@@ -586,7 +678,10 @@ function ChatProviderFromShell({
       agent,
       listSessionsAgentId,
       initialSessionId: pendingSessionId ?? hostInitialSessionId,
-      children: /* @__PURE__ */ jsx12(DraftCatalogProvider, { children })
+      children: /* @__PURE__ */ jsxs6(DraftCatalogProvider, { children: [
+        /* @__PURE__ */ jsx12(DraftSpecPreferenceBridge, {}),
+        children
+      ] })
     },
     runtimeKey
   );
@@ -597,7 +692,7 @@ function TrueforgeUI(props) {
 }
 function TrueforgeUIShell(props) {
   const {
-    layout,
+    layout = "sidebar",
     overrides,
     theme,
     className,
@@ -624,7 +719,7 @@ import {
   mergeAgentSpec,
   trueFoundryAttachmentAdapter as trueFoundryAttachmentAdapter2,
   useTrueFoundryAgentRuntime as useTrueFoundryAgentRuntime2,
-  useTrueFoundryAgentSpec,
+  useTrueFoundryAgentSpec as useTrueFoundryAgentSpec2,
   useTrueFoundryCancel,
   useTrueFoundryDownloadSandboxFile,
   useTrueFoundryHistoryPagination,
@@ -632,7 +727,7 @@ import {
   useTrueFoundryRespondToToolApproval,
   useTrueFoundryToolResponses,
   useTrueFoundryTurnId,
-  useTrueFoundryUpdateAgentSpec
+  useTrueFoundryUpdateAgentSpec as useTrueFoundryUpdateAgentSpec2
 } from "@truefoundry/assistant-ui-runtime";
 export {
   AgentStepsCard,
@@ -768,7 +863,7 @@ export {
   useToaster,
   useToasterOptional,
   useTrueFoundryAgentRuntime2 as useTrueFoundryAgentRuntime,
-  useTrueFoundryAgentSpec,
+  useTrueFoundryAgentSpec2 as useTrueFoundryAgentSpec,
   useTrueFoundryCancel,
   useTrueFoundryDownloadSandboxFile,
   useTrueFoundryHistoryPagination,
@@ -776,6 +871,6 @@ export {
   useTrueFoundryRespondToToolApproval,
   useTrueFoundryToolResponses,
   useTrueFoundryTurnId,
-  useTrueFoundryUpdateAgentSpec
+  useTrueFoundryUpdateAgentSpec2 as useTrueFoundryUpdateAgentSpec
 };
 //# sourceMappingURL=index.js.map