daytona 0.200.1 → 0.201.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 714770a5cd03bbfb27825fd9c018333d0377db70c0ce32b7ed484ff62946b2da
-  data.tar.gz: 1ea5e87028fffadb41659ceb7d76e6ec635ce2801e27f53bf1319d02b66ea6a7
+  metadata.gz: 684e4304ffae6f21d39f795fbcd75754e0148adf43a86963e3c8303b6b9eb87e
+  data.tar.gz: 3d4dc419e7b759cce74db9df2cec0bb36900493c39d08b5640b04dd5c4863b11
 SHA512:
-  metadata.gz: 9f8bec4d5a367c8c1f2cc48bab297903dd2b75f03460c0c72728842a639da4dd00c781ec5668f64aef5375af1955f3e1cb6221577dfa4d3fd237c97fbb6056e2
-  data.tar.gz: 19a3b2b8bafb2c5a9fc14777411de38abe6a6ac17b1c0d132af9334142ab915ef3aa325aa8184e1ac999df7e0c867a47e8d3691975b09be3ca6a98a80584b8b8
+  metadata.gz: 59a7fea3c20c801d9e3d09bad69f6a4318f5b2ae64d397e2848ee893e8f5e1117c973f626ba825fce6bb4604f2875e1f4d393391b0c7dc8b850459a7d7073bd1
+  data.tar.gz: ade5a786482e476b4e3baf42eea6828e0d422ec83e1dc1363130165a8ae09b9a5e0d99be687848d5c50f6e4c0ce09342244ddfa1553d9effbff882ad3bd58267

data/lib/daytona/code_interpreter.rb

@@ -232,8 +232,8 @@ module Daytona
     def create_context(cwd: nil)
       request = DaytonaToolboxApiClient::CreateContextRequest.new(cwd:)
       @toolbox_api.create_interpreter_context(request)
-    rescue StandardError => e
-      raise Sdk::Error, "Failed to create interpreter context: #{e.message}"
+    rescue *Sdk::API_ERROR_CLASSES => e
+      raise Sdk.wrap_error(e, 'Failed to create interpreter context')
     end
 
     # List all user-created interpreter contexts.
@@ -252,8 +252,8 @@ module Daytona
     def list_contexts
       response = @toolbox_api.list_interpreter_contexts
       response.contexts || []
-    rescue StandardError => e
-      raise Sdk::Error, "Failed to list interpreter contexts: #{e.message}"
+    rescue *Sdk::API_ERROR_CLASSES => e
+      raise Sdk.wrap_error(e, 'Failed to list interpreter contexts')
     end
 
     # Delete an interpreter context and shut down all associated processes.
@@ -272,8 +272,8 @@ module Daytona
     def delete_context(context)
       @toolbox_api.delete_interpreter_context(context.id)
       nil
-    rescue StandardError => e
-      raise Sdk::Error, "Failed to delete interpreter context: #{e.message}"
+    rescue *Sdk::API_ERROR_CLASSES => e
+      raise Sdk.wrap_error(e, 'Failed to delete interpreter context')
     end
 
     instrument :run_code, :create_context, :list_contexts, :delete_context,

data/lib/daytona/common/response.rb

@@ -89,6 +89,8 @@ module Daytona
     :created_at_before,
     :last_activity_after,
     :last_activity_before,
+    :auto_destroy_at_after,
+    :auto_destroy_at_before,
     :sort,
     :order
   ) do

data/lib/daytona/computer_use.rb

@@ -33,8 +33,8 @@ module Daytona
       #   puts "Mouse is at: #{position.x}, #{position.y}"
       def position
         toolbox_api.get_mouse_position
-      rescue StandardError => e
-        raise Sdk::Error, "Failed to get mouse position: #{e.message}"
+      rescue *Sdk::API_ERROR_CLASSES => e
+        raise Sdk.wrap_error(e, 'Failed to get mouse position')
       end
 
       # Moves the mouse cursor to the specified coordinates.
@@ -50,8 +50,8 @@ module Daytona
       def move(x:, y:)
         request = DaytonaToolboxApiClient::MouseMoveRequest.new(x:, y:)
         toolbox_api.move_mouse(request)
