@axiom-lattice/react-sdk 2.1.44 → 2.1.46

package/dist/index.d.mts

@@ -717,16 +717,28 @@ declare const ChatUIContext: React$1.Context<{
         component_key: string;
         data: any;
         message?: string;
+        context?: {
+            thread_id?: string;
+            assistant_id?: string;
+        };
     } | null;
     setSideAppSelectedCard: (card: {
         component_key: string;
         data: any;
         message?: string;
+        context?: {
+            thread_id?: string;
+            assistant_id?: string;
+        };
     } | null) => void;
     openSideApp: (card: {
         component_key: string;
         data: any;
         message?: string;
+        context?: {
+            thread_id?: string;
+            assistant_id?: string;
+        };
     }) => void;
     closeSideApp: () => void;
     menuCollapsed: boolean;
@@ -744,16 +756,28 @@ declare const useChatUIContext: () => {
         component_key: string;
         data: any;
         message?: string;
+        context?: {
+            thread_id?: string;
+            assistant_id?: string;
+        };
     } | null;
     setSideAppSelectedCard: (card: {
         component_key: string;
         data: any;
         message?: string;
+        context?: {
+            thread_id?: string;
+            assistant_id?: string;
+        };
     } | null) => void;
     openSideApp: (card: {
         component_key: string;
         data: any;
         message?: string;
+        context?: {
+            thread_id?: string;
+            assistant_id?: string;
+        };
     }) => void;
     closeSideApp: () => void;
     menuCollapsed: boolean;

package/dist/index.d.ts

@@ -717,16 +717,28 @@ declare const ChatUIContext: React$1.Context<{
         component_key: string;
         data: any;
         message?: string;
+        context?: {
+            thread_id?: string;
+            assistant_id?: string;
+        };
     } | null;
     setSideAppSelectedCard: (card: {
         component_key: string;
         data: any;
         message?: string;
+        context?: {
+            thread_id?: string;
+            assistant_id?: string;
+        };
     } | null) => void;
     openSideApp: (card: {
         component_key: string;
         data: any;
         message?: string;
+        context?: {
+            thread_id?: string;
+            assistant_id?: string;
+        };
     }) => void;
     closeSideApp: () => void;
     menuCollapsed: boolean;
@@ -744,16 +756,28 @@ declare const useChatUIContext: () => {
         component_key: string;
         data: any;
         message?: string;
+        context?: {
+            thread_id?: string;
+            assistant_id?: string;
+        };
     } | null;
     setSideAppSelectedCard: (card: {
         component_key: string;
         data: any;
         message?: string;
+        context?: {
+            thread_id?: string;
+            assistant_id?: string;
+        };
     } | null) => void;
     openSideApp: (card: {
         component_key: string;
         data: any;
         message?: string;
+        context?: {
+            thread_id?: string;
+            assistant_id?: string;
+        };
     }) => void;
     closeSideApp: () => void;
     menuCollapsed: boolean;

package/dist/index.js

