qq群聊机器人接入ChatGPT-简介和源码

时间:2022-12-10 22:50:46

qq群聊机器人接入ChatGPT


最近 ChatGPT 很火,也注册了账号玩了玩,确实灰常强大。但是也有的小伙伴可能没办法注册账号,我就想着把qq群机器人接入ChatGPT。 过程还是比较简单顺利的。下面简单介绍一下

直接跳过介绍,查项目代码

1. ChatGPT 网页的几个接口介绍。

1.1 第一个接口 https://chat.openai.com/backend-api/moderations

控制台上看到每次都有调用这个接口,但是阻塞这个接口,也可以正常运行。所有可以忽略这个接口。

1.2 第二个接口 https://chat.openai.com/backend-api/conversation

这个接口就是发送请求,获取响应的接口。控制台上看是fetch请求,但把请求直接copy出来,发现请求总是403。

1.3 第三个接口 https://chat.openai.com/api/auth/session

最后看到这个接口,大概就是刷新token的,第二个接口之前没有调用这个接口,所以请求可能鉴权失败。

1.4 总结: 经过我的捣鼓,再加上 github 看了下别人写的插件,可以像下面这样正确的发送请求已经接受响应。

2. ChatGPT 具体请求示例

1. 复制cookie

从浏览器控制台请求的请求头,或者 application -> cookie 复制出 __Secure-next-auth.session-token 的key和value。

示例:

"__Secure-next-auth.session-token=eyJhbGciOiJkaXIiLCJlbmMiOiJBMjU2R0NNIn0..R3Kc7SzojKpBGjqD.A8_9DrrtRoHFzEiJXrfWzePQg后面省略..."

ps: (__Secure-next-auth.callback-url 和 __Host-next-auth.csrf-token 可要可不要)

2. 发起session请求,这里用java okhttp实现,没有多余封装,直接硬码

把结果 nextAuthSessionToken 和 authorization 保留下来。nextAuthSessionToken 作为下一次请求 sessionReq
方法的参数(就不用每次都复制了),nextAuthSessionToken 和 authorization 都传递到下一个接口作为参数

public static String[] sessionReq(String copyCookie) throws IOException {
        //copyCookie就是从浏览器请求的请求头,或者 application -> cookie 复制出 __Secure-next-auth.session-token 的key和value。   
        //ps: (__Secure-next-auth.callback-url 和 __Host-next-auth.csrf-token 可要可不要)
        //示例: "__Secure-next-auth.session-token=eyJhbGciOiJkaXIiLCJlbmMiOiJBMjU2R0NNIn0....后面省略"

        HashMap<String, String> headerMap = Maps.newHashMap();
        headerMap.put("user-agent", "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/107.0.0.0 Safari/537.36");
        headerMap.put("cookie", copyCookie);
        headerMap.put("accept-encoding", "gzip, deflate, br");
        //session接口
        Request getRequest = new Request.Builder()
                .url("https://chat.openai.com/api/auth/session")
                .get()
                .headers(Headers.of(headerMap))
                .build();

        Response responseSession = client.newCall(getRequest).execute();
        String result = CharStreams.toString(new InputStreamReader(responseSession.body().byteStream(), StandardCharsets.UTF_8));
        logger.info("session response: {}", result);
        Map map = objectMapper.readValue(result, Map.class);
        String setCookie = responseSession.headers().get("set-cookie");
        Map<String, String> collect = Splitter.on(";").splitToList(setCookie)
                .stream().filter(it -> it.contains("=")).map(it -> it.split("="))
                .collect(Collectors.toMap(it -> it[0], it -> it[1]));
        String nextAuthSessionToken = collect.get("__Secure-next-auth.session-token");
        String authorization = (String) map.get("accessToken");
        logger.info("nextAuthSessionToken: {}", nextAuthSessionToken);
        ChatGPTHandle.sessionToken = nextAuthSessionToken;
        logger.info("authorization: {}", authorization);
        return new String[]{nextAuthSessionToken, authorization};
    }

3. 发起 conversation 请求,同样用java okhttp实现,没有多余封装,直接硬码

requestBody 只用替换 query就行,conCookieMap 就是之前复制的cookie里的内容,这里我把三组都写上了,
__Host-next-auth.csrf-token 和 __Secure-next-auth.callback-url 都是固定值

public static String conversation(String[] auths, String requestBody) throws IOException {
        //这个 requestBody 可以作为模板写死,不同的请求只需要修改里面的query 
//        String requestBody = "{\"parent_message_id\":\"" + UUID.randomUUID()
//                + "\",\"action\":\"next\",\"messages\":[{\"role\":\"user\",\"id\":\""
//                + UUID.randomUUID() + "\",\"content\":{\"content_type\":\"text\",\"parts\":[\"" + "query" + "\"]}}]," +
//                "\"model\":\"text-davinci-002-render\"}";

        String nextAuthSessionToken = auths[0];
        String authorization = auths[1];
        //替换那个 cookie
        Map<String, String> conCookieMap = new HashMap<>(4, 1);
        //这个就是上面复制的cookie里的内容
        conCookieMap.put("__Secure-next-auth.session-token", nextAuthSessionToken);
        StringBuilder sb = new StringBuilder();
        conCookieMap.forEach((k, v) -> sb.append(k).append("=").append(v).append("; "));
        sb.deleteCharAt(sb.length() - 2);


        HashMap<String, String> hashMap = Maps.newHashMap();
        hashMap.put("accept-encoding", "gzip, deflate, br");
        hashMap.put("accept-language", "zh-CN,zh;q=0.9");
        hashMap.put("authorization", "Bearer " + authorization);
        hashMap.put("content-type", "application/json");
        hashMap.put("cookie", sb.toString().trim());
        hashMap.put("origin", "https: //chat.openai.com");
        hashMap.put("referer", "https: //chat.openai.com/chat");
        hashMap.put("user-agent", "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/107.0.0.0 Safari/537.36");
        Request post = new Request.Builder()
                .headers(Headers.of(hashMap))
                .url("https://chat.openai.com/backend-api/conversation")
                .post(RequestBody.create(MediaType.parse("application/json"), requestBody)).build();
        
        Call call = client.newCall(post);
        Response response = call.execute();
        if (response.isSuccessful()) {
            //处理response的响应消息
            String res = CharStreams.toString(new InputStreamReader(response.body().byteStream(), StandardCharsets.UTF_8));
            //这里是连续多行
            //a
            //a b
            //a b c
            //这直接取倒数第二行
            String[] split = res.split("\n");
            List<String> collect1 = Arrays.stream(split).filter(Strings::isNotBlank)
                    .collect(Collectors.toList());
            String fullLine = collect1.get(collect1.size() - 2);
            Map map1 = objectMapper.readValue(fullLine.substring(5), Map.class);
            ArrayList list = (ArrayList) ((Map) ((Map) map1.get("message")).get("content")).get("parts");
            return (String) list.get(0);
        } else {
            logger.info(response.code() + "  " + response.toString());
        }
        return "服务出错了, g了";
    }

这样基本就能返回成功了,需要的依赖是 okhttp3 和 jackson 和 guava。普通java web项目里应该都有

3. 接入qq机器人,我用的是 github 开源 mirai

3.1 基于 mirai 开发一个简单的 qq机器人,支持艾特回复。

想了一下,还是直接上代码吧,介绍累死我了。(以后有空再加)

项目地址

这里跟一下趋势,用的springboot3.0 java17开发

mirai-qq-bot

最后效果演示

qq群聊机器人接入ChatGPT-简介和源码