a couple tweaks

Author: ericallam

packages/core/src/v3/workers/taskExecutor.ts

@@ -134,8 +134,6 @@ export class TaskExecutor {
 
             const output = await this.#callRun(parsedPayload, ctx, initOutput, signal);
 
-            await this.#callOnSuccessFunctions(parsedPayload, output, ctx, initOutput, signal);
-
             try {
               const stringifiedOutput = await stringifyIO(output);
 
@@ -155,6 +153,8 @@ export class TaskExecutor {
                 span.setAttributes(attributes);
               }
 
+              await this.#callOnSuccessFunctions(parsedPayload, output, ctx, initOutput, signal);
+
               // Call onComplete with success result
               await this.#callOnCompleteFunctions(
                 parsedPayload,
@@ -173,15 +173,6 @@ export class TaskExecutor {
             } catch (outputError) {
               recordSpanException(span, outputError);
 
-              // Call onComplete with error result
-              await this.#callOnCompleteFunctions(
-                parsedPayload,
-                { ok: false, error: outputError },
-                ctx,
-                initOutput,
-                signal
-              );
-
               return {
                 ok: false,
                 id: execution.run.id,
@@ -243,15 +234,6 @@ export class TaskExecutor {
             } catch (handleErrorError) {
               recordSpanException(span, handleErrorError);
 
-              // Call onComplete with error result
-              await this.#callOnCompleteFunctions(
-                parsedPayload,
-                { ok: false, error: handleErrorError },
-                ctx,
-                initOutput,
-                signal
-              );
-
               return {
                 ok: false,
                 id: execution.run.id,