jeo-code 0.12.4 → 0.12.5

package/CHANGELOG.md

@@ -9,6 +9,19 @@ The README mirrors the latest 5 entries — regenerate with `bun run changelog:s
 
 ## [Unreleased]
 
+## [0.12.5] - 2026-10-17
+_Antigravity and Kimi provider support in interactive setup, Kimi live model cache rehydration, immediate discovery caching on provider probing._
+
+### Added
+- **Interactive Setup Provider Options** (`src/commands/setup.ts`, `src/commands/setup-helpers.ts`) — added `antigravity` (Google Cloud Code Assist OAuth) and `kimi` (Moonshot API key / Kimi Code OAuth) to interactive setup provider selection, enabling browser OAuth, bearer token paste, or API key configuration with default models (`antigravity/gemini-3-flash`, `kimi-k2.5`). Updated `buildEnabledProviders` to recognize configured `kimi` credentials.
+- **Kimi Live Model Cache Rehydration** (`src/ai/model-cache.ts`) — cache hits for Kimi OAuth (`provider === 'kimi' && source === 'oauth'`) now rehydrate `recordLiveKimiCodeModels(entry.models)` alongside OpenAI Codex models.
+- **Immediate Model Cache on Provider Probe** (`src/commands/provider.ts`) — successful `jeo provider test <id>` and `jeo provider add` probing now immediately writes discovered custom/compatible models to the disk cache (`writeModelCache([result])`).
+
+### Fixed
+- **Model Cache Key Scoping** (`src/ai/model-cache.ts`) — fixed `keyOf` helper scoping and included `kimi` in source distinction so OAuth and API key catalogs do not clobber each other.
+- **OAuth Token Preservation in Setup** (`src/commands/setup.ts`) — fixed `delete next.oauth` condition to check `Object.keys(next.oauth).length === 0` so non-Anthropic/OpenAI/Gemini OAuth credentials (such as `antigravity` or `kimi`) are not inadvertently dropped on save.
+- **Auth Import Model Reporting** (`src/commands/auth.ts`) — added `antigravity` reporting to `reportLiveModels` following Gemini OAuth import.
+
 ## [0.12.4] - 2026-10-17
 _Refresh Antigravity provider (`antigravity/*`) account catalogs with `antigravity/hub/2.23.1` so entitled Gemini 4 Argon and Gemini 3.8 Flash tiers appear directly in `/model provider antigravity`._
 

package/README.ja.md

@@ -416,10 +416,10 @@ CI は `.github/workflows/npm-publish.yml` で公開します — GitHub リリ
 
 <!-- CHANGELOG:START (auto-generated from CHANGELOG.md — run `bun run changelog:sync`) -->
 - **[Unreleased]**
+- **[0.12.5]** (2026-10-17) — Antigravity and Kimi provider support in interactive setup, Kimi live model cache rehydration, immediate discovery caching on provider probing.
 - **[0.12.4]** (2026-10-17) — Refresh Antigravity provider (`antigravity/*`) account catalogs with `antigravity/hub/2.23.1` so entitled Gemini 4 Argon and Gemini 3.8 Flash tiers appear directly in `/model provider antigravity`.
 - **[0.12.3]** (2026-10-17) — Anthropic static model catalog trimmed to claude-5.x with live API discovery, and `/route off` strictly fixes the active model with no failover.
 - **[0.12.2]** (2026-10-16) — Select Argon through the official agy full agent, with its own live catalog and session, separately from jeo's Cloud Code Assist backend.
-- **[0.12.1]** (2026-10-16) — Logged-in OpenAI subscription models were invisible on proxy-configured machines.
 
 See [CHANGELOG.md](CHANGELOG.md) for the full history.
 <!-- CHANGELOG:END -->

package/README.ko.md

@@ -417,10 +417,10 @@ CI는 `.github/workflows/npm-publish.yml`로 배포합니다 — GitHub 릴리
 
 <!-- CHANGELOG:START (auto-generated from CHANGELOG.md — run `bun run changelog:sync`) -->
 - **[Unreleased]**
+- **[0.12.5]** (2026-10-17) — Antigravity and Kimi provider support in interactive setup, Kimi live model cache rehydration, immediate discovery caching on provider probing.
 - **[0.12.4]** (2026-10-17) — Refresh Antigravity provider (`antigravity/*`) account catalogs with `antigravity/hub/2.23.1` so entitled Gemini 4 Argon and Gemini 3.8 Flash tiers appear directly in `/model provider antigravity`.
 - **[0.12.3]** (2026-10-17) — Anthropic static model catalog trimmed to claude-5.x with live API discovery, and `/route off` strictly fixes the active model with no failover.
 - **[0.12.2]** (2026-10-16) — Select Argon through the official agy full agent, with its own live catalog and session, separately from jeo's Cloud Code Assist backend.
-- **[0.12.1]** (2026-10-16) — Logged-in OpenAI subscription models were invisible on proxy-configured machines.
 
 See [CHANGELOG.md](CHANGELOG.md) for the full history.
 <!-- CHANGELOG:END -->

package/README.md

@@ -409,10 +409,10 @@ Huge thanks to [gajae-code](https://github.com/Yeachan-Heo/gajae-code) for the i
 
 <!-- CHANGELOG:START (auto-generated from CHANGELOG.md — run `bun run changelog:sync`) -->
 - **[Unreleased]**
+- **[0.12.5]** (2026-10-17) — Antigravity and Kimi provider support in interactive setup, Kimi live model cache rehydration, immediate discovery caching on provider probing.
 - **[0.12.4]** (2026-10-17) — Refresh Antigravity provider (`antigravity/*`) account catalogs with `antigravity/hub/2.23.1` so entitled Gemini 4 Argon and Gemini 3.8 Flash tiers appear directly in `/model provider antigravity`.
 - **[0.12.3]** (2026-10-17) — Anthropic static model catalog trimmed to claude-5.x with live API discovery, and `/route off` strictly fixes the active model with no failover.
 - **[0.12.2]** (2026-10-16) — Select Argon through the official agy full agent, with its own live catalog and session, separately from jeo's Cloud Code Assist backend.
-- **[0.12.1]** (2026-10-16) — Logged-in OpenAI subscription models were invisible on proxy-configured machines.
 
 See [CHANGELOG.md](CHANGELOG.md) for the full history.
 <!-- CHANGELOG:END -->

package/README.zh.md

@@ -414,10 +414,10 @@ CI 通过 `.github/workflows/npm-publish.yml` 发布 — GitHub 发布 release
 
 <!-- CHANGELOG:START (auto-generated from CHANGELOG.md — run `bun run changelog:sync`) -->
 - **[Unreleased]**
+- **[0.12.5]** (2026-10-17) — Antigravity and Kimi provider support in interactive setup, Kimi live model cache rehydration, immediate discovery caching on provider probing.
 - **[0.12.4]** (2026-10-17) — Refresh Antigravity provider (`antigravity/*`) account catalogs with `antigravity/hub/2.23.1` so entitled Gemini 4 Argon and Gemini 3.8 Flash tiers appear directly in `/model provider antigravity`.
 - **[0.12.3]** (2026-10-17) — Anthropic static model catalog trimmed to claude-5.x with live API discovery, and `/route off` strictly fixes the active model with no failover.
 - **[0.12.2]** (2026-10-16) — Select Argon through the official agy full agent, with its own live catalog and session, separately from jeo's Cloud Code Assist backend.
-- **[0.12.1]** (2026-10-16) — Logged-in OpenAI subscription models were invisible on proxy-configured machines.
 
 See [CHANGELOG.md](CHANGELOG.md) for the full history.
 <!-- CHANGELOG:END -->

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "jeo-code",
-  "version": "0.12.4",
+  "version": "0.12.5",
   "description": "Clean, highly optimized AI coding agent using spec-first loop",
   "type": "module",
   "main": "src/cli.ts",

package/src/ai/model-cache.ts

@@ -19,7 +19,7 @@ import { mkdir, readFile, writeFile } from "node:fs/promises";
 import * as os from "node:os";
 import * as path from "node:path";
 import { jeoEnv } from "../util/env";
-import { recordLiveCodexModels, recordLiveProviderModels, setAntigravityAccountScope, setOpenAIOauthAccountScope } from "./model-catalog";
+import { recordLiveCodexModels, recordLiveKimiCodeModels, recordLiveProviderModels, setAntigravityAccountScope, setOpenAIOauthAccountScope } from "./model-catalog";
 import type { Credential } from "../auth";
 import { getStoredOAuth, snapshotProvider } from "../auth/storage";
 import { getAntigravityUserAgent } from "./providers/antigravity";
@@ -158,7 +158,7 @@ export function mergeCacheEntries(
   const isScopedOAuth = (provider: ProviderName, source: ModelCacheSource): boolean =>
     (provider === "openai" || provider === "antigravity") && source === "oauth";
   const keyOf = (provider: ProviderName, baseUrl?: string, accountId?: string, source?: ModelCacheSource) =>
-    `${provider}\u0000${baseUrl?.replace(/\/+$/, "") ?? ""}\u0000${isScopedOAuth(provider, source ?? "none") ? accountId ?? "" : ""}\u0000${provider === "openai" || provider === "antigravity" ? source ?? "none" : ""}`;
+    `${provider}\u0000${baseUrl?.replace(/\/+$/, "") ?? ""}\u0000${isScopedOAuth(provider, source ?? "none") ? accountId ?? "" : ""}\u0000${provider === "openai" || provider === "antigravity" || provider === "kimi" ? source ?? "none" : ""}`;
   for (const entry of previous) {
     const scoped = isScopedOAuth(entry.provider, entry.source);
     if (entry.provider === "agy") continue;
@@ -240,6 +240,9 @@ export function applyCachedModels(cache: ModelCacheFile | null, openAIOauthAccou
     if (isOpenAIOAuth && entry.accountId && !entry.baseUrl) {
       recordLiveCodexModels(entry.models, entry.accountId);
     }
+    if (entry.provider === "kimi" && entry.source === "oauth") {
+      recordLiveKimiCodeModels(entry.models);
+    }
     applied += entry.models.length;
   }
   return applied;

package/src/ai/model-catalog.ts

@@ -501,6 +501,30 @@ export const KIMI_CODE_MODELS: readonly string[] = [
   "kimi-k2",
 ];
 
+const liveKimiCodeModels = new Set<string>();
+
+/** Record ids confirmed by the Kimi Code OAuth `/v1/models` endpoint so brand-new
+ *  subscription models work immediately without waiting for a static catalog release. */
+export function recordLiveKimiCodeModels(ids: readonly string[]): void {
+  for (const id of ids) {
+    const trimmed = id.trim();
+    if (!trimmed) continue;
+    const wire = trimmed.startsWith("kimi/") ? trimmed.slice(5) : trimmed;
+    if (wire) liveKimiCodeModels.add(wire);
+  }
+}
+
+/** Whether `model` is served by the Kimi Code OAuth subscription endpoint. */
+export function isKimiCodeModel(model: string): boolean {
+  const wire = model.startsWith("kimi/") ? model.slice(5) : model;
+  return KIMI_CODE_MODELS.includes(wire) || liveKimiCodeModels.has(wire);
+}
+
+/** Test-only reset for live Kimi Code observations. */
+export function resetLiveKimiCodeModels(): void {
+  liveKimiCodeModels.clear();
+}
+
 /** Recency compare for `releaseDate` ("YYYY-MM"): `>0` when `a` is newer; a
  *  missing date sorts oldest, so an unconfirmed date never outranks a
  *  verified-newer model. */
