@ahmadposten/talos-wire 0.1.2 → 0.1.4

package/dist/index.mjs

@@ -352,22 +352,39 @@ const WorkflowAgentSchema = z$1.object({
   revision: z$1.number().int().positive(),
   name: z$1.string().min(1).max(60),
   description: z$1.string().max(300),
-  provider: z$1.literal("codex"),
+  provider: z$1.enum(["codex", "claude", "muse"]),
   model: z$1.string().min(1).max(200),
+  modelLabel: z$1.string().max(300).optional(),
   effort: z$1.string().max(30).nullable(),
   permissionMode: z$1.enum(["default", "read-only"]),
   instructions: text,
   documents: z$1.array(z$1.object({ name: z$1.string().max(120), content: z$1.string().max(16e3) })).max(5)
 });
 const WorkflowSlotSchema = z$1.object({ agent: WorkflowAgentSchema, assignment: text });
+const WorkflowStepSchema = z$1.object({
+  id: z$1.string().uuid(),
+  name: z$1.string().trim().min(1).max(80),
+  kind: z$1.enum(["plan", "execute", "review"]),
+  agents: z$1.array(WorkflowSlotSchema).min(1).max(3),
+  criteria: z$1.string().trim().max(24e3),
+  checks: z$1.array(z$1.object({ name: z$1.string().trim().min(1).max(100), command: z$1.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$1.object({
   id: z$1.string().uuid(),
   revision: z$1.number().int().positive(),
   name: z$1.string().trim().min(1).max(80),
   description: z$1.string().max(1e3),
-  planners: z$1.array(WorkflowSlotSchema).min(2).max(4),
+  planners: z$1.array(WorkflowSlotSchema).min(1).max(4),
   executor: WorkflowSlotSchema,
-  reviewers: z$1.array(WorkflowSlotSchema).min(2).max(4),
+  reviewers: z$1.array(WorkflowSlotSchema).min(1).max(4),
+  steps: z$1.array(WorkflowStepSchema).min(3).max(8).optional(),
   criteria: text,
   checks: z$1.array(z$1.object({ name: z$1.string().trim().min(1).max(100), command: z$1.string().trim().min(1).max(2e3) })).min(1).max(8),
   planningRounds: z$1.number().int().min(1).max(5),
@@ -378,11 +395,40 @@ const WorkflowDefinitionSchema = z$1.object({
   updatedAt: z$1.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$1.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$1.enum(["propose", "consolidate", "plan_vote", "execute", "review", "verify"]);
 const WorkflowDecisionSchema = z$1.object({
   decision: z$1.enum(["approve", "changes", "information", "replan"]),
@@ -396,6 +442,8 @@ const WorkflowTaskSchema = z$1.object({
   round: z$1.number(),
   agentId: z$1.string(),
   agentName: z$1.string(),
+  stepId: z$1.string().uuid().optional(),
+  attempt: z$1.number().int().positive().optional(),
   clarifications: z$1.number().int().optional(),
   assignment: z$1.string(),
   version: z$1.string(),
@@ -421,6 +469,10 @@ const WorkflowRunSchema = z$1.object({
   baseCommit: z$1.string(),
   status: z$1.enum(["running", "paused", "needs_input", "complete", "cancelled"]),
   stage: WorkflowStageSchema,
+  stepIndex: z$1.number().int().nonnegative().optional(),
+  stepAttempt: z$1.number().int().positive().optional(),
+  stepRounds: z$1.record(z$1.string(), z$1.number().int().positive()).optional(),
+  completedSteps: z$1.array(z$1.string().uuid()).max(8).optional(),
   planningRound: z$1.number().int(),
   reviewRound: z$1.number().int(),
   planVersion: z$1.number().int(),
@@ -434,6 +486,16 @@ const WorkflowRunSchema = z$1.object({
   approvedPlanVersion: z$1.number().nullable(),
   createdAt: z$1.number(),
   updatedAt: z$1.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$1.object({ id: z$1.string().uuid(), definition: WorkflowDefinitionSchema, task: text, directory: z$1.string().min(1).max(4e3) });
 const WorkflowActionSchema = z$1.object({
@@ -448,5 +510,8 @@ 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");
+}
 
-export { AgentMessageSchema, ApiMessageSchema, ApiUpdateMachineStateSchema, ApiUpdateNewMessageSchema, ApiUpdateSessionStateSchema, CoreUpdateBodySchema, CoreUpdateContainerSchema, LegacyMessageContentSchema, MessageContentSchema, MessageMetaSchema, ProviderUsageBalanceSchema, ProviderUsageRequestSchema, ProviderUsageSnapshotSchema, ProviderUsageWindowSchema, SessionMessageContentSchema, SessionMessageSchema, SessionProtocolMessageSchema, UpdateBodySchema, UpdateMachineBodySchema, UpdateNewMessageBodySchema, UpdateSchema, UpdateSessionBodySchema, UsageProviderSchema, UserMessageSchema, VersionedEncryptedValueSchema, VersionedMachineEncryptedValueSchema, VersionedNullableEncryptedValueSchema, VoiceConversationDeniedSchema, VoiceConversationGrantedSchema, VoiceConversationResponseSchema, VoiceUsageResponseSchema, WorkflowActionSchema, WorkflowAgentSchema, WorkflowDecisionSchema, WorkflowDefinitionSchema, WorkflowLibrarySchema, WorkflowRunSchema, WorkflowSlotSchema, WorkflowStageSchema, WorkflowStartSchema, WorkflowTaskSchema, authenticationContexts, createEnvelope, encryptionContexts, legacyInstallation, legacyServerBanner, normalizeMetadata, rpcMethods, sessionEnvelopeSchema, sessionEventSchema, sessionFileEventSchema, sessionRoleSchema, sessionServiceMessageEventSchema, sessionStartEventSchema, sessionStopEventSchema, sessionTextEventSchema, sessionToolCallEndEventSchema, sessionToolCallStartEventSchema, sessionTurnEndEventSchema, sessionTurnEndStatusSchema, sessionTurnStartEventSchema, toWireMetadata, workflowEnabled, workflowStageLabel };
+export { AgentMessageSchema, ApiMessageSchema, ApiUpdateMachineStateSchema, ApiUpdateNewMessageSchema, ApiUpdateSessionStateSchema, CoreUpdateBodySchema, CoreUpdateContainerSchema, LegacyMessageContentSchema, MessageContentSchema, MessageMetaSchema, ProviderUsageBalanceSchema, ProviderUsageRequestSchema, ProviderUsageSnapshotSchema, ProviderUsageWindowSchema, SessionMessageContentSchema, SessionMessageSchema, SessionProtocolMessageSchema, UpdateBodySchema, UpdateMachineBodySchema, UpdateNewMessageBodySchema, UpdateSchema, UpdateSessionBodySchema, UsageProviderSchema, UserMessageSchema, VersionedEncryptedValueSchema, VersionedMachineEncryptedValueSchema, VersionedNullableEncryptedValueSchema, VoiceConversationDeniedSchema, VoiceConversationGrantedSchema, VoiceConversationResponseSchema, VoiceUsageResponseSchema, WorkflowActionSchema, WorkflowAgentSchema, WorkflowDecisionSchema, WorkflowDefinitionSchema, WorkflowLibrarySchema, WorkflowRunSchema, WorkflowSlotSchema, WorkflowStageSchema, WorkflowStartSchema, WorkflowStepSchema, WorkflowTaskSchema, authenticationContexts, createEnvelope, encryptionContexts, legacyInstallation, legacyServerBanner, normalizeMetadata, rpcMethods, sessionEnvelopeSchema, sessionEventSchema, sessionFileEventSchema, sessionRoleSchema, sessionServiceMessageEventSchema, sessionStartEventSchema, sessionStopEventSchema, sessionTextEventSchema, sessionToolCallEndEventSchema, sessionToolCallStartEventSchema, sessionTurnEndEventSchema, sessionTurnEndStatusSchema, sessionTurnStartEventSchema, toWireMetadata, workflowEnabled, workflowNeedsProviders, workflowProjection, workflowSlots, workflowStageLabel };

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@ahmadposten/talos-wire",
-  "version": "0.1.2",
+  "version": "0.1.4",
   "description": "Shared message wire types and Zod schemas for Talos clients and services",
   "author": "Kirill Dubovitskiy",
   "license": "MIT",