-      rescue StandardError => e
-        raise Sdk::Error, "Failed to move mouse: #{e.message}"
+      rescue *Sdk::API_ERROR_CLASSES => e
+        raise Sdk.wrap_error(e, 'Failed to move mouse')
       end
 
       # Clicks the mouse at the specified coordinates.
@@ -75,8 +75,8 @@ module Daytona
       def click(x:, y:, button: 'left', double: false)
         request = DaytonaToolboxApiClient::MouseClickRequest.new(x:, y:, button:, double:)
         toolbox_api.click(request)
-      rescue StandardError => e
-        raise Sdk::Error, "Failed to click mouse: #{e.message}"
+      rescue *Sdk::API_ERROR_CLASSES => e
+        raise Sdk.wrap_error(e, 'Failed to click mouse')
       end
 
       # Drags the mouse from start coordinates to end coordinates.
@@ -95,8 +95,8 @@ module Daytona
       def drag(start_x:, start_y:, end_x:, end_y:, button: 'left')
         request = DaytonaToolboxApiClient::MouseDragRequest.new(start_x:, start_y:, end_x:, end_y:, button:)
         toolbox_api.drag(request)
-      rescue StandardError => e
-        raise Sdk::Error, "Failed to drag mouse: #{e.message}"
+      rescue *Sdk::API_ERROR_CLASSES => e
+        raise Sdk.wrap_error(e, 'Failed to drag mouse')
       end
 
       # Scrolls the mouse wheel at the specified coordinates.
@@ -118,8 +118,8 @@ module Daytona
         request = DaytonaToolboxApiClient::MouseScrollRequest.new(x:, y:, direction:, amount:)
         toolbox_api.scroll(request)
         true
-      rescue StandardError => e
-        raise Sdk::Error, "Failed to scroll mouse: #{e.message}"
+      rescue *Sdk::API_ERROR_CLASSES => e
+        raise Sdk.wrap_error(e, 'Failed to scroll mouse')
       end
 
       instrument :position, :move, :click, :drag, :scroll, component: 'Mouse'
@@ -164,8 +164,8 @@ module Daytona
       def type(text:, delay: nil)
         request = DaytonaToolboxApiClient::KeyboardTypeRequest.new(text:, delay:)
         toolbox_api.type_text(request)
-      rescue StandardError => e
-        raise Sdk::Error, "Failed to type text: #{e.message}"
+      rescue *Sdk::API_ERROR_CLASSES => e
+        raise Sdk.wrap_error(e, 'Failed to type text')
       end
 
       # Presses a key with optional modifiers.
@@ -187,8 +187,8 @@ module Daytona
       def press(key:, modifiers: nil)
         request = DaytonaToolboxApiClient::KeyboardPressRequest.new(key:, modifiers: modifiers || [])
         toolbox_api.press_key(request)
-      rescue StandardError => e
-        raise Sdk::Error, "Failed to press key: #{e.message}"
+      rescue *Sdk::API_ERROR_CLASSES => e
+        raise Sdk.wrap_error(e, 'Failed to press key')
       end
 
       # Presses a hotkey combination.
@@ -209,8 +209,8 @@ module Daytona
       def hotkey(keys:)
         request = DaytonaToolboxApiClient::KeyboardHotkeyRequest.new(keys:)
         toolbox_api.press_hotkey(request)
-      rescue StandardError => e
-        raise Sdk::Error, "Failed to press hotkey: #{e.message}"
+      rescue *Sdk::API_ERROR_CLASSES => e
+        raise Sdk.wrap_error(e, 'Failed to press hotkey')
       end
 
       instrument :type, :press, :hotkey, component: 'Keyboard'
@@ -254,8 +254,8 @@ module Daytona
       #   with_cursor = sandbox.computer_use.screenshot.take_full_screen(show_cursor: true)
       def take_full_screen(show_cursor: false)
         toolbox_api.take_screenshot(show_cursor:)
-      rescue StandardError => e
-        raise Sdk::Error, "Failed to take screenshot: #{e.message}"
+      rescue *Sdk::API_ERROR_CLASSES => e
+        raise Sdk.wrap_error(e, 'Failed to take screenshot')
       end
 
       # Takes a screenshot of a specific region.