@@ -639,6 +663,20 @@ export function inferCatalogMetadata(modelId: string): CatalogModel | undefined
       company: "xAI",
     };
   }
+  // Kimi / Moonshot
+  const kimi = id.match(/^kimi-(k\d+|latest|for-coding)/);
+  if (kimi) {
+    return {
+      canonical: raw,
+      provider: "kimi",
+      providerModel: id,
+      contextTokens: 256_000,
+      maxOutputTokens: 16_384,
+      thinking: FULL,
+      images: false,
+      company: "Moonshot",
+    };
+  }
 
   return undefined;
 }

package/src/ai/model-discovery.ts

@@ -16,6 +16,7 @@ import {
   catalogByProvider,
   KIMI_CODE_MODELS,
   recordLiveCodexModels,
+  recordLiveKimiCodeModels,
   recordLiveProviderModels,
   setOpenAIOauthAccountScope,
   setAntigravityAccountScope,
@@ -26,7 +27,7 @@ import { openaiCompatDef } from "./providers/openai-compatible-catalog";
 import { antigravityAccountScope } from "./model-cache";
 import { getAntigravityUserAgent, resolveAntigravityProjectId } from "./providers/antigravity";
 import { discoverGoogleProjectId, ANTIGRAVITY_DISCOVERY_METADATA } from "../auth/flows/google-project";
-
+import { getKimiCommonHeaders } from "../auth/flows/kimi";
 export interface ProviderModelsResult {
   provider: ProviderName;
   /** Discovered model ids (provider-qualified where the router expects it). */
@@ -185,6 +186,15 @@ export function discoveryRequest(
       return { url: "https://api.x.ai/v1/models", headers: token ? { Authorization: `Bearer ${token}` } : {} };
     }
     case "kimi": {
+      if (cred?.kind === "oauth") {
+        return {
+          url: "https://api.kimi.com/coding/v1/models",
+          headers: {
+            ...getKimiCommonHeaders(),
+            Authorization: `Bearer ${cred.token}`,
+          },
+        };
+      }
       const token = cred?.kind === "api_key" ? cred.token : "";
       return { url: "https://api.moonshot.ai/v1/models", headers: token ? { Authorization: `Bearer ${token}` } : {} };
     }
@@ -585,6 +595,7 @@ export async function listProviderModels(
       ...(source === "oauth" && scope.accountId ? { accountId: scope.accountId } : {}),
     });
     if (provider === "openai" && source === "oauth" && !baseUrl) recordLiveCodexModels(allModels, oauthAccountId);
+    if (provider === "kimi" && source === "oauth") recordLiveKimiCodeModels(allModels);
     return {
       provider,
       models,
@@ -659,6 +670,12 @@ export async function discoverModels(
           return [primary, await listProviderModels(p, { ...discoveryOpts, baseUrl: undefined, preferOAuth: true, codexCatalog: true })];
         }
       }
+      if (p === "kimi" && primary.source !== "oauth") {
+        const credential = await resolveCredential("kimi");
+        if (credential.kind === "oauth") {
+          return [primary, await listProviderModels(p, { ...discoveryOpts, preferOAuth: true })];
+        }
+      }
       return [primary];
     }),
   );

