feat: add company builder wizard with AI-powered org generation (#73)
* feat: add company builder wizard with AI-powered org generation Adds a new "Build Your Company" step to the onboarding wizard that lets users describe their business and generates a full agent org structure using OpenClaw's AI. Includes company plan generation, role deduplication, agent bootstrap with main-agent reuse, org chart preview, confetti on success, CSS voxel running-avatar loader, amber theme unification, and best-effort SSH workspace cleanup. Made-with: Cursor * fix: resolve lint errors in CompanyBuilderModal Replace setState-in-effect pattern with a direct callback, escape apostrophes in JSX text, and derive org chart hover state without side effects. Made-with: Cursor --------- Co-authored-by: iamlukethedev <lucas.guilherme@smartwayslfl.com>
This commit is contained in:
Luke The Dev
@@ -0,0 +1,464 @@
|
||||
import {
|
||||
type CommandModeId,
|
||||
} from "@/features/agents/operations/agentPermissionsOperation";
|
||||
import {
|
||||
createEmptyPersonalityDraft,
|
||||
serializePersonalityFiles,
|
||||
type PersonalityBuilderDraft,
|
||||
} from "@/lib/agents/personalityBuilder";
|
||||
import type { AgentFileName } from "@/lib/agents/agentFiles";
|
||||
import type {
|
||||
CompanyAgentBlueprint,
|
||||
CompanyBuilderPlan,
|
||||
CompanyBuilderRole,
|
||||
CompanyBuilderStoredSnapshot,
|
||||
} from "@/features/company-builder/types";
|
||||
|
||||
type ParsedCompanyPlan = {
|
||||
companyName?: unknown;
|
||||
summary?: unknown;
|
||||
sharedRules?: unknown;
|
||||
plannerNotes?: unknown;
|
||||
roles?: unknown;
|
||||
};
|
||||
|
||||
type ParsedCompanyRole = {
|
||||
id?: unknown;
|
||||
name?: unknown;
|
||||
title?: unknown;
|
||||
purpose?: unknown;
|
||||
soul?: unknown;
|
||||
responsibilities?: unknown;
|
||||
collaborators?: unknown;
|
||||
tools?: unknown;
|
||||
heartbeat?: unknown;
|
||||
emoji?: unknown;
|
||||
creature?: unknown;
|
||||
vibe?: unknown;
|
||||
userContext?: unknown;
|
||||
commandMode?: unknown;
|
||||
};
|
||||
|
||||
const COMPANY_FENCE_RE = /^```(?:json)?\s*|\s*```$/gim;
|
||||
const MAX_ROLE_COUNT = 8;
|
||||
|
||||
const normalizeLine = (value: string) => value.replace(/\r\n/g, "\n").trim();
|
||||
|
||||
const coerceString = (value: unknown) => (typeof value === "string" ? value.trim() : "");
|
||||
|
||||
const coerceStringArray = (value: unknown): string[] => {
|
||||
if (!Array.isArray(value)) return [];
|
||||
return value
|
||||
.filter((entry): entry is string => typeof entry === "string")
|
||||
.map((entry) => entry.trim())
|
||||
.filter((entry) => entry.length > 0);
|
||||
};
|
||||
|
||||
const uniqueStrings = (values: string[]) => Array.from(new Set(values));
|
||||
|
||||
const slugify = (value: string) =>
|
||||
value
|
||||
.trim()
|
||||
.toLowerCase()
|
||||
.replace(/[^a-z0-9]+/g, "-")
|
||||
.replace(/^-+|-+$/g, "");
|
||||
|
||||
const toPascalCaseWord = (value: string) =>
|
||||
value
|
||||
.replace(/[^a-zA-Z0-9]+/g, " ")
|
||||
.split(/\s+/)
|
||||
.map((entry) => entry.trim())
|
||||
.filter((entry) => entry.length > 0)
|
||||
.map((entry) => `${entry.charAt(0).toUpperCase()}${entry.slice(1).toLowerCase()}`)
|
||||
.join("");
|
||||
|
||||
const normalizeRoleName = (value: string) => {
|
||||
const trimmed = value.trim();
|
||||
if (!trimmed) return "";
|
||||
const singleWord = trimmed.replace(/[^a-zA-Z0-9]/g, "");
|
||||
if (singleWord.length > 0 && !/\s/.test(trimmed)) {
|
||||
return singleWord.slice(0, 18);
|
||||
}
|
||||
const compact = toPascalCaseWord(trimmed);
|
||||
return compact.slice(0, 18);
|
||||
};
|
||||
|
||||
const dedupeCompactNames = (values: string[], fallbackPrefix: string) => {
|
||||
const used = new Set<string>();
|
||||
return values.map((value, index) => {
|
||||
const fallback = `${fallbackPrefix}${index + 1}`;
|
||||
const baseName = normalizeRoleName(value) || normalizeRoleName(fallback) || fallback;
|
||||
let nextName = baseName;
|
||||
let suffix = 2;
|
||||
while (used.has(nextName.toLowerCase())) {
|
||||
const suffixText = String(suffix);
|
||||
const trimmedBase = baseName.slice(0, Math.max(1, 18 - suffixText.length));
|
||||
nextName = `${trimmedBase}${suffixText}`;
|
||||
suffix += 1;
|
||||
}
|
||||
used.add(nextName.toLowerCase());
|
||||
return nextName;
|
||||
});
|
||||
};
|
||||
|
||||
const toSentenceList = (values: string[]) =>
|
||||
values
|
||||
.map((entry) => entry.trim())
|
||||
.filter((entry) => entry.length > 0)
|
||||
.map((entry) => (/[.!?]$/.test(entry) ? entry : `${entry}.`));
|
||||
|
||||
const resolveCommandMode = (value: unknown, roleText: string): CommandModeId => {
|
||||
const normalized = typeof value === "string" ? value.trim().toLowerCase() : "";
|
||||
if (normalized === "off" || normalized === "ask" || normalized === "auto") {
|
||||
return normalized;
|
||||
}
|
||||
const lowered = roleText.toLowerCase();
|
||||
if (/\b(developer|engineer|automation|devops|ops)\b/.test(lowered)) {
|
||||
return "auto";
|
||||
}
|
||||
if (/\b(manager|lead|qa|support|analyst|marketing|social)\b/.test(lowered)) {
|
||||
return "ask";
|
||||
}
|
||||
return "ask";
|
||||
};
|
||||
|
||||
const buildRoleIdentity = (role: CompanyBuilderRole) => ({
|
||||
emoji: role.emoji || "🤖",
|
||||
creature: role.creature || "specialist",
|
||||
vibe: role.vibe || "helpful and focused",
|
||||
});
|
||||
|
||||
const buildRoleAgentsMarkdown = (params: {
|
||||
plan: CompanyBuilderPlan;
|
||||
role: CompanyBuilderRole;
|
||||
}) => {
|
||||
const collaborators =
|
||||
params.role.collaborators.length > 0
|
||||
? params.role.collaborators
|
||||
: params.plan.roles
|
||||
.filter((entry) => entry.id !== params.role.id)
|
||||
.slice(0, 3)
|
||||
.map((entry) => entry.title);
|
||||
const responsibilityLines = toSentenceList(params.role.responsibilities);
|
||||
const sharedRules = toSentenceList(params.plan.sharedRules);
|
||||
const plannerNotes = toSentenceList(params.plan.plannerNotes);
|
||||
return [
|
||||
`# ${params.plan.companyName} Team Operating Guide`,
|
||||
"",
|
||||
`You are the ${params.role.title} inside ${params.plan.companyName}.`,
|
||||
"",
|
||||
"## Mission",
|
||||
"",
|
||||
params.role.purpose || `Own the ${params.role.title.toLowerCase()} function for the company.`,
|
||||
"",
|
||||
"## Responsibilities",
|
||||
"",
|
||||
...(responsibilityLines.length > 0
|
||||
? responsibilityLines.map((entry) => `- ${entry}`)
|
||||
: ["- Keep your area moving and surface blockers quickly."]),
|
||||
"",
|
||||
"## Collaborators",
|
||||
"",
|
||||
...(collaborators.length > 0
|
||||
? collaborators.map((entry) => `- Work closely with ${entry}.`)
|
||||
: ["- Coordinate with the rest of the company when work crosses team boundaries."]),
|
||||
"",
|
||||
"## Shared Rules",
|
||||
"",
|
||||
...(sharedRules.length > 0
|
||||
? sharedRules.map((entry) => `- ${entry}`)
|
||||
: [
|
||||
"- Keep updates concise, practical, and action-oriented.",
|
||||
"- Hand off work clearly when another role should take over.",
|
||||
]),
|
||||
"",
|
||||
"## Planning Notes",
|
||||
"",
|
||||
...(plannerNotes.length > 0
|
||||
? plannerNotes.map((entry) => `- ${entry}`)
|
||||
: ["- Treat the user's company brief as the source of truth."]),
|
||||
"",
|
||||
].join("\n");
|
||||
};
|
||||
|
||||
const buildRoleToolsMarkdown = (role: CompanyBuilderRole) =>
|
||||
[
|
||||
"# TOOLS.md",
|
||||
"",
|
||||
`Preferred operating mode: ${role.commandMode}.`,
|
||||
"",
|
||||
"## Tool Preferences",
|
||||
"",
|
||||
...(role.tools.length > 0
|
||||
? role.tools.map((entry) => `- ${entry}.`)
|
||||
: [
|
||||
"- Use the tools that best match your role.",
|
||||
"- Ask for help when another teammate has better context.",
|
||||
]),
|
||||
"",
|
||||
].join("\n");
|
||||
|
||||
const buildRoleHeartbeatMarkdown = (role: CompanyBuilderRole) =>
|
||||
[
|
||||
"# HEARTBEAT.md",
|
||||
"",
|
||||
"When your heartbeat runs:",
|
||||
"",
|
||||
...(role.heartbeat.length > 0
|
||||
? role.heartbeat.map((entry) => `- ${entry}.`)
|
||||
: [
|
||||
"- Check your most important queue or active work.",
|
||||
"- Report blockers before they become expensive.",
|
||||
"- Coordinate with collaborators if handoffs are waiting.",
|
||||
]),
|
||||
"",
|
||||
].join("\n");
|
||||
|
||||
const buildRoleMemoryMarkdown = (params: {
|
||||
plan: CompanyBuilderPlan;
|
||||
role: CompanyBuilderRole;
|
||||
}) =>
|
||||
[
|
||||
"# MEMORY.md",
|
||||
"",
|
||||
`Company: ${params.plan.companyName}.`,
|
||||
`Role: ${params.role.title}.`,
|
||||
"",
|
||||
"Remember:",
|
||||
"",
|
||||
`- ${params.plan.summary || "The company plan should guide your decisions."}`,
|
||||
`- ${params.role.purpose || "Protect the quality of your function."}`,
|
||||
"",
|
||||
].join("\n");
|
||||
|
||||
const buildRoleSoulDraft = (params: {
|
||||
plan: CompanyBuilderPlan;
|
||||
role: CompanyBuilderRole;
|
||||
}): PersonalityBuilderDraft => {
|
||||
const draft = createEmptyPersonalityDraft();
|
||||
const identity = buildRoleIdentity(params.role);
|
||||
draft.identity.name = params.role.title;
|
||||
draft.identity.emoji = identity.emoji;
|
||||
draft.identity.creature = identity.creature;
|
||||
draft.identity.vibe = identity.vibe;
|
||||
draft.user.context = normalizeLine(
|
||||
[
|
||||
`Company brief: ${params.plan.summary}`,
|
||||
params.role.userContext,
|
||||
]
|
||||
.filter((entry) => entry.trim().length > 0)
|
||||
.join("\n\n")
|
||||
);
|
||||
draft.soul.coreTruths = normalizeLine(
|
||||
[
|
||||
params.role.soul,
|
||||
`Your job is to help ${params.plan.companyName} succeed as the ${params.role.title}.`,
|
||||
]
|
||||
.filter((entry) => entry.trim().length > 0)
|
||||
.join("\n\n")
|
||||
);
|
||||
draft.soul.boundaries = normalizeLine(
|
||||
[
|
||||
"Do not invent decisions that should be handed to another specialist.",
|
||||
"Escalate blockers early and keep handoffs explicit.",
|
||||
].join("\n")
|
||||
);
|
||||
draft.soul.vibe = normalizeLine(
|
||||
params.role.vibe || `${params.role.title} energy: practical, collaborative, and sharp.`
|
||||
);
|
||||
draft.soul.continuity = normalizeLine(
|
||||
`Keep continuity around ${params.plan.companyName}'s goals, teammates, and operating rules.`
|
||||
);
|
||||
draft.agents = buildRoleAgentsMarkdown(params);
|
||||
draft.tools = buildRoleToolsMarkdown(params.role);
|
||||
draft.heartbeat = buildRoleHeartbeatMarkdown(params.role);
|
||||
draft.memory = buildRoleMemoryMarkdown(params);
|
||||
return draft;
|
||||
};
|
||||
|
||||
const normalizeRole = (value: ParsedCompanyRole, index: number): CompanyBuilderRole | null => {
|
||||
const title = normalizeRoleName(coerceString(value.name) || coerceString(value.title));
|
||||
if (!title) return null;
|
||||
const purpose = coerceString(value.purpose);
|
||||
const soul = coerceString(value.soul);
|
||||
const responsibilities = uniqueStrings(coerceStringArray(value.responsibilities)).slice(0, 8);
|
||||
const collaborators = uniqueStrings(coerceStringArray(value.collaborators)).slice(0, 8);
|
||||
const tools = uniqueStrings(coerceStringArray(value.tools)).slice(0, 8);
|
||||
const heartbeat = uniqueStrings(coerceStringArray(value.heartbeat)).slice(0, 8);
|
||||
const emoji = coerceString(value.emoji);
|
||||
const creature = coerceString(value.creature);
|
||||
const vibe = coerceString(value.vibe);
|
||||
const userContext = coerceString(value.userContext);
|
||||
const id = coerceString(value.id) || slugify(title) || `role-${index + 1}`;
|
||||
const roleText = [title, purpose, soul, ...responsibilities, ...tools].join(" ");
|
||||
return {
|
||||
id,
|
||||
title,
|
||||
purpose,
|
||||
soul,
|
||||
responsibilities,
|
||||
collaborators,
|
||||
tools,
|
||||
heartbeat,
|
||||
emoji,
|
||||
creature,
|
||||
vibe,
|
||||
userContext,
|
||||
commandMode: resolveCommandMode(value.commandMode, roleText),
|
||||
};
|
||||
};
|
||||
|
||||
export const buildImproveCompanyBriefPrompt = (businessDescription: string) =>
|
||||
[
|
||||
"You are helping a user describe the company they want to build inside Claw3D.",
|
||||
"Rewrite their brief so another OpenClaw agent can generate a clean org structure from it.",
|
||||
"Keep the answer short, concrete, and useful.",
|
||||
"Return markdown with these sections only:",
|
||||
"## Company",
|
||||
"## Goals",
|
||||
"## Constraints",
|
||||
"## Suggested Roles",
|
||||
"",
|
||||
"User brief:",
|
||||
businessDescription.trim(),
|
||||
].join("\n");
|
||||
|
||||
export const buildGenerateCompanyPlanPrompt = (brief: string) =>
|
||||
[
|
||||
"You are designing an AI company org structure for Claw3D.",
|
||||
"Return only valid JSON with no markdown fence.",
|
||||
"Each role name must be one concise word only with no spaces.",
|
||||
"Schema:",
|
||||
"{",
|
||||
' "companyName": "string",',
|
||||
' "summary": "string",',
|
||||
' "sharedRules": ["string"],',
|
||||
' "plannerNotes": ["string"],',
|
||||
' "roles": [',
|
||||
" {",
|
||||
' "id": "string",',
|
||||
' "name": "string",',
|
||||
' "purpose": "string",',
|
||||
' "soul": "string",',
|
||||
' "responsibilities": ["string"],',
|
||||
' "collaborators": ["string"],',
|
||||
' "tools": ["string"],',
|
||||
' "heartbeat": ["string"],',
|
||||
' "emoji": "string",',
|
||||
' "creature": "string",',
|
||||
' "vibe": "string",',
|
||||
' "userContext": "string",',
|
||||
' "commandMode": "off|ask|auto"',
|
||||
" }",
|
||||
" ]",
|
||||
"}",
|
||||
"Create between 2 and 6 roles unless the brief clearly needs more or less.",
|
||||
"Prefer silly but useful role titles when it helps the brand, but keep the org practical.",
|
||||
"Role names should be short single words like Builder, Analyst, Closer, Captain, Scout, or Designer.",
|
||||
"All role names must be unique.",
|
||||
"Make collaborators reference role names.",
|
||||
"",
|
||||
"Company brief:",
|
||||
brief.trim(),
|
||||
].join("\n");
|
||||
|
||||
export const extractJsonFromAssistantText = (value: string) => {
|
||||
const trimmed = value.trim();
|
||||
if (!trimmed) {
|
||||
throw new Error("The planning agent returned an empty response.");
|
||||
}
|
||||
const unfenced = trimmed.replace(COMPANY_FENCE_RE, "").trim();
|
||||
const firstBrace = unfenced.indexOf("{");
|
||||
const lastBrace = unfenced.lastIndexOf("}");
|
||||
if (firstBrace < 0 || lastBrace < firstBrace) {
|
||||
throw new Error("The planning agent did not return valid JSON.");
|
||||
}
|
||||
return unfenced.slice(firstBrace, lastBrace + 1);
|
||||
};
|
||||
|
||||
export const parseCompanyPlanFromAssistantText = (value: string): CompanyBuilderPlan => {
|
||||
let parsed: ParsedCompanyPlan;
|
||||
try {
|
||||
parsed = JSON.parse(extractJsonFromAssistantText(value)) as ParsedCompanyPlan;
|
||||
} catch (error) {
|
||||
if (error instanceof Error) {
|
||||
throw error;
|
||||
}
|
||||
throw new Error("Failed to parse the planning agent response.");
|
||||
}
|
||||
|
||||
const rolesRaw = Array.isArray(parsed.roles) ? parsed.roles : [];
|
||||
const normalizedRoles = rolesRaw
|
||||
.map((entry, index) => normalizeRole((entry ?? {}) as ParsedCompanyRole, index))
|
||||
.filter((entry): entry is CompanyBuilderRole => Boolean(entry))
|
||||
.slice(0, MAX_ROLE_COUNT);
|
||||
if (normalizedRoles.length === 0) {
|
||||
throw new Error("The planning agent did not return any company roles.");
|
||||
}
|
||||
const uniqueTitles = dedupeCompactNames(
|
||||
normalizedRoles.map((entry) => entry.title),
|
||||
"Agent",
|
||||
);
|
||||
const usedIds = new Set<string>();
|
||||
const roles = normalizedRoles.map((role, index) => {
|
||||
const title = uniqueTitles[index] ?? role.title;
|
||||
const baseId = role.id.trim() || slugify(title) || `role-${index + 1}`;
|
||||
let nextId = baseId;
|
||||
let suffix = 2;
|
||||
while (usedIds.has(nextId)) {
|
||||
nextId = `${baseId}-${suffix}`;
|
||||
suffix += 1;
|
||||
}
|
||||
usedIds.add(nextId);
|
||||
return {
|
||||
...role,
|
||||
id: nextId,
|
||||
title,
|
||||
};
|
||||
});
|
||||
return {
|
||||
companyName: coerceString(parsed.companyName) || "New Company",
|
||||
summary: coerceString(parsed.summary) || "A company plan generated from the user's brief.",
|
||||
sharedRules: uniqueStrings(coerceStringArray(parsed.sharedRules)).slice(0, 12),
|
||||
plannerNotes: uniqueStrings(coerceStringArray(parsed.plannerNotes)).slice(0, 12),
|
||||
roles,
|
||||
};
|
||||
};
|
||||
|
||||
export const buildCompanyAgentBlueprints = (plan: CompanyBuilderPlan): CompanyAgentBlueprint[] => {
|
||||
const usedNames = new Set<string>();
|
||||
return plan.roles.map((role, index) => {
|
||||
const baseName = role.title.trim() || `Agent ${index + 1}`;
|
||||
let nextName = baseName;
|
||||
let dedupe = 2;
|
||||
while (usedNames.has(nextName.toLowerCase())) {
|
||||
nextName = `${baseName} ${dedupe}`;
|
||||
dedupe += 1;
|
||||
}
|
||||
usedNames.add(nextName.toLowerCase());
|
||||
const roleWithName = { ...role, title: nextName };
|
||||
const draft = buildRoleSoulDraft({ plan, role: roleWithName });
|
||||
const files = serializePersonalityFiles(draft) as Record<AgentFileName, string>;
|
||||
return {
|
||||
agentName: nextName,
|
||||
role: roleWithName,
|
||||
draft,
|
||||
files,
|
||||
};
|
||||
});
|
||||
};
|
||||
|
||||
export const buildStoredCompanySnapshot = (params: {
|
||||
prompt: string;
|
||||
improvedBrief: string;
|
||||
plan: CompanyBuilderPlan;
|
||||
now?: () => string;
|
||||
}): CompanyBuilderStoredSnapshot => ({
|
||||
companyName: params.plan.companyName,
|
||||
prompt: params.prompt.trim(),
|
||||
improvedBrief: params.improvedBrief.trim(),
|
||||
summary: params.plan.summary,
|
||||
generatedAt: (params.now ?? (() => new Date().toISOString()))(),
|
||||
roleTitles: params.plan.roles.map((entry) => entry.title),
|
||||
planJson: JSON.stringify(params.plan),
|
||||
});
|
||||
Reference in New Issue
Block a user