fix types casting

Author: whoiskatrin

packages/sandbox/src/sandbox.ts

@@ -45,9 +45,7 @@ export function getSandbox(
   }
 
   if (options?.sleepAfter !== undefined) {
-    // Set sleepAfter property on the stub
-    // Cast to any to work around RPC type inference issues
-    (stub as any).sleepAfter = options.sleepAfter;
+    stub.setSleepAfter(options.sleepAfter);
   }
 
   return stub;
@@ -127,6 +125,11 @@ export class Sandbox<Env = unknown> extends Container<Env> implements ISandbox {
     }
   }
 
+  // RPC method to set the sleep timeout
+  async setSleepAfter(sleepAfter: string | number): Promise<void> {
+    this.sleepAfter = sleepAfter;
+  }
+
   // RPC method to set environment variables
   async setEnvVars(envVars: Record<string, string>): Promise<void> {
     // Update local state for new sessions

packages/sandbox/tests/get-sandbox.test.ts

@@ -27,6 +27,9 @@ describe('getSandbox', () => {
       sleepAfter: '3m',
       setSandboxName: vi.fn(),
       setBaseUrl: vi.fn(),
+      setSleepAfter: vi.fn((value: string | number) => {
+        mockStub.sleepAfter = value;
+      }),
     };
 
     // Mock getContainer to return our stub