inngest 4.18.0 → 4.19.0

package/components/execution/engine.cjs

@@ -189,6 +189,11 @@ var InngestExecutionEngine = class extends require_InngestExecution.InngestExecu
 					}
 				}, async () => {
 					return tracer.startActiveSpan("inngest.execution", (span) => {
+						let aiMetadataScope;
+						if (this.options.client.aiMetadataEnabled) aiMetadataScope = require_metadataProcessor.metadataSpanProcessor.startScope({
+							runId: this.options.runId,
+							span
+						});
 						this.rootSpanId = span.spanContext().spanId;
 						require_access.clientProcessorMap.get(this.options.client)?.declareStartingSpan({
 							span,
@@ -196,7 +201,8 @@ var InngestExecutionEngine = class extends require_InngestExecution.InngestExecu
 							traceparent: this.options.headers[require_consts.headerKeys.TraceParent],
 							tracestate: this.options.headers[require_consts.headerKeys.TraceState]
 						});
-						if (this.options.client.aiMetadataEnabled) require_metadataProcessor.metadataSpanProcessor.declareStartingSpan({
+						if (aiMetadataScope) require_metadataProcessor.metadataSpanProcessor.declareStartingSpan({
+							scope: aiMetadataScope,
 							span,
 							traceparent: this.options.headers[require_consts.headerKeys.TraceParent],
 							onAIMetadata: (aiMetadata) => {
@@ -204,11 +210,12 @@ var InngestExecutionEngine = class extends require_InngestExecution.InngestExecu
 								this.state.executingStepAIMetadata = this.state.executingStepAIMetadata ? require_aiExtractor.aggregate(this.state.executingStepAIMetadata, aiMetadata) : aiMetadata;
 							}
 						});
-						return this._start().then((result) => {
+						return this._start(() => {
+							span.end();
+							if (aiMetadataScope) require_metadataProcessor.metadataSpanProcessor.endScope(aiMetadataScope);
+						}).then((result) => {
 							this.devDebug("result:", result);
 							return result;
-						}).finally(() => {
-							span.end();
 						});
 					});
 				});
@@ -230,10 +237,14 @@ var InngestExecutionEngine = class extends require_InngestExecution.InngestExecu
 	}
 	/**
 	* Starts execution of the user's function and the core loop.
+	*
+	* @param onCoreLoopSettled - Called after this invocation has finished
+	* running user code and processing checkpoints, even if this method returned
+	* earlier. The run may re-enter later in a new SDK request.
 	*/
-	async _start() {
+	async _start(onCoreLoopSettled) {
 		if (this.options.stepMode === require_types.StepMode.Sync && this.options.acceptsSse) this.earlyStreamResponse = require_promises.createDeferredPromise();
-		const coreLoop = this.runCoreLoop();
+		const coreLoop = this.runCoreLoop().finally(onCoreLoopSettled);
 		if (this.earlyStreamResponse) {
 			coreLoop.catch((err) => {
 				this.options.client[require_Inngest.internalLoggerSymbol].error({ err }, "Core loop rejected after early stream response was sent");
@@ -1533,6 +1544,13 @@ var InngestExecutionEngine = class extends require_InngestExecution.InngestExecu
 				let deferMeta;
 				try {
 					deferMeta = require_sessions.normalizeEventMeta(meta);
+					if (this.options.client[require_Inngest.sessionPropagationSymbol]) {
+						const propagated = require_sessions.normalizePropagatedSessions(this.fnArg.sessions);
+						if (propagated) deferMeta = {
+							...deferMeta,
+							propagated_sessions: propagated
+						};
+					}
 				} catch (err) {
 					log.error({
 						runId,
@@ -1540,13 +1558,6 @@ var InngestExecutionEngine = class extends require_InngestExecution.InngestExecu
 					}, "defer skipped: invalid meta");
 					return noopHandle;
 				}
-				if (this.options.client[require_Inngest.sessionPropagationSymbol]) {
-					const propagated = this.fnArg.sessions;
-					if (propagated && Object.keys(propagated).length > 0) deferMeta = {
-						...deferMeta,
-						propagated_sessions: propagated
-					};
-				}
 				stepHandler({
 					args: [stepOptions, finalInput],
 					matchOp: (stepOptions$1, inputArg) => ({