@townco/agent 0.1.124 → 0.1.125

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@townco/agent",
-  "version": "0.1.124",
+  "version": "0.1.125",
   "type": "module",
   "module": "index.ts",
   "files": [
@@ -83,12 +83,12 @@
     "@opentelemetry/sdk-trace-base": "^1.28.0",
     "@opentelemetry/sdk-trace-node": "^1.28.0",
     "@opentelemetry/semantic-conventions": "^1.28.0",
-    "@townco/apiclient": "0.0.36",
-    "@townco/core": "0.0.94",
-    "@townco/gui-template": "0.1.113",
-    "@townco/tsconfig": "0.1.113",
-    "@townco/tui-template": "0.1.113",
-    "@townco/ui": "0.1.116",
+    "@townco/apiclient": "0.0.37",
+    "@townco/core": "0.0.95",
+    "@townco/gui-template": "0.1.114",
+    "@townco/tsconfig": "0.1.114",
+    "@townco/tui-template": "0.1.114",
+    "@townco/ui": "0.1.117",
     "ai-tokenizer": "^1.0.6",
     "exa-js": "^2.0.0",
     "hono": "^4.10.4",

package/templates/index.ts

@@ -4,6 +4,14 @@ import type { AgentDefinition, McpConfigSchema } from "../definition";
 
 type McpConfig = z.infer<typeof McpConfigSchema>;
 
+/** Config for a subagent to wire up via makeSubagentsTool */
+export interface SubagentConfig {
+  agentName: string;
+  toolDescription: string;
+  displayName?: string;
+  path: string;
+}
+
 export interface TemplateVars {
   name: string;
   model: string;
@@ -24,6 +32,7 @@ export interface TemplateVars {
       }
   >;
   mcps?: McpConfig[];
+  subagents?: SubagentConfig[];
   systemPrompt: string | null;
   hasWebSearch: boolean;
   hasGenerateImage: boolean;
@@ -83,6 +92,14 @@ export function getTemplateVars(
     result.mcps = definition.mcps;
   }
 
+  // Handle subagents if defined (using type assertion since AgentDefinition may not have subagents yet)
+  const defWithSubagents = definition as AgentDefinition & {
+    subagents?: SubagentConfig[];
+  };
+  if (defWithSubagents.subagents && defWithSubagents.subagents.length > 0) {
+    result.subagents = defWithSubagents.subagents;
+  }
+
   return result;
 }
 
@@ -123,41 +140,67 @@ export function generatePackageJson(vars: TemplateVars): string {
 }
 
 export async function generateIndexTs(vars: TemplateVars): Promise<string> {
-  // Build agent definition with fields in a logical order
-  const agentDef: Record<string, unknown> = {
-    model: vars.model,
-  };
+  const hasSubagents = vars.subagents && vars.subagents.length > 0;
 
+  let imports =
+    'import type { AgentDefinition } from "@townco/agent/definition";';
+  if (hasSubagents) {
+    imports += '\nimport { makeSubagentsTool } from "@townco/agent/utils";';
+  }
+
+  // Build tools code - either plain JSON or with makeSubagentsTool call
+  const toolsCode: string = (() => {
+    if (!hasSubagents) {
+      return JSON.stringify(vars.tools);
+    }
+    // makeSubagentsTool expects 'description' not 'toolDescription'
+    const subagentConfigs = vars.subagents!.map((s) => ({
+      agentName: s.agentName,
+      description: s.toolDescription,
+      ...(s.displayName && { displayName: s.displayName }),
+      path: s.path,
+    }));
+    const subagentToolCall = `makeSubagentsTool(${JSON.stringify(subagentConfigs, null, 2)})`;
+    const toolItems = vars.tools.map((t) => JSON.stringify(t));
+    toolItems.push(subagentToolCall);
+    return `[${toolItems.join(", ")}]`;
+  })();
+
+  // Build optional fields
+  const optionalFields: string[] = [];
   if (vars.displayName) {
-    agentDef.displayName = vars.displayName;
+    optionalFields.push(`displayName: ${JSON.stringify(vars.displayName)},`);
   }
   if (vars.description) {
-    agentDef.description = vars.description;
+    optionalFields.push(`description: ${JSON.stringify(vars.description)},`);
   }
   if (vars.suggestedPrompts) {
-    agentDef.suggestedPrompts = vars.suggestedPrompts;
+    optionalFields.push(
+      `suggestedPrompts: ${JSON.stringify(vars.suggestedPrompts)},`,
+    );
   }
-
-  agentDef.systemPrompt = vars.systemPrompt;
-  agentDef.tools = vars.tools;
-
   if (vars.mcps && vars.mcps.length > 0) {
-    agentDef.mcps = vars.mcps;
+    optionalFields.push(`mcps: ${JSON.stringify(vars.mcps)},`);
   }
-
   if (vars.hooks) {
-    agentDef.hooks = vars.hooks;
+    optionalFields.push(`hooks: ${JSON.stringify(vars.hooks)},`);
   }
 
-  return prettier.format(
-    `import type { AgentDefinition } from "@townco/agent/definition";
+  const optionalFieldsStr =
+    optionalFields.length > 0 ? `\n  ${optionalFields.join("\n  ")}` : "";
 
-const agent: AgentDefinition = ${JSON.stringify(agentDef)};
+  const code = `${imports}
+
+const agent: AgentDefinition = {
+  model: ${JSON.stringify(vars.model)},${optionalFieldsStr}
+  systemPrompt: ${JSON.stringify(vars.systemPrompt)},
+  tools: ${toolsCode},
+};
 
 export default agent;
-`,
-    { parser: "typescript" },
-  );
+`;
+
+  return prettier.format(code, { parser: "typescript" });
 }
 
 export function generateBinTs(): string {