文章详情

短信预约-IT技能 免费直播动态提醒

请输入下面的图形验证码

提交验证

短信预约提醒成功

java调用chatgpt接口,实现专属于自己的人工智能助手

2023-09-10 05:35

关注

文章目录

今天突然突发奇想,就想要用java来调用chatget的接口,实现自己的聊天机器人,但是网上找文章,属实是少的可怜(可能是不让发吧)。找到了一些文章,但是基本都是通过调用别人的库来完成的,导入其他的jar还有不低的学习成本,于是就自己使用HttpClient5写了一个,在这里讲解一下思路。

对于http调用,我使用的是比较流行的httpclient5,然后直接创建了一个springboot项目,方便以后对外提供接口。

    <parent>        <artifactId>spring-boot-starter-parentartifactId>        <groupId>org.springframework.bootgroupId>        <version>2.5.3version>    parent>    <dependencies>        <dependency>            <groupId>org.springframework.bootgroupId>            <artifactId>spring-boot-starter-webartifactId>        dependency>        <dependency>            <groupId>org.springframework.bootgroupId>            <artifactId>spring-boot-starter-testartifactId>        dependency>        <dependency>            <groupId>org.projectlombokgroupId>            <artifactId>lombokartifactId>        dependency>                <dependency>            <groupId>org.apache.httpcomponents.client5groupId>            <artifactId>httpclient5artifactId>            <version>5.2.1version>        dependency>    dependencies>

在编写代码之前,这个先给出HttpClient的Api文档 api文档

在这里插入图片描述

我们在编写代码之前需要了解官方提供的接口如何进行访问以及返回的结果是什么

请求参数

官方文档地址为 文档,请求参数必须填写的内容如下

{  "model": "gpt-3.5-turbo",  "messages": [{"role": "user", "content": "Hello!"}]}

一个是model,一个是messages。model根据自己的情况来选择,聊天的话就是gpt-3.5-turbo,下面的messages里面包含n个对象,每个对象有role和content,role表示角色,content表示内容。
下面为官方文档中的解释

在这里插入图片描述

简单理解就是我们要问问题,role就是user。如果要实现连续对话,那么就将返回的返回内容设置到messages中,role设置为返回的role。

响应参数

下面直接给出响应的内容

{ 'id': 'chatcmpl-6p9XYPYSTTRi0xEviKjjilqrWU2Ve', 'object': 'chat.completion', 'created': 1677649420, 'model': 'gpt-3.5-turbo', 'usage': {'prompt_tokens': 56, 'completion_tokens': 31, 'total_tokens': 87}, 'choices': [   {    'message': {      'role': 'assistant',      'content': 'The 2020 World Series was played in Arlington, Texas at the Globe Life Field, which was the new home stadium for the Texas Rangers.'},    'finish_reason': 'stop',    'index': 0   }  ]}

我们问问题的答案就在choices.message下的content中,而role就代表了chatGpt扮演的角色。看到这我们就应该知道该干嘛了吧肯定是创建对应的VO类啊。

创建请求和响应的VO类

下面5个类就对应了我们发送和接收的各种信息

ChatGptMessage类

@Data@NoArgsConstructor@AllArgsConstructorpublic class ChatGptMessage {        String role;        String content;}

ChatGptRequestParameter 类

@Data@NoArgsConstructor@AllArgsConstructorpublic class ChatGptRequestParameter {    String model = "gpt-3.5-turbo";    List<ChatGptMessage> messages = new ArrayList<>();    public void addMessages(ChatGptMessage message) {        this.messages.add(message);    }}

ChatGptResponseParameter 类

@Data@NoArgsConstructor@AllArgsConstructorpublic class ChatGptResponseParameter {    String id;    String object;    String created;    String model;    Usage usage;    List<Choices> choices;}

Choices 类

@Data@NoArgsConstructor@AllArgsConstructorpublic class Choices {    ChatGptMessage message;    String finish_reason;    Integer index;}

Usage 类

@Data@NoArgsConstructor@AllArgsConstructorpublic class Usage {    String prompt_tokens;    String completion_tokens;    String total_tokens;}

不说废话,首先创建一个CustomChatGpt类

public class CustomChatGpt {}

然后定义一些成员属性

        private String apiKey;        private String model = "gpt-3.5-turbo-0301";        private String url = "https://api.openai.com/v1/chat/completions";        private Charset charset = StandardCharsets.UTF_8;            private ChatGptRequestParameter chatGptRequestParameter = new ChatGptRequestParameter();

提供一个ApiKey的构造器,创建该对象必须要传入ApiKey

    public CustomChatGpt(String apiKey) {        this.apiKey = apiKey;    }

定义一个响应超时时间

        private int responseTimeout = 10000;    public void setResponseTimeout(int responseTimeout) {        this.responseTimeout = responseTimeout;    }

