@burtson-labs/bandit-engine 2.0.36 → 2.0.38

package/dist/cli/cli.js

@@ -30,7 +30,7 @@ var import_commander = require("commander");
 // package.json
 var package_default = {
   name: "@burtson-labs/bandit-engine",
-  version: "2.0.36",
+  version: "2.0.38",
   license: "BUSL-1.1",
   main: "dist/index.js",
   module: "dist/index.mjs",
@@ -162,14 +162,29 @@ var toTitleCase = (value) => {
 };
 var formatJson = (value) => `${JSON.stringify(value, null, 2)}
 `;
+var KNOWN_PROVIDERS = /* @__PURE__ */ new Set(["openai", "azure", "azure-openai", "azureopenai", "anthropic", "xai", "ollama"]);
 var sanitizeModelIdentifier = (value) => {
   const trimmed = value.trim();
   if (!trimmed.includes(":")) {
     return trimmed.toLowerCase();
   }
-  const [provider, model] = trimmed.split(/:(.+)/).filter(Boolean);
-  const cleanModel = model.replace(/[^a-zA-Z0-9_.-]/g, "-").replace(/-+/g, "-").toLowerCase();
-  return `${provider.toLowerCase()}:${cleanModel}`;
+  const segments = trimmed.split(/:(.+)/).filter(Boolean);
+  if (segments.length < 2) {
+    return trimmed.toLowerCase();
+  }
+  const [candidateProvider, rest] = segments;
+  const provider = candidateProvider.toLowerCase();
+  const cleanRest = rest.trim().replace(/[^a-zA-Z0-9_.:-]/g, "-").replace(/-+/g, "-").toLowerCase();
+  if (KNOWN_PROVIDERS.has(provider)) {
+    if (provider === "azure-openai" || provider === "azureopenai") {
+      return `azure:${cleanRest}`;
+    }
+    if (provider === "ollama") {
+      return cleanRest;
+    }
+    return `${provider}:${cleanRest}`;
+  }
+  return [candidateProvider, rest].filter(Boolean).join(":").replace(/[^a-zA-Z0-9_.:-]/g, "-").replace(/-+/g, "-").toLowerCase();
 };
 var normalizeLineEndings = (content) => content.replace(/\r\n/g, "\n");
 var ensureTrailingNewline = (content) => content.endsWith("\n") ? content : `${content}
@@ -3331,7 +3346,8 @@ var createQuickstartProject = async (options) => {
   const defaultModelId = sanitizeModelIdentifier(
     options.defaultModelId ?? inferDefaultModelId(provider)
   );
-  const fallbackModelId = options.fallbackModelId ? sanitizeModelIdentifier(options.fallbackModelId) : inferFallbackModelId(provider, defaultModelId);
+  const fallbackModelRaw = options.fallbackModelId ? options.fallbackModelId : inferFallbackModelId(provider, defaultModelId);
+  const fallbackModelId = fallbackModelRaw ? sanitizeModelIdentifier(fallbackModelRaw) : void 0;
   const inputs = {
     targetDir: resolvedDir,
     projectTitle,
@@ -3395,7 +3411,7 @@ var normalizeProvider = (value) => {
 };
 var inferDefaultModelId = (provider) => {
   if (provider === "ollama") {
-    return "ollama:llama3.1";
+    return "llama3.1";
   }
   if (provider === "azure") {
     return "azure:gpt-4o";
@@ -3410,7 +3426,11 @@ var inferDefaultModelId = (provider) => {
 };
 var inferFallbackModelId = (provider, defaultId) => {
   if (provider === "ollama") {
-    return defaultId === "ollama:llama3" ? "ollama:llama2" : "ollama:llama3";
+    const normalized = defaultId.toLowerCase();
+    if (normalized.startsWith("llama3")) {
+      return "llama2";
+    }
+    return "llama3";
   }
   if (provider === "azure") {
     return defaultId === "azure:gpt-4o-mini" ? "azure:gpt-4o" : "azure:gpt-4o-mini";