package/src/ai/model-manager.ts

@@ -6,7 +6,7 @@ import "./register-providers"; // side-effect: registers built-in adapters into
 import type { CallOptions, Message, ProviderAdapter, ProviderName } from "./types";
 import { expandAlias, resolveModelId, effectiveAliasesFor, BUILTIN_ALIASES } from "./model-registry";
 import { findCatalogEntry, type ModelCatalogEntry } from "./model-catalog-compat";
-import { toProviderModel, CODEX_MODELS, KIMI_CODE_MODELS, findCatalogModel, isCodexModel, findLiveCatalogModel } from "./model-catalog";
+import { toProviderModel, CODEX_MODELS, KIMI_CODE_MODELS, findCatalogModel, isCodexModel, isKimiCodeModel, findLiveCatalogModel } from "./model-catalog";
 import { xaiCredential } from "./providers/xai";
 import { allCompatNames, isOpenAICompatProvider } from "./providers/openai-compatible-catalog";
 import { withRetry, defaultRetryable, withConnectionContext, isConnectionError, ConnectionContextError, type RetryOptions } from "../util/retry";
@@ -329,8 +329,7 @@ export function oauthServesModel(provider: AuthProvider, model: string): boolean
     return isCodexModel(model);
   }
   if (provider === "kimi") {
-    const wire = model.startsWith("kimi/") ? model.slice(5) : model;
-    return KIMI_CODE_MODELS.includes(wire);
+    return isKimiCodeModel(model);
   }
   if (isOAuthProvider(provider) && OAUTH_FLOW_REGISTRY[provider].verifiedEndToEnd === false) return false;
   return true;

