lazycodex-ai 5.0.0-beta.6 → 5.0.0-beta.61

package/packages/omo-codex/plugin/components/ulw-loop/src/sdk/factory.ts

@@ -0,0 +1,183 @@
+import { checkpointUlwLoop } from "../checkpoint.js";
+import { checkpointTemplate } from "../checkpoint-template.js";
+import { recordEvidence } from "../evidence.js";
+import { type UlwLoopScope, ulwLoopAttemptEvidenceDir } from "../paths.js";
+import { addUlwLoopGoal, createUlwLoopPlan, startNextUlwLoop, summarizeUlwLoopPlan } from "../plan-crud.js";
+import { readUlwLoopPlan } from "../plan-io.js";
+import { recordFinalReviewBlockers } from "../review-blockers.js";
+import { UlwLoopError } from "../runtime.js";
+import { statusNextActions } from "../status-next-actions.js";
+import { steerUlwLoop } from "../steering.js";
+import type { UlwLoopOperation } from "./manifest.js";
+import { ULW_LOOP_MANIFEST } from "./manifest.js";
+import type {
+	AgentToolkit,
+	AgentToolkitDependencies,
+	CheckpointArgs,
+	ToolkitContext,
+	ToolkitDispatchRequest,
+	ToolkitDispatchResponse,
+	ToolkitError,
+	ToolkitFailure,
+	ToolkitResponseFor,
+	ToolkitResultFor,
+	ToolkitUnknownRequest,
+} from "./types.js";
+
+// The SDK takes inline values only, so a snapshot that is not JSON is a caller error - normalize it
+// to the same stable code the CLI reports instead of letting it surface as a generic failure.
+function validateCodexGoalJson(raw: string | undefined): void {
+	if (raw === undefined) return;
+	try {
+		JSON.parse(raw);
+	} catch (error) {
+		throw new UlwLoopError(
+			`Invalid codexGoal: ${error instanceof Error ? error.message : "not valid JSON"}`,
+			"ULW_LOOP_CODEX_GOAL_JSON_INVALID",
+			{ cause: error },
+		);
+	}
+}
+
+function validateContext(context: ToolkitContext): void {
+	if (!context.cwd.trim()) throw new UlwLoopError("cwd is required.", "ULW_LOOP_CWD_REQUIRED");
+	if (!context.sessionId.trim())
+		throw new UlwLoopError("ULW_LOOP_SESSION_ID_REQUIRED: sessionId is required.", "ULW_LOOP_SESSION_ID_REQUIRED");
+	if (context.surface !== "omo-senpi" && context.surface !== "lazycodex")
+		throw new UlwLoopError("surface must be omo-senpi or lazycodex.", "ULW_LOOP_SURFACE_INVALID");
+}
+
+function errorDetails(error: UlwLoopError): ToolkitError {
+	const details =
+		error.details === undefined
+			? undefined
+			: Object.fromEntries(Object.entries(error.details).map(([key, value]) => [key, String(value)]));
+	return details === undefined
+		? { code: error.code, message: error.message }
+		: { code: error.code, message: error.message, details };
+}
+
+function failure<Operation extends string>(operation: Operation, error: UlwLoopError): ToolkitFailure<Operation> {
+	return { ok: false, operation, error: errorDetails(error) };
+}
+
+function caught<Operation extends string>(operation: Operation, error: Error): ToolkitFailure<Operation> {
+	return failure(operation, error instanceof UlwLoopError ? error : new UlwLoopError(error.message, "ULW_LOOP_ERROR"));
+}
+
+function isKnownRequest(request: ToolkitDispatchRequest | ToolkitUnknownRequest): request is ToolkitDispatchRequest {
+	return ULW_LOOP_MANIFEST.operations.some((operation) => operation.name === request.operation);
+}
+
+function unreachable(value: never): never {
+	throw new UlwLoopError(`Unhandled operation: ${String(value)}`, "ULW_LOOP_OPERATION_UNHANDLED");
+}
+
+function nextActionsFrom(result: object): readonly string[] {
+	if (!("nextActions" in result) || !Array.isArray(result.nextActions)) return [];
+	return result.nextActions.filter((action): action is string => typeof action === "string").slice(0, 8);
+}
+
+function checkpointWithValidatedSnapshot(
+	context: ToolkitContext,
+	scope: UlwLoopScope,
+	args: Exclude<CheckpointArgs, { readonly printTemplate: true }>,
+): ReturnType<typeof checkpointUlwLoop> {
+	validateCodexGoalJson(args.codexGoalJson);
+	return checkpointUlwLoop(context.cwd, args, scope, { surface: context.surface });
+}
+
+export function createAgentToolkit(context: ToolkitContext, deps: AgentToolkitDependencies = {}): AgentToolkit {
+	validateContext(context);
+	const scope: UlwLoopScope = { sessionId: context.sessionId };
+	const notify = async <Operation extends UlwLoopOperation>(
+		operation: Operation,
+		response: ToolkitResponseFor<Operation>,
+	): Promise<ToolkitResponseFor<Operation>> => {
+		if (deps.hooks?.onOperation !== undefined) await deps.hooks.onOperation({ operation, context, response });
+		return response;
+	};
+	const invoke = async <Operation extends UlwLoopOperation>(
+		operation: Operation,
+		fn: () => Promise<ToolkitResultFor<Operation>>,
+	): Promise<ToolkitResponseFor<Operation>> => {
+		try {
+			const result = await fn();
+			const nextActions = typeof result === "object" && result !== null ? nextActionsFrom(result) : [];
+			return await notify(operation, { ok: true, operation, result, nextActions });
+		} catch (error) {
+			const response = caught(operation, error instanceof Error ? error : new Error("ULW_LOOP_ERROR"));
+			return notify(operation, response);
+		}
+	};
+	const toolkit: AgentToolkit = {
+		dispatch: async (request): Promise<ToolkitDispatchResponse> => {
+			if (!isKnownRequest(request))
+				return failure(
+					request.operation,
+					new UlwLoopError(`Unknown operation: ${request.operation}`, "ULW_LOOP_OPERATION_UNKNOWN"),
+				);
+			switch (request.operation) {
+				case "help":
+					return toolkit.help();
+				case "create-goals":
+					return toolkit.createGoals(request.args);
+				case "status":
+					return toolkit.status();
+				case "complete-goals":
+					return toolkit.completeGoals(request.args);
+				case "checkpoint":
+					return toolkit.checkpoint(request.args);
+				case "steer":
+					return toolkit.steer(request.args);
+				case "add-goal":
+					return toolkit.addGoal(request.args);
+				case "criteria":
+					return toolkit.criteria(request.args);
+				case "record-evidence":
+					return toolkit.recordEvidence(request.args);
+				case "record-review-blockers":
+					return toolkit.recordReviewBlockers(request.args);
+				default:
+					return unreachable(request);
+			}
+		},
+		help: () => invoke("help", async () => ULW_LOOP_MANIFEST),
+		createGoals: (args) => invoke("create-goals", () => createUlwLoopPlan(context.cwd, args, scope)),
+		status: () =>
+			invoke("status", async () => {
+				const plan = await readUlwLoopPlan(context.cwd, scope);
+				const active = plan.goals.find((goal) => goal.id === plan.activeGoalId);
+				return {
+					plan,
+					summary: summarizeUlwLoopPlan(plan),
+					nextActions: statusNextActions(plan),
+					// Attempt directories are an evidence-layout v2 concept; a v1 plan must not advertise one.
+					...(active === undefined || plan.evidenceLayoutVersion !== 2
+						? {}
+						: { currentAttemptDir: ulwLoopAttemptEvidenceDir(active.id, active.attempt, scope) }),
+				};
+			}),
+		completeGoals: (args = {}) => invoke("complete-goals", () => startNextUlwLoop(context.cwd, args, scope)),
+		checkpoint: (args: CheckpointArgs) =>
+			invoke("checkpoint", () =>
+				args.printTemplate === true
+					? checkpointTemplate(context.cwd, scope, args.goalId, { surface: context.surface })
+					: checkpointWithValidatedSnapshot(context, scope, args),
+			),
+		steer: (args) => invoke("steer", () => steerUlwLoop(context.cwd, args, scope)),
+		addGoal: (args) => invoke("add-goal", () => addUlwLoopGoal(context.cwd, args, scope)),
+		criteria: (args) =>
+			invoke("criteria", async () => {
+				const plan = await readUlwLoopPlan(context.cwd, scope);
+				const goal = plan.goals.find((candidate) => candidate.id === args.goalId);
+				if (goal === undefined)
+					throw new UlwLoopError(`Unknown ulw-loop id: ${args.goalId}.`, "ULW_LOOP_GOAL_NOT_FOUND");
+				return { goalId: goal.id, criteria: goal.successCriteria };
+			}),
+		recordEvidence: (args) => invoke("record-evidence", () => recordEvidence(context.cwd, args, scope)),
+		recordReviewBlockers: (args) =>
+			invoke("record-review-blockers", () => recordFinalReviewBlockers(context.cwd, args, scope)),
+	};
+	return toolkit;
+}

