Java开发者实战指南：如何高效调用ChatGPT API并处理常见问题

1次阅读

共计 3782 个字符，预计需要花费 10 分钟才能阅读完成。

ChatGPT API 是 OpenAI 提供的一项强大的人工智能服务，它允许开发者将自然语言处理功能集成到自己的应用程序中。对于 Java 开发者来说，这意味着可以在企业级应用、后台服务或移动应用中添加智能对话、内容生成等功能。常见的适用场景包括：

客服聊天机器人
内容自动生成
代码辅助工具
语言翻译服务

Java 开发者在调用 ChatGPT API 时经常会遇到几个典型问题：

认证问题 ：如何正确设置 API 密钥
请求构造 ：如何构建符合要求的 JSON 请求体
响应解析 ：如何处理返回的复杂 JSON 结构
性能瓶颈 ：如何优化 API 调用的响应时间
错误处理 ：如何应对 API 限流和错误响应

我们将使用 OkHttp 作为 HTTP 客户端来实现 ChatGPT API 的调用，这是因为它简单易用且性能优越。以下是实现的关键步骤：

首先需要在项目中添加 OkHttp 依赖（Maven 示例）：

<dependency>
    <groupId>com.squareup.okhttp3</groupId>
    <artifactId>okhttp</artifactId>
    <version>4.9.3</version>
</dependency>

ChatGPT API 使用 Bearer Token 进行认证，需要在请求头中添加 Authorization 字段：

String apiKey = "your-api-key";
Request request = new Request.Builder()
    .url("https://api.openai.com/v1/chat/completions")
    .addHeader("Authorization", "Bearer" + apiKey)
    .addHeader("Content-Type", "application/json")
    .build();

ChatGPT API 需要特定格式的 JSON 请求体。我们可以使用 JSON 库（如 Jackson 或 Gson）来构建：

ObjectMapper mapper = new ObjectMapper();
ObjectNode requestBody = mapper.createObjectNode();
requestBody.put("model", "gpt-3.5-turbo");
ArrayNode messages = requestBody.putArray("messages");
ObjectNode message = messages.addObject();
message.put("role", "user");
message.put("content", "Hello, how are you?");
String jsonBody = mapper.writeValueAsString(requestBody);

RequestBody body = RequestBody.create(jsonBody, MediaType.parse("application/json"));
Request request = new Request.Builder()
    .url("https://api.openai.com/v1/chat/completions")
    .post(body)
    .addHeader("Authorization", "Bearer" + apiKey)
    .build();

try (Response response = client.newCall(request).execute()) {if (!response.isSuccessful()) {throw new IOException("Unexpected code" + response);
    }
    String responseBody = response.body().string();
    JsonNode responseJson = mapper.readTree(responseBody);
    // 处理响应数据
}

下面是一个完整的 Java 类示例，实现了 ChatGPT API 的调用：

import com.fasterxml.jackson.databind.JsonNode;
import com.fasterxml.jackson.databind.ObjectMapper;
import com.fasterxml.jackson.databind.node.ArrayNode;
import com.fasterxml.jackson.databind.node.ObjectNode;
import okhttp3.*;

import java.io.IOException;

public class ChatGPTClient {
    private static final String API_URL = "https://api.openai.com/v1/chat/completions";
    private final OkHttpClient client;
    private final ObjectMapper mapper;
    private final String apiKey;

    public ChatGPTClient(String apiKey) {
        this.apiKey = apiKey;
        this.client = new OkHttpClient.Builder()
                .connectTimeout(30, TimeUnit.SECONDS)
                .writeTimeout(30, TimeUnit.SECONDS)
                .readTimeout(30, TimeUnit.SECONDS)
                .build();
        this.mapper = new ObjectMapper();}

    public String getCompletion(String prompt) throws IOException {ObjectNode requestBody = mapper.createObjectNode();
        requestBody.put("model", "gpt-3.5-turbo");
        ArrayNode messages = requestBody.putArray("messages");
        ObjectNode message = messages.addObject();
        message.put("role", "user");
        message.put("content", prompt);

        RequestBody body = RequestBody.create(mapper.writeValueAsString(requestBody),
                MediaType.parse("application/json")
        );

        Request request = new Request.Builder()
                .url(API_URL)
                .post(body)
                .addHeader("Authorization", "Bearer" + apiKey)
                .addHeader("Content-Type", "application/json")
                .build();

        try (Response response = client.newCall(request).execute()) {if (!response.isSuccessful()) {throw new IOException("API request failed:" + response.code() + " " + response.message());
            }
            JsonNode responseJson = mapper.readTree(response.body().string());
            return responseJson.path("choices").get(0).path("message").path("content").asText();}
    }

    public void close() {client.dispatcher().executorService().shutdown();
        client.connectionPool().evictAll();
    }
}