fix(orchestrator): le lancement d'une cible structurée froide lie la session au fil A-B résolu par ask_agent — #238

ensure_structured_session reçoit désormais le conversation_id déjà résolu et le
transmet au plan de spawn au lieu de retomber sur le fallback User-B. WorkState
et la session lancée pointent ainsi sur le même fil. Test de régression
ask_cold_structured_agent_requester_launches_on_resolved_pair_thread.
This commit is contained in:
2026-08-22 11:30:35 +02:00
parent 963b51a277
commit 5449526157
2 changed files with 77 additions and 5 deletions

View File

@ -4293,17 +4293,22 @@ impl AgentSession for EchoSession {
struct CountingFactory {
starts: Arc<AtomicUsize>,
next_id: Arc<Mutex<u128>>,
session_plans: Arc<Mutex<Vec<SessionPlan>>>,
}
impl CountingFactory {
fn new() -> Self {
Self {
starts: Arc::new(AtomicUsize::new(0)),
next_id: Arc::new(Mutex::new(9000)),
session_plans: Arc::new(Mutex::new(Vec::new())),
}
}
fn start_count(&self) -> usize {
self.starts.load(Ordering::SeqCst)
}
fn session_plans(&self) -> Vec<SessionPlan> {
self.session_plans.lock().unwrap().clone()
}
}
#[async_trait]
impl AgentSessionFactory for CountingFactory {
@ -4315,13 +4320,14 @@ impl AgentSessionFactory for CountingFactory {
_profile: &AgentProfile,
_ctx: &PreparedContext,
_cwd: &ProjectPath,
_session: &SessionPlan,
session: &SessionPlan,
_requester: Option<&str>,
_env: &[(String, String)],
_sandbox: Option<&domain::sandbox::SandboxPlan>,
_structured_policy: Option<&domain::ports::StructuredProviderLaunchPolicy>,
) -> Result<Arc<dyn AgentSession>, AgentSessionError> {
self.starts.fetch_add(1, Ordering::SeqCst);
self.session_plans.lock().unwrap().push(session.clone());
let id = {
let mut n = self.next_id.lock().unwrap();
let id = SessionId::from_uuid(Uuid::from_u128(*n));
@ -4500,6 +4506,7 @@ struct StructuredAskFixture {
session_limits: Arc<SessionLimitService>,
background_store: Arc<RecordingBackgroundStore>,
background_runner: Arc<RecordingBackgroundRunner>,
conversations: Arc<TestConversations>,
}
fn structured_ask_fixture(contexts: FakeContexts) -> StructuredAskFixture {
@ -4574,7 +4581,7 @@ fn structured_ask_fixture_with_background(
Arc::clone(&mailbox),
Arc::new(pty.clone()) as Arc<dyn PtyPort>,
));
let conversations = Arc::new(TestConversations::new()) as Arc<dyn ConversationRegistry>;
let conversations = Arc::new(TestConversations::new());
let session_limits = Arc::new(SessionLimitService::new(
Arc::new(FixedMillisClock(1_000)) as Arc<dyn Clock>,
Arc::new(NoopScheduler::default()) as Arc<dyn Scheduler>,
@ -4596,7 +4603,7 @@ fn structured_ask_fixture_with_background(
Arc::clone(&mediator) as Arc<dyn InputMediator>,
Arc::clone(&mailbox) as Arc<dyn AgentMailbox>,
)
.with_conversations(conversations)
.with_conversations(Arc::clone(&conversations) as Arc<dyn ConversationRegistry>)
.with_events(Arc::new(bus.clone()))
.with_session_limits(Arc::clone(&session_limits))
// Même registre que le launcher : `ask_agent` y relit la session fraîchement
@ -4626,6 +4633,7 @@ fn structured_ask_fixture_with_background(
session_limits,
background_store,
background_runner,
conversations,
}
}
@ -4677,6 +4685,63 @@ async fn ask_cold_structured_target_autolaunches_session_and_final_unblocks() {
);
}
/// Régression #238 : quand une cible structurée froide est lancée pour un ask A→B,
/// le launcher doit recevoir l'id du fil A↔B déjà résolu par `ask_agent`. Sans cela,
/// `launch_structured` retombe sur le fallback User↔B : WorkState pointe alors vers le
/// ticket/log A↔B tandis que la session/vue fraîchement lancée est liée à User↔B.
#[tokio::test]
async fn ask_cold_structured_agent_requester_launches_on_resolved_pair_thread() {
let agent = scratch_agent(aid(1), "architect", "agents/architect.md");
let fx = structured_ask_fixture(FakeContexts::with_agent(&agent, "# persona"));
let requester = aid(7);
let requester_uuid = uuid::Uuid::from_u128(7);
let ask_json = format!(
r#"{{ "type":"agent.message", "requestedBy":"{requester_uuid}", "targetAgent":"architect", "task":"Analyse §17" }}"#
);
let out = fx
.service
.dispatch(&project(), cmd(&ask_json))
.await
.expect("ask ok");
assert_eq!(out.reply.as_deref(), Some("structured: Analyse §17"));
assert_eq!(
fx.factory.start_count(),
1,
"the cold structured target is launched exactly once"
);
let expected = fx
.conversations
.resolve(
project_id(),
domain::ConversationParty::agent(requester),
domain::ConversationParty::agent(aid(1)),
)
.id;
let user_target = fx
.conversations
.resolve(
project_id(),
domain::ConversationParty::User,
domain::ConversationParty::agent(aid(1)),
)
.id;
assert_ne!(
expected, user_target,
"the regression guard must distinguish agent-requester and user threads"
);
assert_eq!(
fx.factory.session_plans(),
vec![SessionPlan::Resume {
conversation_id: expected.to_string()
}],
"cold launch must reuse the conversation id resolved by ask_agent"
);
}
/// Réutilisation : une **deuxième** délégation vers la **même** cible (désormais chaude)
/// ne relance **pas** de session — elle réutilise celle insérée au premier `ask`. Prouve
/// que le `session_for_agent` court-circuite l'auto-lancement (idempotence du chemin).