Fixed issue with asResponse and withResponse not working on runs.retrieve

.changeset/lazy-carpets-reply.md

@@ -0,0 +1,6 @@
+---
+"@trigger.dev/sdk": patch
+"@trigger.dev/core": patch
+---
+
+Fixed issue with asResponse and withResponse not working on runs.retrieve

packages/core/src/v3/apiClient/core.ts

@@ -26,15 +26,18 @@ export const defaultRetryOptions = {
   randomize: false,
 } satisfies RetryOptions;
 
-export type ZodFetchOptions = {
+export type ZodFetchOptions<T = unknown> = {
   retry?: RetryOptions;
   tracer?: TriggerTracer;
   name?: string;
   attributes?: Attributes;
   icon?: string;
   onResponseBody?: (body: unknown, span: Span) => void;
+  prepareData?: (data: T) => Promise<T> | T;
 };
 
+export type AnyZodFetchOptions = ZodFetchOptions<any>;
+
 export type ApiRequestOptions = Pick<ZodFetchOptions, "retry">;
 
 type KeysEnum<T> = { [P in keyof Required<T>]: true };
@@ -67,7 +70,7 @@ export function zodfetch<TResponseBodySchema extends z.ZodTypeAny>(
   schema: TResponseBodySchema,
   url: string,
   requestInit?: RequestInit,
-  options?: ZodFetchOptions
+  options?: ZodFetchOptions<z.output<TResponseBodySchema>>
 ): ApiPromise<z.output<TResponseBodySchema>> {
   return new ApiPromise(_doZodFetch(schema, url, requestInit, options));
 }
@@ -197,6 +200,10 @@ async function _doZodFetch<TResponseBodySchema extends z.ZodTypeAny>(
       options.onResponseBody(result.data, span);
     }
 
+    if (options?.prepareData) {
+      result.data = await options.prepareData(result.data);
+    }
+
     return result;
   });
 }

packages/core/src/v3/apiClient/index.ts

@@ -32,6 +32,7 @@ import {
 import { taskContext } from "../task-context-api.js";
 import { AnyRunTypes, TriggerJwtOptions } from "../types/tasks.js";
 import {
+  AnyZodFetchOptions,
   ApiRequestOptions,
   CursorPagePromise,
   ZodFetchOptions,
@@ -863,9 +864,9 @@ function createSearchQueryForListRuns(query?: ListRunsQueryParams): URLSearchPar
 }
 
 export function mergeRequestOptions(
-  defaultOptions: ZodFetchOptions,
+  defaultOptions: AnyZodFetchOptions,
   options?: ApiRequestOptions
-): ZodFetchOptions {
+): AnyZodFetchOptions {
   if (!options) {
     return defaultOptions;
   }

packages/trigger-sdk/src/v3/runs.ts

@@ -14,9 +14,9 @@ import type {
   TaskRunShape,
   AnyBatchedRunHandle,
   AsyncIterableStream,
+  ApiPromise,
 } from "@trigger.dev/core/v3";
 import {
-  ApiPromise,
   CanceledRunResponse,
   CursorPagePromise,
   ListRunResponseItem,
@@ -187,15 +187,14 @@ function retrieveRun<TRunId extends AnyRunHandle | AnyBatchedRunHandle | AnyTask
           style: "codepath",
         }),
       },
+      prepareData: resolvePayloadAndOutputUrls,
     },
     requestOptions
   );
 
   const $runId = typeof runId === "string" ? runId : runId.id;
 
-  return apiClient.retrieveRun($runId, $requestOptions).then((retrievedRun) => {
-    return resolvePayloadAndOutputUrls(retrievedRun);
-  }) as ApiPromise<RetrieveRunResult<TRunId>>;
+  return apiClient.retrieveRun($runId, $requestOptions) as ApiPromise<RetrieveRunResult<TRunId>>;
 }
 
 async function resolvePayloadAndOutputUrls(run: AnyRetrieveRunResult) {