编写一个getAnswer方法,要求传入一个CloseableHttpClient和一个问题

    public String getAnswer(CloseableHttpClient client, String question) {        }

继续实现方法,下面会完成一些参数的创建和设置

        // 创建一个HttpPost        HttpPost httpPost = new HttpPost(url);        // 创建一个ObjectMapper,用于解析和创建json        ObjectMapper objectMapper = new ObjectMapper();        // 设置请求参数        chatGptRequestParameter.addMessages(new ChatGptMessage("user", question));        HttpEntity httpEntity = null;        try {            // 对象转换为json字符串            httpEntity = new StringEntity(objectMapper.writeValueAsString(chatGptRequestParameter), charset);        } catch (JsonProcessingException e) {            System.out.println(question + "->json转换异常");            return null;        }        httpPost.setEntity(httpEntity);

下面会完成一些配置的设置

        // 设置请求头        httpPost.setHeader(HttpHeaders.CONTENT_TYPE, "application/json");        // 设置登录凭证        httpPost.setHeader(HttpHeaders.AUTHORIZATION, "Bearer " + apiKey);        // 用于设置超时时间        RequestConfig config = RequestConfig                .custom()                .setResponseTimeout(responseTimeout, TimeUnit.MILLISECONDS)                .build();        httpPost.setConfig(config);

