"""
模型工厂 — 集中管理所有 LLM 模型实例

用法：
    from common.models import minimax_model, deepseek_model
"""

import os
from google.adk.models.lite_llm import LiteLlm

# ============================================================
# MiniMax 模型
# ============================================================
minimax_model = LiteLlm(
    model=os.getenv("MINIMAX_MODEL"),
    api_base=os.getenv("MINIMAX_API_BASE"),
    api_key=os.getenv("MINIMAX_API_KEY"),
)

# ============================================================
# DeepSeek 模型
# ============================================================
deepseek_model = LiteLlm(
    model=os.getenv("DEEPSEEK_MODEL"),
    api_base=os.getenv("DEEPSEEK_API_BASE"),
    api_key=os.getenv("DEEPSEEK_API_KEY"),
)