feat(mode): set current_mode from the last message

Author: sudo-tee

lua/opencode/core.lua

@@ -31,9 +31,11 @@ end
 
 function M.switch_session(session_id)
   local selected_session = session.get_by_id(session_id)
-  -- clear the model so it can be set by the session. If it doesn't get set
-  -- then core.get_model() will reset it to the default
+
   state.current_model = nil
+  state.current_mode = nil
+  M.ensure_current_mode()
+
   state.active_session = selected_session
   if state.windows then
     state.restore_points = {}
@@ -73,8 +75,11 @@ function M.open(opts)
   if opts.new_session then
     state.active_session = nil
     state.last_sent_context = nil
-    -- clear current_model here so it can be reset to the default (if one is set)
+
     state.current_model = nil
+    state.current_mode = nil
+    M.ensure_current_mode()
+
     state.active_session = M.create_new_session()
   else
     if not state.active_session then

lua/opencode/ui/renderer.lua

@@ -177,7 +177,7 @@ function M._render_full_session_data(session_data)
   end
 
   if set_mode_from_messages then
-    M._set_model_from_messages()
+    M._set_model_and_mode_from_messages()
   end
   M.scroll_to_bottom()
 
@@ -213,17 +213,22 @@ function M.on_emit_events_finished()
 end
 
 ---Find the most recently used model from the messages
-function M._set_model_from_messages()
+function M._set_model_and_mode_from_messages()
   if not state.messages then
     return
   end
 
   for i = #state.messages, 1, -1 do
     local message = state.messages[i]
 
-    if message and message.info and message.info.modelID and message.info.providerID then
-      state.current_model = message.info.providerID .. '/' .. message.info.modelID
-      return
+    if message and message.info then
+      if message.info.modelID and message.info.providerID then
+        state.current_model = message.info.providerID .. '/' .. message.info.modelID
+        if message.info.mode then
+          state.current_mode = message.info.mode
+        end
+        return
+      end
     end
   end