getcompanion-ai/co-mono
1
0
Fork 0
mirror of https://github.com/getcompanion-ai/co-mono.git synced 2026-04-18 17:04:49 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Add cwd parameter to subagent example tool (#291)

Browse source
This commit is contained in:
Nico Bailon 2025-12-23 15:58:43 -08:00 • committed by GitHub
parent fb9c25f51e
commit ce8a1c8eb1
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 8 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

12
packages/coding-agent/examples/custom-tools/subagent/index.ts
View file

@ -203,6 +203,7 @@ async function runSingleAgent(
agents: AgentConfig[], agents: AgentConfig[],
agentName: string, agentName: string,
task: string, task: string,
cwd: string | undefined,
step: number | undefined, step: number | undefined,
signal: AbortSignal | undefined, signal: AbortSignal | undefined,
onUpdate: OnUpdateCallback | undefined, onUpdate: OnUpdateCallback | undefined,
@ -263,7 +264,7 @@ async function runSingleAgent(
let wasAborted = false; let wasAborted = false;
const exitCode = await new Promise<number>((resolve) => { const exitCode = await new Promise<number>((resolve) => {
const proc = spawn("pi", args, { cwd: pi.cwd, shell: false, stdio: ["ignore", "pipe", "pipe"] }); const proc = spawn("pi", args, { cwd: cwd ?? pi.cwd, shell: false, stdio: ["ignore", "pipe", "pipe"] });
let buffer = ""; let buffer = "";
const processLine = (line: string) => { const processLine = (line: string) => {
@ -338,11 +339,13 @@ async function runSingleAgent(
const TaskItem = Type.Object({ const TaskItem = Type.Object({
agent: Type.String({ description: "Name of the agent to invoke" }), agent: Type.String({ description: "Name of the agent to invoke" }),
task: Type.String({ description: "Task to delegate to the agent" }), task: Type.String({ description: "Task to delegate to the agent" }),
cwd: Type.Optional(Type.String({ description: "Working directory for the agent process" })),
}); });
const ChainItem = Type.Object({ const ChainItem = Type.Object({
agent: Type.String({ description: "Name of the agent to invoke" }), agent: Type.String({ description: "Name of the agent to invoke" }),
task: Type.String({ description: "Task with optional {previous} placeholder for prior output" }), task: Type.String({ description: "Task with optional {previous} placeholder for prior output" }),
cwd: Type.Optional(Type.String({ description: "Working directory for the agent process" })),
}); });
const AgentScopeSchema = StringEnum(["user", "project", "both"] as const, { const AgentScopeSchema = StringEnum(["user", "project", "both"] as const, {
@ -357,6 +360,7 @@ const SubagentParams = Type.Object({
chain: Type.Optional(Type.Array(ChainItem, { description: "Array of {agent, task} for sequential execution" })), chain: Type.Optional(Type.Array(ChainItem, { description: "Array of {agent, task} for sequential execution" })),
agentScope: Type.Optional(AgentScopeSchema), agentScope: Type.Optional(AgentScopeSchema),
confirmProjectAgents: Type.Optional(Type.Boolean({ description: "Prompt before running project-local agents. Default: true.", default: true })), confirmProjectAgents: Type.Optional(Type.Boolean({ description: "Prompt before running project-local agents. Default: true.", default: true })),
cwd: Type.Optional(Type.String({ description: "Working directory for the agent process (single mode)" })),
}); });
const factory: CustomToolFactory = (pi) => { const factory: CustomToolFactory = (pi) => {
@ -441,7 +445,7 @@ const factory: CustomToolFactory = (pi) => {
} }
} : undefined; } : undefined;
const result = await runSingleAgent(pi, agents, step.agent, taskWithContext, i + 1, signal, chainUpdate, makeDetails("chain")); const result = await runSingleAgent(pi, agents, step.agent, taskWithContext, step.cwd, i + 1, signal, chainUpdate, makeDetails("chain"));
results.push(result); results.push(result);
const isError = result.exitCode !== 0 || result.stopReason === "error" || result.stopReason === "aborted"; const isError = result.exitCode !== 0 || result.stopReason === "error" || result.stopReason === "aborted";
@ -486,7 +490,7 @@ const factory: CustomToolFactory = (pi) => {
const results = await mapWithConcurrencyLimit(params.tasks, MAX_CONCURRENCY, async (t, index) => { const results = await mapWithConcurrencyLimit(params.tasks, MAX_CONCURRENCY, async (t, index) => {
const result = await runSingleAgent( const result = await runSingleAgent(
pi, agents, t.agent, t.task, undefined, signal, pi, agents, t.agent, t.task, t.cwd, undefined, signal,
// Per-task update callback // Per-task update callback
(partial) => { (partial) => {
if (partial.details?.results[0]) { if (partial.details?.results[0]) {
@ -511,7 +515,7 @@ const factory: CustomToolFactory = (pi) => {
} }
if (params.agent && params.task) { if (params.agent && params.task) {
const result = await runSingleAgent(pi, agents, params.agent, params.task, undefined, signal, onUpdate, makeDetails("single")); const result = await runSingleAgent(pi, agents, params.agent, params.task, params.cwd, undefined, signal, onUpdate, makeDetails("single"));
const isError = result.exitCode !== 0 || result.stopReason === "error" || result.stopReason === "aborted"; const isError = result.exitCode !== 0 || result.stopReason === "error" || result.stopReason === "aborted";
if (isError) { if (isError) {
const errorMsg = result.errorMessage || result.stderr || getFinalOutput(result.messages) || "(no output)"; const errorMsg = result.errorMessage || result.stderr || getFinalOutput(result.messages) || "(no output)";