@ahmadposten/talos-wire 0.1.1 → 0.1.3

package/dist/index.mjs

@@ -1,4 +1,5 @@
 import * as z from 'zod';
+import { z as z$1 } from 'zod';
 import { isCuid, createId } from '@paralleldrive/cuid2';
 
 const sessionRoleSchema = z.enum(["user", "agent"]);
@@ -345,4 +346,168 @@ const ProviderUsageSnapshotSchema = z.object({
   retryAt: z.number().finite().optional()
 });
 
-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, authenticationContexts, createEnvelope, encryptionContexts, legacyInstallation, legacyServerBanner, normalizeMetadata, rpcMethods, sessionEnvelopeSchema, sessionEventSchema, sessionFileEventSchema, sessionRoleSchema, sessionServiceMessageEventSchema, sessionStartEventSchema, sessionStopEventSchema, sessionTextEventSchema, sessionToolCallEndEventSchema, sessionToolCallStartEventSchema, sessionTurnEndEventSchema, sessionTurnEndStatusSchema, sessionTurnStartEventSchema, toWireMetadata };
+const text = z$1.string().trim().min(1).max(24e3);
+const WorkflowAgentSchema = z$1.object({
+  id: z$1.string().min(1).max(100),
+  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"),
+  model: z$1.string().min(1).max(200),
+  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(1).max(4),
+  executor: WorkflowSlotSchema,
+  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),
+  reviewRounds: z$1.number().int().min(1).max(5),
+  turnMinutes: z$1.number().int().min(1).max(30),
+  maxTurns: z$1.number().int().min(8).max(100),
+  approvePlan: z$1.boolean(),
+  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 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"]),
+  summary: text,
+  document: z$1.string().max(24e3),
+  findings: z$1.array(z$1.object({ title: z$1.string().min(1).max(300), evidence: text, correction: text, blocking: z$1.boolean() })).max(20)
+});
+const WorkflowTaskSchema = z$1.object({
+  id: z$1.string(),
+  stage: WorkflowStageSchema,
+  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(),
+  status: z$1.enum(["running", "done", "interrupted"]),
+  startedAt: z$1.number(),
+  completedAt: z$1.number().optional(),
+  sessionId: z$1.string().optional(),
+  threadId: z$1.string().optional(),
+  prompt: z$1.string(),
+  result: WorkflowDecisionSchema.optional(),
+  error: z$1.string().optional()
+});
+const WorkflowRunSchema = z$1.object({
+  id: z$1.string().uuid(),
+  revision: z$1.number().int(),
+  definition: WorkflowDefinitionSchema,
+  machineId: z$1.string(),
+  task: text,
+  requestedDirectory: z$1.string().optional(),
+  sourceDirectory: z$1.string(),
+  directory: z$1.string(),
+  branch: z$1.string(),
+  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(),
+  plan: z$1.string(),
+  artifactVersion: z$1.string(),
+  reason: z$1.string(),
+  tasks: z$1.array(WorkflowTaskSchema).max(200),
+  checks: z$1.array(z$1.object({ name: z$1.string(), exitCode: z$1.number().nullable(), output: z$1.string(), version: z$1.string() })),
+  events: z$1.array(z$1.object({ at: z$1.number(), text: z$1.string() })).max(500),
+  notes: z$1.array(z$1.string()).max(30),
+  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({
+  id: z$1.string().uuid(),
+  expectedRevision: z$1.number().int(),
+  action: z$1.enum(["pause", "resume", "cancel", "approve_plan", "revise_plan", "retry_review", "replace_agent"]),
+  note: z$1.string().trim().max(24e3).default(""),
+  agentId: z$1.string().optional(),
+  replacement: WorkflowAgentSchema.optional()
+});
+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" };
+
+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, workflowProjection, workflowSlots, workflowStageLabel };

package/package.json

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