@mastra/react 1.1.2-alpha.5 → 1.2.0-alpha.6

package/dist/index.d.ts

@@ -4,6 +4,7 @@ export * from './agent/types.js';
 export type { MastraClientCredentials, MastraClientProviderProps } from './mastra-client-context.js';
 export { useMastraClient } from './mastra-client-context.js';
 export * from './lib/mastra-db/index.js';
+export type { TaskItem } from '@mastra/core/signals';
 export type { MastraDBMessage, MastraMessageContentV2, MastraMessagePart, MastraToolApproval, MastraToolInvocation, MastraToolInvocationPart, MessageSource, MemoryInfo, } from '@mastra/core/agent/message-list';
 export * from './ui/index.js';
 export * from './workflows/index.js';

package/dist/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,cAAc,yBAAyB,CAAC;AACxC,cAAc,eAAe,CAAC;AAC9B,cAAc,eAAe,CAAC;AAC9B,YAAY,EAAE,uBAAuB,EAAE,yBAAyB,EAAE,MAAM,yBAAyB,CAAC;AAClG,OAAO,EAAE,eAAe,EAAE,MAAM,yBAAyB,CAAC;AAC1D,cAAc,iBAAiB,CAAC;AAChC,YAAY,EACV,eAAe,EACf,sBAAsB,EACtB,iBAAiB,EACjB,kBAAkB,EAClB,oBAAoB,EACpB,wBAAwB,EACxB,aAAa,EACb,UAAU,GACX,MAAM,iCAAiC,CAAC;AACzC,cAAc,MAAM,CAAC;AACrB,cAAc,aAAa,CAAC;AAC5B,cAAc,SAAS,CAAC"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,cAAc,yBAAyB,CAAC;AACxC,cAAc,eAAe,CAAC;AAC9B,cAAc,eAAe,CAAC;AAC9B,YAAY,EAAE,uBAAuB,EAAE,yBAAyB,EAAE,MAAM,yBAAyB,CAAC;AAClG,OAAO,EAAE,eAAe,EAAE,MAAM,yBAAyB,CAAC;AAC1D,cAAc,iBAAiB,CAAC;AAChC,YAAY,EAAE,QAAQ,EAAE,MAAM,sBAAsB,CAAC;AACrD,YAAY,EACV,eAAe,EACf,sBAAsB,EACtB,iBAAiB,EACjB,kBAAkB,EAClB,oBAAoB,EACpB,wBAAwB,EACxB,aAAa,EACb,UAAU,GACX,MAAM,iCAAiC,CAAC;AACzC,cAAc,MAAM,CAAC;AACrB,cAAc,aAAa,CAAC;AAC5B,cAAc,SAAS,CAAC"}

package/dist/index.js

