create-agentmark 0.8.3 → 0.9.0

package/dist/index.js

@@ -9,34 +9,6 @@ import prompts2 from "prompts";
 import fs4 from "fs-extra";
 import * as path2 from "path";
 
-// src/utils/providers.ts
-var Providers = {
-  openai: {
-    label: "OpenAI",
-    languageModels: [
-      "gpt-4o",
-      "gpt-4o-mini",
-      "gpt-4",
-      "gpt-5",
-      "gpt-4-turbo",
-      "gpt-3.5-turbo"
-    ],
-    imageModels: ["dall-e-3", "dall-e-2"],
-    speechModels: ["tts-1", "tts-1-hd"]
-  },
-  anthropic: {
-    label: "Anthropic",
-    languageModels: [
-      "claude-sonnet-4-20250514",
-      "claude-opus-4-20250514",
-      "claude-3-5-sonnet-20241022",
-      "claude-3-5-haiku-20241022"
-    ],
-    imageModels: [],
-    speechModels: []
-  }
-};
-
 // src/utils/examples/templates/adapters.ts
 var createAdapterConfig = (provider) => {
   return {
@@ -599,7 +571,7 @@ var getAnimalDrawingPrompt = () => {
   return `---
 name: animal-drawing
 image_config:
-  model_name: dall-e-3
+  model_name: openai/dall-e-3
   num_images: 1
   size: 1024x1024
   aspect_ratio: 1:1
@@ -701,7 +673,7 @@ var getStoryTellerPrompt = () => {
   return `---
 name: story-teller
 speech_config:
-  model_name: tts-1-hd
+  model_name: openai/tts-1-hd
   voice: "nova"
   speed: 1.0
   output_format: "mp3"
@@ -746,20 +718,23 @@ var getStoryDataset = () => {
 import fs3 from "fs-extra";
 var createExamplePrompts = (model, targetPath = ".", adapter = "ai-sdk") => {
   fs3.ensureDirSync(`${targetPath}/agentmark`);
-  const noImageSupport = ["mastra", "claude-agent-sdk"];
-  const noSpeechSupport = ["mastra", "claude-agent-sdk"];
+  const noImageSupport = ["mastra", "claude-agent-sdk", "pydantic-ai"];
+  const noSpeechSupport = ["mastra", "claude-agent-sdk", "pydantic-ai"];
   const skipImagePrompts = noImageSupport.includes(adapter);
   const skipSpeechPrompts = noSpeechSupport.includes(adapter);
+  const usedModels = [];
   if (!skipImagePrompts) {
     const animalDrawingPrompt = getAnimalDrawingPrompt();
     fs3.writeFileSync(`${targetPath}/agentmark/animal-drawing.prompt.mdx`, animalDrawingPrompt);
     const animalDataset = getAnimalDataset();
     fs3.writeFileSync(`${targetPath}/agentmark/animal.jsonl`, animalDataset);
+    usedModels.push("openai/dall-e-3");
   }
   const customerSupportPrompt = getCustomerSupportPrompt(model);
   fs3.writeFileSync(`${targetPath}/agentmark/customer-support-agent.prompt.mdx`, customerSupportPrompt);
   const customerQueryDataset = getCustomerQueryDataset();
   fs3.writeFileSync(`${targetPath}/agentmark/customer-query.jsonl`, customerQueryDataset);
+  usedModels.push(model);
   const partyPlannerPrompt = getPartyPlannerPrompt(model);
   fs3.writeFileSync(`${targetPath}/agentmark/party-planner.prompt.mdx`, partyPlannerPrompt);
   const partyDataset = getPartyDataset();
@@ -769,18 +744,18 @@ var createExamplePrompts = (model, targetPath = ".", adapter = "ai-sdk") => {
     fs3.writeFileSync(`${targetPath}/agentmark/story-teller.prompt.mdx`, storyTellerPrompt);
     const storyDataset = getStoryDataset();
     fs3.writeFileSync(`${targetPath}/agentmark/story.jsonl`, storyDataset);
+    usedModels.push("openai/tts-1-hd");
   }
+  return [...new Set(usedModels)];
 };
 
 // src/utils/examples/templates/user-client-config.ts
 var getClientConfigContent = (options) => {
-  const { provider, languageModels, adapter, deploymentMode = "cloud" } = options;
+  const { provider, adapter, deploymentMode = "cloud" } = options;
   const adapterConfig = getAdapterConfig(adapter, provider);
   const { modelRegistry, toolRegistry } = adapterConfig.classes;
   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";`;
   const loaderSetup = deploymentMode === "cloud" ? `  // ApiLoader works for both development and production
@@ -810,8 +785,7 @@ import { FileLoader } from "@agentmark-ai/loader-file";`;
   return modelRegistry;
 }` : `function createModelRegistry() {
   const modelRegistry = new ${modelRegistry}()
-    .registerModels(${JSON.stringify(languageModels)}, (name: string) => ${provider}(name))
-    ${extraModelRegs};
+    .registerProviders({ ${provider} });
   return modelRegistry;
 }`;
   const adapterOptionsImport = isClaudeAgentSdk ? `
@@ -997,38 +971,6 @@ function displayProjectDetectionSummary(projectInfo) {
   console.log("");
 }
 
-// src/utils/git-init.ts
-import { execSync as execSync2 } from "child_process";
-function initGitRepo(targetPath) {
-  try {
-    try {
-      execSync2("git --version", { stdio: "ignore" });
-    } catch {
-      console.log("\u26A0\uFE0F  git not found \u2014 skipping repository initialization");
-      return false;
-    }
-    try {
-      execSync2("git rev-parse --is-inside-work-tree", {
-        cwd: targetPath,
-        stdio: "ignore"
-      });
-      return false;
-    } catch {
-    }
-    execSync2("git init", { cwd: targetPath, stdio: "ignore" });
-    execSync2("git add -A", { cwd: targetPath, stdio: "ignore" });
-    execSync2(
-      'git -c user.name="create-agentmark" -c user.email="noreply" commit -m "Initial commit from create-agentmark"',
-      { cwd: targetPath, stdio: "ignore" }
-    );
-    console.log("\u2705 Initialized git repository with initial commit");
-    return true;
-  } catch {
-    console.log("\u26A0\uFE0F  Could not initialize git repository");
-    return false;
-  }
-}
-
 // src/utils/examples/create-example-app.ts
 var setupMCPServer = (client, targetPath) => {
   if (client === "skip") {
@@ -1147,7 +1089,7 @@ var setupMCPServer = (client, targetPath) => {
 var createExampleApp = async (client, targetPath = ".", apiKey = "", adapter = "ai-sdk", deploymentMode = "cloud", projectInfo = null, resolutions = []) => {
   try {
     const modelProvider = adapter === "claude-agent-sdk" ? "anthropic" : "openai";
-    const model = adapter === "claude-agent-sdk" ? "claude-sonnet-4-20250514" : "gpt-4o";
+    const model = adapter === "claude-agent-sdk" ? "anthropic/claude-sonnet-4-20250514" : "openai/gpt-4o";
     const isExistingProject = projectInfo?.isExistingProject ?? false;
     if (isExistingProject) {
       console.log("Adding AgentMark to existing project...");
@@ -1157,12 +1099,11 @@ var createExampleApp = async (client, targetPath = ".", apiKey = "", adapter = "
     const folderName = targetPath;
     fs4.ensureDirSync(`${targetPath}/agentmark`);
     setupMCPServer(client, targetPath);
-    createExamplePrompts(model, targetPath, adapter);
+    const usedModels = createExamplePrompts(model, targetPath, adapter);
     console.log(`\u2705 Example prompts and datasets created in ${folderName}/agentmark/`);
-    const langModels = Providers[modelProvider].languageModels.slice(0, 1);
     fs4.writeFileSync(
       `${targetPath}/agentmark.client.ts`,
-      getClientConfigContent({ provider: modelProvider, languageModels: langModels, adapter, deploymentMode })
+      getClientConfigContent({ provider: modelProvider, adapter, deploymentMode })
     );
     if (shouldMergeFile(".env", projectInfo, resolutions)) {
       const envVars = {};
@@ -1283,9 +1224,6 @@ main().catch((err) => {
       fs4.writeFileSync(devEntryPath, devEntryContent);
       console.log(`\u2705 Created dev-entry.ts at project root`);
     }
-    if (!isExistingProject) {
-      initGitRepo(targetPath);
-    }
     console.log("\n\u2705 Agentmark initialization completed successfully!");
     console.log(
       `
@@ -1321,6 +1259,7 @@ main().catch((err) => {
     console.log("\u2500".repeat(70));
     console.log("  Documentation: https://docs.agentmark.co");
     console.log("\u2550".repeat(70) + "\n");
+    return usedModels;
   } catch (error) {
     console.error("Error creating example app:", error);
     throw error;
@@ -1807,7 +1746,7 @@ htmlcov/
 };
 var createPythonApp = async (client, targetPath = ".", apiKey = "", deploymentMode = "cloud", adapter = "pydantic-ai", projectInfo = null, resolutions = []) => {
   try {
-    const model = adapter === "claude-agent-sdk" ? "claude-sonnet-4-20250514" : "gpt-4o";
+    const model = adapter === "claude-agent-sdk" ? "anthropic/claude-sonnet-4-20250514" : "openai/gpt-4o";
     const adapterDisplayName = adapter === "claude-agent-sdk" ? "Claude Agent SDK" : "Pydantic AI";
     const isExistingProject = projectInfo?.isExistingProject ?? false;
     if (isExistingProject) {
@@ -1818,7 +1757,7 @@ var createPythonApp = async (client, targetPath = ".", apiKey = "", deploymentMo
     const folderName = targetPath;
     fs5.ensureDirSync(`${targetPath}/agentmark`);
     setupMCPServer2(client, targetPath);
-    createExamplePrompts(model, targetPath, adapter);
+    const usedModels = createExamplePrompts(model, targetPath, adapter);
     console.log(`Example prompts and datasets created in ${folderName}/agentmark/`);
     if (!isExistingProject) {
       const projectName = path3.basename(targetPath).replace(/[^a-zA-Z0-9_-]/g, "-");
@@ -1893,9 +1832,6 @@ var createPythonApp = async (client, targetPath = ".", apiKey = "", deploymentMo
       console.log("Note: You'll need to set up a virtual environment and install dependencies.");
       console.log("");
     }
-    if (!isExistingProject) {
-      initGitRepo(targetPath);
-    }
     console.log("\n\u2705 AgentMark Python initialization completed successfully!");
     console.log(
       `
@@ -1930,6 +1866,7 @@ var createPythonApp = async (client, targetPath = ".", apiKey = "", deploymentMo
     console.log("\u2500".repeat(70));
     console.log("  Documentation: https://docs.agentmark.co");
     console.log("\u2550".repeat(70) + "\n");
+    return usedModels;
   } catch (error) {
     console.error("Error creating Python app:", error);
     throw error;
@@ -2046,6 +1983,38 @@ function isCurrentDirectory(folderName) {
   return folderName === "." || folderName === "./" || folderName === ".\\";
 }
 
+// src/utils/git-init.ts
+import { execSync as execSync2 } from "child_process";
+function initGitRepo(targetPath) {
+  try {
+    try {
+      execSync2("git --version", { stdio: "ignore" });
+    } catch {
+      console.log("\u26A0\uFE0F  git not found \u2014 skipping repository initialization");
+      return false;
+    }
+    try {
+      execSync2("git rev-parse --is-inside-work-tree", {
+        cwd: targetPath,
+        stdio: "ignore"
+      });
+      return false;
+    } catch {
+    }
+    execSync2("git init", { cwd: targetPath, stdio: "ignore" });
+    execSync2("git add -A", { cwd: targetPath, stdio: "ignore" });
+    execSync2(
+      'git -c user.name="create-agentmark" -c user.email="noreply" commit -m "Initial commit from create-agentmark"',
+      { cwd: targetPath, stdio: "ignore" }
+    );
+    console.log("\u2705 Initialized git repository with initial commit");
+    return true;
+  } catch {
+    console.log("\u26A0\uFE0F  Could not initialize git repository");
+    return false;
+  }
+}
+
 // src/index.ts
 var parseArgs = () => {
   const args = process.argv.slice(2);
@@ -2127,7 +2096,6 @@ var main = async () => {
     });
     adapter = response.adapter;
   }
-  config.builtInModels = adapter === "claude-agent-sdk" ? ["claude-sonnet-4-20250514"] : ["gpt-4o"];
   const apiKeyName = adapter === "claude-agent-sdk" ? "Anthropic" : "OpenAI";
   let apiKey = "";
   const { providedApiKey } = await prompts2({
@@ -2170,11 +2138,13 @@ var main = async () => {
       { title: "Skip", value: "skip" }
     ]
   });
+  let usedModels;
   if (language === "python") {
-    await createPythonApp(client, targetPath, apiKey, deploymentMode, adapter, projectInfo, resolutions);
+    usedModels = await createPythonApp(client, targetPath, apiKey, deploymentMode, adapter, projectInfo, resolutions);
   } else {
-    await createExampleApp(client, targetPath, apiKey, adapter, deploymentMode, projectInfo, resolutions);
+    usedModels = await createExampleApp(client, targetPath, apiKey, adapter, deploymentMode, projectInfo, resolutions);
   }
+  config.builtInModels = usedModels;
   const agentmarkJsonPath = path6.join(targetPath, "agentmark.json");
   const agentmarkJsonResolution = resolutions.find((r) => r.path === "agentmark.json");
   if (!fs8.existsSync(agentmarkJsonPath) || agentmarkJsonResolution?.action === "overwrite") {
@@ -2182,6 +2152,9 @@ var main = async () => {
   } else if (agentmarkJsonResolution?.action === "skip") {
     console.log("\u23ED\uFE0F  Skipped agentmark.json (keeping existing file)");
   }
+  if (!projectInfo.isExistingProject) {
+    initGitRepo(targetPath);
+  }
 };
 main().catch((error) => {
   console.error("Error:", error);