frontend/src/lib/models.ts

export interface ModelConfig {
  id: string;
  name: string;
  description: string;
  category: "standard" | "reasoning";
  recommended?: boolean;
  speed: "fast" | "medium" | "slow";
  contextWindow: string;
  costLevel: "low" | "medium" | "high";
}

export const AVAILABLE_MODELS = {
  standard: [
    {
      id: "gpt-5-mini",
      name: "GPT-5 Mini",
      description: "Latest model for most tasks",
      category: "standard" as const,
      recommended: true,
      speed: "fast" as const,
      contextWindow: "128K",
      costLevel: "low" as const,
    },
    {
      id: "gpt-5",
      name: "GPT-5",
      description: "Latest model for most tasks",
      category: "standard" as const,
      recommended: true,
      speed: "fast" as const,
      contextWindow: "128K",
      costLevel: "low" as const,
    },
    {
      id: "gpt-4o-mini",
      name: "GPT-4o Mini",
      description: "Cost-effective model for most tasks",
      category: "standard" as const,
      recommended: true,
      speed: "fast" as const,
      contextWindow: "128K",
      costLevel: "low" as const,
    },
    {
      id: "gpt-4o",
      name: "GPT-4o",
      description: "Balanced performance and capabilities",
      category: "standard" as const,
      speed: "medium" as const,
      contextWindow: "128K",
      costLevel: "medium" as const,
    },
    {
      id: "gpt-4.1-nano",
      name: "GPT-4.1 Nano",
      description: "Fastest and most cost-effective latest model",
      category: "standard" as const,
      speed: "fast" as const,
      contextWindow: "1M",
      costLevel: "low" as const,
    },
    {
      id: "gpt-4.1-mini",
      name: "GPT-4.1 Mini",
      description: "Balanced performance with large context",
      category: "standard" as const,
      speed: "medium" as const,
      contextWindow: "1M",
      costLevel: "medium" as const,
    },
    {
      id: "gpt-4.1",
      name: "GPT-4.1",
      description: "Latest flagship model with 1M token context",
      category: "standard" as const,
      speed: "medium" as const,
      contextWindow: "1M",
      costLevel: "high" as const,
    },
  ],
  reasoning: [
    {
      id: "o4-mini",
      name: "o4-mini",
      description: "Latest reasoning model for complex analysis",
      category: "reasoning" as const,
      speed: "slow" as const,
      contextWindow: "200K",
      costLevel: "medium" as const,
    },
    {
      id: "o3",
      name: "o3",
      description: "Advanced reasoning for complex problems",
      category: "reasoning" as const,
      speed: "slow" as const,
      contextWindow: "128K",
      costLevel: "high" as const,
    },
    {
      id: "o1",
      name: "o1",
      description: "Original reasoning model",
      category: "reasoning" as const,
      speed: "slow" as const,
      contextWindow: "128K",
      costLevel: "high" as const,
    },
  ],
};

export const getModelById = (modelId: string): ModelConfig | undefined => {
  const allModels = [
    ...AVAILABLE_MODELS.standard,
    ...AVAILABLE_MODELS.reasoning,
  ];
  return allModels.find((model) => model.id === modelId);
};

export const getDefaultModel = (): string => "gpt-5-mini";

export function getCurrentModelName(): string {
  try {
    const savedModel = localStorage.getItem("agentgraph_model_preference");
    if (savedModel && getModelById(savedModel)) {
      return getModelById(savedModel)?.name || "Unknown Model";
    }
    return getModelById(getDefaultModel())?.name || "Unknown Model";
  } catch (error) {
    console.warn("Failed to get current model name:", error);
    return getModelById(getDefaultModel())?.name || "Unknown Model";
  }
}