Fixing some typecheck errors

Author: ericallam

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

@@ -54,14 +54,12 @@ export function runShapeStream<TRunTypes extends AnyRunTypes>(
   url: string,
   options?: RunShapeStreamOptions
 ): RunSubscription<TRunTypes> {
-  const subscription = new RunSubscription<TRunTypes>(url, options);
-
-  return subscription.init();
+  return new RunSubscription<TRunTypes>(url, options);
 }
 
 export class RunSubscription<TRunTypes extends AnyRunTypes> {
   private abortController: AbortController;
-  private unsubscribeShape: () => void;
+  private unsubscribeShape?: () => void;
   private stream: AsyncIterableStream<RunShape<TRunTypes>>;
   private packetCache = new Map<string, any>();
 
@@ -70,9 +68,7 @@ export class RunSubscription<TRunTypes extends AnyRunTypes> {
     private options?: RunShapeStreamOptions
   ) {
     this.abortController = new AbortController();
-  }
 
-  init(): this {
     const source = new ReadableStream<SubscribeRunRawShape>({
       start: async (controller) => {
         this.unsubscribeShape = await zodShapeStream(
@@ -108,8 +104,6 @@ export class RunSubscription<TRunTypes extends AnyRunTypes> {
         controller.enqueue(run);
       },
     });
-
-    return this;
   }
 
   unsubscribe(): void {

packages/core/src/v3/errors.ts

@@ -19,6 +19,8 @@ export class AbortTaskRunError extends Error {
 }
 
 export class TaskPayloadParsedError extends Error {
+  public readonly cause: unknown;
+
   constructor(cause: unknown) {
     const causeMessage = cause instanceof Error ? cause.message : String(cause);
 

packages/core/src/v3/types/tasks.ts

@@ -32,11 +32,13 @@ export class SubtaskUnwrapError extends Error {
 
   constructor(taskId: string, runId: string, subtaskError: unknown) {
     if (subtaskError instanceof Error) {
-      super(`Error in ${taskId}: ${subtaskError.message}`, { cause: subtaskError });
+      super(`Error in ${taskId}: ${subtaskError.message}`);
+      this.cause = subtaskError;
       this.name = "SubtaskUnwrapError";
     } else {
-      super(`Error in ${taskId}`, { cause: subtaskError });
+      super(`Error in ${taskId}`);
       this.name = "SubtaskUnwrapError";
+      this.cause = subtaskError;
     }
 
     this.taskId = taskId;