28 lines
785 B
Python
28 lines
785 B
Python
"""
|
|
模型工厂 — 集中管理所有 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"),
|
|
)
|