package/packages/omo-codex/plugin/components/ulw-loop/src/sdk/manifest.ts

@@ -0,0 +1,42 @@
+export const ULW_LOOP_OPERATIONS = [
+	"help",
+	"create-goals",
+	"status",
+	"complete-goals",
+	"checkpoint",
+	"steer",
+	"add-goal",
+	"criteria",
+	"record-evidence",
+	"record-review-blockers",
+] as const;
+
+export type UlwLoopOperation = (typeof ULW_LOOP_OPERATIONS)[number];
+
+export interface ToolkitOperationManifest {
+	readonly name: UlwLoopOperation;
+	readonly mutating: boolean;
+}
+
+export interface ToolkitManifest {
+	readonly version: 1;
+	readonly name: "ulw-loop";
+	readonly operations: readonly ToolkitOperationManifest[];
+}
+
+export const ULW_LOOP_MANIFEST = {
+	version: 1,
+	name: "ulw-loop",
+	operations: [
+		{ name: "help", mutating: false },
+		{ name: "create-goals", mutating: true },
+		{ name: "status", mutating: false },
+		{ name: "complete-goals", mutating: true },
+		{ name: "checkpoint", mutating: true },
+		{ name: "steer", mutating: true },
+		{ name: "add-goal", mutating: true },
+		{ name: "criteria", mutating: false },
+		{ name: "record-evidence", mutating: true },
+		{ name: "record-review-blockers", mutating: true },
+	] satisfies readonly ToolkitOperationManifest[],
+} satisfies ToolkitManifest;

