Inline create-API-key flow on orchestration cards (warpdotdev#11124)

Author: advait-m

app/src/ai/agent/conversation.rs

@@ -2491,6 +2491,7 @@ impl AIConversation {
                                         ctx.emit(
                                             BlocklistAIHistoryEvent::OrchestrationConfigUpdated {
                                                 conversation_id: self.id,
+                                                from_restore: false,
                                             },
                                         );
                                     }
@@ -2651,6 +2652,7 @@ impl AIConversation {
                             ) {
                                 ctx.emit(BlocklistAIHistoryEvent::OrchestrationConfigUpdated {
                                     conversation_id: self.id,
+                                    from_restore: false,
                                 });
                             }
                         }

app/src/ai/ai_document_view.rs

@@ -292,17 +292,15 @@ impl AIDocumentView {
                     }
                     BlocklistAIHistoryEvent::OrchestrationConfigUpdated {
                         conversation_id: cid,
+                        from_restore,
                     } => {
-                        // Re-render so the config block picks up changes
-                        // only for our document's conversation.
                         let our_conv = AIDocumentModel::as_ref(ctx)
                             .get_conversation_id_for_document_id(&document_id);
                         if our_conv.as_ref() == Some(cid) {
-                            // Lazily create the config block view if it
-                            // wasn't available at construction time (the
-                            // plan sidebar can open before the server
-                            // sends the orchestration config).
-                            if me.orchestration_config_block.is_none() {
+                            // Lazily create the config block view if the
+                            // plan sidebar opened before the orchestration
+                            // config arrived.
+                            let was_freshly_created = if me.orchestration_config_block.is_none() {
                                 let conv_id = *cid;
                                 // TODO: introduce DocumentId / PlanId newtypes to make this
                                 // conversion type-safe.
@@ -311,6 +309,18 @@ impl AIDocumentView {
                                     Some(ctx.add_typed_action_view(move |ctx| {
                                         OrchestrationConfigBlockView::new(conv_id, plan_id, ctx)
                                     }));
+                                true
+                            } else {
+                                false
+                            };
+                            // Arm auto-pop for live agent dispatches but
+                            // not for restore-hydrated events.
+                            if was_freshly_created && !*from_restore {
+                                if let Some(block) = &me.orchestration_config_block {
+                                    block.update(ctx, |block, ctx| {
+                                        block.arm_for_fresh_dispatch(ctx);
+                                    });
+                                }
                             }
                             ctx.notify();
                         }

app/src/ai/blocklist/history_model.rs

@@ -2402,10 +2402,12 @@ pub enum BlocklistAIHistoryEvent {
         conversation_id: AIConversationId,
     },
 
-    /// Emitted when a conversation's orchestration config is updated reactively
-    /// from an incoming `OrchestrationConfigSnapshot` message.
+    /// Emitted when a conversation's orchestration config is updated
+    /// (live wire snapshot, user edit, or restore-hydration).
+    /// Consumers that perform UI side effects should gate on `!from_restore`.
     OrchestrationConfigUpdated {
         conversation_id: AIConversationId,
+        from_restore: bool,
     },
 
     /// Emitted when a conversation's `conversation_usage_metadata` is updated