chatGPT-SDK组件工程简单功能实现

public interface IOpenAiApi {

    /**
     * 文本问答
     * @param qaCompletionRequest 请求信息
     * @return                    返回结果
     */
    @POST("v1/completions")
    Single<QACompletionResponse> completions(@Body QACompletionRequest qaCompletionRequest);

    /**
     * 默认 GPT-3.5 问答模型
     * @param chatCompletionRequest 请求信息
     * @return                      返回结果
     */
    @POST("v1/chat/completions")
    Single<ChatCompletionResponse> completions(@Body ChatCompletionRequest chatCompletionRequest);
}

public interface OpenAiSession {

    /**
     * 文本问答 24年1月，方法废弃
     * @param qaCompletionRequest 请求信息
     * @return                    返回结果
     */
    QACompletionResponse completions(QACompletionRequest qaCompletionRequest);

    /**
     * 文本问答；简单请求 24年1月，方法废弃
     * @param question 请求信息
     * @return         返回结果
     */
    QACompletionResponse completions(String question);

    /**
     * 默认 GPT-3.5 问答模型
     * @param chatCompletionRequest 请求信息
     * @return                      返回结果
     */
    ChatCompletionResponse completions(ChatCompletionRequest chatCompletionRequest);

}

public class DefaultOpenAiSessionFactory implements OpenAiSessionFactory {

    private final Configuration configuration;

    public DefaultOpenAiSessionFactory(Configuration configuration) {
        this.configuration = configuration;
    }

    @Override
    public OpenAiSession openSession() {
        // 1. 日志配置
        HttpLoggingInterceptor httpLoggingInterceptor = new HttpLoggingInterceptor();
        httpLoggingInterceptor.setLevel(HttpLoggingInterceptor.Level.HEADERS);

        // 2. 开启 Http 客户端
        OkHttpClient okHttpClient = new OkHttpClient
                .Builder()
                .addInterceptor(httpLoggingInterceptor)
                .addInterceptor(new OpenAiInterceptor(configuration.getApiKey(), configuration.getAuthToken()))
                .connectTimeout(450, TimeUnit.SECONDS)
                .writeTimeout(450, TimeUnit.SECONDS)
                .readTimeout(450, TimeUnit.SECONDS)
                .build();

        // 3. 创建 API 服务
        IOpenAiApi openAiApi = new Retrofit.Builder()
                .baseUrl(configuration.getApiHost())
                .client(okHttpClient)
                .addCallAdapterFactory(RxJava2CallAdapterFactory.create())
                .addConverterFactory(JacksonConverterFactory.create())
                .build().create(IOpenAiApi.class);

        return new DefaultOpenAiSession(openAiApi);
    }
}

public static void main(String[] args) {
    HttpLoggingInterceptor httpLoggingInterceptor = new HttpLoggingInterceptor();
    httpLoggingInterceptor.setLevel(HttpLoggingInterceptor.Level.BODY);
    OkHttpClient okHttpClient = new OkHttpClient
            .Builder()
            .addInterceptor(httpLoggingInterceptor)
            .addInterceptor(chain -> {
                Request original = chain.request();
                // 从请求中获取 token 参数，并将其添加到请求路径中
                HttpUrl url = original.url().newBuilder()
                        .addQueryParameter("token", "eyJhbGciOiJIUzI1NiJ9.eyJzdWIiOiJ4ZmciLCJleHAiOjE2ODMyNzIyMjAsImlhdCI6MTY4MzI2ODYyMCwianRpIjoiOTkwMmM");
                        .build();
                Request request = original.newBuilder()
                        .url(url)
                        .header(Header.AUTHORIZATION.getValue(), "Bearer " + "sk-****" // 替换为你自己的 API Key，切勿提交真实密钥到代码仓库)
                        .header(Header.CONTENT_TYPE.getValue(), ContentType.JSON.getValue())
                        .method(original.method(), original.body())
                        .build();
                return chain.proceed(request);
            })
            .build();
    IOpenAiApi openAiApi = new Retrofit.Builder()
            .baseUrl("https://api.xfg.im/b8b6/") // 你看可以替换任意你可以使用的接口。
            .client(okHttpClient)
            .addCallAdapterFactory(RxJava2CallAdapterFactory.create())
            .addConverterFactory(JacksonConverterFactory.create())
            .build().create(IOpenAiApi.class);
    Message message = Message.builder().role(Constants.Role.USER).content("写一个java冒泡排序").build();
    ChatCompletionRequest chatCompletion = ChatCompletionRequest
            .builder()
            .messages(Collections.singletonList(message))
            .model(ChatCompletionRequest.Model.GPT_3_5_TURBO.getCode())
            .build();
    Single<ChatCompletionResponse> chatCompletionResponseSingle = openAiApi.completions(chatCompletion);
    ChatCompletionResponse chatCompletionResponse = chatCompletionResponseSingle.blockingGet();
    chatCompletionResponse.getChoices().forEach(e -> {
        System.out.println(e.getMessage());
    });
}

@Before
public void test_OpenAiSessionFactory() {
    // 1. 配置文件【如果你从小傅哥获取key会给你提供apihost，你可以分别替换下】
    Configuration configuration = new Configuration();
    configuration.setApiHost("https://api.xfg.im/b8b6/");
    configuration.setApiKey("sk-****"); // 替换为你自己的 API Key
    // 可以根据课程首页评论置顶说明获取 apihost、apikey；https://t.zsxq.com/0d3o5FKvc
    configuration.setAuthToken("如果是直接访问官网地址或者小傅哥新给的地址和key，就不需要token了");
    // 2. 会话工厂
    OpenAiSessionFactory factory = new DefaultOpenAiSessionFactory(configuration);
    // 3. 开启会话
    this.openAiSession = factory.openSession();
}

@Test
public void test_qa_completions() throws JsonProcessingException {
    QACompletionResponse response01 = openAiSession.completions("写个java冒泡排序");
    log.info("测试结果：{}", new ObjectMapper().writeValueAsString(response01.getChoices()));
}

@Test
public void test_chat_completions() {
    // 1. 创建参数
    ChatCompletionRequest chatCompletion = ChatCompletionRequest
            .builder()
            .messages(Collections.singletonList(Message.builder().role(Constants.Role.USER).content("写一个java冒泡排序").build()))
            .model(ChatCompletionRequest.Model.GPT_3_5_TURBO.getCode())
            .build();
    // 2. 发起请求
    ChatCompletionResponse chatCompletionResponse = openAiSession.completions(chatCompletion);
    // 3. 解析结果
    chatCompletionResponse.getChoices().forEach(e -> {
        log.info("测试结果：{}", e.getMessage());
    });
}