package/src/commands/auth.ts

@@ -152,7 +152,7 @@ async function runAuthLogin(rest: string[]): Promise<void> {
  * already succeeded and is not rolled back by this check. Prints only model
  * ids/count — no tokens/secrets.
  */
-const LIVE_MODEL_REPORT_PROVIDERS = new Set<AuthProvider>(["openai", "anthropic", "antigravity"]);
+const LIVE_MODEL_REPORT_PROVIDERS = new Set<AuthProvider>(["openai", "anthropic", "antigravity", "kimi"]);
 
 async function reportLiveModels(provider: AuthProvider): Promise<void> {
   try {
@@ -249,6 +249,7 @@ async function runAuthImport(rest: string[]): Promise<void> {
     }
     const expiryMsg = expires ? fmtExpiry(expires) : "";
     console.log(`Expiry status:${expiryMsg || " valid"}`);
+    await reportLiveModels("antigravity");
   } catch (err: any) {
     console.log(`[FAILED] ${err.message}`);
     process.exitCode = 1;
@@ -335,8 +336,16 @@ async function selectProvider(
   console.log("  1) anthropic   (real PKCE OAuth, verified end-to-end)");
   console.log("  2) openai      (real PKCE OAuth, Codex backend)");
   console.log("  3) gemini      (real Google OAuth, Cloud Code Assist)");
-  const ans = (await rl.question("Choose [1-3]: ")).trim();
-  const map: Record<string, AuthProvider> = { "1": "anthropic", "2": "openai", "3": "gemini" };
+  console.log("  4) antigravity (Google Cloud Code Assist OAuth)");
+  console.log("  5) kimi        (Kimi Code OAuth device authorization)");
+  const ans = (await rl.question("Choose [1-5]: ")).trim();
+  const map: Record<string, AuthProvider> = {
+    "1": "anthropic",
+    "2": "openai",
+    "3": "gemini",
+    "4": "antigravity",
+    "5": "kimi",
+  };
   const p = map[ans];
   if (!p) {
     console.log("Invalid choice.");

package/src/commands/provider.ts

@@ -31,6 +31,7 @@ import {
   normalizeProviderId,
 } from "../ai/providers/custom-providers";
 import { listProviderModels } from "../ai/model-discovery";
+import { writeModelCache } from "../ai/model-cache";
 import { describeProvider } from "../ai/provider-status";
 
 const USAGE = [
@@ -186,6 +187,7 @@ async function probe(id: string, opts: { failOnError: boolean }): Promise<void>
     if (shown.length) console.log(`              ${shown.join(", ")}${result.models.length > shown.length ? ", …" : ""}`);
     const first = result.models[0];
     console.log(`  status    : OK — use it with 'jeo launch --model ${first ? qualify(first) : `${id}/<model>`}'`);
+    await writeModelCache([result]);
     return;
   }
 

package/src/commands/setup-helpers.ts

@@ -76,6 +76,7 @@ export function buildEnabledProviders(config: Config): string[] {
   if (cfg.providers?.anthropic || cfg.oauth?.anthropic) enabled.push("anthropic");
   if (cfg.providers?.openai || cfg.oauth?.openai) enabled.push("openai");
   if (cfg.providers?.gemini || cfg.oauth?.gemini || cfg.providers?.antigravity || cfg.oauth?.antigravity) enabled.push("google-oauth");
+  if (cfg.providers?.kimi || cfg.oauth?.kimi) enabled.push("kimi");
   if (cfg.ollamaBaseUrl) enabled.push(`ollama(${cfg.ollamaBaseUrl})`);
   if (cfg.openaiBaseUrl) enabled.push(`openai-compatible(${cfg.openaiBaseUrl})`);
   return enabled;

package/src/commands/setup.ts

@@ -24,12 +24,14 @@ function reportModelChoice(r: { warning?: string; suggestions: string[] }): void
   if (r.suggestions.length) console.log(`  Did you mean: ${r.suggestions.join(", ")}?`);
 }
 
-type ProviderChoice = "anthropic" | "openai" | "gemini" | "ollama" | "lmstudio" | "openai-compatible";
+type ProviderChoice = "anthropic" | "openai" | "gemini" | "antigravity" | "kimi" | "ollama" | "lmstudio" | "openai-compatible";
 
 const DEFAULT_MODELS: Record<ProviderChoice, string> = {
   anthropic: "claude-sonnet-4-6",
   openai: "gpt-4o",
   gemini: "gemini-2.0-flash",
+  antigravity: "antigravity/gemini-3-flash",
+  kimi: "kimi-k2.5",
   ollama: "ollama/llama3.1:8b",
   lmstudio: "openai/local-model",
   "openai-compatible": "openai/local-model",
@@ -87,16 +89,19 @@ export async function runSetupCommand(): Promise<void> {
   console.log("  1) anthropic           — Claude (API key or OAuth bearer)");
   console.log("  2) openai              — GPT (API key or OAuth bearer)");
   console.log("  3) gemini              — Google AI Studio (API key or OAuth bearer)");
-  console.log("  4) ollama              — local Ollama (no auth)");
-  console.log("  5) lmstudio            — local LM Studio (OpenAI-compatible)");
-  console.log("  6) openai-compatible   — custom OpenAI-compatible (vLLM, llama-cpp-server, ...)");
-  console.log("  7) skip                — keep existing values, just change defaults\n");
-
-  const sel = (await rl.question("Configure which provider? [1-7]: ")).trim();
+  console.log("  4) antigravity         — Google Cloud Code Assist (OAuth)");
+  console.log("  5) kimi                — Kimi / Moonshot (API key or Kimi Code OAuth)");
+  console.log("  6) ollama              — local Ollama (no auth)");
+  console.log("  7) lmstudio            — local LM Studio (OpenAI-compatible)");
+  console.log("  8) openai-compatible   — custom OpenAI-compatible (vLLM, llama-cpp-server, ...)");
+  console.log("  9) skip                — keep existing values, just change defaults\n");
+
+  const sel = (await rl.question("Configure which provider? [1-9]: ")).trim();
   const map: Record<string, ProviderChoice | "skip"> = {
     "1": "anthropic", "2": "openai", "3": "gemini",
-    "4": "ollama",    "5": "lmstudio", "6": "openai-compatible",
-    "7": "skip",
+    "4": "antigravity", "5": "kimi",
+    "6": "ollama",    "7": "lmstudio", "8": "openai-compatible",
+    "9": "skip",
   };
   const choice = map[sel] ?? "skip";
 
@@ -108,7 +113,7 @@ export async function runSetupCommand(): Promise<void> {
   next.providers = next.providers || {};
   next.oauth = next.oauth || {};
 
-  if (choice === "anthropic" || choice === "openai" || choice === "gemini") {
+  if (choice === "anthropic" || choice === "openai" || choice === "gemini" || choice === "antigravity" || choice === "kimi") {
     const authMode = (
       await rl.question("Auth mode: (b)rowser OAuth login, (t)oken paste, or api (k)ey? [b]: ")
     ).trim().toLowerCase();
@@ -223,7 +228,7 @@ export async function runSetupCommand(): Promise<void> {
   rl.close();
 
   // Drop empty oauth/providers to keep config tidy.
-  if (next.oauth && !next.oauth.anthropic && !next.oauth.openai && !next.oauth.gemini) delete next.oauth;
+  if (next.oauth && Object.keys(next.oauth).length === 0) delete next.oauth;
 
   await saveGlobalConfig(next);
   console.log("\n[SUCCESS] Configuration saved to ~/.jeo/config.json");