Files
clawmates/crates/cm-api/src/routes/teams.rs
T
Omar Sobh ac8c689f50
ci / gates (push) Successful in 6s
ci / frontend (push) Successful in 37s
ci / rust (push) Successful in 2m58s
ci / e2e (push) Skipped
ci / publish (push) Successful in 2m53s
logs + team parity: pretty step/container renderers; quota + audit on team creation
Two bundled changes:

── LiveRunLogs prettification ──────────────────────────────────
The Steps + Container tabs were plain mono lines with a single
color per event. Now they get structured layout:

Steps:
- Color-hashed actor pill (stable palette so [Distiller] and
  [Novelty Analyst] each get their own hue across the session).
- Phase pill (plan=cyan, work=green, synth=amber, aggregate=purple).
- Token count pill formatted 1.2k / 14.3k / etc.
- Gated-action warning pill in amber when > 0.
- Left-border color strip keyed to the actor for at-a-glance
  visual grouping.
- Long outputs collapse to their first 300 chars with a '+ N more'
  toggle to expand the full text.
- 'done' events get a green (or red for error) border strip +
  pill instead of blending into the stream.

Container:
- Splits '[actor] action (outcome) · msg' into colored spans —
  actor pill (deterministic color), action in dim, outcome pill
  green/red/dim by state.
- Non-line events (info/error/done) get their own left-border
  strip so bash echoes and stack traces don't drown in the daemon
  chatter.
- Timestamps switch to HH:MM:SS.mmm — dense but scannable.

Small palette (LOG constants) keeps the color budget bounded — no
new UI vocabulary, just cleaner reads of what was already there.

── Team-wizard governance parity ───────────────────────────────
build_team_with_lifecycle now matches POST /api/claws' governance:
- enforce_new_agent quota check per member (previously bypassed
  workspace agent quotas entirely for team/auto-provision paths).
- audit::append('agent.created', ..., {source: 'team_wizard'}) per
  member so team-created claws appear in the same audit trail as
  individually-created ones. Adding a 'source' key distinguishes
  provenance without changing consumers.

.brain (h5) handling was already consistent between the two paths —
both use the lazy on-first-access load_brain hook seeded from
agents.system_prompt. No change there.
2026-07-17 18:20:56 -07:00

765 lines
26 KiB
Rust

