21 lines
552 B
Python
21 lines
552 B
Python
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="glm-5.2"
|
|
# MODEL_BASE_URL="https://dashscope.aliyuncs.com/compatible-mode/v1"
|
|
# MODEL_API_KEY="sk-52bcd98e9c1d45908437c4e8706eefff" |