@@ -271,8 +271,8 @@ module Daytona
       #   puts "Captured region: #{screenshot.region.width}x#{screenshot.region.height}"
       def take_region(region:, show_cursor: false)
         toolbox_api.take_region_screenshot(region.height, region.width, region.y, region.x, show_cursor:)
-      rescue StandardError => e
-        raise Sdk::Error, "Failed to take region screenshot: #{e.message}"
+      rescue *Sdk::API_ERROR_CLASSES => e
+        raise Sdk.wrap_error(e, 'Failed to take region screenshot')
       end
 
       # Takes a compressed screenshot of the entire screen.
@@ -303,8 +303,8 @@ module Daytona
           format: options.fmt,
           show_cursor: options.show_cursor
         )
-      rescue StandardError => e
-        raise Sdk::Error, "Failed to take compressed screenshot: #{e.message}"
+      rescue *Sdk::API_ERROR_CLASSES => e
+        raise Sdk.wrap_error(e, 'Failed to take compressed screenshot')
       end
 
       # Takes a compressed screenshot of a specific region.
@@ -334,8 +334,8 @@ module Daytona
           format: options.fmt,
           show_cursor: options.show_cursor
         )
-      rescue StandardError => e
-        raise Sdk::Error, "Failed to take compressed region screenshot: #{e.message}"
+      rescue *Sdk::API_ERROR_CLASSES => e
+        raise Sdk.wrap_error(e, 'Failed to take compressed region screenshot')
       end
 
       instrument :take_full_screen, :take_region, :take_compressed, :take_compressed_region,
@@ -380,8 +380,8 @@ module Daytona
       #   end
       def info
         toolbox_api.get_display_info
-      rescue StandardError => e
-        raise Sdk::Error, "Failed to get display info: #{e.message}"
+      rescue *Sdk::API_ERROR_CLASSES => e
+        raise Sdk.wrap_error(e, 'Failed to get display info')
       end
 
       # Gets the list of open windows.
@@ -397,8 +397,8 @@ module Daytona
       #   end
       def windows
         toolbox_api.get_windows
-      rescue StandardError => e
-        raise Sdk::Error, "Failed to get windows: #{e.message}"
+      rescue *Sdk::API_ERROR_CLASSES => e
+        raise Sdk.wrap_error(e, 'Failed to get windows')
       end
 
       instrument :info, :windows, component: 'Display'
@@ -446,8 +446,8 @@ module Daytona
         opts[:max_depth] = max_depth unless max_depth.nil?
 
         toolbox_api.get_accessibility_tree(opts)
-      rescue StandardError => e
-        raise Sdk::Error, "Failed to get accessibility tree: #{e.message}"
+      rescue *Sdk::API_ERROR_CLASSES => e
+        raise Sdk.wrap_error(e, 'Failed to get accessibility tree')
       end
 
       # Finds AT-SPI accessibility nodes matching the provided filters.
@@ -482,8 +482,8 @@ module Daytona
 
         request = DaytonaToolboxApiClient::FindAccessibilityNodesRequest.new(attrs)
         toolbox_api.find_accessibility_nodes(request)
-      rescue StandardError => e
-        raise Sdk::Error, "Failed to find accessibility nodes: #{e.message}"
+      rescue *Sdk::API_ERROR_CLASSES => e
+        raise Sdk.wrap_error(e, 'Failed to find accessibility nodes')
       end
 
       # Focuses an AT-SPI accessibility node.
@@ -496,8 +496,8 @@ module Daytona
       def focus_node(id:)
         request = DaytonaToolboxApiClient::AccessibilityNodeRequest.new(id:)
         toolbox_api.focus_accessibility_node(request)
-      rescue StandardError => e
-        raise Sdk::Error, "Failed to focus accessibility node: #{e.message}"
+      rescue *Sdk::API_ERROR_CLASSES => e
+        raise Sdk.wrap_error(e, 'Failed to focus accessibility node')
       end
 
       # Invokes an AT-SPI accessibility node action.