//! Team endpoints — deploy a baseline topology staffed with real claws, then run
//! it on the durable topology runner. The first rung of the deploy ladder.
use axum::extract::{Path, State};
use axum::http::StatusCode;
use axum::Json;
use cm_domain::{AccessPolicy, Agent, AgentId, AgentStatus};
use cm_llm::{ChatMessage, ChatRequest, ChatRole, ContentPart, LlmEvent};
use cm_topology::{build, TopologyKind};
use serde::{Deserialize, Serialize};
use serde_json::Value;
use time::format_description::well_known::Rfc3339;
use uuid::Uuid;
use crate::runtime_provision::{claw_alias, RuntimeProvisioner};
use crate::{ApiError, AppState, Authed};
/// One staffed role in the team (becomes a claw + a topology node).
#[derive(Deserialize)]
pub struct TeamMemberInput {
pub role: String,
pub name: String,
/// Model selector: claude | glm | glm-5.2 | kimi | gemini | groq.
#[serde(default)]
pub model: String,
#[serde(default)]
pub system_prompt: String,
#[serde(default)]
pub accent: String,
}
#[derive(Deserialize)]
pub struct CreateTeamRequest {
pub name: String,
/// TopologyKind (snake_case), e.g. "hierarchical", "pipeline".
pub kind: String,
pub members: Vec<TeamMemberInput>,
/// Optional parent company — when provided, the freshly-created team
/// gets bound to it via `company_teams` so it never lands orphaned.
/// Wizards typically fetch this from `POST /api/structure/ensure-chain`
/// so the workspace always has a valid parent before team creation.
#[serde(default)]
pub attach_to_company_id: Option<Uuid>,
}
#[derive(Serialize)]
pub struct TeamCreated {
pub team_id: String,
}
fn parse_kind(s: &str) -> Result<TopologyKind, ApiError> {
serde_json::from_value(Value::String(s.to_string())).map_err(|_| ApiError::BadRequest)
}
/// Create a team end-to-end: for each member create a claw + provision a runtime
/// agent, build the baseline topology, bind node→claw, persist. Returns the team
/// id + the created claw ids (in member order) so callers (e.g. the Master
/// Planner scaffold) can attach brains afterward.
pub(crate) async fn build_team(
state: &AppState,
workspace_id: cm_domain::WorkspaceId,
user_id: cm_domain::UserId,
name: &str,
kind_str: &str,
members: &[TeamMemberInput],
) -> Result<(Uuid, Vec<Uuid>), ApiError> {
build_team_with_lifecycle(
state,
workspace_id,
user_id,
name,
kind_str,
members,
"permanent",
)
.await
}
/// Same as `build_team` but with an explicit `lifecycle` (`permanent` |
/// `ephemeral`). Ephemeral teams are torn down by the topology_worker after
/// their last run terminates — used by the Scheduled + Triggered planner modes.
pub(crate) async fn build_team_with_lifecycle(
state: &AppState,
workspace_id: cm_domain::WorkspaceId,
user_id: cm_domain::UserId,
name: &str,
kind_str: &str,
members: &[TeamMemberInput],
lifecycle: &str,
) -> Result<(Uuid, Vec<Uuid>), ApiError> {
if members.is_empty() {
return Err(ApiError::BadRequest);
}
let kind = parse_kind(kind_str)?;
let provisioner = RuntimeProvisioner::from_env().ok_or(ApiError::Internal)?;
let mut claw_ids: Vec<Uuid> = Vec::with_capacity(members.len());
for m in members {
// Parity with the individual `POST /api/claws` handler — each
// claw counts against the workspace's agent quota + emits an
// audit row. Without these the auto-provision + team-wizard
// paths silently bypassed both governance rails.
crate::quota::enforce_new_agent(state, workspace_id).await?;
let agent = Agent {
id: AgentId::new(),
workspace_id,
name: m.name.clone(),
job_title: m.role.clone(),
system_prompt: m.system_prompt.clone(),
avatar: String::new(),
accent: m.accent.clone(),
wallpaper: String::new(),
managed_by: user_id,
status: AgentStatus::Online,
};
cm_db::repo::agents::insert(&state.pool, &agent, &AccessPolicy::default()).await?;
cm_db::repo::audit::append(
&state.pool,
workspace_id,
cm_db::repo::audit::Actor::User(user_id),
"agent.created",
"agent",
&agent.id.to_string(),
serde_json::json!({
"name": agent.name,
"job_title": agent.job_title,
"source": "team_wizard",
}),
)
.await?;
let claw_id = agent.id.as_uuid();
provisioner
.provision_claw(claw_id, &m.model)
.await
.map_err(|e| {
eprintln!("teams: provision claw {claw_id} failed: {e}");
ApiError::Internal
})?;
cm_db::repo::agents::set_model_binding(&state.pool, agent.id, &m.model).await?;
claw_ids.push(claw_id);
}
let roles: Vec<&str> = members.iter().map(|m| m.role.as_str()).collect();
let mut graph = build(kind, &roles).map_err(|_| ApiError::BadRequest)?;
for (i, node) in graph.nodes.iter_mut().enumerate() {
if let Some(cid) = claw_ids.get(i) {
node.attrs.insert("agent".into(), claw_alias(*cid));
node.attrs.insert("claw_id".into(), cid.to_string());
}
}
let team_id = Uuid::now_v7();
let graph_json = serde_json::to_value(&graph).map_err(|_| ApiError::Internal)?;
cm_db::repo::teams::insert_team_with_lifecycle(
&state.pool,
team_id,
workspace_id,
name,
kind.as_str(),
&graph_json,
lifecycle,
)
.await?;
for (i, node) in graph.nodes.iter().enumerate() {
if let Some(cid) = claw_ids.get(i) {
cm_db::repo::teams::add_member(&state.pool, team_id, &node.id, *cid, &node.role)
.await?;
}
}
Ok((team_id, claw_ids))
}
/// `POST /api/teams` — create a team from explicit members.
pub async fn create_team(
State(state): State<AppState>,
Authed(user): Authed,
Json(body): Json<CreateTeamRequest>,
) -> Result<(StatusCode, Json<TeamCreated>), ApiError> {
let (team_id, _) = build_team(
&state,
user.workspace_id,
user.user_id,
&body.name,
&body.kind,
&body.members,
)
.await?;
// Auto-parent the new team when the wizard fetched a company via
// `POST /api/structure/ensure-chain`. Ownership check + node_id
// allocation happen inline so the team never lands orphaned mid-turn.
if let Some(company_id) = body.attach_to_company_id {
let company = cm_db::repo::companies::get(&state.pool, company_id, user.workspace_id)
.await
.map_err(|_| ApiError::NotFound)?;
let existing = cm_db::repo::companies::teams_for_company(&state.pool, company.id).await?;
let node_id = format!("n{}", existing.len());
cm_db::repo::companies::add_team(&state.pool, company.id, &node_id, team_id, "team")
.await?;
}
Ok((
StatusCode::CREATED,
Json(TeamCreated {
team_id: team_id.to_string(),
}),
))
}
#[derive(Deserialize)]
pub struct ComposeTeamRequest {
pub name: String,
/// TopologyKind (snake_case); defaults to `hub_spoke` when omitted.
#[serde(default)]
pub kind: String,
/// Existing claws (agents) to group into the new team.
pub claw_ids: Vec<Uuid>,
/// Optional parent company id. When set, the new team is bound under
/// this company via `companies::add_team` inside the same handler so
/// the team never lands orphaned. Mirrors `create_team`'s field —
/// wizards fetch this via `POST /api/structure/ensure-chain`.
#[serde(default)]
pub attach_to_company_id: Option<Uuid>,
}
/// `POST /api/teams/from-claws` — create a team from EXISTING claws (no new
/// provisioning): verify each claw is in the caller's workspace, build the
/// baseline topology over their roles, bind node→claw, persist.
pub async fn create_team_from_claws(
State(state): State<AppState>,
Authed(user): Authed,
Json(body): Json<ComposeTeamRequest>,
) -> Result<(StatusCode, Json<TeamCreated>), ApiError> {
if body.claw_ids.is_empty() {
return Err(ApiError::BadRequest);
}
let kind = parse_kind(if body.kind.is_empty() {
"hub_spoke"
} else {
&body.kind
})?;
// Resolve + authorize each claw, collecting its role for the topology.
let mut roles: Vec<String> = Vec::with_capacity(body.claw_ids.len());
for cid in &body.claw_ids {
let agent =
crate::routes::claws::workspace_agent(&state, &user, AgentId::from(*cid)).await?;
roles.push(if agent.job_title.is_empty() {
"claw".into()
} else {
agent.job_title
});
}
let role_refs: Vec<&str> = roles.iter().map(|s| s.as_str()).collect();
let mut graph = build(kind, &role_refs).map_err(|_| ApiError::BadRequest)?;
for (i, node) in graph.nodes.iter_mut().enumerate() {
if let Some(cid) = body.claw_ids.get(i) {
node.attrs.insert("agent".into(), claw_alias(*cid));
node.attrs.insert("claw_id".into(), cid.to_string());
}
}
let team_id = Uuid::now_v7();
let graph_json = serde_json::to_value(&graph).map_err(|_| ApiError::Internal)?;
cm_db::repo::teams::insert_team(
&state.pool,
team_id,
user.workspace_id,
&body.name,
kind.as_str(),
&graph_json,
)
.await?;
for (i, node) in graph.nodes.iter().enumerate() {
if let Some(cid) = body.claw_ids.get(i) {
cm_db::repo::teams::add_member(&state.pool, team_id, &node.id, *cid, &node.role)
.await?;
}
}
// Same auto-parent path as create_team — if the caller preflighted
// ensure-chain to get a company_id, bind the new team under it here.
if let Some(company_id) = body.attach_to_company_id {
let company = cm_db::repo::companies::get(&state.pool, company_id, user.workspace_id)
.await
.map_err(|_| ApiError::NotFound)?;
let existing = cm_db::repo::companies::teams_for_company(&state.pool, company.id).await?;
let node_id = format!("n{}", existing.len());
cm_db::repo::companies::add_team(&state.pool, company.id, &node_id, team_id, "team")
.await?;
}
Ok((
StatusCode::CREATED,
Json(TeamCreated {
team_id: team_id.to_string(),
}),
))
}
#[derive(Serialize)]
pub struct TeamSummaryOut {
pub id: String,
pub name: String,
pub kind: String,
pub status: String,
pub created_at: String,
}
/// `GET /api/teams` — recent teams for the workspace.
pub async fn list_teams(
State(state): State<AppState>,
Authed(user): Authed,
) -> Result<Json<Vec<TeamSummaryOut>>, ApiError> {
let rows = cm_db::repo::teams::list_for_workspace(&state.pool, user.workspace_id, 50).await?;
Ok(Json(
rows.into_iter()
.map(|t| TeamSummaryOut {
id: t.id.to_string(),
name: t.name,
kind: t.kind,
status: t.status,
created_at: t.created_at.format(&Rfc3339).unwrap_or_default(),
})
.collect(),
))
}
#[derive(Serialize)]
pub struct TeamMemberOut {
pub node_id: String,
pub claw_id: String,
pub role: String,
}
#[derive(Serialize)]
pub struct TeamDetail {
pub id: String,
pub name: String,
pub kind: String,
pub status: String,
pub created_at: String,
pub graph: Value,
pub members: Vec<TeamMemberOut>,
/// Per-team runtime posture (0045). `None` risk_profile ⇒
/// container inherits the template default at spawn time.
#[serde(skip_serializing_if = "Option::is_none")]
pub risk_profile: Option<String>,
#[serde(default)]
pub mcp_bundles: Vec<String>,
}
/// `PATCH /api/teams/{id}/runtime-config` — set the per-team
/// risk_profile + mcp_bundles. Wizards call this after creating a team
/// to bind it as either "read-heavy research" or "write-capable coding"
/// without weakening the sibling team's posture.
#[derive(Deserialize)]
pub struct RuntimeConfigRequest {
#[serde(default)]
pub risk_profile: Option<String>,
#[serde(default)]
pub mcp_bundles: Vec<String>,
}
pub async fn set_runtime_config(
State(state): State<AppState>,
Authed(user): Authed,
Path(id): Path<Uuid>,
Json(body): Json<RuntimeConfigRequest>,
) -> Result<StatusCode, ApiError> {
// Workspace-scope check via the existing get_team.
let _ = cm_db::repo::teams::get_team(&state.pool, id, user.workspace_id).await?;
cm_db::repo::teams::set_team_runtime_config(
&state.pool,
id,
user.workspace_id,
&cm_db::repo::teams::TeamRuntimeConfig {
risk_profile: body.risk_profile,
mcp_bundles: body.mcp_bundles,
},
)
.await?;
Ok(StatusCode::NO_CONTENT)
}
/// `GET /api/teams/{id}` — a team's graph + node→claw bindings.
pub async fn get_team(
State(state): State<AppState>,
Authed(user): Authed,
Path(id): Path<Uuid>,
) -> Result<Json<TeamDetail>, ApiError> {
let team = cm_db::repo::teams::get_team(&state.pool, id, user.workspace_id).await?;
let members = cm_db::repo::teams::members_for_team(&state.pool, id).await?;
let runtime = cm_db::repo::teams::get_team_runtime_config(&state.pool, id, user.workspace_id)
.await
.ok()
.flatten()
.unwrap_or_default();
Ok(Json(TeamDetail {
id: team.id.to_string(),
name: team.name,
kind: team.kind,
status: team.status,
created_at: team.created_at.format(&Rfc3339).unwrap_or_default(),
graph: team.graph,
members: members
.into_iter()
.map(|m| TeamMemberOut {
node_id: m.node_id,
claw_id: m.claw_id.to_string(),
role: m.role,
})
.collect(),
risk_profile: runtime.risk_profile,
mcp_bundles: runtime.mcp_bundles,
}))
}
#[derive(Deserialize)]
pub struct PatchTeamRequest {
/// New TopologyKind (snake_case), e.g. "hierarchical", "hub_spoke".
pub kind: String,
}
/// `PATCH /api/teams/{id}` — change a team's topology: rebuild the graph over the
/// existing members' roles (stable node ids keep the node→claw bindings valid)
/// and persist the new kind + graph.
pub async fn patch_team(
State(state): State<AppState>,
Authed(user): Authed,
Path(id): Path<Uuid>,
Json(body): Json<PatchTeamRequest>,
) -> Result<StatusCode, ApiError> {
let team = cm_db::repo::teams::get_team(&state.pool, id, user.workspace_id).await?;
let kind = parse_kind(&body.kind)?;
let members = cm_db::repo::teams::members_for_team(&state.pool, team.id).await?;
let by_node: std::collections::HashMap<String, (Uuid, String)> = members
.into_iter()
.map(|m| (m.node_id, (m.claw_id, m.role)))
.collect();
let n = by_node.len();
let roles: Vec<String> = (0..n)
.map(|i| {
by_node
.get(&format!("n{i}"))
.map(|(_, r)| r.clone())
.unwrap_or_else(|| "claw".into())
})
.collect();
let role_refs: Vec<&str> = roles.iter().map(String::as_str).collect();
let mut graph = build(kind, &role_refs).map_err(|_| ApiError::BadRequest)?;
for node in graph.nodes.iter_mut() {
if let Some((cid, _)) = by_node.get(&node.id) {
node.attrs.insert("agent".into(), claw_alias(*cid));
node.attrs.insert("claw_id".into(), cid.to_string());
}
}
let graph_json = serde_json::to_value(&graph).map_err(|_| ApiError::Internal)?;
cm_db::repo::teams::set_topology(
&state.pool,
team.id,
user.workspace_id,
kind.as_str(),
&graph_json,
)
.await?;
Ok(StatusCode::NO_CONTENT)
}
/// `PATCH /api/teams/{id}/name` — inline rename from the sidebar.
#[derive(Deserialize)]
pub struct RenameTeamRequest {
pub name: String,
}
pub async fn rename_team(
State(state): State<AppState>,
Authed(user): Authed,
Path(id): Path<Uuid>,
Json(body): Json<RenameTeamRequest>,
) -> Result<StatusCode, ApiError> {
let name = body.name.trim();
if name.is_empty() {
return Err(ApiError::BadRequest);
}
cm_db::repo::teams::rename_team(&state.pool, id, user.workspace_id, name).await?;
Ok(StatusCode::NO_CONTENT)
}
/// `DELETE /api/teams/{id}` — remove a team and its node→claw bindings (the claws
/// themselves remain in the workspace).
pub async fn delete_team(
State(state): State<AppState>,
Authed(user): Authed,
Path(id): Path<Uuid>,
) -> Result<StatusCode, ApiError> {
cm_db::repo::teams::delete_team(&state.pool, id, user.workspace_id).await?;
Ok(StatusCode::NO_CONTENT)
}
#[derive(Deserialize)]
pub struct RunTeamRequest {
pub task: String,
}
#[derive(Serialize)]
pub struct RunAccepted {
pub run_id: String,
pub status: String,
}
/// `POST /api/teams/{id}/run` — enqueue a durable run of the team's topology
/// (drives the bound claws). Poll/stream via `/api/topology-runs/{run_id}`.
pub async fn run_team(
State(state): State<AppState>,
Authed(user): Authed,
Path(id): Path<Uuid>,
Json(body): Json<RunTeamRequest>,
) -> Result<(StatusCode, Json<RunAccepted>), ApiError> {
let team = cm_db::repo::teams::get_team(&state.pool, id, user.workspace_id).await?;
crate::quota::enforce_new_run(&state, user.workspace_id).await?;
let run_id = Uuid::now_v7();
cm_db::repo::topology_runs::enqueue_run_for_team(
&state.pool,
run_id,
user.workspace_id,
&body.task,
&team.graph,
id,
)
.await?;
Ok((
StatusCode::ACCEPTED,
Json(RunAccepted {
run_id: run_id.to_string(),
status: "queued".into(),
}),
))
}
// ── auto-provision (0047 fold): LLM-derived team ─────────────────────
#[derive(Deserialize)]
pub struct AutoProvisionRequest {
/// Topic title (short — used to name the team).
pub title: String,
/// Topic description (drives the LLM roster derivation).
pub description: String,
/// Outcome kind (spec / prod_plan / roadmap / paper / integrations)
/// — steers the roster + topology.
pub outcome_kind: String,
/// Optional user hint for topology. When absent, derived from
/// outcome_kind (integrations → pipeline; everything else →
/// hub_spoke). Accepts the same strings as the wizard.
#[serde(default)]
pub topology_kind: Option<String>,
/// Optional preferred model for every provisioned agent. Falls back
/// to `claude-sonnet-5` when omitted. Kept overridable so the same
/// endpoint serves cost-conscious topics too.
#[serde(default)]
pub model: Option<String>,
/// Optional `["file_read","web_search",...]` risk-profile hint —
/// when omitted the endpoint picks by outcome_kind (research →
/// research_web_readonly; coding-adjacent → coding_readwrite).
#[serde(default)]
pub risk_profile: Option<String>,
/// MCP bundle aliases — same fall-back rule applies (always
/// clawmates_door; gitea_forge when a repo is bound; deep-research
/// skill for research profiles).
#[serde(default)]
pub mcp_bundles: Vec<String>,
}
#[derive(Serialize)]
pub struct AutoProvisionedAgent {
pub agent_id: String,
pub name: String,
pub role_slot: String,
pub model: String,
}
#[derive(Serialize)]
pub struct AutoProvisionResponse {
pub team_id: String,
pub topology_kind: String,
pub agents: Vec<AutoProvisionedAgent>,
/// Echo of the runtime-config applied to the team row (0045 fields).
#[serde(skip_serializing_if = "Option::is_none")]
pub risk_profile: Option<String>,
pub mcp_bundles: Vec<String>,
}
const AUTOPROVISION_SYSTEM: &str = "You are a team-composition assistant. \
Given a research/coding topic prompt + outcome kind, produce a compact roster of \
3 to 5 agents that could execute the pipeline end to end. For each agent output:\n\
- role_slot: short lowercase snake_case slot name (e.g. \"harvester\", \"tdd_implementer\"). \
Must be unique within the roster.\n\
- name: short human-friendly display name (1-2 words, ASCII, distinct per agent).\n\
- system_prompt: 2-4 sentence system prompt describing this agent's job in the \
topology. Should be actionable and reference the topic where relevant.\n\
Return a SINGLE JSON object with no code fences and no additional keys:\n\
{\"roles\": [ {\"role_slot\": \"...\", \"name\": \"...\", \"system_prompt\": \"...\"}, ... ]}\n\
Order matters: it dictates the pipeline / hub_spoke position. First role is the \
coordinator or first stage.";
/// `POST /api/teams/auto-provision` — LLM-derive a roster then materialize
/// a team + all claws + node→claw bindings, and stamp the runtime posture
/// (risk_profile + mcp_bundles) so it's ready for wizard step-5 to bind.
/// Returns the shape the wizard's `agents[]` submit expects, so the caller
/// can flow straight into `POST /api/research` (or `/api/loops`) with
/// `agents: response.agents` and no user-visible detour to the roster page.
pub async fn auto_provision(
State(state): State<AppState>,
Authed(user): Authed,
Json(body): Json<AutoProvisionRequest>,
) -> Result<(StatusCode, Json<AutoProvisionResponse>), ApiError> {
if body.title.trim().is_empty() || body.description.trim().is_empty() {
return Err(ApiError::BadRequest);
}
// Derive topology + defaults from outcome_kind if the caller didn't override.
let outcome_kind = body.outcome_kind.trim().to_string();
let topology_kind = body.topology_kind.clone().unwrap_or_else(|| {
if outcome_kind == "integrations" {
"pipeline".to_string()
} else {
"hub_spoke".to_string()
}
});
let model = body
.model
.clone()
.filter(|s| !s.trim().is_empty())
.unwrap_or_else(|| "claude-sonnet-5".to_string());
// Auto-pick runtime posture when the caller didn't. Research topics
// default to the read-only + web-fetch profile so agents can fetch
// papers; coding-adjacent kinds don't apply here (loops.wizard picks
// them from a different path).
let risk_profile = body.risk_profile.clone().or_else(|| {
Some(match outcome_kind.as_str() {
"integrations" | "paper" | "spec" | "prod_plan" | "roadmap" => {
"research_web_readonly".to_string()
}
_ => "research_web_readonly".to_string(),
})
});
let mut mcp_bundles = body.mcp_bundles.clone();
if mcp_bundles.is_empty() {
mcp_bundles.push("clawmates_door".to_string());
// gitea_forge is scoped to teams that will touch repos; the
// wizard's downstream repo-binding step is what earns it.
// Always safe to add now — the MCP layer no-ops when the token
// isn't present in the container env.
mcp_bundles.push("gitea_forge".to_string());
}
// 1) LLM plan pass → roster JSON.
let user_message = format!(
"Outcome kind: {outcome_kind}\nTopology: {topology_kind}\n\nTopic title: {}\n\nTopic description:\n{}",
body.title.trim(),
body.description.trim(),
);
let request = ChatRequest {
system: AUTOPROVISION_SYSTEM.into(),
messages: vec![ChatMessage {
role: ChatRole::User,
parts: vec![ContentPart::Text { text: user_message }],
}],
tools: Vec::new(),
model: state.runtime.model().to_string(),
max_tokens: 2048,
web_search: false,
};
let provider = state.runtime.provider();
let mut stream = provider
.stream(request)
.await
.map_err(|_| ApiError::Internal)?;
let mut buf = String::new();
use futures::StreamExt;
while let Some(event) = stream.next().await {
match event.map_err(|_| ApiError::Internal)? {
LlmEvent::TextDelta(delta) => buf.push_str(&delta),
LlmEvent::Stop(_) => break,
_ => {}
}
}
#[derive(Deserialize)]
struct DerivedRole {
role_slot: String,
name: String,
system_prompt: String,
}
#[derive(Deserialize)]
struct DerivedRoster {
roles: Vec<DerivedRole>,
}
let roster: DerivedRoster = serde_json::from_str(buf.trim()).map_err(|_| ApiError::Internal)?;
if roster.roles.is_empty() || roster.roles.len() > 8 {
return Err(ApiError::Internal);
}
// 2) Materialize the team + all claws via the existing build_team pipeline.
let members: Vec<TeamMemberInput> = roster
.roles
.iter()
.map(|r| TeamMemberInput {
role: r.role_slot.trim().to_string(),
name: r.name.trim().to_string(),
model: model.clone(),
system_prompt: r.system_prompt.trim().to_string(),
accent: String::new(),
})
.collect();
let team_name = format!("Auto · {}", body.title.trim());
let (team_id, claw_ids) = build_team(
&state,
user.workspace_id,
user.user_id,
&team_name,
&topology_kind,
&members,
)
.await?;
// 3) Stamp the runtime posture so the container spawn slice (3b) has
// the right risk_profile + bundles when it fires.
if let Err(e) = cm_db::repo::teams::set_team_runtime_config(
&state.pool,
team_id,
user.workspace_id,
&cm_db::repo::teams::TeamRuntimeConfig {
risk_profile: risk_profile.clone(),
mcp_bundles: mcp_bundles.clone(),
},
)
.await
{
eprintln!("auto_provision({team_id}): runtime-config write failed: {e:?}");
}
// 4) Shape the response for the wizard: agent_id + role_slot in the
// exact form the /api/research submit body expects.
let agents: Vec<AutoProvisionedAgent> = claw_ids
.iter()
.zip(members.iter())
.map(|(cid, m)| AutoProvisionedAgent {
agent_id: cid.to_string(),
name: m.name.clone(),
role_slot: m.role.clone(),
model: model.clone(),
})
.collect();
Ok((
StatusCode::CREATED,
Json(AutoProvisionResponse {
team_id: team_id.to_string(),
topology_kind,
agents,
risk_profile,
mcp_bundles,
}),
))
}