Files
NBATransfer/NBATransfer-backend/modelapiservice/__init__.py
2025-12-14 15:40:49 +08:00

11 lines
377 B
Python

from .DeepSeek.service import DeepSeekService
from .NanoBanana.service import NanoBananaService
def get_model_service(model_name: str):
"""根据模型名称获取对应的服务实例"""
if model_name.startswith('deepseek-'):
return DeepSeekService()
else:
# 默认使用 Nano Banana 服务 (包括文生图等)
return NanoBananaService()