18 lines
406 B
Python
18 lines
406 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
|