@@ -6414,10 +6414,26 @@ var Todo = ({
   data,
   component_key,
   interactive = true,
-  focusMode = false
+  focusMode = false,
+  smartCollapse = true
 }) => {
   const { styles } = useStyle3();
-  const [isExpanded, setIsExpanded] = (0, import_react26.useState)(false);
+  const getDefaultExpanded = () => {
+    if (!data || !Array.isArray(data) || data.length <= 3) {
+      return true;
+    }
+    const allPending = data.every((item) => item.status === "pending");
+    const allCompleted = data.every((item) => item.status === "completed");
+    return allPending || allCompleted;
+  };
+  const [isExpanded, setIsExpanded] = (0, import_react26.useState)(
+    () => smartCollapse ? getDefaultExpanded() : false
+  );
+  (0, import_react26.useEffect)(() => {
+    if (smartCollapse) {
+      setIsExpanded(getDefaultExpanded());
+    }
+  }, [data, smartCollapse]);
   const getStatusIcon3 = (status) => {
     switch (status) {
       case "completed":
@@ -7672,8 +7688,12 @@ var TaskCard = ({
   const { description, subagent_type, assignee } = toolCallData?.args || {};
   const status = toolCallData.status || "pending";
   const response = toolCallData.response || null;
-  const { threadId } = useAgentChat();
-  const subagent_thread_id = (threadId || "") + "____" + subagent_type + "_" + toolCallData.id;
+  const { threadId, assistantId } = useAgentChat();
+  let new_assistant_id = subagent_type;
+  if (subagent_type === "general-purpose") {
+    new_assistant_id = assistantId + "-general-purpose";
+  }
+  const subagent_thread_id = (threadId || "") + "____" + new_assistant_id + "_" + toolCallData.id;
   const getStatusConfig3 = (status2) => {
     switch (status2) {
       case "success":
@@ -7709,7 +7729,7 @@ var TaskCard = ({
       data: {
         thread_id: subagent_thread_id,
         description,
-        subagent_type
+        subagent_type: new_assistant_id
       }
     });
   };
@@ -8473,7 +8493,7 @@ var import_antd30 = require("antd");
 var import_icons16 = require("@ant-design/icons");
 var import_jsx_runtime48 = require("react/jsx-runtime");
 var TeamButton = () => {
-  const { agentState, threadId } = useAgentChat();
+  const { agentState, threadId, assistantId } = useAgentChat();
   const { openSideApp } = useChatUIContext();
   const teamData = agentState?.values?.team;
   const teammates = teamData?.teammates || [];
@@ -8492,6 +8512,10 @@ var TeamButton = () => {
           description: t.description
         })),
         parent_thread_id: threadId
+      },
+      context: {
+        thread_id: threadId,
+        assistant_id: assistantId
       }
     });
   };
@@ -12699,7 +12723,7 @@ var TeamGraph = ({ data }) => {
   const tasks = safeGetTasks(data);
   const teamId = safeGetTeamId(data);
   const { openSideApp } = useChatUIContext();
-  const { threadId } = useAgentChat();
+  const { threadId, assistantId } = useAgentChat();
   const colors3 = [
     "#1890ff",
     "#52c41a",
@@ -12729,6 +12753,10 @@ var TeamGraph = ({ data }) => {
         teamId,
         teammates,
         parent_thread_id: threadId
+      },
+      context: {
+        thread_id: threadId,
+        assistant_id: assistantId
       }
     });
   };
@@ -13085,8 +13113,8 @@ var import_antd_style19 = require("antd-style");
 
 // src/components/GenUI/elements/TeamWorkspace/hooks/useTeamWorkspaceData.ts
 var import_react47 = require("react");
-function useTeamWorkspaceData(threadId) {
-  const { agentState, isLoading, error, refresh } = useAgentState(threadId, {
+function useTeamWorkspaceData(threadId, assistantId) {
+  const { agentState, isLoading, error, refresh } = useAgentState(threadId, assistantId, {
     pollingInterval: 5e3,
     autoStart: true
   });
@@ -15562,10 +15590,12 @@ var getTeammateAssistantId = (teamId, agentName) => {
   return `team:${teamId}:${agentName}`;
 };
 var TeamWorkspace = ({
-  data
+  data,
+  context
 }) => {
   const { styles } = useStyles16();
   const { teamId, teammates, parent_thread_id } = data || {};
+  const assistantId = context?.assistant_id || "";
   const {
     team,
     tasks,
@@ -15575,7 +15605,7 @@ var TeamWorkspace = ({
     recentActivities,
     isLoading,
     refresh
-  } = useTeamWorkspaceData(parent_thread_id || null);
+  } = useTeamWorkspaceData(parent_thread_id || null, assistantId);
   const [activeMenuId, setActiveMenuId] = (0, import_react53.useState)("dashboard");
   const menuItems = (0, import_react53.useMemo)(() => {
     const items = [
@@ -15663,13 +15693,13 @@ var TeamWorkspace = ({
             teamId,
             activeItem.data.name
           );
-          const assistantId = getTeammateAssistantId(teamId, activeItem.data.name);
+          const assistantId2 = getTeammateAssistantId(teamId, activeItem.data.name);
           return /* @__PURE__ */ (0, import_jsx_runtime79.jsx)(
             TeamMemberChat,
             {
               teammate: activeItem.data,
               threadId,
-              assistantId
+              assistantId: assistantId2
             }
           );
         }
@@ -16866,7 +16896,8 @@ var SideAppViewBrowser = () => {
         SideAppView,
         {
           component_key: sideAppSelectedCard?.component_key || "",
-          data: sideAppSelectedCard?.data
+          data: sideAppSelectedCard?.data,
+          context: sideAppSelectedCard?.context
         }
       ),
       sideAppSelectedCard?.component_key
@@ -19335,6 +19366,7 @@ var AgentConfigPanel = ({
       const values = await form.validateFields();
       const configToSave = {
         ...values,
+        type: agentType,
         tools: selectedTools,
         subAgents: selectedSubAgents,
         prompt: agentPrompt,