create-oke 0.6.1 → 0.8.0

package/src/ai-setup/detect-ollama.ts

@@ -0,0 +1,180 @@
+/**
+ * Detect local Ollama — CLI (`ollama list` / `ps`) and HTTP `/api/tags`.
+ */
+
+import { ALL_CURATED, type CatalogModel } from "./catalog.ts";
+
+/** Default Ollama HTTP origin (create-oke copy — no okengine driver import). */
+const OLLAMA_DEFAULT_BASE_URL = "http://127.0.0.1:11434";
+
+/**
+ * Normalize an Ollama base URL (strip trailing slash / accidental `/v1`).
+ *
+ * @param raw - User or env value
+ */
+function normalizeOllamaBaseUrl(raw: string): string {
+  const trimmed = raw.trim().replace(/\/$/, "");
+  if (!trimmed) return OLLAMA_DEFAULT_BASE_URL;
+  if (/^https?:\/\//i.test(trimmed)) return trimmed.replace(/\/v1\/?$/i, "");
+  return `http://${trimmed}`;
+}
+
+/** Result of probing the local Ollama install. */
+export type OllamaDetectResult = {
+  readonly available: boolean;
+  readonly baseUrl: string;
+  readonly installed: readonly string[];
+  readonly running: readonly string[];
+  readonly curatedInstalled: readonly CatalogModel[];
+  readonly curatedMissing: readonly CatalogModel[];
+};
+
+/**
+ * Parse `ollama list` tabular output into model names.
+ *
+ * @param stdout - CLI stdout
+ */
+export function parseOllamaList(stdout: string): string[] {
+  const lines = stdout
+    .split("\n")
+    .map((l) => l.trim())
+    .filter(Boolean);
+  const out: string[] = [];
+  for (const line of lines) {
+    if (/^NAME\b/i.test(line)) continue;
+    const name = line.split(/\s+/)[0];
+    if (name) out.push(name);
+  }
+  return out;
+}
+
+/**
+ * Parse `ollama ps` tabular output into running model names.
+ *
+ * @param stdout - CLI stdout
+ */
+export function parseOllamaPs(stdout: string): string[] {
+  return parseOllamaList(stdout);
+}
+
+/**
+ * Normalize installed tag for comparison (`qwen3.5:9b` vs `qwen3.5:9b-mlx`).
+ *
+ * @param id - Catalog or installed id
+ * @param installed - Installed names
+ */
+export function isInstalled(id: string, installed: readonly string[]): boolean {
+  const base = id.split(":")[0] ?? id;
+  return installed.some((name) => {
+    if (name === id) return true;
+    if (name.startsWith(`${id}-`)) return true;
+    if (name.startsWith(`${base}:`) && id.startsWith(`${base}:`)) {
+      // treat mlx / quant suffixes as matching the same family:tag when prefix matches
+      const instTag = name.slice(base.length + 1);
+      const wantTag = id.slice(base.length + 1);
+      return (
+        instTag === wantTag || instTag.startsWith(`${wantTag}-`) || wantTag.startsWith(instTag)
+      );
+    }
+    return false;
+  });
+}
+
+/**
+ * Detect Ollama via CLI then HTTP fallback.
+ *
+ * @param options - Base URL / spawn / fetch seams
+ */
+export async function detectOllama(
+  options: {
+    readonly baseUrl?: string;
+    readonly fetch?: typeof globalThis.fetch;
+    readonly run?: (cmd: readonly string[]) => Promise<{ code: number; stdout: string }>;
+  } = {},
+): Promise<OllamaDetectResult> {
+  const baseUrl = normalizeOllamaBaseUrl(options.baseUrl ?? OLLAMA_DEFAULT_BASE_URL);
+  const run =
+    options.run ??
+    (async (cmd) => {
+      const proc = Bun.spawn([...cmd], { stdout: "pipe", stderr: "pipe" });
+      const stdout = await new Response(proc.stdout).text();
+      const code = await proc.exited;
+      return { code, stdout };
+    });
+
+  let installed: string[] = [];
+  let running: string[] = [];
+  let available = false;
+
+  try {
+    const list = await run(["ollama", "list"]);
+    if (list.code === 0) {
+      available = true;
+      installed = parseOllamaList(list.stdout);
+    }
+    const ps = await run(["ollama", "ps"]);
+    if (ps.code === 0) {
+      running = parseOllamaPs(ps.stdout);
+    }
+  } catch {
+    // CLI missing — try HTTP
+  }
+
+  if (!available) {
+    const fetchFn = options.fetch ?? globalThis.fetch;
+    try {
+      const res = await fetchFn(`${baseUrl}/api/tags`, {
+        method: "GET",
+        signal: AbortSignal.timeout(2_000),
+      });
+      if (res.ok) {
+        available = true;
+        const json = (await res.json()) as { models?: readonly { name?: string }[] };
+        installed = (json.models ?? [])
+          .map((m) => m.name)
+          .filter((n): n is string => typeof n === "string");
+      }
+    } catch {
+      // unavailable
+    }
+  }
+
+  const curatedInstalled = ALL_CURATED.filter((m) => isInstalled(m.id, installed));
+  const curatedMissing = ALL_CURATED.filter((m) => !isInstalled(m.id, installed));
+
+  return {
+    available,
+    baseUrl,
+    installed,
+    running,
+    curatedInstalled,
+    curatedMissing,
+  };
+}
+
+/**
+ * Best-effort total system RAM in GB.
+ */
+export function detectTotalRamGb(): number | null {
+  try {
+    if (process.platform === "darwin") {
+      const proc = Bun.spawnSync(["sysctl", "-n", "hw.memsize"], { stdout: "pipe" });
+      if (proc.exitCode === 0) {
+        const bytes = Number(proc.stdout.toString().trim());
+        if (Number.isFinite(bytes) && bytes > 0) return Math.round(bytes / 1024 ** 3);
+      }
+    }
+    if (process.platform === "linux") {
+      const proc = Bun.spawnSync(["awk", "/MemTotal/ {print $2}", "/proc/meminfo"], {
+        stdout: "pipe",
+      });
+      if (proc.exitCode === 0) {
+        const kb = Number(proc.stdout.toString().trim());
+        if (Number.isFinite(kb) && kb > 0) return Math.round(kb / 1024 ** 2);
+      }
+    }
+  } catch {
+    // ignore
+  }
+  return null;
+}

package/src/ai-setup/from-pref.ts

@@ -0,0 +1,99 @@
+/**
+ * Convert persisted create-defaults AI prefs ↔ apply input.
+ */
+
+import type { CreateAiPref } from "../create-defaults.ts";
+import type { AiSetupApplyInput } from "./apply.ts";
+
+/**
+ * Build apply input from saved create-defaults (reuse path).
+ *
+ * @param pref - Saved AI preference
+ */
+export function applyInputFromAiPref(pref: CreateAiPref): AiSetupApplyInput | null {
+  if (!pref.enabled || !pref.driver) return null;
+  const driver = pref.driver as AiSetupApplyInput["driver"];
+  if (
+    driver !== "ollama" &&
+    driver !== "anthropic" &&
+    driver !== "openai-compatible" &&
+    driver !== "mock"
+  ) {
+    return null;
+  }
+  return {
+    driver,
+    ...(pref.baseUrl ? { baseUrl: pref.baseUrl } : {}),
+    ...(pref.chatModel ? { chatModel: pref.chatModel } : {}),
+    ...(pref.visionModel !== undefined && pref.visionModel !== null
+      ? { visionModel: pref.visionModel }
+      : {}),
+    ...(pref.embedModel !== undefined && pref.embedModel !== null
+      ? { embedModel: pref.embedModel }
+      : {}),
+    ...(pref.apiKeyEnv ? { apiKeyEnv: pref.apiKeyEnv } : {}),
+  };
+}
+
+/**
+ * Merge model choices into a CreateAiPref for persistence.
+ *
+ * @param base - Provider / driver prefs
+ * @param apply - Model apply input
+ */
+export function aiPrefWithModels(
+  base: CreateAiPref,
+  apply: AiSetupApplyInput | null,
+): CreateAiPref {
+  if (!apply) return base;
+  return {
+    ...base,
+    driver: apply.driver,
+    chatModel: apply.chatModel ?? null,
+    visionModel: apply.visionModel ?? null,
+    embedModel: apply.embedModel ?? null,
+    baseUrl: apply.baseUrl ?? null,
+    apiKeyEnv: apply.apiKeyEnv ?? null,
+  };
+}
+
+/**
+ * Non-interactive defaults for `--yes --ai` (mirrors `oke ai setup --yes`).
+ *
+ * @param provider - Menu id
+ */
+export function nonInteractiveAiApply(provider: string): AiSetupApplyInput {
+  if (provider === "ollama") {
+    return {
+      driver: "ollama",
+      baseUrl: process.env.OKE_AI_URL ?? "http://127.0.0.1:11434",
+      chatModel: "gemma4:e4b",
+      visionModel: "qwen3-vl:4b",
+      embedModel: "nomic-embed-text",
+    };
+  }
+  if (provider === "anthropic") {
+    return {
+      driver: "anthropic",
+      chatModel: "claude-sonnet-4-20250514",
+      visionModel: null,
+      embedModel: null,
+      apiKeyEnv: "ANTHROPIC_API_KEY",
+    };
+  }
+  const baseUrls: Record<string, string | undefined> = {
+    openai: "https://api.openai.com/v1",
+    openrouter: "https://openrouter.ai/api/v1",
+    lmstudio: "http://127.0.0.1:1234/v1",
+    gemini: undefined,
+    custom: undefined,
+  };
+  return {
+    driver: "openai-compatible",
+    ...(baseUrls[provider] !== undefined ? { baseUrl: baseUrls[provider] } : {}),
+    chatModel: "gpt-4o-mini",
+    visionModel: null,
+    embedModel: null,
+    apiKeyEnv: "OPENAI_API_KEY",
+  };
+}