@mastra/react 0.2.9-alpha.0 → 0.2.10-alpha.0

package/dist/index.js

@@ -1,14 +1,12 @@
 import { MastraClient } from '@mastra/client-js';
 import { createContext, useContext, useRef, useState, useEffect, useLayoutEffect, useCallback, Fragment } from 'react';
-import { jsxs, jsx } from 'react/jsx-runtime';
+import { jsx, jsxs } from 'react/jsx-runtime';
 import { v4 } from '@lukeed/uuid';
-import { formatStreamCompletionFeedback, formatCompletionFeedback } from '@mastra/core/loop';
 import { ChevronDownIcon, CheckIcon, CopyIcon } from 'lucide-react';
 import { twMerge } from 'tailwind-merge';
 import { TooltipProvider, Root, TooltipPortal, TooltipContent as TooltipContent$1, TooltipTrigger as TooltipTrigger$1 } from '@radix-ui/react-tooltip';
 import { toJsxRuntime } from 'hast-util-to-jsx-runtime';
 import { codeToHast } from 'shiki/bundle/web';
-import { RequestContext } from '@mastra/core/request-context';
 
 // src/mastra-client-context.tsx
 var MastraClientContext = createContext({});
@@ -42,6 +40,53 @@ var createMastraClient = (baseUrl, mastraClientHeaders = {}, apiPrefix) => {
 var MastraReactProvider = ({ children, baseUrl, headers, apiPrefix }) => {
   return /* @__PURE__ */ jsx(MastraClientProvider, { baseUrl, headers, apiPrefix, children });
 };
+
+// src/lib/ai-sdk/utils/formatCompletionFeedback.ts
+var formatBaseCompletionFeedback = (result, maxIterationReached, formatScorerHeading, incompleteMessage) => {
+  const lines = [];
+  lines.push("#### Completion Check Results");
+  lines.push("");
+  lines.push(`Overall: ${result.complete ? "\u2705 COMPLETE" : "\u274C NOT COMPLETE"}`);
+  lines.push(`Duration: ${result.totalDuration}ms`);
+  if (result.timedOut) {
+    lines.push("\u26A0\uFE0F Scoring timed out");
+  }
+  lines.push("");
+  for (const scorer of result.scorers) {
+    lines.push(formatScorerHeading(scorer));
+    lines.push(`Score: ${scorer.score} ${scorer.passed ? "\u2705" : "\u274C"}`);
+    if (scorer.reason) {
+      lines.push(`Reason: ${scorer.reason}`);
+    }
+    lines.push("");
+  }
+  if (result.complete) {
+    lines.push("\u2705 The task is complete.");
+  } else if (maxIterationReached) {
+    lines.push("\u26A0\uFE0F Max iterations reached.");
+  } else {
+    lines.push(incompleteMessage);
+  }
+  return lines.join("\n");
+};
+var formatCompletionFeedback = (result, maxIterationReached) => {
+  return formatBaseCompletionFeedback(
+    result,
+    maxIterationReached,
+    (scorer) => `###### ${scorer.scorerName} (${scorer.scorerId})`,
+    "\u{1F504} Will continue working on the task."
+  );
+};
+var formatStreamCompletionFeedback = (result, maxIterationReached) => {
+  return formatBaseCompletionFeedback(
+    result,
+    maxIterationReached,
+    (scorer) => `**${scorer.scorerName}** (${scorer.scorerId})`,
+    "\u{1F504} The task is not yet complete. Please continue working based on the feedback above."
+  );
+};
+
+// src/lib/ai-sdk/utils/toUIMessage.ts
 var mapWorkflowStreamChunkToWatchResult = (prev, chunk) => {
   if (chunk.type === "workflow-start") {
     return {
@@ -459,9 +504,7 @@ var toUIMessage = ({ chunk, conversation, metadata }) => {
           complete: chunk.payload.passed,
           scorers: chunk.payload.results,
           totalDuration: chunk.payload.duration,
-          timedOut: chunk.payload.timedOut,
-          completionReason: chunk.payload.reason
-        },
+          timedOut: chunk.payload.timedOut},
         chunk.payload.maxIterationReached
       );
       const newMessage = {
@@ -1035,6 +1078,8 @@ var resolveToChildMessages = (messages) => {
   }
   return childMessages;
 };
+
+// src/lib/ai-sdk/transformers/AISdkNetworkTransformer.ts
 var AISdkNetworkTransformer = class {
   transform({ chunk, conversation, metadata }) {
     const newConversation = [...conversation];
@@ -1057,9 +1102,7 @@ var AISdkNetworkTransformer = class {
           complete: chunk.payload.passed,
           scorers: chunk.payload.results,
           totalDuration: chunk.payload.duration,
-          timedOut: chunk.payload.timedOut,
-          completionReason: chunk.payload.reason
-        },
+          timedOut: chunk.payload.timedOut},
         chunk.payload.maxIterationReached
       );
       const newMessage = {
@@ -2406,16 +2449,12 @@ function useStreamWorkflow({ debugMode, tracingOptions, onError }) {
       if (!isMountedRef.current) return;
       setIsStreaming(true);
       setStreamResult({ input: inputData });
-      const requestContext = new RequestContext();
-      Object.entries(playgroundRequestContext).forEach(([key, value]) => {
-        requestContext.set(key, value);
-      });
       const workflow = client.getWorkflow(workflowId);
       const run = await workflow.createRun({ runId });
       const stream = await run.stream({
         inputData,
         initialState,
-        requestContext,
+        requestContext: playgroundRequestContext,
         closeOnSuspend: true,
         tracingOptions,
         perStep: perStep ?? debugMode
@@ -2521,15 +2560,11 @@ function useStreamWorkflow({ debugMode, tracingOptions, onError }) {
       if (!isMountedRef.current) return;
       setIsStreaming(true);
       const workflow = client.getWorkflow(workflowId);
-      const requestContext = new RequestContext();
-      Object.entries(playgroundRequestContext).forEach(([key, value]) => {
-        requestContext.set(key, value);
-      });
       const run = await workflow.createRun({ runId });
       const stream = await run.resumeStream({
         step,
         resumeData,
-        requestContext,
+        requestContext: playgroundRequestContext,
         tracingOptions,
         perStep: perStep ?? debugMode
       });
@@ -2580,15 +2615,11 @@ function useStreamWorkflow({ debugMode, tracingOptions, onError }) {
       if (!isMountedRef.current) return;
       setIsStreaming(true);
       const workflow = client.getWorkflow(workflowId);
-      const requestContext = new RequestContext();
-      Object.entries(playgroundRequestContext).forEach(([key, value]) => {
-        requestContext.set(key, value);
-      });
       const run = await workflow.createRun({ runId });
       const stream = await run.timeTravelStream({
         ...params,
         perStep: perStep ?? debugMode,
-        requestContext,
+        requestContext: playgroundRequestContext,
         tracingOptions
       });
       if (!stream) {