daytona 0.182.0 → 0.184.0.alpha.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 061f7ac81675bdb4cd1fe06c69ca710cda155bbfb7ee7a9c4e3c9ee21d9b7ed7
-  data.tar.gz: cf7d7e25ed8dfdbe49276458e1d26602f7a38a29218218dd817706dd113bcd7e
+  metadata.gz: 51f6c85ff29165b3645e8752fede62f4f0e006b6bece9cd4541ad90be2960f93
+  data.tar.gz: 9178ec77c86d9c28f65baa3e0faf76f861f282930259dc2c44f0fbbfd2643a39
 SHA512:
-  metadata.gz: 78ffeff8eae2db105b97e81f039b6aff95e3c57f552b6e99ac99cac98dd1c23eb48cac61bb229857fac28de453b3e35d4e44f0d01fb2f1aa06c6f4b6cc698e96
-  data.tar.gz: edf6e665cc52a2b7a177157fb629d16c11a9e5b095cf6d3e7126fb75f5ece1fc67f7ea3eaf586bf7ddfeded74121e4d276aa18a0bb65cac04fe322b51bba81ff
+  metadata.gz: d7eb785153c1bbfe977cf0d00823cbcc0937baa52637c20001cab8237035dcb8c2a18fb2458f8fda6a1bc292a5fb1bec73d9af6853944205176713cb2438dfb8
+  data.tar.gz: 5883ceea8ae8d4ab9e593f4a55b31df562afce803d0119ed71f4958bc53d82b1ca73a3e64152b5c200c165597b64cd1933bbc9169d33cf25cf5b6dd3cc643850

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/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

@@ -90,9 +90,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.
@@ -178,6 +181,8 @@ module Daytona
       }.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'
@@ -262,6 +267,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.
@@ -114,8 +114,8 @@ module Daytona
     #   puts "Subdirectories: #{dirs.map(&:name).join(', ')}"
     def list_files(path)
       toolbox_api.list_files({ path: })
-    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.
@@ -150,8 +150,8 @@ module Daytona
       else
         file
       end
-    rescue StandardError => e
-      raise Sdk::Error, "Failed to download file: #{e.message}"
+    rescue *Sdk::API_ERROR_CLASSES => e
+      raise Sdk.wrap_error(e, 'Failed to download file')
     end
 
     # Downloads a single file from the Sandbox as a stream without buffering the entire
@@ -189,8 +189,8 @@ module Daytona
                                    timeout: timeout, on_progress: on_progress,
                                    cancel_event: cancel_event, &)
       nil
-    rescue StandardError => e
-      raise Sdk::Error, "Failed to download file: #{e.message}"
+    rescue *Sdk::API_ERROR_CLASSES, Sdk::Error => e
+      raise Sdk.wrap_error(e, 'Failed to download file')
     end
 
     # Uploads a file to the specified path in the Sandbox. If a file already exists at
@@ -229,8 +229,8 @@ module Daytona
           toolbox_api.upload_file(remote_path, file)
         end
       end
-    rescue StandardError => e
-      raise Sdk::Error, "Failed to upload file: #{e.message}"
+    rescue *Sdk::API_ERROR_CLASSES => e
+      raise Sdk.wrap_error(e, 'Failed to upload file')
     end
 
     # Streams +source+ to the Sandbox without buffering its contents in memory, with
@@ -259,8 +259,8 @@ 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
-      raise Sdk::Error, "Failed to upload file: #{e.message}"
+    rescue *Sdk::API_ERROR_CLASSES, Sdk::Error => e
+      raise Sdk.wrap_error(e, 'Failed to upload file')
     end
 
     # Uploads multiple files to the Sandbox. If files already exist at the destination paths,
@@ -280,8 +280,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 => e
+      raise Sdk.wrap_error(e, 'Failed to upload files')
     end
 
     # Searches for files containing a pattern, similar to the grep command.
@@ -301,8 +301,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.
@@ -325,8 +325,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.
@@ -358,8 +358,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.
@@ -394,8 +394,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
@@ -429,8 +429,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,