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

package/dist/index.d.ts

@@ -1,22 +1 @@
-
-export * from './domains/agents/index';
-export * from './domains/networks/index';
-export * from './domains/workflows/index';
-export * from './domains/resizable-panel';
-export * from './components/dynamic-form/index';
-export * from './components/ui/data-table';
-export * from './components/threads';
-export * from './types';
-export * from './ds/components/Badge/index';
-export * from './ds/components/Button/index';
-export * from './ds/components/Breadcrumb/index';
-export * from './ds/components/Header/index';
-export * from './ds/components/Logo/index';
-export * from './ds/components/Table/index';
-export * from './ds/components/Txt/index';
-export * from './ds/icons/index';
-export * from './lib/polls';
-export { useTraces } from './hooks/index';
-export { TraceContext, TraceProvider } from './domains/traces/context/trace-context';
-export type { TraceContextType } from './domains/traces/context/trace-context';
-export { refineTraces } from './domains/traces/utils';
+export * from './src/index'

package/dist/index.es.js

@@ -15,7 +15,10 @@ import { jsonLanguage } from '@codemirror/lang-json';
 import { tags } from '@lezer/highlight';
 import { draculaInit } from '@uiw/codemirror-theme-dracula';
 import CodeMirror from '@uiw/react-codemirror';
+import { RuntimeContext } from '@mastra/core/di';
 import { MastraClient } from '@mastra/client-js';
+import { create } from 'zustand';
+import { persist } from 'zustand/middleware';
 import { format, formatDistanceToNow, isValid } from 'date-fns';
 import * as TabsPrimitive from '@radix-ui/react-tabs';
 import { toast } from 'sonner';
@@ -23,6 +26,7 @@ import { AnimatePresence } from 'motion/react';
 import { processDataStream } from '@ai-sdk/ui-utils';
 import Markdown from 'react-markdown';
 import { useDebouncedCallback } from 'use-debounce';
+import { RuntimeContext as RuntimeContext$1 } from '@mastra/core/runtime-context';
 import { MarkerType, Handle, Position, useNodesState, useEdgesState, ReactFlow, Controls, MiniMap, Background, BackgroundVariant, ReactFlowProvider } from '@xyflow/react';
 import '@xyflow/react/dist/style.css';
 import Dagre from '@dagrejs/dagre';
