feat: wire real AgentOS execution

This commit is contained in:
-Puter
2026-07-28 16:52:49 +05:30
parent 4d9f6da41b
commit 4edd456b5b
17 changed files with 1090 additions and 308 deletions

View File

@@ -17,7 +17,9 @@ DAEMON_NAME=Local MacBook
DAEMON_VERSION=0.0.0 DAEMON_VERSION=0.0.0
DAEMON_HEARTBEAT_MS=15000 DAEMON_HEARTBEAT_MS=15000
DAEMON_COMMAND_LEASE_MS=60000 DAEMON_COMMAND_LEASE_MS=60000
# RIVET_ENDPOINT=http://localhost:6420 RIVET_ENDPOINT=http://localhost:6420
RIVET_PUBLIC_ENDPOINT=http://localhost:6420
RIVET_WORKSPACE_TOKEN=replace-with-a-long-random-workspace-token
# Flue persistence adapter # Flue persistence adapter
FLUE_DB_TOKEN=replace-with-a-long-random-token FLUE_DB_TOKEN=replace-with-a-long-random-token

View File

@@ -5,8 +5,10 @@ import {
} from "@code/ui/components/ai-elements/conversation"; } from "@code/ui/components/ai-elements/conversation";
import { Button } from "@code/ui/components/button"; import { Button } from "@code/ui/components/button";
import { import {
AlertTriangle,
ChevronRight, ChevronRight,
Check, Check,
FileCode2,
FolderGit2, FolderGit2,
Hammer, Hammer,
LoaderCircle, LoaderCircle,
@@ -15,6 +17,7 @@ import {
ImagePlus, ImagePlus,
Play, Play,
RotateCcw, RotateCcw,
ScrollText,
Send, Send,
Sparkles, Sparkles,
Settings, Settings,
@@ -36,6 +39,28 @@ import {
type SliceWork = NonNullable<ReturnType<typeof useSliceOne>["works"]>[number]; type SliceWork = NonNullable<ReturnType<typeof useSliceOne>["works"]>[number];
const EMPTY_WORKS: readonly SliceWork[] = []; const EMPTY_WORKS: readonly SliceWork[] = [];
type SliceArtifact = NonNullable<
NonNullable<SliceWork["runs"][number]["artifacts"]>[number]
>;
interface ArtifactMetadata {
readonly changedFiles?: readonly string[];
}
const parseArtifactMetadata = (artifact: SliceArtifact): ArtifactMetadata => {
if (!artifact.metadataJson) {
return {};
}
try {
return JSON.parse(artifact.metadataJson) as ArtifactMetadata;
} catch {
return {};
}
};
const changedFilesFor = (artifact: SliceArtifact): readonly string[] =>
parseArtifactMetadata(artifact).changedFiles ?? [];
interface WorkCardProps { interface WorkCardProps {
readonly onSourceSelect: (rawText: string) => void; readonly onSourceSelect: (rawText: string) => void;
readonly work: SliceWork; readonly work: SliceWork;
@@ -96,6 +121,7 @@ const starterDesign = (work: SliceWork) => ({
const WorkCard = ({ onSourceSelect, work, slice }: WorkCardProps) => { const WorkCard = ({ onSourceSelect, work, slice }: WorkCardProps) => {
const [sourcesOpen, setSourcesOpen] = useState(false); const [sourcesOpen, setSourcesOpen] = useState(false);
const [expanded, setExpanded] = useState(false); const [expanded, setExpanded] = useState(false);
const [logsOpen, setLogsOpen] = useState(false);
const sources = work.signals.flatMap((signal) => signal.sources); const sources = work.signals.flatMap((signal) => signal.sources);
const { definition } = work; const { definition } = work;
const { design } = work; const { design } = work;
@@ -258,6 +284,22 @@ const WorkCard = ({ onSourceSelect, work, slice }: WorkCardProps) => {
<p className="font-semibold uppercase tracking-wide text-[#65713a]"> <p className="font-semibold uppercase tracking-wide text-[#65713a]">
Build Build
</p> </p>
{slice.operationError ? (
<p className="mt-2 flex items-start gap-1.5 border border-red-300 bg-red-50 px-2 py-1.5 text-red-800">
<AlertTriangle className="mt-0.5 size-3.5 shrink-0" />
<span className="min-w-0 flex-1 leading-5">
{slice.operationError.message}
</span>
<button
aria-label="Dismiss error"
className="shrink-0"
onClick={() => slice.clearOperationError()}
type="button"
>
<X className="size-3.5" />
</button>
</p>
) : null}
{latestRun ? ( {latestRun ? (
<div className="mt-1 space-y-2 text-[#626057]"> <div className="mt-1 space-y-2 text-[#626057]">
<p className="leading-5"> <p className="leading-5">
@@ -269,31 +311,75 @@ const WorkCard = ({ onSourceSelect, work, slice }: WorkCardProps) => {
latestRun.terminalClassification ?? latestRun.terminalClassification ??
"activity is still arriving"} "activity is still arriving"}
</p> </p>
{latestRun.attemptEvents?.slice(-5).map((item) => (
<p
className="border-l-2 border-[#b8c760] pl-2"
key={item._id}
>
{item.message}
</p>
))}
{latestRun.baseRevision ? ( {latestRun.baseRevision ? (
<p className="font-mono text-[10px] text-[#747168]"> <p className="font-mono text-[10px] text-[#747168]">
{latestRun.baseRevision.slice(0, 8)} {" "} {latestRun.baseRevision.slice(0, 8)} {" "}
{latestRun.candidateRevision?.slice(0, 8) ?? "working"} {latestRun.candidateRevision?.slice(0, 8) ?? "working"}
</p> </p>
) : null} ) : null}
{latestRun.artifacts?.map((artifact) => ( {latestRun.artifacts?.map((artifact) => {
<a const changedFiles = changedFilesFor(artifact);
className="block font-medium underline" return (
href={artifact.uri} <div key={artifact._id} className="space-y-1">
key={artifact._id} {artifact.uri ? (
rel="noreferrer" <a
target="_blank" className="block font-medium underline"
> href={artifact.uri}
{artifact.title} rel="noreferrer"
</a> target="_blank"
))} >
{artifact.title}
</a>
) : (
<p className="font-medium">{artifact.title}</p>
)}
{changedFiles.length > 0 ? (
<ul className="space-y-0.5">
{changedFiles.map((file) => (
<li
className="flex items-start gap-1.5 font-mono text-[11px] leading-5 text-[#747168]"
key={file}
>
<FileCode2 className="mt-0.5 size-3 shrink-0 text-[#9a985f]" />
<span className="min-w-0 break-all">{file}</span>
</li>
))}
</ul>
) : null}
</div>
);
})}
{latestRun.attemptEvents &&
latestRun.attemptEvents.length > 0 ? (
<div className="space-y-1">
{(logsOpen
? latestRun.attemptEvents
: latestRun.attemptEvents.slice(-3)
).map((item) => (
<p
className="border-l-2 border-[#b8c760] pl-2 leading-5"
key={item._id}
>
{item.message}
</p>
))}
{latestRun.attemptEvents.length > 3 ? (
<button
className="flex items-center gap-1 font-medium text-[#65713a] hover:underline"
onClick={() => setLogsOpen((open) => !open)}
type="button"
>
<ScrollText className="size-3.5" />
{logsOpen
? "Show recent activity"
: `Show full activity log (${latestRun.attemptEvents.length})`}
<ChevronRight
className={`size-3.5 transition-transform ${logsOpen ? "rotate-90" : ""}`}
/>
</button>
) : null}
</div>
) : null}
</div> </div>
) : ( ) : (
<p className="mt-1 text-[#747168]">No implementation Run yet.</p> <p className="mt-1 text-[#747168]">No implementation Run yet.</p>
@@ -303,9 +389,7 @@ const WorkCard = ({ onSourceSelect, work, slice }: WorkCardProps) => {
<Button <Button
size="sm" size="sm"
disabled={!slice.projectGitConnection} disabled={!slice.projectGitConnection}
onClick={() => onClick={() => void slice.startExecution(work._id)}
void slice.startExecution(work._id, design?.slices?.[0]?.id)
}
> >
<Play className="size-3.5" /> Run <Play className="size-3.5" /> Run
</Button> </Button>
@@ -315,11 +399,7 @@ const WorkCard = ({ onSourceSelect, work, slice }: WorkCardProps) => {
size="sm" size="sm"
variant="outline" variant="outline"
onClick={() => onClick={() =>
void slice.startSimulation( void slice.startSimulation(work._id, "success")
work._id,
"success",
design?.slices?.[0]?.id
)
} }
> >
<Play className="size-3.5" /> Simulate <Play className="size-3.5" /> Simulate
@@ -338,7 +418,8 @@ const WorkCard = ({ onSourceSelect, work, slice }: WorkCardProps) => {
<X className="size-3.5" /> Cancel <X className="size-3.5" /> Cancel
</Button> </Button>
) : null} ) : null}
{latestRun?.status === "terminal" && {latestRun?.executionKind !== "real" &&
latestRun?.status === "terminal" &&
latestRun.terminalClassification === "RetryableFailure" ? ( latestRun.terminalClassification === "RetryableFailure" ? (
<Button <Button
size="sm" size="sm"
@@ -428,6 +509,7 @@ const ConnectProject = ({ slice }: { slice: SliceOneState }) => (
</main> </main>
); );
// oxlint-disable-next-line complexity -- this page coordinates the existing mobile shell without owning domain logic.
export const SliceOnePage = () => { export const SliceOnePage = () => {
const slice = useSliceOne(); const slice = useSliceOne();
const viewportStyle = useVisualViewportStyle(); const viewportStyle = useVisualViewportStyle();
@@ -553,6 +635,22 @@ export const SliceOnePage = () => {
? `${slice.projectGitConnection.provider} · ${slice.projectGitConnection.serverUrl}` ? `${slice.projectGitConnection.provider} · ${slice.projectGitConnection.serverUrl}`
: "No Git credentials attached"} : "No Git credentials attached"}
</p> </p>
{slice.operationError ? (
<p className="mt-2 flex items-start gap-1.5 border border-red-300 bg-red-50 px-2 py-1.5 text-xs text-red-800">
<AlertTriangle className="mt-0.5 size-3.5 shrink-0" />
<span className="min-w-0 flex-1 leading-5">
{slice.operationError.message}
</span>
<button
aria-label="Dismiss error"
className="shrink-0"
onClick={() => slice.clearOperationError()}
type="button"
>
<X className="size-3.5" />
</button>
</p>
) : null}
<div className="mt-4 flex gap-2"> <div className="mt-4 flex gap-2">
<Button <Button
size="sm" size="sm"

View File

@@ -40,7 +40,9 @@ interface WorkRecord {
_id: string; _id: string;
kind: string; kind: string;
message: string; message: string;
metadataJson: string;
occurredAt: number; occurredAt: number;
sequence: number;
}[]; }[];
readonly baseRevision?: string; readonly baseRevision?: string;
readonly candidateRevision?: string; readonly candidateRevision?: string;
@@ -179,6 +181,20 @@ export const useSliceOne = () => {
const [repository, setRepository] = useState(""); const [repository, setRepository] = useState("");
const [pending, setPending] = useState(false); const [pending, setPending] = useState(false);
const [error, setError] = useState<Error>(); const [error, setError] = useState<Error>();
const [operationError, setOperationError] = useState<Error>();
const captureOperationError =
<TArgs extends unknown[], TResult>(
operation: (...args: TArgs) => Promise<TResult>
) =>
async (...args: TArgs): Promise<TResult> => {
setOperationError(undefined);
try {
return await operation(...args);
} catch (caughtError) {
setOperationError(toError(caughtError));
throw caughtError;
}
};
const selectedProjectStillExists = projects?.some( const selectedProjectStillExists = projects?.some(
(project) => project.id === (selectedProjectId as unknown as string) (project) => project.id === (selectedProjectId as unknown as string)
); );
@@ -262,15 +278,18 @@ export const useSliceOne = () => {
| "permanent-failure" | "permanent-failure"
| "cancelled", | "cancelled",
sliceId?: string sliceId?: string
) => startSimulationMutation({ scenario, sliceId, workId }); ) =>
captureOperationError(() =>
startSimulationMutation({ scenario, sliceId, workId })
)();
const cancelSimulation = (runId: Id<"workRuns">) => const cancelSimulation = (runId: Id<"workRuns">) =>
cancelSimulationMutation({ runId }); captureOperationError(() => cancelSimulationMutation({ runId }))();
const retrySimulation = (runId: Id<"workRuns">) => const retrySimulation = (runId: Id<"workRuns">) =>
retrySimulationMutation({ runId }); captureOperationError(() => retrySimulationMutation({ runId }))();
const startExecution = (workId: Id<"works">, sliceId?: string) => const startExecution = (workId: Id<"works">, sliceId?: string) =>
startExecutionMutation({ sliceId, workId }); captureOperationError(() => startExecutionMutation({ sliceId, workId }))();
const cancelExecution = (runId: Id<"workRuns">) => const cancelExecution = (runId: Id<"workRuns">) =>
cancelExecutionMutation({ runId }); captureOperationError(() => cancelExecutionMutation({ runId }))();
const attachConnection = async (connectionId: Id<"gitConnections">) => { const attachConnection = async (connectionId: Id<"gitConnections">) => {
if (!activeProjectId) { if (!activeProjectId) {
throw new Error("Select a project first"); throw new Error("Select a project first");
@@ -280,18 +299,16 @@ export const useSliceOne = () => {
projectId: activeProjectId, projectId: activeProjectId,
}); });
}; };
const connectGitea = async (input: { const connectGitea = captureOperationError(
serverUrl: string; async (input: { serverUrl: string; token: string; username?: string }) => {
token: string; const result = await connectGiteaAction(input);
username?: string; await attachConnection(result.connectionId);
}) => { }
const result = await connectGiteaAction(input); );
await attachConnection(result.connectionId); const connectLinkedGithub = captureOperationError(async () => {
};
const connectLinkedGithub = async () => {
const result = await connectGithubAction({}); const result = await connectGithubAction({});
await attachConnection(result.connectionId); await attachConnection(result.connectionId);
}; });
return { return {
agent, agent,
@@ -300,11 +317,13 @@ export const useSliceOne = () => {
authorizeGithub, authorizeGithub,
cancelExecution, cancelExecution,
cancelSimulation, cancelSimulation,
clearOperationError: () => setOperationError(undefined),
connectGitea, connectGitea,
connectLinkedGithub, connectLinkedGithub,
connectRepository, connectRepository,
error, error,
gitConnections, gitConnections,
operationError,
pending, pending,
projectGitConnection, projectGitConnection,
projects, projects,

View File

@@ -155,7 +155,6 @@
"version": "0.0.0", "version": "0.0.0",
"dependencies": { "dependencies": {
"@agentos-software/codex": "0.2.7", "@agentos-software/codex": "0.2.7",
"@agentos-software/codex-cli": "0.3.4",
"@agentos-software/git": "0.3.3", "@agentos-software/git": "0.3.3",
"@rivet-dev/agentos": "catalog:", "@rivet-dev/agentos": "catalog:",
"effect": "catalog:", "effect": "catalog:",

View File

@@ -44,21 +44,22 @@ AGENT_MODEL_CONTEXT_WINDOW=262000
AGENT_MODEL_MAX_TOKENS=131072 AGENT_MODEL_MAX_TOKENS=131072
# --------------------------------------------------------------------------- # ---------------------------------------------------------------------------
# 4. AgentOS / RivetKit — OPTIONAL (defaults shown) # 4. AgentOS / Rivet Engine — REQUIRED for the execution runner
# registry.start() in the daemon boots an in-process RivetKit engine # The agent service and runner connect through the public engine endpoint.
# (envoy mode) backed by a native Rust sidecar. createClient() connects # RIVET_WORKSPACE_TOKEN authenticates every workspace actor connection.
# back to RIVET_ENDPOINT. No separate Rivet Engine process is required # RIVET_ENVOY_VERSION must change for each runner deployment.
# for single-node operation. Leave RIVET_ENDPOINT unset to use the
# library default (http://localhost:6420).
# --------------------------------------------------------------------------- # ---------------------------------------------------------------------------
#RIVET_ENDPOINT=http://localhost:6420 RIVET_ENDPOINT=https://default:@rivet.example.com
RIVET_PUBLIC_ENDPOINT=https://default@rivet.example.com
RIVET_ENVOY_VERSION=1
RIVET_WORKSPACE_TOKEN=replace-with-a-long-random-workspace-token
# --------------------------------------------------------------------------- # ---------------------------------------------------------------------------
# 5. Zopu agent service (Flue) # 5. Zopu agent service (Flue)
# FLUE_DB_TOKEN authenticates the Flue persistence adapter. # FLUE_DB_TOKEN authenticates the Flue persistence adapter.
# zopu-agent.service pins the Flue Node server to port 3583. # zopu-agent.service pins the Flue Node server to port 3583.
# --------------------------------------------------------------------------- # ---------------------------------------------------------------------------
FLUE_DB_TOKEN=replace-with-long-random-token FLUE_DB_TOKEN=replace-with-long-random-token
AGENT_BACKEND_URL=https://zopu-agent.example.com
# --------------------------------------------------------------------------- # ---------------------------------------------------------------------------
# 6. Daemon identity # 6. Daemon identity

View File

@@ -4,8 +4,19 @@ services:
context: ../../.. context: ../../..
dockerfile: deploy/zopu-runtime/runner/Dockerfile dockerfile: deploy/zopu-runtime/runner/Dockerfile
environment: environment:
AGENT_MODEL_API: ${AGENT_MODEL_API}
AGENT_MODEL_API_KEY: ${AGENT_MODEL_API_KEY}
AGENT_MODEL_BASE_URL: ${AGENT_MODEL_BASE_URL}
AGENT_MODEL_CONTEXT_WINDOW: ${AGENT_MODEL_CONTEXT_WINDOW}
AGENT_MODEL_MAX_TOKENS: ${AGENT_MODEL_MAX_TOKENS}
AGENT_MODEL_NAME: ${AGENT_MODEL_NAME}
AGENT_MODEL_PROVIDER: ${AGENT_MODEL_PROVIDER}
CONVEX_URL: ${CONVEX_URL}
DAEMON_ID: zopu-agentos-runner
FLUE_DB_TOKEN: ${FLUE_DB_TOKEN}
RIVET_ENDPOINT: ${RIVET_ENDPOINT} RIVET_ENDPOINT: ${RIVET_ENDPOINT}
RIVET_PUBLIC_ENDPOINT: ${RIVET_PUBLIC_ENDPOINT} RIVET_PUBLIC_ENDPOINT: ${RIVET_PUBLIC_ENDPOINT}
RIVET_ENVOY_VERSION: ${RIVET_ENVOY_VERSION} RIVET_ENVOY_VERSION: ${RIVET_ENVOY_VERSION}
RIVET_WORKSPACE_TOKEN: ${RIVET_WORKSPACE_TOKEN}
RIVET_POOL: default RIVET_POOL: default
restart: unless-stopped restart: unless-stopped

View File

@@ -1,4 +1,5 @@
import { parseAgentEnv } from "@code/env/agent"; import { parseAgentEnv } from "@code/env/agent";
import { WorkAttemptExecutionError } from "@code/primitives/execution-runtime";
import { registerProvider } from "@flue/runtime"; import { registerProvider } from "@flue/runtime";
import type { Fetchable } from "@flue/runtime/routing"; import type { Fetchable } from "@flue/runtime/routing";
import { flue } from "@flue/runtime/routing"; import { flue } from "@flue/runtime/routing";
@@ -42,8 +43,23 @@ app.post("/internal/work-attempts/execute", async (context) => {
try { try {
return context.json(await executeAgentOsAttempt(await context.req.json())); return context.json(await executeAgentOsAttempt(await context.req.json()));
} catch (error) { } catch (error) {
const failure =
error instanceof WorkAttemptExecutionError
? error
: new WorkAttemptExecutionError({
message:
error instanceof Error ? error.message : "Execution failed",
reason: "HarnessFailed",
retryable: false,
});
return context.json( return context.json(
{ error: error instanceof Error ? error.message : "Execution failed" }, {
error: {
message: failure.message,
reason: failure.reason,
retryable: failure.retryable,
},
},
500 500
); );
} }
@@ -55,7 +71,11 @@ app.post("/internal/work-attempts/:workspaceKey/cancel", async (context) => {
) { ) {
return context.json({ error: "Unauthorized" }, 401); return context.json({ error: "Unauthorized" }, 401);
} }
await cancelAgentOsAttempt(context.req.param("workspaceKey")); const body = (await context.req.json()) as { attemptId?: unknown };
if (typeof body.attemptId !== "string" || body.attemptId.length === 0) {
return context.json({ error: "attemptId is required" }, 400);
}
await cancelAgentOsAttempt(context.req.param("workspaceKey"), body.attemptId);
return context.json({ cancelled: true }); return context.json({ cancelled: true });
}); });

View File

@@ -7,13 +7,24 @@ import {
codexSessionEnv, codexSessionEnv,
makeCodexAgentOsConfig, makeCodexAgentOsConfig,
} from "../../../primitives/src/agent-os"; } from "../../../primitives/src/agent-os";
import { decodeWorkAttemptExecutionInput } from "../../../primitives/src/execution-runtime"; import {
decodeWorkAttemptExecutionInput,
WorkAttemptExecutionError,
} from "../../../primitives/src/execution-runtime";
import type { import type {
ExecutionEvent, ExecutionEvent,
WorkAttemptExecutionResult, WorkAttemptExecutionResult,
} from "../../../primitives/src/execution-runtime"; } from "../../../primitives/src/execution-runtime";
const workspace = agentOS(makeCodexAgentOsConfig() as never); const codexConfig = makeCodexAgentOsConfig();
const workspace = agentOS<undefined, { token: string }>({
onBeforeConnect: (_context, params) => {
if (params.token !== process.env.RIVET_WORKSPACE_TOKEN) {
throw new Error("Unauthorized workspace connection");
}
},
software: codexConfig.software,
});
export const runtimeRegistry = setup({ use: { workspace } }); export const runtimeRegistry = setup({ use: { workspace } });
@@ -43,6 +54,32 @@ const requireSuccess = (
return result.stdout.trim(); return result.stdout.trim();
}; };
const executionError = (
message: string,
reason: WorkAttemptExecutionError["reason"],
retryable: boolean
) => new WorkAttemptExecutionError({ message, reason, retryable });
const classifyRuntimeFailure = (cause: unknown): WorkAttemptExecutionError => {
if (cause instanceof WorkAttemptExecutionError) {
return cause;
}
const message = cause instanceof Error ? cause.message : "Execution failed";
if (/auth|credential|401|403/iu.test(message)) {
return executionError(message, "Authentication", false);
}
if (/clone|checkout|repository|git /iu.test(message)) {
return executionError(message, "RepositoryFailed", false);
}
if (/timeout|timed out/iu.test(message)) {
return executionError(message, "Timeout", true);
}
if (/connect|network|unavailable|502|503|504/iu.test(message)) {
return executionError(message, "ProviderUnavailable", true);
}
return executionError(message, "HarnessFailed", false);
};
const gitAuthEnv = (username: string, credential: string) => ({ const gitAuthEnv = (username: string, credential: string) => ({
GIT_CONFIG_COUNT: "1", GIT_CONFIG_COUNT: "1",
GIT_CONFIG_KEY_0: "http.extraHeader", GIT_CONFIG_KEY_0: "http.extraHeader",
@@ -52,162 +89,189 @@ const gitAuthEnv = (username: string, credential: string) => ({
export const executeAgentOsAttempt = async ( export const executeAgentOsAttempt = async (
rawInput: unknown rawInput: unknown
): Promise<WorkAttemptExecutionResult> => { ): Promise<WorkAttemptExecutionResult> => {
const input = await Effect.runPromise( try {
decodeWorkAttemptExecutionInput(rawInput) const input = await Effect.runPromise(
); decodeWorkAttemptExecutionInput(rawInput)
const env = parseAgentEnv(process.env);
const client = createClient<typeof runtimeRegistry>({
endpoint: env.RIVET_PUBLIC_ENDPOINT ?? env.RIVET_ENDPOINT,
});
const vm = client.workspace.getOrCreate([input.workspaceKey]);
const events: ExecutionEvent[] = [
event(0, "runtime.preparing", "AgentOS workspace selected", {
workspaceKey: input.workspaceKey,
}),
];
const authEnv = gitAuthEnv(
input.auth.username ??
(input.auth.provider === "github" ? "x-access-token" : "git"),
input.auth.credential
);
await vm.mkdir("/workspace", { recursive: true });
if (!(await vm.exists("/workspace/repository/.git"))) {
events.push(event(1, "repository.cloning", "Cloning project repository"));
const clone = await vm.execArgv(
"git",
["clone", input.repositoryUrl, "/workspace/repository"],
{
cwd: "/workspace",
env: authEnv,
}
); );
requireSuccess(clone, "Repository clone"); const env = parseAgentEnv(process.env);
} const client = createClient<typeof runtimeRegistry>({
disableMetadataLookup: true,
const checkout = await vm.exec( endpoint: env.RIVET_PUBLIC_ENDPOINT ?? env.RIVET_ENDPOINT,
`git fetch origin ${shellQuote(input.baseBranch)} && git checkout -B ${shellQuote(`zopu/${input.runId}`)} ${shellQuote(`origin/${input.baseBranch}`)}`, });
{ cwd: "/workspace/repository", env: authEnv } const vm = client.workspace.getOrCreate([input.workspaceKey], {
); params: { token: env.RIVET_WORKSPACE_TOKEN },
requireSuccess(checkout, "Repository checkout"); });
const baseRevision = requireSuccess( const events: ExecutionEvent[] = [
await vm.execArgv("git", ["rev-parse", "HEAD"], { event(0, "runtime.preparing", "AgentOS workspace selected", {
cwd: "/workspace/repository", workspaceKey: input.workspaceKey,
}), }),
"Base revision lookup" ];
); const authEnv = gitAuthEnv(
events.push( input.auth.username ??
event(2, "repository.ready", "Repository checkout is ready", { (input.auth.provider === "github" ? "x-access-token" : "git"),
baseRevision, input.auth.credential
})
);
const sessionId = `codex-${input.attemptId}`;
await vm.openSession({
additionalInstructions:
"Work only inside /workspace/repository. Do not reveal credentials. Make the requested change and run focused verification. Do not push or open a pull request.",
agent: "codex",
cwd: "/workspace/repository",
env: codexSessionEnv({
apiKey: env.AGENT_MODEL_API_KEY,
baseUrl: env.AGENT_MODEL_BASE_URL,
model: env.AGENT_MODEL_NAME,
}),
permissionPolicy: "allow_all",
sessionId,
});
events.push(
event(3, "harness.started", "Codex implementation session started")
);
const promptResult = await vm.prompt({
content: [{ text: input.prompt, type: "text" }],
idempotencyKey: input.attemptId,
sessionId,
});
events.push(
event(4, "harness.progress", "Codex implementation turn completed", {
stopReason: String(promptResult.stopReason),
})
);
const status = requireSuccess(
await vm.execArgv("git", ["status", "--porcelain"], {
cwd: "/workspace/repository",
}),
"Changed file lookup"
);
const diff = requireSuccess(
await vm.execArgv("git", ["diff", "--binary", "HEAD"], {
cwd: "/workspace/repository",
}),
"Diff collection"
);
const changedFiles = status
.split("\n")
.filter(Boolean)
.map((line) => line.slice(3).trim());
let candidateRevision = baseRevision;
if (changedFiles.length > 0) {
requireSuccess(
await vm.execArgv("git", ["add", "-A"], { cwd: "/workspace/repository" }),
"Candidate staging"
); );
const tree = requireSuccess(
await vm.execArgv("git", ["write-tree"], { await vm.mkdir("/workspace", { recursive: true });
if (!(await vm.exists("/workspace/repository/.git"))) {
events.push(event(1, "repository.cloning", "Cloning project repository"));
const clone = await vm.execArgv(
"git",
["clone", input.repositoryUrl, "/workspace/repository"],
{
cwd: "/workspace",
env: authEnv,
}
);
requireSuccess(clone, "Repository clone");
}
const checkout = await vm.exec(
`git fetch origin ${shellQuote(input.baseBranch)} && git checkout -B ${shellQuote(`zopu/${input.runId}`)} ${shellQuote(`origin/${input.baseBranch}`)}`,
{ cwd: "/workspace/repository", env: authEnv }
);
requireSuccess(checkout, "Repository checkout");
const baseRevision = requireSuccess(
await vm.execArgv("git", ["rev-parse", "HEAD"], {
cwd: "/workspace/repository", cwd: "/workspace/repository",
}), }),
"Candidate tree creation" "Base revision lookup"
); );
candidateRevision = requireSuccess( events.push(
await vm.exec( event(2, "repository.ready", "Repository checkout is ready", {
`printf %s ${shellQuote(`Zopu candidate for ${input.attemptId}`)} | git commit-tree ${shellQuote(tree)} -p ${shellQuote(baseRevision)}`, baseRevision,
{ })
);
const sessionId = `codex-${input.attemptId}`;
await vm.openSession({
additionalInstructions:
"Work only inside /workspace/repository. Do not reveal credentials. Make the requested change and run focused verification. Do not push or open a pull request.",
agent: "codex",
cwd: "/workspace/repository",
env: codexSessionEnv({
apiKey: env.AGENT_MODEL_API_KEY,
baseUrl: env.AGENT_MODEL_BASE_URL,
model: env.AGENT_MODEL_NAME,
}),
permissionPolicy: "allow_all",
sessionId,
});
events.push(
event(3, "harness.started", "Codex implementation session started")
);
const promptResult = await vm.prompt({
content: [{ text: input.prompt, type: "text" }],
idempotencyKey: input.attemptId,
sessionId,
});
if (promptResult.stopReason !== "end_turn") {
const reason =
promptResult.stopReason === "cancelled" ? "Cancelled" : "HarnessFailed";
throw executionError(
`Codex stopped with ${promptResult.stopReason}`,
reason,
promptResult.stopReason === "max_tokens" ||
promptResult.stopReason === "max_turn_requests"
);
}
events.push(
event(4, "harness.progress", "Codex implementation turn completed", {
stopReason: promptResult.stopReason,
})
);
const status = requireSuccess(
await vm.execArgv("git", ["status", "--porcelain"], {
cwd: "/workspace/repository",
}),
"Changed file lookup"
);
const diff = requireSuccess(
await vm.execArgv("git", ["diff", "--binary", "HEAD"], {
cwd: "/workspace/repository",
}),
"Diff collection"
);
const changedFiles = status
.split("\n")
.filter(Boolean)
.map((line) => line.slice(3).trim());
let candidateRevision = baseRevision;
if (changedFiles.length > 0) {
requireSuccess(
await vm.execArgv("git", ["add", "-A"], {
cwd: "/workspace/repository", cwd: "/workspace/repository",
env: { }),
GIT_AUTHOR_EMAIL: "agent@zopu.dev", "Candidate staging"
GIT_AUTHOR_NAME: "Zopu Agent", );
GIT_COMMITTER_EMAIL: "agent@zopu.dev", const tree = requireSuccess(
GIT_COMMITTER_NAME: "Zopu Agent", await vm.execArgv("git", ["write-tree"], {
}, cwd: "/workspace/repository",
}),
"Candidate tree creation"
);
candidateRevision = requireSuccess(
await vm.exec(
`printf %s ${shellQuote(`Zopu candidate for ${input.attemptId}`)} | git commit-tree ${shellQuote(tree)} -p ${shellQuote(baseRevision)}`,
{
cwd: "/workspace/repository",
env: {
GIT_AUTHOR_EMAIL: "agent@zopu.dev",
GIT_AUTHOR_NAME: "Zopu Agent",
GIT_COMMITTER_EMAIL: "agent@zopu.dev",
GIT_COMMITTER_NAME: "Zopu Agent",
},
}
),
"Candidate revision creation"
);
requireSuccess(
await vm.execArgv("git", ["reset"], { cwd: "/workspace/repository" }),
"Candidate index reset"
);
}
events.push(
event(
5,
"repository.changed",
`${changedFiles.length} changed file(s) collected`,
{
candidateRevision,
} }
), ),
"Candidate revision creation" event(6, "runtime.completed", "AgentOS execution completed")
); );
requireSuccess(
await vm.execArgv("git", ["reset"], { cwd: "/workspace/repository" }),
"Candidate index reset"
);
}
events.push(
event(
5,
"repository.changed",
`${changedFiles.length} changed file(s) collected`,
{
candidateRevision,
}
),
event(6, "runtime.completed", "AgentOS execution completed")
);
return { return {
baseRevision, baseRevision,
candidateRevision, candidateRevision,
changedFiles, changedFiles,
diff, diff,
environmentId: input.workspaceKey, environmentId: input.workspaceKey,
events, events,
summary: summary:
changedFiles.length > 0 changedFiles.length > 0
? `Codex changed ${changedFiles.length} file(s)` ? `Codex changed ${changedFiles.length} file(s)`
: "Codex completed without repository changes", : "Codex completed without repository changes",
}; };
} catch (error) {
throw classifyRuntimeFailure(error);
}
}; };
export const cancelAgentOsAttempt = async (workspaceKey: string) => { export const cancelAgentOsAttempt = async (
workspaceKey: string,
attemptId: string
) => {
const env = parseAgentEnv(process.env); const env = parseAgentEnv(process.env);
const client = createClient<typeof runtimeRegistry>({ const client = createClient<typeof runtimeRegistry>({
disableMetadataLookup: true,
endpoint: env.RIVET_PUBLIC_ENDPOINT ?? env.RIVET_ENDPOINT, endpoint: env.RIVET_PUBLIC_ENDPOINT ?? env.RIVET_ENDPOINT,
}); });
await client.workspace.getOrCreate([workspaceKey]).cancelPrompt({}); await client.workspace
.getOrCreate([workspaceKey], {
params: { token: env.RIVET_WORKSPACE_TOKEN },
})
.cancelPrompt({ sessionId: `codex-${attemptId}` });
}; };

View File

@@ -3,6 +3,21 @@ import { ConvexError, v } from "convex/values";
import { internalMutation, mutation, query } from "./_generated/server"; import { internalMutation, mutation, query } from "./_generated/server";
import { requireCurrentOrganization, requireProjectMember } from "./authz"; import { requireCurrentOrganization, requireProjectMember } from "./authz";
// Provider ("forge") that owns a repository host. A project's source must be
// served by the same forge whose credentials are attached, so a Gitea token is
// never offered to GitHub and vice versa. Unknown hosts return null, leaving
// the caller free to attach without a forge constraint.
export const forgeForHost = (host: string): "github" | "gitea" | null => {
const normalized = host.toLowerCase();
if (normalized === "github.com" || normalized.endsWith(".githost.com")) {
return "github";
}
if (normalized === "git.openputer.com") {
return "gitea";
}
return null;
};
export const persist = internalMutation({ export const persist = internalMutation({
args: { args: {
credentialCiphertext: v.string(), credentialCiphertext: v.string(),
@@ -110,6 +125,15 @@ export const attachToProject = mutation({
if (!connection || connection.organizationId !== organizationId) { if (!connection || connection.organizationId !== organizationId) {
throw new ConvexError("Git connection not found"); throw new ConvexError("Git connection not found");
} }
const project = await ctx.db.get(args.projectId);
if (project) {
const expected = forgeForHost(project.sourceHost);
if (expected && connection.provider !== expected) {
throw new ConvexError(
`Git credential provider (${connection.provider}) does not match this project's forge (${expected})`
);
}
}
await ctx.db.patch(args.projectId, { await ctx.db.patch(args.projectId, {
gitConnectionId: connection._id, gitConnectionId: connection._id,
updatedAt: Date.now(), updatedAt: Date.now(),

View File

@@ -368,6 +368,17 @@ export default defineSchema({
workAttempts: defineTable({ workAttempts: defineTable({
classification: v.optional(attemptClassification), classification: v.optional(attemptClassification),
endedAt: v.optional(v.number()), endedAt: v.optional(v.number()),
failureReason: v.optional(
v.union(
v.literal("Authentication"),
v.literal("Cancelled"),
v.literal("HarnessFailed"),
v.literal("InvalidInput"),
v.literal("ProviderUnavailable"),
v.literal("RepositoryFailed"),
v.literal("Timeout")
)
),
leaseExpiresAt: v.optional(v.number()), leaseExpiresAt: v.optional(v.number()),
leaseOwner: v.optional(v.string()), leaseOwner: v.optional(v.string()),
number: v.number(), number: v.number(),

View File

@@ -1,7 +1,11 @@
"use node"; "use node";
import { env } from "@code/env/convex"; import { env } from "@code/env/convex";
import { decodeWorkAttemptExecutionResult } from "@code/primitives/execution-runtime"; import {
WorkAttemptExecutionError,
decodeWorkAttemptExecutionFailure,
decodeWorkAttemptExecutionResult,
} from "@code/primitives/execution-runtime";
import { ConvexError, v } from "convex/values"; import { ConvexError, v } from "convex/values";
import { Effect } from "effect"; import { Effect } from "effect";
@@ -56,23 +60,36 @@ export const executeAttempt = internalAction({
); );
const payload = (await response.json()) as unknown; const payload = (await response.json()) as unknown;
if (!response.ok) { if (!response.ok) {
throw new ConvexError( // Decode the agent's classified failure envelope and re-throw as the
typeof payload === "object" && payload && "error" in payload // typed runtime error so the workflow handler maps reason/retryable to
? String(payload.error) // a durable attempt classification. A malformed envelope falls back to
: `Agent backend returned ${response.status}` // an InvalidInput failure (non-retryable).
const failure = await Effect.runPromise(
decodeWorkAttemptExecutionFailure(payload)
); );
throw new WorkAttemptExecutionError(failure.error);
} }
return await Effect.runPromise(decodeWorkAttemptExecutionResult(payload)); return await Effect.runPromise(decodeWorkAttemptExecutionResult(payload));
}, },
}); });
export const cancelAttempt = internalAction({ export const cancelAttempt = internalAction({
args: { workspaceKey: v.string() }, args: {
attemptId: v.string(),
workspaceKey: v.string(),
},
handler: async (_ctx, args) => { handler: async (_ctx, args) => {
// The workspace key remains the URL path segment (workspace identity),
// while the body carries the attemptId so the runtime can target the
// codex session `codex-${attemptId}` for cancellation.
await fetch( await fetch(
`${backendUrl()}/internal/work-attempts/${encodeURIComponent(args.workspaceKey)}/cancel`, `${backendUrl()}/internal/work-attempts/${encodeURIComponent(args.workspaceKey)}/cancel`,
{ {
headers: { authorization: `Bearer ${env.FLUE_DB_TOKEN}` }, body: JSON.stringify({ attemptId: args.attemptId }),
headers: {
authorization: `Bearer ${env.FLUE_DB_TOKEN}`,
"content-type": "application/json",
},
method: "POST", method: "POST",
} }
); );

View File

@@ -7,94 +7,116 @@ import schema from "./schema";
const modules = import.meta.glob("./**/*.ts"); const modules = import.meta.glob("./**/*.ts");
const api = anyApi; const api = anyApi;
type TestContext = ReturnType<typeof convexTest>;
interface Seeded {
attemptId: string;
runId: string;
t: TestContext;
workId: string;
}
// Seed an in-flight real attempt so each mutation under test starts from a
// realistic running state: Work "executing", Run "running", Attempt "running".
const seedRunningAttempt = async (
overrides: {
attemptStatus?: "queued" | "claimed" | "running" | "terminal";
runStatus?: "ready" | "running" | "terminal" | "cancelled";
} = {}
): Promise<Seeded> => {
const t = convexTest({ modules, schema });
const seeded = await t.run(async (ctx) => {
const organizationId = await ctx.db.insert("organizations", {
createdAt: 1,
createdBy: "user",
kind: "personal",
name: "Personal",
});
const projectId = await ctx.db.insert("projects", {
createdAt: 1,
name: "Zopu",
normalizedSourceUrl: "https://github.com/puter/zopu",
organizationId,
repositoryPath: "puter/zopu",
sourceHost: "github.com",
sourceUrl: "https://github.com/puter/zopu",
updatedAt: 1,
});
const workId = await ctx.db.insert("works", {
createdAt: 1,
objective: "Implement a real slice",
organizationId,
projectId,
status: "executing",
title: "Real execution",
updatedAt: 1,
});
const sliceRowId = await ctx.db.insert("workSlices", {
designVersion: 1,
objective: "Change the repo",
observableBehavior: "A diff exists",
ordinal: 0,
payloadJson: "{}",
sliceId: "slice-1",
status: "running",
title: "Implementation",
workId,
});
const runId = await ctx.db.insert("workRuns", {
createdAt: 1,
designVersion: 1,
executionKind: "real",
kitId: "coding-v0",
kitVersion: "1",
scenario: "success",
sliceId: "slice-1",
sliceRowId,
status: overrides.runStatus ?? "running",
workId,
});
const attemptId = await ctx.db.insert("workAttempts", {
number: 1,
runId,
status: overrides.attemptStatus ?? "running",
workId,
workspaceKey: "workspace-1",
});
return { attemptId, runId, workId };
});
return { ...seeded, t };
};
const successResult = {
baseRevision: "base123",
candidateRevision: "candidate456",
changedFiles: ["src/index.ts"],
diff: "+export const ready = true;",
environmentId: "workspace-1",
events: [
{
kind: "runtime.completed",
message: "completed",
metadata: {},
occurredAt: 2,
sequence: 0,
},
],
summary: "Changed one file",
};
describe("real work execution persistence", () => { describe("real work execution persistence", () => {
test("records revisions, activity, diff, and terminal state atomically", async () => { test("records revisions, activity, diff, and terminal state atomically", async () => {
const t = convexTest({ modules, schema }); const { t, ...seeded } = await seedRunningAttempt();
const seeded = await t.run(async (ctx) => {
const organizationId = await ctx.db.insert("organizations", {
createdAt: 1,
createdBy: "user",
kind: "personal",
name: "Personal",
});
const projectId = await ctx.db.insert("projects", {
createdAt: 1,
name: "Zopu",
normalizedSourceUrl: "https://example.com/zopu",
organizationId,
repositoryPath: "puter/zopu",
sourceHost: "example.com",
sourceUrl: "https://example.com/zopu",
updatedAt: 1,
});
const workId = await ctx.db.insert("works", {
createdAt: 1,
objective: "Implement a real slice",
organizationId,
projectId,
status: "executing",
title: "Real execution",
updatedAt: 1,
});
const sliceRowId = await ctx.db.insert("workSlices", {
designVersion: 1,
objective: "Change the repo",
observableBehavior: "A diff exists",
ordinal: 0,
payloadJson: "{}",
sliceId: "slice-1",
status: "running",
title: "Implementation",
workId,
});
const runId = await ctx.db.insert("workRuns", {
createdAt: 1,
designVersion: 1,
executionKind: "real",
kitId: "coding-v0",
kitVersion: "1",
scenario: "success",
sliceId: "slice-1",
sliceRowId,
status: "running",
workId,
});
const attemptId = await ctx.db.insert("workAttempts", {
number: 1,
runId,
status: "running",
workId,
workspaceKey: "workspace-1",
});
return { attemptId, runId, workId };
});
await t.mutation(api.workExecutionWorkflow.completeAttempt, { await t.mutation(api.workExecutionWorkflow.completeAttempt, {
attemptId: seeded.attemptId, attemptId: seeded.attemptId,
result: { result: successResult,
baseRevision: "base123",
candidateRevision: "candidate456",
changedFiles: ["src/index.ts"],
diff: "+export const ready = true;",
environmentId: "workspace-1",
events: [
{
kind: "runtime.completed",
message: "completed",
metadata: {},
occurredAt: 2,
sequence: 0,
},
],
summary: "Changed one file",
},
}); });
const state = await t.run(async (ctx) => ({ const state = await t.run(async (ctx) => ({
artifacts: await ctx.db.query("workArtifacts").collect(), artifacts: await ctx.db.query("workArtifacts").collect(),
attempt: await ctx.db.get(seeded.attemptId), attempt: await ctx.db.get(seeded.attemptId as any),
run: await ctx.db.get(seeded.runId), run: await ctx.db.get(seeded.runId as any),
work: await ctx.db.get(seeded.workId), work: await ctx.db.get(seeded.workId as any),
})); }));
expect(state.attempt?.classification).toBe("Succeeded"); expect(state.attempt?.classification).toBe("Succeeded");
expect(state.run).toMatchObject({ expect(state.run).toMatchObject({
@@ -109,3 +131,251 @@ describe("real work execution persistence", () => {
}); });
}); });
}); });
describe("classified failure mapping", () => {
test("maps a transient runtime reason to a retryable classification", async () => {
const { t, ...seeded } = await seedRunningAttempt();
await t.mutation(api.workExecutionWorkflow.failAttempt, {
attemptId: seeded.attemptId,
reason: "ProviderUnavailable",
retryable: true,
summary: "model provider timed out",
});
const attempt = await t.run((ctx) => ctx.db.get(seeded.attemptId as any));
expect(attempt?.classification).toBe("RetryableFailure");
expect(attempt?.failureReason).toBe("ProviderUnavailable");
});
test("queues a new attempt without terminalizing a retryable run", async () => {
const { t, ...seeded } = await seedRunningAttempt();
await t.mutation(api.workExecutionWorkflow.failAttempt, {
attemptId: seeded.attemptId,
reason: "ProviderUnavailable",
retryable: true,
summary: "provider unavailable",
});
const state = await t.run(async (ctx) => ({
attempts: await ctx.db.query("workAttempts").collect(),
run: await ctx.db.get(seeded.runId as any),
work: await ctx.db.get(seeded.workId as any),
}));
expect(state.attempts).toHaveLength(2);
expect(state.attempts[1]).toMatchObject({
number: 2,
status: "queued",
workspaceKey: "workspace-1",
});
expect(state.run?.status).toBe("running");
expect(state.work?.status).toBe("executing");
});
test("maps an authentication reason to a permanent failure", async () => {
const { t, ...seeded } = await seedRunningAttempt();
await t.mutation(api.workExecutionWorkflow.failAttempt, {
attemptId: seeded.attemptId,
reason: "Authentication",
retryable: false,
summary: "token rejected",
});
const state = await t.run(async (ctx) => ({
attempt: await ctx.db.get(seeded.attemptId as any),
run: await ctx.db.get(seeded.runId as any),
work: await ctx.db.get(seeded.workId as any),
}));
expect(state.attempt?.classification).toBe("PermanentFailure");
expect(state.attempt?.failureReason).toBe("Authentication");
expect(state.run?.terminalClassification).toBe("PermanentFailure");
expect(state.work?.status).toBe("failed");
});
});
describe("cancellation fencing", () => {
test("a cancelled attempt cannot later settle success", async () => {
const { t, ...seeded } = await seedRunningAttempt();
// Simulate cancellation marking the attempt/run terminal as Cancelled.
await t.run(async (ctx) => {
await ctx.db.patch(seeded.attemptId as any, {
classification: "Cancelled",
endedAt: 5,
status: "terminal",
summary: "Execution cancelled",
});
await ctx.db.patch(seeded.runId as any, {
endedAt: 5,
status: "cancelled",
terminalClassification: "Cancelled",
terminalSummary: "Execution cancelled",
});
});
// A late-arriving completion must be ignored.
await t.mutation(api.workExecutionWorkflow.completeAttempt, {
attemptId: seeded.attemptId,
result: successResult,
});
const attempt = await t.run((ctx) => ctx.db.get(seeded.attemptId as any));
expect(attempt?.classification).toBe("Cancelled");
expect(attempt?.status).toBe("terminal");
});
test("a late failure does not overwrite a cancelled run", async () => {
const { t, ...seeded } = await seedRunningAttempt();
await t.run(async (ctx) => {
await ctx.db.patch(seeded.attemptId as any, {
classification: "Cancelled",
endedAt: 5,
status: "terminal",
});
await ctx.db.patch(seeded.runId as any, { status: "cancelled" });
});
await t.mutation(api.workExecutionWorkflow.failAttempt, {
attemptId: seeded.attemptId,
reason: "HarnessFailed",
retryable: true,
summary: "late failure",
});
const attempt = await t.run((ctx) => ctx.db.get(seeded.attemptId as any));
expect(attempt?.classification).toBe("Cancelled");
});
});
describe("empty-change rejection", () => {
test("a no-op result with no changed files fails as permanent", async () => {
const { t, ...seeded } = await seedRunningAttempt();
await t.mutation(api.workExecutionWorkflow.completeAttempt, {
attemptId: seeded.attemptId,
result: {
...successResult,
changedFiles: [],
candidateRevision: "base123",
},
});
const state = await t.run(async (ctx) => ({
attempt: await ctx.db.get(seeded.attemptId as any),
work: await ctx.db.get(seeded.workId as any),
}));
expect(state.attempt?.classification).toBe("PermanentFailure");
expect(state.attempt?.failureReason).toBe("InvalidInput");
expect(state.work?.status).toBe("failed");
});
test("identical base and candidate revisions are rejected", async () => {
const { t, ...seeded } = await seedRunningAttempt();
await t.mutation(api.workExecutionWorkflow.completeAttempt, {
attemptId: seeded.attemptId,
result: {
...successResult,
baseRevision: "same",
candidateRevision: "same",
},
});
const attempt = await t.run((ctx) => ctx.db.get(seeded.attemptId as any));
expect(attempt?.classification).toBe("PermanentFailure");
});
});
describe("attempt re-entry", () => {
test("markAttemptRunning re-claims an already-running attempt", async () => {
const { t, ...seeded } = await seedRunningAttempt();
// First claim (already "running" from seed) must still succeed so a
// workflow replay can re-enter the same live attempt.
const first = await t.mutation(
api.workExecutionWorkflow.markAttemptRunning,
{ attemptId: seeded.attemptId }
);
expect(first).toBe(true);
const attempt = await t.run((ctx) => ctx.db.get(seeded.attemptId as any));
expect(attempt?.status).toBe("running");
expect(attempt?.leaseExpiresAt).toBeGreaterThan(0);
});
test("a terminal attempt is not re-runnable", async () => {
const { t, ...seeded } = await seedRunningAttempt({
attemptStatus: "terminal",
});
const result = await t.mutation(
api.workExecutionWorkflow.markAttemptRunning,
{ attemptId: seeded.attemptId }
);
expect(result).toBe(false);
});
test("a cancelled run cannot re-enter a queued attempt", async () => {
const { t, ...seeded } = await seedRunningAttempt({
attemptStatus: "queued",
runStatus: "cancelled",
});
const result = await t.mutation(
api.workExecutionWorkflow.markAttemptRunning,
{ attemptId: seeded.attemptId }
);
expect(result).toBe(false);
const attempt = await t.run((ctx) => ctx.db.get(seeded.attemptId as any));
expect(attempt?.status).toBe("queued");
});
});
describe("forge credential validation", () => {
const identity = { tokenIdentifier: "https://convex.test|forge-user" };
const seedForgableProject = async (
provider: "github" | "gitea",
sourceHost: string
) => {
const t = convexTest({ modules, schema }).withIdentity(identity);
const ids = await t.run(async (ctx) => {
const organizationId = await ctx.db.insert("organizations", {
createdAt: 1,
createdBy: identity.tokenIdentifier,
kind: "personal",
name: "Personal",
});
await ctx.db.insert("organizationMembers", {
createdAt: 1,
organizationId,
role: "owner",
userId: identity.tokenIdentifier,
});
const projectId = await ctx.db.insert("projects", {
createdAt: 1,
name: "Zopu",
normalizedSourceUrl: `https://${sourceHost}/puter/zopu`,
organizationId,
repositoryPath: "puter/zopu",
sourceHost,
sourceUrl: `https://${sourceHost}/puter/zopu`,
updatedAt: 1,
});
const connectionId = await ctx.db.insert("gitConnections", {
connectedAt: 1,
credentialCiphertext: "x",
credentialIv: "y",
credentialKind: provider === "github" ? "oauth" : "token",
organizationId,
provider,
serverUrl: `https://${sourceHost}`,
updatedAt: 1,
username: "zopu",
});
return { connectionId, projectId };
});
return { t, ids };
};
test("rejects a git connection whose provider mismatches the project forge", async () => {
// Gitea token attached to a GitHub-hosted project.
const { t, ids } = await seedForgableProject("gitea", "github.com");
await expect(
t.mutation(api.gitConnectionData.attachToProject, {
connectionId: ids.connectionId,
projectId: ids.projectId,
})
).rejects.toThrow(/does not match this project's forge/u);
});
test("accepts a matching provider for the project forge", async () => {
const { t, ids } = await seedForgableProject("github", "github.com");
const result = await t.mutation(api.gitConnectionData.attachToProject, {
connectionId: ids.connectionId,
projectId: ids.projectId,
});
expect(result).toMatchObject({ attached: true });
});
});

View File

@@ -1,4 +1,7 @@
import { defaultCodingKitV0 } from "@code/primitives/resolver"; import { WorkAttemptExecutionError } from "@code/primitives/execution-runtime";
import type { WorkAttemptExecutionErrorReason } from "@code/primitives/execution-runtime";
import { defaultCodingKitV0, resolveOutcome } from "@code/primitives/resolver";
import type { AttemptClassification } from "@code/primitives/resolver";
import { WorkflowManager } from "@convex-dev/workflow"; import { WorkflowManager } from "@convex-dev/workflow";
import type { WorkflowId } from "@convex-dev/workflow"; import type { WorkflowId } from "@convex-dev/workflow";
import { ConvexError, v } from "convex/values"; import { ConvexError, v } from "convex/values";
@@ -8,9 +11,68 @@ import type { Doc, Id } from "./_generated/dataModel";
import { internalMutation, internalQuery, mutation } from "./_generated/server"; import { internalMutation, internalQuery, mutation } from "./_generated/server";
import type { MutationCtx } from "./_generated/server"; import type { MutationCtx } from "./_generated/server";
import { requireProjectMember } from "./authz"; import { requireProjectMember } from "./authz";
import { forgeForHost } from "./gitConnectionData";
export const workflow = new WorkflowManager(components.workflow); export const workflow = new WorkflowManager(components.workflow);
// Lease window for a running real attempt. Long enough to outlast a single
// agent turn; the workflow's own retry/lease reconciliation covers crashes.
const LEASE_MS = 5 * 60_000;
// Runtime failure reason -> durable attempt classification. The Convex
// workflow stores classifications, not provider-specific reasons, so the
// resolver and Work lifecycle stay forge-agnostic. `retryable` from the
// runtime is carried alongside and consulted by the kit retry policy.
const FAILURE_REASON_CLASSIFICATION = {
Authentication: "PermanentFailure",
Cancelled: "Cancelled",
HarnessFailed: "RetryableFailure",
InvalidInput: "PermanentFailure",
ProviderUnavailable: "RetryableFailure",
RepositoryFailed: "PermanentFailure",
Timeout: "RetryableFailure",
} as const satisfies Record<
WorkAttemptExecutionErrorReason,
AttemptClassification
>;
export const classifyFailure = (
reason: WorkAttemptExecutionErrorReason
): AttemptClassification => FAILURE_REASON_CLASSIFICATION[reason];
// Normalize an error thrown from the agent action into the durable failure
// triple (reason/retryable/summary). WorkAttemptExecutionError is the
// classified runtime failure; anything else is a Convex/infrastructure error
// treated as a transient HarnessFailed so the workflow retry policy decides.
const toExecutionFailure = (
error: unknown
): {
message: string;
reason: WorkAttemptExecutionErrorReason;
retryable: boolean;
} =>
error instanceof WorkAttemptExecutionError
? {
message: error.message,
reason: error.reason,
retryable: error.retryable,
}
: {
message: error instanceof Error ? error.message : "Execution failed",
reason: "HarnessFailed",
retryable: true,
};
const failureReasonValues = v.union(
v.literal("Authentication"),
v.literal("Cancelled"),
v.literal("HarnessFailed"),
v.literal("InvalidInput"),
v.literal("ProviderUnavailable"),
v.literal("RepositoryFailed"),
v.literal("Timeout")
);
const resolveReadySlice = async ( const resolveReadySlice = async (
ctx: MutationCtx, ctx: MutationCtx,
work: Doc<"works">, work: Doc<"works">,
@@ -36,6 +98,34 @@ const resolveReadySlice = async (
return slice; return slice;
}; };
// Deployment prerequisite gate for real execution. Rejects before any
// workflow/sandbox starts when the project lacks an attached, forge-matched
// Git connection with a cloneable source URL and default branch. Returns the
// validated connection so the caller can pass it to the agent unchanged.
const validateProjectDeployment = async (
ctx: MutationCtx,
work: Doc<"works">
): Promise<Doc<"gitConnections">> => {
const project = await ctx.db.get(work.projectId);
if (!project) {
throw new ConvexError("Project not found");
}
if (!project.gitConnectionId) {
throw new ConvexError("Connect Git credentials to this project first");
}
const connection = await ctx.db.get(project.gitConnectionId);
if (!connection) {
throw new ConvexError("Git connection not found");
}
const expected = forgeForHost(project.sourceHost);
if (expected && connection.provider !== expected) {
throw new ConvexError(
`Git credential provider (${connection.provider}) does not match this project's forge (${expected})`
);
}
return connection;
};
export const execute = workflow export const execute = workflow
.define({ args: { attemptId: v.id("workAttempts") } }) .define({ args: { attemptId: v.id("workAttempts") } })
.handler(async (step, args): Promise<void> => { .handler(async (step, args): Promise<void> => {
@@ -57,9 +147,12 @@ export const execute = workflow
}, },
}); });
} catch (error) { } catch (error) {
const failure = toExecutionFailure(error);
await step.runMutation(internal.workExecutionWorkflow.failAttempt, { await step.runMutation(internal.workExecutionWorkflow.failAttempt, {
attemptId: args.attemptId, attemptId: args.attemptId,
summary: error instanceof Error ? error.message : "Execution failed", reason: failure.reason,
retryable: failure.retryable,
summary: failure.message,
}); });
} }
}); });
@@ -91,10 +184,7 @@ export const startExecution = mutation({
) { ) {
throw new ConvexError("Execution requires exact approved versions"); throw new ConvexError("Execution requires exact approved versions");
} }
const project = await ctx.db.get(work.projectId); await validateProjectDeployment(ctx, work);
if (!project?.gitConnectionId) {
throw new ConvexError("Connect Git credentials to this project first");
}
const slice = await resolveReadySlice(ctx, work, args.sliceId); const slice = await resolveReadySlice(ctx, work, args.sliceId);
const createdAt = Date.now(); const createdAt = Date.now();
const runId = await ctx.db.insert("workRuns", { const runId = await ctx.db.insert("workRuns", {
@@ -159,6 +249,12 @@ export const executionContext = internalQuery({
if (!connection) { if (!connection) {
throw new ConvexError("Git connection not found"); throw new ConvexError("Git connection not found");
} }
const expectedForge = forgeForHost(project.sourceHost);
if (expectedForge && connection.provider !== expectedForge) {
throw new ConvexError(
`Git credential provider (${connection.provider}) does not match this project's forge (${expectedForge})`
);
}
return { return {
attempt, attempt,
connection, connection,
@@ -179,11 +275,16 @@ export const markAttemptRunning = internalMutation({
args: { attemptId: v.id("workAttempts") }, args: { attemptId: v.id("workAttempts") },
handler: async (ctx, args) => { handler: async (ctx, args) => {
const attempt = await ctx.db.get(args.attemptId); const attempt = await ctx.db.get(args.attemptId);
if (!attempt || attempt.status !== "queued") { if (!attempt || attempt.status === "terminal") {
return false;
}
const run = await ctx.db.get(attempt.runId);
if (!run || run.status !== "running") {
return false; return false;
} }
await ctx.db.patch(attempt._id, { await ctx.db.patch(attempt._id, {
startedAt: Date.now(), leaseExpiresAt: Date.now() + LEASE_MS,
startedAt: attempt.startedAt ?? Date.now(),
status: "running", status: "running",
}); });
return true; return true;
@@ -193,7 +294,8 @@ export const markAttemptRunning = internalMutation({
const settleSliceAndWork = async ( const settleSliceAndWork = async (
ctx: MutationCtx, ctx: MutationCtx,
run: Doc<"workRuns">, run: Doc<"workRuns">,
succeeded: boolean succeeded: boolean,
workStatus: Doc<"works">["status"]
) => { ) => {
if (!run.sliceRowId) { if (!run.sliceRowId) {
return; return;
@@ -202,12 +304,18 @@ const settleSliceAndWork = async (
if (!slice) { if (!slice) {
return; return;
} }
await ctx.db.patch(slice._id, { status: succeeded ? "completed" : "ready" }); let sliceStatus: "blocked" | "completed" | "ready" = "ready";
if (succeeded) {
sliceStatus = "completed";
} else if (workStatus === "blocked") {
sliceStatus = "blocked";
}
await ctx.db.patch(slice._id, { status: sliceStatus });
const work = await ctx.db.get(run.workId); const work = await ctx.db.get(run.workId);
if (!work) { if (!work) {
return; return;
} }
let status: Doc<"works">["status"] = succeeded ? "completed" : "failed"; let status = workStatus;
if (succeeded) { if (succeeded) {
const slices = await ctx.db const slices = await ctx.db
.query("workSlices") .query("workSlices")
@@ -249,6 +357,8 @@ export const completeAttempt = internalMutation({
}, },
handler: async (ctx, args) => { handler: async (ctx, args) => {
const attempt = await ctx.db.get(args.attemptId); const attempt = await ctx.db.get(args.attemptId);
// Cancellation fencing: a terminal or cancelled attempt/run must never
// later settle success, even if the agent action raced to completion.
if (!attempt || attempt.status === "terminal") { if (!attempt || attempt.status === "terminal") {
return; return;
} }
@@ -257,6 +367,33 @@ export const completeAttempt = internalMutation({
if (!run || !work) { if (!run || !work) {
throw new ConvexError("Execution records not found"); throw new ConvexError("Execution records not found");
} }
if (run.status === "terminal" || run.status === "cancelled") {
return;
}
// Empty-change rejection: a no-op result (no changed files, or identical
// base/candidate revisions) is not a successful implementation. Fail it
// as a permanent InvalidInput so the resolver does not mark Work done.
const noOp =
args.result.changedFiles.length === 0 ||
args.result.baseRevision === args.result.candidateRevision;
if (noOp) {
await ctx.db.patch(attempt._id, {
classification: "PermanentFailure",
endedAt: Date.now(),
failureReason: "InvalidInput",
leaseExpiresAt: undefined,
status: "terminal",
summary: "Execution produced no repository changes",
});
await ctx.db.patch(run._id, {
endedAt: Date.now(),
status: "terminal",
terminalClassification: "PermanentFailure",
terminalSummary: "Execution produced no repository changes",
});
await settleSliceAndWork(ctx, run, false, "failed");
return;
}
for (const item of args.result.events) { for (const item of args.result.events) {
await ctx.db.insert("workAttemptEvents", { await ctx.db.insert("workAttemptEvents", {
attemptId: attempt._id, attemptId: attempt._id,
@@ -271,6 +408,7 @@ export const completeAttempt = internalMutation({
await ctx.db.patch(attempt._id, { await ctx.db.patch(attempt._id, {
classification: "Succeeded", classification: "Succeeded",
endedAt, endedAt,
leaseExpiresAt: undefined,
status: "terminal", status: "terminal",
summary: args.result.summary, summary: args.result.summary,
}); });
@@ -309,7 +447,7 @@ export const completeAttempt = internalMutation({
} }
: {}), : {}),
}); });
await settleSliceAndWork(ctx, run, true); await settleSliceAndWork(ctx, run, true, "completed");
await ctx.db.insert("workEvents", { await ctx.db.insert("workEvents", {
createdAt: endedAt, createdAt: endedAt,
idempotencyKey: `real-run-completed:${run._id}`, idempotencyKey: `real-run-completed:${run._id}`,
@@ -321,31 +459,107 @@ export const completeAttempt = internalMutation({
}, },
}); });
export const failAttempt = internalMutation({ export const launchAttemptWorkflow = internalMutation({
args: { attemptId: v.id("workAttempts"), summary: v.string() }, args: { attemptId: v.id("workAttempts") },
handler: async (ctx, args) => { handler: async (ctx, args) => {
const attempt = await ctx.db.get(args.attemptId); const attempt = await ctx.db.get(args.attemptId);
if (!attempt || attempt.status === "terminal") { if (!attempt || attempt.status === "terminal") {
return; return;
} }
const run = await ctx.db.get(attempt.runId); const run = await ctx.db.get(attempt.runId);
if (!run) { if (!run || run.status !== "running") {
return; return;
} }
const workflowId: WorkflowId = await workflow.start(
ctx,
internal.workExecutionWorkflow.execute,
args
);
await ctx.db.patch(run._id, { workflowId });
},
});
export const failAttempt = internalMutation({
args: {
attemptId: v.id("workAttempts"),
reason: failureReasonValues,
retryable: v.boolean(),
summary: v.string(),
},
handler: async (ctx, args) => {
const attempt = await ctx.db.get(args.attemptId);
// Cancellation fencing: once an attempt or run is terminal/cancelled, a
// late failure from the workflow catch block must not overwrite the
// settled state. This also prevents a cancelled attempt from settling
// as a different classification after cancelExecution ran.
if (!attempt || attempt.status === "terminal") {
return;
}
const run = await ctx.db.get(attempt.runId);
if (!run || run.status === "terminal" || run.status === "cancelled") {
return;
}
const classification = classifyFailure(args.reason);
const endedAt = Date.now(); const endedAt = Date.now();
await ctx.db.patch(attempt._id, { await ctx.db.patch(attempt._id, {
classification: "PermanentFailure", classification,
endedAt, endedAt,
failureReason: args.reason,
leaseExpiresAt: undefined,
status: "terminal", status: "terminal",
summary: args.summary, summary: args.summary,
}); });
const outcome = resolveOutcome(
{ classification, retryable: args.retryable, summary: args.summary },
attempt.number,
defaultCodingKitV0.retryPolicy
);
await ctx.db.insert("resolverDecisions", {
attemptId: attempt._id,
attemptNumber: attempt.number,
classification,
createdAt: endedAt,
decision: outcome.kind,
...(outcome.kind === "terminal"
? { resultingWorkStatus: outcome.workStatus }
: {}),
runId: run._id,
summary: args.summary,
workId: attempt.workId,
});
if (outcome.kind === "retry") {
const nextAttemptId = await ctx.db.insert("workAttempts", {
number: attempt.number + 1,
runId: run._id,
status: "queued",
workId: attempt.workId,
workspaceKey: attempt.workspaceKey,
});
await ctx.scheduler.runAfter(
0,
internal.workExecutionWorkflow.launchAttemptWorkflow,
{ attemptId: nextAttemptId }
);
return;
}
await ctx.db.patch(run._id, { await ctx.db.patch(run._id, {
endedAt, endedAt,
status: "terminal", status: "terminal",
terminalClassification: "PermanentFailure", terminalClassification: classification,
terminalSummary: args.summary, terminalSummary: args.summary,
}); });
await settleSliceAndWork(ctx, run, false); await settleSliceAndWork(ctx, run, false, outcome.workStatus);
const work = await ctx.db.get(run.workId);
if (work) {
await ctx.db.insert("workEvents", {
createdAt: endedAt,
idempotencyKey: `real-run-failed:${run._id}`,
kind: "run.completed",
payloadJson: JSON.stringify({ classification }),
referenceId: String(run._id),
workId: work._id,
});
}
}, },
}); });
@@ -375,15 +589,30 @@ export const cancelExecution = mutation({
0, 0,
internal.workExecutionAgent.cancelAttempt, internal.workExecutionAgent.cancelAttempt,
{ {
attemptId: String(attempt._id),
workspaceKey: attempt.workspaceKey, workspaceKey: attempt.workspaceKey,
} }
); );
const cancelledAt = Date.now();
await ctx.db.patch(attempt._id, { await ctx.db.patch(attempt._id, {
classification: "Cancelled", classification: "Cancelled",
endedAt: Date.now(), endedAt: cancelledAt,
failureReason: "Cancelled",
leaseExpiresAt: undefined,
status: "terminal", status: "terminal",
summary: "Execution cancelled", summary: "Execution cancelled",
}); });
await ctx.db.insert("resolverDecisions", {
attemptId: attempt._id,
attemptNumber: attempt.number,
classification: "Cancelled",
createdAt: cancelledAt,
decision: "terminal",
resultingWorkStatus: "ready",
runId: run._id,
summary: "Execution cancelled",
workId: run.workId,
});
} }
await ctx.db.patch(run._id, { await ctx.db.patch(run._id, {
endedAt: Date.now(), endedAt: Date.now(),

View File

@@ -16,6 +16,7 @@ const agentEnvSchema = z.object({
OPENROUTER_API_KEY: z.string().min(1).optional(), OPENROUTER_API_KEY: z.string().min(1).optional(),
RIVET_ENDPOINT: z.url(), RIVET_ENDPOINT: z.url(),
RIVET_PUBLIC_ENDPOINT: z.url().optional(), RIVET_PUBLIC_ENDPOINT: z.url().optional(),
RIVET_WORKSPACE_TOKEN: z.string().min(32),
}); });
export type AgentEnv = z.infer<typeof agentEnvSchema>; export type AgentEnv = z.infer<typeof agentEnvSchema>;

View File

@@ -31,7 +31,6 @@
}, },
"dependencies": { "dependencies": {
"@agentos-software/codex": "0.2.7", "@agentos-software/codex": "0.2.7",
"@agentos-software/codex-cli": "0.3.4",
"@agentos-software/git": "0.3.3", "@agentos-software/git": "0.3.3",
"@rivet-dev/agentos": "catalog:", "@rivet-dev/agentos": "catalog:",
"effect": "catalog:" "effect": "catalog:"

View File

@@ -1,5 +1,5 @@
/* eslint-disable max-classes-per-file -- the AgentOS service and its tagged error form one adapter contract. */ /* eslint-disable max-classes-per-file -- the AgentOS service and its tagged error form one adapter contract. */
import codex from "@agentos-software/codex-cli"; import codex from "@agentos-software/codex";
import { agentOS as createAgentOsActor } from "@rivet-dev/agentos"; import { agentOS as createAgentOsActor } from "@rivet-dev/agentos";
import type { AgentOSConfigInput } from "@rivet-dev/agentos"; import type { AgentOSConfigInput } from "@rivet-dev/agentos";
import { Context, Effect, Layer, Schema } from "effect"; import { Context, Effect, Layer, Schema } from "effect";
@@ -99,6 +99,7 @@ export const codexSessionEnv = (
model: CodexModelEnv, model: CodexModelEnv,
extra: Readonly<Record<string, string>> = {} extra: Readonly<Record<string, string>> = {}
): Record<string, string> => ({ ): Record<string, string> => ({
CODEX_MODEL: model.model,
OPENAI_API_KEY: model.apiKey, OPENAI_API_KEY: model.apiKey,
OPENAI_BASE_URL: model.baseUrl, OPENAI_BASE_URL: model.baseUrl,
...extra, ...extra,

View File

@@ -121,6 +121,16 @@ export class WorkAttemptExecutionError extends Schema.TaggedErrorClass<WorkAttem
} }
) {} ) {}
export const WorkAttemptExecutionFailure = Schema.Struct({
error: Schema.Struct({
message: Text,
reason: WorkAttemptExecutionErrorReason,
retryable: Schema.Boolean,
}),
});
export type WorkAttemptExecutionFailure =
typeof WorkAttemptExecutionFailure.Type;
const invalidInput = (message: string) => const invalidInput = (message: string) =>
new WorkAttemptExecutionError({ new WorkAttemptExecutionError({
message, message,
@@ -143,6 +153,11 @@ export const decodeWorkAttemptExecutionResult = (input: unknown) =>
Effect.mapError((cause) => invalidInput(cause.message)) Effect.mapError((cause) => invalidInput(cause.message))
); );
export const decodeWorkAttemptExecutionFailure = (input: unknown) =>
Schema.decodeUnknownEffect(WorkAttemptExecutionFailure)(input).pipe(
Effect.mapError((cause) => invalidInput(cause.message))
);
export interface SandboxRuntime { export interface SandboxRuntime {
readonly name: string; readonly name: string;
readonly execute: ( readonly execute: (
@@ -150,6 +165,7 @@ export interface SandboxRuntime {
signal?: AbortSignal signal?: AbortSignal
) => Effect.Effect<WorkAttemptExecutionResult, WorkAttemptExecutionError>; ) => Effect.Effect<WorkAttemptExecutionResult, WorkAttemptExecutionError>;
readonly cancel: ( readonly cancel: (
workspaceKey: string workspaceKey: string,
attemptId: string
) => Effect.Effect<void, WorkAttemptExecutionError>; ) => Effect.Effect<void, WorkAttemptExecutionError>;
} }