docs: fix PTY Python examples to match the real SDK API

Author: beran-t

docs/sandbox/pty.mdx

@@ -25,8 +25,8 @@ const terminal = await sandbox.pty.create({
   cols: 80,              // Terminal width in characters
   rows: 24,              // Terminal height in characters
   onData: (data) => {
-    // Called whenever terminal outputs data
-    process.stdout.write(data)
+    // Called whenever terminal outputs data (decode the Uint8Array to text)
+    process.stdout.write(new TextDecoder().decode(data))
   },
   envs: { MY_VAR: 'hello' },  // Optional environment variables
   cwd: '/home/user',          // Optional working directory
@@ -38,21 +38,24 @@ console.log('Terminal PID:', terminal.pid)
 ```
 
 ```python Python
-from e2b import Sandbox
+from e2b import Sandbox, PtySize
 
 sandbox = Sandbox()
 
+# PtySize takes the terminal dimensions in characters
 terminal = sandbox.pty.create(
-    cols=80,              # Terminal width in characters
-    rows=24,              # Terminal height in characters
-    on_data=lambda data: print(data.decode(), end=''),  # end='' prevents print from adding extra newline
+    PtySize(rows=24, cols=80),
     envs={'MY_VAR': 'hello'},  # Optional environment variables
     cwd='/home/user',          # Optional working directory
     user='root',               # Optional user to run as
 )
 
 # terminal.pid contains the process ID
 print('Terminal PID:', terminal.pid)
+
+# Handle output by passing on_pty to wait(). The data is bytes, so decode it.
+# end='' prevents print from adding an extra newline (PTY output already contains newlines).
+terminal.wait(on_pty=lambda data: print(data.decode(), end=''))
 ```
 </CodeGroup>
 
@@ -73,30 +76,30 @@ const sandbox = await Sandbox.create()
 const terminal = await sandbox.pty.create({
   cols: 80,
   rows: 24,
-  onData: (data) => process.stdout.write(data),
+  onData: (data) => process.stdout.write(new TextDecoder().decode(data)),
   timeoutMs: 0,  // Keep the session open indefinitely
 })
 ```
 
 ```python Python
-from e2b import Sandbox
+from e2b import Sandbox, PtySize
 
 sandbox = Sandbox()
 
-# end='' prevents print() from adding an extra newline
-# (PTY output already contains newlines)
 terminal = sandbox.pty.create(
-    cols=80,
-    rows=24,
-    on_data=lambda data: print(data.decode(), end=''),
+    PtySize(rows=24, cols=80),
     timeout=0,  # Keep the session open indefinitely
 )
+
+# Handle output by passing on_pty to wait(). The data is bytes, so decode it.
+# end='' prevents print() from adding an extra newline (PTY output already contains newlines).
+terminal.wait(on_pty=lambda data: print(data.decode(), end=''))
 ```
 </CodeGroup>
 
 ## Send input to PTY
 
-Use `sendInput()` in JavaScript or `send_stdin()` in Python to send data to the terminal. These methods return a Promise (JavaScript) or complete synchronously (Python) - the actual output will be delivered to your `onData` callback.
+Use `sendInput()` in JavaScript or `send_stdin()` in Python to send data to the terminal. These methods return a Promise (JavaScript) or complete synchronously (Python) - the actual output is delivered to your output callback (`onData` in JavaScript, or `on_pty` passed to `wait()` in Python).
 
 <CodeGroup>
 ```js JavaScript & TypeScript
@@ -107,7 +110,7 @@ const sandbox = await Sandbox.create()
 const terminal = await sandbox.pty.create({
   cols: 80,
   rows: 24,
-  onData: (data) => process.stdout.write(data),
+  onData: (data) => process.stdout.write(new TextDecoder().decode(data)),
 })
 
 // Send a command (don't forget the newline!)
@@ -118,19 +121,18 @@ await sandbox.pty.sendInput(
 ```
 
 ```python Python
-from e2b import Sandbox
+from e2b import Sandbox, PtySize
 
 sandbox = Sandbox()
 
-terminal = sandbox.pty.create(
-    cols=80,
-    rows=24,
-    on_data=lambda data: print(data.decode(), end=''),  # end='' prevents extra newline
-)
+terminal = sandbox.pty.create(PtySize(rows=24, cols=80))
 
 # Send a command as bytes (b'...' is Python's byte string syntax)
 # Don't forget the newline!
 sandbox.pty.send_stdin(terminal.pid, b'echo "Hello from PTY"\n')
+
+# Receive output via on_pty on wait() (data is bytes, so decode it)
+terminal.wait(on_pty=lambda data: print(data.decode(), end=''))
 ```
 </CodeGroup>
 
@@ -147,7 +149,7 @@ const sandbox = await Sandbox.create()
 const terminal = await sandbox.pty.create({
   cols: 80,
   rows: 24,
-  onData: (data) => process.stdout.write(data),
+  onData: (data) => process.stdout.write(new TextDecoder().decode(data)),
 })
 
 // Resize to new dimensions (in characters)
@@ -158,18 +160,14 @@ await sandbox.pty.resize(terminal.pid, {
 ```
 
 ```python Python
-from e2b import Sandbox
+from e2b import Sandbox, PtySize
 
 sandbox = Sandbox()
 
-terminal = sandbox.pty.create(
-    cols=80,
-    rows=24,
-    on_data=lambda data: print(data.decode(), end=''),  # end='' prevents extra newline
-)
+terminal = sandbox.pty.create(PtySize(rows=24, cols=80))
 
 # Resize to new dimensions (in characters)
-sandbox.pty.resize(terminal.pid, cols=120, rows=40)
+sandbox.pty.resize(terminal.pid, PtySize(rows=40, cols=120))
 ```
 </CodeGroup>
 
@@ -214,38 +212,29 @@ await reconnected.wait()
 ```
 
 ```python Python
-import time
-from e2b import Sandbox
+from e2b import Sandbox, PtySize
 
 sandbox = Sandbox()
 
 # Create a PTY session
-terminal = sandbox.pty.create(
-    cols=80,
-    rows=24,
-    on_data=lambda data: print('Handler 1:', data.decode()),
-)
+terminal = sandbox.pty.create(PtySize(rows=24, cols=80))
 
 pid = terminal.pid
 
 # Send a command
 sandbox.pty.send_stdin(pid, b'echo hello\n')
-time.sleep(0.5)
 
 # Disconnect - PTY keeps running in the background
 terminal.disconnect()
 
-# Later: reconnect with a new data handler
-reconnected = sandbox.pty.connect(
-    pid,
-    on_data=lambda data: print('Handler 2:', data.decode()),
-)
+# Later: reconnect to the running session
+reconnected = sandbox.pty.connect(pid)
 
 # Continue using the session
 sandbox.pty.send_stdin(pid, b'echo world\n')
 
-# Wait for the terminal to exit
-reconnected.wait()
+# Receive output via on_pty on wait() and wait for the terminal to exit
+reconnected.wait(on_pty=lambda data: print('Handler 2:', data.decode()))
 ```
 </CodeGroup>
 
@@ -262,7 +251,7 @@ const sandbox = await Sandbox.create()
 const terminal = await sandbox.pty.create({
   cols: 80,
   rows: 24,
-  onData: (data) => process.stdout.write(data),
+  onData: (data) => process.stdout.write(new TextDecoder().decode(data)),
 })
 
 // Kill the PTY
@@ -274,15 +263,11 @@ console.log('Killed:', killed)  // true if successful
 ```
 
 ```python Python
-from e2b import Sandbox
+from e2b import Sandbox, PtySize
 
 sandbox = Sandbox()
 
-terminal = sandbox.pty.create(
-    cols=80,
-    rows=24,
-    on_data=lambda data: print(data.decode(), end=''),  # end='' prevents extra newline
-)
+terminal = sandbox.pty.create(PtySize(rows=24, cols=80))
 
 # Kill the PTY
 killed = sandbox.pty.kill(terminal.pid)
@@ -306,7 +291,7 @@ const sandbox = await Sandbox.create()
 const terminal = await sandbox.pty.create({
   cols: 80,
   rows: 24,
-  onData: (data) => process.stdout.write(data),
+  onData: (data) => process.stdout.write(new TextDecoder().decode(data)),
 })
 
 // Send exit command
@@ -318,21 +303,17 @@ console.log('Exit code:', result.exitCode)
 ```
 
 ```python Python
-from e2b import Sandbox
+from e2b import Sandbox, PtySize
 
 sandbox = Sandbox()
 
-terminal = sandbox.pty.create(
-    cols=80,
-    rows=24,
-    on_data=lambda data: print(data.decode(), end=''),  # end='' prevents extra newline
-)
+terminal = sandbox.pty.create(PtySize(rows=24, cols=80))
 
 # Send exit command
 sandbox.pty.send_stdin(terminal.pid, b'exit\n')
 
-# Wait for the terminal to exit
-result = terminal.wait()
+# Wait for the terminal to exit, handling output via on_pty (data is bytes)
+result = terminal.wait(on_pty=lambda data: print(data.decode(), end=''))
 print('Exit code:', result.exit_code)
 ```
 </CodeGroup>