DeepseekUtil.py 929 B

12345678910111213141516171819202122232425262728293031323334
  1. import requests
  2. import json
  3. def chat(question):
  4. url = "https://qianfan.baidubce.com/v2/chat/completions"
  5. payload = json.dumps({
  6. "model": "deepseek-v3",
  7. "messages": [
  8. {
  9. "role": "user",
  10. "content": question
  11. }
  12. ]
  13. }, ensure_ascii=False)
  14. headers = {
  15. 'Content-Type': 'application/json',
  16. 'appid': '',
  17. 'Authorization': 'Bearer bce-v3/ALTAK-4724E5sCJcdxRCBCilJoL/641317ddac7137060721e65e688876f6c772115d'
  18. }
  19. response = requests.request("POST", url, headers=headers, data=payload.encode("utf-8"))
  20. #response.text是个json,获取choices数组中第一个元素的message的content字段
  21. answer = json.loads(response.text)
  22. answer = answer["choices"][0]["message"]["content"]
  23. print(answer)
  24. #返回answer
  25. return answer
  26. if __name__ == '__main__':
  27. print(chat('1985年今年几岁'))