Add session forking capability (#5882)

Co-authored-by: Zane Staggs <zane@squareup.com>
2026-01-22 12:49:04 -05:00
parent f1a36b2bc3
commit 13bdff4bb5
13 changed files with 378 additions and 275 deletions
@@ -1285,14 +1285,23 @@ impl SessionStorage {
            )
            .await?;

-        session_manager
+        let mut builder = session_manager
            .update(&new_session.id)
            .extension_data(original_session.extension_data)
            .schedule_id(original_session.schedule_id)
            .recipe(original_session.recipe)
-            .user_recipe_values(original_session.user_recipe_values)
-            .apply()
-            .await?;
+            .user_recipe_values(original_session.user_recipe_values);
+
+        // Preserve provider and model config from original session
+        if let Some(provider_name) = original_session.provider_name {
+            builder = builder.provider_name(provider_name);
+        }
+
+        if let Some(model_config) = original_session.model_config {
+            builder = builder.model_config(model_config);
+        }
+
+        builder.apply().await?;

        if let Some(conversation) = original_session.conversation {
            self.replace_conversation(&new_session.id, &conversation)