下面代码会提交请求,解析响应,最后返回对应问题的答案

        try {            // 提交请求            return client.execute(httpPost, response -> {                // 得到返回的内容                String resStr = EntityUtils.toString(response.getEntity(), charset);                // 转换为对象                ChatGptResponseParameter responseParameter = objectMapper.readValue(resStr, ChatGptResponseParameter.class);                String ans = "";                // 遍历所有的Choices(一般都只有一个)                for (Choices choice : responseParameter.getChoices()) {                    ChatGptMessage message = choice.getMessage();                    chatGptRequestParameter.addMessages(new ChatGptMessage(message.getRole(), message.getContent()));                    String s = message.getContent().replaceAll("\n+", "\n");                    ans += s;                }                // 返回信息                return ans;            });        } catch (IOException e) {            e.printStackTrace();        }        // 发生异常,移除刚刚添加的ChatGptMessage        chatGptRequestParameter.getMessages().remove(chatGptRequestParameter.getMessages().size()-1);        return "您当前的网络无法访问";

下面给出这个类的完整代码

import com.fasterxml.jackson.core.JsonProcessingException;import com.fasterxml.jackson.databind.ObjectMapper;import com.ttpfx.vo.ChatGptMessage;import com.ttpfx.vo.ChatGptRequestParameter;import com.ttpfx.vo.ChatGptResponseParameter;import com.ttpfx.vo.Choices;import org.apache.hc.client5.http.classic.methods.HttpPost;import org.apache.hc.client5.http.config.RequestConfig;import org.apache.hc.client5.http.impl.classic.CloseableHttpClient;import org.apache.hc.core5.http.HttpEntity;import org.apache.hc.core5.http.HttpHeaders;import org.apache.hc.core5.http.io.entity.EntityUtils;import org.apache.hc.core5.http.io.entity.StringEntity;import java.io.IOException;import java.nio.charset.Charset;import java.nio.charset.StandardCharsets;import java.util.concurrent.TimeUnit;public class CustomChatGpt {        private String apiKey;        private String model = "gpt-3.5-turbo-0301";        private String url = "https://api.openai.com/v1/chat/completions";        private Charset charset = StandardCharsets.UTF_8;        private ChatGptRequestParameter chatGptRequestParameter = new ChatGptRequestParameter();        private int responseTimeout = 1000;    public void setResponseTimeout(int responseTimeout) {        this.responseTimeout = responseTimeout;    }    public CustomChatGpt(String apiKey) {        this.apiKey = apiKey;    }    public String getAnswer(CloseableHttpClient client, String question) {        // 创建一个HttpPost        HttpPost httpPost = new HttpPost(url);        // 创建一个ObjectMapper,用于解析和创建json        ObjectMapper objectMapper = new ObjectMapper();        // 设置请求参数        chatGptRequestParameter.addMessages(new ChatGptMessage("user", question));        HttpEntity httpEntity = null;        try {            // 对象转换为json字符串            httpEntity = new StringEntity(objectMapper.writeValueAsString(chatGptRequestParameter), charset);        } catch (JsonProcessingException e) {            System.out.println(question + "->json转换异常");            return null;        }        httpPost.setEntity(httpEntity);        // 设置请求头        httpPost.setHeader(HttpHeaders.CONTENT_TYPE, "application/json");        // 设置登录凭证        httpPost.setHeader(HttpHeaders.AUTHORIZATION, "Bearer " + apiKey);        // 用于设置超时时间        RequestConfig config = RequestConfig                .custom()                .setResponseTimeout(responseTimeout, TimeUnit.MILLISECONDS)                .build();        httpPost.setConfig(config);        try {            // 提交请求            return client.execute(httpPost, response -> {                // 得到返回的内容                String resStr = EntityUtils.toString(response.getEntity(), charset);                // 转换为对象                ChatGptResponseParameter responseParameter = objectMapper.readValue(resStr, ChatGptResponseParameter.class);                String ans = "";                // 遍历所有的Choices(一般都只有一个)                for (Choices choice : responseParameter.getChoices()) {                    ChatGptMessage message = choice.getMessage();                    chatGptRequestParameter.addMessages(new ChatGptMessage(message.getRole(), message.getContent()));                    String s = message.getContent().replaceAll("\n+", "\n");                    ans += s;                }                // 返回信息                return ans;            });        } catch (IOException e) {            e.printStackTrace();        }        // 发生异常,移除刚刚添加的ChatGptMessage        chatGptRequestParameter.getMessages().remove(chatGptRequestParameter.getMessages().size()-1);        return "您当前的网络无法访问";    }}

下面就是测试代码,我们只需要传入一个CloseableHttpClient 和 question 即可

public class Test {    public static void main(String[] args) throws IOException {        CloseableHttpClient httpClient = HttpClients.createDefault();        String apiKey = "自己的ApiKey";        CustomChatGpt customChatGpt = new CustomChatGpt(apiKey);        // 根据自己的网络设置吧        customChatGpt.setResponseTimeout(20000);        while (true) {            System.out.print("\n请输入问题(q退出):");            String question = new Scanner(System.in).nextLine();            if ("q".equals(question)) break;            long start = System.currentTimeMillis();            String answer = customChatGpt.getAnswer(httpClient, question);            long end = System.currentTimeMillis();            System.out.println("该回答花费时间为:" + (end - start) / 1000.0 + "秒");            System.out.println(answer);        }        httpClient.close();    }}

下面为运行图片

在这里插入图片描述
在这里插入图片描述

对于ApiKey,只能说难者不会,会者不难,这个没办法教。

如果代码无法运行,或者运行速度及其缓慢,请使用代理,在HttpClient里面可以很轻松的使用代理

        String proxyIp = "127.0.0.1";        int proxyPort = 7890;        HttpHost httpHost = new HttpHost(proxyIp, proxyPort);

上面就是一个示例,对于代理,这里也就无法继续进行说明了。

如果我们完成了上面的功能,是不是就能够对外提供接口,然后写一个自己的网页端的ChatGpt或者弄一个聊天机器人呢?当然没问题啊


2023/3/27 23:27更新

很多人找我要这个小程序的demo,其实源代码上面都已经给出了。既然有人问,那么就给出这个demo的github地址 git地址

使用指令 git clone https://github.com/c-ttpfx/chatGPT.git 可以轻松的下载到本地


2023/3/31 19:54更新
上面这个代码大家应该发现了问题,就是返回很慢,原因就是我们请求不是异步的,是等待所有答案都给出了再返回,所以速度很慢,并且容易出现超时。

解决方法就是使用异步请求,大家请参照 使用chatgpt实现微信聊天小程序(秒回复),github开源(附带链接) ,基本可以实现秒回复

2023/3/31 14:40更新
最新的代码已经支持代理,参考使用chatgpt实现微信聊天小程序(秒回复),github开源(附带链接)

来源地址:https://blog.csdn.net/m0_51545690/article/details/129739621

阅读原文内容投诉

免责声明:

① 本站未注明“稿件来源”的信息均来自网络整理。其文字、图片和音视频稿件的所属权归原作者所有。本站收集整理出于非商业性的教育和科研之目的,并不意味着本站赞同其观点或证实其内容的真实性。仅作为临时的测试数据,供内部测试之用。本站并未授权任何人以任何方式主动获取本站任何信息。

② 本站未注明“稿件来源”的临时测试数据将在测试完成后最终做删除处理。有问题或投稿请发送至: 邮箱/279061341@qq.com QQ/279061341

软考中级精品资料免费领

  • 历年真题答案解析
  • 备考技巧名师总结
  • 高频考点精准押题
  • 2024年上半年信息系统项目管理师第二批次真题及答案解析(完整版)

    难度     813人已做
    查看
  • 【考后总结】2024年5月26日信息系统项目管理师第2批次考情分析

    难度     354人已做
    查看
  • 【考后总结】2024年5月25日信息系统项目管理师第1批次考情分析

    难度     318人已做
    查看
  • 2024年上半年软考高项第一、二批次真题考点汇总(完整版)

    难度     435人已做
    查看
  • 2024年上半年系统架构设计师考试综合知识真题

    难度     224人已做
    查看

相关文章

发现更多好内容

猜你喜欢

AI推送时光机
位置:首页-资讯-后端开发
咦!没有更多了?去看看其它编程学习网 内容吧
首页课程
资料下载
问答资讯