@@ -514,8 +514,8 @@ module Daytona
 
         request = DaytonaToolboxApiClient::AccessibilityInvokeRequest.new(attrs)
         toolbox_api.invoke_accessibility_node(request)
-      rescue StandardError => e
-        raise Sdk::Error, "Failed to invoke accessibility node: #{e.message}"
+      rescue *Sdk::API_ERROR_CLASSES => e
+        raise Sdk.wrap_error(e, 'Failed to invoke accessibility node')
       end
 
       # Sets an AT-SPI accessibility node value.
@@ -529,8 +529,8 @@ module Daytona
       def set_node_value(id:, value:)
         request = DaytonaToolboxApiClient::AccessibilitySetValueRequest.new(id:, value:)
         toolbox_api.set_accessibility_node_value(request)
-      rescue StandardError => e
-        raise Sdk::Error, "Failed to set accessibility node value: #{e.message}"
+      rescue *Sdk::API_ERROR_CLASSES => e
+        raise Sdk.wrap_error(e, 'Failed to set accessibility node value')
       end
 
       instrument :get_tree, :find_nodes, :focus_node, :invoke_node, :set_node_value,
@@ -627,8 +627,8 @@ module Daytona
       def start(label: nil)
         request = DaytonaToolboxApiClient::StartRecordingRequest.new(label:)
         toolbox_api.start_recording(request: request)
-      rescue StandardError => e
-        raise Sdk::Error, "Failed to start recording: #{e.message}"
+      rescue *Sdk::API_ERROR_CLASSES => e
+        raise Sdk.wrap_error(e, 'Failed to start recording')
       end
 
       # Stops an active screen recording session.
@@ -644,8 +644,8 @@ module Daytona
       def stop(id:)
         request = DaytonaToolboxApiClient::StopRecordingRequest.new(id: id)
         toolbox_api.stop_recording(request)
-      rescue StandardError => e
-        raise Sdk::Error, "Failed to stop recording: #{e.message}"
+      rescue *Sdk::API_ERROR_CLASSES => e
+        raise Sdk.wrap_error(e, 'Failed to stop recording')
       end
 
       # Lists all recordings (active and completed).
@@ -661,8 +661,8 @@ module Daytona
       #   end
       def list
         toolbox_api.list_recordings
-      rescue StandardError => e
-        raise Sdk::Error, "Failed to list recordings: #{e.message}"
+      rescue *Sdk::API_ERROR_CLASSES => e
+        raise Sdk.wrap_error(e, 'Failed to list recordings')
       end
 
       # Gets details of a specific recording by ID.
@@ -678,8 +678,8 @@ module Daytona
       #   puts "Duration: #{recording.duration_seconds} seconds"
       def get(id:)
         toolbox_api.get_recording(id)
-      rescue StandardError => e
-        raise Sdk::Error, "Failed to get recording: #{e.message}"
+      rescue *Sdk::API_ERROR_CLASSES => e
+        raise Sdk.wrap_error(e, 'Failed to get recording')
       end
 
       # Deletes a recording by ID.
@@ -693,8 +693,8 @@ module Daytona
       #   puts "Recording deleted"
       def delete(id:)
         toolbox_api.delete_recording(id)
-      rescue StandardError => e
-        raise Sdk::Error, "Failed to delete recording: #{e.message}"
+      rescue *Sdk::API_ERROR_CLASSES => e
+        raise Sdk.wrap_error(e, 'Failed to delete recording')
       end
 
       # Downloads a recording file and saves it to a local path.
@@ -815,8 +815,8 @@ module Daytona
     #   puts "Computer use processes started: #{result.message}"
     def start
       toolbox_api.start_computer_use
-    rescue StandardError => e
-      raise Sdk::Error, "Failed to start computer use: #{e.message}"
+    rescue *Sdk::API_ERROR_CLASSES => e
+      raise Sdk.wrap_error(e, 'Failed to start computer use')
     end
 
     # Stops all computer use processes.
@@ -829,8 +829,8 @@ module Daytona
     #   puts "Computer use processes stopped: #{result.message}"
     def stop
       toolbox_api.stop_computer_use
