@mastra/core 1.11.0 → 1.12.0-alpha.0

package/dist/{chunk-S5BMBKUR.js → chunk-Z7VOJIJU.js}

@@ -1,13 +1,13 @@
-import { ModelsDevGateway, parseModelRouterId, ToolLoopAgent } from './chunk-Y2MVLQFW.js';
-import { NetlifyGateway } from './chunk-VW4MYD6N.js';
-import { EventSourceParserStream, MastraModelGateway, createOpenAICompatible, createOpenAI, createGoogleGenerativeAI } from './chunk-GOGAIHBU.js';
-import { PROVIDER_REGISTRY, GatewayRegistry } from './chunk-RF2EPUM4.js';
+import { ModelsDevGateway, parseModelRouterId, ToolLoopAgent } from './chunk-TZE75LDQ.js';
+import { NetlifyGateway } from './chunk-PYRU2AFG.js';
+import { EventSourceParserStream, MastraModelGateway, createOpenAICompatible, createOpenAI, createGoogleGenerativeAI, MASTRA_USER_AGENT } from './chunk-55I2D5RA.js';
+import { PROVIDER_REGISTRY, GatewayRegistry } from './chunk-4RE5DA24.js';
 import { MessageList, output_exports, generateText, generateObject, streamText, streamObject, coreContentToString, DefaultGeneratedFile, DefaultGeneratedFileWithType, sanitizeToolName } from './chunk-WD467XGW.js';
 import { DefaultVoice } from './chunk-ISDIKE46.js';
 import { generateId, parsePartialJson, isDeepEqualData, APICallError, tool, asSchema, stepCountIs } from './chunk-76ZCW5L6.js';
 import { PUBSUB_SYMBOL, STREAM_FORMAT_SYMBOL } from './chunk-2QXNHEDL.js';
 import { BM25Index, createWorkspaceTools, createSkillTools } from './chunk-7FVJIJGY.js';
-import { TABLE_SCHEMAS, TABLE_SCORERS, listTracesArgsSchema, toTraceSpans, TABLE_WORKFLOW_SNAPSHOT, saveScorePayloadSchema } from './chunk-W3LSSG4T.js';
+import { TABLE_SCHEMAS, TABLE_SCORERS, listTracesArgsSchema, toTraceSpans, TABLE_WORKFLOW_SNAPSHOT, saveScorePayloadSchema } from './chunk-5X4N6M7R.js';
 import { resolveObservabilityContext, executeWithContext, executeWithContextSync, createObservabilityContext, getOrCreateSpan, NoOpObservability, noOpLoggerContext, noOpMetricsContext, wrapMastra } from './chunk-YY3T5AJK.js';
 import { registerHook, executeHook } from './chunk-L54GIUCB.js';
 import { ToolStream } from './chunk-KQUKNTG7.js';
@@ -15,10 +15,10 @@ import { createTool, Tool, isVercelTool, validateToolInput, validateToolOutput,
 import { isZodArray, getZodDef, isZodObject } from './chunk-RT2LCNDN.js';
 import { toStandardSchema, standardSchemaToJSONSchema, isStandardSchemaWithJSON } from './chunk-6VAVASZ5.js';
 import { noopLogger } from './chunk-PSCMWPLC.js';
+import { InMemoryServerCache } from './chunk-GMSAGYTB.js';
 import { RequestContext, MASTRA_THREAD_ID_KEY, MASTRA_RESOURCE_ID_KEY } from './chunk-JGOH7RWL.js';
 import { MastraError, getErrorFromUnknown } from './chunk-FJEVLHJT.js';
 import { EventEmitterPubSub } from './chunk-X2HAI3OX.js';
-import { InMemoryServerCache } from './chunk-GMSAGYTB.js';
 import { MastraBase } from './chunk-WCAFTXGK.js';
 import { RegisteredLogger, ConsoleLogger, LogLevel } from './chunk-X2WMFSPB.js';
 import { __commonJS, __export, __reExport, __toESM } from './chunk-TM6AOUSD.js';
@@ -8077,14 +8077,16 @@ var AzureOpenAIGateway = class extends MastraModelGateway {
   }
   async resolveLanguageModel({
     modelId,
-    apiKey
+    apiKey,
+    headers
   }) {
     const apiVersion = this.config.apiVersion || "2024-04-01-preview";
     return createAzure({
       resourceName: this.config.resourceName,
       apiKey,
       apiVersion,
-      useDeploymentBasedUrls: true
+      useDeploymentBasedUrls: true,
+      headers: { "User-Agent": MASTRA_USER_AGENT, ...headers }
     })(modelId);
   }
 };
