关于ChatGPT3.5模型的使用（java）-Toy模板网

这篇具有很好参考价值的文章主要介绍了关于ChatGPT3.5模型的使用（java）。希望对大家有所帮助。如果存在错误或未考虑完全的地方，请大家不吝赐教，您也可以点击"举报违法"按钮提交疑问。

目的

最近网上关于ChatGPT的内容特别多，但是浏览大多数文章，能够完整拿来即用的代码却少之又少，索性自己写了一份以** gpt-3.5-turbo ** 模型为基础的demo，技术有限，仅供参考，不喜勿喷！

开发准备

想要进行利用chatGPT进行开发，前提是要申请OpenAI账号，拿到自己的秘钥。这一步劝退了大部分人，但是对于一个技术人来讲，如果连这最基础的准备都无法达到，那也不必继续研究了，网上白嫖其他人的就可以。这里我就不做赘述了！接下来直接进入主题！

封装模型调用方法

	final static private String chatCompletionUrl = "https://api.openai.com/v1/chat/completions";
    final static private String model = "gpt-3.5-turbo";
    final static private String Authorization = "Bearer sk-XXXXXXXXXXXXXXXXX";//这里使用你自己的认证信息
    public static CompletionChatResponse chatCompletion(String prompt){//prompt即要将要发送至gtp的内容
    	//这里是我根据模型请求封装的模型实体
        CompletionChatRequest completionChatRequest = new CompletionChatRequest();
        
        //封装http请求
        HttpRequest post = HttpUtil.createPost(chatCompletionUrl).timeout(300000);
        Map<String, String> headers = new HashMap<>();
        headers.put("Authorization", Authorization);
        post.addHeaders(headers);
        post.contentType("application/json;charset=UTF-8");
        completionChatRequest.setModel(model);
        ChoiceMessage message = new ChoiceMessage("user", prompt);
        
		//这里我把用户发送和机器人返回的20条内容内容存在了redis中，用于关联对话上下文场景，如果不需要，注释掉即可，直接new一个ArrayList
        List<Object> messages = getUserCacheMessages();
        int size = messages.size();

        if (size>19){
            messages.remove(0);
        }

        messages.add(message);
        setUserCacheMessages(messages);//把新的放到redis
        completionChatRequest.setMessages(messages);
        post.body(JSON.toJSONString(completionChatRequest));
        String body = post.execute().body();
        CompletionChatResponse completionChatResponse = JSON.parseObject(body, CompletionChatResponse.class);
        return completionChatResponse;
    }

获取已缓存对话上下文方法（可忽略）

	public static List<Object> getUserCacheMessages(){
        RedisCache redisCache = SpringUtils.getBean(RedisCache.class);
        List<Object> cacheMessages = redisCache.getCacheObject(JwtTokenUtil.getUserId + ":messages");
        if (StringUtils.isEmpty(cacheMessages)){
            cacheMessages= new ArrayList<>();
        }
        return cacheMessages;
    }

请求模型实体

参数的具体意义，可以自行查阅OpenAI官网进行调整


import lombok.AllArgsConstructor;
import lombok.Builder;
import lombok.Data;
import lombok.NoArgsConstructor;

import java.util.List;
import java.util.Map;

@Builder
@NoArgsConstructor
@AllArgsConstructor
@Data
public class CompletionChatRequest {
    String model;
    List<Object> messages;
    Integer maxTokens;
    Double temperature;
    Double topP;
    Integer n;
    Boolean stream;
    Integer logprobs;
    Boolean echo;
    List<String> stop;
    Double presencePenalty;
    Double frequencyPenalty;
    Integer bestOf;
    Map<String, Integer> logitBias;
    String user;
}

响应实体

import com.theokanning.openai.Usage;
import lombok.AllArgsConstructor;
import lombok.Builder;
import lombok.Data;
import lombok.NoArgsConstructor;

import java.util.List;

@Builder
@NoArgsConstructor
@AllArgsConstructor
@Data
public class CompletionChatResponse {
    String id;
    String object;
    long created;
    String model;
    List<CompletionChatChoice> choices;
    Usage usage;
}

messages实体

这里role我只用到了两种，一种是用户，另外就是机器人，其实还可以设置一个系统场景角色

import lombok.AllArgsConstructor;
import lombok.Builder;
import lombok.Data;
import lombok.NoArgsConstructor;

@Builder
@NoArgsConstructor
@AllArgsConstructor
@Data
public class ChoiceMessage {
    String role;
    String content;
}

choices实体

import lombok.AllArgsConstructor;
import lombok.Builder;
import lombok.Data;
import lombok.NoArgsConstructor;

@Builder
@NoArgsConstructor
@AllArgsConstructor
@Data
public class ChoiceMessage {
    String role;
    String content;
}