-    rescue StandardError => e
-      raise Sdk::Error, "Failed to stop computer use: #{e.message}"
+    rescue *Sdk::API_ERROR_CLASSES => e
+      raise Sdk.wrap_error(e, 'Failed to stop computer use')
     end
 
     # Gets the status of all computer use processes.
@@ -843,8 +843,8 @@ module Daytona
     #   puts "Computer use status: #{response.status}"
     def status
       toolbox_api.get_computer_use_status
-    rescue StandardError => e
-      raise Sdk::Error, "Failed to get computer use status: #{e.message}"
+    rescue *Sdk::API_ERROR_CLASSES => e
+      raise Sdk.wrap_error(e, 'Failed to get computer use status')
     end
 
     # Gets the status of a specific VNC process.
@@ -858,8 +858,8 @@ module Daytona
     #   no_vnc_status = sandbox.computer_use.get_process_status("novnc")
     def get_process_status(process_name:)
       toolbox_api.get_process_status(process_name, sandbox_id)
-    rescue StandardError => e
-      raise Sdk::Error, "Failed to get process status: #{e.message}"
+    rescue *Sdk::API_ERROR_CLASSES => e
+      raise Sdk.wrap_error(e, 'Failed to get process status')
     end
 
     # Restarts a specific VNC process.
@@ -873,8 +873,8 @@ module Daytona
     #   puts "XFCE4 process restarted: #{result.message}"
     def restart_process(process_name:)
       toolbox_api.restart_process(process_name, sandbox_id)
-    rescue StandardError => e
-      raise Sdk::Error, "Failed to restart process: #{e.message}"
+    rescue *Sdk::API_ERROR_CLASSES => e
+      raise Sdk.wrap_error(e, 'Failed to restart process')
     end
 
     # Gets logs for a specific VNC process.
@@ -888,8 +888,8 @@ module Daytona
     #   puts "NoVNC logs: #{logs}"
     def get_process_logs(process_name:)
       toolbox_api.get_process_logs(process_name, sandbox_id)
-    rescue StandardError => e
-      raise Sdk::Error, "Failed to get process logs: #{e.message}"
+    rescue *Sdk::API_ERROR_CLASSES => e
+      raise Sdk.wrap_error(e, 'Failed to get process logs')
     end
 
     # Gets error logs for a specific VNC process.
@@ -903,8 +903,8 @@ module Daytona
     #   puts "X11VNC errors: #{errors}"
     def get_process_errors(process_name:)
       toolbox_api.get_process_errors(process_name, sandbox_id)
-    rescue StandardError => e
-      raise Sdk::Error, "Failed to get process errors: #{e.message}"
+    rescue *Sdk::API_ERROR_CLASSES => e
+      raise Sdk.wrap_error(e, 'Failed to get process errors')
     end
 
     instrument :start, :stop, :status, :get_process_status, :restart_process,

data/lib/daytona/daytona.rb

@@ -127,9 +127,12 @@ module Daytona
     #
     # @param id [String]
     # @return [Daytona::Sandbox]
+    # @raise [Daytona::Sdk::Error]
     def get(id)
       sandbox_dto = sandbox_api.get_sandbox(id)
       to_sandbox(sandbox_dto:)
+    rescue *Sdk::API_ERROR_CLASSES => e
+      raise Sdk.wrap_error(e, "Failed to get sandbox #{id}")
     end
 
     # Iterates over Sandboxes matching the given query.
@@ -222,11 +225,15 @@ module Daytona
         created_at_before: q.created_at_before,
         last_event_after: q.last_activity_after,
         last_event_before: q.last_activity_before,
+        auto_destroy_at_after: q.auto_destroy_at_after,
+        auto_destroy_at_before: q.auto_destroy_at_before,
         sort: q.sort,
         order: q.order
       }.compact
 
       sandbox_api.list_sandboxes(opts)
+    rescue *Sdk::API_ERROR_CLASSES => e
+      raise Sdk.wrap_error(e, 'Failed to list sandboxes')
     end
 
     instrument :fetch_sandbox_page, component: 'Daytona.list'
@@ -334,6 +341,8 @@ module Daytona
       end
 
       sandbox
