当前位置:   article > 正文

怎么调用文心一言的api接口生成一个简单的聊天机器人(python代码)_文心一言python接口

文心一言python接口

寒假在学习大模型,但也没弄出多少眉目,电脑性能还有点小问题,大模型总跑不起来,只会简单调用一下现有的大模型的接口,例如:文心一言,下面展示一下代码:

  1. import tkinter as tk
  2. import requests
  3. import json
  4. #此处需填入自己的API_KEY以及SECRET_KEY
  5. API_KEY = "*****"
  6. SECRET_KEY = "*******"
  7. class ChatBot:
  8. def __init__(self, master):
  9. self.master = master
  10. master.title("Chat Bot")
  11. # 创建控件
  12. self.message_input = tk.Entry(master, width=50)
  13. self.send_button = tk.Button(master, text="Send", command=self.send_message)
  14. self.chat_box = tk.Text(master, width=60, height=20)
  15. # 布局控件
  16. self.message_input.pack(side=tk.LEFT, padx=10)
  17. self.send_button.pack(side=tk.LEFT)
  18. self.chat_box.pack(side=tk.BOTTOM, pady=20)
  19. def send_message(self):
  20. message = self.message_input.get()
  21. self.message_input.delete(0, tk.END)
  22. self.chat_box.insert(tk.END, "You: " + message + "\n")
  23. response = self.get_response(message)
  24. self.chat_box.insert(tk.END, "Bot: " + response + "\n")
  25. def get_response(self, message):
  26. url = "https://aip.baidubce.com/rpc/2.0/ai_custom/v1/wenxinworkshop/chat/completions?access_token=" + self.get_access_token()
  27. payload = json.dumps({
  28. "messages": [
  29. {
  30. "role": "user",
  31. "content": message
  32. }
  33. ]
  34. })
  35. headers = {
  36. 'Content-Type': 'application/json'
  37. }
  38. response = requests.request("POST", url, headers=headers, data=payload)
  39. return response.json()["result"]
  40. def get_access_token(self):
  41. url = "https://aip.baidubce.com/oauth/2.0/token"
  42. params = {"grant_type": "client_credentials", "client_id": API_KEY, "client_secret": SECRET_KEY}
  43. return str(requests.post(url, params=params).json().get("access_token"))
  44. root = tk.Tk()
  45. chat_bot = ChatBot(root)
  46. root.mainloop()

至于如何获取这其中的两个key,网上有太多文章了,大家可以去找找看,比如这篇文章:百度文心一言API申请和python/Go使用教程 - 知乎 (zhihu.com)

不过内容我没细看,如果不满足要求的话可以再找找。之前我看过的那篇很有用的文章本来想推荐给大家,刚刚发现不存在了。因为网上相关太多了,我就不写了,当然,如果大家想看,我也可以出。

结果如下所示:

之前录的视频没保存,找不着了,人又比较懒,下面是我之前发在b站的视频,大家可以看看,当然,愿意关注的话,也可以关注下,虽然我b站发布的内容好像和技术没啥关系哈:

python调用文心一言大模型接口生成一个简陋版聊天机器人_哔哩哔哩_bilibili

这次就这样啦,与大家共勉。

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/小小林熬夜学编程/article/detail/497179
推荐阅读
相关标签
  

闽ICP备14008679号