@agentskit/cli 0.7.1 → 0.8.2

package/dist/index.cjs

@@ -22,6 +22,7 @@ var prompts = require('@inquirer/prompts');
 var kleur = require('kleur');
 var chokidar = require('chokidar');
 var rag = require('@agentskit/rag');
+var agentSchema = require('@agentskit/core/agent-schema');
 
 function _interopDefault (e) { return e && e.__esModule ? e : { default: e }; }
 
@@ -2959,6 +2960,187 @@ function registerRagCommand(program) {
   });
 }
 
+// src/ai/scaffold.ts
+function snakeToCamel(name) {
+  return name.replace(/[-_](\w)/g, (_, c) => c.toUpperCase());
+}
+function toolStub(tool) {
+  const fnName = snakeToCamel(tool.name);
+  const schema = tool.schema ? JSON.stringify(tool.schema, null, 2) : '{ type: "object", properties: {} }';
+  return `import { defineTool } from '@agentskit/core'
+
+/**
+ * ${tool.description ?? tool.name}
+ * ${tool.implementation ? `
+ * Implementation hint: ${tool.implementation}` : ""}
+ */
+export const ${fnName} = defineTool({
+  name: '${tool.name}',
+  description: ${JSON.stringify(tool.description ?? "")},
+  schema: ${schema} as const,
+  ${tool.requiresConfirmation ? "requiresConfirmation: true," : ""}
+  async execute(args) {
+    // TODO: implement ${tool.name}
+    return { args }
+  },
+})
+`;
+}
+function indexTs(schema) {
+  const toolImports2 = (schema.tools ?? []).map((t) => `import { ${snakeToCamel(t.name)} } from './tools/${t.name}'`).join("\n");
+  const toolList2 = (schema.tools ?? []).map((t) => snakeToCamel(t.name)).join(", ");
+  return `import { createRuntime } from '@agentskit/runtime'
+${toolImports2}
+
+export const agent = {
+  name: '${schema.name}',
+  systemPrompt: ${JSON.stringify(schema.systemPrompt ?? "")},
+  tools: [${toolList2}],
+}
+
+export function createAgent(adapter: Parameters<typeof createRuntime>[0]['adapter']) {
+  return createRuntime({
+    adapter,
+    systemPrompt: agent.systemPrompt,
+    tools: agent.tools,
+  })
+}
+`;
+}
+function readme(schema) {
+  const lines = [];
+  lines.push(`# ${schema.name}`);
+  lines.push("");
+  if (schema.description) lines.push(schema.description, "");
+  lines.push(`## Model`);
+  lines.push("");
+  lines.push(`- Provider: \`${schema.model.provider}\``);
+  if (schema.model.model) lines.push(`- Model: \`${schema.model.model}\``);
+  lines.push("");
+  if (schema.tools && schema.tools.length > 0) {
+    lines.push("## Tools");
+    lines.push("");
+    for (const t of schema.tools) {
+      lines.push(`- \`${t.name}\` \u2014 ${t.description ?? ""}`);
+    }
+    lines.push("");
+  }
+  if (schema.skills && schema.skills.length > 0) {
+    lines.push("## Skills");
+    lines.push("");
+    for (const s of schema.skills) lines.push(`- ${s}`);
+    lines.push("");
+  }
+  lines.push("## Generated by");
+  lines.push("");
+  lines.push('`npx agentskit ai "<description>"`');
+  lines.push("");
+  return lines.join("\n");
+}
+function scaffoldAgent(schema) {
+  const files = [
+    { path: "agent.json", content: JSON.stringify(schema, null, 2) + "\n" },
+    { path: "agent.ts", content: indexTs(schema) },
+    { path: "README.md", content: readme(schema) }
+  ];
+  for (const tool of schema.tools ?? []) {
+    files.push({ path: `tools/${tool.name}.ts`, content: toolStub(tool) });
+  }
+  return files;
+}
+async function writeScaffold(files, outDir, opts = {}) {
+  const { writeFile: writeFile2, mkdir: mkdir2, access } = await import('fs/promises');
+  const { dirname, join: join5 } = await import('path');
+  const written = [];
+  for (const f of files) {
+    const full = join5(outDir, f.path);
+    await mkdir2(dirname(full), { recursive: true });
+    if (!opts.overwrite) {
+      try {
+        await access(full);
+        continue;
+      } catch {
+      }
+    }
+    await writeFile2(full, f.content, "utf8");
+    written.push(f.path);
+  }
+  return written;
+}
+var PLANNER_SYSTEM_PROMPT = `You design AI agents for the AgentsKit framework.
+Given a user's plain-language description, return a single JSON object
+matching the AgentsKit AgentSchema type. Do not include commentary or
+markdown fences \u2014 emit raw JSON only.
+
+Schema:
+{
+  "name": "kebab-or-snake-case",
+  "description": "short summary",
+  "systemPrompt": "you are...",
+  "model": { "provider": "anthropic"|"openai"|"gemini"|..., "model": "..." },
+  "tools": [{ "name": "search", "description": "...", "schema": {...}, "implementation": "fetch('/api/search?q=...')" }],
+  "memory": { "kind": "inMemory"|"localStorage", "key": "..." },
+  "skills": ["researcher", "critic"]
+}
+
+Prefer fewer well-scoped tools over many overlapping ones. Name tools
+in snake_case.`;
+function extractJson(text) {
+  const fenced = text.match(/```(?:json)?\s*([\s\S]+?)```/);
+  const raw = fenced?.[1] ?? text;
+  const start = raw.indexOf("{");
+  const end = raw.lastIndexOf("}");
+  if (start < 0 || end <= start) throw new Error("Planner response did not contain a JSON object");
+  return raw.slice(start, end + 1);
+}
+function createAdapterPlanner(adapter) {
+  return async (description) => {
+    const messages = [
+      { id: "sys", role: "system", content: PLANNER_SYSTEM_PROMPT, status: "complete", createdAt: /* @__PURE__ */ new Date(0) },
+      { id: "user", role: "user", content: description, status: "complete", createdAt: /* @__PURE__ */ new Date(0) }
+    ];
+    const request = { messages, context: { temperature: 0 } };
+    const source = adapter.createSource(request);
+    let text = "";
+    for await (const chunk of source.stream()) {
+      if (chunk.type === "text" && chunk.content) text += chunk.content;
+    }
+    const json = extractJson(text);
+    return agentSchema.validateAgentSchema(JSON.parse(json));
+  };
+}
+
+// src/commands/ai.ts
+function registerAiCommand(program) {
+  program.command("ai <description...>").description("Generate an agent (config + tools + skill wiring) from a natural-language description.").option("-o, --out <dir>", "Output directory", "./my-agent").option("--provider <provider>", "Planner provider (openai | anthropic | ...)", "anthropic").option("--model <model>", "Planner model id").option("--api-key <key>", "API key (falls back to env)").option("--base-url <url>", "Override base URL (for OpenAI-compatible endpoints)").option("--overwrite", "Overwrite existing files", false).option("--dry-run", "Print the plan + files without writing", false).action(async (descriptionWords, options) => {
+    const description = descriptionWords.join(" ").trim();
+    if (!description) {
+      process.stderr.write("agentskit ai: missing description.\n");
+      process.exit(1);
+    }
+    const resolved = resolveChatProvider({
+      provider: options.provider,
+      model: options.model,
+      apiKey: options.apiKey,
+      baseUrl: options.baseUrl
+    });
+    const planner2 = createAdapterPlanner(resolved.adapter);
+    process.stderr.write(`Planning agent for: "${description}"
+`);
+    const schema = await planner2(description);
+    const files = scaffoldAgent(schema);
+    if (options.dryRun) {
+      process.stdout.write(JSON.stringify({ schema, files: files.map((f) => f.path) }, null, 2) + "\n");
+      return;
+    }
+    const written = await writeScaffold(files, options.out, { overwrite: options.overwrite });
+    process.stderr.write(`Wrote ${written.length} file(s) to ${options.out}
+`);
+    for (const f of written) process.stderr.write(`  + ${f}
+`);
+  });
+}
+
 // src/commands/index.ts
 function createCli() {
   const program = new commander.Command();
@@ -2971,6 +3153,7 @@ function createCli() {
   registerConfigCommand(program);
   registerTunnelCommand(program);
   registerRagCommand(program);
+  registerAiCommand(program);
   return program;
 }