+    rescue *Sdk::API_ERROR_CLASSES => e
+      raise Sdk.wrap_error(e, 'Failed to create sandbox')
     end
 
     # @return [void]

data/lib/daytona/file_system.rb

@@ -46,8 +46,8 @@ module Daytona
     def create_folder(path, mode)
       Sdk.logger.debug("Creating folder #{path} with mode #{mode}")
       toolbox_api.create_folder(path, mode)
-    rescue StandardError => e
-      raise Sdk::Error, "Failed to create folder: #{e.message}"
+    rescue *Sdk::API_ERROR_CLASSES => e
+      raise Sdk.wrap_error(e, 'Failed to create folder')
     end
 
     # Deletes a file from the Sandbox.
@@ -65,8 +65,8 @@ module Daytona
     #   sandbox.fs.delete_file("workspace/old_dir", recursive: true)
     def delete_file(path, recursive: false)
       toolbox_api.delete_file(path, { recursive: })
-    rescue StandardError => e
-      raise Sdk::Error, "Failed to delete file: #{e.message}"
+    rescue *Sdk::API_ERROR_CLASSES => e
+      raise Sdk.wrap_error(e, 'Failed to delete file')
     end
 
     # Gets detailed information about a file or directory, including its
@@ -89,8 +89,8 @@ module Daytona
     #   puts "Path is a directory" if info.is_dir
     def get_file_info(path)
       toolbox_api.get_file_info(path)
-    rescue StandardError => e
-      raise Sdk::Error, "Failed to get file info: #{e.message}"
+    rescue *Sdk::API_ERROR_CLASSES => e
+      raise Sdk.wrap_error(e, 'Failed to get file info')
     end
 
     # Lists files and directories in a given path and returns their information, similar to the ls -l command.
@@ -123,8 +123,8 @@ module Daytona
       toolbox_api.list_files({ path:, depth: }.compact)
     rescue Sdk::Error
       raise
-    rescue StandardError => e
-      raise Sdk::Error, "Failed to list files: #{e.message}"
+    rescue *Sdk::API_ERROR_CLASSES => e
+      raise Sdk.wrap_error(e, 'Failed to list files')
     end
 
     # Downloads a file from the Sandbox. Returns the file contents as a string.
@@ -159,7 +159,9 @@ module Daytona
       else
         file
       end
-    rescue StandardError => e
+    rescue *Sdk::API_ERROR_CLASSES => e
+      raise Sdk.wrap_error(e, 'Failed to download file')
+    rescue IOError, SystemCallError => e
       raise Sdk::Error, "Failed to download file: #{e.message}"
     end
 
@@ -198,7 +200,9 @@ module Daytona
                                    timeout: timeout, on_progress: on_progress,
                                    cancel_event: cancel_event, &)
       nil
-    rescue StandardError => e
+    rescue *Sdk::API_ERROR_CLASSES, Sdk::Error => e
+      raise Sdk.wrap_error(e, 'Failed to download file')
+    rescue IOError, SystemCallError => e
       raise Sdk::Error, "Failed to download file: #{e.message}"
     end
 
@@ -222,13 +226,13 @@ module Daytona
     #   # Upload binary data
     #   data = { key: "value" }.to_json
     #   sandbox.fs.upload_file(data, "tmp/config.json")
-    def upload_file(source, remote_path)
+    def upload_file(source, remote_path) # rubocop:disable Metrics/AbcSize, Metrics/MethodLength
       if source.is_a?(String) && File.exist?(source)
         # Source is a file path
-        File.open(source, 'rb') { |file| toolbox_api.upload_file(remote_path, file) }
+        File.open(source, 'rb') { |file| toolbox_api.upload_file(remote_path, file: file) }
       elsif source.respond_to?(:read)
         # Source is an IO object
-        toolbox_api.upload_file(remote_path, source)
+        toolbox_api.upload_file(remote_path, file: source)
       else
         # Tempfile.create yields a ::File (works with Typhoeus) but deletes
         # on block exit — too early if curl reads asynchronously. Write via