@@ -26286,7 +26288,8 @@ async function persistStepUpdate(engine, params) {
     workflowStatus,
     result,
     error,
-    requestContext
+    requestContext,
+    tracingContext
   } = params;
   const operationId = `workflow.${workflowId}.run.${runId}.path.${JSON.stringify(executionContext.executionPath)}.stepUpdate`;
   await engine.wrapDurableOperation(operationId, async () => {
@@ -26316,7 +26319,9 @@ async function persistStepUpdate(engine, params) {
         result,
         error,
         requestContext: requestContextObj,
-        timestamp: Date.now()
+        timestamp: Date.now(),
+        // Persist tracing context for span continuity on resume
+        tracingContext
       }
     });
   });
@@ -27948,6 +27953,11 @@ var DefaultExecutionEngine = class extends ExecutionEngine {
           void 0,
           stepExecutionPath
         );
+        const persistTracingContext = result2.status === "suspended" && workflowSpan ? {
+          traceId: workflowSpan.traceId,
+          spanId: workflowSpan.id,
+          parentSpanId: workflowSpan.getParentSpanId()
+        } : void 0;
         await this.persistStepUpdate({
           workflowId,
           runId,
@@ -27958,7 +27968,8 @@ var DefaultExecutionEngine = class extends ExecutionEngine {
           workflowStatus: result2.status,
           result: result2.result,
           error: result2.error,
-          requestContext: currentRequestContext
+          requestContext: currentRequestContext,
+          tracingContext: persistTracingContext
         });
         if (result2.error) {
           workflowSpan?.error({
@@ -30444,18 +30455,30 @@ var Run = class {
         requestContextToUse.set(key, value);
       }
     });
+    const persistedTracingContext = snapshot?.tracingContext;
+    const userProvidedTraceId = params.tracingOptions?.traceId;
+    const userProvidedParentSpanId = params.tracingOptions?.parentSpanId;
+    const effectiveTraceId = userProvidedTraceId ?? (!userProvidedParentSpanId ? persistedTracingContext?.traceId : void 0);
+    const shouldUsePersistedParentSpan = !userProvidedParentSpanId && (!userProvidedTraceId || userProvidedTraceId === persistedTracingContext?.traceId);
+    const resumeTracingOptions = {
+      ...params.tracingOptions,
+      traceId: effectiveTraceId,
+      parentSpanId: shouldUsePersistedParentSpan ? persistedTracingContext?.spanId : params.tracingOptions?.parentSpanId
+    };
     const workflowSpan = getOrCreateSpan({
       type: "workflow_run" /* WORKFLOW_RUN */,
-      name: `workflow run: '${this.workflowId}'`,
+      name: `workflow run: '${this.workflowId}' (resumed)`,
       entityType: "workflow_run" /* WORKFLOW_RUN */,
       entityId: this.workflowId,
       input: resumeDataToUse,
       metadata: {
         resourceId: this.resourceId,
-        runId: this.runId
+        runId: this.runId,
+        resumed: true,
+        resumedFromSpanId: persistedTracingContext?.spanId
       },
       tracingPolicy: this.tracingPolicy,
-      tracingOptions: params.tracingOptions,
+      tracingOptions: resumeTracingOptions,
       tracingContext: observabilityContext.tracingContext,
       requestContext: requestContextToUse,
       mastra: this.#mastra
@@ -34036,6 +34059,13 @@ function createAgenticLoopWorkflow(params) {
             }
           } else if (iterationResult.continue === false && !hasFinishedSteps) {
             hasFinishedSteps = true;
+          } else if (iterationResult.continue === true && (hasFinishedSteps || !typedInputData.stepResult?.isContinued)) {
+            if (rest.maxSteps && accumulatedSteps.length < rest.maxSteps || !rest.maxSteps) {
+              hasFinishedSteps = false;
+              if (typedInputData.stepResult) {
+                typedInputData.stepResult.isContinued = true;
+              }
+            }
           }
         }
       } catch (error) {
@@ -40484,6 +40514,8 @@ ${errorMessages}`,
               source: "agent",
               entityId: agentName
             }) || `${slugify.default(this.id)}-${agentName}`;
+            const subAgentDefaultOptions = await agent.getDefaultOptions?.({ requestContext });
+            const subAgentHasOwnMemoryConfig = subAgentDefaultOptions?.memory !== void 0;
             const savedMastraMemory = requestContext.get("MastraMemory");
             if ((methodType === "generate" || methodType === "generateLegacy" || methodType === "stream" || methodType === "streamLegacy") && supportedLanguageModelSpecifications.includes(modelVersion)) {
               if (!agent.hasOwnMemory() && this.#memory) {
@@ -40637,7 +40669,7 @@ ${effectiveInstructions}`;
                   ...resolveObservabilityContext(context ?? {}),
                   ...effectiveInstructions && { instructions: effectiveInstructions },
                   ...effectiveMaxSteps && { maxSteps: effectiveMaxSteps },
-                  ...resourceId && threadId ? {
+                  ...resourceId && threadId && !subAgentHasOwnMemoryConfig ? {
                     memory: {
                       resource: subAgentResourceId,
                       thread: subAgentThreadId,
@@ -40649,7 +40681,7 @@ ${effectiveInstructions}`;
                   ...resolveObservabilityContext(context ?? {}),
                   ...effectiveInstructions && { instructions: effectiveInstructions },
                   ...effectiveMaxSteps && { maxSteps: effectiveMaxSteps },
-                  ...resourceId && threadId ? {
+                  ...resourceId && threadId && !subAgentHasOwnMemoryConfig ? {
                     memory: {
                       resource: subAgentResourceId,
                       thread: subAgentThreadId,
@@ -40720,7 +40752,7 @@ ${effectiveInstructions}`;
                   ...resolveObservabilityContext(context ?? {}),
                   ...effectiveInstructions && { instructions: effectiveInstructions },
                   ...effectiveMaxSteps && { maxSteps: effectiveMaxSteps },
-                  ...resourceId && threadId ? {
+                  ...resourceId && threadId && !subAgentHasOwnMemoryConfig ? {
                     memory: {
                       resource: subAgentResourceId,
                       thread: subAgentThreadId,
@@ -40734,7 +40766,7 @@ ${effectiveInstructions}`;
                   ...resolveObservabilityContext(context ?? {}),
                   ...effectiveInstructions && { instructions: effectiveInstructions },
                   ...effectiveMaxSteps && { maxSteps: effectiveMaxSteps },
-                  ...resourceId && threadId ? {
+                  ...resourceId && threadId && !subAgentHasOwnMemoryConfig ? {
                     memory: {
                       resource: subAgentResourceId,
                       thread: subAgentThreadId,
@@ -44722,8 +44754,7 @@ var StepExecutor = class extends MastraBase {
             [STREAM_FORMAT_SYMBOL]: params.format,
             engine: {},
             abortSignal: abortController?.signal,
-            // TODO
-            ...createObservabilityContext()
+            ...createObservabilityContext(params.tracingContext)
           },
           {
             paramName: "runCount",
@@ -51680,7 +51711,7 @@ var Mastra = class {
       if (process.env.MASTRA_DEV !== "true" && process.env.MASTRA_DEV !== "1") {
         return;
       }
-      import('./provider-registry-6FITK73S.js').then(async ({ GatewayRegistry: GatewayRegistry2 }) => {
+      import('./provider-registry-E5XXTB3V.js').then(async ({ GatewayRegistry: GatewayRegistry2 }) => {
         const registry = GatewayRegistry2.getInstance();
         const customGateways = Object.values(this.#gateways || {});
         registry.registerCustomGateways(customGateways);
@@ -52540,7 +52571,7 @@ async function fetchWithRetry(url, options = {}, maxRetries = 3) {
       if (retryCount >= maxRetries) {
         break;
       }
-      const delay3 = Math.min(1e3 * Math.pow(2, retryCount) * 1e3, 1e4);
+      const delay3 = Math.min(1e3 * Math.pow(2, retryCount), 1e4);
       await new Promise((resolve4) => setTimeout(resolve4, delay3));
     }
   }
@@ -52586,5 +52617,5 @@ var removeUndefinedValues = (obj) => {
 };
 
 export { AISDKV5LanguageModel, Agent, AgentsStorage, AzureOpenAIGateway, BaseProcessor, BatchPartsProcessor, BlobStore, ChunkFrom, Dataset, DatasetsInMemory, DatasetsManager, DatasetsStorage, DefaultExecutionEngine, EDITOR_DOMAINS, EMBEDDING_MODELS, EventedExecutionEngine, EventedRun, EventedWorkflow, ExecutionEngine, ExperimentsInMemory, ExperimentsStorage, FilePartSchema, FilesystemAgentsStorage, FilesystemDB, FilesystemMCPClientsStorage, FilesystemMCPServersStorage, FilesystemPromptBlocksStorage, FilesystemScorerDefinitionsStorage, FilesystemSkillsStorage, FilesystemStore, FilesystemVersionedHelpers, FilesystemWorkspacesStorage, GitHistory, ImagePartSchema, InMemoryAgentsStorage, InMemoryBlobStore, InMemoryDB, InMemoryMCPClientsStorage, InMemoryMCPServersStorage, InMemoryMemory, InMemoryPromptBlocksStorage, InMemoryScorerDefinitionsStorage, InMemorySkillsStorage, InMemoryStore, InMemoryWorkspacesStorage, LanguageDetector, MCPClientsStorage, MCPServersStorage, Mastra, MastraAgentNetworkStream, MastraCompositeStore, MastraLLMV1, MastraMemory, MastraModelOutput, MastraScorer, MastraStorage, MemoryProcessor, MemoryStorage, MessageContentSchema, MessageHistory, MessagePartSchema, MockMemory, MockStore, ModelRouterEmbeddingModel, ModelRouterLanguageModel, ModerationProcessor, ObservabilityInMemory, ObservabilityStorage, PIIDetector, ProcessorInputPhaseSchema, ProcessorInputStepPhaseSchema, ProcessorMessageContentSchema, ProcessorMessageSchema, ProcessorOutputResultPhaseSchema, ProcessorOutputStepPhaseSchema, ProcessorOutputStreamPhaseSchema, ProcessorRunner, ProcessorState, ProcessorStepInputSchema, ProcessorStepOutputSchema, ProcessorStepSchema, PromptBlocksStorage, PromptInjectionDetector, ReasoningPartSchema, Run, SchemaUpdateValidationError, SchemaValidationError, SchemaValidator, ScorerDefinitionsStorage, ScoresInMemory, ScoresStorage, SemanticRecall, SkillsProcessor, SkillsStorage, SourcePartSchema, StepExecutor, StepStartPartSchema, StorageDomain, StoreOperations, StoreOperationsInMemory, StructuredOutputProcessor, SystemPromptScrubber, TextPartSchema, TokenLimiterProcessor, ToolCallFilter, ToolInvocationPartSchema, ToolSearchProcessor, TripWire, UnicodeNormalizer, VersionedStorageDomain, WORKING_MEMORY_END_TAG, WORKING_MEMORY_START_TAG, Workflow, WorkflowEventProcessor, WorkflowRunOutput, WorkflowsInMemory, WorkflowsStorage, WorkingMemory, WorkspaceInstructionsProcessor, WorkspacesStorage, calculatePagination, checkEvalStorageFields, cleanStepResult, cloneStep, cloneStep2, cloneWorkflow, cloneWorkflow2, compareExperiments, computeMean, computeScorerStats, convertFullStreamChunkToMastra, convertFullStreamChunkToUIMessageStream, convertMastraChunkToAISDKv5, createDeprecationProxy, createMastraProxy, createScorer, createStep, createStep2, createStorageErrorId, createStoreErrorId, createTimeTravelExecutionParams, createValidator, createVectorErrorId, createWorkflow, createWorkflow2, deepEqual, deepMerge, delay2 as delay, ensureDate, ensureToolProperties, executeTarget, extractWorkingMemoryContent, extractWorkingMemoryTags, fetchWithRetry, filterByDateRange, formatCheckFeedback, formatCompletionFeedback, formatStreamCompletionFeedback, formatValidationFeedback, generateEmptyFromSchema, generateFinalResult, generateStructuredFinalResult, getDefaultValue, getNestedValue, getResumeLabelsByStepId, getSchemaValidator, getSettings, getSqlType, getStepIds, getStepResult, getThreadOMMetadata, globalEmbeddingCache, hydrateSerializedStepErrors, isCoreMessage, isObservationalMemoryEnabled, isProcessor, isProcessorWorkflow, isRegression, isSupportedLanguageModel, isToolLoopAgentLike, isUiMessage, isZodType, jsonValueEquals, loop, makeCoreTool, makeCoreToolV5, mapVariable, maskStreamTags, memoryDefaultOptions, normalizePerPage, omitKeys, parseFieldKey, parseMemoryRequestContext, parseSqlIdentifier, removeUndefinedValues, removeWorkingMemoryTags, resolveModelConfig, resolveScorers, resolveSerializedZodOutput, resolveThreadIdFromArgs, runChecks, runCompletionScorers, runCountDeprecationMessage, runDefaultCompletionCheck, runEvals, runExperiment, runScorersForItem, runStreamCompletionScorers, runValidation, safelyParseJSON, selectFields, serializeDate, setNestedValue, setThreadOMMetadata, supportedLanguageModelSpecifications, toolLoopAgentToMastraAgent, transformRow, transformScoreRow, tryGenerateWithJsonFallback, tryStreamWithJsonFallback, validateStepInput, validateStepRequestContext, validateStepResumeData, validateStepStateData, validateStepSuspendData };
-//# sourceMappingURL=chunk-S5BMBKUR.js.map
-//# sourceMappingURL=chunk-S5BMBKUR.js.map
+//# sourceMappingURL=chunk-Z7VOJIJU.js.map
+//# sourceMappingURL=chunk-Z7VOJIJU.js.map