fix(server): event capturing and necessary logic for smoke test.

This commit is contained in:
2026-05-17 03:17:12 +04:00
parent 0c1876a549
commit 7d46f46f01
39 changed files with 1097 additions and 67 deletions
+12
View File
@@ -15,9 +15,21 @@ dependencies {
implementation project(':core:kernel')
implementation project(':core:inference')
implementation project(':core:transitions')
implementation project(':core:context')
implementation project(':core:validation')
implementation project(':core:risk')
implementation project(':core:artifacts')
implementation project(':infrastructure')
implementation project(':infrastructure:workflow')
implementation project(':infrastructure:persistence')
implementation project(':infrastructure:inference')
implementation project(':infrastructure:inference:llama_cpp')
implementation project(':infrastructure:inference:commons')
implementation "com.github.ajalt.clikt:clikt:5.0.1"
implementation "io.ktor:ktor-client-cio:$ktor_version"
implementation "io.ktor:ktor-client-logging:$ktor_version"
implementation "io.ktor:ktor-server-core:$ktor_version"
implementation "io.ktor:ktor-server-netty:$ktor_version"
implementation "io.ktor:ktor-server-websockets:$ktor_version"
@@ -1,5 +1,101 @@
package com.correx.apps.server
import com.correx.apps.server.registry.FileSystemWorkflowRegistry
import com.correx.apps.server.registry.ProviderRegistry
import com.correx.core.approvals.domain.DefaultApprovalEngine
import com.correx.core.context.builder.DefaultContextPackBuilder
import com.correx.core.context.compression.DefaultContextCompressor
import com.correx.core.inference.DefaultInferenceRouter
import com.correx.core.inference.InferenceProjector
import com.correx.core.inference.InferenceRepository
import com.correx.core.kernel.orchestration.DefaultOrchestrationReducer
import com.correx.core.kernel.orchestration.DefaultSessionOrchestrator
import com.correx.core.kernel.orchestration.OrchestrationProjector
import com.correx.core.kernel.orchestration.OrchestrationRepository
import com.correx.core.kernel.orchestration.OrchestratorEngines
import com.correx.core.kernel.orchestration.OrchestratorRepositories
import com.correx.core.kernel.retry.DefaultRetryCoordinator
import com.correx.core.risk.DefaultRiskAssessor
import com.correx.core.sessions.DefaultSessionReducer
import com.correx.core.sessions.DefaultSessionRepository
import com.correx.core.sessions.SessionProjector
import com.correx.core.sessions.projections.replay.DefaultEventReplayer
import com.correx.core.transitions.evaluation.PromptResolver
import com.correx.core.transitions.resolution.DefaultTransitionResolver
import com.correx.core.validation.pipeline.ValidationPipeline
import com.correx.infrastructure.InfrastructureModule
import com.correx.infrastructure.inference.DefaultProviderRegistry
import com.correx.infrastructure.inference.FirstAvailableRoutingStrategy
import io.ktor.server.engine.embeddedServer
import io.ktor.server.netty.Netty
fun main() {
println("correx :: apps/server")
val eventStore = InfrastructureModule.createEventStore()
val llamaProvider = InfrastructureModule.createLlamaCppProvider(
modelId = System.getenv("CORREX_MODEL_ID") ?: "default",
modelPath = System.getenv("CORREX_MODEL_PATH") ?: "",
baseUrl = System.getenv("CORREX_LLAMA_URL") ?: "http://127.0.0.1:10000",
)
val infraRegistry = InfrastructureModule.createProviderRegistry(listOf(llamaProvider))
val artifactRepository = InfrastructureModule.createArtifactRepository(eventStore)
val sessionRepository = DefaultSessionRepository(
DefaultEventReplayer(eventStore, SessionProjector(DefaultSessionReducer()))
)
val inferenceRepository = InferenceRepository(
DefaultEventReplayer(eventStore, InferenceProjector())
)
val orchestrationRepository = OrchestrationRepository(
DefaultEventReplayer(eventStore, OrchestrationProjector(DefaultOrchestrationReducer()))
)
val inferenceRouter = DefaultInferenceRouter(infraRegistry, FirstAvailableRoutingStrategy())
val promptLoader = InfrastructureModule.createPromptLoader()
val promptResolver = PromptResolver { path -> promptLoader.load(path) }
val repositories = OrchestratorRepositories(
eventStore = eventStore,
inferenceRepository = inferenceRepository,
orchestrationRepository = orchestrationRepository,
sessionRepository = sessionRepository,
artifactRepository = artifactRepository,
)
val engines = OrchestratorEngines(
transitionResolver = DefaultTransitionResolver { condition, ctx -> condition.evaluate(ctx) },
contextPackBuilder = DefaultContextPackBuilder(DefaultContextCompressor()),
inferenceRouter = inferenceRouter,
validationPipeline = ValidationPipeline(validators = emptyList()),
approvalEngine = DefaultApprovalEngine(),
riskAssessor = DefaultRiskAssessor(),
promptResolver = promptResolver,
)
val orchestrator = DefaultSessionOrchestrator(
repositories = repositories,
engines = engines,
retryCoordinator = DefaultRetryCoordinator(eventStore),
)
val workflowRegistry = FileSystemWorkflowRegistry(InfrastructureModule.createWorkflowLoader())
val module = ServerModule(
orchestrator = orchestrator,
eventStore = eventStore,
sessionRepository = sessionRepository,
workflowRegistry = workflowRegistry,
providerRegistry = infraRegistry.asServerRegistry(),
)
embeddedServer(Netty, port = 8080) {
configureServer(module)
}.start(wait = true)
}
private fun DefaultProviderRegistry.asServerRegistry(): ProviderRegistry = object : ProviderRegistry {
override fun listAll() = this@asServerRegistry.listAll()
override suspend fun healthCheckAll() = this@asServerRegistry.healthCheckAll()
}
@@ -0,0 +1,39 @@
package com.correx.apps.server.registry
import com.correx.core.transitions.graph.WorkflowGraph
import com.correx.infrastructure.workflow.WorkflowLoader
import java.nio.file.Path
import kotlin.io.path.exists
import kotlin.io.path.listDirectoryEntries
class FileSystemWorkflowRegistry(
private val loader: WorkflowLoader,
private val workflowsDir: Path = Path.of(
System.getProperty("user.home"), ".config", "correx", "workflows"
),
) : WorkflowRegistry {
private val cache: Map<String, WorkflowGraph> by lazy { loadAll() }
override fun listAll(): List<WorkflowSummary> =
cache.values.map { WorkflowSummary(workflowId = it.id, description = it.id) }
override fun find(workflowId: String): WorkflowGraph? = cache[workflowId]
private fun loadAll(): Map<String, WorkflowGraph> {
if (!workflowsDir.exists()) return emptyMap()
return workflowsDir.listDirectoryEntries("*.toml")
.mapNotNull { file ->
runCatching { loader.load(file) }
.onFailure { logWarning(file, it) }
.getOrNull()
}
.associateBy { graph ->
graph.id.ifBlank { graph.let { it } .let { file -> file.start.value } }
}
}
private fun logWarning(file: Path, ex: Throwable) {
System.err.println("[WorkflowRegistry] Skipping $file: ${ex.message}")
}
}
@@ -3,10 +3,16 @@ package com.correx.apps.server.routes
import com.correx.apps.server.ServerModule
import com.correx.apps.server.protocol.SessionConfigDto
import com.correx.apps.server.ws.SessionStreamHandler
import com.correx.core.events.events.EventMetadata
import com.correx.core.events.events.NewEvent
import com.correx.core.events.events.WorkflowFailedEvent
import com.correx.core.events.types.EventId
import com.correx.core.events.types.SessionId
import com.correx.core.kernel.orchestration.OrchestrationConfig
import com.correx.core.utils.TypeId
import io.ktor.http.HttpStatusCode
import io.ktor.server.application.call
import io.ktor.server.application.application
import io.ktor.server.request.receive
import io.ktor.server.response.respond
import io.ktor.server.routing.Route
@@ -14,6 +20,8 @@ import io.ktor.server.routing.get
import io.ktor.server.routing.post
import io.ktor.server.routing.route
import io.ktor.server.websocket.webSocket
import kotlinx.coroutines.launch
import kotlinx.datetime.Clock
import kotlinx.serialization.Serializable
import java.util.UUID
@@ -34,67 +42,95 @@ data class StartSessionResponse(val sessionId: String)
fun Route.sessionRoutes(module: ServerModule) {
val streamHandler = SessionStreamHandler(module)
route("/sessions") {
get {
val sessions = emptyList<SessionSummaryResponse>()
call.respond(sessions)
}
post {
val body = call.receive<StartSessionRequest>()
module.workflowRegistry.find(body.workflowId)
?: return@post call.respond(HttpStatusCode.BadRequest, "Unknown workflowId: ${body.workflowId}")
val sessionId: SessionId = TypeId(UUID.randomUUID().toString())
call.respond(HttpStatusCode.Created, StartSessionResponse(sessionId.value))
}
get { call.respond(emptyList<SessionSummaryResponse>()) }
startSessionRoute(module)
route("/{id}") {
get {
val id = call.parameters["id"]
?: return@get call.respond(HttpStatusCode.BadRequest, "Missing session id")
val sessionId: SessionId = TypeId(id)
val session = runCatching { module.sessionRepository.getSession(sessionId) }.getOrNull()
?: return@get call.respond(HttpStatusCode.NotFound, "Session not found")
val response = SessionStateResponse(
sessionId = session.sessionId.value,
status = session.state.status.name,
createdAt = session.state.createdAt?.toString(),
)
call.respond(response)
}
post("/cancel") {
val id = call.parameters["id"]
?: return@post call.respond(HttpStatusCode.BadRequest, "Missing session id")
val sessionId: SessionId = TypeId(id)
module.orchestrator.cancel(sessionId)
call.respond(HttpStatusCode.OK)
}
get("/events") {
val id = call.parameters["id"]
?: return@get call.respond(HttpStatusCode.BadRequest, "Missing session id")
val from = call.request.queryParameters["from"]?.toLongOrNull() ?: 0L
val sessionId: SessionId = TypeId(id)
val events = module.eventStore.readFrom(sessionId, from)
val response = events.map { e ->
EventResponse(
eventId = e.metadata.eventId.value,
sequence = e.sequence,
sessionId = e.metadata.sessionId.value,
)
}
call.respond(response)
}
getSessionRoute(module)
cancelSessionRoute(module)
getEventsRoute(module)
webSocket("/stream") {
val id = call.parameters["id"] ?: return@webSocket
val lastEventId = call.request.queryParameters["lastEventId"]?.toLongOrNull()
val sessionId: SessionId = TypeId(id)
streamHandler.handle(this, sessionId, lastEventId)
streamHandler.handle(this, TypeId(id), lastEventId)
}
}
}
}
private fun Route.startSessionRoute(module: ServerModule) {
post {
val body = call.receive<StartSessionRequest>()
val graph = module.workflowRegistry.find(body.workflowId)
?: return@post call.respond(HttpStatusCode.BadRequest, "Unknown workflowId: ${body.workflowId}")
val sessionId: SessionId = TypeId(UUID.randomUUID().toString())
call.application.launch {
runCatching { module.orchestrator.run(sessionId, graph, OrchestrationConfig()) }
.onFailure { ex ->
System.err.println("[SessionRoutes] run failed for $sessionId: ${ex.message}")
module.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 = graph.start,
reason = ex.message ?: "unexpected orchestrator failure",
retryExhausted = false,
),
)
)
}
}
call.respond(HttpStatusCode.Accepted, StartSessionResponse(sessionId.value))
}
}
private fun Route.getSessionRoute(module: ServerModule) {
get {
val id = call.parameters["id"]
?: return@get call.respond(HttpStatusCode.BadRequest, "Missing session id")
val session = runCatching { module.sessionRepository.getSession(TypeId(id)) }.getOrNull()
?: return@get call.respond(HttpStatusCode.NotFound, "Session not found")
call.respond(
SessionStateResponse(
sessionId = session.sessionId.value,
status = session.state.status.name,
createdAt = session.state.createdAt?.toString(),
)
)
}
}
private fun Route.cancelSessionRoute(module: ServerModule) {
post("/cancel") {
val id = call.parameters["id"]
?: return@post call.respond(HttpStatusCode.BadRequest, "Missing session id")
module.orchestrator.cancel(TypeId(id))
call.respond(HttpStatusCode.OK)
}
}
private fun Route.getEventsRoute(module: ServerModule) {
get("/events") {
val id = call.parameters["id"]
?: return@get call.respond(HttpStatusCode.BadRequest, "Missing session id")
val from = call.request.queryParameters["from"]?.toLongOrNull() ?: 0L
val events = module.eventStore.readFrom(TypeId(id), from)
call.respond(
events.map { e ->
EventResponse(
eventId = e.metadata.eventId.value,
sequence = e.sequence,
sessionId = e.metadata.sessionId.value,
)
}
)
}
}
@@ -3,7 +3,16 @@ package com.correx.apps.server.ws
import com.correx.apps.server.ServerModule
import com.correx.apps.server.protocol.ClientMessage
import com.correx.apps.server.protocol.ProtocolSerializer
import com.correx.apps.server.protocol.ProviderHealthDto
import com.correx.apps.server.protocol.ServerMessage
import com.correx.core.events.events.EventMetadata
import com.correx.core.events.events.NewEvent
import com.correx.core.events.events.WorkflowFailedEvent
import com.correx.core.events.types.EventId
import com.correx.core.events.types.SessionId
import com.correx.core.inference.ProviderHealth
import com.correx.core.kernel.orchestration.OrchestrationConfig
import com.correx.core.utils.TypeId
import io.ktor.server.websocket.DefaultWebSocketServerSession
import io.ktor.websocket.Frame
import io.ktor.websocket.readText
@@ -11,12 +20,15 @@ import kotlinx.coroutines.channels.ClosedReceiveChannelException
import kotlinx.coroutines.delay
import kotlinx.coroutines.isActive
import kotlinx.coroutines.launch
import kotlinx.datetime.Clock
import java.util.*
private const val HEARTBEAT_INTERVAL_MS = 30_000L
class GlobalStreamHandler(private val module: ServerModule) {
suspend fun handle(session: DefaultWebSocketServerSession) {
System.err.println("[GlobalStream] client connected")
sendInitialSnapshot(session)
val heartbeatJob = session.launch {
@@ -32,26 +44,39 @@ class GlobalStreamHandler(private val module: ServerModule) {
if (frame is Frame.Text) {
val text = frame.readText()
runCatching { ProtocolSerializer.decodeClientMessage(text) }
.onSuccess { msg -> handleClientMessage(session, msg) }
.onSuccess { msg ->
System.err.println("[GlobalStream] recv: ${msg::class.simpleName}")
handleClientMessage(session, msg)
}
.onFailure {
System.err.println("[GlobalStream] decode error: ${it.message}")
val error = ServerMessage.ProtocolError("Unknown message: ${it.message}")
session.send(Frame.Text(ProtocolSerializer.encodeServerMessage(error)))
}
}
}
} catch (_: ClosedReceiveChannelException) {
// client disconnected
System.err.println("[GlobalStream] client disconnected")
} finally {
heartbeatJob.cancel()
}
}
private suspend fun sendInitialSnapshot(session: DefaultWebSocketServerSession) {
val providerHealth = runCatching { module.providerRegistry.healthCheckAll() }.getOrDefault(emptyMap())
providerHealth.forEach { (providerId, _) ->
val providerHealth = runCatching { module.providerRegistry.healthCheckAll() }
.onFailure { System.err.println("[GlobalStream] healthCheckAll failed: ${it.message}") }
.getOrDefault(emptyMap())
System.err.println("[GlobalStream] snapshot: ${providerHealth.size} provider(s)")
providerHealth.forEach { (providerId, health) ->
val status = when (health) {
is ProviderHealth.Healthy -> "healthy"
is ProviderHealth.Degraded -> "degraded"
is ProviderHealth.Unavailable -> "unavailable"
}
System.err.println("[GlobalStream] provider=${providerId.value} status=$status")
val msg = ServerMessage.ProviderStatusChanged(
providerId = providerId.value,
status = com.correx.apps.server.protocol.ProviderHealthDto(providerId.value, "unknown", null),
status = ProviderHealthDto(providerId.value, status, null),
)
session.send(Frame.Text(ProtocolSerializer.encodeServerMessage(msg)))
}
@@ -60,8 +85,55 @@ class GlobalStreamHandler(private val module: ServerModule) {
private suspend fun handleClientMessage(session: DefaultWebSocketServerSession, msg: ClientMessage) {
when (msg) {
is ClientMessage.Ping -> Unit
else -> {
val error = ServerMessage.ProtocolError("Unexpected message type in global stream")
is ClientMessage.StartSession -> {
val graph = module.workflowRegistry.find(msg.workflowId)
if (graph == null) {
val error = ServerMessage.ProtocolError("Unknown workflowId: ${msg.workflowId}")
session.send(Frame.Text(ProtocolSerializer.encodeServerMessage(error)))
return
}
val sessionId: SessionId = TypeId(UUID.randomUUID().toString())
System.err.println("[GlobalStream] starting session=${sessionId.value} workflow=${msg.workflowId}")
session.launch {
runCatching { module.orchestrator.run(sessionId, graph, OrchestrationConfig()) }
.onFailure { ex ->
System.err.println("[GlobalStreamHandler] run failed for $sessionId: ${ex.message}")
module.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 = graph.start,
reason = ex.message ?: "unexpected orchestrator failure",
retryExhausted = false,
),
),
)
}
}
val started = ServerMessage.SessionStarted(sessionId = sessionId, workflowId = msg.workflowId)
session.send(Frame.Text(ProtocolSerializer.encodeServerMessage(started)))
}
is ClientMessage.CancelSession -> {
module.orchestrator.cancel(msg.sessionId)
}
is ClientMessage.ResumeSession -> {
val error = ServerMessage.ProtocolError("ResumeSession not supported")
session.send(Frame.Text(ProtocolSerializer.encodeServerMessage(error)))
}
is ClientMessage.ApprovalResponse -> {
val error = ServerMessage.ProtocolError("ApprovalResponse not supported on global stream")
session.send(Frame.Text(ProtocolSerializer.encodeServerMessage(error)))
}
}
@@ -0,0 +1,13 @@
package com.correx.core.artifacts.repository
import com.correx.core.artifacts.ArtifactState
import com.correx.core.events.types.ArtifactId
import com.correx.core.events.types.SessionId
import com.correx.core.events.types.StageId
interface ArtifactRepository {
fun getBySession(sessionId: SessionId): Map<ArtifactId, ArtifactState>
fun getByStage(sessionId: SessionId, stageId: StageId): Map<ArtifactId, ArtifactState>
fun getById(sessionId: SessionId, artifactId: ArtifactId): ArtifactState?
fun getByIds(sessionId: SessionId, artifactIds: Set<ArtifactId>): Map<ArtifactId, ArtifactState>
}
@@ -4,6 +4,7 @@ import com.correx.core.approvals.domain.ApprovalEngine
import com.correx.core.context.builder.ContextPackBuilder
import com.correx.core.inference.InferenceRouter
import com.correx.core.risk.RiskAssessor
import com.correx.core.transitions.evaluation.PromptResolver
import com.correx.core.transitions.resolution.TransitionResolver
import com.correx.core.validation.pipeline.ValidationPipeline
@@ -14,4 +15,5 @@ data class OrchestratorEngines(
val validationPipeline: ValidationPipeline,
val approvalEngine: ApprovalEngine,
val riskAssessor: RiskAssessor,
val promptResolver: PromptResolver = PromptResolver { "" },
)
@@ -1,5 +1,6 @@
package com.correx.core.kernel.orchestration
import com.correx.core.artifacts.repository.ArtifactRepository
import com.correx.core.events.stores.EventStore
import com.correx.core.inference.InferenceRepository
import com.correx.core.sessions.DefaultSessionRepository
@@ -9,4 +10,5 @@ data class OrchestratorRepositories(
val inferenceRepository: InferenceRepository,
val orchestrationRepository: OrchestrationRepository,
val sessionRepository: DefaultSessionRepository,
val artifactRepository: ArtifactRepository,
)
+1
View File
@@ -7,4 +7,5 @@ plugins {
dependencies {
implementation(project(":core:events"))
implementation(project(":core:inference"))
implementation(project(":core:artifacts"))
}
@@ -0,0 +1,8 @@
package com.correx.core.transitions.conditions
import com.correx.core.transitions.evaluation.EvaluationContext
import com.correx.core.transitions.graph.TransitionCondition
object AlwaysTrue : TransitionCondition {
override fun evaluate(context: EvaluationContext) = true
}
@@ -0,0 +1,21 @@
package com.correx.core.transitions.conditions
import com.correx.core.events.types.ArtifactId
import com.correx.core.events.types.ArtifactLifecyclePhase
import com.correx.core.transitions.evaluation.EvaluationContext
import com.correx.core.transitions.graph.TransitionCondition
data class ArtifactPresent(val artifactId: ArtifactId) : TransitionCondition {
override fun evaluate(context: EvaluationContext) =
context.artifacts[artifactId] != null
}
data class ArtifactAbsent(val artifactId: ArtifactId) : TransitionCondition {
override fun evaluate(context: EvaluationContext) =
context.artifacts[artifactId] == null
}
data class ArtifactValidated(val artifactId: ArtifactId) : TransitionCondition {
override fun evaluate(context: EvaluationContext) =
context.artifacts[artifactId]?.phase == ArtifactLifecyclePhase.VALIDATED
}
@@ -0,0 +1,19 @@
package com.correx.core.transitions.conditions
import com.correx.core.transitions.evaluation.EvaluationContext
import com.correx.core.transitions.graph.TransitionCondition
data class AllOf(val conditions: List<TransitionCondition>) : TransitionCondition {
override fun evaluate(context: EvaluationContext) =
conditions.all { it.evaluate(context) }
}
data class AnyOf(val conditions: List<TransitionCondition>) : TransitionCondition {
override fun evaluate(context: EvaluationContext) =
conditions.any { it.evaluate(context) }
}
data class Not(val condition: TransitionCondition) : TransitionCondition {
override fun evaluate(context: EvaluationContext) =
!condition.evaluate(context)
}
@@ -0,0 +1,9 @@
package com.correx.core.transitions.conditions
import com.correx.core.transitions.evaluation.EvaluationContext
import com.correx.core.transitions.graph.TransitionCondition
data class VariableEquals(val key: String, val value: String) : TransitionCondition {
override fun evaluate(context: EvaluationContext) =
context.variables[key] == value
}
@@ -1,10 +1,13 @@
package com.correx.core.transitions.evaluation
import com.correx.core.artifacts.ArtifactState
import com.correx.core.events.types.ArtifactId
import com.correx.core.events.types.SessionId
import com.correx.core.events.types.StageId
data class EvaluationContext(
val sessionId: SessionId,
val currentStage: StageId,
val variables: Map<String, String>
val artifacts: Map<ArtifactId, ArtifactState> = emptyMap(),
val variables: Map<String, String> = emptyMap(),
)
@@ -0,0 +1,5 @@
package com.correx.core.transitions.evaluation
fun interface PromptResolver {
fun resolve(path: String): String
}
@@ -2,7 +2,7 @@ package com.correx.core.transitions.evaluation
import com.correx.core.transitions.graph.TransitionCondition
interface TransitionConditionEvaluator {
fun interface TransitionConditionEvaluator {
fun evaluate(
condition: TransitionCondition,
context: EvaluationContext
@@ -1,5 +1,6 @@
package com.correx.core.transitions.graph
import com.correx.core.events.types.ArtifactId
import com.correx.core.inference.GenerationConfig
import com.correx.core.inference.ModelCapability
import kotlinx.serialization.Serializable
@@ -15,4 +16,7 @@ data class StageConfig(
),
val allowedTools: Set<String> = emptySet(),
val maxRetries: Int = 3,
val produces: Set<ArtifactId> = emptySet(),
val needs: Set<ArtifactId> = emptySet(),
val metadata: Map<String, String> = emptyMap(),
)
@@ -3,6 +3,7 @@ package com.correx.core.transitions.graph
import com.correx.core.events.types.StageId
data class WorkflowGraph(
val id: String = "",
val stages: Map<StageId, StageConfig>,
val transitions: Set<TransitionEdge>,
val start: StageId,
@@ -0,0 +1,159 @@
package com.correx.core.transitions.conditions
import com.correx.core.artifacts.ArtifactState
import com.correx.core.events.types.ArtifactId
import com.correx.core.events.types.ArtifactLifecyclePhase
import com.correx.core.events.types.SessionId
import com.correx.core.events.types.StageId
import com.correx.core.transitions.evaluation.EvaluationContext
import org.junit.jupiter.api.Assertions.assertFalse
import org.junit.jupiter.api.Assertions.assertTrue
import org.junit.jupiter.api.Test
class TransitionConditionTest {
private val baseContext = EvaluationContext(
sessionId = SessionId("test-session"),
currentStage = StageId("stage-1"),
)
private fun contextWith(
artifacts: Map<ArtifactId, ArtifactState> = emptyMap(),
variables: Map<String, String> = emptyMap(),
) = baseContext.copy(artifacts = artifacts, variables = variables)
// AlwaysTrue
@Test
fun `AlwaysTrue returns true`() {
assertTrue(AlwaysTrue.evaluate(baseContext))
}
// ArtifactPresent
@Test
fun `ArtifactPresent returns true when artifact in map`() {
val id = ArtifactId("a1")
val ctx = contextWith(artifacts = mapOf(id to ArtifactState()))
assertTrue(ArtifactPresent(id).evaluate(ctx))
}
@Test
fun `ArtifactPresent returns false when artifact absent`() {
val ctx = contextWith()
assertFalse(ArtifactPresent(ArtifactId("a1")).evaluate(ctx))
}
// ArtifactAbsent
@Test
fun `ArtifactAbsent returns false when artifact in map`() {
val id = ArtifactId("a1")
val ctx = contextWith(artifacts = mapOf(id to ArtifactState()))
assertFalse(ArtifactAbsent(id).evaluate(ctx))
}
@Test
fun `ArtifactAbsent returns true when artifact absent`() {
val ctx = contextWith()
assertTrue(ArtifactAbsent(ArtifactId("a1")).evaluate(ctx))
}
// ArtifactValidated
@Test
fun `ArtifactValidated returns true when phase is VALIDATED`() {
val id = ArtifactId("a1")
val ctx = contextWith(artifacts = mapOf(id to ArtifactState(phase = ArtifactLifecyclePhase.VALIDATED)))
assertTrue(ArtifactValidated(id).evaluate(ctx))
}
@Test
fun `ArtifactValidated returns false when phase is CREATED`() {
val id = ArtifactId("a1")
val ctx = contextWith(artifacts = mapOf(id to ArtifactState(phase = ArtifactLifecyclePhase.CREATED)))
assertFalse(ArtifactValidated(id).evaluate(ctx))
}
@Test
fun `ArtifactValidated returns false when phase is REJECTED`() {
val id = ArtifactId("a1")
val ctx = contextWith(artifacts = mapOf(id to ArtifactState(phase = ArtifactLifecyclePhase.REJECTED)))
assertFalse(ArtifactValidated(id).evaluate(ctx))
}
@Test
fun `ArtifactValidated returns false when artifact absent`() {
val ctx = contextWith()
assertFalse(ArtifactValidated(ArtifactId("a1")).evaluate(ctx))
}
// VariableEquals
@Test
fun `VariableEquals returns true when key matches value`() {
val ctx = contextWith(variables = mapOf("status" to "done"))
assertTrue(VariableEquals("status", "done").evaluate(ctx))
}
@Test
fun `VariableEquals returns false when key missing`() {
val ctx = contextWith()
assertFalse(VariableEquals("status", "done").evaluate(ctx))
}
@Test
fun `VariableEquals returns false when value differs`() {
val ctx = contextWith(variables = mapOf("status" to "pending"))
assertFalse(VariableEquals("status", "done").evaluate(ctx))
}
// AllOf
@Test
fun `AllOf with empty list returns true (vacuous truth)`() {
assertTrue(AllOf(emptyList()).evaluate(baseContext))
}
@Test
fun `AllOf with all true conditions returns true`() {
assertTrue(AllOf(listOf(AlwaysTrue, AlwaysTrue)).evaluate(baseContext))
}
@Test
fun `AllOf with one false condition returns false`() {
val alwaysFalse = Not(AlwaysTrue)
assertFalse(AllOf(listOf(AlwaysTrue, alwaysFalse)).evaluate(baseContext))
}
// AnyOf
@Test
fun `AnyOf with empty list returns false (vacuous falsity)`() {
assertFalse(AnyOf(emptyList()).evaluate(baseContext))
}
@Test
fun `AnyOf with one true condition returns true`() {
val alwaysFalse = Not(AlwaysTrue)
assertTrue(AnyOf(listOf(alwaysFalse, AlwaysTrue)).evaluate(baseContext))
}
@Test
fun `AnyOf with all false conditions returns false`() {
val alwaysFalse = Not(AlwaysTrue)
assertFalse(AnyOf(listOf(alwaysFalse, alwaysFalse)).evaluate(baseContext))
}
// Not
@Test
fun `Not inverts true to false`() {
assertFalse(Not(AlwaysTrue).evaluate(baseContext))
}
@Test
fun `Not inverts false to true`() {
assertTrue(Not(Not(AlwaysTrue)).evaluate(baseContext))
}
}
+2
View File
@@ -16,6 +16,8 @@ dependencies {
implementation project(":infrastructure:persistence")
implementation project(":infrastructure:tools")
implementation project(":infrastructure:tools:filesystem")
implementation project(":infrastructure:workflow")
implementation project(":core:artifacts")
implementation "io.ktor:ktor-client-core:$ktor_version"
implementation "io.ktor:ktor-client-cio:$ktor_version"
implementation "io.ktor:ktor-client-content-negotiation:$ktor_version"
+1
View File
@@ -8,6 +8,7 @@ dependencies {
implementation "org.jetbrains.kotlinx:kotlinx-coroutines-core:$kotlinx_coroutines_version"
implementation(project(":core:events"))
implementation(project(":core:sessions"))
implementation(project(":core:artifacts"))
implementation "org.xerial:sqlite-jdbc"
testImplementation(testFixtures(project(":testing:contracts")))
testImplementation "org.junit.jupiter:junit-jupiter"
@@ -0,0 +1,96 @@
package com.correx.infrastructure.persistence.artifact
import com.correx.core.artifacts.ArtifactProjector
import com.correx.core.artifacts.ArtifactReducer
import com.correx.core.artifacts.ArtifactState
import com.correx.core.artifacts.repository.ArtifactRepository
import com.correx.core.events.events.ArtifactArchivedEvent
import com.correx.core.events.events.ArtifactCreatedEvent
import com.correx.core.events.events.ArtifactRejectedEvent
import com.correx.core.events.events.ArtifactRelationshipAddedEvent
import com.correx.core.events.events.ArtifactSupersededEvent
import com.correx.core.events.events.ArtifactValidatedEvent
import com.correx.core.events.events.ArtifactValidatingEvent
import com.correx.core.events.events.StoredEvent
import com.correx.core.events.stores.EventStore
import com.correx.core.events.types.ArtifactId
import com.correx.core.events.types.SessionId
import com.correx.core.events.types.StageId
import kotlinx.coroutines.CoroutineScope
import kotlinx.coroutines.Dispatchers
import kotlinx.coroutines.Job
import kotlinx.coroutines.SupervisorJob
import kotlinx.coroutines.flow.collect
import kotlinx.coroutines.launch
import java.util.concurrent.ConcurrentHashMap
class LiveArtifactRepository(
private val eventStore: EventStore,
private val reducer: ArtifactReducer,
private val scope: CoroutineScope = CoroutineScope(Dispatchers.IO + SupervisorJob()),
) : ArtifactRepository {
private val artifactCache = ConcurrentHashMap<SessionId, ConcurrentHashMap<ArtifactId, ArtifactState>>()
private val stageIndex = ConcurrentHashMap<SessionId, ConcurrentHashMap<ArtifactId, StageId>>()
private val subscriptions = ConcurrentHashMap<SessionId, Job>()
private val projector = ArtifactProjector(reducer)
override fun getBySession(sessionId: SessionId): Map<ArtifactId, ArtifactState> {
ensureSubscribed(sessionId)
return artifactCache[sessionId]?.toMap() ?: emptyMap()
}
override fun getByStage(sessionId: SessionId, stageId: StageId): Map<ArtifactId, ArtifactState> {
ensureSubscribed(sessionId)
val cache = artifactCache[sessionId] ?: emptyMap<ArtifactId, ArtifactState>()
val index = stageIndex[sessionId] ?: emptyMap<ArtifactId, StageId>()
return cache.filter { (artifactId, _) -> index[artifactId] == stageId }
}
override fun getById(sessionId: SessionId, artifactId: ArtifactId): ArtifactState? {
ensureSubscribed(sessionId)
return artifactCache[sessionId]?.get(artifactId)
}
override fun getByIds(sessionId: SessionId, artifactIds: Set<ArtifactId>): Map<ArtifactId, ArtifactState> {
ensureSubscribed(sessionId)
val cache = artifactCache[sessionId] ?: return emptyMap()
return artifactIds.mapNotNull { id -> cache[id]?.let { id to it } }.toMap()
}
private fun ensureSubscribed(sessionId: SessionId) {
subscriptions.computeIfAbsent(sessionId) {
scope.launch {
eventStore.subscribe(sessionId).collect { event ->
processEvent(sessionId, event)
}
}
}
}
private fun processEvent(sessionId: SessionId, event: StoredEvent) {
val artifactId = extractArtifactId(event) ?: return
val sessionArtifacts = artifactCache.computeIfAbsent(sessionId) { ConcurrentHashMap() }
val currentState = sessionArtifacts[artifactId] ?: projector.initial()
val newState = projector.apply(currentState, event)
sessionArtifacts[artifactId] = newState
val payload = event.payload
if (payload is ArtifactCreatedEvent) {
val sessionStages = stageIndex.computeIfAbsent(sessionId) { ConcurrentHashMap() }
sessionStages[artifactId] = payload.stageId
}
}
private fun extractArtifactId(event: StoredEvent): ArtifactId? =
when (val p = event.payload) {
is ArtifactCreatedEvent -> p.artifactId
is ArtifactValidatingEvent -> p.artifactId
is ArtifactValidatedEvent -> p.artifactId
is ArtifactRejectedEvent -> p.artifactId
is ArtifactSupersededEvent -> p.artifactId
is ArtifactArchivedEvent -> p.artifactId
is ArtifactRelationshipAddedEvent -> p.sourceId
else -> null
}
}
@@ -12,21 +12,37 @@ import com.correx.core.tools.contract.ToolExecutor
import com.correx.core.tools.registry.ToolRegistry
import com.correx.infrastructure.inference.DefaultProviderRegistry
import com.correx.infrastructure.inference.FirstAvailableRoutingStrategy
import com.correx.infrastructure.inference.commons.ModelDescriptor
import com.correx.infrastructure.inference.commons.ModelManager
import com.correx.infrastructure.inference.commons.ResidencyMode
import com.correx.infrastructure.inference.llama.cpp.DefaultModelManager
import com.correx.infrastructure.inference.llama.cpp.LlamaCppInferenceProvider
import com.correx.infrastructure.persistence.SqliteEventStore
import com.correx.infrastructure.tools.DefaultToolRegistry
import com.correx.infrastructure.tools.DispatchingToolExecutor
import com.correx.infrastructure.tools.SandboxedToolExecutor
import com.correx.infrastructure.tools.ToolConfig
import com.correx.infrastructure.tools.buildTools
import com.correx.infrastructure.workflow.FileSystemPromptLoader
import com.correx.infrastructure.workflow.PromptLoader
import com.correx.infrastructure.workflow.TomlWorkflowLoader
import com.correx.infrastructure.workflow.WorkflowLoader
import com.correx.infrastructure.persistence.artifact.LiveArtifactRepository
import com.correx.core.artifacts.repository.ArtifactRepository
import com.correx.core.artifacts.DefaultArtifactReducer
import io.ktor.client.HttpClient
import java.nio.file.Path
import java.sql.DriverManager
object InfrastructureModule {
fun createEventStore(dbPath: String): EventStore =
SqliteEventStore(DriverManager.getConnection("jdbc:sqlite:$dbPath"))
private val defaultDbPath: String =
"${System.getProperty("user.home")}/.config/correx/correx.db"
fun createEventStore(dbPath: String = defaultDbPath): EventStore {
val dir = java.io.File(dbPath).parentFile
if (!dir.exists()) dir.mkdirs()
return SqliteEventStore(DriverManager.getConnection("jdbc:sqlite:$dbPath"))
}
fun createModelManager(
eventStore: EventStore,
@@ -47,6 +63,31 @@ object InfrastructureModule {
strategy: RoutingStrategy = FirstAvailableRoutingStrategy(),
): InferenceRouter = DefaultInferenceRouter(registry, strategy)
fun createLlamaCppProvider(
modelId: String,
modelPath: String,
baseUrl: String = "http://127.0.0.1:10000",
residencyMode: ResidencyMode = ResidencyMode.PERSISTENT,
): LlamaCppInferenceProvider = LlamaCppInferenceProvider(
descriptor = ModelDescriptor(
modelId = modelId,
modelPath = modelPath,
residencyMode = residencyMode,
),
baseUrl = baseUrl,
)
fun createProviderRegistry(
providers: List<InferenceProvider> = emptyList(),
): DefaultProviderRegistry = DefaultProviderRegistry(providers)
fun createArtifactRepository(eventStore: EventStore): ArtifactRepository =
LiveArtifactRepository(eventStore, DefaultArtifactReducer())
fun createWorkflowLoader(): WorkflowLoader = TomlWorkflowLoader()
fun createPromptLoader(): PromptLoader = FileSystemPromptLoader()
fun createToolExecutor(
registry: ToolRegistry,
approvalEngine: ApprovalEngine,
+14
View File
@@ -0,0 +1,14 @@
plugins {
id 'java-library'
id 'org.jetbrains.kotlin.jvm'
}
dependencies {
implementation(project(":core:transitions"))
implementation(project(":core:inference"))
implementation(project(":core:events"))
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"
testImplementation "org.jetbrains.kotlin:kotlin-test-junit5"
}
@@ -0,0 +1,39 @@
package com.correx.infrastructure.workflow
import com.correx.core.events.types.ArtifactId
import com.correx.core.transitions.conditions.AllOf
import com.correx.core.transitions.conditions.AlwaysTrue
import com.correx.core.transitions.conditions.AnyOf
import com.correx.core.transitions.conditions.ArtifactAbsent
import com.correx.core.transitions.conditions.ArtifactPresent
import com.correx.core.transitions.conditions.ArtifactValidated
import com.correx.core.transitions.conditions.Not
import com.correx.core.transitions.conditions.VariableEquals
import com.correx.core.transitions.graph.TransitionCondition
fun ConditionSpec.toCondition(): TransitionCondition = when (type) {
"always_true" -> AlwaysTrue
"artifact_present" -> buildArtifactPresent()
"artifact_absent" -> buildArtifactAbsent()
"artifact_validated" -> buildArtifactValidated()
"variable_equals" -> buildVariableEquals()
"all_of" -> AllOf(conditions.map { it.toCondition() })
"any_of" -> AnyOf(conditions.map { it.toCondition() })
"not" -> Not((condition ?: error("condition required for not")).toCondition())
else -> throw WorkflowValidationException("unknown condition type: $type")
}
private fun ConditionSpec.buildArtifactPresent() =
ArtifactPresent(ArtifactId(artifactId ?: error("artifact_id required for artifact_present")))
private fun ConditionSpec.buildArtifactAbsent() =
ArtifactAbsent(ArtifactId(artifactId ?: error("artifact_id required for artifact_absent")))
private fun ConditionSpec.buildArtifactValidated() =
ArtifactValidated(ArtifactId(artifactId ?: error("artifact_id required for artifact_validated")))
private fun ConditionSpec.buildVariableEquals() =
VariableEquals(
key ?: error("key required for variable_equals"),
value ?: error("value required for variable_equals"),
)
@@ -0,0 +1,10 @@
package com.correx.infrastructure.workflow
data class ConditionSpec(
val type: String,
val artifactId: String? = null,
val key: String? = null,
val value: String? = null,
val conditions: List<ConditionSpec> = emptyList(),
val condition: ConditionSpec? = null,
)
@@ -0,0 +1,19 @@
package com.correx.infrastructure.workflow
import java.nio.file.Path
import kotlin.io.path.exists
import kotlin.io.path.readText
class FileSystemPromptLoader(
private val configDir: Path = Path.of(System.getProperty("user.home"), ".config", "correx", "prompts"),
) : PromptLoader {
override fun load(path: String): String {
val local = Path.of(path)
if (local.exists()) return local.readText()
val fromConfig = configDir.resolve(path)
if (fromConfig.exists()) return fromConfig.readText()
throw PromptNotFoundException("Prompt not found: $path (searched: $local, $fromConfig)")
}
}
@@ -0,0 +1,5 @@
package com.correx.infrastructure.workflow
interface PromptLoader {
fun load(path: String): String
}
@@ -0,0 +1,3 @@
package com.correx.infrastructure.workflow
class PromptNotFoundException(message: String) : Exception(message)
@@ -0,0 +1,130 @@
package com.correx.infrastructure.workflow
import com.correx.core.events.types.ArtifactId
import com.correx.core.events.types.StageId
import com.correx.core.events.types.TransitionId
import com.correx.core.transitions.graph.StageConfig
import com.correx.core.transitions.graph.TransitionEdge
import com.correx.core.transitions.graph.WorkflowGraph
import com.fasterxml.jackson.annotation.JsonProperty
import com.fasterxml.jackson.databind.DeserializationFeature
import com.fasterxml.jackson.dataformat.toml.TomlMapper
import com.fasterxml.jackson.module.kotlin.kotlinModule
import com.fasterxml.jackson.module.kotlin.readValue
import java.nio.file.Path
import kotlin.io.path.readText
private data class WorkflowFile(
val id: String = "",
val start: String = "",
val stages: List<StageSection> = emptyList(),
val transitions: List<TransitionSection> = emptyList(),
)
private data class StageSection(
val id: String = "",
val prompt: String? = null,
val produces: List<String> = emptyList(),
val needs: List<String> = emptyList(),
@JsonProperty("allowed_tools") val allowedTools: List<String> = emptyList(),
@JsonProperty("token_budget") val tokenBudget: Int = 4096,
@JsonProperty("max_retries") val maxRetries: Int = 3,
)
// condition fields flattened into the transition row
private data class TransitionSection(
val id: String = "",
val from: String = "",
val to: String = "",
@JsonProperty("condition_type") val conditionType: String = "",
@JsonProperty("condition_artifact_id") val conditionArtifactId: String? = null,
@JsonProperty("condition_key") val conditionKey: String? = null,
@JsonProperty("condition_value") val conditionValue: String? = null,
)
private const val TERMINAL = "done"
private val mapper = TomlMapper.builder()
.addModule(kotlinModule())
.disable(DeserializationFeature.FAIL_ON_UNKNOWN_PROPERTIES)
.build()
class TomlWorkflowLoader : WorkflowLoader {
override fun load(path: Path): WorkflowGraph {
val raw = path.readText()
val file = runCatching { mapper.readValue<WorkflowFile>(raw) }
.getOrElse { throw WorkflowValidationException("Failed to parse workflow TOML: ${it.message}") }
return file.toWorkflowGraph()
}
private fun WorkflowFile.toWorkflowGraph(): WorkflowGraph {
val startId = StageId(start)
val stageMap = stages.associate { s ->
StageId(s.id) to StageConfig(
produces = s.produces.map { ArtifactId(it) }.toSet(),
needs = s.needs.map { ArtifactId(it) }.toSet(),
allowedTools = s.allowedTools.toSet(),
tokenBudget = s.tokenBudget,
maxRetries = s.maxRetries,
metadata = buildMap { s.prompt?.let { put("prompt", it) } },
)
}
val declaredIds = stageMap.keys.map { it.value }.toSet()
validate(start, declaredIds, transitions)
val allProduced = stageMap.values.flatMap { it.produces }.toSet()
stageMap.forEach { (stageId, config) ->
config.needs.forEach { needed ->
if (needed !in allProduced) {
throw WorkflowValidationException(
"Stage '${stageId.value}' needs artifact '${needed.value}' but no stage produces it"
)
}
}
}
val transitionSet = transitions.map { t ->
TransitionEdge(
id = TransitionId(t.id),
from = StageId(t.from),
to = StageId(t.to),
condition = ConditionSpec(
type = t.conditionType,
artifactId = t.conditionArtifactId,
key = t.conditionKey,
value = t.conditionValue,
).toCondition(),
)
}.toSet()
return WorkflowGraph(id = id, stages = stageMap, transitions = transitionSet, start = startId)
}
@Suppress("ThrowsCount")
private fun validate(
start: String,
declaredStages: Set<String>,
transitions: List<TransitionSection>,
) {
if (start !in declaredStages) {
throw WorkflowValidationException("start stage '$start' is not declared in stages")
}
val seenIds = mutableSetOf<String>()
transitions.forEach { t ->
if (!seenIds.add(t.id)) {
throw WorkflowValidationException("duplicate transition id: '${t.id}'")
}
checkTransitionStage(t, "from", t.from, declaredStages)
checkTransitionStage(t, "to", t.to, declaredStages)
}
}
private fun checkTransitionStage(t: TransitionSection, field: String, stageRef: String, declared: Set<String>) {
if (stageRef !in declared && (field != "to" || stageRef != TERMINAL)) {
throw WorkflowValidationException(
"transition '${t.id}' references unknown $field-stage '$stageRef'"
)
}
}
}
@@ -0,0 +1,8 @@
package com.correx.infrastructure.workflow
import com.correx.core.transitions.graph.WorkflowGraph
import java.nio.file.Path
interface WorkflowLoader {
fun load(path: Path): WorkflowGraph
}
@@ -0,0 +1,3 @@
package com.correx.infrastructure.workflow
class WorkflowValidationException(message: String) : Exception(message)
@@ -0,0 +1,45 @@
package com.correx.infrastructure.workflow
import org.junit.jupiter.api.Test
import org.junit.jupiter.api.assertThrows
import java.nio.file.Files
import kotlin.io.path.writeText
import kotlin.test.assertEquals
import kotlin.test.assertContains
class FileSystemPromptLoaderTest {
@Test
fun `loads from absolute path when file exists`() {
val file = Files.createTempFile("prompt", ".md").also { it.writeText("hello world") }
val loader = FileSystemPromptLoader()
assertEquals("hello world", loader.load(file.toAbsolutePath().toString()))
}
@Test
fun `falls back to configDir when local path not found`() {
val configDir = Files.createTempDirectory("correx-prompts")
configDir.resolve("test.md").also { it.writeText("from config") }
val loader = FileSystemPromptLoader(configDir = configDir)
assertEquals("from config", loader.load("test.md"))
}
@Test
fun `throws PromptNotFoundException with both paths when not found`() {
val configDir = Files.createTempDirectory("correx-prompts-empty")
val loader = FileSystemPromptLoader(configDir = configDir)
val ex = assertThrows<PromptNotFoundException> { loader.load("missing.md") }
assertContains(ex.message!!, "missing.md")
assertContains(ex.message!!, configDir.toString())
}
@Test
fun `local path takes precedence over configDir`() {
val localFile = Files.createTempFile("prompt", ".md").also { it.writeText("local") }
val configDir = Files.createTempDirectory("correx-prompts")
configDir.resolve(localFile.fileName.toString()).also { it.writeText("from config") }
val loader = FileSystemPromptLoader(configDir = configDir)
assertEquals("local", loader.load(localFile.toAbsolutePath().toString()))
}
}
@@ -0,0 +1,93 @@
package com.correx.infrastructure.workflow
import org.junit.jupiter.api.Test
import org.junit.jupiter.api.assertThrows
import java.nio.file.Files
import kotlin.io.path.writeText
import kotlin.test.assertEquals
class TomlWorkflowLoaderTest {
private val loader = TomlWorkflowLoader()
private val validToml = """
id = "healthcheck"
start = "collect"
[[stages]]
id = "collect"
prompt = "prompts/collect.md"
produces = ["system_stats"]
allowed_tools = ["ShellTool"]
token_budget = 2048
max_retries = 2
[[stages]]
id = "report"
prompt = "prompts/report.md"
needs = ["system_stats"]
produces = ["report"]
token_budget = 2048
[[transitions]]
id = "collect-to-report"
from = "collect"
to = "report"
condition_type = "artifact_present"
condition_artifact_id = "system_stats"
[[transitions]]
id = "report-to-done"
from = "report"
to = "done"
condition_type = "always_true"
""".trimIndent()
@Test
fun `valid toml produces correct WorkflowGraph`() {
val path = Files.createTempFile("workflow", ".toml").also { it.writeText(validToml) }
val graph = loader.load(path)
assertEquals("collect", graph.start.value)
assertEquals(2, graph.stages.size)
assertEquals(2, graph.transitions.size)
val collectStage = graph.stages[graph.start]!!
assertEquals(setOf("system_stats"), collectStage.produces.map { it.value }.toSet())
assertEquals("prompts/collect.md", collectStage.metadata["prompt"])
val reportStage = graph.stages.values.first { it.produces.any { a -> a.value == "report" } }
assertEquals(setOf("system_stats"), reportStage.needs.map { it.value }.toSet())
}
@Test
fun `missing stage reference throws WorkflowValidationException`() {
val bad = validToml.replace("from = \"collect\"", "from = \"nonexistent\"")
val path = Files.createTempFile("workflow", ".toml").also { it.writeText(bad) }
assertThrows<WorkflowValidationException> { loader.load(path) }
}
@Test
fun `duplicate transition ids throw WorkflowValidationException`() {
val bad = validToml.replace("id = \"report-to-done\"", "id = \"collect-to-report\"")
val path = Files.createTempFile("workflow", ".toml").also { it.writeText(bad) }
assertThrows<WorkflowValidationException> { loader.load(path) }
}
@Test
fun `unknown start stage throws WorkflowValidationException`() {
val bad = validToml.replace("start = \"collect\"", "start = \"missing\"")
val path = Files.createTempFile("workflow", ".toml").also { it.writeText(bad) }
assertThrows<WorkflowValidationException> { loader.load(path) }
}
@Test
fun `unsatisfied needs throws WorkflowValidationException`() {
val bad = validToml.replace(
"needs = [\"system_stats\"]",
"needs = [\"nonexistent_artifact\"]",
)
val path = Files.createTempFile("workflow", ".toml").also { it.writeText(bad) }
assertThrows<WorkflowValidationException> { loader.load(path) }
}
}
+4
View File
@@ -0,0 +1,4 @@
# healthcheck_execute.md
Run `scripts/healthcheck.sh` and report the results.
If the script fails, report the error and exit code.
+10
View File
@@ -0,0 +1,10 @@
# healthcheck_write.md
Write a bash script to `scripts/healthcheck.sh` that reports:
- RAM usage %
- CPU usage %
- GPU usage % and VRAM usage % if an NVIDIA or AMD GPU is detected
Use standard tools only (free, top/vmstat, nvidia-smi, rocm-smi).
If no GPU is detected, skip that section gracefully.
Make the script executable.
+1
View File
@@ -40,6 +40,7 @@ include ':infrastructure:tools:filesystem'
include ':infrastructure:security'
include ':infrastructure:scheduler'
include ':infrastructure:telemetry'
include ':infrastructure:workflow'
include ':interfaces:api'
include ':interfaces:cli'
+1
View File
@@ -15,6 +15,7 @@ dependencies {
testImplementation(project(":core:kernel"))
testImplementation(project(":core:risk"))
testImplementation(project(":infrastructure:persistence"))
testImplementation(project(":core:artifacts"))
testImplementation(project(":testing:fixtures"))
testImplementation(project(":testing:kernel"))
testImplementation(project(":testing:contracts"))
@@ -35,7 +35,9 @@ import com.correx.core.transitions.graph.StageConfig
import com.correx.core.transitions.graph.WorkflowGraph
import com.correx.core.validation.approval.ApprovalTrigger
import com.correx.core.validation.pipeline.ValidationPipeline
import com.correx.core.artifacts.DefaultArtifactReducer
import com.correx.infrastructure.persistence.InMemoryEventStore
import com.correx.infrastructure.persistence.artifact.LiveArtifactRepository
import com.correx.testing.fixtures.InferenceFixtures
import com.correx.testing.fixtures.context.ContextFixtures
import com.correx.testing.fixtures.cyclePolicyMissingValidator
@@ -82,6 +84,7 @@ class SessionOrchestratorIntegrationTest {
inferenceRepository = inferenceRepository,
orchestrationRepository = orchestrationRepository,
sessionRepository = sessionRepository,
artifactRepository = LiveArtifactRepository(eventStore, DefaultArtifactReducer()),
)
private val engines = OrchestratorEngines(