Compare commits
10 Commits
9e62097c97
...
9ac32c9e93
| Author | SHA1 | Date | |
|---|---|---|---|
| 9ac32c9e93 | |||
| b43ab77eee | |||
| 95b16a5047 | |||
| 0a001c42c7 | |||
| 68c56b6af6 | |||
| 1b58bc325e | |||
| 7b90944b61 | |||
| 6010f6b6c0 | |||
| 0e1095e9ba | |||
| e25e9e46fd |
@@ -21,6 +21,7 @@ All sources under `apps/server/src/`.
|
||||
- `GET /metrics/tool-reliability` — per-model tool-call validity across the event log (`ToolReliabilityInspectionService`); groundwork for capability-aware routing
|
||||
- Optional `[git]` transport creates `run/<sessionId>` from a server-local checkout and pushes it at terminal state; clients review with ordinary Git and never supply a remote URL as `cwd`.
|
||||
- Repo-map L3 embeddings use bounded, recorded source descriptors (module/package, imports, leading purpose comment, symbols); raw file bodies are never embedded. Their versioned `repomap:v2` namespace forces a one-time re-embed when the semantic document format changes.
|
||||
- At boot, `tools.workspace_root` is the authoritative tool jail and project-observation root. A configured `tools.working_dir` may only remain distinct when it is contained by that root; an outside value is clamped to `workspace_root`. Project memory and repo-map indexing are also rebound to `workspace_root`, so a stale `[project].root` cannot inject files from outside the session workspace.
|
||||
|
||||
### WebSocket protocol (`/ws`)
|
||||
- **ServerMessage** (server → client): sealed hierarchy — `SessionMessage` (event-derived, carries `sequence` + `sessionSequence`) and `NonEventMessage` (control/infra). Variants include session lifecycle, approval requests, clarification requests, narration, proposed workflows, health/metrics pushes.
|
||||
|
||||
@@ -0,0 +1,35 @@
|
||||
package com.correx.apps.server
|
||||
|
||||
import com.correx.core.config.ProjectConfig
|
||||
import java.nio.file.Path
|
||||
|
||||
internal data class BootWorkspace(
|
||||
val workspaceRoot: Path,
|
||||
val workingDir: Path,
|
||||
val workingDirWasClamped: Boolean,
|
||||
)
|
||||
|
||||
internal fun resolveBootWorkspace(
|
||||
explicitWorkspaceRoot: Path?,
|
||||
explicitWorkingDir: Path?,
|
||||
processWorkingDir: Path,
|
||||
): BootWorkspace {
|
||||
val workspaceRoot = (explicitWorkspaceRoot ?: explicitWorkingDir ?: processWorkingDir)
|
||||
.toAbsolutePath()
|
||||
.normalize()
|
||||
val requestedWorkingDir = (explicitWorkingDir ?: workspaceRoot)
|
||||
.toAbsolutePath()
|
||||
.normalize()
|
||||
val workingDirIsContained = requestedWorkingDir.startsWith(workspaceRoot)
|
||||
|
||||
return BootWorkspace(
|
||||
workspaceRoot = workspaceRoot,
|
||||
workingDir = requestedWorkingDir.takeIf { workingDirIsContained } ?: workspaceRoot,
|
||||
workingDirWasClamped = !workingDirIsContained,
|
||||
)
|
||||
}
|
||||
|
||||
/** Keep repo-map observation and L3 project memory inside the authoritative boot workspace. */
|
||||
internal fun ProjectConfig.boundToWorkspace(workspaceRoot: Path): ProjectConfig = copy(
|
||||
root = workspaceRoot.toAbsolutePath().normalize().toString(),
|
||||
)
|
||||
@@ -83,6 +83,7 @@ import com.correx.core.events.types.SessionId
|
||||
import com.correx.infrastructure.InfrastructureModule
|
||||
import com.correx.infrastructure.inference.DefaultProviderRegistry
|
||||
import com.correx.infrastructure.workflow.ExecutionPlanCompiler
|
||||
import com.correx.infrastructure.workflow.Lsp4jDiagnosticsRunner
|
||||
import com.correx.infrastructure.workflow.PlanLinter
|
||||
import com.correx.infrastructure.inference.CapabilityAwareRoutingStrategy
|
||||
import com.correx.infrastructure.inference.commons.ManagedInferenceRouter
|
||||
@@ -296,6 +297,14 @@ fun main() {
|
||||
})
|
||||
}
|
||||
val mcpTools = mountedMcpServers.flatMap { it.tools }
|
||||
// Index the workspace into codebase-memory (#242) off the startup path — a fast index makes its
|
||||
// search tools usable by the first session instead of failing "not indexed"; a slow index just
|
||||
// means the first stage or two miss it, not a blocked boot.
|
||||
if (mcpTools.any { it.name.endsWith("__index_repository") }) {
|
||||
Thread {
|
||||
runBlocking { bootstrapCodebaseIndex(mcpTools, workspaceRoot, log) }
|
||||
}.apply { isDaemon = true; name = "mcp-index-bootstrap" }.start()
|
||||
}
|
||||
val extraTools = taskTools + toolOutputTool + mcpTools
|
||||
val toolRegistry = InfrastructureModule.createToolRegistry(
|
||||
buildToolConfig(
|
||||
@@ -376,6 +385,7 @@ fun main() {
|
||||
workspacePolicy = workspacePolicy,
|
||||
workspaceToolRegistryProvider = wsToolRegistryProvider,
|
||||
staticAnalysisRunner = ProcessStaticAnalysisRunner(),
|
||||
lspDiagnosticsRunner = Lsp4jDiagnosticsRunner(),
|
||||
contractAssertionEvaluator = FileSystemContractEvaluator(),
|
||||
semanticReviewer = SemanticReviewerImpl(inferenceRouter),
|
||||
)
|
||||
@@ -456,6 +466,7 @@ fun main() {
|
||||
maxClarificationRounds = maxClarificationRounds,
|
||||
reviewBlockMinConfidence = reviewBlockMinConfidence,
|
||||
reviewBlockRetryCap = reviewBlockRetryCap,
|
||||
reviewLoopMaxCycles = reviewLoopMaxCycles,
|
||||
defaultMaxRefinement = defaultMaxRefinement,
|
||||
recoveryRouteBudget = recoveryRouteBudget,
|
||||
intentRouteBudget = intentRouteBudget,
|
||||
@@ -613,6 +624,20 @@ fun main() {
|
||||
requestPlanApproval = { sid, planJson -> orchestrator.requestPlanApproval(sid, planJson) },
|
||||
toolCapabilities = toolRegistry.all().associate { it.name to it.requiredCapabilities },
|
||||
reflector = com.correx.apps.server.inference.CapabilityGapReflectorImpl(inferenceRouter),
|
||||
// Return-to-architect: re-run the planning workflow from the architect stage so it emits a
|
||||
// corrected plan after a grounding rejection. rehydrate before (architect needs the analyst's
|
||||
// dod, evicted on the planning graph's completion) and after (the fresh execution_plan is
|
||||
// evicted again when this re-run completes — lockAndRun's planContent must read it back).
|
||||
rerunArchitect = { sid ->
|
||||
orchestrator.rehydrate(sid)
|
||||
val planningGraph = workflowRegistry.find("freestyle_planning")
|
||||
?: error("freestyle_planning workflow not registered")
|
||||
val result = orchestrator.runFrom(
|
||||
sid, planningGraph, defaultOrchestrationConfig, com.correx.core.events.types.StageId("architect"),
|
||||
)
|
||||
orchestrator.rehydrate(sid)
|
||||
result
|
||||
},
|
||||
)
|
||||
// observability-spec §4: continuous health watch. Seed the monitor's last-status from the
|
||||
// recorded system-session events so a restart doesn't re-emit a degraded already in the log.
|
||||
|
||||
@@ -0,0 +1,40 @@
|
||||
package com.correx.apps.server
|
||||
|
||||
import com.correx.core.events.events.ToolRequest
|
||||
import com.correx.core.events.types.SessionId
|
||||
import com.correx.core.events.types.StageId
|
||||
import com.correx.core.events.types.ToolInvocationId
|
||||
import com.correx.core.tools.contract.Tool
|
||||
import com.correx.core.tools.contract.ToolExecutor
|
||||
import com.correx.core.tools.contract.ToolResult
|
||||
import org.slf4j.Logger
|
||||
import java.nio.file.Path
|
||||
import java.util.UUID
|
||||
|
||||
/**
|
||||
* Bootstraps the codebase-memory index for the boot workspace (#242): its search tools are dead on
|
||||
* arrival ("project not found or not indexed") until `index_repository` runs once against the repo,
|
||||
* and nothing in a workflow calls it. If a mounted MCP server advertises an `index_repository` tool
|
||||
* we invoke it here, at working-dir open, with the workspace root — through the normal ToolExecutor
|
||||
* path, so no special-casing.
|
||||
*
|
||||
* ponytail: single-workspace boot only, "fast" mode. When per-connection working dirs land, hang
|
||||
* this off the WS-open hook as one entry in a bootstrap registry; one action doesn't earn a registry
|
||||
* yet. "fast" skips similarity/semantic edges — quickest to ready; upgrade the mode if search recall
|
||||
* proves thin.
|
||||
*/
|
||||
internal suspend fun bootstrapCodebaseIndex(mcpTools: List<Tool>, workspaceRoot: Path, log: Logger) {
|
||||
val tool = mcpTools.firstOrNull { it.name.endsWith("__index_repository") } ?: return
|
||||
val executor = tool as? ToolExecutor ?: return
|
||||
val request = ToolRequest(
|
||||
invocationId = ToolInvocationId(UUID.randomUUID().toString()),
|
||||
sessionId = SessionId("boot"),
|
||||
stageId = StageId("boot"),
|
||||
toolName = tool.name,
|
||||
parameters = mapOf("repo_path" to workspaceRoot.toString(), "mode" to "fast"),
|
||||
)
|
||||
when (val result = executor.execute(request)) {
|
||||
is ToolResult.Success -> log.info("Indexed workspace into codebase-memory ({})", tool.name)
|
||||
is ToolResult.Failure -> log.warn("codebase-memory index bootstrap failed: {}", result.reason)
|
||||
}
|
||||
}
|
||||
@@ -380,10 +380,11 @@ class ServerModule(
|
||||
// Record the repo map + seed prior-session memory before the run so stages
|
||||
// see both in context.
|
||||
projectMemory?.let { pm ->
|
||||
val root = sessionWorkspaceRoot(sessionId)
|
||||
runCatching {
|
||||
pm.observeAndRecord(sessionId, pm.repoRoot())
|
||||
pm.indexAndRecord(sessionId, pm.repoRoot())
|
||||
pm.retrieveAndSeed(sessionId, pm.repoRoot())
|
||||
pm.observeAndRecord(sessionId, root)
|
||||
pm.indexAndRecord(sessionId, root)
|
||||
pm.retrieveAndSeed(sessionId, root)
|
||||
}
|
||||
}
|
||||
// Bind operator profile snapshot as an event so replay reads the recorded
|
||||
@@ -416,7 +417,7 @@ class ServerModule(
|
||||
val result = orchestrator.run(sessionId, graph, sessionConfig)
|
||||
freestyleHandoff(sessionId, graph, result)
|
||||
// Distil this run's decisions into durable project memory on completion.
|
||||
projectMemory?.let { pm -> pm.persist(sessionId, pm.repoRoot()) }
|
||||
projectMemory?.let { pm -> pm.persist(sessionId, sessionWorkspaceRoot(sessionId)) }
|
||||
// Propose learned profile adaptations based on session journal (opt-in, never auto-applied).
|
||||
operatorProfile?.let { profile ->
|
||||
profileAdaptationService?.let { svc ->
|
||||
@@ -498,6 +499,17 @@ class ServerModule(
|
||||
* router chat triage, and replay read the recorded snapshot, never the live file
|
||||
* (invariants #8/#9). Shared by the workflow path and the chat-session path.
|
||||
*/
|
||||
/**
|
||||
* The session's bound workspace root — the same one the tool jail and [SessionWorkspaceBoundEvent]
|
||||
* use. The repo-map/index/L3-memory pipeline MUST key off this, not [ProjectMemoryService.repoRoot]
|
||||
* (a session-independent config/cwd default): when they diverge the repo map is computed for a
|
||||
* different tree than the session operates in, so grounding "proves" real paths absent
|
||||
* (session 5fe538f5, 2026-07-19). Falls back to the server default only when unbound.
|
||||
*/
|
||||
private fun sessionWorkspaceRoot(sessionId: SessionId): String =
|
||||
runCatching { sessionRepository.getSession(sessionId).state.boundWorkspace?.workspaceRoot }
|
||||
.getOrNull() ?: projectMemory?.repoRoot() ?: "."
|
||||
|
||||
suspend fun bindProjectProfile(sessionId: SessionId) {
|
||||
val workspaceRoot = runCatching {
|
||||
sessionRepository.getSession(sessionId).state.boundWorkspace?.workspaceRoot
|
||||
|
||||
@@ -1,5 +1,6 @@
|
||||
package com.correx.apps.server.freestyle
|
||||
|
||||
import com.correx.core.events.events.ArtifactValidatedEvent
|
||||
import com.correx.core.events.events.CapabilityGapDetectedEvent
|
||||
import com.correx.core.events.events.CapabilityGapReflectedEvent
|
||||
import com.correx.core.events.events.CapabilityGapVerdict
|
||||
@@ -12,6 +13,7 @@ import com.correx.core.events.events.PlanGroundingVerdict
|
||||
import com.correx.core.events.events.PlanLintCompletedEvent
|
||||
import com.correx.core.events.events.ProjectProfileBoundEvent
|
||||
import com.correx.core.events.events.RepoMapComputedEvent
|
||||
import com.correx.core.events.events.WorkflowFailedEvent
|
||||
import com.correx.infrastructure.workflow.PlanGrounder
|
||||
import com.correx.core.tools.contract.ToolCapability
|
||||
import com.correx.infrastructure.workflow.CapabilityGap
|
||||
@@ -55,37 +57,73 @@ class FreestyleDriver(
|
||||
// Vikunja #30 part 2: bounded LLM "are you sure?" pass over capability gaps, consulted before
|
||||
// requestPlanApproval. Null = feature degrades to part-1 behavior (gaps recorded, never reflected).
|
||||
private val reflector: CapabilityGapReflector? = null,
|
||||
// Return-to-architect loop: on a grounding rejection, re-run the planning workflow from the
|
||||
// architect stage so it emits a corrected plan (the grounding findings are injected into its
|
||||
// context by buildGroundingFeedbackEntry). Null = no loop; grounding rejection is terminal.
|
||||
private val rerunArchitect: (suspend (SessionId) -> WorkflowResult)? = null,
|
||||
// Max grounding-driven architect re-runs before the plan is rejected for good.
|
||||
private val maxGroundingRetries: Int = 2,
|
||||
) {
|
||||
@Suppress("ReturnCount") // sequential gate pipeline: each gate is a guard-return, same as the engines
|
||||
suspend fun lockAndRun(sessionId: SessionId) {
|
||||
val json = planContent(sessionId) ?: run {
|
||||
log.warn("freestyle: no execution_plan content for session={}", sessionId.value)
|
||||
emitRejected(sessionId, "no execution_plan content produced by planning phase", "missing_content")
|
||||
return
|
||||
}
|
||||
val graph = runCatching { compiler.compile(json, "freestyle-${sessionId.value}") }
|
||||
.getOrElse {
|
||||
log.error("freestyle: plan failed to compile: {}", it.message)
|
||||
emitRejected(sessionId, "plan failed to compile: ${it.message}", "compile")
|
||||
// Return-to-architect loop: each pass compiles + gates the current execution_plan. A grounding
|
||||
// rejection with budget left re-runs the architect (which emits a corrected plan) and loops;
|
||||
// every other gate failure — and grounding once the budget is spent — is terminal.
|
||||
var groundingRetries = 0
|
||||
while (true) {
|
||||
val json = planContent(sessionId) ?: run {
|
||||
log.warn("freestyle: no execution_plan content for session={}", sessionId.value)
|
||||
emitRejected(sessionId, "no execution_plan content produced by planning phase", "missing_content")
|
||||
return
|
||||
}
|
||||
// Deterministic lint (plan-pipeline-spec §5) before the plan is surfaced/locked: a hard
|
||||
// failure (unproduced need, trap state) means the plan would fail at runtime, so reject it
|
||||
// now rather than execute a broken plan. Soft findings are recorded for display only.
|
||||
val lint = PlanLinter.lint(graph)
|
||||
emitPlanLint(sessionId, graph.id, lint)
|
||||
if (!lint.passed) {
|
||||
val summary = lint.hardFailures.joinToString("; ") {
|
||||
"${it.code}${it.stageId?.let { s -> " @$s" } ?: ""}: ${it.detail}"
|
||||
val graph = runCatching {
|
||||
compiler.compile(json, "freestyle-${sessionId.value}", sessionArtifacts(sessionId))
|
||||
}
|
||||
log.warn("freestyle: plan failed lint for session={}: {}", sessionId.value, summary)
|
||||
emitRejected(sessionId, "plan failed lint: $summary", "lint")
|
||||
.getOrElse {
|
||||
log.error("freestyle: plan failed to compile: {}", it.message)
|
||||
emitRejected(sessionId, "plan failed to compile: ${it.message}", "compile")
|
||||
return
|
||||
}
|
||||
// Deterministic lint (plan-pipeline-spec §5) before the plan is surfaced/locked: a hard
|
||||
// failure (unproduced need, trap state) means the plan would fail at runtime, so reject it
|
||||
// now rather than execute a broken plan. Soft findings are recorded for display only.
|
||||
val lint = PlanLinter.lint(graph)
|
||||
emitPlanLint(sessionId, graph.id, lint)
|
||||
if (!lint.passed) {
|
||||
val summary = lint.hardFailures.joinToString("; ") {
|
||||
"${it.code}${it.stageId?.let { s -> " @$s" } ?: ""}: ${it.detail}"
|
||||
}
|
||||
log.warn("freestyle: plan failed lint for session={}: {}", sessionId.value, summary)
|
||||
emitRejected(sessionId, "plan failed lint: $summary", "lint")
|
||||
return
|
||||
}
|
||||
// Plan grounding (design 2026-07-15 seam 1) before lock: check the compiled plan against the
|
||||
// session's recorded workspace facts (repo map + profile commands). A build stage aimed at a
|
||||
// prerequisite nothing creates is doomed. Deterministic + pure over recorded events (#8/#9).
|
||||
val groundingFailure = groundPlan(sessionId, graph)
|
||||
if (groundingFailure != null) {
|
||||
if (groundingRetries < maxGroundingRetries && rerunArchitect != null) {
|
||||
groundingRetries++
|
||||
log.info(
|
||||
"freestyle: grounding returned plan to architect (attempt {}/{}) session={}: {}",
|
||||
groundingRetries, maxGroundingRetries, sessionId.value, groundingFailure,
|
||||
)
|
||||
// Re-run the architect stage; it sees the recorded grounding findings via
|
||||
// buildGroundingFeedbackEntry and emits a corrected plan. Then loop and re-gate.
|
||||
rerunArchitect.invoke(sessionId)
|
||||
continue
|
||||
}
|
||||
log.warn("freestyle: plan failed grounding for session={}: {}", sessionId.value, groundingFailure)
|
||||
emitRejected(sessionId, "plan failed grounding: $groundingFailure", "grounding")
|
||||
return
|
||||
}
|
||||
lockAndRunGrounded(sessionId, graph, json)
|
||||
return
|
||||
}
|
||||
// Plan grounding (design 2026-07-15 seam 1) before lock: check the compiled plan against the
|
||||
// session's recorded workspace facts (repo map + profile commands). A build stage aimed at a
|
||||
// prerequisite nothing creates is doomed — reject now so it fails at stage 0, not after
|
||||
// downstream files pile up. Deterministic + pure over recorded events (invariants #8/#9).
|
||||
if (groundPlan(sessionId, graph)) return
|
||||
}
|
||||
|
||||
/** Post-grounding tail of [lockAndRun]: capability gaps, plan-approval, lock, phase-2 handoff. */
|
||||
private suspend fun lockAndRunGrounded(sessionId: SessionId, graph: WorkflowGraph, json: String) {
|
||||
// Capability-gap detector (Vikunja #30 part 1): advisory only — recorded, never blocking.
|
||||
// A gap does not fail the gate and does not grant the missing tool (invariants #3/#4/#5).
|
||||
val gaps = CapabilityGapDetector.detect(graph, toolCapabilities)
|
||||
@@ -137,23 +175,33 @@ class FreestyleDriver(
|
||||
*/
|
||||
fun compiledGraph(sessionId: SessionId): WorkflowGraph? {
|
||||
val json = planContent(sessionId) ?: return null
|
||||
return runCatching { compiler.compile(json, "freestyle-${sessionId.value}") }
|
||||
return runCatching { compiler.compile(json, "freestyle-${sessionId.value}", sessionArtifacts(sessionId)) }
|
||||
.onFailure { log.error("freestyle: plan recompile for resume failed: {}", it.message) }
|
||||
.getOrNull()
|
||||
}
|
||||
|
||||
/** Artifact ids validated earlier in the session (e.g. planning-phase `dod`) for the #264 needs seam. */
|
||||
private fun sessionArtifacts(sessionId: SessionId): Set<String> =
|
||||
eventStore.read(sessionId)
|
||||
.mapNotNull { (it.payload as? ArtifactValidatedEvent)?.artifactId?.value }
|
||||
.toSet()
|
||||
|
||||
/**
|
||||
* Grounds [graph] against recorded workspace facts and emits [PlanGroundingEvaluatedEvent].
|
||||
* Returns true if the plan was rejected (caller must stop before lock). Missing repo map/profile
|
||||
* (fresh session, no scan) grounds vacuously — an empty path set with the manifest-produced-by-plan
|
||||
* check still catches the "build with nothing to build" case.
|
||||
* Returns null when the plan grounds (PASS), else the findings summary — the caller decides
|
||||
* whether to return the plan to architect or reject it. Missing repo map/profile (fresh session,
|
||||
* no scan) grounds vacuously; the manifest-produced-by-plan check still catches "build with
|
||||
* nothing to build".
|
||||
*/
|
||||
private suspend fun groundPlan(sessionId: SessionId, graph: WorkflowGraph): Boolean {
|
||||
private suspend fun groundPlan(sessionId: SessionId, graph: WorkflowGraph): String? {
|
||||
val events = eventStore.read(sessionId)
|
||||
val repoMap = events.mapNotNull { it.payload as? RepoMapComputedEvent }.lastOrNull()
|
||||
val profile = events.mapNotNull { it.payload as? ProjectProfileBoundEvent }.lastOrNull()
|
||||
val paths = repoMap?.entries?.map { it.path }?.toSet().orEmpty()
|
||||
val result = PlanGrounder.ground(graph, paths, profile?.commands.orEmpty())
|
||||
// No RepoMapComputedEvent = no scan ran, so `paths` is unknown, not "empty workspace".
|
||||
// Tell the grounder not to prove paths absent from a set it never observed (the false
|
||||
// "apps/server/** doesn't exist" reject); the build-manifest check still runs.
|
||||
val result = PlanGrounder.ground(graph, paths, profile?.commands.orEmpty(), scanned = repoMap != null)
|
||||
eventStore.append(
|
||||
NewEvent(
|
||||
metadata = EventMetadata(
|
||||
@@ -173,11 +221,8 @@ class FreestyleDriver(
|
||||
),
|
||||
),
|
||||
)
|
||||
if (result.verdict == PlanGroundingVerdict.PASS) return false
|
||||
val summary = result.findings.joinToString("; ")
|
||||
log.warn("freestyle: plan failed grounding for session={}: {}", sessionId.value, summary)
|
||||
emitRejected(sessionId, "plan failed grounding: $summary", "grounding")
|
||||
return true
|
||||
if (result.verdict == PlanGroundingVerdict.PASS) return null
|
||||
return result.findings.joinToString("; ")
|
||||
}
|
||||
|
||||
private suspend fun emitPlanLint(sessionId: SessionId, candidateId: String, lint: PlanLintResult) {
|
||||
@@ -307,6 +352,28 @@ class FreestyleDriver(
|
||||
),
|
||||
),
|
||||
)
|
||||
// A rejected plan ends the session — but the last workflow verdict on record was the
|
||||
// planning phase's WorkflowCompleted, so the session read as SUCCESS (the "COMPLETED-lie").
|
||||
// Emit a session-terminal WorkflowFailed so the run's true outcome is on the log. stageId is
|
||||
// architect: the plan's producer and where a fix (or future return-to-architect loop) lands.
|
||||
eventStore.append(
|
||||
NewEvent(
|
||||
metadata = EventMetadata(
|
||||
eventId = EventId(UUID.randomUUID().toString()),
|
||||
sessionId = sessionId,
|
||||
timestamp = Clock.System.now(),
|
||||
schemaVersion = 1,
|
||||
causationId = null,
|
||||
correlationId = null,
|
||||
),
|
||||
payload = WorkflowFailedEvent(
|
||||
sessionId = sessionId,
|
||||
stageId = StageId("architect"),
|
||||
reason = "execution plan rejected ($source): $reason",
|
||||
retryExhausted = false,
|
||||
),
|
||||
),
|
||||
)
|
||||
}
|
||||
|
||||
companion object {
|
||||
|
||||
@@ -0,0 +1,38 @@
|
||||
package com.correx.apps.server.routes
|
||||
|
||||
import com.correx.apps.server.ServerModule
|
||||
import com.correx.core.events.events.ClarificationAnswer
|
||||
import com.correx.core.events.types.SessionId
|
||||
import com.correx.core.utils.TypeId
|
||||
import io.ktor.http.HttpStatusCode
|
||||
import io.ktor.server.request.receive
|
||||
import io.ktor.server.response.respond
|
||||
import io.ktor.server.routing.Route
|
||||
import io.ktor.server.routing.post
|
||||
import kotlinx.serialization.Serializable
|
||||
|
||||
// REST parity for stage clarifications (WS already has ClientMessage.ClarificationResponse). Keyed by
|
||||
// session — the server resolves the live pending (stageId, requestId) so a headless run (a script,
|
||||
// `correx run`) can clear a discovery-stage clarification without a WebSocket. If the caller omits the
|
||||
// answer for a question, its value is the empty string (free-text skip). Fixes Vikunja #42.
|
||||
@Serializable
|
||||
data class ClarifyStageRequest(val answers: List<ClarificationAnswer>)
|
||||
|
||||
internal fun Route.clarifyStageRoute(module: ServerModule) {
|
||||
post("/clarify") {
|
||||
val id = call.parameters["id"]
|
||||
if (id == null) {
|
||||
call.respond(HttpStatusCode.BadRequest, "Missing session id")
|
||||
return@post
|
||||
}
|
||||
val sessionId: SessionId = TypeId(id)
|
||||
val pending = module.orchestrator.pendingClarificationFor(sessionId)
|
||||
if (pending == null) {
|
||||
call.respond(HttpStatusCode.NotFound, "No pending clarification for session $id")
|
||||
return@post
|
||||
}
|
||||
val answers = call.receive<ClarifyStageRequest>().answers
|
||||
module.orchestrator.submitClarification(sessionId, pending.stageId, pending.requestId, answers)
|
||||
call.respond(HttpStatusCode.OK)
|
||||
}
|
||||
}
|
||||
@@ -100,6 +100,7 @@ fun Route.sessionRoutes(module: ServerModule) {
|
||||
getSessionRoute(module)
|
||||
cancelSessionRoute(module)
|
||||
approveStageRoute(module)
|
||||
clarifyStageRoute(module)
|
||||
approveSourcesRoute(module)
|
||||
undoSessionRoute(module)
|
||||
resumeSessionRoute(module)
|
||||
|
||||
@@ -0,0 +1,61 @@
|
||||
package com.correx.apps.server
|
||||
|
||||
import com.correx.core.config.ProjectConfig
|
||||
import java.nio.file.Path
|
||||
import kotlin.test.assertEquals
|
||||
import kotlin.test.assertFalse
|
||||
import kotlin.test.assertTrue
|
||||
import org.junit.jupiter.api.Test
|
||||
|
||||
class BootWorkspaceTest {
|
||||
@Test
|
||||
fun `workspace root clamps an outside configured working directory`() {
|
||||
val resolved = resolveBootWorkspace(
|
||||
explicitWorkspaceRoot = Path.of("/tmp/audition"),
|
||||
explicitWorkingDir = Path.of("/home/user/repo"),
|
||||
processWorkingDir = Path.of("/home/user/repo"),
|
||||
)
|
||||
|
||||
assertEquals(Path.of("/tmp/audition"), resolved.workspaceRoot)
|
||||
assertEquals(Path.of("/tmp/audition"), resolved.workingDir)
|
||||
assertTrue(resolved.workingDirWasClamped)
|
||||
}
|
||||
|
||||
@Test
|
||||
fun `working directory inside workspace root remains intact`() {
|
||||
val resolved = resolveBootWorkspace(
|
||||
explicitWorkspaceRoot = Path.of("/tmp/audition"),
|
||||
explicitWorkingDir = Path.of("/tmp/audition/frontend"),
|
||||
processWorkingDir = Path.of("/home/user/repo"),
|
||||
)
|
||||
|
||||
assertEquals(Path.of("/tmp/audition"), resolved.workspaceRoot)
|
||||
assertEquals(Path.of("/tmp/audition/frontend"), resolved.workingDir)
|
||||
assertFalse(resolved.workingDirWasClamped)
|
||||
}
|
||||
|
||||
@Test
|
||||
fun `configured working directory supplies the root when workspace root is absent`() {
|
||||
val resolved = resolveBootWorkspace(
|
||||
explicitWorkspaceRoot = null,
|
||||
explicitWorkingDir = Path.of("/tmp/configured"),
|
||||
processWorkingDir = Path.of("/home/user/repo"),
|
||||
)
|
||||
|
||||
assertEquals(Path.of("/tmp/configured"), resolved.workspaceRoot)
|
||||
assertEquals(Path.of("/tmp/configured"), resolved.workingDir)
|
||||
assertFalse(resolved.workingDirWasClamped)
|
||||
}
|
||||
|
||||
@Test
|
||||
fun `project memory root follows the authoritative workspace root`() {
|
||||
val configured = ProjectConfig(
|
||||
enabled = true,
|
||||
root = "/home/user/repo",
|
||||
)
|
||||
|
||||
val resolved = configured.boundToWorkspace(Path.of("/tmp/audition/../audition"))
|
||||
|
||||
assertEquals("/tmp/audition", resolved.root)
|
||||
}
|
||||
}
|
||||
@@ -0,0 +1,47 @@
|
||||
package com.correx.apps.server
|
||||
|
||||
import com.correx.core.approvals.Tier
|
||||
import com.correx.core.events.events.ToolRequest
|
||||
import com.correx.core.tools.contract.Tool
|
||||
import com.correx.core.tools.contract.ToolCapability
|
||||
import com.correx.core.tools.contract.ToolExecutor
|
||||
import com.correx.core.tools.contract.ToolResult
|
||||
import com.correx.core.tools.contract.ValidationResult
|
||||
import kotlinx.coroutines.runBlocking
|
||||
import kotlinx.serialization.json.JsonObject
|
||||
import org.junit.jupiter.api.Test
|
||||
import org.slf4j.LoggerFactory
|
||||
import java.nio.file.Path
|
||||
import kotlin.test.assertEquals
|
||||
import kotlin.test.assertNull
|
||||
|
||||
class McpIndexBootstrapTest {
|
||||
private val log = LoggerFactory.getLogger("test")
|
||||
|
||||
private class FakeTool(override val name: String) : Tool, ToolExecutor {
|
||||
var seen: ToolRequest? = null
|
||||
override val description = ""
|
||||
override val parametersSchema = JsonObject(emptyMap())
|
||||
override val tier = Tier.T2
|
||||
override val requiredCapabilities = emptySet<ToolCapability>()
|
||||
override fun validateRequest(request: ToolRequest) = ValidationResult.Valid
|
||||
override suspend fun execute(request: ToolRequest): ToolResult {
|
||||
seen = request
|
||||
return ToolResult.Success(request.invocationId, "indexed")
|
||||
}
|
||||
}
|
||||
|
||||
@Test
|
||||
fun `calls index_repository with repo_path set to the workspace root`() = runBlocking {
|
||||
val index = FakeTool("mcp__codebase-memory__index_repository")
|
||||
bootstrapCodebaseIndex(listOf(FakeTool("mcp__x__search_code"), index), Path.of("/repo"), log)
|
||||
assertEquals("/repo", index.seen?.parameters?.get("repo_path"))
|
||||
}
|
||||
|
||||
@Test
|
||||
fun `no-op when no index_repository tool is mounted`() = runBlocking {
|
||||
val other = FakeTool("mcp__x__search_code")
|
||||
bootstrapCodebaseIndex(listOf(other), Path.of("/repo"), log)
|
||||
assertNull(other.seen)
|
||||
}
|
||||
}
|
||||
@@ -6,9 +6,14 @@ import com.correx.core.artifacts.kind.JsonSchema
|
||||
import com.correx.core.events.events.CapabilityGapDetectedEvent
|
||||
import com.correx.core.events.events.CapabilityGapReflectedEvent
|
||||
import com.correx.core.events.events.CapabilityGapVerdict
|
||||
import com.correx.core.events.events.EventMetadata
|
||||
import com.correx.core.events.events.ExecutionPlanLockedEvent
|
||||
import com.correx.core.events.events.ExecutionPlanRejectedEvent
|
||||
import com.correx.core.events.events.NewEvent
|
||||
import com.correx.core.events.events.PlanLintCompletedEvent
|
||||
import com.correx.core.events.events.RepoMapComputedEvent
|
||||
import com.correx.core.events.events.RepoMapEntry
|
||||
import com.correx.core.events.types.EventId
|
||||
import com.correx.core.events.types.SessionId
|
||||
import com.correx.core.kernel.execution.WorkflowResult
|
||||
import com.correx.core.kernel.orchestration.CapabilityGapReflection
|
||||
@@ -18,6 +23,8 @@ import com.correx.core.transitions.graph.WorkflowGraph
|
||||
import com.correx.infrastructure.persistence.InMemoryEventStore
|
||||
import com.correx.infrastructure.workflow.ExecutionPlanCompiler
|
||||
import kotlinx.coroutines.runBlocking
|
||||
import kotlinx.datetime.Clock
|
||||
import java.util.UUID
|
||||
import org.junit.jupiter.api.Assertions.assertEquals
|
||||
import org.junit.jupiter.api.Assertions.assertFalse
|
||||
import org.junit.jupiter.api.Assertions.assertTrue
|
||||
@@ -471,4 +478,121 @@ class FreestyleDriverTest {
|
||||
assertTrue(payloads.filterIsInstance<CapabilityGapReflectedEvent>().isEmpty())
|
||||
assertEquals(1, payloads.filterIsInstance<ExecutionPlanLockedEvent>().size)
|
||||
}
|
||||
|
||||
// "apply" is scoped to frontend/** — with a scan recorded whose only path is backend/, and no
|
||||
// stage creating a frontend file, scope grounding fails (RETURN_TO_ARCHITECT). Passes lint.
|
||||
private val groundingFailPlanJson = """
|
||||
{
|
||||
"goal": "plan scoped to a path that doesn't exist",
|
||||
"stages": [
|
||||
{ "id": "analyse", "prompt": "Analyse", "produces": "patch", "needs": [], "tools": [] },
|
||||
{ "id": "apply", "prompt": "Apply", "produces": "patch", "needs": ["patch"], "tools": [],
|
||||
"touches": ["frontend/**"] }
|
||||
],
|
||||
"edges": [
|
||||
{ "from": "analyse", "to": "apply", "condition": { "type": "always_true" } },
|
||||
{ "from": "apply", "to": "done", "condition": { "type": "always_true" } }
|
||||
]
|
||||
}
|
||||
""".trimIndent()
|
||||
|
||||
// Records a scan (scanned=true) whose only path is backend/, so frontend/** grounds to nothing.
|
||||
private suspend fun recordBackendScan(store: InMemoryEventStore, sessionId: SessionId) {
|
||||
store.append(
|
||||
NewEvent(
|
||||
metadata = EventMetadata(
|
||||
eventId = EventId(UUID.randomUUID().toString()),
|
||||
sessionId = sessionId,
|
||||
timestamp = Clock.System.now(),
|
||||
schemaVersion = 1,
|
||||
causationId = null,
|
||||
correlationId = null,
|
||||
),
|
||||
payload = RepoMapComputedEvent(
|
||||
sessionId = sessionId,
|
||||
repoRoot = "/ws",
|
||||
entries = listOf(RepoMapEntry(path = "backend/main.kt", score = 1.0)),
|
||||
computedAt = Clock.System.now(),
|
||||
),
|
||||
),
|
||||
)
|
||||
}
|
||||
|
||||
@Test
|
||||
fun `grounding failure returns plan to architect then locks once the re-run yields a grounded plan`(): Unit =
|
||||
runBlocking {
|
||||
val sessionId = SessionId("driver-grounding-retry-session")
|
||||
val eventStore = InMemoryEventStore()
|
||||
val compiler = ExecutionPlanCompiler(buildRegistry())
|
||||
recordBackendScan(eventStore, sessionId)
|
||||
|
||||
var corrected = false
|
||||
var rerunInvocations = 0
|
||||
var runPhase2Invocations = 0
|
||||
|
||||
val driver = FreestyleDriver(
|
||||
eventStore = eventStore,
|
||||
compiler = compiler,
|
||||
// Fails grounding until the architect re-run "fixes" it (flips to the unconstrained plan).
|
||||
planContent = { if (corrected) validPlanJson else groundingFailPlanJson },
|
||||
config = OrchestrationConfig(),
|
||||
runPhase2 = { sid, graph, _ ->
|
||||
runPhase2Invocations++
|
||||
WorkflowResult.Completed(sid, graph.start)
|
||||
},
|
||||
rerunArchitect = { sid ->
|
||||
rerunInvocations++
|
||||
corrected = true
|
||||
WorkflowResult.Completed(sid, com.correx.core.events.types.StageId("architect"))
|
||||
},
|
||||
)
|
||||
|
||||
driver.lockAndRun(sessionId)
|
||||
|
||||
assertEquals(1, rerunInvocations, "architect should be re-run exactly once")
|
||||
val payloads = eventStore.read(sessionId).map { it.payload }
|
||||
assertEquals(1, payloads.filterIsInstance<ExecutionPlanLockedEvent>().size, "corrected plan should lock")
|
||||
assertEquals(1, runPhase2Invocations, "runPhase2 should run once on the corrected plan")
|
||||
assertTrue(
|
||||
payloads.filterIsInstance<ExecutionPlanRejectedEvent>().isEmpty(),
|
||||
"no rejection once the re-run grounds",
|
||||
)
|
||||
}
|
||||
|
||||
@Test
|
||||
fun `grounding failure that never resolves is rejected with source grounding after exhausting retries`(): Unit =
|
||||
runBlocking {
|
||||
val sessionId = SessionId("driver-grounding-exhaust-session")
|
||||
val eventStore = InMemoryEventStore()
|
||||
val compiler = ExecutionPlanCompiler(buildRegistry())
|
||||
recordBackendScan(eventStore, sessionId)
|
||||
|
||||
var rerunInvocations = 0
|
||||
var runPhase2Invocations = 0
|
||||
|
||||
val driver = FreestyleDriver(
|
||||
eventStore = eventStore,
|
||||
compiler = compiler,
|
||||
planContent = { groundingFailPlanJson }, // never corrected
|
||||
config = OrchestrationConfig(),
|
||||
runPhase2 = { sid, graph, _ ->
|
||||
runPhase2Invocations++
|
||||
WorkflowResult.Completed(sid, graph.start)
|
||||
},
|
||||
rerunArchitect = { sid ->
|
||||
rerunInvocations++
|
||||
WorkflowResult.Completed(sid, com.correx.core.events.types.StageId("architect"))
|
||||
},
|
||||
maxGroundingRetries = 2,
|
||||
)
|
||||
|
||||
driver.lockAndRun(sessionId)
|
||||
|
||||
assertEquals(2, rerunInvocations, "architect re-run should be capped at maxGroundingRetries")
|
||||
assertEquals(0, runPhase2Invocations, "runPhase2 must not run when grounding never clears")
|
||||
val payloads = eventStore.read(sessionId).map { it.payload }
|
||||
assertTrue(payloads.filterIsInstance<ExecutionPlanLockedEvent>().isEmpty(), "never locks")
|
||||
val rejected = payloads.filterIsInstance<ExecutionPlanRejectedEvent>().single()
|
||||
assertEquals("grounding", rejected.source)
|
||||
}
|
||||
}
|
||||
|
||||
@@ -24,6 +24,7 @@ CORREX kernel team. Config schema changes affect all consumers — coordinate wi
|
||||
- `ConfigHolder` is the shared mutable reference injected into consumers. Never read the file directly in domain code — always go through `ConfigHolder`.
|
||||
- `CorrexConfigWriter` regenerates TOML from the in-memory model; round-tripping loses comments by design.
|
||||
- `[git]` is opt-in server transport configuration (`enabled`, `remote`, `base_branch`, optional `author`); it is off by default.
|
||||
- `[orchestration].review_loop_max_cycles` bounds review→rework cycles before recovery escalation; default `3`.
|
||||
- `ModelConfig.contextSize` defaults to 24,576 tokens; explicit `context_size` remains the operator override for smaller local-model windows.
|
||||
|
||||
## Verification
|
||||
|
||||
@@ -217,6 +217,7 @@ object ConfigLoader {
|
||||
private const val DEFAULT_MAX_CLARIFICATION_ROUNDS = 3
|
||||
private const val DEFAULT_REVIEW_BLOCK_MIN_CONFIDENCE = 0.7
|
||||
private const val DEFAULT_REVIEW_BLOCK_RETRY_CAP = 20
|
||||
private const val DEFAULT_REVIEW_LOOP_MAX_CYCLES = 3
|
||||
private const val DEFAULT_MAX_REFINEMENT = 3
|
||||
private const val DEFAULT_RECOVERY_ROUTE_BUDGET = 2
|
||||
private const val DEFAULT_INTENT_ROUTE_BUDGET = 2
|
||||
@@ -716,6 +717,8 @@ object ConfigLoader {
|
||||
reviewBlockMinConfidence =
|
||||
asDouble(orchestrationSection["review_block_min_confidence"], DEFAULT_REVIEW_BLOCK_MIN_CONFIDENCE),
|
||||
reviewBlockRetryCap = asInt(orchestrationSection["review_block_retry_cap"], DEFAULT_REVIEW_BLOCK_RETRY_CAP),
|
||||
reviewLoopMaxCycles =
|
||||
asInt(orchestrationSection["review_loop_max_cycles"], DEFAULT_REVIEW_LOOP_MAX_CYCLES),
|
||||
defaultMaxRefinement = asInt(orchestrationSection["default_max_refinement"], DEFAULT_MAX_REFINEMENT),
|
||||
recoveryRouteBudget = asInt(orchestrationSection["recovery_route_budget"], DEFAULT_RECOVERY_ROUTE_BUDGET),
|
||||
intentRouteBudget = asInt(orchestrationSection["intent_route_budget"], DEFAULT_INTENT_ROUTE_BUDGET),
|
||||
|
||||
@@ -130,6 +130,8 @@ data class OrchestrationKnobs(
|
||||
val maxClarificationRounds: Int = 3,
|
||||
val reviewBlockMinConfidence: Double = 0.7,
|
||||
val reviewBlockRetryCap: Int = 20,
|
||||
/** Review→rework cycles before deterministic escalation to the recovery stage. */
|
||||
val reviewLoopMaxCycles: Int = 3,
|
||||
val defaultMaxRefinement: Int = 3,
|
||||
val recoveryRouteBudget: Int = 2,
|
||||
val intentRouteBudget: Int = 2,
|
||||
|
||||
@@ -98,6 +98,7 @@ object CorrexConfigWriter {
|
||||
b.kv("max_clarification_rounds", cfg.orchestration.maxClarificationRounds)
|
||||
b.kv("review_block_min_confidence", cfg.orchestration.reviewBlockMinConfidence)
|
||||
b.kv("review_block_retry_cap", cfg.orchestration.reviewBlockRetryCap)
|
||||
b.kv("review_loop_max_cycles", cfg.orchestration.reviewLoopMaxCycles)
|
||||
b.kv("default_max_refinement", cfg.orchestration.defaultMaxRefinement)
|
||||
b.kv("recovery_route_budget", cfg.orchestration.recoveryRouteBudget)
|
||||
b.kv("intent_route_budget", cfg.orchestration.intentRouteBudget)
|
||||
|
||||
@@ -20,6 +20,7 @@ CORREX kernel team. This is the most cross-cutting module in the codebase — ch
|
||||
- `JsonEventSerializer` / `EventSerializer` — serialize/deserialize `StoredEvent` to JSON.
|
||||
- `EventDispatcher` — broadcasts events to in-process listeners.
|
||||
- Domain event files: `ApprovalEvents`, `ArtifactEvents`, `ContextEvents`, `InferenceEvents`, `OrchestrationEvents`, `RouterEvents`, `SessionEvents`, `TaskEvents`, `ToolEvents`, `IntentEvents`, `RiskAssessedEvent`, `JournalCompactedEvent`, and many more — all payload definitions live here.
|
||||
- `LspDiagnosticsCompletedEvent` records pulled language-server diagnostics or a graceful skip reason; replay consumes this observation and never contacts the server.
|
||||
- Shared vocabulary: `IdentityTypes` (SessionId, TaskId, etc.), `Tier`, `TokenUsage`, `ToolReceipt`, `ToolRequest`, `RiskLevel`, `RetryPolicy`, `GrantScope`, `GrantLedger`.
|
||||
|
||||
## Work Guidance
|
||||
|
||||
@@ -0,0 +1,27 @@
|
||||
package com.correx.core.events.events
|
||||
|
||||
import com.correx.core.events.types.SessionId
|
||||
import com.correx.core.events.types.StageId
|
||||
import kotlinx.serialization.SerialName
|
||||
import kotlinx.serialization.Serializable
|
||||
|
||||
@Serializable
|
||||
data class LspDiagnostic(
|
||||
val path: String,
|
||||
val line: Int,
|
||||
val character: Int,
|
||||
val severity: String,
|
||||
val code: String? = null,
|
||||
val message: String,
|
||||
)
|
||||
|
||||
/** Recorded LSP 3.17 pull-diagnostic observation; replay never re-queries a language server. */
|
||||
@Serializable
|
||||
@SerialName("LspDiagnosticsCompleted")
|
||||
data class LspDiagnosticsCompletedEvent(
|
||||
val sessionId: SessionId,
|
||||
val stageId: StageId,
|
||||
val server: String?,
|
||||
val diagnostics: List<LspDiagnostic>,
|
||||
val skippedReason: String? = null,
|
||||
) : EventPayload
|
||||
@@ -16,6 +16,7 @@ import com.correx.core.events.events.BriefEchoMismatchEvent
|
||||
import com.correx.core.events.events.BriefGroundingCheckedEvent
|
||||
import com.correx.core.events.events.ConceptPromotedEvent
|
||||
import com.correx.core.events.events.StaticAnalysisCompletedEvent
|
||||
import com.correx.core.events.events.LspDiagnosticsCompletedEvent
|
||||
import com.correx.core.events.events.ContractGateEvaluatedEvent
|
||||
import com.correx.core.events.events.PlanLintCompletedEvent
|
||||
import com.correx.core.events.events.ChatSessionStartedEvent
|
||||
@@ -173,6 +174,7 @@ val eventModule = SerializersModule {
|
||||
subclass(BriefGroundingCheckedEvent::class)
|
||||
subclass(BriefEchoMismatchEvent::class)
|
||||
subclass(StaticAnalysisCompletedEvent::class)
|
||||
subclass(LspDiagnosticsCompletedEvent::class)
|
||||
subclass(ContractGateEvaluatedEvent::class)
|
||||
subclass(PlanLintCompletedEvent::class)
|
||||
subclass(RiskAssessedEvent::class)
|
||||
|
||||
+23
@@ -0,0 +1,23 @@
|
||||
package com.correx.core.events.serialization
|
||||
|
||||
import com.correx.core.events.events.LspDiagnostic
|
||||
import com.correx.core.events.events.LspDiagnosticsCompletedEvent
|
||||
import com.correx.core.events.types.SessionId
|
||||
import com.correx.core.events.types.StageId
|
||||
import kotlin.test.Test
|
||||
import kotlin.test.assertEquals
|
||||
import kotlin.test.assertIs
|
||||
|
||||
class LspDiagnosticsCompletedEventSerializationTest {
|
||||
@Test
|
||||
fun `round-trips as polymorphic EventPayload`() {
|
||||
val event = LspDiagnosticsCompletedEvent(
|
||||
SessionId("s"), StageId("impl"), "tsserver",
|
||||
listOf(LspDiagnostic("src/App.tsx", 1, 2, "error", "2322", "not assignable")),
|
||||
)
|
||||
val encoded = eventJson.encodeToString(com.correx.core.events.events.EventPayload.serializer(), event)
|
||||
val decoded = eventJson.decodeFromString(com.correx.core.events.events.EventPayload.serializer(), encoded)
|
||||
assertIs<LspDiagnosticsCompletedEvent>(decoded)
|
||||
assertEquals("src/App.tsx", decoded.diagnostics.single().path)
|
||||
}
|
||||
}
|
||||
@@ -19,6 +19,8 @@ CORREX kernel team. This is the integration point for all other `core/` modules.
|
||||
- `ReplayOrchestrator` / `ReplayInferenceProvider` / `ReplayStrategy` — deterministic replay of a session from its event log. `ReplayInferenceProvider` returns recorded responses — no live LLM (Hard Invariant #8).
|
||||
- `SubagentRunner` / `InSessionSubagentRunner` — runs sub-agent invocations within an active session.
|
||||
- `StaticAnalysisRunner` / `ProcessStaticAnalysisRunner` — runs static analysis tools and records results as events.
|
||||
- `LspDiagnosticsRunner` — injected pull-diagnostics seam; diagnostics are filtered to stage-written files, recorded, and enforced before build/review.
|
||||
- Review→rework loops use the configured three-cycle default, then route accumulated notes to recovery once and fail if the fixed DoD still cannot be approved.
|
||||
- `StageCheckpointReconciler` — reconciles checkpoint state across stage transitions.
|
||||
- Capability-gated failures first retry in place when the stage holds the required tool; an unchanged-fingerprint gate-budget exhaustion routes to the recovery/intent-holder stage when one is available, so capability possession alone cannot cause a frozen owner loop to fail the workflow.
|
||||
- Three repeated `REFERENCE_EXISTS` blocks for the same build prerequisite within one stage become a `workspace_precondition` gate failure, which is eligible for file-write recovery rather than remaining disconnected tool-call noise.
|
||||
|
||||
+47
-2
@@ -10,6 +10,8 @@ import com.correx.core.context.model.ContextLayer
|
||||
import com.correx.core.context.model.EntryRole
|
||||
import com.correx.core.events.events.FailureTicketOpenedEvent
|
||||
import com.correx.core.events.events.InitialIntentEvent
|
||||
import com.correx.core.events.events.PlanGroundingEvaluatedEvent
|
||||
import com.correx.core.events.events.PlanGroundingVerdict
|
||||
import com.correx.core.events.events.RefinementIterationEvent
|
||||
import com.correx.core.events.events.RetryAttemptedEvent
|
||||
import com.correx.core.events.events.StoredEvent
|
||||
@@ -40,6 +42,35 @@ fun buildRetryFeedbackEntry(events: List<StoredEvent>, stageId: StageId): Contex
|
||||
)
|
||||
}
|
||||
|
||||
/**
|
||||
* Feeds the deterministic plan-grounding findings back into the architect stage when the freestyle
|
||||
* driver returned its plan for another attempt (grounding verdict != PASS). The findings are already
|
||||
* recorded on [PlanGroundingEvaluatedEvent] (invariant #9), so this reads them rather than re-deriving.
|
||||
* Gated to the plan-producing stage — "architect" in freestyle_planning, the same stage id the driver
|
||||
* stamps on rejection. Last event wins: a re-run that grounds PASS clears the feedback automatically.
|
||||
*/
|
||||
fun buildGroundingFeedbackEntry(events: List<StoredEvent>, stageId: StageId): ContextEntry? {
|
||||
if (stageId.value != "architect") return null
|
||||
val latest = events.mapNotNull { it.payload as? PlanGroundingEvaluatedEvent }.lastOrNull() ?: return null
|
||||
if (latest.verdict == PlanGroundingVerdict.PASS) return null
|
||||
val content = "## Plan grounding feedback\n" +
|
||||
"Your previous execution_plan was returned — it does not hold against the workspace facts:\n" +
|
||||
latest.findings.joinToString("\n") { "- $it" } + "\n" +
|
||||
"Emit a corrected plan that resolves every point above. Typical fixes: declare the manifest a " +
|
||||
"build stage needs (have an earlier stage create it via `writes`/`expectedFiles`), narrow a " +
|
||||
"stage's `touches` to paths that exist or that an earlier stage creates, or drop a build stage " +
|
||||
"that has nothing to build. Do not repeat the identical plan."
|
||||
return ContextEntry(
|
||||
id = ContextEntryId(UUID.randomUUID().toString()),
|
||||
layer = ContextLayer.L1,
|
||||
content = content,
|
||||
sourceType = "groundingFeedback",
|
||||
sourceId = stageId.value,
|
||||
tokenEstimate = content.length / 4,
|
||||
role = EntryRole.SYSTEM,
|
||||
)
|
||||
}
|
||||
|
||||
/**
|
||||
* Feeds the open failure ticket into the recovery stage's context. The recovery stage was entered
|
||||
* by kernel routing (not a normal edge) precisely because an upstream write-less stage failed a gate
|
||||
@@ -51,10 +82,24 @@ fun buildRecoveryTicketEntry(events: List<StoredEvent>, stageId: StageId): Conte
|
||||
val ticket = events
|
||||
.mapNotNull { it.payload as? FailureTicketOpenedEvent }
|
||||
.lastOrNull { it.routeTo == stageId } ?: return null
|
||||
val content = if (ticket.escalated) {
|
||||
val intent = events.mapNotNull { it.payload as? InitialIntentEvent }.lastOrNull()?.intent
|
||||
val content = if (ticket.gate == STAGE_LOOP_BREAK_GATE) {
|
||||
// A stuck-loop route is NOT a cross-file contract dispute — the gate output names no files, it
|
||||
// is the SAME action (often a malformed tool call) repeated until the loop-break tripped. The
|
||||
// arbitration prompt (read/reconcile the named files) sends the model hunting for files that
|
||||
// don't exist. Tell it plainly: the last action is futile, take a materially different one.
|
||||
"## Stuck-loop ticket\n" +
|
||||
"Stage '${ticket.stageId.value}' repeated the SAME failing action until it tripped the " +
|
||||
"'${ticket.gate}' gate. Retrying that action again is futile — it will fail identically. The " +
|
||||
"exact failure is below; it is a single stuck step, NOT a dispute between files, so do not go " +
|
||||
"looking for files to reconcile. If it is a malformed tool call, fix the call's shape and " +
|
||||
"continue the task; otherwise take a materially different route to the same goal. Serve the " +
|
||||
"intent" + (intent?.let { " below" } ?: "") + ", then control returns to verification." +
|
||||
(intent?.let { "\n### Intent (authoritative)\n$it" } ?: "") +
|
||||
"\n### Gate output (the failing action)\n${ticket.evidence}"
|
||||
} else if (ticket.escalated) {
|
||||
// Tier 2: the owner loop couldn't settle it — a cross-file contract dispute. The arbiter holds
|
||||
// the intent and reconciles ALL sides in one pass.
|
||||
val intent = events.mapNotNull { it.payload as? InitialIntentEvent }.lastOrNull()?.intent
|
||||
"## Contract arbitration ticket\n" +
|
||||
"Stage '${ticket.stageId.value}' keeps failing the '${ticket.gate}' gate even after the " +
|
||||
"file owners repaired their own layers — so this is NOT a bug in one file. The files named " +
|
||||
|
||||
+32
-9
@@ -15,6 +15,7 @@ import com.correx.core.events.events.OrchestrationResumedEvent
|
||||
import com.correx.core.events.events.SteeringNoteAddedEvent
|
||||
import com.correx.core.events.events.StoredEvent
|
||||
import com.correx.core.events.events.TransitionExecutedEvent
|
||||
import com.correx.core.events.events.WorkflowStartedEvent
|
||||
import com.correx.core.events.orchestration.OrchestrationState
|
||||
import com.correx.core.events.types.ApprovalDecisionId
|
||||
import com.correx.core.events.types.ApprovalRequestId
|
||||
@@ -130,15 +131,30 @@ class DefaultSessionOrchestrator(
|
||||
sessionId: SessionId,
|
||||
graph: WorkflowGraph,
|
||||
config: OrchestrationConfig,
|
||||
): WorkflowResult {
|
||||
log.debug("[Orchestrator] session={} workflow={} start={}", sessionId.value, graph.id, graph.start.value)
|
||||
emitWorkflowStarted(sessionId, graph, config)
|
||||
): WorkflowResult = runFrom(sessionId, graph, config, graph.start)
|
||||
|
||||
val base = ExecutionContext(graph, sessionId, 0, graph.start, config, null, null)
|
||||
/**
|
||||
* Runs [graph] entering at [startStage] instead of [graph].start. Used by the freestyle
|
||||
* return-to-architect loop to re-enter just the plan-producing stage (with grounding feedback
|
||||
* already in its L1 context) without redoing discovery/analyst. [startStage] must be in [graph].
|
||||
*/
|
||||
suspend fun runFrom(
|
||||
sessionId: SessionId,
|
||||
graph: WorkflowGraph,
|
||||
config: OrchestrationConfig,
|
||||
startStage: StageId,
|
||||
): WorkflowResult {
|
||||
require(graph.stages.containsKey(startStage)) {
|
||||
"startStage '${startStage.value}' is not a stage of workflow '${graph.id}'"
|
||||
}
|
||||
log.debug("[Orchestrator] session={} workflow={} start={}", sessionId.value, graph.id, startStage.value)
|
||||
emitWorkflowStarted(sessionId, graph, config, startStage)
|
||||
|
||||
val base = ExecutionContext(graph, sessionId, 0, startStage, config, null, null)
|
||||
val enriched = base.enrich()
|
||||
|
||||
// Execute the start stage before entering the step loop
|
||||
return when (val result = enterStage(enriched, graph.start)) {
|
||||
return when (val result = enterStage(enriched, startStage)) {
|
||||
is StepResult.Continue -> step(result.ctx)
|
||||
is StepResult.Terminal -> result.result
|
||||
}
|
||||
@@ -226,8 +242,8 @@ class DefaultSessionOrchestrator(
|
||||
|
||||
/**
|
||||
* Delivers the operator's answers to a pending clarification raised by a stage. The waiting
|
||||
* stage (parked in [requestClarificationIfNeeded]) completes and re-runs with the answers in
|
||||
* context. If the server restarted while the clarification was pending there is no live
|
||||
* stage (parked in [requestClarificationIfNeeded]) unparks and the run advances to the next
|
||||
* stage with the answers in context. If the server restarted while the clarification was pending there is no live
|
||||
* coroutine to complete, so the answers are recorded directly and the session resumed.
|
||||
*/
|
||||
suspend fun submitClarification(
|
||||
@@ -280,10 +296,17 @@ class DefaultSessionOrchestrator(
|
||||
)
|
||||
}
|
||||
|
||||
// A back-edge re-enters a stage already transitioned into this run (e.g. reviewer→implementer).
|
||||
// A back-edge re-enters any stage already visited this run (e.g. reviewer→implementer). The start
|
||||
// stage is visited via WorkflowStarted rather than TransitionExecuted, so it must participate too.
|
||||
// Top-level (not a member) to keep the orchestrator off the TooManyFunctions threshold.
|
||||
internal fun isBackEdge(events: List<StoredEvent>, target: StageId): Boolean =
|
||||
events.any { (it.payload as? TransitionExecutedEvent)?.to == target }
|
||||
events.any {
|
||||
when (val payload = it.payload) {
|
||||
is WorkflowStartedEvent -> payload.startStageId == target
|
||||
is TransitionExecutedEvent -> payload.to == target
|
||||
else -> false
|
||||
}
|
||||
}
|
||||
|
||||
internal sealed class StepResult {
|
||||
data class Continue(val ctx: EnrichedExecutionContext) : StepResult()
|
||||
|
||||
+45
-6
@@ -5,6 +5,7 @@ import com.correx.core.artifacts.ArtifactState
|
||||
import com.correx.core.events.events.ApprovalDecisionResolvedEvent
|
||||
import com.correx.core.events.events.ApprovalRequestedEvent
|
||||
import com.correx.core.events.events.ArtifactValidatedEvent
|
||||
import com.correx.core.events.events.FailureTicketOpenedEvent
|
||||
import com.correx.core.events.events.RefinementIterationEvent
|
||||
import com.correx.core.events.events.RetrySalvageDecidedEvent
|
||||
import com.correx.core.events.events.SalvageDecision
|
||||
@@ -135,6 +136,7 @@ internal tailrec suspend fun DefaultSessionOrchestrator.step(ctx: EnrichedExecut
|
||||
* re-execute the same stage if attempts remain, else fail terminally with retryExhausted=true.
|
||||
*/
|
||||
|
||||
@Suppress("LongMethod", "ReturnCount", "NestedBlockDepth")
|
||||
internal suspend fun DefaultSessionOrchestrator.executeMove(
|
||||
ctx: EnrichedExecutionContext,
|
||||
decision: TransitionDecision.Move,
|
||||
@@ -154,10 +156,43 @@ internal suspend fun DefaultSessionOrchestrator.executeMove(
|
||||
// terminal failure instead of looping forever.
|
||||
if (isBackEdge(repositories.eventStore.read(ctx.sessionId), nextStageId)) {
|
||||
val cycleKey = "${ctx.currentStageId.value}->${nextStageId.value}"
|
||||
val maxIterations = ctx.graph.stages[nextStageId]?.maxRetries ?: tuning.defaultMaxRefinement
|
||||
val reviewerRole = ctx.graph.stages[ctx.currentStageId]?.metadata?.get("role")?.lowercase()
|
||||
val reviewLoop = reviewerRole in setOf("review", "reviewer")
|
||||
val maxIterations = if (reviewLoop) {
|
||||
tuning.reviewLoopMaxCycles
|
||||
} else {
|
||||
ctx.graph.stages[nextStageId]?.maxRetries ?: tuning.defaultMaxRefinement
|
||||
}
|
||||
val iteration = (orchestrationRepository.getState(ctx.sessionId).refinementIterations[cycleKey] ?: 0) + 1
|
||||
emit(ctx.sessionId, RefinementIterationEvent(ctx.sessionId, cycleKey, iteration, maxIterations))
|
||||
if (iteration > maxIterations) {
|
||||
if (reviewLoop) {
|
||||
val events = repositories.eventStore.read(ctx.sessionId)
|
||||
val alreadyRecovered = events.mapNotNull { it.payload as? FailureTicketOpenedEvent }
|
||||
.any { it.gate == REVIEW_LOOP_GATE && it.stageId == ctx.currentStageId }
|
||||
val notes = ctx.graph.stages[ctx.currentStageId]?.produces
|
||||
?.mapNotNull { artifactContentCache["${ctx.sessionId.value}:${it.name.value}"] }
|
||||
?.joinToString("\n\n")
|
||||
.orEmpty()
|
||||
val reason = "review loop exhausted after exactly $maxIterations cycles. " +
|
||||
"The fixed DoD was not approved. Accumulated review notes:\n" +
|
||||
notes.ifBlank { "(review stage emitted no retained notes)" }
|
||||
if (!alreadyRecovered) {
|
||||
return routeToRecovery(
|
||||
ctx,
|
||||
ctx.currentStageId,
|
||||
REVIEW_LOOP_GATE,
|
||||
"review_convergence",
|
||||
reason,
|
||||
orchestrationRepository.getState(ctx.sessionId),
|
||||
) ?: StepResult.Terminal(
|
||||
failWorkflow(ctx.sessionId, ctx.currentStageId, reason, retryExhausted = true),
|
||||
)
|
||||
}
|
||||
return StepResult.Terminal(
|
||||
failWorkflow(ctx.sessionId, ctx.currentStageId, reason, retryExhausted = true),
|
||||
)
|
||||
}
|
||||
return StepResult.Terminal(
|
||||
failWorkflow(
|
||||
ctx.sessionId,
|
||||
@@ -179,6 +214,8 @@ internal suspend fun DefaultSessionOrchestrator.executeMove(
|
||||
}
|
||||
}
|
||||
|
||||
private const val REVIEW_LOOP_GATE = "review_loop"
|
||||
|
||||
@Suppress("ReturnCount")
|
||||
internal suspend fun DefaultSessionOrchestrator.enterStage(
|
||||
ctx: EnrichedExecutionContext,
|
||||
@@ -224,11 +261,13 @@ internal suspend fun DefaultSessionOrchestrator.enterStage(
|
||||
).outcome
|
||||
when (result) {
|
||||
is StageExecutionResult.Success -> {
|
||||
if (requestClarificationIfNeeded(ctx.sessionId, stageId, ctx.graph)) {
|
||||
// The stage raised open questions and the operator answered them; loop to
|
||||
// re-run the stage with the answers injected (no failure-retry budget spent).
|
||||
continue
|
||||
}
|
||||
// The stage may emit open questions in its artifact (discovery does). Park, let the
|
||||
// operator answer, and record the answers — then ADVANCE, do not re-run the stage.
|
||||
// A re-run restarts inference with no prior CoT, so the stage re-explores instead of
|
||||
// converging (2026-07-18). The answers are recorded as ClarificationAnsweredEvents
|
||||
// and injected into every later stage's L0 context (buildClarificationAnswerEntries),
|
||||
// so the next stage (e.g. analyst) sees them without discovery running again.
|
||||
requestClarificationIfNeeded(ctx.sessionId, stageId, ctx.graph)
|
||||
compactionService?.let { svc ->
|
||||
val journalState = decisionJournalRepository.getJournal(ctx.sessionId)
|
||||
val journalText = DecisionJournalRenderer().render(journalState)
|
||||
|
||||
+12
-10
@@ -43,16 +43,18 @@ class JournalCompactionService(
|
||||
}
|
||||
|
||||
val summaryArtifactId = artifactStore.put(summaryText.toByteArray(Charsets.UTF_8))
|
||||
artifactStore.flushBefore {
|
||||
emit(
|
||||
JournalCompactedEvent(
|
||||
sessionId = sessionId,
|
||||
coversThroughSequence = throughSequence,
|
||||
summaryArtifactId = summaryArtifactId,
|
||||
lowSalienceOmittedCount = lowCount,
|
||||
),
|
||||
)
|
||||
}
|
||||
// Emit directly, NOT inside flushBefore { }: append() already fsyncs artifacts before it
|
||||
// persists the event, so wrapping the emit here re-acquires the (non-reentrant) artifact
|
||||
// Mutex that flushBefore already holds → self-deadlock. Only surfaces on long runs, which
|
||||
// are the ones that cross the compaction threshold.
|
||||
emit(
|
||||
JournalCompactedEvent(
|
||||
sessionId = sessionId,
|
||||
coversThroughSequence = throughSequence,
|
||||
summaryArtifactId = summaryArtifactId,
|
||||
lowSalienceOmittedCount = lowCount,
|
||||
),
|
||||
)
|
||||
return true
|
||||
}
|
||||
}
|
||||
|
||||
+20
@@ -0,0 +1,20 @@
|
||||
package com.correx.core.kernel.orchestration
|
||||
|
||||
import com.correx.core.events.events.LspDiagnostic
|
||||
import java.nio.file.Path
|
||||
|
||||
data class LspDiagnosticsRequest(
|
||||
val workspaceRoot: Path,
|
||||
val paths: List<String>,
|
||||
)
|
||||
|
||||
data class LspDiagnosticsResult(
|
||||
val server: String? = null,
|
||||
val diagnostics: List<LspDiagnostic> = emptyList(),
|
||||
val skippedReason: String? = null,
|
||||
)
|
||||
|
||||
/** Nondeterministic LSP boundary. The orchestrator records its result before using it. */
|
||||
fun interface LspDiagnosticsRunner {
|
||||
suspend fun pull(request: LspDiagnosticsRequest): LspDiagnosticsResult
|
||||
}
|
||||
+2
@@ -35,6 +35,8 @@ data class OrchestrationTuning(
|
||||
val reviewBlockMinConfidence: Double = 0.7,
|
||||
/** Pathological backstop: max review-driven retries before the stage is let through. */
|
||||
val reviewBlockRetryCap: Int = 20,
|
||||
/** Review→rework cycles before deterministic escalation to recovery. */
|
||||
val reviewLoopMaxCycles: Int = 3,
|
||||
/** Max review→refine cycles for a stage (freestyle default refinement budget). */
|
||||
val defaultMaxRefinement: Int = 3,
|
||||
/** Budget for routing a failed write-less stage to a recovery stage. */
|
||||
|
||||
+1
@@ -31,6 +31,7 @@ data class OrchestratorEngines(
|
||||
// Runs operator-configured static-analysis commands as a harness step (role-reliability §5).
|
||||
// Null = no static-first step; a stage declaring `static_analysis` then no-ops with a warning.
|
||||
val staticAnalysisRunner: StaticAnalysisRunner? = null,
|
||||
val lspDiagnosticsRunner: LspDiagnosticsRunner? = null,
|
||||
// Evaluates Gate 2 contract assertions against a stage's produced files (design §1). Null = no
|
||||
// contract gate; the deterministic funnel then rests on produces-presence + static analysis only.
|
||||
val contractAssertionEvaluator: ContractAssertionEvaluator? = null,
|
||||
|
||||
+49
-9
@@ -4,6 +4,7 @@ import com.correx.core.tools.process.ChildProcess
|
||||
import kotlinx.coroutines.Dispatchers
|
||||
import kotlinx.coroutines.TimeoutCancellationException
|
||||
import kotlinx.coroutines.async
|
||||
import kotlinx.coroutines.coroutineScope
|
||||
import kotlinx.coroutines.withContext
|
||||
import kotlinx.coroutines.withTimeout
|
||||
import java.nio.file.Path
|
||||
@@ -26,31 +27,70 @@ class ProcessStaticAnalysisRunner(
|
||||
withContext(Dispatchers.IO) {
|
||||
val argv = command.trim().split(WHITESPACE).filter { it.isNotEmpty() }
|
||||
if (argv.isEmpty()) return@withContext StaticAnalysisRunResult(EXIT_NOT_RUN, "empty command")
|
||||
if (argv.first() == STATIC_FLOOR_COMMAND) return@withContext runStaticFloor(workingDir, argv)
|
||||
runProcess(workingDir, argv, command)
|
||||
}
|
||||
|
||||
@Suppress("ReturnCount")
|
||||
private suspend fun runStaticFloor(workingDir: Path, argv: List<String>): StaticAnalysisRunResult {
|
||||
val paths = argv.dropWhile { it != "--" }.drop(1)
|
||||
if (paths.isEmpty()) return StaticAnalysisRunResult(0, "static floor skipped: no concrete files")
|
||||
val checks = paths.mapNotNull { path -> checkerFor(path)?.let { it + path } }
|
||||
if (checks.isEmpty()) {
|
||||
return StaticAnalysisRunResult(0, "static floor skipped: no resolvable checker for ${paths.joinToString()}")
|
||||
}
|
||||
val outputs = mutableListOf<String>()
|
||||
for (check in checks) {
|
||||
val result = runProcess(workingDir, check, check.joinToString(" "))
|
||||
outputs += "${check.joinToString(" ")}: ${result.output}".trim()
|
||||
if (result.exitCode != 0) return StaticAnalysisRunResult(result.exitCode, outputs.joinToString("\n"))
|
||||
}
|
||||
val skipped = paths.size - checks.size
|
||||
if (skipped > 0) outputs += "static floor skipped $skipped file(s) without a checker"
|
||||
return StaticAnalysisRunResult(0, outputs.joinToString("\n"))
|
||||
}
|
||||
|
||||
private fun checkerFor(path: String): List<String>? = when (path.substringAfterLast('.', "").lowercase()) {
|
||||
"js", "mjs", "cjs" -> listOf("node", "--check")
|
||||
"py" -> listOf("python3", "-m", "py_compile")
|
||||
"sh", "bash" -> listOf("bash", "-n")
|
||||
"rb" -> listOf("ruby", "-c")
|
||||
else -> null
|
||||
}
|
||||
|
||||
private suspend fun runProcess(
|
||||
workingDir: Path,
|
||||
argv: List<String>,
|
||||
displayCommand: String,
|
||||
): StaticAnalysisRunResult {
|
||||
val process = runCatching {
|
||||
ChildProcess.builder(argv, workingDir.toFile()).redirectErrorStream(true).start()
|
||||
}.getOrElse {
|
||||
return@withContext StaticAnalysisRunResult(EXIT_NOT_RUN, "failed to start '$command': ${it.message}")
|
||||
return StaticAnalysisRunResult(EXIT_NOT_RUN, "failed to start '$displayCommand': ${it.message}")
|
||||
}
|
||||
runCatching {
|
||||
withTimeout(timeoutMs) {
|
||||
val outputDeferred = async { process.inputStream.bufferedReader().use { it.readText() } }
|
||||
val exit = process.waitFor()
|
||||
StaticAnalysisRunResult(exit, outputDeferred.await())
|
||||
return runCatching {
|
||||
coroutineScope {
|
||||
withTimeout(timeoutMs) {
|
||||
val outputDeferred = async { process.inputStream.bufferedReader().use { it.readText() } }
|
||||
val exit = process.waitFor()
|
||||
StaticAnalysisRunResult(exit, outputDeferred.await())
|
||||
}
|
||||
}
|
||||
}.getOrElse {
|
||||
process.destroyForcibly()
|
||||
val reason = if (it is TimeoutCancellationException) {
|
||||
"static analysis '$command' timed out after ${timeoutMs}ms"
|
||||
"static analysis '$displayCommand' timed out after ${timeoutMs}ms"
|
||||
} else {
|
||||
it.message ?: "error running '$command'"
|
||||
it.message ?: "error running '$displayCommand'"
|
||||
}
|
||||
StaticAnalysisRunResult(EXIT_NOT_RUN, reason)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
private companion object {
|
||||
const val DEFAULT_TIMEOUT_MS = 300_000L
|
||||
const val EXIT_NOT_RUN = -1
|
||||
const val STATIC_FLOOR_COMMAND = "correx-static-floor"
|
||||
val WHITESPACE = Regex("\\s+")
|
||||
}
|
||||
}
|
||||
|
||||
+19
@@ -9,6 +9,8 @@ import com.correx.core.artifactstore.ArtifactStore
|
||||
import com.correx.core.context.builder.ContextPackBuilder
|
||||
import com.correx.core.context.model.ContextPack
|
||||
import com.correx.core.events.events.ClarificationAnswer
|
||||
import com.correx.core.events.events.ClarificationAnsweredEvent
|
||||
import com.correx.core.events.events.ClarificationRequestedEvent
|
||||
import com.correx.core.events.events.InferenceCompletedEvent
|
||||
import com.correx.core.events.events.InferenceFailedEvent
|
||||
import com.correx.core.events.events.InferenceStartedEvent
|
||||
@@ -212,6 +214,7 @@ abstract class SessionOrchestrator(
|
||||
internal val workspacePolicy: WorkspacePolicy? = engines.workspacePolicy
|
||||
internal val worldProbe: WorldProbe = engines.worldProbe
|
||||
internal val staticAnalysisRunner: StaticAnalysisRunner? = engines.staticAnalysisRunner
|
||||
internal val lspDiagnosticsRunner: LspDiagnosticsRunner? = engines.lspDiagnosticsRunner
|
||||
internal val contractAssertionEvaluator: ContractAssertionEvaluator? = engines.contractAssertionEvaluator
|
||||
internal val planCompilationCheck: PlanCompilationCheck? = engines.planCompilationCheck
|
||||
internal val semanticReviewer: SemanticReviewer? = engines.semanticReviewer
|
||||
@@ -259,6 +262,22 @@ abstract class SessionOrchestrator(
|
||||
fun liveClarificationRequestIds(): Set<String> =
|
||||
pendingClarifications.keys.mapTo(mutableSetOf()) { it.value }
|
||||
|
||||
/**
|
||||
* The clarification a headless caller should answer for this session: the newest still-live,
|
||||
* unanswered request. Null if the session is not parked on a clarification. Lets a REST answer
|
||||
* route resolve the (stageId, requestId) server-side so a script need only supply answer values —
|
||||
* the WS path knows them because it received the [ClarificationRequestedEvent]; a curl caller does not.
|
||||
*/
|
||||
suspend fun pendingClarificationFor(sessionId: SessionId): ClarificationRequestedEvent? {
|
||||
val live = liveClarificationRequestIds()
|
||||
if (live.isEmpty()) return null
|
||||
val events = eventStore.read(sessionId)
|
||||
val answered = events.mapNotNull { it.payload as? ClarificationAnsweredEvent }
|
||||
.mapTo(mutableSetOf()) { it.requestId }
|
||||
return events.mapNotNull { it.payload as? ClarificationRequestedEvent }
|
||||
.lastOrNull { it.requestId.value in live && it.requestId !in answered }
|
||||
}
|
||||
|
||||
/** Public seam for out-of-band gates (e.g. freestyle plan review) that reuse the stage-approval flow. */
|
||||
suspend fun requestPlanApproval(sessionId: SessionId, preview: String): Boolean =
|
||||
requestStageApproval(sessionId, StageId("plan_review"), preview)
|
||||
|
||||
+7
-4
@@ -76,7 +76,7 @@ internal suspend fun SessionOrchestrator.repairArtifact(
|
||||
if (eligible && bestCandidate != null && !isCancelled(sessionId)) {
|
||||
emitArtifactRepairAttempted(sessionId, stageId, slot, unresolved.classification, "LLM")
|
||||
val repaired = runArtifactRepairInference(
|
||||
sessionId, stageId, slot, bestCandidate, stageConfig, effectives, timeoutMs,
|
||||
sessionId, stageId, slot, bestCandidate, unresolved.detail, stageConfig, effectives, timeoutMs,
|
||||
)
|
||||
val reRun = repaired?.let { artifactExtractionPipeline.run(it, schema) }
|
||||
if (reRun is ArtifactExtractionPipeline.ExtractionResult.Resolved) {
|
||||
@@ -115,15 +115,18 @@ internal suspend fun SessionOrchestrator.runArtifactRepairInference(
|
||||
stageId: StageId,
|
||||
slot: TypedArtifactSlot,
|
||||
bestCandidate: String,
|
||||
validationError: String,
|
||||
stageConfig: StageConfig,
|
||||
effectives: RunEffectives,
|
||||
timeoutMs: Long,
|
||||
): String? {
|
||||
val schema = slot.kind.deriveJsonSchema()
|
||||
val schemaJson = Json.encodeToString(JsonSchema.serializer(), schema)
|
||||
val prompt = "The previous output for the '${slot.name.value}' artifact was malformed and did not " +
|
||||
"match the required schema.\n\nMalformed output:\n$bestCandidate\n\nReturn ONLY a single JSON " +
|
||||
"object matching this schema. Do not add fields, prose, or code fences.\nSchema: $schemaJson"
|
||||
val prompt = "The previous output for the '${slot.name.value}' artifact failed schema validation.\n\n" +
|
||||
"Validation error:\n$validationError\n\nMalformed output:\n$bestCandidate\n\nFix ONLY what the " +
|
||||
"validation error names — every field must sit at the level the schema defines; do not nest a " +
|
||||
"top-level field inside another object. Return ONLY a single JSON object matching this schema. " +
|
||||
"Do not add fields, prose, or code fences.\nSchema: $schemaJson"
|
||||
val entry = ContextEntry(
|
||||
id = ContextEntryId(UUID.randomUUID().toString()),
|
||||
layer = ContextLayer.L2,
|
||||
|
||||
+7
-3
@@ -190,9 +190,13 @@ internal suspend fun SessionOrchestrator.buildClarificationAnswerEntries(session
|
||||
/**
|
||||
* If [stageId] just produced an LLM artifact carrying a non-empty `questions` array, park the
|
||||
* run: record the questions (invariant #9 — observed at parse time), await the operator's
|
||||
* answers, record them, and return true so the caller re-runs the stage with the answers in
|
||||
* context. Bounded by [MAX_CLARIFICATION_ROUNDS] so a stage that keeps re-asking eventually
|
||||
* proceeds. Returns false when there are no questions or the round budget is spent.
|
||||
* answers, and record them. The caller then ADVANCES to the next stage — the stage is not
|
||||
* re-run. The answers are session-wide events, injected into every later stage's L0 context
|
||||
* (see [buildClarificationAnswerEntries]), so the next stage acts on them directly; re-running
|
||||
* the questioning stage only restarts inference with no prior reasoning and makes it re-explore.
|
||||
* Bounded by [OrchestrationTuning.maxClarificationRounds] so retries can't re-park unboundedly.
|
||||
* Returns true when it parked and collected answers, false when there are no questions or the
|
||||
* round budget is spent.
|
||||
*/
|
||||
|
||||
internal suspend fun SessionOrchestrator.requestClarificationIfNeeded(
|
||||
|
||||
+25
-5
@@ -229,6 +229,8 @@ internal suspend fun SessionOrchestrator.executeStage(
|
||||
val agentInstructionsEntries = emptyList<ContextEntry>()
|
||||
val retryFeedbackEntries = buildRetryFeedbackEntry(sessionEvents, stageId)
|
||||
?.let { listOf(it) } ?: emptyList()
|
||||
val groundingFeedbackEntries = buildGroundingFeedbackEntry(sessionEvents, stageId)
|
||||
?.let { listOf(it) } ?: emptyList()
|
||||
val recoveryTicketEntries = buildRecoveryTicketEntry(sessionEvents, stageId)
|
||||
?.let { listOf(it) } ?: emptyList()
|
||||
val vocabularyEntries = artifactKindRegistry
|
||||
@@ -266,7 +268,8 @@ internal suspend fun SessionOrchestrator.executeStage(
|
||||
systemPrompt + operatingGuidance + promotedConcepts + successfulPlanShapes + verifiedBaseline + intentEntries + profileEntries + projectProfileEntries + agentInstructionsEntries +
|
||||
journalEntries + repoMapEntries + claimedTaskEntries +
|
||||
needsEntries + schemaEntries + vocabularyEntries + promptEntries + steeringEntries +
|
||||
clarificationEntries + retryFeedbackEntries + recoveryTicketEntries + remainingDeltaEntries,
|
||||
clarificationEntries + retryFeedbackEntries + groundingFeedbackEntries + recoveryTicketEntries +
|
||||
remainingDeltaEntries,
|
||||
)
|
||||
val contextPack = runCatching {
|
||||
contextPackBuilder.build(
|
||||
@@ -374,12 +377,29 @@ internal suspend fun SessionOrchestrator.executeStage(
|
||||
continue
|
||||
}
|
||||
// emit_artifact is a meta-tool: the LLM calls it with the artifact's fields as arguments.
|
||||
// Capture those arguments as the artifact content and exit the loop straight to validation
|
||||
// (the post-loop capture honours this override instead of the empty assistant text).
|
||||
// Validate those arguments INLINE, like any other tool call: a valid artifact captures and
|
||||
// exits the loop; an invalid one is handed straight back into the SAME conversation as a
|
||||
// tool-result error, so the model fixes it with its full context (and CoT) intact. Breaking
|
||||
// the loop on a bad artifact would instead drop into a cold repair/retry that re-explores
|
||||
// from scratch and tends to re-make the identical schema mistake (2026-07-18).
|
||||
val emitCall = inferenceResult.response.toolCalls.firstOrNull { it.function.name == EMIT_ARTIFACT_TOOL }
|
||||
if (emitCall != null && llmEmittedSlots.isNotEmpty()) {
|
||||
llmArtifactOverride = emitCall.function.arguments
|
||||
break
|
||||
val emitSlot = llmEmittedSlots.first()
|
||||
when (val res = artifactExtractionPipeline.run(emitCall.function.arguments, emitSlot.kind.deriveJsonSchema())) {
|
||||
is ArtifactExtractionPipeline.ExtractionResult.Resolved -> {
|
||||
llmArtifactOverride = res.canonicalJson.toString()
|
||||
break
|
||||
}
|
||||
is ArtifactExtractionPipeline.ExtractionResult.Unresolved -> {
|
||||
inferenceResult = pushBack(
|
||||
"ERROR: your emit_artifact call for '${emitSlot.name.value}' did not match the " +
|
||||
"schema: ${res.detail}. Call emit_artifact again with only that corrected. Every " +
|
||||
"field must sit at the level the schema defines — do not nest a top-level field " +
|
||||
"(e.g. ready, questions) inside another object.",
|
||||
)
|
||||
continue
|
||||
}
|
||||
}
|
||||
}
|
||||
// stage_complete is a meta-tool: the LLM calls it to signal the stage's goal is met.
|
||||
// Skip tool dispatch — the loop exits and normal validation/transition follows
|
||||
|
||||
+1
@@ -199,6 +199,7 @@ internal suspend fun SessionOrchestrator.runPostStageGates(
|
||||
{ runContractGate(sessionId, stageId, stageConfig, effectives) },
|
||||
{ runPlanCompileGate(sessionId, stageId, stageConfig) },
|
||||
{ runStaticAnalysis(sessionId, stageId, stageConfig, effectives) },
|
||||
{ runLspDiagnostics(sessionId, stageId, stageConfig, effectives) },
|
||||
{ runExecutionGate(sessionId, stageId, stageConfig, effectives, profileCommands) },
|
||||
{ runReviewGate(sessionId, stageId, stageConfig, effectives) },
|
||||
)
|
||||
|
||||
+56
@@ -5,6 +5,7 @@ import com.correx.core.events.events.ReviewVerdict
|
||||
import com.correx.core.events.events.ArtifactCreatedEvent
|
||||
import com.correx.core.events.events.StaticAnalysisCompletedEvent
|
||||
import com.correx.core.events.events.StaticAnalysisFinding
|
||||
import com.correx.core.events.events.LspDiagnosticsCompletedEvent
|
||||
import com.correx.core.events.events.ArtifactValidatedEvent
|
||||
import com.correx.core.events.events.ArtifactValidatingEvent
|
||||
import com.correx.core.events.types.SessionId
|
||||
@@ -66,6 +67,55 @@ internal suspend fun SessionOrchestrator.runStaticAnalysis(
|
||||
)
|
||||
}
|
||||
|
||||
@Suppress("ReturnCount")
|
||||
internal suspend fun SessionOrchestrator.runLspDiagnostics(
|
||||
sessionId: SessionId,
|
||||
stageId: StageId,
|
||||
stageConfig: StageConfig,
|
||||
effectives: RunEffectives,
|
||||
): StageExecutionResult {
|
||||
val runner = lspDiagnosticsRunner ?: return StageExecutionResult.Success(emptyList())
|
||||
val workspaceRoot = effectives.policy?.workspaceRoot ?: return StageExecutionResult.Success(emptyList())
|
||||
val stagePaths = stageWrittenPaths(sessionId, stageId)
|
||||
val paths = if (stagePaths.isNotEmpty()) {
|
||||
stagePaths
|
||||
} else if (stageConfig.autoBuildGate) {
|
||||
sessionWrittenPaths(sessionId)
|
||||
} else {
|
||||
emptyList()
|
||||
}
|
||||
if (paths.isEmpty()) return StageExecutionResult.Success(emptyList())
|
||||
|
||||
val result = runner.pull(LspDiagnosticsRequest(workspaceRoot, paths))
|
||||
val plannedButUnwritten = stageConfig.writeManifest.filterNot { it in paths }
|
||||
val diagnostics = result.diagnostics.filter { diagnostic ->
|
||||
diagnostic.path in paths && plannedButUnwritten.none { planned ->
|
||||
diagnostic.message.contains(planned.substringAfterLast('/'), ignoreCase = true)
|
||||
}
|
||||
}
|
||||
emit(
|
||||
sessionId,
|
||||
LspDiagnosticsCompletedEvent(sessionId, stageId, result.server, diagnostics, result.skippedReason),
|
||||
)
|
||||
val errors = diagnostics.filter { it.severity.equals("error", ignoreCase = true) }
|
||||
if (errors.isEmpty()) return StageExecutionResult.Success(emptyList())
|
||||
val detail = errors.joinToString("\n") {
|
||||
"- ${it.path}:${it.line + 1}:${it.character + 1} ${it.code.orEmpty()} ${it.message}".trim()
|
||||
}
|
||||
return StageExecutionResult.Failure(
|
||||
"stage ${stageId.value} has LSP diagnostics in files it wrote. Fix these before proceeding:\n$detail",
|
||||
retryable = true,
|
||||
gate = "lsp_diagnostics",
|
||||
)
|
||||
}
|
||||
|
||||
internal fun SessionOrchestrator.sessionWrittenPaths(sessionId: SessionId): List<String> =
|
||||
eventStore.read(sessionId)
|
||||
.mapNotNull { it.payload as? com.correx.core.events.events.FileWrittenEvent }
|
||||
.filter { it.postImageHash != null }
|
||||
.map { it.path }
|
||||
.distinct()
|
||||
|
||||
/**
|
||||
* Gate 4 — execution (staged-verification §1/§2). A stage's [StageConfig.buildExpectation]
|
||||
* (none/module/project/tests) resolves to the bound project profile's typecheck/build/test
|
||||
@@ -76,6 +126,7 @@ internal suspend fun SessionOrchestrator.runStaticAnalysis(
|
||||
* [StaticAnalysisCompletedEvent], invariant #9, so replay never re-runs it).
|
||||
*/
|
||||
|
||||
@Suppress("ReturnCount")
|
||||
internal suspend fun SessionOrchestrator.runExecutionGate(
|
||||
sessionId: SessionId,
|
||||
stageId: StageId,
|
||||
@@ -93,6 +144,11 @@ internal suspend fun SessionOrchestrator.runExecutionGate(
|
||||
stageConfig.autoBuildGate && sessionProducedBuildTarget(sessionId) -> BuildExpectation.PROJECT
|
||||
else -> null
|
||||
}
|
||||
// LSP pull diagnostics are the compiler-front-end/typecheck gate. Keep PROJECT bundlers and
|
||||
// TESTS as real commands, but do not invoke the profile's flat `typecheck` alias as well.
|
||||
if (expectation == BuildExpectation.MODULE && lspDiagnosticsRunner != null) {
|
||||
return StageExecutionResult.Success(emptyList())
|
||||
}
|
||||
val alias = expectation?.commandAlias ?: return StageExecutionResult.Success(emptyList())
|
||||
val runner = staticAnalysisRunner
|
||||
val workspaceRoot = effectives.policy?.workspaceRoot
|
||||
|
||||
+7
-2
@@ -69,13 +69,18 @@ internal suspend fun SessionOrchestrator.advanceStage(
|
||||
|
||||
// --- terminal state helpers ---
|
||||
|
||||
internal suspend fun SessionOrchestrator.emitWorkflowStarted(sessionId: SessionId, graph: WorkflowGraph, config: OrchestrationConfig) {
|
||||
internal suspend fun SessionOrchestrator.emitWorkflowStarted(
|
||||
sessionId: SessionId,
|
||||
graph: WorkflowGraph,
|
||||
config: OrchestrationConfig,
|
||||
startStage: StageId = graph.start,
|
||||
) {
|
||||
emit(
|
||||
sessionId,
|
||||
WorkflowStartedEvent(
|
||||
sessionId = sessionId,
|
||||
workflowId = graph.id,
|
||||
startStageId = graph.start,
|
||||
startStageId = startStage,
|
||||
retryPolicy = config.retryPolicy,
|
||||
),
|
||||
)
|
||||
|
||||
+12
-11
@@ -48,17 +48,18 @@ class TierContextSummarizer(
|
||||
}
|
||||
val summaryText = summarize(prompt)
|
||||
val summaryArtifactId = artifactStore.put(summaryText.toByteArray(Charsets.UTF_8))
|
||||
artifactStore.flushBefore {
|
||||
emit(
|
||||
ContextSummarizedEvent(
|
||||
sessionId = sessionId,
|
||||
coversThroughSequence = coversThroughSequence,
|
||||
summaryArtifactId = summaryArtifactId,
|
||||
tier = tier,
|
||||
entriesOmittedCount = freeformEntries.size,
|
||||
),
|
||||
)
|
||||
}
|
||||
// Emit directly, NOT inside flushBefore { }: append() already fsyncs artifacts before it
|
||||
// persists the event, so wrapping the emit here re-acquires the (non-reentrant) artifact
|
||||
// Mutex that flushBefore already holds → self-deadlock (same bug as JournalCompactionService).
|
||||
emit(
|
||||
ContextSummarizedEvent(
|
||||
sessionId = sessionId,
|
||||
coversThroughSequence = coversThroughSequence,
|
||||
summaryArtifactId = summaryArtifactId,
|
||||
tier = tier,
|
||||
entriesOmittedCount = freeformEntries.size,
|
||||
),
|
||||
)
|
||||
return true
|
||||
}
|
||||
}
|
||||
|
||||
+27
@@ -9,6 +9,9 @@ import com.correx.core.journal.model.DecisionKind
|
||||
import com.correx.core.journal.model.DecisionRecord
|
||||
import com.correx.core.utils.TypeId
|
||||
import kotlinx.coroutines.runBlocking
|
||||
import kotlinx.coroutines.sync.Mutex
|
||||
import kotlinx.coroutines.sync.withLock
|
||||
import kotlinx.coroutines.withTimeout
|
||||
import org.junit.jupiter.api.Assertions.assertEquals
|
||||
import org.junit.jupiter.api.Assertions.assertFalse
|
||||
import org.junit.jupiter.api.Assertions.assertTrue
|
||||
@@ -70,6 +73,30 @@ class JournalCompactionServiceTest {
|
||||
assertEquals(fixedArtifactId, event.summaryArtifactId)
|
||||
}
|
||||
|
||||
// Regression: append() flushes artifacts inside the SAME non-reentrant Mutex that flushBefore
|
||||
// holds. If compaction wraps its emit in flushBefore, the emit->append->flushBefore re-acquires
|
||||
// that Mutex and self-deadlocks. This store mirrors production locking so the deadlock is real;
|
||||
// the fake above cannot catch it because its flushBefore takes no lock.
|
||||
private fun reentrantHostileArtifactStore(): ArtifactStore = object : ArtifactStore {
|
||||
val mutex = Mutex()
|
||||
override suspend fun put(bytes: ByteArray) = fixedArtifactId
|
||||
override suspend fun get(id: TypeId): ByteArray? = null
|
||||
override suspend fun flushBefore(commit: suspend () -> Unit) = mutex.withLock { commit() }
|
||||
}
|
||||
|
||||
@Test
|
||||
fun `does not deadlock when emit re-enters the artifact store (as append does)`(): Unit = runBlocking {
|
||||
val store = reentrantHostileArtifactStore()
|
||||
val svc = JournalCompactionService(store, { "summary" }, tokenThreshold = { 100 })
|
||||
val state = stateWithRecords(makeRecord(1, DecisionKind.INTENT))
|
||||
// emit simulates SqliteEventStore.append: it flushes artifacts inside the store mutex.
|
||||
withTimeout(2000) {
|
||||
svc.compactIfNeeded(SessionId("s1"), state, renderedTokenEstimate = 500) {
|
||||
store.flushBefore { /* persist event row */ }
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
@Test
|
||||
fun `only HIGH-salience records are included in the summarize prompt`(): Unit = runBlocking {
|
||||
val capturedPrompts = mutableListOf<String>()
|
||||
|
||||
+17
@@ -51,4 +51,21 @@ class ProcessStaticAnalysisRunnerTest {
|
||||
val result = runner.run(dir, " ")
|
||||
assertTrue(result.exitCode != 0, "exit: ${result.exitCode}")
|
||||
}
|
||||
|
||||
@Test
|
||||
fun `static floor skips an unsupported filetype without failing`() = runBlocking {
|
||||
val dir = createTempDirectory("sa-skip")
|
||||
dir.resolve("Main.kt").writeText("fun main() = Unit")
|
||||
val result = runner.run(dir, "correx-static-floor -- Main.kt")
|
||||
assertEquals(0, result.exitCode)
|
||||
assertTrue(result.output.contains("skipped"), "output: ${result.output}")
|
||||
}
|
||||
|
||||
@Test
|
||||
fun `static floor checks javascript syntax without project dependencies`() = runBlocking {
|
||||
val dir = createTempDirectory("sa-js")
|
||||
dir.resolve("broken.js").writeText("const = ;")
|
||||
val result = runner.run(dir, "correx-static-floor -- broken.js")
|
||||
assertTrue(result.exitCode != 0, "exit: ${result.exitCode}")
|
||||
}
|
||||
}
|
||||
|
||||
+25
-5
@@ -19,8 +19,11 @@ import java.nio.file.Path
|
||||
* workspace but outside the declared set is scope creep — the dominant local-implementer
|
||||
* failure that tests do not catch — and is raised as PATH_OUTSIDE_MANIFEST → BLOCK.
|
||||
*
|
||||
* Empty manifest = unrestricted (workspace containment still applies via
|
||||
* [PathContainmentRule]). Out-of-workspace targets are that rule's concern, not this one.
|
||||
* A write already inside the active task's affected_paths is allowed even if the manifest is
|
||||
* narrower: those paths are the agent-widenable, recorded authority (see [WriteScopeRule]), and
|
||||
* the stage manifest is a planner hint that defers to them. Empty manifest = unrestricted
|
||||
* (workspace containment still applies via [PathContainmentRule]). Out-of-workspace targets are
|
||||
* that rule's concern, not this one.
|
||||
* Path resolution goes through WorldProbe (symlink-safe) and the facts are recorded as
|
||||
* observations, so replay reads them back rather than re-globbing (invariant #9).
|
||||
*/
|
||||
@@ -36,6 +39,14 @@ class ManifestContainmentRule : ToolCallRule {
|
||||
val matchers = input.writeManifest.map {
|
||||
FileSystems.getDefault().getPathMatcher("glob:$it")
|
||||
}
|
||||
// The active task's affected_paths are the agent-widenable, recorded authority (see
|
||||
// WriteScopeRule). A write already inside that scope is not manifest scope-creep — the
|
||||
// stage manifest is a narrower planner hint that must defer to it, otherwise a too-tight
|
||||
// manifest becomes an unrecoverable dead-end (the escape hatch is task_update affected_paths).
|
||||
val taskScope = input.session.activeTask?.scope.orEmpty()
|
||||
val taskMatchers = taskScope.map {
|
||||
FileSystems.getDefault().getPathMatcher("glob:${it.removePrefix("./")}")
|
||||
}
|
||||
|
||||
val issues = mutableListOf<ValidationIssue>()
|
||||
val observations = mutableListOf<ToolCallObservation>()
|
||||
@@ -48,7 +59,9 @@ class ManifestContainmentRule : ToolCallRule {
|
||||
val resolvedReal = input.probe.resolveReal(resolvedInput)
|
||||
val inWorkspace = resolvedReal.startsWith(workspaceReal)
|
||||
val relative = if (inWorkspace) workspaceReal.relativize(resolvedReal).toString() else raw
|
||||
val inManifest = inWorkspace && matchers.any { it.matches(Path.of(relative)) }
|
||||
val relPath = Path.of(relative)
|
||||
val inManifest = inWorkspace && matchers.any { it.matches(relPath) }
|
||||
val inTaskScope = inWorkspace && taskMatchers.any { it.matches(relPath) }
|
||||
|
||||
observations += ToolCallObservation(
|
||||
ruleCode = RULE_CODE,
|
||||
@@ -57,14 +70,21 @@ class ManifestContainmentRule : ToolCallRule {
|
||||
"relative" to relative,
|
||||
"inWorkspace" to inWorkspace.toString(),
|
||||
"inManifest" to inManifest.toString(),
|
||||
"inTaskScope" to inTaskScope.toString(),
|
||||
),
|
||||
)
|
||||
|
||||
if (inWorkspace && !inManifest) {
|
||||
if (inWorkspace && !inManifest && !inTaskScope) {
|
||||
val remedy = if (taskScope.isEmpty()) {
|
||||
" Claim a task whose affected_paths cover this path (task_update), then retry."
|
||||
} else {
|
||||
" If this path is genuinely part of the task, widen its affected_paths via " +
|
||||
"task_update (note why), then retry."
|
||||
}
|
||||
issues += ValidationIssue(
|
||||
code = "PATH_OUTSIDE_MANIFEST",
|
||||
message = "Tool '${input.request.toolName}' writes '$raw' outside the stage's " +
|
||||
"declared manifest ${input.writeManifest}",
|
||||
"declared manifest ${input.writeManifest}.$remedy",
|
||||
severity = ValidationSeverity.ERROR,
|
||||
)
|
||||
disposition = maxAction(disposition, RiskAction.BLOCK)
|
||||
|
||||
+42
-11
@@ -24,17 +24,20 @@ class ManifestContainmentRuleTest {
|
||||
override fun resolveReal(path: Path): Path = path.toAbsolutePath().normalize()
|
||||
}
|
||||
|
||||
private fun input(pathArg: String, manifest: List<String>) = ToolCallAssessmentInput(
|
||||
request = ToolRequest(
|
||||
ToolInvocationId("i"), SessionId("s"), StageId("st"), "file_write",
|
||||
mapOf("path" to pathArg, "mode" to "0644"),
|
||||
),
|
||||
capabilities = setOf(ToolCapability.FILE_WRITE),
|
||||
workspace = WorkspacePolicy(workspace),
|
||||
probe = FakeProbe(),
|
||||
paramRoles = mapOf("path" to ParamRole.PATH),
|
||||
writeManifest = manifest,
|
||||
)
|
||||
private fun input(pathArg: String, manifest: List<String>, taskScope: List<String> = emptyList()) =
|
||||
ToolCallAssessmentInput(
|
||||
request = ToolRequest(
|
||||
ToolInvocationId("i"), SessionId("s"), StageId("st"), "file_write",
|
||||
mapOf("path" to pathArg, "mode" to "0644"),
|
||||
),
|
||||
capabilities = setOf(ToolCapability.FILE_WRITE),
|
||||
workspace = WorkspacePolicy(workspace),
|
||||
probe = FakeProbe(),
|
||||
paramRoles = mapOf("path" to ParamRole.PATH),
|
||||
writeManifest = manifest,
|
||||
session = if (taskScope.isEmpty()) SessionContext()
|
||||
else SessionContext(activeTask = ActiveTask("t1", taskScope)),
|
||||
)
|
||||
|
||||
@Test
|
||||
fun `applies only to write capability`() {
|
||||
@@ -67,6 +70,34 @@ class ManifestContainmentRuleTest {
|
||||
assertEquals("false", r.observations.single().facts["inManifest"])
|
||||
}
|
||||
|
||||
@Test
|
||||
fun `write outside manifest but inside task scope is allowed`() {
|
||||
// The scaffold dead-end: manifest lists only two files, but the claimed task's
|
||||
// affected_paths cover the whole subtree. The recorded task scope wins.
|
||||
val r = rule.assess(
|
||||
input(
|
||||
"/work/project/frontend/src/main.tsx",
|
||||
manifest = listOf("frontend/package.json", "frontend/vite.config.ts"),
|
||||
taskScope = listOf("frontend/**"),
|
||||
),
|
||||
)
|
||||
assertEquals(RiskAction.PROCEED, r.disposition)
|
||||
assertTrue(r.issues.isEmpty())
|
||||
val facts = r.observations.single().facts
|
||||
assertEquals("false", facts["inManifest"])
|
||||
assertEquals("true", facts["inTaskScope"])
|
||||
}
|
||||
|
||||
@Test
|
||||
fun `write outside both manifest and task scope is blocked with widen remedy`() {
|
||||
val r = rule.assess(
|
||||
input("/work/project/apps/x.kt", manifest = listOf("core/**"), taskScope = listOf("frontend/**")),
|
||||
)
|
||||
assertEquals(RiskAction.BLOCK, r.disposition)
|
||||
assertEquals("PATH_OUTSIDE_MANIFEST", r.issues.single().code)
|
||||
assertTrue(r.issues.single().message.contains("task_update"))
|
||||
}
|
||||
|
||||
@Test
|
||||
fun `relative path arg is resolved against the workspace before matching`() {
|
||||
val r = rule.assess(input("core/a/B.kt", listOf("core/**")))
|
||||
|
||||
+1
-1
@@ -14,7 +14,7 @@ Maintained alongside the features they describe. Any agent shipping a significan
|
||||
- `decisions/` — ADRs (adr-NNNN-*.md). Append-only. Never delete or retroactively alter a decided ADR; write a superseding one instead.
|
||||
- `qa/` — QA run plans (QA-*.md). Each maps to a shipped feature or an explicitly pending controlled audition. `TEMPLATE.md` is the canonical shape. `ENV.md` describes the required live environment. `README.md` explains the QA process.
|
||||
- `specs/` — feature specs by date-slug. Inputs for planned or in-flight work.
|
||||
- `schemas/` — canonical JSON schemas for structured outputs (analysis, brief_echo, design, execution_plan, impl_plan). Shared across the router and validator.
|
||||
- `schemas/` — canonical JSON schemas for structured outputs (analysis, discovery brief, DoD, brief_echo, design, execution_plan, impl_plan). Shared across the router and validator.
|
||||
- `epics/`, `modules/`, `diagrams/`, `design/`, `reviews/`, `visual/` — supporting reference material.
|
||||
|
||||
**⚠️ STALE / DO NOT TRUST FOR STATUS:**
|
||||
|
||||
@@ -0,0 +1,150 @@
|
||||
# Correx — Catch-up for a ~2-month-old memory
|
||||
|
||||
**If your last memory of Correx is the epic-13/epic-14 era (mid-May 2026), read this.**
|
||||
|
||||
Back then the project was a *skeleton*: epics 0–14 had landed the module structure
|
||||
(events → sessions → transitions → validation → approvals → artifacts → context →
|
||||
tools → inference → kernel → infra → interfaces → router). It compiled and replayed,
|
||||
but the orchestration loop was thin and mostly unproven against real model runs.
|
||||
|
||||
Two months and ~250 feature commits later, the architecture is the *same* (all 9 Hard
|
||||
Invariants still hold; the event log is still the only source of truth), but nearly the
|
||||
entire *behavioral* layer — how a session actually drives a model through real work —
|
||||
was built and hardened by live QA. This document groups the major additions. It is a
|
||||
map, not a changelog; the dated files in `docs/plans/` are the real changelog.
|
||||
|
||||
Work is no longer tracked as "epics." It's tracked as **plans** (`docs/plans/`) and a
|
||||
**Vikunja backlog** (project_id 4).
|
||||
|
||||
---
|
||||
|
||||
## The single biggest correction: the TUI is now Go
|
||||
|
||||
The Kotlin TUI (Mosaic, then tamboui) was **retired**. The interactive terminal UI is
|
||||
now a **Go / Bubble Tea** app at `apps/tui-go` — a WS client of the server. `apps/cli`
|
||||
(Clikt, Kotlin) and `apps/server` (Ktor) are the only Gradle app modules; the old
|
||||
`apps/desktop` / `apps/worker` stubs were deleted. If you remember tamboui, forget it.
|
||||
|
||||
## New whole subsystems (did not meaningfully exist at epic-14)
|
||||
|
||||
**Native task tracking (`core:tasks`).** A dependency-aware work graph: `task_decompose`
|
||||
splits a goal into a DEPENDS_ON graph in one approval; tasks have claim gates
|
||||
(dependency-hard-gate, cite-before-claim, duplicate-title guard), per-task history,
|
||||
markdown export, a full REST surface, a `correx task` CLI, a TUI task board with
|
||||
readiness, and **git-driven status** (a commit mention → IN_PROGRESS, a closing keyword
|
||||
→ DONE, idempotent). The execution loop is now *tasked* — it replaced the old linear
|
||||
role_pipeline.
|
||||
|
||||
**Plane-2 tool-call intent validation (`core:toolintent`).** An anti-hallucination /
|
||||
safety layer that assesses every tool call *before* it runs and records
|
||||
`ToolCallAssessedEvent`: path-containment, read-before-write, reference-must-exist,
|
||||
write-scope adherence, stale-write, exec-interpreter and network-host rules, and a
|
||||
per-session egress allowlist. This is the layer that stops an agent citing files it
|
||||
never read or writing outside its declared scope.
|
||||
|
||||
**Compression pipeline (`infrastructure:compression`).** A staged, replay-safe context
|
||||
compressor: level policy → fact sheet + strict allocation → tier summarizer → token
|
||||
pruning + embedding relevance + ToMe merge → static-doc pruning (CLAUDE.md/AGENTS.md).
|
||||
Distinct from the required/optional *bucket* rework in `core:context`.
|
||||
|
||||
**Staged verification & review gates.** Stage output passes a gate ladder instead of
|
||||
being trusted: build gate, plan-compile gate, contract/execution gate, and a
|
||||
semantic-review gate (`core:critique`, with structured `CritiqueFinding` + per-model
|
||||
calibration). Per-gate retry budgets with progress-aware charging and hybrid salvage;
|
||||
static-first reviewer framing; capability-gap reflection; brief echo-back gate.
|
||||
|
||||
**Research workflow.** `WebSearchTool` (local SearXNG), `WebFetchTool` (bounded fetch),
|
||||
deterministic HTML→markdown extraction, a research workflow graph, batch source-fetch
|
||||
approval, and dedicated `SourceFetched` / `LowQualityExtraction` events. On by default.
|
||||
|
||||
**Cross-session repo memory (L3 + repo-map + project profile).** `RepoMapIndexer`
|
||||
(symbols for Kotlin, GDScript, C#/Godot-Mono, markdown headings), `RepoKnowledgeRetriever`
|
||||
with recency fallback, L3 ANN retrieval (`in_memory` or `turbovec` sidecar) recorded as
|
||||
environment observations, `ProjectMemoryService`/`Distiller`, and `ProjectProfile`
|
||||
(per-repo standing context at `.correx/project.toml`) injected as L0. Note: this is
|
||||
*repo-scoped* memory — the "cross-session memory" anti-feature (conversational recall)
|
||||
is still deliberately out.
|
||||
|
||||
**Git run-branch transport (the "Gitea transport") — SHIPPED.** The server owns a Git
|
||||
checkout; each run pushes to a run branch (`GitRunBranchTransport`, `RunBranchPushedEvent`);
|
||||
`GitTaskSync`/`GitCommandCommitReader` drive task status from commits. "gitea" is just
|
||||
the configured remote name; the transport is plain Git.
|
||||
|
||||
**Observability & replay tooling (Epic 15, largely shipped).** Event-stream inspector
|
||||
(`correx events`, `/sessions/{id}/events`), session replay + determinism digest,
|
||||
causation-graph DOT export, metrics as a replayable projection (`correx stats`),
|
||||
event-sourced health checks (EventStore latency, llama-server liveness, disk watermark,
|
||||
ROCm/VRAM probe), and correlation-structured MDC logging.
|
||||
|
||||
**Model lifecycle management (`[[models]]`).** Correx spawns/owns the llama-server
|
||||
process: per-stage model selection, manual swap + pin, resource telemetry, VRAM gauge.
|
||||
(The autonomous *residency scheduler* is still unbuilt — see below.)
|
||||
|
||||
**Workspace scoping & undo.** `WorkspaceResolver` trust pipeline + `allowed_workspace_roots`,
|
||||
per-session workspace-scoped tools/policy, client→server workspace handshake, crash-safe
|
||||
atomic file writes with pre/post-image captured to CAS (`FileWrittenEvent`), and a
|
||||
`FileMutationReverser` undo primitive (server endpoint + CLI). Agents can also READ
|
||||
outside `workspace_root` via an operator approval prompt (writes stay jailed).
|
||||
|
||||
**Approvals & grants.** Cross-session grant scopes (PROJECT/GLOBAL) + revoke,
|
||||
grant-aware approval engine, risk rationale surfaced to the operator; approval gates now
|
||||
block indefinitely instead of auto-rejecting on timeout, and steering actually affects
|
||||
the run.
|
||||
|
||||
**Router interaction layer.** Beyond CHAT+STEERING: grounded system prompts, triage
|
||||
directive (no bare refusals), structured **workflow proposals** from triage, a
|
||||
**clarification-question loop** (analyst asks, operator answers, producer-exit rehydrate
|
||||
on reconnect), **rubber-duck idea board** (capture/feed/promote to project profile), and
|
||||
grounded **narration** (pinned narration model, latency/token metrics surfaced).
|
||||
|
||||
**Freestyle / execution-plan engine.** Two-phase planning→execution driver,
|
||||
`ExecutionPlanCompiler` + `ExecutionPlanLockedEvent`, post-plan operator approval gate,
|
||||
freestyle graph re-routing (deterministic override), soft-lock steering preemption,
|
||||
plan grounding + positive-pattern mining + a stage-prompt audition rig.
|
||||
|
||||
**Decision journal (`core:journal`).** Decision-journal projection injected into stage
|
||||
context, salience-aware compaction (`JournalCompactionService`, CAS-rendered).
|
||||
|
||||
**Personalization (`core:config`).** `OperatorProfile` + `ProfileLoader` bound at start
|
||||
and surfaced into L0; propose-only `ProfileAdaptationService`.
|
||||
|
||||
**Native MCP host.** The server mounts external stdio MCP servers as first-class Correx
|
||||
tools (inherit tier / receipt / replay). `codebase-memory-mcp` is installed and granted
|
||||
per code-intel stage.
|
||||
|
||||
**Recovery loop.** Failed write-less stages route to recovery instead of futile retry;
|
||||
tier-2 intent-holder arbiter; remaining-delta pinning; failure-ticket routing +
|
||||
review-gate RECOVER + return-to-sender; structured package-404 recovery evidence;
|
||||
stage-global repeated-failure loop breaker.
|
||||
|
||||
**Godot / game-dev support.** GDScript + C#/Godot-Mono symbol extraction in the repo
|
||||
map, and a structural `.tscn`/`.tres` scene validator.
|
||||
|
||||
## Refactors & hardening you'll notice
|
||||
|
||||
- `DefaultSessionOrchestrator` god-class decomposed into per-concern extensions;
|
||||
`DomainEventMapper`'s god-`when` split per-domain. Detekt baseline ratcheted 120 → 90.
|
||||
- Persistence hardened: sequences assigned atomically inside INSERT, reads serialized
|
||||
with the append transaction, slow WS clients no longer stall the kernel.
|
||||
- Tool results bounded + framed, full output spilled to CAS with a `tool_output`
|
||||
retrieval tool; `glob`/`grep` search tools; web_fetch SSRF fix; interruptible shell
|
||||
timeouts with process-tree kill; shell allowlist validates every command position.
|
||||
- Config: orchestration loop/threshold/budget constants moved to `[orchestration]`;
|
||||
operator-tunable sampling knobs (top_k/min_p/repeat_penalty) per stage.
|
||||
|
||||
## Still planned / deliberately NOT built (don't assume these exist)
|
||||
|
||||
- Full **remote always-on / thin-client** deployment topology. Its Git *transport*
|
||||
shipped (above); the hosting topology itself is not yet stood up.
|
||||
- Full router **context isolation** (persistent conversation history, session-scoped
|
||||
boundaries). CHAT+STEERING works; the isolation layer is deferred.
|
||||
- Autonomous GPU **residency scheduling** (lifecycle *is* built; idle-eviction policy
|
||||
is not).
|
||||
- Still explicitly out: parallel agent execution, conversational cross-session memory,
|
||||
three-critic ensemble, fine-tuning, streaming inference.
|
||||
|
||||
## Where to look now
|
||||
|
||||
- `CLAUDE.md` — current session guide, Module Map, invariants.
|
||||
- `docs/plans/` — the real changelog of intent since epic-14 (dated files).
|
||||
- Vikunja project 4 — live backlog. `scripts/ctx.py <query>` — ranked file/symbol lookup.
|
||||
@@ -135,6 +135,10 @@ backend = "in_memory" # or "turbovec"
|
||||
# schema_path = "schemas/review_report.json"
|
||||
# llm_emitted = true
|
||||
#
|
||||
# Review loops escalate to the synthesized recovery stage after this many review→rework cycles.
|
||||
# [orchestration]
|
||||
# review_loop_max_cycles = 3
|
||||
#
|
||||
# The bundled examples/workflows/review_loop.toml uses this kind: the reviewer emits a
|
||||
# review_report whose `verdict` field gates the implementer↔reviewer loop via
|
||||
# artifact_field_equals transitions. See docs/schemas/review_report.json for the schema.
|
||||
@@ -143,3 +147,8 @@ backend = "in_memory" # or "turbovec"
|
||||
id = "execution_plan"
|
||||
schema_path = "schemas/execution_plan.json"
|
||||
llm_emitted = true
|
||||
|
||||
[[artifacts]]
|
||||
id = "dod"
|
||||
schema_path = "schemas/dod.json"
|
||||
llm_emitted = true
|
||||
|
||||
@@ -1,6 +1,20 @@
|
||||
{
|
||||
"type": "object",
|
||||
"properties": {
|
||||
"brief": {
|
||||
"type": "object",
|
||||
"properties": {
|
||||
"what": { "type": "string" },
|
||||
"why": { "type": "string" },
|
||||
"who": { "type": "array", "items": { "type": "string" } },
|
||||
"scope": { "type": "array", "items": { "type": "string" } },
|
||||
"non_goals": { "type": "array", "items": { "type": "string" } },
|
||||
"constraints": { "type": "array", "items": { "type": "string" } },
|
||||
"assumptions": { "type": "array", "items": { "type": "string" } }
|
||||
},
|
||||
"required": ["what", "why", "who", "scope", "non_goals", "constraints", "assumptions"],
|
||||
"additionalProperties": false
|
||||
},
|
||||
"ready": {
|
||||
"type": "boolean",
|
||||
"description": "true when the request is clear and grounded enough to plan; false when you are raising questions"
|
||||
@@ -33,6 +47,6 @@
|
||||
}
|
||||
}
|
||||
},
|
||||
"required": ["ready", "questions"],
|
||||
"required": ["brief", "ready", "questions"],
|
||||
"additionalProperties": false
|
||||
}
|
||||
|
||||
@@ -0,0 +1,24 @@
|
||||
{
|
||||
"type": "object",
|
||||
"properties": {
|
||||
"summary": { "type": "string" },
|
||||
"criteria": {
|
||||
"type": "array",
|
||||
"minItems": 1,
|
||||
"items": {
|
||||
"type": "object",
|
||||
"properties": {
|
||||
"id": { "type": "string" },
|
||||
"statement": { "type": "string" },
|
||||
"part": { "type": "string" },
|
||||
"verified_by": { "type": "string", "description": "one of: gate, reviewer" }
|
||||
},
|
||||
"required": ["id", "statement", "part", "verified_by"],
|
||||
"additionalProperties": false
|
||||
}
|
||||
},
|
||||
"out_of_scope": { "type": "array", "items": { "type": "string" } }
|
||||
},
|
||||
"required": ["summary", "criteria", "out_of_scope"],
|
||||
"additionalProperties": false
|
||||
}
|
||||
@@ -14,6 +14,10 @@
|
||||
"produces": { "type": "string" },
|
||||
"kind": { "type": "string" },
|
||||
"tools": { "type": "array", "items": { "type": "string" } }
|
||||
,"writes": { "type": "array", "items": { "type": "string" } }
|
||||
,"touches": { "type": "array", "items": { "type": "string" } }
|
||||
,"build_expectation": { "type": "string", "description": "one of: none, module, project, tests" }
|
||||
,"semantic_review": { "type": "boolean" }
|
||||
},
|
||||
"required": ["id", "prompt", "produces"]
|
||||
}
|
||||
|
||||
@@ -22,6 +22,7 @@ Each TOML file in `workflows/` is a valid workflow loadable by the server. Keep
|
||||
- Add a new workflow example when a major new workflow type ships.
|
||||
- Prompts referenced by TOML files go in `workflows/prompts/`.
|
||||
- Freestyle architect prompts specify stage constraints, boundaries, and verification goals; they do not prescribe an exact resulting file list when the authoritative intent leaves implementation details open.
|
||||
- Freestyle discovery emits a structured comprehension brief; the analyst emits the addressable `dod` artifact used as the fixed implementation/review rubric.
|
||||
- Do not add configs/plugins/stages stubs speculatively — populate when there is real content.
|
||||
|
||||
## Verification
|
||||
|
||||
@@ -19,6 +19,11 @@ id = "analysis"
|
||||
schema_path = "schemas/analysis.json"
|
||||
llm_emitted = true
|
||||
|
||||
[[artifacts]]
|
||||
id = "dod"
|
||||
schema_path = "schemas/dod.json"
|
||||
llm_emitted = true
|
||||
|
||||
[[artifacts]]
|
||||
id = "design"
|
||||
schema_path = "schemas/design.json"
|
||||
|
||||
@@ -12,7 +12,7 @@ allowed_tools = ["file_read", "list_dir", "shell"]
|
||||
token_budget = 16384
|
||||
max_retries = 2
|
||||
|
||||
# analyst writes no files, but it owns task framing: task_search/task_context (read-only) find
|
||||
# analyst writes no files, but it owns task framing and the fixed definition of done:
|
||||
# existing work; task_create (T2, approval-gated — a task is an event-log entry, not a file write)
|
||||
# opens a single task; task_decompose (T2, one approval for the whole graph) splits a goal with
|
||||
# dependency seams into parent + DEPENDS_ON-linked children. Either way the analysis names the task
|
||||
@@ -20,7 +20,8 @@ max_retries = 2
|
||||
[[stages]]
|
||||
id = "analyst"
|
||||
prompt = "prompts/analyst_freestyle.md"
|
||||
produces = [{ name = "analysis", kind = "analysis" }]
|
||||
needs = ["discovery"]
|
||||
produces = [{ name = "dod", kind = "dod" }]
|
||||
allowed_tools = ["file_read", "list_dir", "shell", "task_search", "task_context", "task_create", "task_decompose"]
|
||||
token_budget = 16384
|
||||
max_retries = 2
|
||||
@@ -30,7 +31,7 @@ id = "architect"
|
||||
requires_approval = true
|
||||
inject_artifact_kinds = true
|
||||
prompt = "prompts/architect_freestyle.md"
|
||||
needs = ["analysis"]
|
||||
needs = ["dod"]
|
||||
produces = [{ name = "execution_plan", kind = "execution_plan" }]
|
||||
token_budget = 16384
|
||||
max_retries = 2
|
||||
@@ -47,7 +48,7 @@ id = "analyst-to-architect"
|
||||
from = "analyst"
|
||||
to = "architect"
|
||||
condition_type = "artifact_validated"
|
||||
condition_artifact_id = "analysis"
|
||||
condition_artifact_id = "dod"
|
||||
|
||||
[[transitions]]
|
||||
id = "architect-to-done"
|
||||
|
||||
@@ -1,33 +1,35 @@
|
||||
You are the **Analyst** in freestyle mode. Understand the user's goal (in the decision history
|
||||
above) and the code it touches. Read-only: `file_read` (also lists a directory's entries when
|
||||
given a directory path), `ls`, `grep`, `cat`, `find`.
|
||||
You are the **Analyst** in freestyle mode. Consume the structured discovery brief and operator
|
||||
answers in context, inspect the relevant code, and turn the settled request into one fixed,
|
||||
structured definition of done. Read-only tools: `file_read`, `list_dir`, `shell`, `task_search`,
|
||||
and `task_context`.
|
||||
|
||||
Before deriving requirements, check for existing work: `task_search` for related, duplicate, or
|
||||
blocking tasks and `task_context` to load any the goal names. Fold what you find into the
|
||||
analysis rather than re-deriving it; flag a duplicate instead of restating it.
|
||||
Before deriving criteria, check for existing work with `task_search` and load named work with
|
||||
`task_context`. Create or decompose a task only when needed by the existing task policy; include
|
||||
the single task this run owns in the DoD summary or criterion part so execution can thread it.
|
||||
|
||||
Then frame the work as a task (per the task policy):
|
||||
- If a task already covers this work, name its id (e.g. `auth-142`) in the analysis.
|
||||
- If the goal is a single coherent unit one run can carry to review, `task_create` one and name its
|
||||
id.
|
||||
- If the goal has **dependency seams** (a thing that must land before another) or **independent
|
||||
review/handoff points** (a piece worth shipping or reviewing on its own), `task_decompose` it into
|
||||
a parent epic + `DEPENDS_ON`-linked children — one approval for the whole graph. A session works
|
||||
one task at a time, so the children are claimed by *later* runs as they unblock; don't over-split.
|
||||
- After decomposing, **name in the analysis the single task this run will work** — the one already
|
||||
ready (no unmet dependency, e.g. the scaffold). Leave the blocked siblings for future runs.
|
||||
Emit the `dod` artifact once. Its criteria are the complete acceptance contract for this run:
|
||||
|
||||
Either way later stages thread the named task through the plan; the rest wait to be claimed.
|
||||
- Give every criterion a stable id (`c1`, `c2`, …), a checkable statement, and its feature area.
|
||||
- Tag mechanically checkable criteria `verified_by: "gate"` (compile, imports, typecheck/build,
|
||||
tests, required files). The reviewer must not adjudicate these.
|
||||
- Tag semantic or UX criteria `verified_by: "reviewer"`.
|
||||
- Copy discovery `brief.non_goals` into `out_of_scope`; this is a hard review boundary.
|
||||
- Cover the entire in-scope brief now. Later stages may not silently add criteria.
|
||||
|
||||
Produce the `analysis` artifact by calling the **`emit_artifact`** tool with these fields:
|
||||
- `summary`: the goal in your own words.
|
||||
- `requirements`: concrete, checkable requirements, one per line.
|
||||
- `affected_areas`: files/modules likely involved, one per line.
|
||||
Call `emit_artifact` with a JSON object matching this shape:
|
||||
`{"summary": string, "criteria": [{"id": string, "statement": string, "part": string,
|
||||
"verified_by": "gate" | "reviewer"}], "out_of_scope": [string]}`.
|
||||
|
||||
Call `emit_artifact` once you have read enough — do not write the JSON as a plain message.
|
||||
Example:
|
||||
```json
|
||||
{
|
||||
"summary": "Deliver the bounded validation gate for task gate-42.",
|
||||
"criteria": [
|
||||
{"id":"c1","statement":"The project typecheck passes before completion","part":"terminal gate","verified_by":"gate"},
|
||||
{"id":"c2","statement":"The operator sees the recorded diagnostic","part":"workflow UX","verified_by":"reviewer"}
|
||||
],
|
||||
"out_of_scope": ["Changing the workflow topology"]
|
||||
}
|
||||
```
|
||||
|
||||
Always produce the analysis — this is your single exit. Open questions and operator forks are the
|
||||
**Discovery** stage's job, and it has already run before you: any ambiguity or contradiction the
|
||||
user needed to resolve was raised and answered upstream, and those answers are in the decision
|
||||
history above. Treat the request as settled, ground your requirements in what you actually read,
|
||||
and do not ask the user anything. Do not design or plan yet.
|
||||
Do not ask questions; discovery owns clarification. Do not design or implement.
|
||||
|
||||
@@ -7,8 +7,8 @@ message; call `emit_artifact` and nothing else.
|
||||
|
||||
## Inputs available to you
|
||||
|
||||
- `analysis` artifact — structured findings from the analyst stage (goal, constraints,
|
||||
risks, open questions).
|
||||
- `dod` artifact — the fixed acceptance contract from the analyst. Every implementation and
|
||||
review stage must consume it and must not widen it.
|
||||
- Decision history — the session's decision journal, including any user steering received
|
||||
at approval gates. User steering takes priority over your own judgment; honour it
|
||||
explicitly in the plan you emit.
|
||||
@@ -48,7 +48,7 @@ Emit a JSON object that validates against the `execution_plan` schema:
|
||||
|
||||
## Rules
|
||||
|
||||
**goal** — one sentence, derived from the analysis artifact and any user steering.
|
||||
**goal** — one sentence, derived from the DoD artifact and any user steering.
|
||||
|
||||
**stages** — ordered list; each stage must:
|
||||
- Have a unique `id` in `snake_case`.
|
||||
@@ -60,6 +60,8 @@ Emit a JSON object that validates against the `execution_plan` schema:
|
||||
llm-emitted kind.
|
||||
- Declare `needs`: every upstream artifact id the stage's prompt references. Every id in
|
||||
`needs` must be `produces`d by a strictly earlier stage.
|
||||
- Every implementation and review stage must include the session-scoped `dod` artifact in
|
||||
`needs`. The compiler also enforces this seam.
|
||||
- Include `tools` per stage as it needs them, using only names from this set:
|
||||
`file_read`, `file_write`, `file_edit`, `list_dir`, `shell`, `task_context`, `task_update`,
|
||||
`task_search`. Stages that write or edit files take the file set
|
||||
@@ -101,7 +103,12 @@ Emit a JSON object that validates against the `execution_plan` schema:
|
||||
always pass the template/preset arg the generator requires (`-- --template react-ts`) since
|
||||
there's no prompt to fall back on. `file_write` is the fallback when no generator fits the
|
||||
stack. Only bare remote runners (`npx`, `bunx`, `pnpx`) stay blocked — they execute arbitrary
|
||||
remote code; reach them via `npm create` instead.
|
||||
remote code; reach them via `npm create` instead. A scaffold/generator stage emits an
|
||||
open-ended file set (a whole `frontend/src/**` tree, not two named files), so declare its
|
||||
`writes` as the covering directory glob — `["frontend/**"]`, **not** an enumerated
|
||||
`["frontend/package.json", "frontend/vite.config.ts"]`. A too-narrow `writes` becomes the
|
||||
stage's write manifest and blocks every file the generator legitimately produces. Use `**`
|
||||
(recursive), not `*` (one level) — `frontend/*` does not cover `frontend/src/main.tsx`.
|
||||
- **Every stage prompt must describe its exact JSON output structure when using a
|
||||
structured kind.** The model that runs the stage never sees the raw JSON schema — it only sees
|
||||
the kind's name (e.g. `analysis`, `research_report`, `review_report`) and the prompt you write
|
||||
@@ -141,12 +148,15 @@ Emit a JSON object that validates against the `execution_plan` schema:
|
||||
`field`: `"verdict"`, `value`: `"approved"`, `operator`: `"eq"` → `to: "done"`
|
||||
- `"type": "artifact_field_equals"`, `artifact_id`: reviewer's produces id,
|
||||
`field`: `"verdict"`, `value`: `"approved"`, `operator`: `"neq"` → `to: "<implement_stage_id>"`
|
||||
- A reviewer prompt must use only DoD rows tagged `verified_by: reviewer`: approve iff every such
|
||||
row is met and no `out_of_scope` item was introduced. It must cite failed criterion ids and may
|
||||
not invent new requirements. Criteria tagged `gate` are already decided upstream.
|
||||
- Every stage except the first must have an inbound edge from an earlier stage; every
|
||||
stage must have an outbound edge. Unreachable stages fail to compile.
|
||||
|
||||
## Constraints
|
||||
|
||||
- Do not add stages, roles, or tools not justified by the analysis artifact.
|
||||
- Do not add stages, roles, or tools not justified by the DoD artifact.
|
||||
- Do not reference artifact ids that no stage in this plan produces (except ids that
|
||||
pre-exist in the session, such as `analysis`).
|
||||
- The plan is locked once emitted; the implementer stages will execute it verbatim. Be
|
||||
|
||||
@@ -22,24 +22,47 @@ Two checks, both grounded in what you actually read:
|
||||
endpoint.
|
||||
|
||||
Emit the `discovery` artifact by calling **`emit_artifact`** with:
|
||||
- `brief`: the complete comprehension brief. Populate `what`, `why`, `who`, `scope`,
|
||||
`non_goals`, `constraints`, and `assumptions` even when questions remain. Use assumptions for
|
||||
reasonable, visible defaults instead of parking on micro-decisions.
|
||||
- `ready`: `true` when the request is clear and grounded enough to hand to the analyst; `false`
|
||||
when you are raising questions.
|
||||
- `questions`: the open questions (empty when `ready` is true). Batch **all** of them into this
|
||||
one list — do not ask one at a time. Each entry is an object:
|
||||
- `prompt` (required): the question, in full.
|
||||
- `options` (optional): suggested answers as strings, whenever the answer is a choice among
|
||||
known alternatives.
|
||||
- `options` (**required whenever the answer is a choice among known alternatives** — and it
|
||||
almost always is: stack, library, endpoint, layout, priority are all choices among things you
|
||||
can name). Provide 2–4 concrete prefilled answers as strings. An open-ended question with no
|
||||
`options` is only acceptable when no candidate set exists at all. Empty `options` on a
|
||||
choice-question is a defect — enumerate the real candidates you found in the repo.
|
||||
- `multiSelect` (optional, default false): true if more than one option may apply.
|
||||
- `header` (optional): a 1–2 word label (e.g. "Scope", "Stack", "Endpoint").
|
||||
|
||||
**Default to proceeding.** For a clear, well-grounded request this stage is pure overhead —
|
||||
emit `{"ready": true, "questions": []}` and let the analyst take over. Only ask when a genuine
|
||||
**Default to proceeding.** First inspect enough of the repository to enumerate the whole question
|
||||
surface, then ask every genuine operator-only question in one batch. For a clear, well-grounded
|
||||
request, proceed with visible assumptions. Do not raise one question, re-enter, and discover
|
||||
another question that the same initial inspection could have exposed.
|
||||
|
||||
**Converge once answered.** If the decision history above already contains the operator's answers
|
||||
to your questions, you are done vetting — emit the brief with `ready: true` and empty `questions`.
|
||||
Do NOT re-explore the repo hunting for new questions after the operator has answered; fold their
|
||||
answers into the brief and hand off. You get **one** clarification round: ask everything up front,
|
||||
then commit. Endless re-inspection is a failure, not diligence.
|
||||
fork or contradiction blocks planning. Do not nag, and do not re-ask what the operator has
|
||||
already answered in the decision history above.
|
||||
|
||||
Example (needs input):
|
||||
```json
|
||||
{
|
||||
"brief": {
|
||||
"what": "Build a browser client for the existing session API.",
|
||||
"why": "Operators need a visual session surface.",
|
||||
"who": ["operators"],
|
||||
"scope": ["browser session client"],
|
||||
"non_goals": ["server protocol redesign"],
|
||||
"constraints": ["reuse the existing endpoint"],
|
||||
"assumptions": []
|
||||
},
|
||||
"ready": false,
|
||||
"questions": [
|
||||
{"prompt": "Which frontend stack should the UI target?",
|
||||
@@ -52,5 +75,17 @@ Example (needs input):
|
||||
|
||||
Example (clear — the common case):
|
||||
```json
|
||||
{ "ready": true, "questions": [] }
|
||||
{
|
||||
"brief": {
|
||||
"what": "Add the requested deterministic validation gate.",
|
||||
"why": "Prevent invalid output from reaching review.",
|
||||
"who": ["workflow authors", "operators"],
|
||||
"scope": ["gate execution and recorded verdict"],
|
||||
"non_goals": ["workflow topology redesign"],
|
||||
"constraints": ["replay uses recorded observations"],
|
||||
"assumptions": ["existing event-store contracts remain authoritative"]
|
||||
},
|
||||
"ready": true,
|
||||
"questions": []
|
||||
}
|
||||
```
|
||||
|
||||
+3
@@ -12,4 +12,7 @@ data class ModelDescriptor(
|
||||
val idleTimeoutMs: Long = 60_000L,
|
||||
val contextSize: Int = 24_576,
|
||||
val capabilities: Set<CapabilityScore> = emptySet(),
|
||||
/** Extra llama-server CLI args (already flattened to token order), e.g. draft-MTP speculative
|
||||
* decoding: `-md <draft.gguf> -ngld 99 --spec-type draft-mtp --spec-draft-n-max 4`. */
|
||||
val extraArgs: List<String> = emptyList(),
|
||||
)
|
||||
|
||||
+1
-1
@@ -66,7 +66,7 @@ class DefaultModelManager(
|
||||
"--ctx-size", descriptor.contextSize.toString(),
|
||||
"--host", host,
|
||||
"--port", port.toString(),
|
||||
)
|
||||
) + descriptor.extraArgs
|
||||
val lp = LlamaProcess(command, logFile)
|
||||
lp.start()
|
||||
lp
|
||||
|
||||
@@ -170,6 +170,10 @@ object InfrastructureModule {
|
||||
residencyMode = ResidencyMode.PERSISTENT,
|
||||
contextSize = config.contextSize,
|
||||
capabilities = parseCapabilitiesFromConfig(config.capabilities),
|
||||
// [[models]] `params` (a flag->value map) passes straight through to llama-server, e.g.
|
||||
// draft-MTP speculative decoding. Flatten to token order; each stays a distinct argv entry
|
||||
// (no shell), so paths with spaces are safe.
|
||||
extraArgs = config.params.flatMap { (flag, value) -> listOf(flag, value) },
|
||||
)
|
||||
|
||||
private fun parseCapabilitiesFromConfig(capsMap: Map<String, Double>): Set<CapabilityScore> {
|
||||
|
||||
+21
@@ -42,6 +42,27 @@ class InfrastructureModuleModelTest {
|
||||
assertEquals(0.8, codingScore!!.score)
|
||||
}
|
||||
|
||||
@Test
|
||||
fun `modelConfigToDescriptor flattens params into extraArgs for llama-server`() {
|
||||
val config = ModelConfig(
|
||||
id = "mtp-model",
|
||||
modelPath = "/models/main.gguf",
|
||||
params = linkedMapOf(
|
||||
"-md" to "/models/draft.gguf",
|
||||
"-ngld" to "99",
|
||||
"--spec-type" to "draft-mtp",
|
||||
"--spec-draft-n-max" to "4",
|
||||
),
|
||||
)
|
||||
|
||||
val descriptor = InfrastructureModule.modelConfigToDescriptor(config)
|
||||
|
||||
assertEquals(
|
||||
listOf("-md", "/models/draft.gguf", "-ngld", "99", "--spec-type", "draft-mtp", "--spec-draft-n-max", "4"),
|
||||
descriptor.extraArgs,
|
||||
)
|
||||
}
|
||||
|
||||
@Test
|
||||
fun `modelConfigToDescriptor handles empty capabilities`() {
|
||||
val config = ModelConfig(
|
||||
|
||||
+31
-8
@@ -154,6 +154,12 @@ class FileEditTool(
|
||||
) ->
|
||||
ValidationResult.Invalid("Missing 'target' or 'replacement' parameter for 'replace' operation.")
|
||||
|
||||
// Anchor validity is checkable now, so reject a bad target BEFORE the approval gate
|
||||
// fires — mirroring read/write's file-exists / read-before-write pre-checks. Without
|
||||
// this the operator is prompted to approve an edit that then dies "Target not found"
|
||||
// at execute time. Recoverable: the model corrects the anchor and retries.
|
||||
operation == "replace" -> validateReplaceAnchor(path, pathString, request)
|
||||
|
||||
operation == "patch" && !request.parameters.containsKey("patch") ->
|
||||
ValidationResult.Invalid("Missing 'patch' parameter for 'patch' operation.")
|
||||
|
||||
@@ -177,6 +183,29 @@ class FileEditTool(
|
||||
private fun isPathAllowed(path: Path): Boolean =
|
||||
PathJail.isContained(path, normalizedAllowedPaths)
|
||||
|
||||
/** Valid iff the replace target occurs exactly once. Same 0/>1 messaging as [replace] at execute. */
|
||||
private fun validateReplaceAnchor(path: Path, pathString: String, request: ToolRequest): ValidationResult {
|
||||
val target = request.parameters["target"] as? String ?: return ValidationResult.Valid
|
||||
val content = Files.readString(path)
|
||||
return when (val occurrences = content.split(target).size - 1) {
|
||||
1 -> ValidationResult.Valid
|
||||
0 -> ValidationResult.Invalid(targetNotFoundMessage(pathString, content, target))
|
||||
else -> ValidationResult.Invalid(targetAmbiguousMessage(pathString, occurrences))
|
||||
}
|
||||
}
|
||||
|
||||
private fun targetNotFoundMessage(pathString: String, content: String, target: String): String =
|
||||
buildString {
|
||||
append("Target not found in $pathString")
|
||||
nearestLine(content, target)?.let {
|
||||
append(". Did you mean (whitespace/content may differ): $it ?")
|
||||
}
|
||||
}
|
||||
|
||||
private fun targetAmbiguousMessage(pathString: String, occurrences: Int): String =
|
||||
"Target found $occurrences times in $pathString, expected exactly once — " +
|
||||
"extend it with surrounding lines to make it unique."
|
||||
|
||||
private fun mapExceptionToValidationResult(e: Throwable): ValidationResult =
|
||||
when (e) {
|
||||
is InvalidPathException -> ValidationResult.Invalid("Invalid path format: ${e.message}")
|
||||
@@ -270,18 +299,12 @@ class FileEditTool(
|
||||
// always whitespace/content drift the model can correct from a nudge.
|
||||
0 -> ToolResult.Failure(
|
||||
invocationId = request.invocationId,
|
||||
reason = buildString {
|
||||
append("Target not found in $pathString")
|
||||
nearestLine(currentContent, target)?.let {
|
||||
append(". Did you mean (whitespace/content may differ): $it ?")
|
||||
}
|
||||
},
|
||||
reason = targetNotFoundMessage(pathString, currentContent, target),
|
||||
recoverable = true,
|
||||
)
|
||||
else -> ToolResult.Failure(
|
||||
invocationId = request.invocationId,
|
||||
reason = "Target found $occurrences times in $pathString, expected exactly once — " +
|
||||
"extend it with surrounding lines to make it unique.",
|
||||
reason = targetAmbiguousMessage(pathString, occurrences),
|
||||
recoverable = true,
|
||||
)
|
||||
}
|
||||
|
||||
+38
@@ -154,6 +154,15 @@ class FileWriteTool(
|
||||
|
||||
val originalContent = runCatching { Files.readString(path) }.getOrDefault("")
|
||||
|
||||
// Clobber guard (#245): a full overwrite that shrinks an existing non-trivial file to an
|
||||
// elided stub (literal `...` placeholders) is almost always the model writing a file from
|
||||
// memory instead of editing it — the incident that silently broke SessionRoutes.kt. Block it
|
||||
// (recoverable) and force a targeted file_edit. Independent of approval tier because the
|
||||
// auto-driver approves T2 writes without a human looking.
|
||||
clobberReason(originalContent, content)?.let { reason ->
|
||||
return@withContext ToolResult.Failure(request.invocationId, reason, recoverable = true)
|
||||
}
|
||||
|
||||
val result = runCatching {
|
||||
Files.createDirectories(path.parent)
|
||||
AtomicFileWriter.write(path, content.toByteArray(Charsets.UTF_8))
|
||||
@@ -180,6 +189,30 @@ class FileWriteTool(
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* Returns a rejection reason when [newContent] would overwrite a non-trivial existing file
|
||||
* ([oldContent]) with a drastically-shrunk elided stub, else null. Requires BOTH a big shrink
|
||||
* and elision markers so legitimate refactors that only shrink (dead-code removal) and new-file
|
||||
* scaffolds that only elide (target didn't exist → oldContent is "") both pass untouched.
|
||||
*/
|
||||
private fun clobberReason(oldContent: String, newContent: String): String? {
|
||||
val oldLines = oldContent.lines().count { it.isNotBlank() }
|
||||
if (oldLines < MIN_GUARDED_LINES) return null
|
||||
val newLines = newContent.lines().count { it.isNotBlank() }
|
||||
val shrankHard = newLines < oldLines * MAX_SHRINK_RATIO
|
||||
if (!shrankHard || !hasElisionMarker(newContent)) return null
|
||||
return "Refusing full overwrite: this replaces $oldLines lines of existing code with a " +
|
||||
"$newLines-line stub containing '...' placeholders. Use file_edit (operation 'replace') " +
|
||||
"for a targeted change instead of rewriting the whole file from memory."
|
||||
}
|
||||
|
||||
/** A line that is just an elision placeholder (`...`, `// ...`, `# ... existing code ...`). */
|
||||
private fun hasElisionMarker(content: String): Boolean =
|
||||
content.lineSequence().any { line ->
|
||||
val t = line.trim().trim('/', '#', '*', '<', '!', '-', ' ')
|
||||
t == "..." || t == "…" || t.endsWith(" ...") || t.endsWith(" …")
|
||||
}
|
||||
|
||||
private fun handleExecutionException(
|
||||
e: Throwable,
|
||||
invocationId: ToolInvocationId,
|
||||
@@ -207,4 +240,9 @@ class FileWriteTool(
|
||||
recoverable = false,
|
||||
)
|
||||
}
|
||||
|
||||
private companion object {
|
||||
const val MIN_GUARDED_LINES = 30
|
||||
const val MAX_SHRINK_RATIO = 0.4
|
||||
}
|
||||
}
|
||||
|
||||
+28
@@ -95,6 +95,34 @@ class FileEditToolTest {
|
||||
assertTrue((result as ValidationResult.Invalid).reason.contains("File not found"))
|
||||
}
|
||||
|
||||
@Test
|
||||
fun `validateRequest rejects a replace whose target is absent or ambiguous, before execute`(): Unit = runBlocking {
|
||||
val tempDir = Files.createTempDirectory("file_edit_anchor")
|
||||
val filePath = tempDir.resolve("test.txt")
|
||||
Files.writeString(filePath, "alpha\nbeta\nbeta\n")
|
||||
val tool = FileEditTool(allowedPaths = setOf(tempDir))
|
||||
|
||||
val absent = tool.validateRequest(
|
||||
createRequest(mapOf("operation" to "replace", "path" to filePath.toString(),
|
||||
"target" to "gamma", "replacement" to "x")),
|
||||
)
|
||||
assertTrue(absent is ValidationResult.Invalid)
|
||||
assertTrue((absent as ValidationResult.Invalid).reason.contains("Target not found"))
|
||||
|
||||
val ambiguous = tool.validateRequest(
|
||||
createRequest(mapOf("operation" to "replace", "path" to filePath.toString(),
|
||||
"target" to "beta", "replacement" to "x")),
|
||||
)
|
||||
assertTrue(ambiguous is ValidationResult.Invalid)
|
||||
assertTrue((ambiguous as ValidationResult.Invalid).reason.contains("expected exactly once"))
|
||||
|
||||
val unique = tool.validateRequest(
|
||||
createRequest(mapOf("operation" to "replace", "path" to filePath.toString(),
|
||||
"target" to "alpha", "replacement" to "x")),
|
||||
)
|
||||
assertEquals(ValidationResult.Valid, unique)
|
||||
}
|
||||
|
||||
@Test
|
||||
fun `validateRequest returns Invalid for missing parameters`(): Unit = runBlocking {
|
||||
val tempDir = Files.createTempDirectory("file_edit_test")
|
||||
|
||||
+29
@@ -132,6 +132,35 @@ class FileWriteToolTest {
|
||||
assertEquals("hello", Files.readString(target))
|
||||
}
|
||||
|
||||
@Test
|
||||
fun `clobber guard blocks overwriting real code with an elided stub`(): Unit = runBlocking {
|
||||
val tempDir = Files.createTempDirectory("file_write_clobber")
|
||||
val target = tempDir.resolve("Routes.kt")
|
||||
Files.writeString(target, (1..273).joinToString("\n") { "line $it = something()" })
|
||||
val tool = FileWriteTool(allowedPaths = setOf(tempDir))
|
||||
val stub = "fun routes() {\n // ... existing routes ...\n ...\n}"
|
||||
val result = tool.execute(createRequest(mapOf("path" to target.toString(), "content" to stub)))
|
||||
|
||||
assertTrue(result is ToolResult.Failure)
|
||||
result as ToolResult.Failure
|
||||
assertTrue(result.recoverable)
|
||||
assertTrue(result.reason.contains("file_edit"))
|
||||
assertEquals(273, Files.readString(target).lines().size) // original untouched
|
||||
}
|
||||
|
||||
@Test
|
||||
fun `clobber guard allows a genuine shrink with no placeholders`(): Unit = runBlocking {
|
||||
val tempDir = Files.createTempDirectory("file_write_shrink")
|
||||
val target = tempDir.resolve("Routes.kt")
|
||||
Files.writeString(target, (1..273).joinToString("\n") { "line $it = something()" })
|
||||
val tool = FileWriteTool(allowedPaths = setOf(tempDir))
|
||||
val trimmed = (1..40).joinToString("\n") { "kept $it = something()" }
|
||||
val result = tool.execute(createRequest(mapOf("path" to target.toString(), "content" to trimmed)))
|
||||
|
||||
assertTrue(result is ToolResult.Success)
|
||||
assertEquals(trimmed, Files.readString(target))
|
||||
}
|
||||
|
||||
@Test
|
||||
fun `write atomically overwrites an existing file and leaves no temp files`(): Unit = runBlocking {
|
||||
val tempDir = Files.createTempDirectory("file_write_overwrite")
|
||||
|
||||
+21
-5
@@ -164,15 +164,30 @@ class ShellTool(
|
||||
}
|
||||
}
|
||||
|
||||
// argv[0] is the program name; embedded quotes/commas/whitespace mean the array collapsed into
|
||||
// one string element and there is no real executable to run.
|
||||
private fun checkExecutable(argv: List<String>): ArgvParse =
|
||||
if (argv[0].any { it == '"' || it == ',' || it.isWhitespace() }) {
|
||||
// argv[0] is the program name. A single element carrying whitespace is the weak-model "whole
|
||||
// command line in one string" collapse (["npm create vite@latest frontend"]) — split it into
|
||||
// tokens so the denylist, allowlist and executor all see a real argv, instead of rejecting it and
|
||||
// looping the stage on a call the model reliably re-emits. A JSON-escape mangle collapses
|
||||
// differently — it leaves quotes/commas in argv[0] (["npm\",\"-v"]) — and stays rejected: there is
|
||||
// no runnable program to recover. (Splitting on whitespace loses quoting of args with spaces —
|
||||
// same ceiling as the `sh -c` join below; fine for agent shell use, not a general shell API.)
|
||||
private fun checkExecutable(raw: List<String>): ArgvParse {
|
||||
val argv = if (raw.size == 1 && raw[0].any { it.isWhitespace() } &&
|
||||
raw[0].none { it == '"' || it == ',' }
|
||||
) {
|
||||
raw[0].trim().split(WHITESPACE_RE)
|
||||
} else {
|
||||
raw
|
||||
}
|
||||
return if (argv[0].any { it == '"' || it == ',' || it.isWhitespace() }) {
|
||||
ArgvParse.Bad(
|
||||
"argv[0] `${argv[0]}` is not a valid executable name — it looks like a collapsed array. " +
|
||||
"Emit each token as a separate string element, e.g. [\"npm\", \"-v\"].",
|
||||
)
|
||||
} else ArgvParse.Ok(argv)
|
||||
} else {
|
||||
ArgvParse.Ok(argv)
|
||||
}
|
||||
}
|
||||
|
||||
override suspend fun execute(request: ToolRequest): ToolResult = withContext(Dispatchers.IO) {
|
||||
validateRequest(request).run {
|
||||
@@ -221,6 +236,7 @@ class ShellTool(
|
||||
// (line 250 of 500) survives the head/tail window instead of being silently dropped.
|
||||
const val SALIENCE_PATTERN = "error|fail|exception|panic|traceback|✗"
|
||||
const val EMPTY_MSG = "Missing or empty 'argv' parameter. Expected a JSON array of strings."
|
||||
val WHITESPACE_RE = Regex("\\s+")
|
||||
val SHELL_BUILTINS = setOf("cd", "export", "source", ".", "pushd", "popd", "umask", "ulimit")
|
||||
val SHELL_OPERATORS = setOf("&&", "||", "|", ">", ">>", "<", ";", "&", "2>", "2>&1")
|
||||
// Operators that chain a new command (its following token is a command position); the rest
|
||||
|
||||
+9
@@ -204,6 +204,15 @@ class ShellToolTest {
|
||||
assertTrue((result as ValidationResult.Invalid).reason.contains("collapsed array"), result.reason)
|
||||
}
|
||||
|
||||
@Test
|
||||
fun `validateRequest splits a collapsed single-string command line into tokens`() {
|
||||
// Repro: weak model emitted ["npm create vite@latest frontend"] — the whole command line as
|
||||
// one whitespace-carrying element (no quotes/commas). Now split into tokens instead of
|
||||
// rejected as a collapsed array, so the stage doesn't loop on a call the model keeps re-emitting.
|
||||
val result = ShellTool().validateRequest(rawArgvRequest(listOf("npm create vite@latest frontend")))
|
||||
assertTrue(result is ValidationResult.Valid, (result as? ValidationResult.Invalid)?.reason ?: "")
|
||||
}
|
||||
|
||||
@Test
|
||||
fun `execute runs a shell command line via sh -c`(): Unit = runBlocking {
|
||||
// Repro: gemma called shell with a builtin/command-line ("cd apps && ..."). Now honoured
|
||||
|
||||
@@ -15,7 +15,9 @@ Adapter for `core:transitions` and `core:inference` workflow interfaces. Depends
|
||||
- `ExecutionPlanCompiler` converts the parsed `ExecutionPlanModel` into a runnable plan for `core:transitions`; compilation is deterministic given the same input model.
|
||||
- `PlanLinter` enforces structural rules (e.g. unreachable stages are rejected); linting failures throw `WorkflowValidationException`.
|
||||
- `PlanDerivedManifest` exposes the write manifest derived from a plan.
|
||||
- For a plan with no explicit build expectation, `ExecutionPlanCompiler` marks every write-declaring stage for the runtime auto build gate; plans with no declared writes are not represented as compiler-verified.
|
||||
- For a plan with no explicit build expectation, `ExecutionPlanCompiler` marks every write-declaring stage and the graph-terminal stage for the runtime auto build gate; plans with no declared writes are not represented as compiler-verified.
|
||||
- When a plan declares a `dod` producer, freestyle implementation/review stages consume that session-scoped artifact; reviewer prompts are compiler-bound to its reviewer-owned criteria.
|
||||
- `Lsp4jDiagnosticsRunner` uses LSP 3.17 pull diagnostics for registered languages and degrades to the static one-shot floor when a server is unavailable.
|
||||
|
||||
## Work Guidance
|
||||
|
||||
|
||||
@@ -9,6 +9,8 @@ dependencies {
|
||||
implementation(project(":core:events"))
|
||||
implementation(project(":core:artifacts"))
|
||||
implementation(project(":core:tools"))
|
||||
implementation(project(":core:kernel"))
|
||||
implementation("org.eclipse.lsp4j:org.eclipse.lsp4j:0.23.1")
|
||||
implementation("com.fasterxml.jackson.dataformat:jackson-dataformat-toml:2.17.0")
|
||||
implementation("com.fasterxml.jackson.module:jackson-module-kotlin:2.17.0")
|
||||
testImplementation "org.junit.jupiter:junit-jupiter"
|
||||
|
||||
+44
-4
@@ -19,6 +19,8 @@ import java.nio.file.Path
|
||||
|
||||
private const val TERMINAL = "done"
|
||||
private const val RECOVERY_STAGE = "recovery"
|
||||
private const val DOD_ARTIFACT = "dod"
|
||||
private const val STATIC_FLOOR_COMMAND = "correx-static-floor"
|
||||
|
||||
// Synthesized recovery stage prompt. Freestyle plans are LLM-emitted and never declare a recovery
|
||||
// stage, but the kernel's retry-agency guard routes a write-less stage's unfixable gate failure to
|
||||
@@ -85,7 +87,16 @@ class ExecutionPlanCompiler(
|
||||
.disable(DeserializationFeature.FAIL_ON_UNKNOWN_PROPERTIES)
|
||||
.build()
|
||||
|
||||
fun compile(planJson: String, workflowId: String): WorkflowGraph {
|
||||
/**
|
||||
* @param sessionArtifactIds artifact ids already produced earlier in the same freestyle session
|
||||
* (e.g. the planning-phase `dod`). The execution plan never re-declares these, so they must be
|
||||
* passed in for the cross-workflow `needs` seam (#264) to thread the DoD into impl/review stages.
|
||||
*/
|
||||
fun compile(
|
||||
planJson: String,
|
||||
workflowId: String,
|
||||
sessionArtifactIds: Set<String> = emptySet(),
|
||||
): WorkflowGraph {
|
||||
val plan = runCatching { mapper.readValue<ExecutionPlanModel>(planJson) }
|
||||
.getOrElse { throw WorkflowValidationException("Failed to parse execution_plan JSON: ${it.message}") }
|
||||
if (plan.stages.isEmpty()) throw WorkflowValidationException("execution_plan has no stages")
|
||||
@@ -115,10 +126,13 @@ class ExecutionPlanCompiler(
|
||||
// FileWritten manifest — the plan's declared kinds don't reveal code-ness, but the written
|
||||
// paths do — so a docs-only plan is left alone and a code plan is always gated.
|
||||
val autoGateStages: Set<String> = if (declaredExpectations.values.all { it == BuildExpectation.NONE }) {
|
||||
plan.stages.filter { it.writes.any { path -> path.isNotBlank() } }.map { it.id }.toSet()
|
||||
val writing = plan.stages.filter { it.writes.any { path -> path.isNotBlank() } }.map { it.id }.toSet()
|
||||
if (writing.isEmpty()) emptySet() else writing + terminalStageId(plan)
|
||||
} else {
|
||||
emptySet()
|
||||
}
|
||||
val sessionArtifacts =
|
||||
plan.stages.mapNotNull { it.produces.takeIf(String::isNotBlank) }.toSet() + sessionArtifactIds
|
||||
|
||||
val stageMap = plan.stages.associate { s ->
|
||||
// `produces` is the unique slot name referenced by needs/edges; `kind` selects the
|
||||
@@ -137,7 +151,7 @@ class ExecutionPlanCompiler(
|
||||
).sorted()
|
||||
StageId(s.id) to StageConfig(
|
||||
produces = listOf(TypedArtifactSlot(name = ArtifactId(s.produces), kind = kind)),
|
||||
needs = s.needs.map { ArtifactId(it) }.toSet(),
|
||||
needs = (s.needs + requiredSessionArtifacts(s, sessionArtifacts)).map { ArtifactId(it) }.toSet(),
|
||||
// Every stage gets the full registered tool universe. The architect's per-stage
|
||||
// `tools` pick was advisory and mis-scoped stages (e.g. an edit-shaped stage granted
|
||||
// file_write but not file_edit → forced full-file rewrites; a repair stage unable to
|
||||
@@ -146,6 +160,7 @@ class ExecutionPlanCompiler(
|
||||
// architect's pick only when no universe was supplied (bare-constructor unit tests).
|
||||
allowedTools = knownTools.ifEmpty { s.tools.toSet() },
|
||||
writeManifest = writeManifest,
|
||||
staticAnalysis = staticFloorCommands(writeManifest),
|
||||
// Option A: the concrete (non-glob) subset of the declared writes are expected files
|
||||
// the contract gate will require to exist. Globs (src/**) are write-permission scopes,
|
||||
// not existence guarantees, so they are excluded here.
|
||||
@@ -156,7 +171,7 @@ class ExecutionPlanCompiler(
|
||||
semanticReview = s.semanticReview,
|
||||
tokenBudget = DEFAULT_STAGE_TOKEN_BUDGET,
|
||||
generationConfig = defaultStageGeneration,
|
||||
metadata = mapOf("promptInline" to s.prompt),
|
||||
metadata = mapOf("role" to s.role, "promptInline" to reviewerBoundPrompt(s, DOD_ARTIFACT in sessionArtifacts)),
|
||||
)
|
||||
}
|
||||
// Inject a recovery stage unless the plan already declares one. Reached only by the kernel's
|
||||
@@ -291,6 +306,31 @@ class ExecutionPlanCompiler(
|
||||
private fun terminalStageId(plan: ExecutionPlanModel): String =
|
||||
plan.edges.firstOrNull { it.to == TERMINAL }?.from ?: plan.stages.last().id
|
||||
|
||||
private fun requiredSessionArtifacts(stage: PlanStage, sessionArtifacts: Set<String>): List<String> =
|
||||
if (
|
||||
DOD_ARTIFACT in sessionArtifacts &&
|
||||
stage.role.lowercase() in setOf("impl", "implementer", "review", "reviewer")
|
||||
) {
|
||||
listOf(DOD_ARTIFACT)
|
||||
} else {
|
||||
emptyList()
|
||||
}
|
||||
|
||||
private fun reviewerBoundPrompt(stage: PlanStage, dodPresent: Boolean): String {
|
||||
if (!dodPresent || stage.role.lowercase() !in setOf("review", "reviewer")) return stage.prompt
|
||||
return stage.prompt + "\n\n" +
|
||||
"The `dod` input artifact is your sole acceptance rubric. Judge only criteria tagged " +
|
||||
"`verified_by: reviewer`; deterministic gates own `verified_by: gate`. Approve if and only if " +
|
||||
"every reviewer criterion is met and no `out_of_scope` item was introduced. On rejection, " +
|
||||
"cite the unmet criterion ids. Never add or imply criteria outside the DoD."
|
||||
}
|
||||
|
||||
private fun staticFloorCommands(writeManifest: List<String>): List<String> {
|
||||
return writeManifest.takeIf { it.isNotEmpty() }
|
||||
?.let { listOf("$STATIC_FLOOR_COMMAND -- ${it.joinToString(" ")}") }
|
||||
?: emptyList()
|
||||
}
|
||||
|
||||
private fun isGlob(path: String): Boolean = path.any { it == '*' || it == '?' || it == '[' }
|
||||
|
||||
private fun validateReachability(
|
||||
|
||||
+1
@@ -10,6 +10,7 @@ data class ExecutionPlanModel(
|
||||
|
||||
data class PlanStage(
|
||||
val id: String = "",
|
||||
val role: String = "",
|
||||
val prompt: String = "",
|
||||
val produces: String = "",
|
||||
val kind: String? = null,
|
||||
|
||||
+142
@@ -0,0 +1,142 @@
|
||||
package com.correx.infrastructure.workflow
|
||||
|
||||
import com.correx.core.events.events.LspDiagnostic
|
||||
import com.correx.core.kernel.orchestration.LspDiagnosticsRequest
|
||||
import com.correx.core.kernel.orchestration.LspDiagnosticsResult
|
||||
import com.correx.core.kernel.orchestration.LspDiagnosticsRunner
|
||||
import kotlinx.coroutines.Dispatchers
|
||||
import kotlinx.coroutines.withContext
|
||||
import org.eclipse.lsp4j.ClientCapabilities
|
||||
import org.eclipse.lsp4j.DidOpenTextDocumentParams
|
||||
import org.eclipse.lsp4j.DocumentDiagnosticParams
|
||||
import org.eclipse.lsp4j.InitializeParams
|
||||
import org.eclipse.lsp4j.InitializedParams
|
||||
import org.eclipse.lsp4j.MessageActionItem
|
||||
import org.eclipse.lsp4j.MessageParams
|
||||
import org.eclipse.lsp4j.PublishDiagnosticsParams
|
||||
import org.eclipse.lsp4j.ShowMessageRequestParams
|
||||
import org.eclipse.lsp4j.TextDocumentIdentifier
|
||||
import org.eclipse.lsp4j.TextDocumentItem
|
||||
import org.eclipse.lsp4j.launch.LSPLauncher
|
||||
import org.eclipse.lsp4j.services.LanguageClient
|
||||
import java.nio.file.Files
|
||||
import java.util.concurrent.CompletableFuture
|
||||
import java.util.concurrent.TimeUnit
|
||||
import kotlin.coroutines.cancellation.CancellationException
|
||||
|
||||
data class LspServerSpec(val id: String, val command: List<String>, val languageId: String)
|
||||
|
||||
class LspServerRegistry(
|
||||
private val byExtension: Map<String, LspServerSpec> = defaults,
|
||||
) {
|
||||
fun serverFor(path: String): LspServerSpec? = byExtension[path.substringAfterLast('.', "").lowercase()]
|
||||
|
||||
companion object {
|
||||
private val typescript = LspServerSpec(
|
||||
"tsserver",
|
||||
listOf("typescript-language-server", "--stdio"),
|
||||
"typescriptreact",
|
||||
)
|
||||
private val rust = LspServerSpec("rust-analyzer", listOf("rust-analyzer"), "rust")
|
||||
private val go = LspServerSpec("gopls", listOf("gopls"), "go")
|
||||
private val clangd = LspServerSpec("clangd", listOf("clangd"), "cpp")
|
||||
val defaults = mapOf(
|
||||
"ts" to typescript.copy(languageId = "typescript"),
|
||||
"tsx" to typescript,
|
||||
"js" to typescript.copy(languageId = "javascript"),
|
||||
"jsx" to typescript.copy(languageId = "javascriptreact"),
|
||||
"rs" to rust,
|
||||
"go" to go,
|
||||
"c" to clangd.copy(languageId = "c"),
|
||||
"cc" to clangd,
|
||||
"cpp" to clangd,
|
||||
"h" to clangd.copy(languageId = "c"),
|
||||
"hpp" to clangd,
|
||||
)
|
||||
}
|
||||
}
|
||||
|
||||
/** LSP 3.17 pull-diagnostics adapter. Missing servers degrade to the static one-shot floor. */
|
||||
class Lsp4jDiagnosticsRunner(
|
||||
private val registry: LspServerRegistry = LspServerRegistry(),
|
||||
private val timeoutSeconds: Long = 30,
|
||||
) : LspDiagnosticsRunner {
|
||||
override suspend fun pull(request: LspDiagnosticsRequest): LspDiagnosticsResult = withContext(Dispatchers.IO) {
|
||||
val groups = request.paths
|
||||
.mapNotNull { path -> registry.serverFor(path)?.let { it to path } }
|
||||
.groupBy({ it.first }, { it.second })
|
||||
if (groups.isEmpty()) {
|
||||
return@withContext LspDiagnosticsResult(skippedReason = "no language server registered for touched files")
|
||||
}
|
||||
val diagnostics = mutableListOf<LspDiagnostic>()
|
||||
val skipped = mutableListOf<String>()
|
||||
groups.forEach { (server, paths) ->
|
||||
runCatching { diagnostics += pullFromServer(request, server, paths) }
|
||||
.onFailure {
|
||||
if (it is CancellationException) throw it
|
||||
skipped += "${server.id}: ${it.message ?: it::class.simpleName}"
|
||||
}
|
||||
}
|
||||
LspDiagnosticsResult(
|
||||
server = groups.keys.joinToString(",") { it.id },
|
||||
diagnostics = diagnostics,
|
||||
skippedReason = skipped.takeIf { it.isNotEmpty() }?.joinToString("; "),
|
||||
)
|
||||
}
|
||||
|
||||
private fun pullFromServer(
|
||||
request: LspDiagnosticsRequest,
|
||||
spec: LspServerSpec,
|
||||
paths: List<String>,
|
||||
): List<LspDiagnostic> {
|
||||
val process = ProcessBuilder(spec.command).directory(request.workspaceRoot.toFile()).start()
|
||||
val stderrDrain = Thread.ofVirtual().start { process.errorStream.bufferedReader().use { it.readText() } }
|
||||
val launcher = LSPLauncher.createClientLauncher(QuietLanguageClient, process.inputStream, process.outputStream)
|
||||
val listening = launcher.startListening()
|
||||
val server = launcher.remoteProxy
|
||||
return try {
|
||||
server.initialize(
|
||||
InitializeParams().apply {
|
||||
rootUri = request.workspaceRoot.toUri().toString()
|
||||
capabilities = ClientCapabilities()
|
||||
},
|
||||
).get(timeoutSeconds, TimeUnit.SECONDS)
|
||||
server.initialized(InitializedParams())
|
||||
paths.flatMap { path ->
|
||||
val file = request.workspaceRoot.resolve(path).normalize()
|
||||
val uri = file.toUri().toString()
|
||||
server.textDocumentService.didOpen(
|
||||
DidOpenTextDocumentParams(TextDocumentItem(uri, spec.languageId, 1, Files.readString(file))),
|
||||
)
|
||||
val report = server.textDocumentService.diagnostic(
|
||||
DocumentDiagnosticParams(TextDocumentIdentifier(uri)),
|
||||
).get(timeoutSeconds, TimeUnit.SECONDS)
|
||||
report.relatedFullDocumentDiagnosticReport.items.orEmpty().map { diagnostic ->
|
||||
LspDiagnostic(
|
||||
path = path,
|
||||
line = diagnostic.range.start.line,
|
||||
character = diagnostic.range.start.character,
|
||||
severity = diagnostic.severity?.name?.lowercase() ?: "error",
|
||||
code = diagnostic.code?.let { if (it.isLeft) it.left else it.right.toString() },
|
||||
message = diagnostic.message,
|
||||
)
|
||||
}
|
||||
}
|
||||
} finally {
|
||||
runCatching { server.shutdown().get(timeoutSeconds, TimeUnit.SECONDS) }
|
||||
server.exit()
|
||||
listening.cancel(true)
|
||||
process.destroy()
|
||||
stderrDrain.join(TimeUnit.SECONDS.toMillis(1))
|
||||
}
|
||||
}
|
||||
|
||||
private object QuietLanguageClient : LanguageClient {
|
||||
override fun telemetryEvent(`object`: Any?) = Unit
|
||||
override fun publishDiagnostics(diagnostics: PublishDiagnosticsParams) = Unit
|
||||
override fun showMessage(messageParams: MessageParams) = Unit
|
||||
override fun showMessageRequest(requestParams: ShowMessageRequestParams): CompletableFuture<MessageActionItem> =
|
||||
CompletableFuture.completedFuture(null)
|
||||
override fun logMessage(message: MessageParams) = Unit
|
||||
}
|
||||
}
|
||||
+53
-12
@@ -35,14 +35,35 @@ object PlanGrounder {
|
||||
graph: WorkflowGraph,
|
||||
repoMapPaths: Set<String>,
|
||||
profileCommands: Map<String, String>,
|
||||
// False = no repo scan was recorded for this session, so [repoMapPaths] is "unknown", not
|
||||
// "empty workspace". Scope grounding proves a path ABSENT, which an unknown set can't do —
|
||||
// running it against a missing scan flags every real path as nonexistent (the apps/server
|
||||
// "doesn't exist?!" false reject). Skip scope grounding then; the build-manifest check still
|
||||
// runs (it degrades safely via manifestProducedByPlan). A recorded scan with 0 entries is a
|
||||
// genuine greenfield workspace — scanned=true, so scope grounding still catches a missing scaffold.
|
||||
scanned: Boolean = true,
|
||||
): PlanGroundingResult {
|
||||
val manifestInWorkspace = repoMapPaths.any(::isBuildManifest)
|
||||
val manifestProducedByPlan = graph.stages.values
|
||||
.flatMap { it.writeManifest + it.expectedFiles }
|
||||
.any(::isBuildManifest)
|
||||
val prerequisiteAvailable = manifestInWorkspace || manifestProducedByPlan
|
||||
|
||||
// Files any stage in the plan will create — the "will exist by run time" set for scope grounding.
|
||||
// A stage that can WRITE files (`file_write`) POPULATES its declared `touches` scope at run time
|
||||
// — including files a scaffolder (`npm create vite …`) generates but the plan never declares in
|
||||
// `writes`/`expectedFiles` (the architect prompt tells the model to use the real scaffolder, not
|
||||
// hand-write package.json). Keyed on file_write, NOT shell: a scaffold stage carries file_write
|
||||
// for the files it authors, whereas a read-only shell stage (log inspection, test runs, grep)
|
||||
// creates nothing and must not have its scope credited. Without this, grounding rejects every
|
||||
// plan that follows the scaffolder guidance (the d038468a "frontend/ doesn't exist / no manifest"
|
||||
// false reject). Runtime precondition handling (#167/#170) stays the backstop for a build whose
|
||||
// prerequisite genuinely never materialises.
|
||||
val writeScopes = graph.stages.values
|
||||
.filter { "file_write" in it.effectiveAllowedTools }
|
||||
.flatMap { it.touches }
|
||||
.mapNotNull(::scopePrefix)
|
||||
val prerequisiteAvailable = manifestInWorkspace || manifestProducedByPlan || writeScopes.isNotEmpty()
|
||||
|
||||
// Files/dirs any stage in the plan will create — the "will exist by run time" set for scope grounding.
|
||||
val createdPaths = graph.stages.values.flatMap { it.expectedFiles + it.writeManifest }
|
||||
|
||||
val findings = graph.stages.entries.flatMap { (id, cfg) ->
|
||||
@@ -60,14 +81,16 @@ object PlanGrounder {
|
||||
// scope that matches NOTHING in the recorded repo map AND that no plan stage creates a file
|
||||
// under is grounded on a directory that will never exist — the exact session-a35cf1e3
|
||||
// "scoped to EXISTING frontend/ that isn't there" failure. Caught at stage 0, not stage 11.
|
||||
cfg.touches
|
||||
.filter { glob -> !scopeSatisfied(glob, repoMapPaths, createdPaths) }
|
||||
.forEach { glob ->
|
||||
add(
|
||||
"stage ${id.value}: scoped to '$glob' but nothing under it exists in the repo map " +
|
||||
"and no plan stage creates a file there — scaffold that path first or fix the scope.",
|
||||
)
|
||||
}
|
||||
if (scanned) {
|
||||
cfg.touches
|
||||
.filter { glob -> !scopeSatisfied(glob, repoMapPaths, createdPaths, writeScopes) }
|
||||
.forEach { glob ->
|
||||
add(
|
||||
"stage ${id.value}: scoped to '$glob' but nothing under it exists in the repo map " +
|
||||
"and no plan stage creates a file there — scaffold that path first or fix the scope.",
|
||||
)
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
@@ -77,8 +100,21 @@ object PlanGrounder {
|
||||
)
|
||||
}
|
||||
|
||||
/** True if any existing repo-map path or any plan-created path falls under the [glob] scope. */
|
||||
private fun scopeSatisfied(glob: String, repoMapPaths: Set<String>, createdPaths: List<String>): Boolean {
|
||||
/**
|
||||
* True if the [glob] scope will be populated by run time: an existing repo-map path or a
|
||||
* plan-created path falls under it, OR it overlaps a scaffold stage's scope ([writeScopes] —
|
||||
* either direction, since a scaffolder under `frontend/src` also makes a `frontend/` scope non-empty
|
||||
* and a scaffolder of `frontend/` covers `frontend/src`).
|
||||
*/
|
||||
private fun scopeSatisfied(
|
||||
glob: String,
|
||||
repoMapPaths: Set<String>,
|
||||
createdPaths: List<String>,
|
||||
writeScopes: List<String>,
|
||||
): Boolean {
|
||||
scopePrefix(glob)?.let { p ->
|
||||
if (writeScopes.any { it == p || it.startsWith("$p/") || p.startsWith("$it/") }) return true
|
||||
}
|
||||
val matcher: PathMatcher = runCatching {
|
||||
FileSystems.getDefault().getPathMatcher("glob:${glob.trimStart('/')}")
|
||||
}.getOrElse { return true } // ponytail: an unparseable glob is not a grounding failure — don't block on it.
|
||||
@@ -86,6 +122,11 @@ object PlanGrounder {
|
||||
return repoMapPaths.any(::matches) || createdPaths.any(::matches)
|
||||
}
|
||||
|
||||
// A scope glob reduced to its literal directory prefix (a "frontend" wildcard glob → "frontend"),
|
||||
// or null when the glob has no literal prefix.
|
||||
private fun scopePrefix(glob: String): String? =
|
||||
glob.trimStart('/').substringBefore('*').trimEnd('/').ifEmpty { null }
|
||||
|
||||
private fun isBuildManifest(path: String): Boolean =
|
||||
path.substringAfterLast('/').lowercase() in BUILD_MANIFEST_FILENAMES
|
||||
|
||||
|
||||
+20
-10
@@ -37,14 +37,17 @@ object PlanLinter {
|
||||
* planning phase, not by any plan stage. The architect prompt explicitly allows `needs` to
|
||||
* reference these (notably `analysis`), so the linter must treat them as available producers.
|
||||
*/
|
||||
private val seedArtifacts = setOf("analysis")
|
||||
private val seedArtifacts = setOf("analysis", "dod")
|
||||
|
||||
fun lint(graph: WorkflowGraph, seeds: Set<String> = seedArtifacts): PlanLintResult =
|
||||
PlanLintResult(
|
||||
hardFailures = unproducedNeeds(graph, seeds) +
|
||||
trapStates(graph) +
|
||||
// HARD = deterministic graph facts, provably broken regardless of wording. SOFT = heuristics
|
||||
// (incl. H3's prose word-match, which cannot tell a forgotten dep from a descriptive mention)
|
||||
// — they score the plan but must NOT block it, or a weak architect burns retries it can't
|
||||
// reword its way out of.
|
||||
hardFailures = unproducedNeeds(graph, seeds) + trapStates(graph),
|
||||
softFindings = stageCount(graph) + fanOut(graph) + emptyBriefs(graph) + duplicateBriefs(graph) +
|
||||
unreferencedPromptArtifacts(graph, seeds),
|
||||
softFindings = stageCount(graph) + fanOut(graph) + emptyBriefs(graph) + duplicateBriefs(graph),
|
||||
)
|
||||
|
||||
/** H1: a stage `needs` an artifact that neither a plan stage `produces` nor a seed provides. */
|
||||
@@ -92,14 +95,21 @@ object PlanLinter {
|
||||
}
|
||||
|
||||
/**
|
||||
* H3: a stage's inline prompt mentions an artifact ID that is produced by the plan or
|
||||
* provided by a seed, but the stage does not declare it in its `needs`. The stage will
|
||||
* not have access to that artifact at runtime, so any reference in the prompt is a dangling
|
||||
* assumption that will confuse or stall the model — force the architect to either add the
|
||||
* artifact to `needs` or reword the prompt.
|
||||
* H3 (SOFT): a stage's inline prompt word-matches a plan-produced artifact ID the stage does not
|
||||
* declare in `needs` — possibly a dangling reference the model can't resolve at runtime. This is a
|
||||
* prose heuristic, not a graph fact: it cannot distinguish a forgotten dependency from a
|
||||
* descriptive mention ("unlike the `report` stage"), so it only penalises the plan's score and
|
||||
* never blocks it. Seeds are excluded (see below); a genuine missing consumption is still caught
|
||||
* structurally by H1 when the stage actually declares the `needs`.
|
||||
*/
|
||||
private fun unreferencedPromptArtifacts(graph: WorkflowGraph, seeds: Set<String>): List<PlanLintFinding> {
|
||||
val produced = graph.stages.values.flatMap { it.produces }.map { it.name.value }.toSet() + seeds
|
||||
// Only plan-PRODUCED artifacts are checked — a prompt that references a distinctly-named one
|
||||
// (e.g. `server_api_analysis`) but omits it from `needs` is a real dangling dependency. Seed
|
||||
// IDs (`analysis`, `dod`) are EXCLUDED: they double as ordinary words, so a prose mention
|
||||
// ("refine the analysis", "meets the DoD") is not a dependency signal — flagging it just churns
|
||||
// architect retries on a false positive the model cannot word around. (A stage that genuinely
|
||||
// consumes a seed still declares it in `needs`; H1 keeps that path honest.)
|
||||
val produced = graph.stages.values.flatMap { it.produces }.map { it.name.value }.toSet() - seeds
|
||||
return graph.stages.entries.flatMap { (id, stage) ->
|
||||
val prompt = briefOf(stage)
|
||||
val needIds = stage.needs.map { it.value }.toSet()
|
||||
|
||||
+57
@@ -499,6 +499,63 @@ class ExecutionPlanCompilerTest {
|
||||
graph.stages[StageId("entry")]!!.autoBuildGate,
|
||||
"implementation stages must not defer compiler coverage to a terminal verifier",
|
||||
)
|
||||
assertTrue(graph.stages[StageId("entry")]!!.staticAnalysis.single().contains("src/main.tsx"))
|
||||
assertTrue(graph.stages[StageId("views")]!!.staticAnalysis.single().contains("src/App.tsx"))
|
||||
}
|
||||
|
||||
@Test
|
||||
fun `non-writing review terminal receives the auto build gate`() {
|
||||
val planned = """
|
||||
{
|
||||
"goal": "implement then review",
|
||||
"stages": [
|
||||
{ "id": "impl", "role": "implementer", "prompt": "write code", "produces": "code",
|
||||
"kind": "react_entry", "writes": ["src/main.tsx"] },
|
||||
{ "id": "review", "role": "reviewer", "prompt": "review against dod", "produces": "verdict",
|
||||
"kind": "react_component", "needs": ["code"] }
|
||||
],
|
||||
"edges": [
|
||||
{ "from": "impl", "to": "review", "condition": { "type": "always_true" } },
|
||||
{ "from": "review", "to": "done", "condition": { "type": "always_true" } }
|
||||
]
|
||||
}
|
||||
""".trimIndent()
|
||||
val graph = codeCompiler.compile(planned, "wf")
|
||||
assertTrue(graph.stages.getValue(StageId("review")).autoBuildGate)
|
||||
// No dod in session → nothing to thread (execution plan never re-produces it).
|
||||
assertTrue(graph.stages.getValue(StageId("review")).needs.none { it.value == "dod" })
|
||||
|
||||
// Planning-phase dod present in the session → threaded into impl + review needs (#264 seam).
|
||||
val withDod = codeCompiler.compile(planned, "wf", sessionArtifactIds = setOf("dod"))
|
||||
assertTrue(withDod.stages.getValue(StageId("review")).needs.any { it.value == "dod" })
|
||||
assertTrue(withDod.stages.getValue(StageId("impl")).needs.any { it.value == "dod" })
|
||||
}
|
||||
|
||||
@Test
|
||||
fun `implementation and review consume dod only when the plan produces it`() {
|
||||
val planned = """
|
||||
{
|
||||
"goal": "define, implement, and review",
|
||||
"stages": [
|
||||
{ "id": "analyst", "role": "analyst", "prompt": "define dod", "produces": "dod",
|
||||
"kind": "react_component" },
|
||||
{ "id": "impl", "role": "implementer", "prompt": "write code", "produces": "code",
|
||||
"kind": "react_entry", "writes": ["src/main.tsx"] },
|
||||
{ "id": "review", "role": "reviewer", "prompt": "review", "produces": "verdict",
|
||||
"kind": "react_component", "needs": ["code"] }
|
||||
],
|
||||
"edges": [
|
||||
{ "from": "analyst", "to": "impl", "condition": { "type": "always_true" } },
|
||||
{ "from": "impl", "to": "review", "condition": { "type": "always_true" } },
|
||||
{ "from": "review", "to": "done", "condition": { "type": "always_true" } }
|
||||
]
|
||||
}
|
||||
""".trimIndent()
|
||||
|
||||
val graph = codeCompiler.compile(planned, "wf-with-dod")
|
||||
|
||||
assertTrue(graph.stages.getValue(StageId("impl")).needs.any { it.value == "dod" })
|
||||
assertTrue(graph.stages.getValue(StageId("review")).needs.any { it.value == "dod" })
|
||||
}
|
||||
|
||||
@Test
|
||||
|
||||
+2
-1
@@ -28,6 +28,7 @@ class FreestylePlanningWorkflowTest {
|
||||
private val registry = DefaultArtifactKindRegistry().apply {
|
||||
register(llmKind("discovery"))
|
||||
register(llmKind("analysis"))
|
||||
register(llmKind("dod"))
|
||||
register(llmKind("execution_plan"))
|
||||
}
|
||||
|
||||
@@ -64,7 +65,7 @@ class FreestylePlanningWorkflowTest {
|
||||
"architect must produce execution_plan",
|
||||
)
|
||||
|
||||
assertTrue(architect.needs.map { it.value }.contains("analysis"), "architect needs analysis")
|
||||
assertTrue(architect.needs.map { it.value }.contains("dod"), "architect needs the fixed DoD")
|
||||
|
||||
val analystToArchitect = graph.transitions.first { it.from == StageId("analyst") }
|
||||
assertEquals(StageId("architect"), analystToArchitect.to)
|
||||
|
||||
+22
@@ -0,0 +1,22 @@
|
||||
package com.correx.infrastructure.workflow
|
||||
|
||||
import kotlin.test.Test
|
||||
import kotlin.test.assertEquals
|
||||
import kotlin.test.assertNull
|
||||
|
||||
class LspServerRegistryTest {
|
||||
private val registry = LspServerRegistry()
|
||||
|
||||
@Test
|
||||
fun `registry maps language extensions to servers`() {
|
||||
assertEquals("tsserver", registry.serverFor("src/App.tsx")?.id)
|
||||
assertEquals("rust-analyzer", registry.serverFor("src/main.rs")?.id)
|
||||
assertEquals("gopls", registry.serverFor("cmd/main.go")?.id)
|
||||
assertEquals("clangd", registry.serverFor("src/main.cpp")?.id)
|
||||
}
|
||||
|
||||
@Test
|
||||
fun `unknown extension falls through`() {
|
||||
assertNull(registry.serverFor("README.md"))
|
||||
}
|
||||
}
|
||||
+35
@@ -65,6 +65,13 @@ class PlanGrounderTest {
|
||||
assertTrue(r.findings.single().contains("frontend/**"))
|
||||
}
|
||||
|
||||
@Test
|
||||
fun `scanned=false skips scope grounding — a missing scan can't prove a path absent`() {
|
||||
val g = graph("impl" to StageConfig(touches = listOf("frontend/**", "apps/server/**")))
|
||||
val r = PlanGrounder.ground(g, repoMapPaths = emptySet(), profileCommands = cmds, scanned = false)
|
||||
assertEquals(PlanGroundingVerdict.PASS, r.verdict)
|
||||
}
|
||||
|
||||
@Test
|
||||
fun `a scope satisfied by an existing repo path grounds`() {
|
||||
val g = graph("impl" to StageConfig(touches = listOf("frontend/**")))
|
||||
@@ -82,6 +89,34 @@ class PlanGrounderTest {
|
||||
assertEquals(PlanGroundingVerdict.PASS, r.verdict)
|
||||
}
|
||||
|
||||
@Test
|
||||
fun `a file-writing scaffold stage grounds its own scope, a later scoped stage, and a build`() {
|
||||
// Repro session d038468a: scaffold stage runs `npm create vite` (file_write + shell) scoped to
|
||||
// frontend/, creating package.json + the tree at run time — invisible to declared writes/
|
||||
// expectedFiles. Grounding must credit that scope, or it false-rejects a plan that follows the
|
||||
// "use the real scaffolder" guidance (frontend/ "doesn't exist", verify stage has "no manifest").
|
||||
val g = graph(
|
||||
"scaffold" to StageConfig(allowedTools = setOf("shell", "file_write"), touches = listOf("frontend/")),
|
||||
"design" to StageConfig(touches = listOf("frontend/**")),
|
||||
"verify" to StageConfig(buildExpectation = BuildExpectation.PROJECT),
|
||||
)
|
||||
val r = PlanGrounder.ground(g, repoMapPaths = emptySet(), profileCommands = cmds)
|
||||
assertEquals(PlanGroundingVerdict.PASS, r.verdict, "findings: ${r.findings}")
|
||||
}
|
||||
|
||||
@Test
|
||||
fun `a read-only shell stage does NOT credit its scope (no file_write, creates nothing)`() {
|
||||
// A stage with shell for log inspection / test runs — not scaffolding — writes nothing, so its
|
||||
// scope must not satisfy a build's manifest prerequisite. Keeps the credit keyed on create-intent.
|
||||
val g = graph(
|
||||
"inspect" to StageConfig(allowedTools = setOf("shell"), touches = listOf("logs/")),
|
||||
"verify" to StageConfig(buildExpectation = BuildExpectation.PROJECT),
|
||||
)
|
||||
val r = PlanGrounder.ground(g, repoMapPaths = emptySet(), profileCommands = cmds)
|
||||
assertEquals(PlanGroundingVerdict.RETURN_TO_ARCHITECT, r.verdict)
|
||||
assertTrue(r.findings.any { it.contains("verify") && it.contains("no build manifest") })
|
||||
}
|
||||
|
||||
@Test
|
||||
fun `NONE expectation never grounds`() {
|
||||
val g = graph("impl" to StageConfig(buildExpectation = BuildExpectation.NONE))
|
||||
|
||||
+19
-3
@@ -107,7 +107,23 @@ class PlanLinterTest {
|
||||
}
|
||||
|
||||
@Test
|
||||
fun `a prompt mentioning another stage's artifact not in needs is a hard failure`() {
|
||||
fun `a prompt using a seed name as an ordinary word is not a failure`() {
|
||||
// Repro: architect burned 3 retries because briefs mentioning "analysis"/"dod" as prose kept
|
||||
// tripping H3 on the seed IDs — words the model cannot reword away.
|
||||
val g = graph(
|
||||
stages = mapOf(
|
||||
"a" to stage(produces = listOf("scope"), prompt = "refine the analysis and meet the dod"),
|
||||
"b" to stage(needs = listOf("scope"), prompt = "build"),
|
||||
),
|
||||
edges = listOf("a" to "b", "b" to "done"),
|
||||
start = "a",
|
||||
)
|
||||
val result = PlanLinter.lint(g)
|
||||
assertTrue(result.passed, "seed names in prose must not fire H3, hard=${result.hardFailures}")
|
||||
}
|
||||
|
||||
@Test
|
||||
fun `a prompt mentioning another stage's artifact not in needs is a soft finding, not a block`() {
|
||||
val g = graph(
|
||||
stages = mapOf(
|
||||
"a" to stage(produces = listOf("plan"), prompt = "write the plan artifact"),
|
||||
@@ -117,8 +133,8 @@ class PlanLinterTest {
|
||||
start = "a",
|
||||
)
|
||||
val result = PlanLinter.lint(g)
|
||||
assertFalse(result.passed)
|
||||
val h = result.hardFailures.single { it.code == "unreferenced_prompt_artifact" }
|
||||
assertTrue(result.passed, "H3 is a prose heuristic — it scores but must not block the plan")
|
||||
val h = result.softFindings.single { it.code == "unreferenced_prompt_artifact" }
|
||||
assertEquals("b", h.stageId)
|
||||
assertTrue(h.detail.contains("plan"))
|
||||
}
|
||||
|
||||
@@ -2,7 +2,7 @@
|
||||
|
||||
## Purpose
|
||||
|
||||
End-to-end integration tests that wire multiple core modules together without live LLM or network. Covers: `SessionOrchestrator` full flows, tool-call gate, freestyle approval gate, clarification loop, steering preemption, graph reroute, artifact injection, workspace-scoped tool registry, validation pipeline integration, critique calibration wiring, session rehydration, repo-map reuse, and subagent runner seam.
|
||||
End-to-end integration tests that wire multiple core modules together without live LLM or network. Covers: `SessionOrchestrator` full flows, tool-call gate, freestyle approval gate, terminal-review gate ordering, bounded review-loop recovery, clarification loop, steering preemption, graph reroute, artifact injection, workspace-scoped tool registry, validation pipeline integration, critique calibration wiring, session rehydration, repo-map reuse, and subagent runner seam.
|
||||
|
||||
## Ownership
|
||||
|
||||
|
||||
@@ -52,14 +52,14 @@ import kotlinx.coroutines.withTimeout
|
||||
import kotlinx.coroutines.yield
|
||||
import org.junit.jupiter.api.Assertions.assertEquals
|
||||
import org.junit.jupiter.api.Assertions.assertNotNull
|
||||
import org.junit.jupiter.api.Assertions.assertTrue
|
||||
import org.junit.jupiter.api.Test
|
||||
|
||||
/**
|
||||
* Integration test for the producer-exit clarification loop: a stage whose produced artifact
|
||||
* Integration test for the producer-exit clarification handoff: a stage whose produced artifact
|
||||
* carries an unanswered `questions` array parks its run, surfaces a [ClarificationRequestedEvent],
|
||||
* and on operator answer re-runs the same stage with the answers in context — without consuming the
|
||||
* failure-retry budget (retry maxAttempts is 1 here). Bounded at three rounds.
|
||||
* and on operator answer the run ADVANCES to the next stage with the answers in context — the
|
||||
* questioning stage is not re-run (a re-run would restart inference with no prior reasoning and
|
||||
* make the stage re-explore; 2026-07-18). The answers reach later stages as session-wide events.
|
||||
*/
|
||||
class ClarificationLoopTest {
|
||||
|
||||
@@ -147,7 +147,7 @@ class ClarificationLoopTest {
|
||||
private val config = OrchestrationConfig(retryPolicy = RetryPolicy(maxAttempts = 1, backoffMs = 0))
|
||||
|
||||
@Test
|
||||
fun `open questions park the stage, the answer re-runs it, then the workflow completes`():
|
||||
fun `open questions park the stage, the answer advances to the next stage`():
|
||||
Unit = runBlocking {
|
||||
val sessionId = SessionId("clarify-happy")
|
||||
val orchestrator = orchestrator(listOf(withQuestions, clean))
|
||||
@@ -177,14 +177,48 @@ class ClarificationLoopTest {
|
||||
)
|
||||
assertNotNull(
|
||||
events.find { it.payload is WorkflowCompletedEvent },
|
||||
"Workflow should complete after the clarification re-run produced a clean analysis",
|
||||
"Workflow should complete once the operator answered and the run advanced",
|
||||
)
|
||||
}
|
||||
|
||||
@Test
|
||||
fun `clarification rounds are capped so a stuck stage still proceeds`(): Unit = runBlocking {
|
||||
val sessionId = SessionId("clarify-cap")
|
||||
// Always returns questions: without a cap this would loop forever.
|
||||
fun `pendingClarificationFor resolves the live request server-side, then clears once answered`():
|
||||
Unit = runBlocking {
|
||||
// Backs the REST /clarify route (#42): a headless caller has no ClarificationRequestedEvent,
|
||||
// so the server must resolve (stageId, requestId) from just the session id, and stop
|
||||
// resolving it once answered.
|
||||
val sessionId = SessionId("clarify-rest")
|
||||
val orchestrator = orchestrator(listOf(withQuestions, clean))
|
||||
val runJob = launch { orchestrator.run(sessionId, graph(), config) }
|
||||
|
||||
val pending = withTimeout(5_000) {
|
||||
var p: ClarificationRequestedEvent? = null
|
||||
while (p == null) {
|
||||
p = orchestrator.pendingClarificationFor(sessionId)
|
||||
if (p == null) yield()
|
||||
}
|
||||
p
|
||||
}
|
||||
assertEquals(analystStage, pending.stageId)
|
||||
assertEquals("Which stack?", pending.questions.single().prompt)
|
||||
|
||||
orchestrator.submitClarification(
|
||||
sessionId, pending.stageId, pending.requestId,
|
||||
listOf(ClarificationAnswer(questionId = "stack", value = "React")),
|
||||
)
|
||||
runJob.join()
|
||||
|
||||
assertEquals(
|
||||
null, orchestrator.pendingClarificationFor(sessionId),
|
||||
"Answered clarification must no longer resolve as pending",
|
||||
)
|
||||
}
|
||||
|
||||
@Test
|
||||
fun `a stage that emits questions parks once and is never re-run`(): Unit = runBlocking {
|
||||
val sessionId = SessionId("clarify-no-rerun")
|
||||
// Provider ALWAYS returns questions. Under the old re-run loop this would park up to the
|
||||
// round cap; under the advance-don't-re-run contract it parks exactly once, then proceeds.
|
||||
val orchestrator = orchestrator(listOf(withQuestions))
|
||||
val runJob = launch { orchestrator.run(sessionId, graph(), config) }
|
||||
|
||||
@@ -207,10 +241,10 @@ class ClarificationLoopTest {
|
||||
runJob.join()
|
||||
|
||||
val rounds = eventStore.read(sessionId).count { it.payload is ClarificationRequestedEvent }
|
||||
assertEquals(3, rounds, "Clarification must be capped at three rounds")
|
||||
assertEquals(1, rounds, "Questioning stage must park exactly once (no re-run)")
|
||||
assertNotNull(
|
||||
eventStore.read(sessionId).find { it.payload is WorkflowCompletedEvent },
|
||||
"Workflow should proceed once the clarification cap is reached",
|
||||
"Workflow should advance to completion after the single answered clarification",
|
||||
)
|
||||
}
|
||||
}
|
||||
|
||||
@@ -144,6 +144,42 @@ class FreestyleApprovalGateTest {
|
||||
start = analystStage,
|
||||
)
|
||||
|
||||
/** Seeds a prior architect-gate request + resolved decision (the retry/resume situation). */
|
||||
private suspend fun seedArchitectDecision(sessionId: SessionId, outcome: ApprovalOutcome) {
|
||||
val requestId = com.correx.core.events.types.ApprovalRequestId("seeded-req-${outcome.name}")
|
||||
eventStore.append(
|
||||
com.correx.testing.fixtures.EventFixtures.newEvent(
|
||||
com.correx.core.events.types.EventId("seed-req-${outcome.name}"),
|
||||
sessionId,
|
||||
ApprovalRequestedEvent(
|
||||
requestId = requestId,
|
||||
tier = Tier.T2,
|
||||
validationReportId = com.correx.core.events.types.ValidationReportId("vr"),
|
||||
riskSummaryId = null,
|
||||
sessionId = sessionId,
|
||||
stageId = architectStage,
|
||||
projectId = null,
|
||||
toolName = null,
|
||||
),
|
||||
),
|
||||
)
|
||||
eventStore.append(
|
||||
com.correx.testing.fixtures.EventFixtures.newEvent(
|
||||
com.correx.core.events.types.EventId("seed-dec-${outcome.name}"),
|
||||
sessionId,
|
||||
com.correx.core.events.events.ApprovalDecisionResolvedEvent(
|
||||
decisionId = com.correx.core.events.types.ApprovalDecisionId("seeded-dec-${outcome.name}"),
|
||||
requestId = requestId,
|
||||
outcome = outcome,
|
||||
status = ApprovalStatus.COMPLETED,
|
||||
tier = Tier.T2,
|
||||
resolutionTimestamp = Clock.System.now(),
|
||||
reason = null,
|
||||
),
|
||||
),
|
||||
)
|
||||
}
|
||||
|
||||
private val analysisSummary = """{"summary":"open questions: feasibility of X, timeline for Y"}"""
|
||||
private val executionPlanJson = """{"plan":"step 1: do A, step 2: do B"}"""
|
||||
|
||||
@@ -274,6 +310,52 @@ class FreestyleApprovalGateTest {
|
||||
runJob.join()
|
||||
}
|
||||
|
||||
@Test
|
||||
fun `runFrom architect reuses a prior APPROVED decision and does not re-prompt on grounding retry`(): Unit =
|
||||
runBlocking {
|
||||
// Return-to-architect loop (FreestyleDriver.rerunArchitect): a grounding-rejected plan re-runs
|
||||
// the architect via runFrom(architect). The operator already approved the analyst→architect
|
||||
// gate on the first pass, so the re-run must NOT re-park — the seeded APPROVED decision below
|
||||
// stands in for that first approval; the gate must reuse it.
|
||||
val sessionId = SessionId("freestyle-gate-rerun")
|
||||
seedArchitectDecision(sessionId, ApprovalOutcome.APPROVED)
|
||||
|
||||
// Provider always yields the execution plan — we enter at architect, not analyst.
|
||||
val router = object : com.correx.core.inference.InferenceRouter {
|
||||
override suspend fun route(stageId: StageId, requiredCapabilities: Set<ModelCapability>) =
|
||||
MockInferenceProvider(fixedResponse = executionPlanJson)
|
||||
}
|
||||
val orchestrator = DefaultSessionOrchestrator(
|
||||
repositories = repositories,
|
||||
engines = OrchestratorEngines(
|
||||
transitionResolver = TransitionFixtures.simpleResolver(),
|
||||
contextPackBuilder = ContextFixtures.simpleBuilder(),
|
||||
inferenceRouter = router,
|
||||
validationPipeline = ValidationPipeline(validators = listOf(cyclePolicyMissingValidator())),
|
||||
approvalEngine = DefaultApprovalEngine(),
|
||||
riskAssessor = DefaultRiskAssessor(),
|
||||
),
|
||||
retryCoordinator = DefaultRetryCoordinator(eventStore),
|
||||
artifactStore = NoopArtifactStore(),
|
||||
decisionJournalRepository = decisionJournalRepository,
|
||||
)
|
||||
val config = OrchestrationConfig(retryPolicy = RetryPolicy(maxAttempts = 1, backoffMs = 0))
|
||||
|
||||
orchestrator.runFrom(sessionId, freestyleGraph(), config, architectStage)
|
||||
|
||||
val architectRequests = eventStore.read(sessionId)
|
||||
.mapNotNull { it.payload as? ApprovalRequestedEvent }
|
||||
.count { it.stageId == architectStage && it.toolName == null }
|
||||
assertTrue(
|
||||
architectRequests == 1,
|
||||
"gate must reuse the prior approval, not re-prompt: got $architectRequests",
|
||||
)
|
||||
assertNotNull(
|
||||
eventStore.read(sessionId).find { it.payload is WorkflowCompletedEvent },
|
||||
"architect should run straight through on the reused approval",
|
||||
)
|
||||
}
|
||||
|
||||
@Test
|
||||
fun `approval resumes workflow and architect produces execution plan`(): Unit = runBlocking {
|
||||
val sessionId = SessionId("freestyle-gate-2")
|
||||
|
||||
@@ -11,6 +11,9 @@ import com.correx.core.approvals.model.DomainApprovalRequest
|
||||
import com.correx.core.artifacts.DefaultArtifactReducer
|
||||
import com.correx.core.events.EventDispatcher
|
||||
import com.correx.core.events.events.FailureTicketOpenedEvent
|
||||
import com.correx.core.events.events.StaticAnalysisCompletedEvent
|
||||
import com.correx.core.events.events.TransitionExecutedEvent
|
||||
import com.correx.core.events.events.WorkflowCompletedEvent
|
||||
import com.correx.core.events.events.WorkflowFailedEvent
|
||||
import com.correx.core.events.execution.RetryPolicy
|
||||
import com.correx.core.events.types.ProviderId
|
||||
@@ -424,6 +427,7 @@ class RecoveryRoutingTest {
|
||||
private fun buildMultiToolOrchestrator(
|
||||
staticOutput: String = "App.tsx: TS2322 type mismatch",
|
||||
provider: InferenceProvider = StageRoutingProvider(),
|
||||
staticExitCode: Int = 1,
|
||||
): Pair<DefaultSessionOrchestrator, InMemoryEventStore> {
|
||||
val eventStore = InMemoryEventStore()
|
||||
val shell = ShellTool(allowedExecutables = setOf("true"), workingDir = workspace)
|
||||
@@ -471,7 +475,7 @@ class RecoveryRoutingTest {
|
||||
workspacePolicy = WorkspacePolicy(workspace),
|
||||
// Gate names the file impl wrote, so route-to-owner can resolve the author from the manifest.
|
||||
staticAnalysisRunner = StaticAnalysisRunner { _, _ ->
|
||||
StaticAnalysisRunResult(exitCode = 1, output = staticOutput)
|
||||
StaticAnalysisRunResult(exitCode = staticExitCode, output = staticOutput)
|
||||
},
|
||||
approvalEngine = object : ApprovalEngine {
|
||||
override fun evaluate(
|
||||
@@ -498,6 +502,84 @@ class RecoveryRoutingTest {
|
||||
return orchestrator to eventStore
|
||||
}
|
||||
|
||||
@Test
|
||||
fun `non-writing review terminal records static analysis before workflow completion`(): Unit = runBlocking {
|
||||
val (orchestrator, eventStore) = buildMultiToolOrchestrator(
|
||||
staticOutput = "static floor clean",
|
||||
staticExitCode = 0,
|
||||
)
|
||||
val sessionId = SessionId("review-terminal-static-floor")
|
||||
val graph = WorkflowGraph(
|
||||
id = "freestyle-review-terminal",
|
||||
stages = mapOf(
|
||||
StageId("impl") to StageConfig(
|
||||
allowedTools = setOf("file_write"),
|
||||
writeManifest = listOf("App.tsx"),
|
||||
),
|
||||
StageId("review") to StageConfig(
|
||||
allowedTools = setOf("shell"),
|
||||
staticAnalysis = listOf("correx-static-floor -- App.tsx"),
|
||||
autoBuildGate = true,
|
||||
metadata = mapOf("role" to "reviewer"),
|
||||
),
|
||||
),
|
||||
transitions = setOf(
|
||||
TransitionEdge(TransitionId("impl->review"), StageId("impl"), StageId("review")) { true },
|
||||
TransitionEdge(TransitionId("review->done"), StageId("review"), StageId("done")) { true },
|
||||
),
|
||||
start = StageId("impl"),
|
||||
)
|
||||
|
||||
orchestrator.run(sessionId, graph, OrchestrationConfig())
|
||||
|
||||
val payloads = eventStore.read(sessionId).map { it.payload }
|
||||
val staticIndex = payloads.indexOfFirst {
|
||||
it is StaticAnalysisCompletedEvent && it.stageId == StageId("review")
|
||||
}
|
||||
val completedIndex = payloads.indexOfFirst { it is WorkflowCompletedEvent }
|
||||
assertTrue(staticIndex >= 0, "the non-writing review terminal must execute its static floor")
|
||||
assertTrue(completedIndex > staticIndex, "static analysis must be durably recorded before completion")
|
||||
}
|
||||
|
||||
@Test
|
||||
fun `review loop routes to recovery after exactly three back edges and never emits a fourth`(): Unit =
|
||||
runBlocking {
|
||||
val (orchestrator, eventStore) = buildMultiToolOrchestrator()
|
||||
val sessionId = SessionId("review-loop-convergence")
|
||||
val graph = WorkflowGraph(
|
||||
id = "review-loop-convergence",
|
||||
stages = mapOf(
|
||||
StageId("impl") to StageConfig(allowedTools = setOf("file_write")),
|
||||
StageId("review") to StageConfig(
|
||||
allowedTools = setOf("shell"),
|
||||
metadata = mapOf("role" to "reviewer"),
|
||||
),
|
||||
StageId("recovery") to StageConfig(
|
||||
allowedTools = setOf("shell", "file_write"),
|
||||
metadata = mapOf("role" to "recovery"),
|
||||
),
|
||||
),
|
||||
transitions = setOf(
|
||||
TransitionEdge(TransitionId("impl->review"), StageId("impl"), StageId("review")) { true },
|
||||
TransitionEdge(TransitionId("review->impl"), StageId("review"), StageId("impl")) { true },
|
||||
),
|
||||
start = StageId("impl"),
|
||||
)
|
||||
|
||||
val result = orchestrator.run(sessionId, graph, OrchestrationConfig())
|
||||
|
||||
assertTrue(result is WorkflowResult.Failed)
|
||||
val payloads = eventStore.read(sessionId).map { it.payload }
|
||||
val reviewBackEdges = payloads.filterIsInstance<TransitionExecutedEvent>()
|
||||
.filter { it.from == StageId("review") && it.to == StageId("impl") }
|
||||
assertEquals(3, reviewBackEdges.size, "the fourth rejected review must not enter rework")
|
||||
val ticket = payloads.filterIsInstance<FailureTicketOpenedEvent>()
|
||||
.single { it.gate == "review_loop" }
|
||||
assertEquals(StageId("review"), ticket.stageId)
|
||||
assertEquals(StageId("recovery"), ticket.routeTo)
|
||||
assertTrue(payloads.any { it is WorkflowFailedEvent })
|
||||
}
|
||||
|
||||
@Test
|
||||
fun `write-less gate failure routes the ticket to the file's author (route-to-owner)`(): Unit = runBlocking {
|
||||
val (orchestrator, eventStore) = buildMultiToolOrchestrator()
|
||||
|
||||
@@ -45,6 +45,10 @@ class LaunchRegistrationRaceTest {
|
||||
}.awaitAll()
|
||||
}
|
||||
|
||||
// computeIfAbsent returns the Job as soon as it's created; the launched body (which
|
||||
// increments launchCount) runs independently on `scope`. Join the registered jobs so the
|
||||
// increments have happened before we assert — otherwise launchCount races the launch bodies.
|
||||
registry.values.forEach { it.join() }
|
||||
assertEquals(1, launchCount.get(), "mapping lambda must execute exactly once")
|
||||
assertEquals(1, registry.size, "registry must contain exactly one entry")
|
||||
scope.cancel()
|
||||
@@ -69,6 +73,9 @@ class LaunchRegistrationRaceTest {
|
||||
}.awaitAll()
|
||||
}
|
||||
|
||||
// Join the launched bodies before asserting: computeIfAbsent returns the Job immediately,
|
||||
// but the increment runs on `scope` independently and may lag awaitAll (the 49-vs-50 flake).
|
||||
registry.values.forEach { it.join() }
|
||||
assertEquals(50, launchCount.get(), "each distinct key must execute its lambda")
|
||||
assertEquals(50, registry.size, "registry must contain 50 entries")
|
||||
scope.cancel()
|
||||
|
||||
Reference in New Issue
Block a user