小言_互联网的博客

【原创】Unity结合OpenAI官方api实现类似chatGPT的AI聊天机器人

1933人阅读  评论(0)

一、什么是ChatGPT

        最近chatGPT爆火,网络铺天盖地的各种文章视频,各种牛逼之声。倒算不上第一时间使用,发布隔了一周多,才从同事那里听说了这么个神奇的技术。这周阳了,持续发烧在家,忙着养病也没时间去了解。等到周五退烧了,病情也缓解了,也就趁着热度试用了一下chatGPT,确实很强大,对话逻辑清晰,甚至感觉不到和我对话的是一个AI。除了能够聊天对话之外,你甚至可以叫他帮助你读代码,写代码,有点牛逼有没有!

       什么是chatGPT呢?从网上查的:chatGPT是OpenAI开发的大型预训练语言模型,是GPT-3模型的一个变体,经过训练可以在对话中生成类似人类的文本响应。

        我问了一下它chatGPT,是这么回答我的:chatGPT是一种基于语言模型的聊天机器人技术,它使用大量文本数据来学习如何与自然语言对话。它可以理解上下文,能够自然而直观的回答用户的查询,它也可以自动生成问题和句子,使对话更加流畅。

        哈哈,对自己还是很了解的嘛。

        不过,今天这篇文章也不是来科普chatGPT的,因为使用chatGPT是需要科学上网的,不用点特殊手段是没办法使用到的。当然,发布chatGPT的openAI提供了api,可以通过api来实现chatGPT的对话功能,api就不需要科学上网可直接访问了。那么我们看看怎么使用unity来开发实现一个AI聊天机器人吧。

二、开始前的准备

        我们要使用openAI提供的api实现AI聊天机器人,首先需要做几个准备工作:

        ①注册一个OpenAI的账号。目前openAI暂不支持大陆地区,怎么注册到openAI的账号,不在本文章解答范围内,请自行解决。

        ②创建API秘钥。登录账号后,在账户管理界面里,找到API Keys页面,创建一个秘钥。这里要注意,创建秘钥之后,站点会提示保存好你的秘钥,这里务必要复制保存,错过了可就没办法再复制了

        ③复制保存一下api地址与代码示例,方便使用。

        官方提供了几个代码示例,从代码示例里查看信息,获取到官方api的地址:

https://api.openai.com/v1/completionshttps://api.openai.com/v1/completions        需要传递的参数:

        "model": "text-davinci-003",
        "prompt": "",
        "temperature": 0,
        "max_tokens": 100,
        "top_p": 1,
        "frequency_penalty": 0.0,
        "presence_penalty": 0.0,
        "stop": ["\n"]

        记录这些信息就差不多了,咱们调用openAI的官方api用到信息基本都有了,现在开始尝试在unity里实现聊天机器人的功能吧。

