create-agentmark 0.6.0 → 0.8.0

package/dist/utils/examples/templates/index.js

@@ -22,6 +22,15 @@ var createAdapterConfig = (provider) => {
         toolRegistry: "MastraToolRegistry",
         webhookHandler: "MastraAdapterWebhookHandler"
       }
+    },
+    "claude-agent-sdk": {
+      package: "@agentmark-ai/claude-agent-sdk-adapter",
+      dependencies: ["@anthropic-ai/claude-agent-sdk@^0.1.0"],
+      classes: {
+        modelRegistry: "ClaudeAgentModelRegistry",
+        toolRegistry: "ClaudeAgentToolRegistry",
+        webhookHandler: "ClaudeAgentWebhookHandler"
+      }
     }
   };
 };
@@ -39,7 +48,61 @@ function getAdapterConfig(adapter, provider) {
 
 // src/utils/examples/templates/app-index.ts
 var getIndexFileContent = (adapter = "ai-sdk") => {
-  if (adapter === "mastra") {
+  if (adapter === "claude-agent-sdk") {
+    return `import "dotenv/config";
+import { query } from "@anthropic-ai/claude-agent-sdk";
+import { withTracing } from "@agentmark-ai/claude-agent-sdk-adapter";
+import { client } from "./agentmark.client";
+
+const telemetry = {
+  isEnabled: true,
+  metadata: {
+    trace_name: "customer-support",
+    user_id: "user-123",
+    session_id: "session-123",
+    session_name: "my-first-session",
+  },
+};
+
+const runCustomerSupport = async (customer_message: string) => {
+  const prompt = await client.loadTextPrompt("customer-support-agent");
+  const adapted = await prompt.format({
+    props: {
+      customer_question: customer_message,
+    },
+    telemetry,
+  });
+
+  // Execute with Claude Agent SDK using withTracing for telemetry
+  // The adapted object contains { query, telemetry } ready for withTracing()
+  const tracedResult = await withTracing(query, adapted);
+
+  // traceId is available immediately
+  console.log("Trace ID:", tracedResult.traceId);
+
+  let result = "";
+  for await (const message of tracedResult) {
+    if (message.type === "result" && message.subtype === "success") {
+      result = message.result || "";
+    }
+  }
+
+  return result;
+};
+
+const main = async () => {
+  try {
+    const user_message = "How long does shipping take?";
+    const assistant = await runCustomerSupport(user_message);
+    console.log("Customer support response:", assistant);
+  } catch (error) {
+    console.error(error);
+  }
+};
+
+main();
+`;
+  } else if (adapter === "mastra") {
     return `import "dotenv/config";
 import { Agent } from "@mastra/core/agent";
 import { client } from "./agentmark.client";
@@ -131,70 +194,207 @@ main();
 };
 
 // src/utils/examples/templates/env.ts
-var getEnvFileContent = (_modelProvider, apiKey = "") => {
+var getEnvFileContent = (_modelProvider, apiKey = "", adapter = "ai-sdk") => {
   const apiKeyValue = apiKey || "your_api_key_here";
+  const apiKeyName = adapter === "claude-agent-sdk" ? "ANTHROPIC_API_KEY" : "OPENAI_API_KEY";
   return `# Cloud deployment: Set these environment variables
 # AGENTMARK_BASE_URL=https://api.agentmark.co
 # AGENTMARK_API_KEY=your_agentmark_api_key
 # AGENTMARK_APP_ID=your_agentmark_app_id
 # Learn more: https://docs.agentmark.co/platform/getting_started/quickstart
 
-OPENAI_API_KEY=${apiKeyValue}
+${apiKeyName}=${apiKeyValue}
 `;
 };
 
 // src/utils/examples/templates/package-setup.ts
+import fs2 from "fs-extra";
+import { execSync } from "child_process";
+
+// src/utils/file-merge.ts
 import fs from "fs-extra";
-import { execSync, execFileSync } from "child_process";
-var setupPackageJson = (targetPath = ".", deploymentMode = "cloud") => {
+import path from "path";
+function mergePackageJson(targetPath, agentmarkDeps, agentmarkDevDeps, agentmarkScripts) {
+  const packageJsonPath = path.join(targetPath, "package.json");
+  const result = {
+    success: false,
+    warnings: [],
+    added: [],
+    skipped: []
+  };
+  try {
+    if (!fs.existsSync(packageJsonPath)) {
+      result.warnings.push("No existing package.json found");
+      return result;
+    }
+    const existing = fs.readJsonSync(packageJsonPath);
+    if (!existing.dependencies) {
+      existing.dependencies = {};
+    }
+    if (!existing.devDependencies) {
+      existing.devDependencies = {};
+    }
+    if (!existing.scripts) {
+      existing.scripts = {};
+    }
+    const deps = existing.dependencies;
+    const devDeps = existing.devDependencies;
+    const scripts = existing.scripts;
+    for (const [pkg, version] of Object.entries(agentmarkDeps)) {
+      if (deps[pkg]) {
+        result.skipped.push(`dependency: ${pkg} (already exists)`);
+      } else {
+        deps[pkg] = version;
+        result.added.push(`dependency: ${pkg}@${version}`);
+      }
+    }
+    for (const [pkg, version] of Object.entries(agentmarkDevDeps)) {
+      if (devDeps[pkg]) {
+        result.skipped.push(`devDependency: ${pkg} (already exists)`);
+      } else {
+        devDeps[pkg] = version;
+        result.added.push(`devDependency: ${pkg}@${version}`);
+      }
+    }
+    for (const [scriptName, scriptCmd] of Object.entries(agentmarkScripts)) {
+      if (scripts[scriptName]) {
+        const namespacedName = `agentmark:${scriptName}`;
+        if (scripts[namespacedName]) {
+          result.skipped.push(`script: ${scriptName}`);
+          result.warnings.push(
+            `Script "${scriptName}" and "${namespacedName}" both already exist. Skipping.`
+          );
+        } else {
+          scripts[namespacedName] = scriptCmd;
+          result.added.push(`script: ${namespacedName} (namespaced due to conflict)`);
+          result.warnings.push(
+            `Script "${scriptName}" already exists. Added as "${namespacedName}" instead.`
+          );
+        }
+      } else {
+        scripts[scriptName] = scriptCmd;
+        result.added.push(`script: ${scriptName}`);
+      }
+    }
+    fs.writeJsonSync(packageJsonPath, existing, { spaces: 2 });
+    result.success = true;
+    result.content = JSON.stringify(existing, null, 2);
+    return result;
+  } catch (error) {
+    result.warnings.push(`Error merging package.json: ${error}`);
+    return result;
+  }
+}
+
+// src/utils/types.ts
+var PACKAGE_MANAGERS = {
+  "yarn.lock": {
+    name: "yarn",
+    lockFile: "yarn.lock",
+    installCmd: "yarn install",
+    addCmd: "yarn add",
+    addDevCmd: "yarn add --dev",
+    runCmd: "yarn"
+  },
+  "pnpm-lock.yaml": {
+    name: "pnpm",
+    lockFile: "pnpm-lock.yaml",
+    installCmd: "pnpm install",
+    addCmd: "pnpm add",
+    addDevCmd: "pnpm add --save-dev",
+    runCmd: "pnpm"
+  },
+  "bun.lockb": {
+    name: "bun",
+    lockFile: "bun.lockb",
+    installCmd: "bun install",
+    addCmd: "bun add",
+    addDevCmd: "bun add --dev",
+    runCmd: "bun run"
+  },
+  "package-lock.json": {
+    name: "npm",
+    lockFile: "package-lock.json",
+    installCmd: "npm install",
+    addCmd: "npm install",
+    addDevCmd: "npm install --save-dev",
+    runCmd: "npm run"
+  }
+};
+var DEFAULT_PACKAGE_MANAGER = PACKAGE_MANAGERS["package-lock.json"];
+
+// src/utils/examples/templates/package-setup.ts
+var setupPackageJson = (targetPath = ".", deploymentMode = "cloud", projectInfo = null) => {
   const packageJsonPath = `${targetPath}/package.json`;
-  if (!fs.existsSync(packageJsonPath)) {
+  const isExistingProject = projectInfo?.isExistingProject ?? false;
+  if (!fs2.existsSync(packageJsonPath)) {
     console.log("Creating package.json...");
     execSync("npm init -y", { cwd: targetPath });
   }
-  const pkgJson = fs.readJsonSync(packageJsonPath);
-  pkgJson.name = pkgJson.name === "test" || !pkgJson.name ? "agentmark-example-app" : pkgJson.name;
-  pkgJson.description = pkgJson.description || "A simple Node.js app using the Agentmark SDK";
-  const devScript = "agentmark dev";
-  const scripts = {
-    ...pkgJson.scripts,
-    "demo": "npx tsx index.ts",
-    "dev": devScript,
-    "prompt": "agentmark run-prompt",
-    "experiment": "agentmark run-experiment"
-  };
-  if (deploymentMode === "static") {
-    scripts["build"] = "agentmark build --out dist/agentmark";
+  if (isExistingProject && fs2.existsSync(packageJsonPath)) {
+    const scriptsToAdd = {
+      "demo": "npx tsx index.ts",
+      "dev": "agentmark dev",
+      "prompt": "agentmark run-prompt",
+      "experiment": "agentmark run-experiment"
+    };
+    if (deploymentMode === "static") {
+      scriptsToAdd["build"] = "agentmark build --out dist/agentmark";
+    }
+    const result = mergePackageJson(targetPath, {}, {}, scriptsToAdd);
+    if (result.added.length > 0) {
+      console.log(`\u2705 Added to package.json: ${result.added.join(", ")}`);
+    }
+    if (result.skipped.length > 0) {
+      console.log(`\u23ED\uFE0F  Skipped existing in package.json: ${result.skipped.join(", ")}`);
+    }
+    if (result.warnings.length > 0) {
+      result.warnings.forEach((w) => console.log(`\u26A0\uFE0F  ${w}`));
+    }
+  } else {
+    const pkgJson = fs2.readJsonSync(packageJsonPath);
+    pkgJson.name = pkgJson.name === "test" || !pkgJson.name ? "agentmark-example-app" : pkgJson.name;
+    pkgJson.description = pkgJson.description || "A simple Node.js app using the Agentmark SDK";
+    const devScript = "agentmark dev";
+    const scripts = {
+      ...pkgJson.scripts,
+      "demo": "npx tsx index.ts",
+      "dev": devScript,
+      "prompt": "agentmark run-prompt",
+      "experiment": "agentmark run-experiment"
+    };
+    if (deploymentMode === "static") {
+      scripts["build"] = "agentmark build --out dist/agentmark";
+    }
+    pkgJson.scripts = scripts;
+    fs2.writeJsonSync(packageJsonPath, pkgJson, { spaces: 2 });
   }
-  pkgJson.scripts = scripts;
-  pkgJson.overrides = {
-    ...pkgJson.overrides,
-    "axios": "^1.7.9"
-  };
-  fs.writeJsonSync(packageJsonPath, pkgJson, { spaces: 2 });
 };
-var installDependencies = (modelProvider, targetPath = ".", adapter = "ai-sdk", deploymentMode = "cloud") => {
+var installDependencies = (modelProvider, targetPath = ".", adapter = "ai-sdk", deploymentMode = "cloud", packageManager = null) => {
   console.log("Installing required packages...");
   console.log("This might take a moment...");
   const adapterConfig = getAdapterConfig(adapter, modelProvider);
+  const pm = packageManager || DEFAULT_PACKAGE_MANAGER;
+  const npmSuffix = pm.name === "npm" ? " --legacy-peer-deps" : "";
   try {
-    const devDepsCmd = "npm install --save-dev typescript ts-node @types/node @agentmark-ai/cli --legacy-peer-deps";
+    const devDeps = ["typescript", "ts-node", "@types/node", "@agentmark-ai/cli"];
+    const devDepsCmd = `${pm.addDevCmd} ${devDeps.join(" ")}${npmSuffix}`;
+    console.log(`Using ${pm.name} to install dependencies...`);
     execSync(devDepsCmd, {
       stdio: "inherit",
       cwd: targetPath
     });
     const loaderPackages = deploymentMode === "static" ? ["@agentmark-ai/loader-api", "@agentmark-ai/loader-file"] : ["@agentmark-ai/loader-api"];
-    const installArgs = [
-      "install",
+    const deps = [
       "dotenv",
       "@agentmark-ai/prompt-core",
       "@agentmark-ai/sdk",
       adapterConfig.package,
       ...loaderPackages,
-      ...adapterConfig.dependencies,
-      "--legacy-peer-deps"
+      ...adapterConfig.dependencies
     ];
-    execFileSync("npm", installArgs, { stdio: "inherit", cwd: targetPath });
+    const depsCmd = `${pm.addCmd} ${deps.join(" ")}${npmSuffix}`;
+    execSync(depsCmd, { stdio: "inherit", cwd: targetPath });
     console.log("Packages installed successfully!");
   } catch (error) {
     console.error("Error installing packages:", error);
@@ -368,28 +568,32 @@ var getStoryDataset = () => {
 };
 
 // src/utils/examples/templates/example-prompts.ts
-import fs2 from "fs-extra";
+import fs3 from "fs-extra";
 var createExamplePrompts = (model, targetPath = ".", adapter = "ai-sdk") => {
-  fs2.ensureDirSync(`${targetPath}/agentmark`);
-  if (adapter !== "mastra") {
+  fs3.ensureDirSync(`${targetPath}/agentmark`);
+  const noImageSupport = ["mastra", "claude-agent-sdk"];
+  const noSpeechSupport = ["mastra", "claude-agent-sdk"];
+  const skipImagePrompts = noImageSupport.includes(adapter);
+  const skipSpeechPrompts = noSpeechSupport.includes(adapter);
+  if (!skipImagePrompts) {
     const animalDrawingPrompt = getAnimalDrawingPrompt();
-    fs2.writeFileSync(`${targetPath}/agentmark/animal-drawing.prompt.mdx`, animalDrawingPrompt);
+    fs3.writeFileSync(`${targetPath}/agentmark/animal-drawing.prompt.mdx`, animalDrawingPrompt);
     const animalDataset = getAnimalDataset();
-    fs2.writeFileSync(`${targetPath}/agentmark/animal.jsonl`, animalDataset);
+    fs3.writeFileSync(`${targetPath}/agentmark/animal.jsonl`, animalDataset);
   }
   const customerSupportPrompt = getCustomerSupportPrompt(model);
-  fs2.writeFileSync(`${targetPath}/agentmark/customer-support-agent.prompt.mdx`, customerSupportPrompt);
+  fs3.writeFileSync(`${targetPath}/agentmark/customer-support-agent.prompt.mdx`, customerSupportPrompt);
   const customerQueryDataset = getCustomerQueryDataset();
-  fs2.writeFileSync(`${targetPath}/agentmark/customer-query.jsonl`, customerQueryDataset);
+  fs3.writeFileSync(`${targetPath}/agentmark/customer-query.jsonl`, customerQueryDataset);
   const partyPlannerPrompt = getPartyPlannerPrompt(model);
-  fs2.writeFileSync(`${targetPath}/agentmark/party-planner.prompt.mdx`, partyPlannerPrompt);
+  fs3.writeFileSync(`${targetPath}/agentmark/party-planner.prompt.mdx`, partyPlannerPrompt);
   const partyDataset = getPartyDataset();
-  fs2.writeFileSync(`${targetPath}/agentmark/party.jsonl`, partyDataset);
-  if (adapter !== "mastra") {
+  fs3.writeFileSync(`${targetPath}/agentmark/party.jsonl`, partyDataset);
+  if (!skipSpeechPrompts) {
     const storyTellerPrompt = getStoryTellerPrompt();
-    fs2.writeFileSync(`${targetPath}/agentmark/story-teller.prompt.mdx`, storyTellerPrompt);
+    fs3.writeFileSync(`${targetPath}/agentmark/story-teller.prompt.mdx`, storyTellerPrompt);
     const storyDataset = getStoryDataset();
-    fs2.writeFileSync(`${targetPath}/agentmark/story.jsonl`, storyDataset);
+    fs3.writeFileSync(`${targetPath}/agentmark/story.jsonl`, storyDataset);
   }
 };
 
@@ -398,8 +602,9 @@ var getClientConfigContent = (options) => {
   const { provider, languageModels, adapter, deploymentMode = "cloud" } = options;
   const adapterConfig = getAdapterConfig(adapter, provider);
   const { modelRegistry, toolRegistry } = adapterConfig.classes;
-  const providerImport = `import { ${provider} } from '@ai-sdk/${provider}';`;
-  const extraModelRegs = provider === "openai" ? `.registerModels(["dall-e-3"], (name: string) => ${provider}.image(name))
+  const isClaudeAgentSdk = adapter === "claude-agent-sdk";
+  const providerImport = isClaudeAgentSdk ? "" : `import { ${provider} } from '@ai-sdk/${provider}';`;
+  const extraModelRegs = provider === "openai" && !isClaudeAgentSdk ? `.registerModels(["dall-e-3"], (name: string) => ${provider}.image(name))
     .registerModels(["tts-1-hd"], (name: string) => ${provider}.speech(name))` : "";
   const loaderImport = deploymentMode === "cloud" ? `import { ApiLoader } from "@agentmark-ai/loader-api";` : `import { ApiLoader } from "@agentmark-ai/loader-api";
 import { FileLoader } from "@agentmark-ai/loader-file";`;
@@ -414,6 +619,51 @@ import { FileLoader } from "@agentmark-ai/loader-file";`;
       });` : `  const loader = process.env.NODE_ENV === 'development'
     ? ApiLoader.local({ baseUrl: process.env.AGENTMARK_BASE_URL || 'http://localhost:9418' })
     : new FileLoader('./dist/agentmark');`;
+  const modelRegistrySetup = isClaudeAgentSdk ? `function createModelRegistry() {
+  // Claude Agent SDK accepts model names directly.
+  // Use createDefault() for simple pass-through of model names.
+  const modelRegistry = ${modelRegistry}.createDefault();
+
+  // To configure specific models (e.g., extended thinking), use:
+  // const modelRegistry = new ${modelRegistry}()
+  //   .registerModels(/claude-.*-thinking/, (name) => ({
+  //     model: name,
+  //     maxThinkingTokens: 10000,  // Enable extended thinking
+  //   }))
+  //   .registerModels("claude-sonnet-4-20250514", (name) => ({ model: name }));
+
+  return modelRegistry;
+}` : `function createModelRegistry() {
+  const modelRegistry = new ${modelRegistry}()
+    .registerModels(${JSON.stringify(languageModels)}, (name: string) => ${provider}(name))
+    ${extraModelRegs};
+  return modelRegistry;
+}`;
+  const adapterOptionsImport = isClaudeAgentSdk ? `
+// Claude Agent SDK adapter options
+// See: https://github.com/anthropics/claude-agent-sdk
+const adapterOptions = {
+  // Permission mode controls tool access:
+  // - 'default': Requires user approval for each tool use
+  // - 'acceptEdits': Auto-approve file edits only
+  // - 'bypassPermissions': Auto-approve all tools (use for automated pipelines)
+  // - 'plan': Planning mode only, no tool execution
+  permissionMode: 'bypassPermissions' as const,
+
+  // Maximum conversation turns before stopping
+  maxTurns: 20,
+
+  // Optional: Set working directory for file operations
+  // cwd: process.cwd(),
+
+  // Optional: Budget limit in USD
+  // maxBudgetUsd: 10.00,
+
+  // Optional: Restrict which tools the agent can use
+  // allowedTools: ['Read', 'Write', 'Glob'],
+  // disallowedTools: ['Bash'],
+};` : "";
+  const createClientCall = isClaudeAgentSdk ? `return createAgentMarkClient<AgentMarkTypes, typeof toolRegistry>({ loader, modelRegistry, toolRegistry, evalRegistry, adapterOptions });` : `return createAgentMarkClient<AgentMarkTypes, typeof toolRegistry>({ loader, modelRegistry, toolRegistry, evalRegistry });`;
   return `// agentmark.client.ts
 import path from 'node:path';
 import dotenv from 'dotenv';
@@ -422,13 +672,9 @@ import { createAgentMarkClient, ${modelRegistry}, ${toolRegistry}, EvalRegistry
 ${loaderImport}
 import AgentMarkTypes, { Tools } from './agentmark.types';
 ${providerImport}
+${adapterOptionsImport}
 
-function createModelRegistry() {
-  const modelRegistry = new ${modelRegistry}()
-    .registerModels(${JSON.stringify(languageModels)}, (name: string) => ${provider}(name))
-    ${extraModelRegs};
-  return modelRegistry;
-}
+${modelRegistrySetup}
 
 function createToolRegistry() {
   const toolRegistry = new ${toolRegistry}<Tools>()
@@ -475,7 +721,7 @@ ${loaderSetup}
   const modelRegistry = createModelRegistry();
   const toolRegistry = createToolRegistry();
   const evalRegistry = createEvalRegistry();
-  return createAgentMarkClient<AgentMarkTypes, typeof toolRegistry>({ loader, modelRegistry, toolRegistry, evalRegistry });
+  ${createClientCall}
 }
 
 export const client = createClient();