pi-cohort 2.0.0

package/src/agents/agent-scope.ts

@@ -0,0 +1,6 @@
+import type { AgentScope } from "./agents.ts";
+
+export function resolveExecutionAgentScope(scope: unknown): AgentScope {
+	if (scope === "user" || scope === "project" || scope === "both") return scope;
+	return "both";
+}

package/src/agents/agent-selection.ts

@@ -0,0 +1,23 @@
+import type { AgentScope, AgentConfig } from "./agents.ts";
+
+export function mergeAgentsForScope(
+	scope: AgentScope,
+	userAgents: AgentConfig[],
+	projectAgents: AgentConfig[],
+	builtinAgents: AgentConfig[] = [],
+): AgentConfig[] {
+	const agentMap = new Map<string, AgentConfig>();
+
+	for (const agent of builtinAgents) agentMap.set(agent.name, agent);
+
+	if (scope === "both") {
+		for (const agent of userAgents) agentMap.set(agent.name, agent);
+		for (const agent of projectAgents) agentMap.set(agent.name, agent);
+	} else if (scope === "user") {
+		for (const agent of userAgents) agentMap.set(agent.name, agent);
+	} else {
+		for (const agent of projectAgents) agentMap.set(agent.name, agent);
+	}
+
+	return Array.from(agentMap.values());
+}

package/src/agents/agent-serializer.ts

@@ -0,0 +1,83 @@
+import type { AgentConfig } from "./agents.ts";
+import { frontmatterNameForConfig } from "./identity.ts";
+
+export const KNOWN_FIELDS = new Set([
+	"name",
+	"package",
+	"description",
+	"tools",
+	"model",
+	"fallbackModels",
+	"thinking",
+	"systemPromptMode",
+	"inheritProjectContext",
+	"inheritSkills",
+	"defaultContext",
+	"skill",
+	"skills",
+	"extensions",
+	"output",
+	"defaultReads",
+	"defaultProgress",
+	"interactive",
+	"maxSubagentDepth",
+	"completionGuard",
+]);
+
+function joinComma(values: string[] | undefined): string | undefined {
+	if (!values || values.length === 0) return undefined;
+	return values.join(", ");
+}
+
+export function serializeAgent(config: AgentConfig): string {
+	const lines: string[] = [];
+	lines.push("---");
+	lines.push(`name: ${frontmatterNameForConfig(config)}`);
+	if (config.packageName) lines.push(`package: ${config.packageName}`);
+	lines.push(`description: ${config.description}`);
+
+	const toolsValue = joinComma(config.tools ?? []);
+	if (toolsValue) lines.push(`tools: ${toolsValue}`);
+
+	if (config.model) lines.push(`model: ${config.model}`);
+	const fallbackModelsValue = joinComma(config.fallbackModels);
+	if (fallbackModelsValue) lines.push(`fallbackModels: ${fallbackModelsValue}`);
+	if (config.thinking && config.thinking !== "off") lines.push(`thinking: ${config.thinking}`);
+	lines.push(`systemPromptMode: ${config.systemPromptMode}`);
+	lines.push(`inheritProjectContext: ${config.inheritProjectContext ? "true" : "false"}`);
+	lines.push(`inheritSkills: ${config.inheritSkills ? "true" : "false"}`);
+	if (config.defaultContext) lines.push(`defaultContext: ${config.defaultContext}`);
+
+	const skillsValue = joinComma(config.skills);
+	if (skillsValue) lines.push(`skills: ${skillsValue}`);
+
+	if (config.extensions !== undefined) {
+		const extensionsValue = joinComma(config.extensions);
+		lines.push(`extensions: ${extensionsValue ?? ""}`);
+	}
+
+	if (config.output) lines.push(`output: ${config.output}`);
+
+	const readsValue = joinComma(config.defaultReads);
+	if (readsValue) lines.push(`defaultReads: ${readsValue}`);
+
+	if (config.defaultProgress) lines.push("defaultProgress: true");
+	if (config.interactive) lines.push("interactive: true");
+	const maxSubagentDepth = config.maxSubagentDepth;
+	if (typeof maxSubagentDepth === "number" && Number.isInteger(maxSubagentDepth) && maxSubagentDepth >= 0) {
+		lines.push(`maxSubagentDepth: ${maxSubagentDepth}`);
+	}
+	if (config.completionGuard === false) lines.push("completionGuard: false");
+
+	if (config.extraFields) {
+		for (const [key, value] of Object.entries(config.extraFields)) {
+			if (KNOWN_FIELDS.has(key)) continue;
+			lines.push(`${key}: ${value}`);
+		}
+	}
+
+	lines.push("---");
+
+	const body = config.systemPrompt ?? "";
+	return `${lines.join("\n")}\n\n${body}\n`;
+}