@mastra/playground-ui 5.0.3-alpha.1 → 5.0.4-alpha.0

package/dist/index.cjs.js

@@ -18,7 +18,10 @@ const langJson = require('@codemirror/lang-json');
 const highlight$1 = require('@lezer/highlight');
 const codemirrorThemeDracula = require('@uiw/codemirror-theme-dracula');
 const CodeMirror = require('@uiw/react-codemirror');
+const di = require('@mastra/core/di');
 const clientJs = require('@mastra/client-js');
+const zustand = require('zustand');
+const middleware = require('zustand/middleware');
 const dateFns = require('date-fns');
 const TabsPrimitive = require('@radix-ui/react-tabs');
 const sonner = require('sonner');
@@ -26,6 +29,7 @@ const react$1 = require('motion/react');
 const uiUtils = require('@ai-sdk/ui-utils');
 const Markdown = require('react-markdown');
 const useDebounce = require('use-debounce');
+const runtimeContext = require('@mastra/core/runtime-context');
 const react$2 = require('@xyflow/react');
 require('@xyflow/react/dist/style.css');
 const Dagre = require('@dagrejs/dagre');
@@ -4053,12 +4057,17 @@ function MastraRuntimeProvider({
   baseUrl,
   refreshThreadList,
   modelSettings = {},
-  chatWithGenerate
+  chatWithGenerate,
+  runtimeContext
 }) {
   const [isRunning, setIsRunning] = React.useState(false);
   const [messages, setMessages] = React.useState([]);
   const [currentThreadId, setCurrentThreadId] = React.useState(threadId);
   const { frequencyPenalty, presencePenalty, maxRetries, maxSteps, maxTokens, temperature, topK, topP, instructions } = modelSettings;
+  const runtimeContextInstance = new di.RuntimeContext();
+  Object.entries(runtimeContext ?? {}).forEach(([key, value]) => {
+    runtimeContextInstance.set(key, value);
+  });
   React.useEffect(() => {
     const hasNewInitialMessages = initialMessages && initialMessages?.length > messages?.length;
     if (messages.length === 0 || currentThreadId !== threadId || hasNewInitialMessages && currentThreadId === threadId) {
@@ -4112,6 +4121,7 @@ function MastraRuntimeProvider({
           topK,
           topP,
           instructions,
+          runtimeContext: runtimeContextInstance,
           ...memory ? { threadId, resourceId: agentId } : {}
         });
         if (generateResponse.response) {
@@ -4220,6 +4230,7 @@ function MastraRuntimeProvider({
           topK,
           topP,
           instructions,
+          runtimeContext: runtimeContextInstance,
           ...memory ? { threadId, resourceId: agentId } : {}
         });
         if (!response.body) {
@@ -4365,6 +4376,18 @@ function AgentProvider({ children }) {
   );
 }
 
+const usePlaygroundStore = zustand.create()(
+  middleware.persist(
+    (set) => ({
+      runtimeContext: {},
+      setRuntimeContext: (runtimeContext) => set({ runtimeContext })
+    }),
+    {
+      name: "mastra-playground-store"
+    }
+  )
+);
+
 const AgentChat = ({
   agentId,
   agentName,
@@ -4375,6 +4398,7 @@ const AgentChat = ({
   refreshThreadList
 }) => {
   const { modelSettings, chatWithGenerate } = React.useContext(AgentContext);
+  const { runtimeContext } = usePlaygroundStore();
   return /* @__PURE__ */ jsxRuntime.jsx(
     MastraRuntimeProvider,
     {
@@ -4387,6 +4411,7 @@ const AgentChat = ({
       refreshThreadList,
       modelSettings,
       chatWithGenerate,
+      runtimeContext,
       children: /* @__PURE__ */ jsxRuntime.jsx("div", { className: "h-full pb-4", children: /* @__PURE__ */ jsxRuntime.jsx(Thread, { agentName: agentName ?? "" }) })
     }
   );
@@ -6222,11 +6247,16 @@ const useExecuteWorkflow = (baseUrl) => {
   const startVNextWorkflowRun = async ({
     workflowId,
     runId,
-    input
+    input,
+    runtimeContext: playgroundRuntimeContext
   }) => {
     try {
+      const runtimeContext$1 = new runtimeContext.RuntimeContext();
+      Object.entries(playgroundRuntimeContext).forEach(([key, value]) => {
+        runtimeContext$1.set(key, value);
+      });
       const workflow = client.getVNextWorkflow(workflowId);
-      await workflow.start({ runId, inputData: input || {} });
+      await workflow.start({ runId, inputData: input || {}, runtimeContext: runtimeContext$1 });
     } catch (error) {
       console.error("Error starting workflow run:", error);
       throw error;
@@ -6235,11 +6265,16 @@ const useExecuteWorkflow = (baseUrl) => {
   const startAsyncVNextWorkflowRun = async ({
     workflowId,
     runId,
-    input
+    input,
+    runtimeContext: playgroundRuntimeContext
   }) => {
     try {
+      const runtimeContext$1 = new runtimeContext.RuntimeContext();
+      Object.entries(playgroundRuntimeContext).forEach(([key, value]) => {
+        runtimeContext$1.set(key, value);
+      });
       const workflow = client.getVNextWorkflow(workflowId);
-      const result = await workflow.startAsync({ runId, inputData: input || {} });
+      const result = await workflow.startAsync({ runId, inputData: input || {}, runtimeContext: runtimeContext$1 });
       return result;
     } catch (error) {
       console.error("Error starting workflow run:", error);
@@ -6367,12 +6402,16 @@ const useResumeWorkflow = (baseUrl) => {
     step,
     runId,
     resumeData,
-    runtimeContext
+    runtimeContext: playgroundRuntimeContext
   }) => {
     try {
       setIsResumingVNextWorkflow(true);
       const client = createMastraClient(baseUrl);
-      const response = await client.getVNextWorkflow(workflowId).resume({ step, runId, resumeData, runtimeContext });
+      const runtimeContext$1 = new runtimeContext.RuntimeContext();
+      Object.entries(playgroundRuntimeContext).forEach(([key, value]) => {
+        runtimeContext$1.set(key, value);
+      });
+      const response = await client.getVNextWorkflow(workflowId).resume({ step, runId, resumeData, runtimeContext: runtimeContext$1 });
       return response;
     } catch (error) {
       console.error("Error resuming workflow:", error);
@@ -8558,6 +8597,7 @@ function VNextWorkflowTrigger({
   baseUrl,
   setRunId
 }) {
+  const { runtimeContext } = usePlaygroundStore();
   const { vNextResult, setVNextResult, payload, setPayload } = React.useContext(WorkflowRunContext);
   const { isLoading, vNextWorkflow } = useVNextWorkflow(workflowId, baseUrl);
   const { createVNextWorkflowRun, startVNextWorkflowRun } = useExecuteWorkflow(baseUrl);
@@ -8574,7 +8614,7 @@ function VNextWorkflowTrigger({
       const { runId } = await createVNextWorkflowRun({ workflowId });
       setRunId?.(runId);
       watchVNextWorkflow({ workflowId, runId });
-      startVNextWorkflowRun({ workflowId, runId, input: data });
+      startVNextWorkflowRun({ workflowId, runId, input: data, runtimeContext });
     } catch (err) {
       setIsRunning(false);
       sonner.toast.error("Error executing workflow");
@@ -8589,7 +8629,8 @@ function VNextWorkflowTrigger({
       step: stepId,
       runId,
       resumeData,
-      workflowId
+      workflowId,
+      runtimeContext
     });
   };
   const watchResultToUse = vNextResult ?? watchVNextResult;
@@ -9141,6 +9182,7 @@ exports.WorkflowRunProvider = WorkflowRunProvider;
 exports.WorkflowTraces = WorkflowTraces;
 exports.WorkflowTrigger = WorkflowTrigger;
 exports.refineTraces = refineTraces;
+exports.usePlaygroundStore = usePlaygroundStore;
 exports.usePolling = usePolling;
 exports.useTraces = useTraces;
 //# sourceMappingURL=index.cjs.js.map