@@ -1701,12 +1701,90 @@ var newUserMessage = (parts) => ({
 var fromCoreUserMessageToMastraDBMessage = (coreUserMessage) => newUserMessage(coreUserMessageToParts(coreUserMessage));
 var fromCoreUserMessagesToMastraDBMessage = (coreUserMessages) => newUserMessage(coreUserMessages.flatMap(coreUserMessageToParts));
 
+// src/agent/extract-tasks.ts
+var TASK_TOOL_NAMES = /* @__PURE__ */ new Set(["task_write", "task_update", "task_complete", "task_check"]);
+var TASK_SIGNAL_ID = "tasks";
+var TASK_TAG_NAMES = /* @__PURE__ */ new Set(["current-task-list", "task-list-update"]);
+var TASK_STATUSES = /* @__PURE__ */ new Set(["pending", "in_progress", "completed"]);
+var isDataChunk2 = (chunk) => typeof chunk.type === "string" && chunk.type.startsWith("data-");
+var isRecord = (value) => typeof value === "object" && value !== null && !Array.isArray(value);
+var isTaskStatus = (value) => typeof value === "string" && TASK_STATUSES.has(value);
+var isTaskItemArray = (value) => Array.isArray(value) && value.every(
+  (item) => isRecord(item) && typeof item.id === "string" && typeof item.content === "string" && isTaskStatus(item.status) && typeof item.activeForm === "string"
+);
+var isTaskSignal = (value) => {
+  if (!isRecord(value)) return false;
+  return value.id === TASK_SIGNAL_ID || typeof value.tagName === "string" && TASK_TAG_NAMES.has(value.tagName);
+};
+var extractTasksFromSignalData = (data) => {
+  if (!isTaskSignal(data) || !isRecord(data)) return void 0;
+  const metadata = data.metadata;
+  if (!isRecord(metadata)) return void 0;
+  const value = metadata.value;
+  if (!isRecord(value)) return void 0;
+  return isTaskItemArray(value.tasks) ? value.tasks : void 0;
+};
+var parseTasksFromResult = (raw) => {
+  if (isRecord(raw) && isTaskItemArray(raw.tasks)) return raw.tasks;
+  if (typeof raw !== "string") return void 0;
+  try {
+    const parsed = JSON.parse(raw);
+    return isRecord(parsed) && isTaskItemArray(parsed.tasks) ? parsed.tasks : void 0;
+  } catch {
+    return void 0;
+  }
+};
+var extractTasksFromSignalChunk = (chunk) => {
+  if (!isDataChunk2(chunk)) return void 0;
+  return extractTasksFromSignalData(chunk.data);
+};
+var extractTasksFromToolResultChunk = (chunk) => {
+  if (chunk.type !== "tool-result") return void 0;
+  const payload = chunk.payload;
+  if (!isRecord(payload) || typeof payload.toolName !== "string" || !TASK_TOOL_NAMES.has(payload.toolName)) {
+    return void 0;
+  }
+  return parseTasksFromResult(payload.result);
+};
+var extractTasksFromToolInvocationPart = (part) => {
+  if (!isRecord(part) || part.type !== "tool-invocation" || !isRecord(part.toolInvocation)) return void 0;
+  const toolInvocation = part.toolInvocation;
+  if (typeof toolInvocation.toolName !== "string" || !TASK_TOOL_NAMES.has(toolInvocation.toolName)) return void 0;
+  return parseTasksFromResult(toolInvocation.result);
+};
+var extractTasksFromDataPart = (part) => {
+  if (!isRecord(part) || typeof part.type !== "string" || !part.type.startsWith("data-")) return void 0;
+  return extractTasksFromSignalData(part.data);
+};
+var extractTasksFromSignalMessage = (message) => {
+  if (message.role !== "signal") return void 0;
+  const metadata = message.content?.metadata;
+  if (!isRecord(metadata)) return void 0;
+  return extractTasksFromSignalData(metadata.signal);
+};
+var extractLatestTasksFromMessages = (messages) => {
+  let latest;
+  for (const message of messages) {
+    for (const part of message.content?.parts ?? []) {
+      const toolTasks = extractTasksFromToolInvocationPart(part);
+      if (toolTasks !== void 0) latest = toolTasks;
+    }
+    const signalTasks = extractTasksFromSignalMessage(message);
+    if (signalTasks !== void 0) latest = signalTasks;
+    for (const part of message.content?.parts ?? []) {
+      const dataTasks = extractTasksFromDataPart(part);
+      if (dataTasks !== void 0) latest = dataTasks;
+    }
+  }
+  return latest ?? [];
+};
+
 // src/agent/extractRunIdFromMessages.ts
-var isRecord = (value) => value !== null && typeof value === "object";
+var isRecord2 = (value) => value !== null && typeof value === "object";
 var runIdMetadataKeys = ["pendingToolApprovals", "requireApprovalMetadata", "suspendedTools"];
-var isRunIdMetadataSource = (value) => isRecord(value) && Object.values(value).every((entry) => isRecord(entry));
+var isRunIdMetadataSource = (value) => isRecord2(value) && Object.values(value).every((entry) => isRecord2(entry));
 var getRunIdMetadataSources = (metadata) => {
-  if (!isRecord(metadata)) return [];
+  if (!isRecord2(metadata)) return [];
   const sources = [];
   for (const key of runIdMetadataKeys) {
     const source = metadata[key];
@@ -1720,7 +1798,7 @@ var extractRunIdFromMessages = (messages) => {
   for (const message of messages) {
     for (const source of getRunIdMetadataSources(message.content?.metadata)) {
       for (const entry of Object.values(source)) {
-        if (isRecord(entry) && typeof entry.runId === "string" && entry.runId.length > 0) {
+        if (isRecord2(entry) && typeof entry.runId === "string" && entry.runId.length > 0) {
           return entry.runId;
         }
       }
@@ -1825,7 +1903,7 @@ var isThreadSignalUnsupportedError = (error) => {
   }
   return status === 400 && typeof error.message === "string" && error.message.includes("No active agent run found for signal target");
 };
-var isDataChunk2 = (chunk) => typeof chunk.type === "string" && chunk.type.startsWith("data-");
+var isDataChunk3 = (chunk) => typeof chunk.type === "string" && chunk.type.startsWith("data-");
 var dbFromServerUiMessages = (uiMessages, metadata) => uiMessages.map((uiMsg) => {
   const dbMsg = AIV5Adapter.fromUIMessage(uiMsg);
   return {
@@ -1866,6 +1944,7 @@ var useChat = ({
   const _threadSubscriptionPromiseRef = useRef(null);
   const _threadSignalsUnsupportedRef = useRef(false);
   const [messages, setMessages] = useState([]);
+  const [tasks, setTasks] = useState([]);
   const [toolCallApprovals, setToolCallApprovals] = useState({});
   const [networkToolCallApprovals, setNetworkToolCallApprovals] = useState({});
   const pendingToolApprovalIdsRef = useRef(/* @__PURE__ */ new Set());
@@ -1875,6 +1954,7 @@ var useChat = ({
   useEffect(() => {
     const formattedMessages = resolveInitialMessages(initialMessages ?? []);
     setMessages(formattedMessages);
+    setTasks(extractLatestTasksFromMessages(formattedMessages));
     pendingToolApprovalIdsRef.current = extractPendingToolApprovalIdsFromMessages(formattedMessages);
     setIsAwaitingToolApproval(pendingToolApprovalIdsRef.current.size > 0);
     _currentRunId.current = extractRunIdFromMessages(formattedMessages);
@@ -1946,7 +2026,11 @@ var useChat = ({
   const processStreamChunk = useCallback(
     async (chunk, onChunk) => {
       setMessages((prev) => accumulateChunk({ chunk, conversation: prev, metadata: { mode: "stream" } }));
-      if (chunk.type === "data-user-message" && isDataChunk2(chunk) && (chunk.data?.type === "user-message" || chunk.data?.type === "user") && typeof chunk.data?.id === "string") {
+      const signalTasks = extractTasksFromSignalChunk(chunk);
+      if (signalTasks !== void 0) setTasks(signalTasks);
+      const toolTasks = extractTasksFromToolResultChunk(chunk);
+      if (toolTasks !== void 0) setTasks(toolTasks);
+      if (chunk.type === "data-user-message" && isDataChunk3(chunk) && (chunk.data?.type === "user-message" || chunk.data?.type === "user") && typeof chunk.data?.id === "string") {
         onSignalEcho?.(chunk.data.id);
       }
       if (chunk.type === "start") {
@@ -2356,7 +2440,7 @@ var useChat = ({
     _onNetworkChunk.current = void 0;
     _requestContext.current = void 0;
   };
-  const approveToolCall = async (toolCallId) => {
+  const approveToolCall = async (toolCallId, resumeData) => {
     const onChunk = _onChunk.current;
     const currentRunId = _currentRunId.current;
     if (!currentRunId)
@@ -2371,6 +2455,7 @@ var useChat = ({
           threadId,
           toolCallId,
           approved: true,
+          ...resumeData !== void 0 ? { resumeData } : {},
           requestContext: _requestContext.current
         });
         pendingToolApprovalIdsRef.current.delete(toolCallId);
@@ -2570,6 +2655,7 @@ var useChat = ({
     isRunning,
     isAwaitingToolApproval,
     messages,
+    tasks,
     approveToolCall,
     declineToolCall,
     approveToolCallGenerate,
@@ -2930,7 +3016,7 @@ var renderPart = (part, renderers, fallback) => {
     case "file":
       return renderers.File?.(part) ?? fallback?.(part) ?? null;
     case "step-start":
-      return renderers.StepStart?.(part) ?? fallback?.(part) ?? null;
+      return renderers.StepStart?.(part) ?? null;
     case "tool-invocation":
       return renderers.ToolInvocation?.(part) ?? fallback?.(part) ?? null;
     case "source":