@@ -238,12 +242,14 @@ module Daytona
           tmp.binmode
           tmp.write(source.to_s.b)
           tmp.close
-          File.open(tmp.path, 'rb') { |file| toolbox_api.upload_file(remote_path, file) }
+          File.open(tmp.path, 'rb') { |file| toolbox_api.upload_file(remote_path, file: file) }
         ensure
           tmp.unlink
         end
       end
-    rescue StandardError => e
+    rescue *Sdk::API_ERROR_CLASSES => e
+      raise Sdk.wrap_error(e, 'Failed to upload file')
+    rescue IOError, SystemCallError => e
       raise Sdk::Error, "Failed to upload file: #{e.message}"
     end
 
@@ -273,7 +279,9 @@ module Daytona
       FileTransfer.stream_upload(api_client: toolbox_api.api_client, remote_path: remote_path,
                                  source: source, timeout: timeout, on_progress: on_progress,
                                  cancel_event: cancel_event)
-    rescue StandardError => e
+    rescue *Sdk::API_ERROR_CLASSES, Sdk::Error => e
+      raise Sdk.wrap_error(e, 'Failed to upload file')
+    rescue IOError, SystemCallError => e
       raise Sdk::Error, "Failed to upload file: #{e.message}"
     end
 
@@ -294,8 +302,8 @@ module Daytona
     #   sandbox.fs.upload_files(files)
     def upload_files(files)
       files.each { |file_upload| upload_file(file_upload.source, file_upload.destination) }
-    rescue StandardError => e
-      raise Sdk::Error, "Failed to upload files: #{e.message}"
+    rescue *Sdk::API_ERROR_CLASSES, Sdk::Error => e
+      raise Sdk.wrap_error(e, 'Failed to upload files')
     end
 
     # Searches for files containing a pattern, similar to the grep command.
@@ -315,8 +323,8 @@ module Daytona
     #   end
     def find_files(path, pattern)
       toolbox_api.find_in_files(path, pattern)
-    rescue StandardError => e
-      raise Sdk::Error, "Failed to find files: #{e.message}"
+    rescue *Sdk::API_ERROR_CLASSES => e
+      raise Sdk.wrap_error(e, 'Failed to find files')
     end
 
     # Searches for files and directories whose names match the specified pattern.
@@ -339,8 +347,8 @@ module Daytona
     #   puts "Found #{result.files.length} test files"
     def search_files(path, pattern)
       toolbox_api.search_files(path, pattern)
-    rescue StandardError => e
-      raise Sdk::Error, "Failed to search files: #{e.message}"
+    rescue *Sdk::API_ERROR_CLASSES => e
+      raise Sdk.wrap_error(e, 'Failed to search files')
     end
 
     # Moves or renames a file or directory. The parent directory of the destination must exist.
@@ -372,8 +380,8 @@ module Daytona
     #   )
     def move_files(source, destination)
       toolbox_api.move_file(source, destination)
-    rescue StandardError => e
-      raise Sdk::Error, "Failed to move files: #{e.message}"
+    rescue *Sdk::API_ERROR_CLASSES => e
+      raise Sdk.wrap_error(e, 'Failed to move files')
     end
 
     # Performs search and replace operations across multiple files.
@@ -408,8 +416,8 @@ module Daytona
         new_value: new_value
       )
       toolbox_api.replace_in_files(replace_request)
-    rescue StandardError => e
-      raise Sdk::Error, "Failed to replace in files: #{e.message}"
+    rescue *Sdk::API_ERROR_CLASSES => e
+      raise Sdk.wrap_error(e, 'Failed to replace in files')
     end
 
     # Sets permissions and ownership for a file or directory. Any of the parameters can be nil
@@ -443,8 +451,8 @@ module Daytona
       opts[:group] = group if group
 
       toolbox_api.set_file_permissions(path, opts)
-    rescue StandardError => e
-      raise Sdk::Error, "Failed to set file permissions: #{e.message}"
+    rescue *Sdk::API_ERROR_CLASSES => e
+      raise Sdk.wrap_error(e, 'Failed to set file permissions')
     end
 
     instrument :create_folder, :delete_file, :get_file_info, :list_files, :download_file,