import os

from dotenv import load_dotenv
from langchain.chat_models import init_chat_model

load_dotenv()

model = init_chat_model(
    model=os.getenv('MODEL_NAME'),
    model_provider="openai",
    base_url=os.getenv('MODEL_BASE_URL'),
    api_key=os.getenv('MODEL_API_KEY')
)

# MODEL_NAME=deepseek-v4-flash
# MODEL_BASE_URL=https://api.deepseek.com
# MODEL_API_KEY=sk-0b237d41f6bc44fc9732ea66bd7eade0

# MODEL_NAME="deepseek-v4-flash-0731"
# MODEL_BASE_URL="https://dashscope.aliyuncs.com/compatible-mode/v1"
# MODEL_API_KEY="sk-52bcd98e9c1d45908437c4e8706eefff"