ds.py 636 B

12345678910111213141516171819
  1. import os
  2. from langchain_openai import ChatOpenAI
  3. from dotenv import load_dotenv
  4. load_dotenv()
  5. ds_api_key = os.getenv('OPENAI_API_KEY')
  6. # 创建 DeepSeek 聊天模型实例
  7. # base_url 指向 DeepSeek 的兼容端点,⽽⾮ OpenAI 官⽅地址
  8. llm = ChatOpenAI(
  9. model_name = "deepseek-v4-flash", # DeepSeek 的对话模型
  10. api_key = ds_api_key, # 在 platform.deepseek.com 获取
  11. base_url = "https://api.deepseek.com" # DeepSeek API 地址
  12. )
  13. # invoke 是 LangChain 统⼀的调⽤⽅法,返回 AIMessage 对象
  14. response = llm.invoke("⽤⼀句话解释什么是微积分")
  15. print(response.content) # .content 拿到纯⽂本