import os
from langchain_openai import ChatOpenAI
from dotenv import load_dotenv

load_dotenv()

ds_api_key = os.getenv('OPENAI_API_KEY')

# 创建 DeepSeek 聊天模型实例
# base_url 指向 DeepSeek 的兼容端点，⽽⾮ OpenAI 官⽅地址
llm = ChatOpenAI(
  model_name = "deepseek-v4-flash", # DeepSeek 的对话模型
  api_key = ds_api_key, # 在 platform.deepseek.com 获取
  base_url = "https://api.deepseek.com" # DeepSeek API 地址
)

# invoke 是 LangChain 统⼀的调⽤⽅法，返回 AIMessage 对象
response = llm.invoke("⽤⼀句话解释什么是微积分")
print(response.content) # .content 拿到纯⽂本