feat: persist mode in forge.yaml (tailcallhq#793)

laststylebender14 · autofix-ci[bot] · tusharmath · web-flow · commit e3e4c85ab18c · 2025-05-13T13:16:17.000+05:30
Co-authored-by: autofix-ci[bot] &lt;114827586+autofix-ci[bot]@users.noreply.github.com&gt;
Co-authored-by: Tushar Mathur &lt;tusharmath@gmail.com&gt;
diff --git a/crates/forge_main/Cargo.toml b/crates/forge_main/Cargo.toml
@@ -43,6 +43,7 @@ update-informer = { version = "1.2.0", default-features = false, features = [
     "rustls-tls",
 ] }
 
+
 [dev-dependencies]
 insta.workspace = true
 pretty_assertions.workspace = true
diff --git a/crates/forge_main/src/state.rs b/crates/forge_main/src/state.rs
@@ -1,12 +1,11 @@
 use derive_setters::Setters;
 use forge_api::{ConversationId, Model, ModelId, Provider, Usage};
-use serde::Deserialize;
+use strum_macros::EnumString;
 
 use crate::prompt::ForgePrompt;
 
-// TODO: convert to a new type
-#[derive(Debug, Clone, Default, Deserialize)]
-#[serde(rename_all = "snake_case")]
+#[derive(Debug, Clone, Default, EnumString)]
+#[strum(ascii_case_insensitive)]
 pub enum Mode {
     Plan,
     #[default]
diff --git a/crates/forge_main/src/ui.rs b/crates/forge_main/src/ui.rs
@@ -1,3 +1,4 @@
+use std::str::FromStr;
 use std::sync::Arc;
 
 use anyhow::{Context, Result};
@@ -101,6 +102,15 @@ impl<F: API> UI<F> {
             self.api.upsert_conversation(conversation).await?;
         }
 
+        // Update the workflow with the new mode
+        self.api
+            .update_workflow(self.cli.workflow.as_deref(), |workflow| {
+                workflow
+                    .variables
+                    .insert("mode".to_string(), Value::from(mode.to_string()));
+            })
+            .await?;
+
         self.writeln(TitleFormat::action(format!(
             "Switched to '{}' mode (context cleared)",
             self.state.mode
@@ -398,8 +408,7 @@ impl<F: API> UI<F> {
                 let mode = workflow
                     .variables
                     .get("mode")
-                    .cloned()
-                    .and_then(|value| serde_json::from_value(value).ok())
+                    .and_then(|value| value.as_str().and_then(|m| Mode::from_str(m).ok()))
                     .unwrap_or(Mode::Act);
 
                 self.state = UIState::new(mode).provider(self.api.environment().provider);
@@ -418,7 +427,7 @@ impl<F: API> UI<F> {
                     self.api.upsert_conversation(conversation).await?;
                     Ok(conversation_id)
                 } else {
-                    let conversation = self.api.init_conversation(workflow.clone()).await?;
+                    let conversation = self.api.init_conversation(workflow).await?;
                     self.state.conversation_id = Some(conversation.id.clone());
                     self.update_model(conversation.main_model()?);
                     Ok(conversation.id)
