Junhoee's picture
Upload 98 files
f646e0b verified
Raw
History Blame Contribute Delete
737 Bytes
"""LLM 모델 설정 — env에서 읽음 (CLAUDE.md LLM 모델 규칙).
- LLM_MODEL: 기본 모델 (director, guard 보조 등 저비용 검증용)
- CHARACTER_MODEL: 캐릭터/문체 품질이 중요한 자리. 미설정 시 LLM_MODEL 사용
"""
import os
from pathlib import Path
from dotenv import load_dotenv
from google.adk.models.lite_llm import LiteLlm
# 저장소 루트 .env 로딩 (OPENAI_API_KEY 등). 이미 설정된 셸 env가 우선.
load_dotenv(Path(__file__).resolve().parent.parent / ".env")
_MODEL_ID = os.environ.get("LLM_MODEL", "openai/gpt-5.4-nano")
LLM_MODEL = LiteLlm(model=_MODEL_ID)
_CHARACTER_MODEL_ID = os.environ.get("CHARACTER_MODEL", _MODEL_ID)
CHARACTER_MODEL = LiteLlm(model=_CHARACTER_MODEL_ID)