Add agent CLI flag for cloud runs (#9935)

## Summary
- Adds `--agent <UID>` to `warp agent run-cloud`.
- Sends the value as `agent_identity_uid`, matching the public agent run
API request field.
- Updates spawn request construction sites and tests for CLI parsing and
request serialization.

## Testing
- `git --no-pager diff --check`
- `cargo nextest run --manifest-path /workspace/warp/Cargo.toml -p
warp_cli agent_run_cloud_accepts_agent_flag`
- `CARGO_BUILD_JOBS=1 cargo check --manifest-path
/workspace/warp/Cargo.toml -p warp --lib`
- `CARGO_BUILD_JOBS=1 cargo nextest run --manifest-path
/workspace/warp/Cargo.toml -p warp
spawn_agent_request_serializes_agent_uid_as_agent_identity_uid`
- Not run: `cargo fmt --manifest-path /workspace/warp/Cargo.toml --all
--check` (`rustfmt` is not installed in the sandbox toolchain)

Co-Authored-By: Oz <oz-agent@warp.dev>

_Conversation:
https://staging.warp.dev/conversation/e3cb2fc1-34d8-43cf-b8cd-fd894946656a_
_Run:
https://oz.staging.warp.dev/runs/019dea24-d12b-719f-9ba4-bb9435d546df_
_This PR was generated with [Oz](https://warp.dev/oz)._

---------

Co-authored-by: Oz <oz-agent@warp.dev>

Author: bnavetta

app/src/ai/agent_sdk/ambient.rs

@@ -484,6 +484,7 @@ impl AmbientAgentRunner {
                     (_, true) => Some(false),
                     _ => None,
                 },
+                agent_identity_uid: args.agent_uid,
                 skill,
                 attachments,
                 interactive: None,

app/src/ai/agent_sdk/mcp_config_tests.rs

@@ -278,6 +278,7 @@ fn serializes_mcp_servers_as_object_not_string() {
         }),
         title: None,
         team: None,
+        agent_identity_uid: None,
         skill: None,
         attachments: vec![],
         interactive: None,

app/src/ai/ambient_agents/spawn_tests.rs

@@ -364,6 +364,7 @@ async fn poll_retries_transient_429_errors() {
         config: None,
         title: None,
         team: None,
+        agent_identity_uid: None,
         skill: None,
         attachments: vec![],
         interactive: None,
@@ -430,6 +431,7 @@ async fn poll_fails_on_permanent_http_error() {
         config: None,
         title: None,
         team: None,
+        agent_identity_uid: None,
         skill: None,
         attachments: vec![],
         interactive: None,
@@ -496,6 +498,7 @@ async fn poll_gives_up_after_max_transient_retries() {
         config: None,
         title: None,
         team: None,
+        agent_identity_uid: None,
         skill: None,
         attachments: vec![],
         interactive: None,
@@ -557,6 +560,7 @@ async fn poll_stops_on_terminal_failure_like_state() {
         config: None,
         title: None,
         team: None,
+        agent_identity_uid: None,
         skill: None,
         attachments: vec![],
         interactive: None,
@@ -701,6 +705,7 @@ async fn poll_for_session_join_info_waits_until_link_is_available() {
         config: None,
         title: None,
         team: None,
+        agent_identity_uid: None,
         skill: None,
         attachments: vec![],
         interactive: None,

app/src/pane_group/pane/terminal_pane.rs

@@ -1564,6 +1564,7 @@ fn launch_remote_child(
         referenced_attachments: vec![],
         conversation_id: None,
         initial_snapshot_token: None,
+        agent_identity_uid: None,
     };
 
     new_terminal_view.update(ctx, |terminal_view, ctx| {

app/src/server/server_api/ai.rs

@@ -213,6 +213,9 @@ pub struct SpawnAgentRequest {
     pub title: Option<String>,
     #[serde(skip_serializing_if = "Option::is_none")]
     pub team: Option<bool>,
+    /// Agent identity UID to use as the execution principal for the run.
+    #[serde(rename = "agent_identity_uid", skip_serializing_if = "Option::is_none")]
+    pub agent_identity_uid: Option<String>,
     /// Use a Claude-compatible skill as the base prompt.
     /// Format: "repo:skill_name" or just "skill_name".
     /// The skill is resolved at runtime in the agent environment.

app/src/server/server_api/ai_tests.rs

@@ -9,7 +9,7 @@ use super::{
     AgentMessageHeader, AgentRunEvent, AgentSource, AmbientAgentTaskState, Artifact,
     ArtifactDownloadResponse, ArtifactType, ExecutionLocation, ForkConversationResponse,
     ListRunsResponse, ReadAgentMessageResponse, RunFollowupRequest, RunSortBy, RunSortOrder,
-    TaskListFilter,
+    SpawnAgentRequest, TaskListFilter, UserQueryMode,
 };
 use crate::notebooks::NotebookId;
 
@@ -34,6 +34,34 @@ fn ambient_agent_headers_for_task_overrides_existing_cloud_agent_header() {
     );
 }
 
+#[test]
+fn spawn_agent_request_serializes_agent_uid_as_agent_identity_uid() {
+    let request = SpawnAgentRequest {
+        prompt: "hello".to_string(),
+        mode: UserQueryMode::Normal,
+        config: None,
+        title: None,
+        team: None,
+        agent_identity_uid: Some("agent_123".to_string()),
+        skill: None,
+        attachments: vec![],
+        interactive: None,
+        parent_run_id: None,
+        runtime_skills: vec![],
+        referenced_attachments: vec![],
+        conversation_id: None,
+        initial_snapshot_token: None,
+    };
+
+    let value = serde_json::to_value(&request).unwrap();
+
+    assert_eq!(
+        value.get("agent_identity_uid").and_then(|v| v.as_str()),
+        Some("agent_123")
+    );
+    assert!(value.get("agent_uid").is_none());
+}
+
 #[test]
 fn test_deserialize_file_artifact_download_response() {
     let json = r#"{

app/src/terminal/view/ambient_agent/model.rs

@@ -877,6 +877,7 @@ impl AmbientAgentViewModel {
             config,
             title: None,
             team: None,
+            agent_identity_uid: None,
             skill: None,
             attachments,
             interactive: None,
@@ -1328,6 +1329,7 @@ impl AmbientAgentViewModel {
             referenced_attachments: vec![],
             conversation_id: Some(forked_conversation_id),
             initial_snapshot_token,
+            agent_identity_uid: None,
         };
         self.spawn_agent_with_request(request, ctx);
     }

app/src/terminal/view_tests.rs

@@ -718,6 +718,7 @@ fn cloud_mode_dispatched_agent_inserts_queued_user_query() {
                             config: None,
                             title: None,
                             team: None,
+                            agent_identity_uid: None,
                             skill: None,
                             attachments: vec![],
                             interactive: None,

crates/warp_cli/src/agent.rs

@@ -438,6 +438,14 @@ pub struct RunCloudArgs {
     #[command(flatten)]
     pub scope: ObjectScope,
 
+    /// UID of the agent to execute this run as.
+    ///
+    /// This will apply the agent's configuration, such
+    /// as its skills and base model, and attribute
+    /// credit usage back to the agent.
+    #[arg(long = "agent", value_name = "UID")]
+    pub agent_uid: Option<String>,
+
     /// Where this job should be hosted. Setting "warp" runs it on Warp's infrastructure. Any other
     /// value is treated is a self-hosted job and the value will be matched with the self-hosted
     /// worker's name.

crates/warp_cli/src/lib_tests.rs

@@ -275,6 +275,29 @@ fn agent_run_cloud_accepts_model() {
     assert_eq!(run_args.model.model.as_deref(), Some("gpt-4o"));
 }
 
+#[test]
+fn agent_run_cloud_accepts_agent_flag() {
+    let args = Args::try_parse_from([
+        "warp",
+        "agent",
+        "run-cloud",
+        "--prompt",
+        "hello",
+        "--agent",
+        "agent_123",
+    ])
+    .unwrap();
+
+    let Some(Command::CommandLine(boxed_cmd)) = args.command else {
+        panic!("Expected `warp agent run-cloud` command");
+    };
+    let CliCommand::Agent(AgentCommand::RunCloud(run_args)) = boxed_cmd.as_ref() else {
+        panic!("Expected `warp agent run-cloud` command");
+    };
+
+    assert_eq!(run_args.agent_uid.as_deref(), Some("agent_123"));
+}
+
 #[test]
 fn agent_run_cloud_accepts_mcp() {
     let uuid = uuid::Uuid::parse_str("550e8400-e29b-41d4-a716-446655440000").unwrap();