@townco/agent 0.1.55 → 0.1.57

package/dist/templates/index.js

@@ -7,6 +7,8 @@ export function getTemplateVars(name, definition) {
         tools,
         systemPrompt: definition.systemPrompt,
         hasWebSearch: tools.some((tool) => typeof tool === "string" && tool === "web_search"),
+        hasGenerateImage: tools.some((tool) => typeof tool === "string" && tool === "generate_image"),
+        hasBrowser: tools.some((tool) => typeof tool === "string" && tool === "browser"),
         hooks: definition.hooks,
     };
     if (definition.displayName) {
@@ -148,9 +150,15 @@ export function generateReadme(vars) {
         })
             .join(", ")
         : "None";
-    const envVars = vars.hasWebSearch
-        ? "\n- `EXA_API_KEY`: Required for web_search tool"
-        : "";
+    const envVars = [
+        vars.hasWebSearch ? "- `EXA_API_KEY`: Required for web_search tool" : "",
+        vars.hasGenerateImage
+            ? "- `GEMINI_API_KEY` (or `GOOGLE_API_KEY`): Required for generate_image tool"
+            : "",
+        vars.hasBrowser ? "- `KERNEL_API_KEY`: Required for browser tool" : "",
+    ]
+        .filter(Boolean)
+        .join("\n");
     return `# ${vars.name}
 
 Agent created with \`town create\`.
@@ -211,10 +219,24 @@ bun run build
 `;
 }
 export function generateEnvExample(vars) {
-    if (!vars.hasWebSearch) {
+    const envs = [];
+    if (vars.hasWebSearch) {
+        envs.push("# Required for web_search tool");
+        envs.push("EXA_API_KEY=your_exa_api_key_here");
+        envs.push("");
+    }
+    if (vars.hasGenerateImage) {
+        envs.push("# Required for generate_image tool");
+        envs.push("GEMINI_API_KEY=your_gemini_api_key_here");
+        envs.push("");
+    }
+    if (vars.hasBrowser) {
+        envs.push("# Required for browser tool");
+        envs.push("KERNEL_API_KEY=your_kernel_api_key_here");
+        envs.push("");
+    }
+    if (envs.length === 0) {
         return null;
     }
-    return `# Required for web_search tool
-EXA_API_KEY=your_exa_api_key_here
-`;
+    return envs.join("\n");
 }