diff --git a/Cargo.toml b/Cargo.toml
index 598784b..3ad65fb 100644
--- a/Cargo.toml
+++ b/Cargo.toml
@@ -10,4 +10,4 @@ path = "src/dart.rs"
 crate-type = ["cdylib"]
 
 [dependencies]
-zed_extension_api = "0.7.0"
+zed_extension_api = "0.8.0"
diff --git a/src/dart.rs b/src/dart.rs
index 66e265c..edff770 100644
--- a/src/dart.rs
+++ b/src/dart.rs
@@ -3,7 +3,8 @@ use zed::settings::LspSettings;
 use zed::{CodeLabel, CodeLabelSpan};
 use zed_extension_api::serde_json::json;
 use zed_extension_api::{
-    self as zed, current_platform, serde_json, DebugAdapterBinary, DebugTaskDefinition, Os, Result,
+    self as zed, current_platform, serde_json, DapCustomAction, DapCustomActionIcon,
+    DapCustomActionTrigger, DebugAdapterBinary, DebugTaskDefinition, Os, Result,
     StartDebuggingRequestArguments, StartDebuggingRequestArgumentsRequest, Worktree,
 };
 
@@ -138,6 +139,12 @@ impl zed::Extension for DartExtension {
 
         let vm_service_uri = user_config.get("vmServiceUri").and_then(|v| v.as_str());
 
+        let tool_args = if debug_mode == "flutter" {
+            vec!["-d".to_string(), device_id.to_string()]
+        } else {
+            vec![]
+        };
+
         let config_json = json!({
             "type": tool,
             "request": request,
@@ -145,6 +152,7 @@ impl zed::Extension for DartExtension {
             "program": program,
             "cwd": cwd.clone().unwrap_or_default(),
             "args": args,
+            "toolArgs": tool_args,
             "flutterMode": "debug",
             "deviceId": device_id,
             "platform": platform,
@@ -217,6 +225,29 @@ impl zed::Extension for DartExtension {
         })))
     }
 
+    fn get_dap_custom_actions(&self, adapter_name: String) -> Vec<DapCustomAction> {
+        if adapter_name != "Flutter" && adapter_name != "Dart" {
+            return Vec::new();
+        }
+
+        vec![
+            DapCustomAction {
+                label: "Hot Reload".to_string(),
+                command: "hotReload".to_string(),
+                arguments: "{}".to_string(),
+                trigger: DapCustomActionTrigger::Both,
+                icon: DapCustomActionIcon::Flame,
+            },
+            DapCustomAction {
+                label: "Hot Restart".to_string(),
+                command: "hotRestart".to_string(),
+                arguments: "{}".to_string(),
+                trigger: DapCustomActionTrigger::Toolbar,
+                icon: DapCustomActionIcon::RotateCw,
+            },
+        ]
+    }
+
     fn label_for_completion(
         &self,
         _language_server_id: &zed::LanguageServerId,