package/packages/omo-codex/plugin/components/ulw-loop/src/sdk/types.ts

@@ -0,0 +1,140 @@
+import type { CheckpointUlwLoopArgs, CheckpointUlwLoopResult } from "../checkpoint.js";
+import type { CheckpointTemplate } from "../checkpoint-template.js";
+import type { recordEvidence } from "../evidence.js";
+import type { addUlwLoopGoal, createUlwLoopPlan, startNextUlwLoop, summarizeUlwLoopPlan } from "../plan-crud.js";
+import type { recordFinalReviewBlockers } from "../review-blockers.js";
+import type { steerUlwLoop } from "../steering.js";
+import type { UlwLoopSteeringProposal } from "../steering-types.js";
+import type { UlwLoopToolkitSurface } from "../surface.js";
+import type { UlwLoopPlan } from "../types.js";
+import type { ULW_LOOP_MANIFEST, UlwLoopOperation } from "./manifest.js";
+
+export type ToolkitSurface = UlwLoopToolkitSurface;
+
+export interface ToolkitContext {
+	readonly cwd: string;
+	readonly sessionId: string;
+	readonly surface: ToolkitSurface;
+}
+
+export type CreateGoalsArgs = Parameters<typeof createUlwLoopPlan>[1];
+export type CompleteGoalsArgs = NonNullable<Parameters<typeof startNextUlwLoop>[1]>;
+export type SteerArgs = UlwLoopSteeringProposal;
+export type AddGoalArgs = Parameters<typeof addUlwLoopGoal>[1];
+export type RecordEvidenceArgs = Parameters<typeof recordEvidence>[1];
+export type CriteriaArgs = { readonly goalId: string };
+export type CheckpointMutationArgs = CheckpointUlwLoopArgs & { readonly printTemplate?: false };
+export type CheckpointTemplateArgs = { readonly printTemplate: true; readonly goalId?: string };
+export type CheckpointArgs = CheckpointMutationArgs | CheckpointTemplateArgs;
+export type RecordReviewBlockersArgs = Parameters<typeof recordFinalReviewBlockers>[1];
+
+export type ToolkitDispatchRequest =
+	| { readonly operation: "help"; readonly args?: Record<never, never> }
+	| { readonly operation: "create-goals"; readonly args: CreateGoalsArgs }
+	| { readonly operation: "status"; readonly args?: Record<never, never> }
+	| { readonly operation: "complete-goals"; readonly args?: CompleteGoalsArgs }
+	| { readonly operation: "checkpoint"; readonly args: CheckpointArgs }
+	| { readonly operation: "steer"; readonly args: SteerArgs }
+	| { readonly operation: "add-goal"; readonly args: AddGoalArgs }
+	| { readonly operation: "criteria"; readonly args: CriteriaArgs }
+	| { readonly operation: "record-evidence"; readonly args: RecordEvidenceArgs }
+	| { readonly operation: "record-review-blockers"; readonly args: RecordReviewBlockersArgs };
+
+export interface ToolkitUnknownRequest {
+	readonly operation: string;
+	readonly args?: Record<never, never>;
+}
+
+export interface ToolkitError {
+	readonly code: string;
+	readonly message: string;
+	readonly details?: Readonly<Record<string, string>>;
+}
+
+export interface ToolkitSuccess<Operation extends UlwLoopOperation, Result> {
+	readonly ok: true;
+	readonly operation: Operation;
+	readonly result: Result;
+	readonly nextActions: readonly string[];
+	readonly warnings?: readonly string[];
+}
+
+export interface ToolkitFailure<Operation extends string = string> {
+	readonly ok: false;
+	readonly operation: Operation;
+	readonly error: ToolkitError;
+}
+
+type StatusResult = {
+	readonly plan: UlwLoopPlan;
+	readonly summary: ReturnType<typeof summarizeUlwLoopPlan>;
+	readonly nextActions: readonly string[];
+	readonly currentAttemptDir?: string;
+};
+
+type ResultFor<Operation extends UlwLoopOperation> = Operation extends "help"
+	? typeof ULW_LOOP_MANIFEST
+	: Operation extends "create-goals"
+		? Awaited<ReturnType<typeof createUlwLoopPlan>>
+		: Operation extends "status"
+			? StatusResult
+			: Operation extends "complete-goals"
+				? Awaited<ReturnType<typeof startNextUlwLoop>>
+				: Operation extends "checkpoint"
+					? CheckpointUlwLoopResult | CheckpointTemplate
+					: Operation extends "steer"
+						? Awaited<ReturnType<typeof steerUlwLoop>>
+						: Operation extends "add-goal"
+							? Awaited<ReturnType<typeof addUlwLoopGoal>>
+							: Operation extends "criteria"
+								? {
+										readonly goalId: string;
+										readonly criteria: UlwLoopPlan["goals"][number]["successCriteria"];
+									}
+								: Operation extends "record-evidence"
+									? Awaited<ReturnType<typeof recordEvidence>>
+									: Operation extends "record-review-blockers"
+										? Awaited<ReturnType<typeof recordFinalReviewBlockers>>
+										: never;
+
+export type ToolkitResultFor<Operation extends UlwLoopOperation> = ResultFor<Operation>;
+export type ToolkitResponseFor<Operation extends UlwLoopOperation> =
+	| ToolkitSuccess<Operation, ResultFor<Operation>>
+	| ToolkitFailure<Operation>;
+export type ToolkitOperationResponse = {
+	[Operation in UlwLoopOperation]: ToolkitResponseFor<Operation>;
+}[UlwLoopOperation];
+export type ToolkitDispatchResponse = ToolkitOperationResponse | ToolkitFailure<string>;
+export type ToolkitResponse<Result> = ToolkitSuccess<UlwLoopOperation, Result> | ToolkitFailure<string>;
+
+export interface ToolkitOperationHookEvent<Operation extends UlwLoopOperation = UlwLoopOperation> {
+	readonly operation: Operation;
+	readonly context: ToolkitContext;
+	readonly response: ToolkitResponseFor<Operation>;
+}
+
+export interface ToolkitHostHooks {
+	readonly onOperation?: <Operation extends UlwLoopOperation>(
+		event: ToolkitOperationHookEvent<Operation>,
+	) => void | Promise<void>;
+}
+
+export interface AgentToolkitDependencies {
+	readonly hooks?: ToolkitHostHooks;
+}
+
+export interface AgentToolkit {
+	readonly dispatch: (request: ToolkitDispatchRequest | ToolkitUnknownRequest) => Promise<ToolkitDispatchResponse>;
+	readonly help: () => Promise<ToolkitResponseFor<"help">>;
+	readonly createGoals: (args: CreateGoalsArgs) => Promise<ToolkitResponseFor<"create-goals">>;
+	readonly status: () => Promise<ToolkitResponseFor<"status">>;
+	readonly completeGoals: (args?: CompleteGoalsArgs) => Promise<ToolkitResponseFor<"complete-goals">>;
+	readonly checkpoint: (args: CheckpointArgs) => Promise<ToolkitResponseFor<"checkpoint">>;
+	readonly steer: (args: SteerArgs) => Promise<ToolkitResponseFor<"steer">>;
+	readonly addGoal: (args: AddGoalArgs) => Promise<ToolkitResponseFor<"add-goal">>;
+	readonly criteria: (args: CriteriaArgs) => Promise<ToolkitResponseFor<"criteria">>;
+	readonly recordEvidence: (args: RecordEvidenceArgs) => Promise<ToolkitResponseFor<"record-evidence">>;
+	readonly recordReviewBlockers: (
+		args: RecordReviewBlockersArgs,
+	) => Promise<ToolkitResponseFor<"record-review-blockers">>;
+}

package/packages/omo-codex/plugin/components/ulw-loop/src/sdk.ts

@@ -0,0 +1,3 @@
+export { createAgentToolkit } from "./sdk/factory.js";
+export * from "./sdk/manifest.js";
+export * from "./sdk/types.js";