怎么使用ChatGPT API來與Java進行交互

怎么使用ChatGPT API來與Java進行交互

隨著人工智能技術的快速發展，Open的ChatGPT API成為了開發者們構建智能對話系統的強大工具。通過API，開發者可以將ChatGPT集成到各種應用程序中，包括Java應用程序。本文將詳細介紹如何使用ChatGPT API與Java進行交互，幫助開發者快速上手。

1. 準備工作

在開始之前，確保你已經完成以下準備工作：

1. 獲取API密鑰：首先，你需要在Open官網上注冊并獲取API密鑰。這個密鑰將用于在Java代碼中與ChatGPT API進行身份驗證。

2. 安裝Java開發環境：確保你的開發環境中已經安裝了Java Development Kit (JDK) 和一個集成開發環境（IDE），如IntelliJ IDEA或Eclipse。

3. 添加HTTP客戶端庫：為了與ChatGPT API進行HTTP通信，你需要一個HTTP客戶端庫。推薦使用OkHttp或Apache HttpClient。本文將使用OkHttp作為示例。

2. 添加依賴

如果你使用的是Maven項目，可以在pom.xml文件中添加以下依賴：

<dependency>
    <groupId>com.squareup.okhttp3</groupId>
    <artifactId>okhttp</artifactId>
    <version>4.9.3</version>
</dependency>
<dependency>
    <groupId>com.google.code.gson</groupId>
    <artifactId>gson</artifactId>
    <version>2.8.9</version>
</dependency>

OkHttp用于發送HTTP請求，Gson用于解析JSON響應。

3. 編寫Java代碼

接下來，我們將編寫一個簡單的Java程序來與ChatGPT API進行交互。

3.1 創建HTTP客戶端

首先，創建一個OkHttpClient實例：

import okhttp3.*;

public class ChatGPTClient {
    private static final String API_URL = "https://api.openai.com/v1/chat/completions";
    private static final String API_KEY = "your-api-key-here";

    private final OkHttpClient client = new OkHttpClient();

    // 其他代碼...
}

3.2 構建請求體

ChatGPT API需要一個JSON格式的請求體，包含對話的上下文和用戶輸入。我們可以使用Gson庫來構建這個請求體：

import com.google.gson.Gson;
import com.google.gson.JsonObject;

public class ChatGPTClient {
    // 其他代碼...

    private String buildRequestBody(String userMessage) {
        JsonObject requestBody = new JsonObject();
        requestBody.addProperty("model", "gpt-3.5-turbo");

        JsonObject message = new JsonObject();
        message.addProperty("role", "user");
        message.addProperty("content", userMessage);

        JsonObject[] messages = new JsonObject[]{message};
        requestBody.add("messages", new Gson().toJsonTree(messages));

        return new Gson().toJson(requestBody);
    }
}

3.3 發送請求并處理響應

接下來，我們編寫一個方法來發送HTTP POST請求并處理響應：

import okhttp3.Request;
import okhttp3.RequestBody;
import okhttp3.Response;

import java.io.IOException;

public class ChatGPTClient {
    // 其他代碼...

    public String sendMessage(String userMessage) throws IOException {
        String requestBody = buildRequestBody(userMessage);

        RequestBody body = RequestBody.create(requestBody, MediaType.parse("application/json"));
        Request request = new Request.Builder()
                .url(API_URL)
                .post(body)
                .addHeader("Authorization", "Bearer " + API_KEY)
                .addHeader("Content-Type", "application/json")
                .build();

        try (Response response = client.newCall(request).execute()) {
            if (!response.isSuccessful()) {
                throw new IOException("Unexpected code " + response);
            }

            return response.body().string();
        }
    }
}

3.4 解析響應

ChatGPT API的響應是一個JSON對象，包含生成的對話內容。我們可以使用Gson來解析這個響應：

import com.google.gson.JsonObject;
import com.google.gson.JsonParser;

public class ChatGPTClient {
    // 其他代碼...

    public String parseResponse(String jsonResponse) {
        JsonObject jsonObject = JsonParser.parseString(jsonResponse).getAsJsonObject();
        String content = jsonObject.getAsJsonArray("choices")
                .get(0).getAsJsonObject()
                .getAsJsonObject("message")
                .get("content").getAsString();

        return content;
    }
}

3.5 完整示例

將上述代碼整合在一起，我們得到一個完整的Java類：

import com.google.gson.Gson;
import com.google.gson.JsonObject;
import com.google.gson.JsonParser;
import okhttp3.*;

import java.io.IOException;

public class ChatGPTClient {
    private static final String API_URL = "https://api.openai.com/v1/chat/completions";
    private static final String API_KEY = "your-api-key-here";

    private final OkHttpClient client = new OkHttpClient();

    public static void main(String[] args) {
        ChatGPTClient chatGPTClient = new ChatGPTClient();
        try {
            String response = chatGPTClient.sendMessage("Hello, how are you?");
            String content = chatGPTClient.parseResponse(response);
            System.out.println("ChatGPT: " + content);
        } catch (IOException e) {
            e.printStackTrace();
        }
    }

    private String buildRequestBody(String userMessage) {
        JsonObject requestBody = new JsonObject();
        requestBody.addProperty("model", "gpt-3.5-turbo");

        JsonObject message = new JsonObject();
        message.addProperty("role", "user");
        message.addProperty("content", userMessage);

        JsonObject[] messages = new JsonObject[]{message};
        requestBody.add("messages", new Gson().toJsonTree(messages));

        return new Gson().toJson(requestBody);
    }

    public String sendMessage(String userMessage) throws IOException {
        String requestBody = buildRequestBody(userMessage);

        RequestBody body = RequestBody.create(requestBody, MediaType.parse("application/json"));
        Request request = new Request.Builder()
                .url(API_URL)
                .post(body)
                .addHeader("Authorization", "Bearer " + API_KEY)
                .addHeader("Content-Type", "application/json")
                .build();

        try (Response response = client.newCall(request).execute()) {
            if (!response.isSuccessful()) {
                throw new IOException("Unexpected code " + response);
            }

            return response.body().string();
        }
    }

    public String parseResponse(String jsonResponse) {
        JsonObject jsonObject = JsonParser.parseString(jsonResponse).getAsJsonObject();
        String content = jsonObject.getAsJsonArray("choices")
                .get(0).getAsJsonObject()
                .getAsJsonObject("message")
                .get("content").getAsString();

        return content;
    }
}

4. 運行程序

運行上述Java程序，你將看到ChatGPT的響應輸出到控制臺。你可以根據需要修改用戶輸入，并觀察ChatGPT的回復。

5. 總結

通過本文的介紹，你已經學會了如何使用Java與ChatGPT API進行交互。我們使用了OkHttp庫來發送HTTP請求，并使用Gson庫來解析JSON響應。你可以在此基礎上進一步擴展功能，例如處理多輪對話、添加錯誤處理等。

希望本文對你有所幫助，祝你在開發過程中取得成功！