三、编写代码实现聊天机器人功能

        接下来,我们开始在unity里编写代码,实现我们所需要的功能。

        1、编写一个类,用来保存Post的参数,参数定义参考上一节中我们记录的传递参数列表。


  
  1. [ System.Serializable] public class PostData{
  2. public string model;
  3. public string prompt;
  4. public int max_tokens;
  5. public float temperature;
  6. public int top_p;
  7. public float frequency_penalty;
  8. public float presence_penalty;
  9. public string stop;
  10. }

        2、编写一个类,用于保存OpenAI返回的数据。这里我们要事先了解一下调用openAI的api后,返回给我们的数据格式,随便找一个API调试助手就可以了,参考官方文档要求的Header和发送的参数要求(Json),拿到返回数据,即可知道数据格式。我只处理我需要的信息,因此编写了以下的类来保存返回的数据。


  
  1. /// <summary>
  2. /// 返回的信息
  3. /// </summary>
  4. [ System.Serializable] public class TextCallback{
  5. public string id;
  6. public string created;
  7. public string model;
  8. public List<TextSample> choices;
  9. [ System.Serializable] public class TextSample{
  10. public string text;
  11. public string index;
  12. public string finish_reason;
  13. }
  14. }

        3、编写Post方法,发送参数至api。这里用到了UnityWebRequest类来实现post方法。使用unity提供的JsonUtility类来实现Json格式的转换。编写代码过程注意一下:发送Raw数据的时候,使用utf-8来读取byte数组,开始的时候没用使用utf-8,openAI那边返回有报错;


  
  1. private IEnumerator GetPostData(string _postWord,System.Action<string> _callback)
  2. {
  3. var request = new UnityWebRequest (m_ApiUrl, "POST");
  4. PostData _postData = new PostData
  5. {
  6. model = m_PostDataSetting.model,
  7. prompt = _postWord,
  8. max_tokens = m_PostDataSetting.max_tokens,
  9. temperature=m_PostDataSetting.temperature,
  10. top_p=m_PostDataSetting.top_p,
  11. frequency_penalty=m_PostDataSetting.frequency_penalty,
  12. presence_penalty=m_PostDataSetting.presence_penalty,
  13. stop=m_PostDataSetting.stop
  14. };
  15. string _jsonText = JsonUtility.ToJson (_postData);
  16. byte[] data = System.Text.Encoding.UTF8.GetBytes (_jsonText);
  17. request.uploadHandler = (UploadHandler) new UploadHandlerRaw (data);
  18. request.downloadHandler = (DownloadHandler) new DownloadHandlerBuffer ();
  19. request.SetRequestHeader ( "Content-Type", "application/json");
  20. request.SetRequestHeader( "Authorization", string.Format( "Bearer {0}",m_OpenAI_Key));
  21. yield return request.SendWebRequest ();
  22. if (request.responseCode == 200) {
  23. string _msg = request.downloadHandler.text;
  24. TextCallback _textback = JsonUtility.FromJson<TextCallback> (_msg);
  25. if (_textback!= null && _textback.choices.Count > 0) {
  26. _callback(_textback.choices [ 0].text);
  27. }
  28. }
  29. }

        4、我的代码里写了一个回调函数,用来处理openAI返回的信息。代码仅做参考,因为我的应用写成了一个聊天对话机器人的形式,所以在获取到openAI返回的信息之后,会创建一个对话框,并且把返回的信息输入到对话框里。获取信息之后更新窗体尺寸并且自动跳到最新的会话位置。


  
  1. private void CallBack(string _callback){
  2. _callback=_callback.Trim();
  3. ChatPrefab _chat=Instantiate(m_RobotChatPrefab,m_ChatParent);
  4. _chat.SetText(_callback);
  5. //重新计算容器尺寸
  6. LayoutRebuilder.ForceRebuildLayoutImmediate(m_rootTrans);
  7. StartCoroutine(TurnToLastLine());
  8. }
  9. private IEnumerator TurnToLastLine(){
  10. yield return new WaitForEndOfFrame();
  11. //滚动到最近的消息
  12. m_ScroTectObject.verticalNormalizedPosition= 0;
  13. }

        5、现在我们基本上已经完成了主要的代码编写。后面的一些工作就是使用unity制作一下聊天应用的界面和一些交互的代码了,就不再赘述,后面是全部代码。


  
  1. using System.Collections;
  2. using System.Collections.Generic;
  3. using UnityEngine;
  4. using UnityEngine.Networking;
  5. using UnityEngine.UI;
  6. public class GetOpenAI : MonoBehaviour
  7. {
  8. //API key
  9. private string m_OpenAI_Key= "你的API KEY";
  10. // 定义Chat API的URL
  11. private string m_ApiUrl = "https://api.openai.com/v1/completions";
  12. //配置参数
  13. [ SerializeField] private PostData m_PostDataSetting;
  14. //输入的信息
  15. [ SerializeField] private InputField m_InputWord;
  16. //聊天文本放置的层
  17. [ SerializeField] private Transform m_ChatParent;
  18. [ SerializeField] private RectTransform m_rootTrans;
  19. //发送聊天气泡
  20. [ SerializeField] private ChatPrefab m_PostChatPrefab;
  21. //回复的聊天气泡
  22. [ SerializeField] private ChatPrefab m_RobotChatPrefab;
  23. //滚动条
  24. [ SerializeField] private ScrollRect m_ScroTectObject;
  25. //发送信息
  26. public void SendData()
  27. {
  28. if(m_InputWord.text.Equals( ""))
  29. return;
  30. string _msg=m_InputWord.text;
  31. ChatPrefab _chat=Instantiate(m_PostChatPrefab,m_ChatParent);
  32. _chat.SetText(_msg);
  33. //重新计算容器尺寸
  34. LayoutRebuilder.ForceRebuildLayoutImmediate(m_rootTrans);
  35. StartCoroutine(TurnToLastLine());
  36. StartCoroutine (GetPostData (_msg,CallBack));
  37. m_InputWord.text= "";
  38. }
  39. //AI回复的信息
  40. private void CallBack(string _callback){
  41. _callback=_callback.Trim();
  42. ChatPrefab _chat=Instantiate(m_RobotChatPrefab,m_ChatParent);
  43. _chat.SetText(_callback);
  44. //重新计算容器尺寸
  45. LayoutRebuilder.ForceRebuildLayoutImmediate(m_rootTrans);
  46. StartCoroutine(TurnToLastLine());
  47. }
  48. private IEnumerator TurnToLastLine(){
  49. yield return new WaitForEndOfFrame();
  50. //滚动到最近的消息
  51. m_ScroTectObject.verticalNormalizedPosition= 0;
  52. }
  53. //设置AI模型
  54. public void SetAIModel(Toggle _modelType){
  55. if(_modelType.isOn){
  56. m_PostDataSetting.model=_modelType.name;
  57. }
  58. }
  59. [ System.Serializable] public class PostData{
  60. public string model;
  61. public string prompt;
  62. public int max_tokens;
  63. public float temperature;
  64. public int top_p;
  65. public float frequency_penalty;
  66. public float presence_penalty;
  67. public string stop;
  68. }
  69. private IEnumerator GetPostData(string _postWord,System.Action<string> _callback)
  70. {
  71. var request = new UnityWebRequest (m_ApiUrl, "POST");
  72. PostData _postData = new PostData
  73. {
  74. model = m_PostDataSetting.model,
  75. prompt = _postWord,
  76. max_tokens = m_PostDataSetting.max_tokens,
  77. temperature=m_PostDataSetting.temperature,
  78. top_p=m_PostDataSetting.top_p,
  79. frequency_penalty=m_PostDataSetting.frequency_penalty,
  80. presence_penalty=m_PostDataSetting.presence_penalty,
  81. stop=m_PostDataSetting.stop
  82. };
  83. string _jsonText = JsonUtility.ToJson (_postData);
  84. byte[] data = System.Text.Encoding.UTF8.GetBytes (_jsonText);
  85. request.uploadHandler = (UploadHandler) new UploadHandlerRaw (data);
  86. request.downloadHandler = (DownloadHandler) new DownloadHandlerBuffer ();
  87. request.SetRequestHeader ( "Content-Type", "application/json");
  88. request.SetRequestHeader( "Authorization", string.Format( "Bearer {0}",m_OpenAI_Key));
  89. yield return request.SendWebRequest ();
  90. if (request.responseCode == 200) {
  91. string _msg = request.downloadHandler.text;
  92. TextCallback _textback = JsonUtility.FromJson<TextCallback> (_msg);
  93. if (_textback!= null && _textback.choices.Count > 0) {
  94. _callback(_textback.choices [ 0].text);
  95. }
  96. }
  97. }
  98. public void Quit(){
  99. Application.Quit();
  100. }
  101. void Update(){
  102. if(Input.GetKeyDown(KeyCode.Escape)){
  103. Application.Quit();
  104. }
  105. }
  106. /// <summary>
  107. /// 返回的信息
  108. /// </summary>
  109. [ System.Serializable] public class TextCallback{
  110. public string id;
  111. public string created;
  112. public string model;
  113. public List<TextSample> choices;
  114. [ System.Serializable] public class TextSample{
  115. public string text;
  116. public string index;
  117. public string finish_reason;
  118. }
  119. }
  120. }

四、结束语

        花了一点时间简单实现了一个基于OpenAI的聊天机器人的功能。完整的工程文件,我挂载在Git和Gee上供参考,unity版本请使用unity2020.1及以上版本。代码水平有限,上述编码过程仅做参考,如有不恰当之处欢迎交流。

        聊天机器人效果可观看视频:
今日头条https://m.toutiao.com/is/hH5j3Tx/      完整代码可自行到到Git上下载,unity版本请使用unity2020.1及以上版本

        Github:https://github.com/zhangliwei7758/OpenAIChatRobotMasterhttps://github.com/zhangliwei7758/OpenAIChatRobotMaster

        Gitee:OpenAIChatRobotMaster: 使用unity实现的基于OpenAI官方api的AI聊天机器人示例https://gitee.com/DammonSpace/open-aichat-robot-master


转载:https://blog.csdn.net/sinat_28962939/article/details/128358522
查看评论
* 以上用户言论只代表其个人观点,不代表本网站的观点或立场