Plumb per-run AWS Bedrock flag (#50)

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

Author: IsaiahWitzke

internal/common/task_utils.go

@@ -59,6 +59,12 @@ func AugmentArgsForTask(task *types.Task, args []string, opts TaskAugmentOptions
 			}
 		}
 
+		if task.AgentConfigSnapshot.BedrockInferenceRole != nil {
+			if role := strings.TrimSpace(*task.AgentConfigSnapshot.BedrockInferenceRole); role != "" {
+				args = append(args, "--bedrock-inference-role", role)
+			}
+		}
+
 		// Pass harness setting if explicitly configured.
 		if task.AgentConfigSnapshot.Harness != nil && task.AgentConfigSnapshot.Harness.Type != nil {
 			if harness := strings.TrimSpace(*task.AgentConfigSnapshot.Harness.Type); harness != "" {

internal/common/task_utils_test.go

@@ -87,6 +87,35 @@ func TestAugmentArgsForTask_IdleOnCompletePrecedence(t *testing.T) {
 			opts:     TaskAugmentOptions{},
 			expected: []string{"agent", "run", "--model", "claude-sonnet-4", "--idle-on-complete", "12m"},
 		},
+		{
+			name: "passes --bedrock-inference-role when set",
+			task: &types.Task{
+				AgentConfigSnapshot: &types.AmbientAgentConfig{
+					ModelID:              strPtr("claude-sonnet-4"),
+					BedrockInferenceRole: strPtr("arn:aws:iam::123456789012:role/BedrockInference"),
+				},
+			},
+			opts: TaskAugmentOptions{},
+			expected: []string{
+				"agent",
+				"run",
+				"--model",
+				"claude-sonnet-4",
+				"--bedrock-inference-role",
+				"arn:aws:iam::123456789012:role/BedrockInference",
+				"--idle-on-complete",
+			},
+		},
+		{
+			name: "skips --bedrock-inference-role when role is empty",
+			task: &types.Task{
+				AgentConfigSnapshot: &types.AmbientAgentConfig{
+					BedrockInferenceRole: strPtr("   "),
+				},
+			},
+			opts:     TaskAugmentOptions{},
+			expected: []string{"agent", "run", "--idle-on-complete"},
+		},
 		{
 			name: "adds --share public:view when session_sharing.public_access is VIEWER",
 			task: &types.Task{

internal/types/messages.go

@@ -98,17 +98,18 @@ type SessionSharingConfig struct {
 
 // AmbientAgentConfig represents the agent configuration.
 type AmbientAgentConfig struct {
-	EnvironmentID      *string                    `json:"environment_id,omitempty"`
-	BasePrompt         *string                    `json:"base_prompt,omitempty"`
-	ModelID            *string                    `json:"model_id,omitempty"`
-	ProfileID          *string                    `json:"profile_id,omitempty"`
-	SkillSpec          *string                    `json:"skill_spec,omitempty"`
-	MCPServers         map[string]json.RawMessage `json:"mcp_servers,omitempty"`
-	ComputerUseEnabled *bool                      `json:"computer_use_enabled,omitempty"`
-	IdleTimeoutMinutes *int                       `json:"idle_timeout_minutes,omitempty"`
-	Harness            *Harness                   `json:"harness,omitempty"`
-	HarnessAuthSecrets *HarnessAuthSecrets        `json:"harness_auth_secrets,omitempty"`
-	SessionSharing     *SessionSharingConfig      `json:"session_sharing,omitempty"`
+	EnvironmentID        *string                    `json:"environment_id,omitempty"`
+	BasePrompt           *string                    `json:"base_prompt,omitempty"`
+	ModelID              *string                    `json:"model_id,omitempty"`
+	ProfileID            *string                    `json:"profile_id,omitempty"`
+	SkillSpec            *string                    `json:"skill_spec,omitempty"`
+	MCPServers           map[string]json.RawMessage `json:"mcp_servers,omitempty"`
+	ComputerUseEnabled   *bool                      `json:"computer_use_enabled,omitempty"`
+	IdleTimeoutMinutes   *int                       `json:"idle_timeout_minutes,omitempty"`
+	Harness              *Harness                   `json:"harness,omitempty"`
+	HarnessAuthSecrets   *HarnessAuthSecrets        `json:"harness_auth_secrets,omitempty"`
+	BedrockInferenceRole *string                    `json:"bedrock_inference_role,omitempty"`
+	SessionSharing       *SessionSharingConfig      `json:"session_sharing,omitempty"`
 }
 
 // Task represents an ambient agent job.