@ahmadposten/talos-wire 0.1.2 → 0.1.4

package/dist/index.cjs

@@ -372,22 +372,39 @@ const WorkflowAgentSchema = z.z.object({
   revision: z.z.number().int().positive(),
   name: z.z.string().min(1).max(60),
   description: z.z.string().max(300),
-  provider: z.z.literal("codex"),
+  provider: z.z.enum(["codex", "claude", "muse"]),
   model: z.z.string().min(1).max(200),
+  modelLabel: z.z.string().max(300).optional(),
   effort: z.z.string().max(30).nullable(),
   permissionMode: z.z.enum(["default", "read-only"]),
   instructions: text,
   documents: z.z.array(z.z.object({ name: z.z.string().max(120), content: z.z.string().max(16e3) })).max(5)
 });
 const WorkflowSlotSchema = z.z.object({ agent: WorkflowAgentSchema, assignment: text });
+const WorkflowStepSchema = z.z.object({
+  id: z.z.string().uuid(),
+  name: z.z.string().trim().min(1).max(80),
+  kind: z.z.enum(["plan", "execute", "review"]),
+  agents: z.z.array(WorkflowSlotSchema).min(1).max(3),
+  criteria: z.z.string().trim().max(24e3),
+  checks: z.z.array(z.z.object({ name: z.z.string().trim().min(1).max(100), command: z.z.string().trim().min(1).max(2e3) })).max(8)
+});
+function workflowProjection(steps) {
+  return {
+    planners: steps.find((s) => s.kind === "plan")?.agents ?? [],
+    executor: steps.find((s) => s.kind === "execute")?.agents[0],
+    reviewers: [...steps].reverse().find((s) => s.kind === "review")?.agents ?? []
+  };
+}
 const WorkflowDefinitionSchema = z.z.object({
   id: z.z.string().uuid(),
   revision: z.z.number().int().positive(),
   name: z.z.string().trim().min(1).max(80),
   description: z.z.string().max(1e3),
-  planners: z.z.array(WorkflowSlotSchema).min(2).max(4),
+  planners: z.z.array(WorkflowSlotSchema).min(1).max(4),
   executor: WorkflowSlotSchema,
-  reviewers: z.z.array(WorkflowSlotSchema).min(2).max(4),
+  reviewers: z.z.array(WorkflowSlotSchema).min(1).max(4),
+  steps: z.z.array(WorkflowStepSchema).min(3).max(8).optional(),
   criteria: text,
   checks: z.z.array(z.z.object({ name: z.z.string().trim().min(1).max(100), command: z.z.string().trim().min(1).max(2e3) })).min(1).max(8),
   planningRounds: z.z.number().int().min(1).max(5),
@@ -398,11 +415,40 @@ const WorkflowDefinitionSchema = z.z.object({
   updatedAt: z.z.number()
 }).superRefine((d, ctx) => {
   if (new TextEncoder().encode(JSON.stringify(d)).length > 64e3) ctx.addIssue({ code: "custom", message: "Workflow definition exceeds 64 KB. Shorten instructions or documents." });
-  const ids = [...d.planners, d.executor, ...d.reviewers].map((s) => s.agent.id);
-  if (new Set(ids).size !== ids.length) ctx.addIssue({ code: "custom", message: "Each workflow participant must be a different saved agent." });
-  if (d.executor.agent.permissionMode === "read-only") ctx.addIssue({ code: "custom", message: "The executor must allow workspace edits." });
+  const issue = (message) => ctx.addIssue({ code: "custom", message });
+  if (d.steps) {
+    if (d.steps[0].kind !== "plan" || d.steps.at(-1)?.kind !== "review" || !d.steps.some((s) => s.kind === "execute")) issue("Start with planning, include execution, and finish with review.");
+    if (new Set(d.steps.map((s) => s.id)).size !== d.steps.length) issue("Each step needs a unique ID.");
+    const writers = new Set(d.steps.filter((s) => s.kind === "execute").flatMap((s) => s.agents.map((a) => a.agent.id)));
+    const identities = /* @__PURE__ */ new Map();
+    let executed = false;
+    for (const step of d.steps) {
+      if (step.kind === "plan") executed = false;
+      if (step.kind === "execute") executed = true;
+      if (step.kind === "review" && !executed) issue("Place an execution step between planning and review.");
+      if (step.kind === "execute" && (step.agents.length !== 1 || step.agents[0].agent.permissionMode === "read-only")) issue("Each execution step needs one agent that allows workspace edits.");
+      if (step.kind !== "review" && step.checks.length) issue("Attach step checks to a review step.");
+      if (new Set(step.agents.map((s) => s.agent.id)).size !== step.agents.length) issue("Choose distinct agents within a consensus step.");
+      for (const slot of step.agents) {
+        if (step.kind === "review" && writers.has(slot.agent.id)) issue("Reviewers must be independent of every executor.");
+        const snapshot = JSON.stringify(slot.agent);
+        if (identities.has(slot.agent.id) && identities.get(slot.agent.id) !== snapshot) issue("A reused agent must have the same configuration in every step.");
+        identities.set(slot.agent.id, snapshot);
+      }
+    }
+    const projection = workflowProjection(d.steps);
+    if (JSON.stringify([d.planners, d.executor, d.reviewers]) !== JSON.stringify([projection.planners, projection.executor, projection.reviewers])) issue("Workflow role summaries must match its steps.");
+  } else {
+    if (d.planners.length < 2 || d.reviewers.length < 2) issue("Legacy workflows require at least two planners and reviewers.");
+    const ids = [...d.planners, d.executor, ...d.reviewers].map((s) => s.agent.id);
+    if (new Set(ids).size !== ids.length) issue("Each workflow participant must be a different saved agent.");
+    if (d.executor.agent.permissionMode === "read-only") issue("The executor must allow workspace edits.");
+  }
 });
 const WorkflowLibrarySchema = z.z.array(WorkflowDefinitionSchema).max(20).refine((x) => JSON.stringify(x).length < 128e3, "Workflow library is too large.");
+function workflowSlots(d) {
+  return d.steps?.flatMap((s) => s.agents) ?? [...d.planners, d.executor, ...d.reviewers];
+}
 const WorkflowStageSchema = z.z.enum(["propose", "consolidate", "plan_vote", "execute", "review", "verify"]);
 const WorkflowDecisionSchema = z.z.object({
   decision: z.z.enum(["approve", "changes", "information", "replan"]),
@@ -416,6 +462,8 @@ const WorkflowTaskSchema = z.z.object({
   round: z.z.number(),
   agentId: z.z.string(),
   agentName: z.z.string(),
+  stepId: z.z.string().uuid().optional(),
+  attempt: z.z.number().int().positive().optional(),
   clarifications: z.z.number().int().optional(),
   assignment: z.z.string(),
   version: z.z.string(),
@@ -441,6 +489,10 @@ const WorkflowRunSchema = z.z.object({
   baseCommit: z.z.string(),
   status: z.z.enum(["running", "paused", "needs_input", "complete", "cancelled"]),
   stage: WorkflowStageSchema,
+  stepIndex: z.z.number().int().nonnegative().optional(),
+  stepAttempt: z.z.number().int().positive().optional(),
+  stepRounds: z.z.record(z.z.string(), z.z.number().int().positive()).optional(),
+  completedSteps: z.z.array(z.z.string().uuid()).max(8).optional(),
   planningRound: z.z.number().int(),
   reviewRound: z.z.number().int(),
   planVersion: z.z.number().int(),
@@ -454,6 +506,16 @@ const WorkflowRunSchema = z.z.object({
   approvedPlanVersion: z.z.number().nullable(),
   createdAt: z.z.number(),
   updatedAt: z.z.number()
+}).superRefine((run, ctx) => {
+  if (!run.definition.steps) return;
+  const steps = run.definition.steps;
+  const step = steps[run.stepIndex ?? -1];
+  if (!step || !run.stepAttempt || !run.completedSteps || !run.stepRounds) {
+    ctx.addIssue({ code: "custom", message: "Editable workflow recovery state is incomplete." });
+    return;
+  }
+  const stages = step.kind === "plan" ? ["propose", "consolidate", "plan_vote"] : step.kind === "execute" ? ["execute"] : ["verify", "review"];
+  if (!stages.includes(run.stage) || run.completedSteps.some((id) => !steps.some((s) => s.id === id)) || run.tasks.some((t) => !t.stepId || !t.attempt || !steps.some((s) => s.id === t.stepId))) ctx.addIssue({ code: "custom", message: "Saved workflow stage does not match its definition." });
 });
 const WorkflowStartSchema = z.z.object({ id: z.z.string().uuid(), definition: WorkflowDefinitionSchema, task: text, directory: z.z.string().min(1).max(4e3) });
 const WorkflowActionSchema = z.z.object({
@@ -468,6 +530,9 @@ function workflowEnabled(s) {
   return s.experiments === true && s.expWorkflows === true;
 }
 const workflowStageLabel = { propose: "Independent proposals", consolidate: "Consolidating plan", plan_vote: "Planning consensus", execute: "Executing", review: "Independent review", verify: "Completion checks" };
+function workflowNeedsProviders(definition) {
+  return workflowSlots(definition).some((slot) => slot.agent.provider !== "codex");
+}
 
 exports.AgentMessageSchema = AgentMessageSchema;
 exports.ApiMessageSchema = ApiMessageSchema;
@@ -509,6 +574,7 @@ exports.WorkflowRunSchema = WorkflowRunSchema;
 exports.WorkflowSlotSchema = WorkflowSlotSchema;
 exports.WorkflowStageSchema = WorkflowStageSchema;
 exports.WorkflowStartSchema = WorkflowStartSchema;
+exports.WorkflowStepSchema = WorkflowStepSchema;
 exports.WorkflowTaskSchema = WorkflowTaskSchema;
 exports.authenticationContexts = authenticationContexts;
 exports.createEnvelope = createEnvelope;
@@ -532,4 +598,7 @@ exports.sessionTurnEndStatusSchema = sessionTurnEndStatusSchema;
 exports.sessionTurnStartEventSchema = sessionTurnStartEventSchema;
 exports.toWireMetadata = toWireMetadata;
 exports.workflowEnabled = workflowEnabled;
+exports.workflowNeedsProviders = workflowNeedsProviders;
+exports.workflowProjection = workflowProjection;
+exports.workflowSlots = workflowSlots;
 exports.workflowStageLabel = workflowStageLabel;