saeeol 1.0.2 → 1.0.4

package/package.json

@@ -1,6 +1,6 @@
 {
   "$schema": "https://json.schemastore.org/package.json",
-  "version": "1.0.2",
+  "version": "1.0.4",
   "name": "saeeol",
   "type": "module",
   "license": "Apache-2.0",

package/src/cli/cmd/init.ts

@@ -5,141 +5,43 @@ import * as prompts from "@clack/prompts"
 import { UI } from "../ui"
 import { Global } from "@saeeol/core/global"
 import { Instance } from "../../project/instance"
-import { Auth } from "../../auth"
 import { AppRuntime } from "../../effect/app-runtime"
 import { Config } from "@/config/config"
 import { Effect } from "effect"
 import { put } from "./providers-auth"
+import { ModelsDev } from "@/provider/models"
+import { map, pipe, sortBy, values } from "remeda"
 
-const { TEXT_HIGHLIGHT: H, TEXT_NORMAL: N, TEXT_DIM: D, TEXT_SUCCESS: S, TEXT_NORMAL_BOLD: B, TEXT_WARNING: W } = UI.Style
+const { TEXT_HIGHLIGHT: H, TEXT_NORMAL: N, TEXT_DIM: D, TEXT_SUCCESS: S, TEXT_NORMAL_BOLD: B } = UI.Style
 
-interface Preset {
-  id: string
-  label: string
-  provider: string
-  envVar: string
-  models: Array<{ id: string; label: string }>
-  hint: string
-  defaultModel?: string
-}
+const getModels = () => AppRuntime.runPromise(ModelsDev.Service.use((s) => s.get()))
+const refreshModels = () => AppRuntime.runPromise(ModelsDev.Service.use((s) => s.refresh(true)))
 
-const presets: Preset[] = [
-  {
-    id: "anthropic",
-    label: "Anthropic (Claude)",
-    provider: "anthropic",
-    envVar: "ANTHROPIC_API_KEY",
-    models: [
-      { id: "anthropic/claude-sonnet-4", label: "Claude Sonnet 4 (추천)" },
-      { id: "anthropic/claude-haiku-4-5", label: "Claude Haiku 4.5 (빠름/저렴)" },
-      { id: "anthropic/claude-opus-4", label: "Claude Opus 4 (강력)" },
-    ],
-    hint: "https://console.anthropic.com/settings/keys",
-    defaultModel: "anthropic/claude-sonnet-4",
-  },
-  {
-    id: "openai",
-    label: "OpenAI (GPT)",
-    provider: "openai",
-    envVar: "OPENAI_API_KEY",
-    models: [
-      { id: "openai/gpt-5", label: "GPT-5" },
-      { id: "openai/gpt-5-mini", label: "GPT-5 Mini (빠름)" },
-      { id: "openai/o3", label: "o3 (추론)" },
-    ],
-    hint: "https://platform.openai.com/api-keys",
-  },
-  {
-    id: "google",
-    label: "Google (Gemini)",
-    provider: "google",
-    envVar: "GOOGLE_GENERATIVE_AI_API_KEY",
-    models: [
-      { id: "google/gemini-2.5-pro", label: "Gemini 2.5 Pro" },
-      { id: "google/gemini-2.5-flash", label: "Gemini 2.5 Flash (빠름)" },
-    ],
-    hint: "https://aistudio.google.com/apikey",
-  },
-  {
-    id: "openrouter",
-    label: "OpenRouter (멀티 provider)",
-    provider: "openrouter",
-    envVar: "OPENROUTER_API_KEY",
-    models: [
-      { id: "openrouter/anthropic/claude-sonnet-4", label: "Claude Sonnet 4 via OpenRouter" },
-      { id: "openrouter/openai/gpt-5", label: "GPT-5 via OpenRouter" },
-      { id: "openrouter/google/gemini-2.5-pro", label: "Gemini 2.5 Pro via OpenRouter" },
-    ],
-    hint: "https://openrouter.ai/keys",
-  },
-  {
-    id: "groq",
-    label: "Groq (초고속 추론)",
-    provider: "groq",
-    envVar: "GROQ_API_KEY",
-    models: [
-      { id: "groq/llama-4-maverick-17b-128e-instruct", label: "Llama 4 Maverick" },
-    ],
-    hint: "https://console.groq.com/keys",
-  },
-  {
-    id: "github-copilot",
-    label: "GitHub Copilot (무료)",
-    provider: "github-copilot",
-    envVar: "",
-    models: [
-      { id: "github-copilot/gpt-5", label: "GPT-5 via Copilot" },
-      { id: "github-copilot/claude-sonnet-4", label: "Claude Sonnet 4 via Copilot" },
-    ],
-    hint: "GitHub 계정으로 OAuth 로그인",
-  },
-  {
-    id: "bedrock",
-    label: "Amazon Bedrock",
-    provider: "amazon-bedrock",
-    envVar: "AWS_ACCESS_KEY_ID",
-    models: [
-      { id: "amazon-bedrock/anthropic.claude-sonnet-4-20250514", label: "Claude Sonnet 4" },
-    ],
-    hint: "AWS 자격 증명 필요",
-  },
-  {
-    id: "custom",
-    label: "직접 입력 (OpenAI 호환)",
-    provider: "",
-    envVar: "",
-    models: [],
-    hint: "",
-  },
-]
+type ProviderInfo = { id: string; name: string; env: string[]; models: Record<string, unknown> }
 
 export const InitCommand = cmd({
   command: "init",
   describe: "최초 설정 — provider, API key, 모델 선택",
   builder: (y) =>
     y
-      .option("preset", {
+      .option("provider", {
         type: "string",
-        describe: "프리셋 ID (anthropic, openai, google, openrouter, groq, github-copilot, bedrock, custom)",
-        choices: presets.map((p) => p.id),
+        alias: "p",
+        describe: "provider id (예: anthropic, openai, google)",
       })
       .option("api-key", {
         type: "string",
-        describe: "API key (인터랙티브 입력 스킵)",
+        alias: "k",
+        describe: "API key",
       })
       .option("model", {
         type: "string",
+        alias: "m",
         describe: "기본 모델 (예: anthropic/claude-sonnet-4)",
       })
       .option("base-url", {
         type: "string",
-        describe: "커스텀 API base URL (custom 프리셋용)",
-      })
-      .option("yes", {
-        type: "boolean",
-        alias: "y",
-        describe: "기본값으로 자동 진행",
-        default: false,
+        describe: "커스텀 API base URL (OpenAI 호환 엔드포인트용)",
       }),
   handler: async (args) => {
     await Instance.provide({
@@ -151,36 +53,90 @@ export const InitCommand = cmd({
         prompts.intro(`${B}SAEEOL 초기 설정${N}`)
         UI.empty()
 
-        // 1. Preset 선택
-        const preset = resolvePreset(args.preset)
-        const selected = preset ?? (await promptPreset())
-        if (!selected) return
+        const config = await AppRuntime.runPromise(Config.Service.use((cfg) => cfg.get()))
 
-        // 2. API Key
-        if (selected.id === "github-copilot") {
-          prompts.log.info("GitHub Copilot은 OAuth 로그인이 필요합니다.")
-          prompts.log.info("나중에 ${H}saeeol auth login -p github-copilot${N} 으로 로그인하세요.")
-        } else if (selected.id === "custom") {
-          await handleCustom(args)
-          return
-        } else if (selected.envVar) {
-          const existingKey = process.env[selected.envVar]
-          if (existingKey) {
-            prompts.log.info(`${D}${selected.envVar} 환경변수가 이미 설정되어 있습니다.${N}`)
-          } else {
-            const key = args["api-key"] ?? (await promptApiKey(selected))
-            if (!key) return
-            await put(selected.provider, { type: "api", key })
-            prompts.log.success("API key 저장 완료")
+        // 1. Provider 목록 동적 로드
+        const s = prompts.spinner()
+        s.start("provider 목록 로드 중...")
+        await refreshModels().catch(() => {})
+        const database = await getModels()
+        s.stop(`${Object.keys(database).length}개 provider 로드 완료`)
+
+        // 2. Provider 선택
+        const disabled = new Set(config.disabled_providers ?? [])
+        const enabled = config.enabled_providers ? new Set(config.enabled_providers) : undefined
+        const providers = Object.entries(database)
+          .filter(([key]) => (enabled ? enabled.has(key) : true) && !disabled.has(key))
+          .map(([id, info]) => ({ id, name: info.name || id, env: info.env, models: info.models || {} } as ProviderInfo))
+
+        const priority: Record<string, number> = {
+          anthropic: 0, openai: 1, google: 2, "github-copilot": 3, openrouter: 4, groq: 5,
+          "amazon-bedrock": 6, azure: 7, mistral: 8, deepinfra: 9, xai: 10, cerebras: 11,
+        }
+
+        let providerId: string
+        if (args.provider) {
+          providerId = args.provider
+          const found = providers.find((p) => p.id === providerId)
+          if (!found) {
+            // 커스텀 provider — base-url 필요
+            if (!args["base-url"]) {
+              prompts.log.warn(`"${providerId}"은(는) 알려진 provider가 아닙니다. --base-url이 필요합니다.`)
+              return
+            }
           }
+        } else {
+          const sorted = pipe(
+            providers,
+            sortBy((x) => priority[x.id] ?? 99, (x) => x.name),
+          )
+          const selected = await prompts.select({
+            message: "사용할 AI Provider를 선택하세요",
+            maxItems: 12,
+            options: [
+              ...sorted.map((p) => ({ label: p.name, value: p.id })),
+              { label: "직접 입력 (OpenAI 호환)", value: "__custom__" },
+            ],
+          })
+          if (prompts.isCancel(selected)) throw new UI.CancelledError()
+          providerId = selected as string
+        }
+
+        // 3. 커스텀 엔드포인트
+        if (providerId === "__custom__") {
+          await handleCustom(args, database)
+          return
         }
 
-        // 3. 모델 선택
-        const model = args.model ?? (await promptModel(selected))
+        // 4. API Key 입력
+        const providerInfo = providers.find((p) => p.id === providerId)
+        const envVar = providerInfo?.env?.[0]
+        const existingKey = envVar ? process.env[envVar] : undefined
+
+        if (existingKey) {
+          prompts.log.info(`${D}${envVar} 환경변수가 이미 설정되어 있습니다.${N}`)
+        } else if (providerId === "github-copilot") {
+          prompts.log.info("GitHub Copilot은 OAuth 로그인이 필요합니다.")
+          prompts.log.info(`  ${H}saeeol auth login -p github-copilot${N} 으로 로그인하세요.`)
+        } else {
+          const hint = providerHint(providerId)
+          if (hint) prompts.log.info(`${D}API key 발급: ${hint}${N}`)
+
+          const key = args["api-key"] ?? (await prompts.password({
+            message: `${providerInfo?.name ?? providerId} API Key`,
+            validate: (x) => (x && x.length > 0 ? undefined : "필수 입력입니다"),
+          }))
+          if (prompts.isCancel(key)) throw new UI.CancelledError()
+          await put(providerId, { type: "api", key })
+          prompts.log.success("API key 저장 완료")
+        }
+
+        // 5. 모델 목록 동적 조회
+        const model = args.model ?? (await pickModel(providerId, database))
         if (!model) return
 
-        // 4. 설정 파일 저장
-        await saveConfig(selected, model, args["base-url"])
+        // 6. 설정 저장
+        await saveConfigFile({ $schema: "https://app.saeeol.ai/config.json", model })
 
         UI.empty()
         prompts.outro(`${S}설정 완료!${N}`)
@@ -194,111 +150,137 @@ export const InitCommand = cmd({
   },
 })
 
-function logo(): string {
-  const lines = [
-    "  ██████╗  █████╗ ███████╗███████╗██╗  ██╗",
-    "  ██╔══██╗██╔══██╗██╔════╝██╔════╝██║ ██╔╝",
-    "  ██████╔╝███████║███████╗███████╗█████╔╝ ",
-    "  ██╔═══╝ ██╔══██║╚════██║╚════██║██╔═██╗ ",
-    "  ██║     ██║  ██║███████║███████║██║  ██╗",
-    "  ╚═╝     ╚═╝  ╚═╝╚══════╝╚══════╝╚═╝  ╚═╝",
-  ]
-  return lines.map((l) => `${H}${l}${N}`).join("\n")
-}
-
-function resolvePreset(id?: string): Preset | undefined {
-  if (!id) return
-  return presets.find((p) => p.id === id)
-}
+async function pickModel(
+  providerId: string,
+  database: Record<string, any>,
+): Promise<string | undefined> {
+  const provider = database[providerId]
+  if (!provider?.models) {
+    prompts.log.warn("해당 provider의 모델 목록을 가져올 수 없습니다.")
+    return `${providerId}/default`
+  }
 
-async function promptPreset(): Promise<Preset | undefined> {
-  const selected = await prompts.select({
-    message: "사용할 AI Provider를 선택하세요",
-    options: presets.map((p) => ({
-      label: p.label,
-      value: p.id,
-      hint: p.id === "custom" ? "OpenAI 호환 API" : p.provider,
-    })),
-  })
-  if (prompts.isCancel(selected)) throw new UI.CancelledError()
-  return presets.find((p) => p.id === selected)
-}
+  const models = Object.entries(provider.models as Record<string, any>)
+    .filter(([, m]) => m.status !== "deprecated" && m.status !== "alpha")
+    .map(([id, m]) => ({
+      id: `${providerId}/${id}`,
+      name: m.name || id,
+      context: m.limit?.context,
+    }))
 
-async function promptApiKey(preset: Preset): Promise<string | undefined> {
-  prompts.log.info(`${D}API key 발급: ${preset.hint}${N}`)
-  const key = await prompts.password({
-    message: `${preset.label} API Key를 입력하세요`,
-    validate: (x) => (x && x.length > 0 ? undefined : "필수 입력입니다"),
-  })
-  if (prompts.isCancel(key)) throw new UI.CancelledError()
-  return key
-}
+  if (models.length === 0) {
+    prompts.log.warn("사용 가능한 모델이 없습니다.")
+    return `${providerId}/default`
+  }
 
-async function promptModel(preset: Preset): Promise<string | undefined> {
-  if (preset.models.length === 0) return
-  if (preset.models.length === 1 && preset.defaultModel) return preset.defaultModel
+  // 컨텍스트 길이 기준 내림차순 정렬 (큰 게 보통 더 강력)
+  models.sort((a, b) => (b.context ?? 0) - (a.context ?? 0))
 
   const selected = await prompts.select({
     message: "기본 모델을 선택하세요",
-    options: preset.models.map((m) => ({
-      label: m.label,
+    maxItems: 10,
+    options: models.map((m) => ({
+      label: m.name,
       value: m.id,
+      hint: m.context ? `${(m.context / 1000).toFixed(0)}K ctx` : undefined,
     })),
   })
   if (prompts.isCancel(selected)) throw new UI.CancelledError()
   return selected as string
 }
 
-async function handleCustom(args: Record<string, unknown>) {
+async function handleCustom(args: Record<string, unknown>, database: Record<string, any>) {
   const baseUrl = (args["base-url"] as string) ?? (await prompts.text({
-    message: "API Base URL을 입력하세요",
+    message: "API Base URL",
     placeholder: "https://api.example.com/v1",
-    validate: (x) => (x && x.startsWith("http") ? undefined : "http:// 또는 https:// 로 시작해야 합니다"),
+    validate: (x) => (x && x.startsWith("http") ? undefined : "http(s)://로 시작해야 합니다"),
   }))
   if (prompts.isCancel(baseUrl)) throw new UI.CancelledError()
 
   const key = (args["api-key"] as string) ?? (await prompts.password({
-    message: "API Key를 입력하세요",
+    message: "API Key",
     validate: (x) => (x && x.length > 0 ? undefined : "필수 입력입니다"),
   }))
   if (prompts.isCancel(key)) throw new UI.CancelledError()
 
-  const modelId = (args.model as string) ?? (await prompts.text({
-    message: "기본 모델 ID를 입력하세요",
-    placeholder: "custom/my-model",
-    validate: (x) => (x && x.length > 0 ? undefined : "필수 입력입니다"),
-  }))
-  if (prompts.isCancel(modelId)) throw new UI.CancelledError()
+  // /v1/models 호출로 동적 모델 조회
+  let modelId: string | undefined
+  try {
+    const resp = await fetch(`${baseUrl.replace(/\/+$/, "")}/models`, {
+      headers: { Authorization: `Bearer ${key}` },
+      signal: AbortSignal.timeout(10000),
+    })
+    if (resp.ok) {
+      const body = (await resp.json()) as { data?: Array<{ id: string }> }
+      const remoteModels = body.data ?? []
+      if (remoteModels.length > 0) {
+        const selected = await prompts.select({
+          message: `${remoteModels.length}개 모델 발견. 기본 모델 선택`,
+          maxItems: 10,
+          options: remoteModels.slice(0, 50).map((m) => ({ label: m.id, value: m.id })),
+        })
+        if (!prompts.isCancel(selected)) modelId = selected as string
+      }
+    }
+  } catch {
+    // /models 엔드포인트 없으면 수동 입력
+  }
 
-  const providerId = modelId.split("/")[0] || "custom"
-  await put(providerId, { type: "api", key })
+  if (!modelId) {
+    modelId = (args.model as string) ?? (await prompts.text({
+      message: "모델 ID",
+      placeholder: "gpt-4o",
+      validate: (x) => (x && x.length > 0 ? undefined : "필수 입력입니다"),
+    })) as string
+    if (prompts.isCancel(modelId)) throw new UI.CancelledError()
+  }
+
+  const pid = `custom-${Date.now().toString(36)}`
+  await put(pid, { type: "api", key })
 
-  const config = buildConfigObj(modelId, { [providerId]: { api: baseUrl } })
-  await saveConfigFile(config)
+  await saveConfigFile({
+    $schema: "https://app.saeeol.ai/config.json",
+    model: `${pid}/${modelId}`,
+    provider: {
+      [pid]: {
+        name: "Custom",
+        npm: "@ai-sdk/openai-compatible",
+        api: baseUrl,
+      },
+    },
+  })
 
   UI.empty()
   prompts.outro(`${S}설정 완료!${N}`)
 }
 
-function buildConfigObj(model: string, provider?: Record<string, unknown>) {
-  const obj: Record<string, unknown> = {
-    $schema: "https://app.saeeol.ai/config.json",
-    model,
-  }
-  if (provider && Object.keys(provider).length > 0) {
-    obj.provider = provider
+function providerHint(id: string): string | undefined {
+  const hints: Record<string, string> = {
+    anthropic: "https://console.anthropic.com/settings/keys",
+    openai: "https://platform.openai.com/api-keys",
+    google: "https://aistudio.google.com/apikey",
+    openrouter: "https://openrouter.ai/keys",
+    groq: "https://console.groq.com/keys",
+    "amazon-bedrock": "AWS 자격 증명 필요",
+    azure: "Azure OpenAI 리소스 필요",
+    mistral: "https://console.mistral.ai/api-keys",
+    deepinfra: "https://deepinfra.com/dash/api_keys",
+    xai: "https://console.x.ai",
+    cerebras: "https://cloud.cerebras.ai",
   }
-  return obj
+  return hints[id]
 }
 
-async function saveConfig(preset: Preset, model: string, baseUrl?: string) {
-  const provider: Record<string, unknown> = {}
-  if (baseUrl) {
-    const pid = model.split("/")[0] || preset.provider
-    provider[pid] = { api: baseUrl }
-  }
-  const config = buildConfigObj(model, provider)
-  await saveConfigFile(config)
+function logo(): string {
+  const lines = [
+    "  ██████╗  █████╗ ███████╗███████╗██╗  ██╗",
+    "  ██╔══██╗██╔══██╗██╔════╝██╔════╝██║ ██╔╝",
+    "  ██████╔╝███████║███████╗███████╗█████╔╝ ",
+    "  ██╔═══╝ ██╔══██║╚════██║╚════██║██╔═██╗ ",
+    "  ██║     ██║  ██║███████║███████║██║  ██╗",
+    "  ╚═╝     ╚═╝  ╚═╝╚══════╝╚══════╝╚═╝  ╚═╝",
+  ]
+  return lines.map((l) => `${H}${l}${N}`).join("\n")
 }
 
 async function saveConfigFile(config: Record<string, unknown>) {

package/src/index.ts

@@ -58,6 +58,7 @@ import { JsonMigration } from "@/storage/json-migration"
 import { Database } from "@/storage/db"
 import { errorMessage } from "./util/error"
 import { PluginCommand } from "./cli/cmd/plug"
+import { InitCommand } from "./cli/cmd/init"
 import { Heap } from "./cli/heap"
 import { drizzle } from "drizzle-orm/bun-sqlite"
 import { ensureProcessMetadata } from "@saeeol/core/util/saeeol-process"
@@ -232,6 +233,7 @@ let cli = yargs(args)
   .command(RemoteCommand) 
   .command(ConfigCLICommand) 
   .command(PluginCommand)
+  .command(InitCommand)
   .command(DbCommand)
 if (InstallationBuildKind !== "release") {
   cli = cli.command(DevSetupCommand).command(DevAliasCommand)