ChatGPT Java SDK: 轻松集成OpenAI强大的语言模型

chatgpt-java

ChatGPT Java SDK简介

ChatGPT Java SDK是一个开源项目,旨在为Java开发者提供一个简单易用的接口来访问OpenAI的ChatGPT语言模型。该SDK封装了与OpenAI API的通信细节,让开发者可以专注于业务逻辑的实现,而无需过多关注底层的API调用。

Image 1: stable

主要功能特性

ChatGPT Java SDK支持以下主要功能:

支持GPT-3.5和GPT-4模型
支持流式对话
支持上下文管理
支持函数调用
支持计算Token数量
支持多API密钥轮询
支持代理和反向代理设置

快速开始

要开始使用ChatGPT Java SDK,首先需要在项目中添加依赖:

Maven:

<dependency>
    <groupId>com.github.plexpt</groupId>
    <artifactId>chatgpt</artifactId>
    <version>5.1.0</version>
</dependency>

Gradle:

implementation group: 'com.github.plexpt', name: 'chatgpt', version: '5.1.0'

然后,您可以使用以下代码快速创建一个ChatGPT实例并发送请求:

Proxy proxy = Proxys.http("127.0.0.1", 1081);

ChatGPT chatGPT = ChatGPT.builder()
    .apiKey("YOUR_API_KEY")
    .proxy(proxy)
    .apiHost("https://api.openai.com/")
    .build()
    .init();

String response = chatGPT.chat("写一段七言绝句诗,题目是:火锅!");
System.out.println(response);

高级用法

流式对话

ChatGPT Java SDK支持流式对话,这对于需要实时响应的应用场景非常有用:

ChatGPTStream chatGPTStream = ChatGPTStream.builder()
    .timeout(600)
    .apiKey("YOUR_API_KEY")
    .proxy(proxy)
    .apiHost("https://api.openai.com/")
    .build()
    .init();

ConsoleStreamListener listener = new ConsoleStreamListener();
Message message = Message.of("写一段七言绝句诗,题目是:火锅!");
ChatCompletion chatCompletion = ChatCompletion.builder()
    .messages(Arrays.asList(message))
    .build();
chatGPTStream.streamChatCompletion(chatCompletion, listener);

函数调用

SDK还支持OpenAI的函数调用功能:

List<ChatFunction> functions = new ArrayList<>();
ChatFunction function = new ChatFunction();
function.setName("getCurrentWeather");
function.setDescription("获取给定位置的当前天气");
// ... 设置函数参数 ...

Message message = Message.of("上海的天气怎么样？");
ChatCompletion chatCompletion = ChatCompletion.builder()
    .model(ChatCompletion.Model.GPT_3_5_TURBO_0613.getName())
    .messages(Arrays.asList(message))
    .functions(functions)
    .build();
ChatCompletionResponse response = chatGPT.chatCompletion(chatCompletion);
// ... 处理响应 ...