import { createOpenRouter } from '@openrouter/ai-sdk-provider';
import {
  DEFAULT_OPENROUTER_MODEL_ID,
  isOpenRouterModelId,
  type OpenRouterModelId,
} from '@/lib/openrouter';

export function getDefaultChatModelId(): OpenRouterModelId {
  const configuredModel = process.env.OPENROUTER_MODEL;
  return isOpenRouterModelId(configuredModel) ? configuredModel : DEFAULT_OPENROUTER_MODEL_ID;
}

export function getChatModel(modelId: OpenRouterModelId = getDefaultChatModelId()) {
  const openrouter = createOpenRouter({
    apiKey: process.env.OPENROUTER_API_KEY,
  });

  return openrouter(modelId);
}