feat: 初始化工程
This commit is contained in:
21
agent/model.py
Normal file
21
agent/model.py
Normal file
@@ -0,0 +1,21 @@
|
||||
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"
|
||||
Reference in New Issue
Block a user