Skip to content
Sign in

flagon-io/g1t

public
0
154 linesCodeBlameRaw
1/**
2 * The agent form (components/agents-mode.tsx) as the agents service takes
3 * it: what each field means, and the checks made before it is sent. Pure,
4 * so it is tested on its own.
5 */
6import type { ModelTier, NewWorkspaceAgent, PersonalityPreset, WorkspaceAgent } from "@g1t/contracts";
7
8/** The tiers, cheapest first: `MODEL_TIERS` in the contracts. */
9const MODEL_TIERS: ModelTier[] = ["small", "large", "frontier"];
10
11export const PRESETS: { value: PersonalityPreset; label: string; about: string }[] = [
12 { value: "crisp", label: "Crisp", about: "Clear and to the point, with enough context to act on." },
13 { value: "friendly", label: "Friendly", about: "Warm and encouraging; explains as it goes." },
14 { value: "socratic", label: "Socratic", about: "Asks before it assumes; draws out what you mean." },
15 { value: "terse", label: "Terse operator", about: "As few words as will do. Status, then done." },
16];
17
18export const TIER_LABELS: Record<ModelTier, string> = {
19 small: "Fast: replies and triage",
20 large: "Standard: making changes, most reviews",
21 frontier: "Most capable: planning and hard work",
22};
23
24export const AUTONOMY = {
25 open_pull_requests: { label: "Open pull requests", options: [["alone", "On its own"], ["approval", "After approval"]] },
26 merge: { label: "Merge", options: [["alone", "On its own"], ["approval", "After approval"], ["never", "Never"]] },
27 deploy_production: { label: "Deploy to production", options: [["approval", "After approval"], ["never", "Never"]] },
28 edit_docs: { label: "Edit docs", options: [["alone", "Directly"], ["suggest", "As suggestions"]] },
29} as const;
30
31/** The handle as people type it: lowercase letters, digits and dashes. */
32export function cleanHandle(typed: string): string {
33 return typed
34 .trim()
35 .toLowerCase()
36 .replace(/^@/, "")
37 .replace(/[\s_]+/g, "-")
38 .replace(/[^a-z0-9-]/g, "")
39 .replace(/-{2,}/g, "-")
40 .replace(/^-|-$/g, "")
41 .slice(0, 39);
42}
43
44/** Handles the site's own addresses use, and g1t's: never an agent's. */
45const RESERVED_HANDLES = new Set(["g1t", "new", "runs", "fleet"]);
46
47function tier(value: FormDataEntryValue | null): ModelTier | null {
48 const text = String(value ?? "");
49 return (MODEL_TIERS as string[]).includes(text) ? (text as ModelTier) : null;
50}
51
52function pick<T extends string>(value: FormDataEntryValue | null, allowed: readonly (readonly [T, string])[], fallback: T): T {
53 const text = String(value ?? "");
54 return (allowed.find(([key]) => key === text)?.[0] ?? fallback) as T;
55}
56
57export type AgentFormResult = { ok: true; input: NewWorkspaceAgent } | { ok: false; errors: Record<string, string> };
58
59/** The form, checked: the agent to make or the changes to save, or what to fix, by field. */
60export function readAgentForm(form: FormData): AgentFormResult {
61 const errors: Record<string, string> = {};
62 const display_name = String(form.get("display_name") ?? "").trim();
63 const handle = cleanHandle(String(form.get("handle") ?? "") || display_name);
64 const role = String(form.get("role") ?? "").trim();
65 const instructions = String(form.get("instructions") ?? "").trim();
66 if (!display_name) errors.display_name = "Give it a name.";
67 if (!handle) errors.handle = "Give it a handle, like @ship.";
68 else if (RESERVED_HANDLES.has(handle)) errors.handle = `@${handle} is taken by g1t.`;
69 if (!role) errors.role = "Say in a line what it does.";
70 if (!instructions) errors.instructions = "Tell it what it's responsible for.";
71 const floor = tier(form.get("floor"));
72 const ceiling = tier(form.get("ceiling"));
73 if (floor && ceiling && MODEL_TIERS.indexOf(floor) > MODEL_TIERS.indexOf(ceiling)) errors.ceiling = "The ceiling can't be below the floor.";
74 const budget = {
75 monthly_micros: microsFromDollars(form.get("monthly")),
76 daily_micros: microsFromDollars(form.get("daily")),
77 task_micros: microsFromDollars(form.get("task")),
78 };
79 for (const [field, value] of [["monthly", budget.monthly_micros], ["daily", budget.daily_micros], ["task", budget.task_micros]] as const) {
80 if (value != null && Number.isNaN(value)) errors[field] = "A sum in dollars, like 25.";
81 }
82 const capacity = Number(form.get("capacity") ?? 3);
83 if (!Number.isInteger(capacity) || capacity < 1 || capacity > 20) errors.capacity = "Between 1 and 20.";
84 const providers = String(form.get("providers") ?? "any");
85 const pinned = String(form.get("pinned") ?? "").trim();
86 if (pinned && !/^[\w.-]+\/[\w.:@-]+$/.test(pinned)) errors.pinned = "As provider/model.";
87 if (Object.keys(errors).length > 0) return { ok: false, errors };
88 return {
89 ok: true,
90 input: {
91 handle,
92 display_name,
93 role,
94 instructions,
95 personality_preset: pick(form.get("personality_preset"), PRESETS.map((p) => [p.value, p.label] as const), "crisp"),
96 personality: String(form.get("personality") ?? "").trim(),
97 routing: {
98 floor,
99 ceiling,
100 providers: providers === "any" ? [] : [providers],
101 pinned: pinned || null,
102 },
103 budget,
104 autonomy: {
105 open_pull_requests: pick(form.get("open_pull_requests"), AUTONOMY.open_pull_requests.options, "alone"),
106 merge: pick(form.get("merge"), AUTONOMY.merge.options, "approval"),
107 deploy_production: pick(form.get("deploy_production"), AUTONOMY.deploy_production.options, "approval"),
108 edit_docs: pick(form.get("edit_docs"), AUTONOMY.edit_docs.options, "suggest"),
109 },
110 capacity,
111 template: String(form.get("template") ?? "") || null,
112 },
113 };
114}
115
116/** What the form starts from: an agent being edited, a template, or nothing. */
117export type AgentDraft = Pick<WorkspaceAgent, "handle" | "display_name" | "role" | "instructions" | "personality_preset" | "personality" | "routing" | "budget" | "autonomy" | "capacity" | "template">;
118
119export const BLANK_DRAFT: AgentDraft = {
120 handle: "",
121 display_name: "",
122 role: "",
123 instructions: "",
124 personality_preset: "crisp",
125 personality: "",
126 routing: { floor: null, ceiling: null, providers: [], pinned: null },
127 budget: { monthly_micros: 50_000_000, daily_micros: null, task_micros: 5_000_000 },
128 autonomy: { open_pull_requests: "alone", merge: "approval", deploy_production: "approval", edit_docs: "suggest" },
129 capacity: 3,
130 template: null,
131};
132
133/** Dollars from micro-dollars, for a form field: empty for none. */
134export function dollarsField(micros: number | null | undefined): string {
135 if (micros == null) return "";
136 const dollars = micros / 1_000_000;
137 return Number.isInteger(dollars) ? String(dollars) : dollars.toFixed(2);
138}
139
140/** Micro-dollars from a form field in dollars: null when empty, NaN when not a sum. */
141export function microsFromDollars(value: FormDataEntryValue | null): number | null {
142 const text = String(value ?? "").trim().replace(/^\$/, "").replaceAll(",", "");
143 if (!text) return null;
144 const dollars = Number(text);
145 if (!Number.isFinite(dollars) || dollars < 0) return Number.NaN;
146 return Math.round(dollars * 1_000_000);
147}
148
149/** "$12.40" from micro-dollars. */
150export function formatDollars(micros: number): string {
151 const dollars = micros / 1_000_000;
152 return `$${dollars < 10 && dollars > 0 && !Number.isInteger(dollars) ? dollars.toFixed(2) : dollars.toLocaleString("en-US", { maximumFractionDigits: 2, minimumFractionDigits: Number.isInteger(dollars) ? 0 : 2 })}`;
153}
154