@datatechsolutions/ui 2.7.110 → 2.7.111

package/dist/workflow/index.mjs

@@ -11012,11 +11012,14 @@ function ResultsTab({ agentId, t }) {
     ] })
   ] });
 }
-function AgentDrawer({ onSaved }) {
+function AgentDrawer({ agent: agentProp, models: modelsProp, open: openProp, onClose: onCloseProp, onSaved }) {
   const t = useTranslations("agents.workflow");
-  const agent = useAgentDrawerStore((state) => state.agent);
-  const models = useAgentDrawerStore((state) => state.models);
-  const open = useAgentDrawerStore((state) => state.open);
+  const agentStore = useAgentDrawerStore((state) => state.agent);
+  const modelsStore = useAgentDrawerStore((state) => state.models);
+  const openStore = useAgentDrawerStore((state) => state.open);
+  const agent = agentProp ?? agentStore;
+  const models = modelsProp ?? modelsStore;
+  const open = openProp ?? openStore;
   const activeTab = useAgentDrawerStore((state) => state.activeTab);
   const selectedModelId = useAgentDrawerStore((state) => state.selectedModelId);
   const selectedFramework = useAgentDrawerStore((state) => state.selectedFramework);
@@ -11032,8 +11035,9 @@ function AgentDrawer({ onSaved }) {
   const markSaved = useAgentDrawerStore((state) => state.markSaved);
   const closeDrawer = useAgentDrawerStore((state) => state.closeDrawer);
   const handleClose = useCallback(() => {
+    onCloseProp?.();
     closeDrawer();
-  }, [closeDrawer]);
+  }, [onCloseProp, closeDrawer]);
   const handleMarkSaved = useCallback(() => {
     markSaved();
     onSaved?.();