Genarrative/apps/ai-game-creator-shell/src/services/llmModelCatalog.ts

import { resolveTauriInvoke } from '../app/tauri';
import type { GameCreatorAppConfigView } from '../app/types';
import { type ClientLlmModelCatalog, loadClientLlmModels } from './clientApi';

export type { ClientLlmModel, ClientLlmModelCatalog } from './clientApi';

let cached: ClientLlmModelCatalog | null = null;
let inFlight: Promise<ClientLlmModelCatalog> | null = null;
let source = '';
let generation = 0;
let localRevision = 0;
export const LLM_CONFIG_CHANGED_EVENT = 'agc-llm-config-changed';
export class LlmModelCatalogConfigError extends Error {}

export function notifyLlmConfigChanged() {
  generation += 1;
  cached = null;
  inFlight = null;
  source = '';
  window.dispatchEvent(new Event(LLM_CONFIG_CHANGED_EVENT));
}

async function loadEffectiveCatalog(epoch: number) {
  const invoke = resolveTauriInvoke();
  if (!invoke) {
    throw new LlmModelCatalogConfigError('本地模型服务不可用');
  }
  let config: GameCreatorAppConfigView | undefined;
  try {
    config = await invoke<GameCreatorAppConfigView>(
      'read_game_creator_app_config',
    );
  } catch (error) {
    if (epoch === generation) cached = null;
    throw new LlmModelCatalogConfigError('读取客户端配置失败');
  }
  if (epoch !== generation) throw new Error('模型配置已更新');
  const llm = config?.config.llm;
  const nextSource = llm?.customEnabled
    ? JSON.stringify(['custom', llm.baseUrl, llm.visibleModels ?? []])
    : 'official';
  if (source !== nextSource) {
    source = nextSource;
    cached = null;
  }
  if (llm?.customEnabled) {
    if (cached) return cached;
    const ids = llm.visibleModels ?? [];
    return {
      defaultModelId: ids[0] ?? '',
      models: ids.map((id) => ({
        id,
        displayName: id,
        agentMode: 'codex' as const,
      })),
      revision: --localRevision,
    };
  }
  // 官方模型目录的凭据、请求和响应解析归 Rust；渲染层只接收目录快照。
  return loadClientLlmModels();
}

/** 最近一次成功读取的模型目录，用于首屏渲染与刷新失败时兜底。 */
export function cachedLlmModelCatalog() {
  return cached;
}

/**
 * 读取模型目录；同一时刻只保留一个在途请求，重复触发复用同一结果。
 * 读取失败时保留上一次成功目录，不覆盖调用方已生效的选择。
 */
export function refreshLlmModelCatalog() {
  if (inFlight) return inFlight;
  const epoch = generation;
  const request = loadEffectiveCatalog(epoch)
    .then((catalog) => {
      if (epoch !== generation) throw new Error('模型配置已更新');
      cached = catalog;
      return catalog;
    })
    .finally(() => {
      if (inFlight === request) inFlight = null;
    });
  inFlight = request;
  return request;
}

export function resetLlmModelCatalogCacheForTest() {
  generation += 1;
  source = '';
  cached = null;
  inFlight = null;
}