@townco/agent 0.1.0

package/templates/index.ts

@@ -0,0 +1,203 @@
+import type { AgentDefinition } from "../definition";
+
+export interface TemplateVars {
+  name: string;
+  model: string;
+  tools: string[];
+  systemPrompt: string | null;
+  hasWebSearch: boolean;
+}
+
+export function getTemplateVars(
+  name: string,
+  definition: AgentDefinition,
+): TemplateVars {
+  const tools = definition.tools || [];
+  return {
+    name,
+    model: definition.model,
+    tools,
+    systemPrompt: definition.systemPrompt,
+    hasWebSearch: tools.includes("web_search"),
+  };
+}
+
+export function generatePackageJson(
+  vars: TemplateVars,
+  useLocalDeps = true,
+): string {
+  // For local development, use link: protocol to reference monorepo packages
+  // After npm publish, users will use the published versions
+  const dependencies: Record<string, string> = useLocalDeps
+    ? {
+        "@townco/agent": "link:../../../../agent_hub/packages/agent",
+        "@townco/ui": "link:../../../../agent_hub/packages/ui",
+      }
+    : {
+        "@townco/agent": "^0.1.0",
+        "@townco/ui": "^0.1.0",
+      };
+
+  const pkg = {
+    name: `@townco/agent-${vars.name}`,
+    version: "0.0.1",
+    type: "module",
+    module: "index.ts",
+    bin: {
+      [`town-agent-${vars.name}`]: "./bin.ts",
+    },
+    scripts: {
+      build: "tsc",
+      check: "tsc --noEmit",
+      start: "bun index.ts stdio",
+      "start-http": "PORT=3100 bun index.ts http",
+    },
+    dependencies,
+  };
+
+  return JSON.stringify(pkg, null, 2);
+}
+
+export function generateIndexTs(): string {
+  return `import { readFileSync } from "node:fs";
+import { join } from "node:path";
+import { makeHttpTransport, makeStdioTransport } from "@townco/agent/acp-server";
+import type { AgentDefinition } from "@townco/agent/definition";
+
+// Load agent definition from JSON file
+const configPath = join(import.meta.dir, "agent.json");
+const agent: AgentDefinition = JSON.parse(readFileSync(configPath, "utf-8"));
+
+const transport = process.argv[2] || "stdio";
+
+if (transport === "http") {
+	makeHttpTransport(agent);
+} else if (transport === "stdio") {
+	makeStdioTransport(agent);
+} else {
+	console.error(\`Invalid transport: \${transport}\`);
+	process.exit(1);
+}
+`;
+}
+
+export function generateAgentJson(vars: TemplateVars): string {
+  const agentDef = {
+    model: vars.model,
+    systemPrompt: vars.systemPrompt,
+    tools: vars.tools,
+  };
+
+  return JSON.stringify(agentDef, null, 2);
+}
+
+export function generateBinTs(): string {
+  return `#!/usr/bin/env bun
+import "./index.js";
+`;
+}
+
+export function generateGitignore(): string {
+  return `node_modules
+dist
+.env
+`;
+}
+
+export function generateTsConfig(): string {
+  const config = {
+    compilerOptions: {
+      target: "ESNext",
+      module: "ESNext",
+      moduleResolution: "bundler",
+      lib: ["ESNext"],
+      outDir: "./dist",
+      strict: true,
+      esModuleInterop: true,
+      skipLibCheck: true,
+      forceConsistentCasingInFileNames: true,
+      resolveJsonModule: true,
+      allowSyntheticDefaultImports: true,
+    },
+    include: ["**/*.ts"],
+    exclude: ["node_modules", "dist"],
+  };
+
+  return JSON.stringify(config, null, 2);
+}
+
+export function generateReadme(vars: TemplateVars): string {
+  const toolsList = vars.tools.length > 0 ? vars.tools.join(", ") : "None";
+  const envVars = vars.hasWebSearch
+    ? "\n- `EXA_API_KEY`: Required for web_search tool"
+    : "";
+
+  return `# ${vars.name}
+
+Agent created with \`town create\`.
+
+## Configuration
+
+- **Model**: ${vars.model}
+- **Tools**: ${toolsList}
+- **System Prompt**: ${vars.systemPrompt ? "Yes" : "No"}
+
+## Setup
+${envVars ? `\n### Environment Variables\n${envVars}\n` : ""}
+Install dependencies:
+
+\`\`\`bash
+bun install
+\`\`\`
+
+## Usage
+
+### CLI Mode (stdio)
+
+\`\`\`bash
+bun start
+\`\`\`
+
+Or using the executable:
+
+\`\`\`bash
+town-agent-${vars.name}
+\`\`\`
+
+### Server Mode (HTTP)
+
+\`\`\`bash
+bun run start-http
+\`\`\`
+
+The server will start on port 3100 with the following endpoints:
+
+- \`/health\`: Health check
+- \`/rpc\`: Request-response messages
+- \`/events\`: SSE stream for real-time updates
+
+## Development
+
+Type check:
+
+\`\`\`bash
+bun run check
+\`\`\`
+
+Build:
+
+\`\`\`bash
+bun run build
+\`\`\`
+`;
+}
+
+export function generateEnvExample(vars: TemplateVars): string | null {
+  if (!vars.hasWebSearch) {
+    return null;
+  }
+
+  return `# Required for web_search tool
+EXA_API_KEY=your_exa_api_key_here
+`;
+}