@axiom-lattice/react-sdk 2.1.31 → 2.1.32

package/dist/index.mjs

@@ -2169,6 +2169,8 @@ var DEFAULT_CONFIG = {
   enableThreadCreation: true,
   enableThreadList: true,
   showSideMenu: true,
+  enableAssistantCreation: true,
+  enableAssistantEditing: true,
   availableMiddlewareTypes: DEFAULT_MIDDLEWARE_TYPES
 };
 var LatticeChatShellContext = createContext4({
@@ -7392,22 +7394,23 @@ import { useContext as useContext8 } from "react";
 import { useState as useState31 } from "react";
 import { Modal as Modal5 } from "antd";
 import {
-  SettingOutlined as SettingOutlined3,
-  ApartmentOutlined,
-  StarOutlined as StarOutlined2,
-  ToolOutlined as ToolOutlined4
-} from "@ant-design/icons";
+  Bot,
+  BrainCircuit,
+  Wrench,
+  Settings
+} from "lucide-react";
 import { createStyles as createStyles13 } from "antd-style";
 
 // src/components/Chat/AssistantFlow.tsx
-import { useMemo as useMemo12, useEffect as useEffect18, useState as useState26, useCallback as useCallback16 } from "react";
+import { useMemo as useMemo12, useEffect as useEffect18, useState as useState26, useCallback as useCallback16, useRef as useRef14 } from "react";
 import {
   ReactFlow,
   Background,
   Controls,
   useNodesState,
   useEdgesState,
-  ReactFlowProvider
+  ReactFlowProvider,
+  useReactFlow as useReactFlow2
 } from "@xyflow/react";
 import "@xyflow/react/dist/style.css";
 
@@ -7898,7 +7901,7 @@ var AssistantNode = ({
               e.currentTarget.style.boxShadow = `0 1px 4px rgba(0, 0, 0, 0.08)`;
             },
             children: [
-              /* @__PURE__ */ jsx47(
+              onConfigure && /* @__PURE__ */ jsx47(
                 "div",
                 {
                   onClick: (e) => {
@@ -8301,7 +8304,8 @@ import {
   GlobalOutlined as GlobalOutlined2,
   DatabaseOutlined,
   FolderOutlined as FolderOutlined2,
-  StarOutlined
+  StarOutlined,
+  DeleteOutlined
 } from "@ant-design/icons";
 import { Fragment as Fragment7, jsx as jsx48, jsxs as jsxs27 } from "react/jsx-runtime";
 var { TextArea } = Input;
@@ -8741,13 +8745,16 @@ var AgentConfigPanel = ({
   allAssistants,
   visible,
   onClose,
-  onSave
+  onSave,
+  onDelete
 }) => {
   const [form] = Form.useForm();
   const { token } = theme3.useToken();
   const { config } = useLatticeChatShellContext();
   const [toolsLoading, setToolsLoading] = useState24(false);
   const [saving, setSaving] = useState24(false);
+  const [deleting, setDeleting] = useState24(false);
+  const [isDeleteModalOpen, setIsDeleteModalOpen] = useState24(false);
   const [tools, setTools] = useState24([]);
   const [selectedTools, setSelectedTools] = useState24([]);
   const [selectedSubAgents, setSelectedSubAgents] = useState24([]);
@@ -8863,6 +8870,20 @@ var AgentConfigPanel = ({
       setSaving(false);
     }
   };
+  const handleDelete = async () => {
+    if (!assistant) return;
+    setDeleting(true);
+    try {
+      await onDelete?.(assistant);
+      setIsDeleteModalOpen(false);
+      onClose();
+    } catch (error) {
+      setIsDeleteModalOpen(false);
+      message5.error(error instanceof Error ? error.message : "Failed to delete assistant");
+    } finally {
+      setDeleting(false);
+    }
+  };
   const handleMiddlewareToggle = (middlewareId) => {
     setMiddleware(
       (prev) => prev.map((m) => {
@@ -9013,15 +9034,27 @@ var AgentConfigPanel = ({
                 /* @__PURE__ */ jsx48(Text16, { strong: true, style: { fontSize: 14 }, children: assistant.name }),
                 /* @__PURE__ */ jsx48(Tag7, { color: "blue", style: { margin: 0, fontSize: 10 }, children: "Config" })
               ] }),
-              /* @__PURE__ */ jsx48(
-                Button19,
-                {
-                  type: "text",
-                  size: "small",
-                  icon: /* @__PURE__ */ jsx48(CloseOutlined2, {}),
-                  onClick: onClose
-                }
-              )
+              /* @__PURE__ */ jsxs27(Space18, { children: [
+                onDelete && /* @__PURE__ */ jsx48(Tooltip10, { title: "Delete Assistant", children: /* @__PURE__ */ jsx48(
+                  Button19,
+                  {
+                    type: "text",
+                    size: "small",
+                    danger: true,
+                    icon: /* @__PURE__ */ jsx48(DeleteOutlined, {}),
+                    onClick: () => setIsDeleteModalOpen(true)
+                  }
+                ) }),
+                /* @__PURE__ */ jsx48(
+                  Button19,
+                  {
+                    type: "text",
+                    size: "small",
+                    icon: /* @__PURE__ */ jsx48(CloseOutlined2, {}),
+                    onClick: onClose
+                  }
+                )
+              ] })
             ]
           }
         ),
@@ -9300,10 +9333,10 @@ var AgentConfigPanel = ({
               alignItems: "center"
             },
             children: [
-              /* @__PURE__ */ jsx48(Text16, { type: "secondary", style: { fontSize: 12 }, children: "Changes will be saved to the agent" }),
+              /* @__PURE__ */ jsx48(Text16, { type: "secondary", style: { fontSize: 12 }, children: onSave ? "Changes will be saved to the agent" : "Viewing agent configuration" }),
               /* @__PURE__ */ jsxs27(Space18, { children: [
                 /* @__PURE__ */ jsx48(Button19, { onClick: onClose, size: "small", children: "Close" }),
-                /* @__PURE__ */ jsx48(
+                onSave && /* @__PURE__ */ jsx48(
                   Button19,
                   {
                     type: "primary",
@@ -9392,6 +9425,33 @@ var AgentConfigPanel = ({
               )
             ]
           }
+        ),
+        /* @__PURE__ */ jsx48(
+          Modal,
+          {
+            title: "Delete Assistant",
+            open: isDeleteModalOpen,
+            onCancel: () => setIsDeleteModalOpen(false),
+            footer: [
+              /* @__PURE__ */ jsx48(Button19, { onClick: () => setIsDeleteModalOpen(false), children: "Cancel" }, "cancel"),
+              /* @__PURE__ */ jsx48(
+                Button19,
+                {
+                  type: "primary",
+                  danger: true,
+                  loading: deleting,
+                  onClick: handleDelete,
+                  children: "Delete"
+                },
+                "delete"
+              )
+            ],
+            children: /* @__PURE__ */ jsxs27(Text16, { children: [
+              "Are you sure you want to delete ",
+              /* @__PURE__ */ jsx48(Text16, { strong: true, children: assistant.name }),
+              "? This action cannot be undone."
+            ] })
+          }
         )
       ]
     }
@@ -9463,7 +9523,7 @@ var CreateAssistantModal = ({
           description: "Assistant created successfully"
         });
         form.resetFields();
-        onSuccess?.();
+        onSuccess?.(data.data);
         onCancel();
       } else {
         message6.error(data.message || "Failed to create assistant");
@@ -9597,6 +9657,7 @@ import { Button as Button21, message as message7, notification as notification2,
 import { PlusOutlined as PlusOutlined3 } from "@ant-design/icons";
 import { jsx as jsx50, jsxs as jsxs29 } from "react/jsx-runtime";
 var AssistantFlowInner = ({ onChat }) => {
+  const { setCenter, fitView } = useReactFlow2();
   const [nodes, setNodes, onNodesChange] = useNodesState([]);
   const [edges, setEdges, onEdgesChange] = useEdgesState([]);
   const [assistants, setAssistants] = useState26([]);
@@ -9606,6 +9667,7 @@ var AssistantFlowInner = ({ onChat }) => {
   const [isCreateModalOpen, setIsCreateModalOpen] = useState26(false);
   const { config } = useLatticeChatShellContext();
   const { token } = theme5.useToken();
+  const { enableAssistantCreation = true, enableAssistantEditing = true } = config;
   const nodeTypes = useMemo12(
     () => ({
       assistant: AssistantNode_default
@@ -9618,6 +9680,7 @@ var AssistantFlowInner = ({ onChat }) => {
   }, []);
   const handleClosePanel = useCallback16(() => {
     setConfigPanelVisible(false);
+    setSelectedAssistant(null);
   }, []);
   const fetchAssistants = async () => {
     setAssistantsLoading(true);
@@ -9796,7 +9859,7 @@ var AssistantFlowInner = ({ onChat }) => {
             type: "assistant",
             data: {
               assistant,
-              onConfigure: () => handleConfigure(assistant),
+              onConfigure: enableAssistantEditing ? () => handleConfigure(assistant) : void 0,
               onChat
             },
             position: { x, y }
@@ -9826,7 +9889,7 @@ var AssistantFlowInner = ({ onChat }) => {
           type: "assistant",
           data: {
             assistant,
-            onConfigure: () => handleConfigure(assistant),
+            onConfigure: enableAssistantEditing ? () => handleConfigure(assistant) : void 0,
             onChat
           },
           position: { x, y }
@@ -9887,9 +9950,82 @@ var AssistantFlowInner = ({ onChat }) => {
       throw error;
     }
   };
-  const handleAssistantCreated = async () => {
-    await fetchAssistants();
+  const handleDeleteAssistant = async (assistant) => {
+    try {
+      const headers = {};
+      if (config.apiKey) {
+        headers["Authorization"] = `Bearer ${config.apiKey}`;
+      }
+      const response = await fetch(
+        `${config.baseURL}/api/assistants/${assistant.id}`,
+        {
+          method: "DELETE",
+          headers,
+          body: JSON.stringify({})
+        }
+      );
+      const data = await response.json();
+      if (response.ok && data.success) {
+        setAssistants((prev) => prev.filter((a) => a.id !== assistant.id));
+        if (selectedAssistant?.id === assistant.id) {
+          setSelectedAssistant(null);
+          setConfigPanelVisible(false);
+        }
+        notification2.success({
+          message: "Deleted",
+          description: "Assistant deleted successfully"
+        });
+      } else {
+        message7.error(data.message || "Failed to delete assistant");
+        throw new Error(data.message || "Failed to delete assistant");
+      }
+    } catch (error) {
+      console.error("Failed to delete assistant:", error);
+      message7.error(error instanceof Error ? error.message : "Failed to delete assistant");
+      throw error;
+    }
+  };
+  const handleAssistantCreated = (createdAssistant) => {
+    setAssistants((prev) => {
+      const exists = prev.some((a) => a.id === createdAssistant.id);
+      if (exists) {
+        return prev.map((a) => a.id === createdAssistant.id ? createdAssistant : a);
+      }
+      return [...prev, createdAssistant];
+    });
+    lastCreatedAssistantRef.current = createdAssistant;
   };
+  useEffect18(() => {
+    if (selectedAssistant) {
+      const node = nodes.find((n) => n.id === selectedAssistant.id);
+      if (node) {
+        setCenter(node.position.x + 140, node.position.y + 100, { zoom: 1, duration: 800 });
+      }
+    }
+  }, [selectedAssistant, nodes, setCenter]);
+  useEffect18(() => {
+    if (assistants.length > 0 && selectedAssistant) {
+      const assistant = assistants.find((a) => a.id === selectedAssistant.id);
+      if (assistant && !configPanelVisible) {
+        setConfigPanelVisible(true);
+      }
+    }
+  }, [assistants, selectedAssistant, configPanelVisible]);
+  const lastCreatedAssistantRef = useRef14(null);
+  useEffect18(() => {
+    if (lastCreatedAssistantRef.current && !assistants.find((a) => a.id === lastCreatedAssistantRef.current?.id)) {
+      lastCreatedAssistantRef.current = null;
+    }
+    if (lastCreatedAssistantRef.current) {
+      const node = nodes.find((n) => n.id === lastCreatedAssistantRef.current.id);
+      if (node) {
+        setCenter(node.position.x + 140, node.position.y + 100, { zoom: 1, duration: 800 });
+        setSelectedAssistant(lastCreatedAssistantRef.current);
+        setConfigPanelVisible(true);
+        lastCreatedAssistantRef.current = null;
+      }
+    }
+  }, [nodes, setCenter]);
   if (assistantsLoading) {
     return /* @__PURE__ */ jsx50(
       "div",
@@ -9932,7 +10068,7 @@ var AssistantFlowInner = ({ onChat }) => {
               ]
             }
           ),
-          /* @__PURE__ */ jsx50(
+          enableAssistantCreation && /* @__PURE__ */ jsx50(
             "div",
             {
               style: {
@@ -9972,7 +10108,8 @@ var AssistantFlowInner = ({ onChat }) => {
         assistant: selectedAssistant,
         allAssistants: assistants,
         onClose: handleClosePanel,
-        onSave: handleSaveConfig
+        onSave: handleSaveConfig,
+        onDelete: handleDeleteAssistant
       }
     ),
     /* @__PURE__ */ jsx50(
@@ -12035,8 +12172,9 @@ var useStyles4 = createStyles13(({ token, css }) => ({
       box-shadow: 0 1px 2px rgba(0, 0, 0, 0.04);
     }
 
-    .anticon {
-      font-size: 22px;
+    .lucide {
+      width: 22px;
+      height: 22px;
     }
   `
 }));
@@ -12071,7 +12209,7 @@ var ChatSidebar = ({
           onClick: handleAssistantFlowClick,
           title: "Assistant Flow Canvas",
           "aria-label": "Assistant Flow Canvas",
-          children: /* @__PURE__ */ jsx55(ApartmentOutlined, {})
+          children: /* @__PURE__ */ jsx55(Bot, {})
         }
       ),
       /* @__PURE__ */ jsx55(
@@ -12081,7 +12219,7 @@ var ChatSidebar = ({
           onClick: handleSkillFlowClick,
           title: "Skill Flow Canvas",
           "aria-label": "Skill Flow Canvas",
-          children: /* @__PURE__ */ jsx55(StarOutlined2, {})
+          children: /* @__PURE__ */ jsx55(BrainCircuit, {})
         }
       ),
       /* @__PURE__ */ jsx55(
@@ -12091,7 +12229,7 @@ var ChatSidebar = ({
           onClick: handleToolsClick,
           title: "Tools List",
           "aria-label": "Tools List",
-          children: /* @__PURE__ */ jsx55(ToolOutlined4, {})
+          children: /* @__PURE__ */ jsx55(Wrench, {})
         }
       ),
       /* @__PURE__ */ jsx55(
@@ -12101,7 +12239,7 @@ var ChatSidebar = ({
           onClick: handleSettingsClick,
           title: "Settings",
           "aria-label": "Settings",
-          children: /* @__PURE__ */ jsx55(SettingOutlined3, {})
+          children: /* @__PURE__ */ jsx55(Settings, {})
         }
       )
     ] }) }),
@@ -12215,11 +12353,11 @@ var LatticeChatView = (props) => {
 };
 
 // src/components/Chat/SettingsModal.tsx
-import { useState as useState32, useEffect as useEffect22, useRef as useRef14 } from "react";
+import { useState as useState32, useEffect as useEffect22, useRef as useRef15 } from "react";
 import {
   Modal as Modal6,
   Input as Input5,
-  Button as Button27,
+  Button as Button26,
   message as message10,
   notification as notification5,
   Typography as Typography22,
@@ -12629,7 +12767,7 @@ var SettingsModal = ({
     return [];
   });
   const [serverConfigs, setServerConfigs] = useState32({});
-  const connectionsRef = useRef14(connections);
+  const connectionsRef = useRef15(connections);
   useEffect22(() => {
     connectionsRef.current = connections;
   }, [connections]);
@@ -13191,7 +13329,7 @@ QUEUE_NAME=tasks`,
                 )
               ] }),
               config.models.length > 1 && /* @__PURE__ */ jsx57(
-                Button27,
+                Button26,
                 {
                   type: "text",
                   danger: true,
@@ -13327,7 +13465,7 @@ QUEUE_NAME=tasks`,
         ] })
       ] }, index)),
       /* @__PURE__ */ jsx57(
-        Button27,
+        Button26,
         {
           type: "dashed",
           onClick: handleAddModel,
@@ -13405,9 +13543,9 @@ QUEUE_NAME=tasks`,
             ] })
           ] }),
           /* @__PURE__ */ jsxs35("div", { className: styles.contentHeaderRight, children: [
-            /* @__PURE__ */ jsx57(Button27, { onClick: onClose, children: "Cancel" }),
+            /* @__PURE__ */ jsx57(Button26, { onClick: onClose, children: "Cancel" }),
             /* @__PURE__ */ jsx57(
-              Button27,
+              Button26,
               {
                 type: "primary",
                 icon: /* @__PURE__ */ jsx57(SaveOutlined2, {}),
@@ -13451,7 +13589,7 @@ QUEUE_NAME=tasks`,
             }
           ),
           /* @__PURE__ */ jsx57(
-            Button27,
+            Button26,
             {
               type: "primary",
               icon: /* @__PURE__ */ jsx57(LinkOutlined, {}),
@@ -13585,11 +13723,21 @@ var AgentServerSetting = () => {
 // src/components/Chat/LatticeChatShell.tsx
 import { jsx as jsx59, jsxs as jsxs36 } from "react/jsx-runtime";
 var LatticeChatShell = (props) => {
-  const { initialConfig } = props;
-  return /* @__PURE__ */ jsxs36(LatticeChatShellContextProvider, { ...props, children: [
-    /* @__PURE__ */ jsx59(AssistantContextProvider, { autoLoad: true, initialAssistantId: initialConfig?.assistantId, children: /* @__PURE__ */ jsx59(ConversationContextProvider, { children: /* @__PURE__ */ jsx59(LatticeChatView, {}) }) }),
-    /* @__PURE__ */ jsx59(AgentServerSetting, {})
-  ] });
+  const { enableAssistantCreation, enableAssistantEditing, ...restProps } = props;
+  return /* @__PURE__ */ jsxs36(
+    LatticeChatShellContextProvider,
+    {
+      initialConfig: {
+        enableAssistantCreation,
+        enableAssistantEditing,
+        ...restProps.initialConfig
+      },
+      children: [
+        /* @__PURE__ */ jsx59(AssistantContextProvider, { autoLoad: true, initialAssistantId: restProps.initialConfig?.assistantId, children: /* @__PURE__ */ jsx59(ConversationContextProvider, { children: /* @__PURE__ */ jsx59(LatticeChatView, {}) }) }),
+        /* @__PURE__ */ jsx59(AgentServerSetting, {})
+      ]
+    }
+  );
 };
 export {
   AgentConversations,