@@ -4021,12 +4025,17 @@ function MastraRuntimeProvider({
   baseUrl,
   refreshThreadList,
   modelSettings = {},
-  chatWithGenerate
+  chatWithGenerate,
+  runtimeContext
 }) {
   const [isRunning, setIsRunning] = useState(false);
   const [messages, setMessages] = useState([]);
   const [currentThreadId, setCurrentThreadId] = useState(threadId);
   const { frequencyPenalty, presencePenalty, maxRetries, maxSteps, maxTokens, temperature, topK, topP, instructions } = modelSettings;
+  const runtimeContextInstance = new RuntimeContext();
+  Object.entries(runtimeContext ?? {}).forEach(([key, value]) => {
+    runtimeContextInstance.set(key, value);
+  });
   useEffect(() => {
     const hasNewInitialMessages = initialMessages && initialMessages?.length > messages?.length;
     if (messages.length === 0 || currentThreadId !== threadId || hasNewInitialMessages && currentThreadId === threadId) {
@@ -4080,6 +4089,7 @@ function MastraRuntimeProvider({
           topK,
           topP,
           instructions,
+          runtimeContext: runtimeContextInstance,
           ...memory ? { threadId, resourceId: agentId } : {}
         });
         if (generateResponse.response) {
@@ -4188,6 +4198,7 @@ function MastraRuntimeProvider({
           topK,
           topP,
           instructions,
+          runtimeContext: runtimeContextInstance,
           ...memory ? { threadId, resourceId: agentId } : {}
         });
         if (!response.body) {
@@ -4333,6 +4344,18 @@ function AgentProvider({ children }) {
   );
 }
 
+const usePlaygroundStore = create()(
+  persist(
+    (set) => ({
+      runtimeContext: {},
+      setRuntimeContext: (runtimeContext) => set({ runtimeContext })
+    }),
+    {
+      name: "mastra-playground-store"
+    }
+  )
+);
+
 const AgentChat = ({
   agentId,
   agentName,
@@ -4343,6 +4366,7 @@ const AgentChat = ({
   refreshThreadList
 }) => {
   const { modelSettings, chatWithGenerate } = useContext(AgentContext);
+  const { runtimeContext } = usePlaygroundStore();
   return /* @__PURE__ */ jsx(
     MastraRuntimeProvider,
     {
@@ -4355,6 +4379,7 @@ const AgentChat = ({
       refreshThreadList,
       modelSettings,
       chatWithGenerate,
+      runtimeContext,
       children: /* @__PURE__ */ jsx("div", { className: "h-full pb-4", children: /* @__PURE__ */ jsx(Thread, { agentName: agentName ?? "" }) })
     }
   );
@@ -6190,11 +6215,16 @@ const useExecuteWorkflow = (baseUrl) => {
   const startVNextWorkflowRun = async ({
     workflowId,
     runId,
-    input
+    input,
+    runtimeContext: playgroundRuntimeContext
   }) => {
     try {
+      const runtimeContext = new RuntimeContext$1();
+      Object.entries(playgroundRuntimeContext).forEach(([key, value]) => {
+        runtimeContext.set(key, value);
+      });
       const workflow = client.getVNextWorkflow(workflowId);
-      await workflow.start({ runId, inputData: input || {} });
+      await workflow.start({ runId, inputData: input || {}, runtimeContext });
     } catch (error) {
       console.error("Error starting workflow run:", error);
       throw error;
@@ -6203,11 +6233,16 @@ const useExecuteWorkflow = (baseUrl) => {
   const startAsyncVNextWorkflowRun = async ({
     workflowId,
     runId,
-    input
+    input,
+    runtimeContext: playgroundRuntimeContext
   }) => {
     try {
+      const runtimeContext = new RuntimeContext$1();
+      Object.entries(playgroundRuntimeContext).forEach(([key, value]) => {
+        runtimeContext.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 });
       return result;
     } catch (error) {
       console.error("Error starting workflow run:", error);
@@ -6335,11 +6370,15 @@ const useResumeWorkflow = (baseUrl) => {
     step,
     runId,
     resumeData,
-    runtimeContext
+    runtimeContext: playgroundRuntimeContext
   }) => {
     try {
       setIsResumingVNextWorkflow(true);
       const client = createMastraClient(baseUrl);
+      const runtimeContext = new RuntimeContext$1();
+      Object.entries(playgroundRuntimeContext).forEach(([key, value]) => {
+        runtimeContext.set(key, value);
+      });
       const response = await client.getVNextWorkflow(workflowId).resume({ step, runId, resumeData, runtimeContext });
       return response;
     } catch (error) {
@@ -8526,6 +8565,7 @@ function VNextWorkflowTrigger({
   baseUrl,
   setRunId
 }) {
+  const { runtimeContext } = usePlaygroundStore();
   const { vNextResult, setVNextResult, payload, setPayload } = useContext(WorkflowRunContext);
   const { isLoading, vNextWorkflow } = useVNextWorkflow(workflowId, baseUrl);
   const { createVNextWorkflowRun, startVNextWorkflowRun } = useExecuteWorkflow(baseUrl);
@@ -8542,7 +8582,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);
       toast.error("Error executing workflow");
@@ -8557,7 +8597,8 @@ function VNextWorkflowTrigger({
       step: stepId,
       runId,
       resumeData,
-      workflowId
+      workflowId,
+      runtimeContext
     });
   };
   const watchResultToUse = vNextResult ?? watchVNextResult;
@@ -9024,5 +9065,5 @@ const useTraces = (componentName, baseUrl, isWorkflow = false) => {
   return { traces, firstCallLoading, error };
 };
 
-export { AgentChat, AgentCoinIcon, AgentContext, AgentEvals, AgentIcon, AgentProvider, AgentTraces, AiIcon, ApiIcon, Badge$1 as Badge, BranchIcon, Breadcrumb, Button, Cell, CheckIcon, ChevronIcon, CommitIcon, CrossIcon, Crumb, DarkLogo, DataTable, DateTimeCell, DbIcon, DebugIcon, DeploymentIcon, DividerIcon, DocsIcon, DynamicForm, EntryCell, EnvIcon, EvaluatorCoinIcon, FiltersIcon, GithubCoinIcon, GithubIcon, GoogleIcon, Header, HeaderAction, HeaderGroup, HeaderTitle, HomeIcon, Icon, InfoIcon, JudgeIcon, LatencyIcon, LogsIcon, MastraResizablePanel, MemoryIcon, NetworkChat, NetworkContext, NetworkProvider, OpenAIIcon, PromptIcon, RepoIcon, Row, ScoreIcon, SettingsIcon, SlashIcon, Table, Tbody, Th, Thead, ThreadDeleteButton, ThreadItem, ThreadLink, ThreadList, Threads, ToolsIcon, TraceContext, TraceIcon, TraceProvider, TsIcon, Txt, TxtCell, UnitCell, VNextWorkflowGraph, VNextWorkflowTrigger, VariablesIcon, WorkflowCoinIcon, WorkflowGraph, WorkflowIcon, WorkflowRunContext, WorkflowRunProvider, WorkflowTraces, WorkflowTrigger, refineTraces, usePolling, useTraces };
+export { AgentChat, AgentCoinIcon, AgentContext, AgentEvals, AgentIcon, AgentProvider, AgentTraces, AiIcon, ApiIcon, Badge$1 as Badge, BranchIcon, Breadcrumb, Button, Cell, CheckIcon, ChevronIcon, CommitIcon, CrossIcon, Crumb, DarkLogo, DataTable, DateTimeCell, DbIcon, DebugIcon, DeploymentIcon, DividerIcon, DocsIcon, DynamicForm, EntryCell, EnvIcon, EvaluatorCoinIcon, FiltersIcon, GithubCoinIcon, GithubIcon, GoogleIcon, Header, HeaderAction, HeaderGroup, HeaderTitle, HomeIcon, Icon, InfoIcon, JudgeIcon, LatencyIcon, LogsIcon, MastraResizablePanel, MemoryIcon, NetworkChat, NetworkContext, NetworkProvider, OpenAIIcon, PromptIcon, RepoIcon, Row, ScoreIcon, SettingsIcon, SlashIcon, Table, Tbody, Th, Thead, ThreadDeleteButton, ThreadItem, ThreadLink, ThreadList, Threads, ToolsIcon, TraceContext, TraceIcon, TraceProvider, TsIcon, Txt, TxtCell, UnitCell, VNextWorkflowGraph, VNextWorkflowTrigger, VariablesIcon, WorkflowCoinIcon, WorkflowGraph, WorkflowIcon, WorkflowRunContext, WorkflowRunProvider, WorkflowTraces, WorkflowTrigger, refineTraces, usePlaygroundStore, usePolling, useTraces };
 //# sourceMappingURL=index.es.js.map