@mastra/playground-ui 6.8.0 → 6.9.0-alpha.0

package/dist/index.es.js

@@ -8711,9 +8711,10 @@ const useAgentMessages = ({
   memory
 }) => {
   const client = useMastraClient();
+  const { runtimeContext } = usePlaygroundStore();
   return useQuery({
-    queryKey: ["memory", "messages", threadId, agentId],
-    queryFn: () => client.getThreadMessages(threadId, { agentId }),
+    queryKey: ["memory", "messages", threadId, agentId, "runtimeContext"],
+    queryFn: () => client.getThreadMessages(threadId, { agentId, runtimeContext }),
     enabled: memory && Boolean(threadId),
     staleTime: 0,
     gcTime: 0,
@@ -10111,6 +10112,7 @@ function useAgentWorkingMemory(agentId, threadId, resourceId) {
   const [workingMemoryFormat, setWorkingMemoryFormat] = useState("markdown");
   const [isLoading, setIsLoading] = useState(true);
   const [isUpdating, setIsUpdating] = useState(false);
+  const { runtimeContext } = usePlaygroundStore();
   const refetch = useCallback(async () => {
     setIsLoading(true);
     try {
@@ -10119,7 +10121,7 @@ function useAgentWorkingMemory(agentId, threadId, resourceId) {
         setIsLoading(false);
         return;
       }
-      const res = await client.getWorkingMemory({ agentId, threadId, resourceId });
+      const res = await client.getWorkingMemory({ agentId, threadId, resourceId, runtimeContext });
       const { workingMemory, source, workingMemoryTemplate, threadExists: threadExists2 } = res;
       setThreadExists(threadExists2);
       setWorkingMemoryData(workingMemory);
@@ -10158,7 +10160,7 @@ function useAgentWorkingMemory(agentId, threadId, resourceId) {
           throw new Error("Invalid JSON working memory");
         }
       }
-      await client.updateWorkingMemory({ agentId, threadId, workingMemory: newMemory, resourceId });
+      await client.updateWorkingMemory({ agentId, threadId, workingMemory: newMemory, resourceId, runtimeContext });
       refetch();
     } catch (error) {
       console.error("Error updating working memory", error);
@@ -14105,6 +14107,7 @@ const AgentMetadataModelSwitcher = ({
   defaultProvider,
   defaultModel,
   updateModel,
+  resetModel,
   apiUrl = "/api/agents/providers"
 }) => {
   const [originalProvider] = useState(defaultProvider);
@@ -14122,6 +14125,10 @@ const AgentMetadataModelSwitcher = ({
   const [providersLoading, setProvidersLoading] = useState(true);
   const [highlightedProviderIndex, setHighlightedProviderIndex] = useState(-1);
   const [highlightedModelIndex, setHighlightedModelIndex] = useState(-1);
+  useEffect(() => {
+    setSelectedModel(defaultModel);
+    setSelectedProvider(defaultProvider || "");
+  }, [defaultModel, defaultProvider]);
   const modelInputRef = useRef(null);
   const providerInputRef = useRef(null);
   useEffect(() => {
@@ -14275,8 +14282,10 @@ const AgentMetadataModelSwitcher = ({
     ] });
   }
   const handleReset = async () => {
-    setSelectedProvider(cleanProviderId(originalProvider));
-    setSelectedModel(originalModel);
+    if (!resetModel) {
+      console.warn("Reset model function not provided");
+      return;
+    }
     setProviderSearch("");
     setModelSearch("");
     setIsSearchingProvider(false);
@@ -14285,10 +14294,7 @@ const AgentMetadataModelSwitcher = ({
     setShowModelSuggestions(false);
     try {
       setLoading(true);
-      await updateModel({
-        provider: originalProvider,
-        modelId: originalModel
-      });
+      await resetModel();
     } catch (error) {
       console.error("Failed to reset model:", error);
     } finally {
@@ -14696,7 +14702,6 @@ Switch.displayName = SwitchPrimitives.Root.displayName;
 
 const AgentMetadataModelList = ({
   modelList,
-  modelProviders,
   updateModelInModelList,
   reorderModelList
 }) => {
@@ -14734,7 +14739,6 @@ const AgentMetadataModelList = ({
       AgentMetadataModelListItem,
       {
         modelConfig,
-        modelProviders,
         updateModelInModelList: updateModel,
         showDragHandle: hasMultipleModels,
         dragHandleProps: provided2.dragHandleProps
@@ -14745,7 +14749,6 @@ const AgentMetadataModelList = ({
 };
 const AgentMetadataModelListItem = ({
   modelConfig,
-  modelProviders,
   updateModelInModelList,
   showDragHandle,
   dragHandleProps
@@ -14759,7 +14762,6 @@ const AgentMetadataModelListItem = ({
         defaultProvider: modelConfig.model.provider,
         defaultModel: modelConfig.model.modelId,
         updateModel: (params) => updateModelInModelList({ modelConfigId: modelConfig.id, model: params }),
-        modelProviders,
         autoSave: true
       }
     ) }),
@@ -14790,7 +14792,7 @@ const AgentMetadata = ({
   promptSlot,
   hasMemoryEnabled,
   updateModel,
-  modelProviders,
+  resetModel,
   updateModelInModelList,
   reorderModelList,
   modelVersion
@@ -14806,7 +14808,6 @@ const AgentMetadata = ({
       AgentMetadataModelList,
       {
         modelList: agent.modelList,
-        modelProviders,
         updateModelInModelList,
         reorderModelList
       }
@@ -14825,7 +14826,7 @@ const AgentMetadata = ({
             defaultProvider: agent.provider,
             defaultModel: agent.modelId,
             updateModel,
-            modelProviders
+            resetModel
           }
         )
       }
@@ -15155,13 +15156,6 @@ const useAgents = () => {
     queryFn: () => client.getAgents(runtimeContext)
   });
 };
-const useModelProviders = () => {
-  const client = useMastraClient();
-  return useQuery({
-    queryKey: ["model-providers"],
-    queryFn: () => client.getModelProviders()
-  });
-};
 const useUpdateAgentModel = (agentId) => {
   const client = useMastraClient();
   const queryClient = useQueryClient();
@@ -15201,6 +15195,19 @@ const useUpdateModelInModelList = (agentId) => {
     }
   });
 };
+const useResetAgentModel = (agentId) => {
+  const client = useMastraClient();
+  const queryClient = useQueryClient();
+  return useMutation({
+    mutationFn: async () => client.getAgent(agentId).resetModel(),
+    onSuccess: () => {
+      queryClient.invalidateQueries({ queryKey: ["agent", agentId] });
+    },
+    onError: (err) => {
+      console.error("Error resetting model", err);
+    }
+  });
+};
 
 const useAgent = (agentId) => {
   const client = useMastraClient();
@@ -15316,9 +15323,10 @@ const AgentToolPanel = ({ toolId, agentId }) => {
 
 const useMemory = (agentId) => {
   const client = useMastraClient();
+  const { runtimeContext } = usePlaygroundStore();
   return useQuery({
     queryKey: ["memory", agentId],
-    queryFn: () => agentId ? client.getMemoryStatus(agentId) : null,
+    queryFn: () => agentId ? client.getMemoryStatus(agentId, runtimeContext) : null,
     enabled: Boolean(agentId),
     staleTime: 5 * 60 * 1e3,
     // 5 minutes
@@ -15329,9 +15337,10 @@ const useMemory = (agentId) => {
 };
 const useMemoryConfig = (agentId) => {
   const client = useMastraClient();
+  const { runtimeContext } = usePlaygroundStore();
   return useQuery({
     queryKey: ["memory", "config", agentId],
-    queryFn: () => agentId ? client.getMemoryConfig({ agentId }) : null,
+    queryFn: () => agentId ? client.getMemoryConfig({ agentId, runtimeContext }) : null,
     enabled: Boolean(agentId),
     staleTime: 5 * 60 * 1e3,
     // 5 minutes
@@ -15347,9 +15356,10 @@ const useThreads = ({
   isMemoryEnabled
 }) => {
   const client = useMastraClient();
+  const { runtimeContext } = usePlaygroundStore();
   return useQuery({
     queryKey: ["memory", "threads", resourceId, agentId],
-    queryFn: () => isMemoryEnabled ? client.getMemoryThreads({ resourceId, agentId }) : null,
+    queryFn: () => isMemoryEnabled ? client.getMemoryThreads({ resourceId, agentId, runtimeContext }) : null,
     enabled: Boolean(isMemoryEnabled),
     staleTime: 0,
     gcTime: 0,
@@ -15360,8 +15370,9 @@ const useThreads = ({
 const useDeleteThread = () => {
   const client = useMastraClient();
   const queryClient = useQueryClient();
+  const { runtimeContext } = usePlaygroundStore();
   return useMutation({
-    mutationFn: ({ threadId, agentId, networkId }) => client.deleteThread(threadId, { agentId, networkId }),
+    mutationFn: ({ threadId, agentId, networkId }) => client.deleteThread(threadId, { agentId, networkId, runtimeContext }),
     onSuccess: (_, variables) => {
       const { agentId, networkId } = variables;
       if (agentId) {
@@ -15382,6 +15393,7 @@ const useMemorySearch = ({
   resourceId,
   threadId
 }) => {
+  const { runtimeContext } = usePlaygroundStore();
   const searchMemory = async (searchQuery, memoryConfig) => {
     if (!searchQuery.trim()) {
       return { results: [], count: 0, query: searchQuery };
@@ -15389,7 +15401,8 @@ const useMemorySearch = ({
     const params = new URLSearchParams({
       searchQuery,
       resourceId,
-      agentId
+      agentId,
+      runtimeContext: btoa(JSON.stringify(runtimeContext))
     });
     if (threadId) {
       params.append("threadId", threadId);
@@ -16749,8 +16762,8 @@ function AgentPromptEnhancer({ agentId }) {
 
 function AgentInformation({ agentId, threadId }) {
   const { data: agent, isLoading } = useAgent(agentId);
-  const { data: modelProviders } = useModelProviders();
   const { mutateAsync: updateModel } = useUpdateAgentModel(agentId);
+  const { mutateAsync: resetModel } = useResetAgentModel(agentId);
   const { mutate: reorderModelList } = useReorderModelList(agentId);
   const { mutateAsync: updateModelInModelList } = useUpdateModelInModelList(agentId);
   const { data: memory, isLoading: isMemoryLoading } = useMemory(agentId);
@@ -16795,9 +16808,9 @@ function AgentInformation({ agentId, threadId }) {
             agentId,
             agent,
             updateModel,
+            resetModel,
             updateModelInModelList,
             reorderModelList,
-            modelProviders: modelProviders || [],
             hasMemoryEnabled: Boolean(memory?.result),
             promptSlot: /* @__PURE__ */ jsx(AgentPromptEnhancer, { agentId }),
             modelVersion: agent.modelVersion
@@ -17377,7 +17390,6 @@ function TemplateForm({
             updateModel: onModelUpdate || (() => Promise.resolve({ message: "Updated" })),
             closeEditor: () => {
             },
-            modelProviders: ["openai", "anthropic", "google", "xai", "groq"],
             autoSave: true,
             selectProviderPlaceholder: "Provider"
           }
@@ -20441,5 +20453,5 @@ const useEvalsByAgentId = (agentId, type) => {
   });
 };
 
-export { AgentChat, AgentCoinIcon, AgentEntityHeader, AgentEvals, AgentIcon, AgentInformation, AgentMetadata, AgentMetadataList, AgentMetadataListEmpty, AgentMetadataListItem, AgentMetadataNetworkList, AgentMetadataPrompt, AgentMetadataScorerList, AgentMetadataSection, AgentMetadataToolList, AgentMetadataWorkflowList, AgentMetadataWrapper, AgentNetworkCoinIcon, AgentSettings, AgentSettingsContext, AgentSettingsProvider, AgentToolPanel, AgentsTable, AiIcon, Alert$1 as Alert, AlertDescription$1 as AlertDescription, AlertDialog, AlertTitle$1 as AlertTitle, ApiIcon, Badge$1 as Badge, BranchIcon, Breadcrumb, Button$1 as Button, ButtonsGroup, Cell, ChatThreads, CheckIcon, ChevronIcon, Collapsible, CollapsibleContent, CollapsibleTrigger, CommitIcon, CrossIcon, Crumb, DarkLogo, DateTimeCell, DateTimePicker, DateTimePickerContent, DbIcon, DebugIcon, DefaultTrigger, DeploymentIcon, DividerIcon, DocsIcon, DynamicForm, EmptyState, Entity, EntityContent, EntityDescription, EntityHeader, EntityIcon, EntityMainHeader, EntityName, Entry, EntryCell, EntryList, EntryListSkeleton, EnvIcon, EvaluatorCoinIcon, FiltersIcon, FolderIcon, FormActions, GithubCoinIcon, GithubIcon, GoogleIcon, Header, HeaderAction, HeaderGroup, HeaderTitle, HomeIcon, Icon, InfoIcon, InputField, JudgeIcon, Kbd, KeyValueList, LatencyIcon, LinkComponentProvider, LogsIcon, MCPDetail, MCPTable, MCPToolPanel, MainContentContent, MainContentLayout, MainSidebar, MainSidebarProvider, MastraResizablePanel, McpCoinIcon, McpServerIcon, MemoryIcon, MemorySearch, ModelResetProvider, Notification, OpenAIIcon, PageHeader, PlaygroundQueryClient, PlaygroundTabs, PromptIcon, RadioGroup, RadioGroupField, RadioGroupItem, RepoIcon, Row, RuntimeContext, RuntimeContextWrapper, ScoreDialog, ScoreIcon, ScorersTable, ScoresList, ScoresTools, SearchField, Searchbar, SearchbarWrapper, Section, Sections, SelectField, SettingsIcon, SideDialog, SlashIcon, SliderField, SpanScoreList, SpanScoring, SpanTabs, Tab$1 as Tab, TabContent$1 as TabContent, TabList$1 as TabList, Table$1 as Table, Tbody, TemplateFailure, TemplateForm, TemplateInfo, TemplateInstallation, TemplateSuccess, TemplatesList, TemplatesTools, TextAndIcon, TextareaField, Th, Thead, ThreadDeleteButton, ThreadInputProvider, ThreadItem, ThreadLink, ThreadList, Threads, ToolCoinIcon, ToolFallback, ToolIconMap, ToolInformation, ToolPanel, ToolTable, ToolsIcon, Tooltip, TooltipContent, TooltipProvider, TooltipTrigger, TraceDialog, TraceIcon, TraceTimeline, TraceTimelineLegend, TraceTimelineSpan, TracesList, TracesTools, TracesView, TracesViewSkeleton, TsIcon, Txt, TxtCell, UnitCell, VariablesIcon, WorkflowCoinIcon, WorkflowGraph, WorkflowIcon, WorkflowRunContext, WorkflowRunDetail, WorkflowRunList, WorkflowRunProvider, WorkflowTable, WorkflowTrigger, WorkingMemoryContext, WorkingMemoryProvider, allowedAiSpanAttributes, cleanString, convertWorkflowRunStateToStreamResult, extractPrompt, formatDuration, formatHierarchicalSpans, formatOtelTimestamp, formatOtelTimestamp2, getColumnTemplate, getShortId, getSpanTypeUi, getToNextEntryFn, getToPreviousEntryFn, parseError, providerMapToIcon, scoresListColumns, spanTypePrefixes, toast, traceScoresListColumns, tracesListColumns, transformKey, useAgent, useAgentSettings, useAgents, useCancelWorkflowRun, useCurrentRun, useDeleteThread, useEvalsByAgentId, useExecuteAgentTool, useExecuteMCPTool, useExecuteTool, useExecuteWorkflow, useInView, useLinkComponent, useMCPServerTool, useMCPServerTools, useMCPServers, useMainSidebar, useMemory, useMemoryConfig, useMemorySearch, useModelProviders, useModelReset, usePlaygroundStore, usePolling, useReorderModelList, useScorer, useScorers, useScoresByEntityId, useScoresByScorerId, useSendWorkflowRunEvent, useSpeechRecognition, useStreamWorkflow, useThreadInput, useThreads, useTool, useTools, useTraceSpanScores, useUpdateAgentModel, useUpdateModelInModelList, useWorkflow, useWorkflowRuns, useWorkflows, useWorkingMemory };
+export { AgentChat, AgentCoinIcon, AgentEntityHeader, AgentEvals, AgentIcon, AgentInformation, AgentMetadata, AgentMetadataList, AgentMetadataListEmpty, AgentMetadataListItem, AgentMetadataNetworkList, AgentMetadataPrompt, AgentMetadataScorerList, AgentMetadataSection, AgentMetadataToolList, AgentMetadataWorkflowList, AgentMetadataWrapper, AgentNetworkCoinIcon, AgentSettings, AgentSettingsContext, AgentSettingsProvider, AgentToolPanel, AgentsTable, AiIcon, Alert$1 as Alert, AlertDescription$1 as AlertDescription, AlertDialog, AlertTitle$1 as AlertTitle, ApiIcon, Badge$1 as Badge, BranchIcon, Breadcrumb, Button$1 as Button, ButtonsGroup, Cell, ChatThreads, CheckIcon, ChevronIcon, Collapsible, CollapsibleContent, CollapsibleTrigger, CommitIcon, CrossIcon, Crumb, DarkLogo, DateTimeCell, DateTimePicker, DateTimePickerContent, DbIcon, DebugIcon, DefaultTrigger, DeploymentIcon, DividerIcon, DocsIcon, DynamicForm, EmptyState, Entity, EntityContent, EntityDescription, EntityHeader, EntityIcon, EntityMainHeader, EntityName, Entry, EntryCell, EntryList, EntryListSkeleton, EnvIcon, EvaluatorCoinIcon, FiltersIcon, FolderIcon, FormActions, GithubCoinIcon, GithubIcon, GoogleIcon, Header, HeaderAction, HeaderGroup, HeaderTitle, HomeIcon, Icon, InfoIcon, InputField, JudgeIcon, Kbd, KeyValueList, LatencyIcon, LinkComponentProvider, LogsIcon, MCPDetail, MCPTable, MCPToolPanel, MainContentContent, MainContentLayout, MainSidebar, MainSidebarProvider, MastraResizablePanel, McpCoinIcon, McpServerIcon, MemoryIcon, MemorySearch, ModelResetProvider, Notification, OpenAIIcon, PageHeader, PlaygroundQueryClient, PlaygroundTabs, PromptIcon, RadioGroup, RadioGroupField, RadioGroupItem, RepoIcon, Row, RuntimeContext, RuntimeContextWrapper, ScoreDialog, ScoreIcon, ScorersTable, ScoresList, ScoresTools, SearchField, Searchbar, SearchbarWrapper, Section, Sections, SelectField, SettingsIcon, SideDialog, SlashIcon, SliderField, SpanScoreList, SpanScoring, SpanTabs, Tab$1 as Tab, TabContent$1 as TabContent, TabList$1 as TabList, Table$1 as Table, Tbody, TemplateFailure, TemplateForm, TemplateInfo, TemplateInstallation, TemplateSuccess, TemplatesList, TemplatesTools, TextAndIcon, TextareaField, Th, Thead, ThreadDeleteButton, ThreadInputProvider, ThreadItem, ThreadLink, ThreadList, Threads, ToolCoinIcon, ToolFallback, ToolIconMap, ToolInformation, ToolPanel, ToolTable, ToolsIcon, Tooltip, TooltipContent, TooltipProvider, TooltipTrigger, TraceDialog, TraceIcon, TraceTimeline, TraceTimelineLegend, TraceTimelineSpan, TracesList, TracesTools, TracesView, TracesViewSkeleton, TsIcon, Txt, TxtCell, UnitCell, VariablesIcon, WorkflowCoinIcon, WorkflowGraph, WorkflowIcon, WorkflowRunContext, WorkflowRunDetail, WorkflowRunList, WorkflowRunProvider, WorkflowTable, WorkflowTrigger, WorkingMemoryContext, WorkingMemoryProvider, allowedAiSpanAttributes, cleanString, convertWorkflowRunStateToStreamResult, extractPrompt, formatDuration, formatHierarchicalSpans, formatOtelTimestamp, formatOtelTimestamp2, getColumnTemplate, getShortId, getSpanTypeUi, getToNextEntryFn, getToPreviousEntryFn, parseError, providerMapToIcon, scoresListColumns, spanTypePrefixes, toast, traceScoresListColumns, tracesListColumns, transformKey, useAgent, useAgentSettings, useAgents, useCancelWorkflowRun, useCurrentRun, useDeleteThread, useEvalsByAgentId, useExecuteAgentTool, useExecuteMCPTool, useExecuteTool, useExecuteWorkflow, useInView, useLinkComponent, useMCPServerTool, useMCPServerTools, useMCPServers, useMainSidebar, useMemory, useMemoryConfig, useMemorySearch, useModelReset, usePlaygroundStore, usePolling, useReorderModelList, useResetAgentModel, useScorer, useScorers, useScoresByEntityId, useScoresByScorerId, useSendWorkflowRunEvent, useSpeechRecognition, useStreamWorkflow, useThreadInput, useThreads, useTool, useTools, useTraceSpanScores, useUpdateAgentModel, useUpdateModelInModelList, useWorkflow, useWorkflowRuns, useWorkflows, useWorkingMemory };
 //# sourceMappingURL=index.es.js.map