当前位置：首页 > article >正文

IDEA接入OpenAI API 方法教程

article 2024/11/7 11:21:31

在 IntelliJ IDEA 中接入 ChatGPT API，你需要以下几个步骤：

步骤 1: 获取 OpenAI API 密钥

访问 OpenAI 官方网站，注册并登录到你的账户。
进入你的账户设置页面，获取 API 密钥。
将 API 密钥保存在一个安全的地方，因为它将用于身份验证。

步骤 2: 创建一个新项目或使用现有项目

打开 IntelliJ IDEA，创建一个新的 Java 项目或打开现有项目。

步骤 3: 添加依赖项（使用 Maven 或 Gradle）

为了方便发送 HTTP 请求和处理 JSON 数据，推荐使用一些库，比如 HttpClient 和 Jackson（用于 JSON 处理）。你可以在项目中使用 Maven 或 Gradle 来管理依赖。

使用 Maven

在 pom.xml 中添加以下依赖项：

<dependencies>
    <!-- 添加 HttpClient 库 -->
    <dependency>
        <groupId>org.apache.httpcomponents</groupId>
        <artifactId>httpclient</artifactId>
        <version>4.5.13</version>
    </dependency>

    <!-- 添加 Jackson 库用于处理 JSON -->
    <dependency>
        <groupId>com.fasterxml.jackson.core</groupId>
        <artifactId>jackson-databind</artifactId>
        <version>2.13.0</version>
    </dependency>
</dependencies>

使用 Gradle

在 build.gradle 中添加以下依赖项：

dependencies {
    // 添加 HttpClient 库
    implementation 'org.apache.httpcomponents:httpclient:4.5.13'

    // 添加 Jackson 库用于处理 JSON
    implementation 'com.fasterxml.jackson.core:jackson-databind:2.13.0'
}

步骤 4: 编写代码来调用 ChatGPT API

在项目中创建一个类来调用 OpenAI API。以下是一个使用 Java 发送 HTTP 请求并处理响应的示例。

import org.apache.http.HttpResponse;
import org.apache.http.client.methods.HttpPost;
import org.apache.http.impl.client.HttpClients;
import org.apache.http.impl.client.CloseableHttpClient;
import org.apache.http.entity.StringEntity;
import com.fasterxml.jackson.databind.JsonNode;
import com.fasterxml.jackson.databind.ObjectMapper;

import java.io.BufferedReader;
import java.io.InputStreamReader;

public class ChatGPTAPIClient {

    // OpenAI API 密钥
    private static final String API_KEY = "your_openai_api_key";  // 替换成你的 API 密钥
    private static final String API_URL = "https://api.openai.com/v1/chat/completions";

    public static void main(String[] args) {
        try {
            // 创建 HTTP 客户端
            CloseableHttpClient httpClient = HttpClients.createDefault();

            // 创建 HTTP 请求对象
            HttpPost httpPost = new HttpPost(API_URL);

            // 设置请求头
            httpPost.setHeader("Content-Type", "application/json");
            httpPost.setHeader("Authorization", "Bearer " + API_KEY);

            // 构造请求体（这里我们使用 ChatGPT 的 API 请求格式）
            String jsonBody = "{"
                    + "\"model\": \"gpt-3.5-turbo\","
                    + "\"messages\": ["
                    + "{\"role\": \"user\", \"content\": \"Hello, ChatGPT!\"}"
                    + "]"
                    + "}";

            // 设置请求体
            StringEntity entity = new StringEntity(jsonBody);
            httpPost.setEntity(entity);

            // 执行请求并获取响应
            HttpResponse response = httpClient.execute(httpPost);

            // 读取响应内容
            BufferedReader reader = new BufferedReader(new InputStreamReader(response.getEntity().getContent()));
            StringBuilder responseString = new StringBuilder();
            String line;
            while ((line = reader.readLine()) != null) {
                responseString.append(line);
            }

            // 解析 JSON 响应
            ObjectMapper objectMapper = new ObjectMapper();
            JsonNode responseJson = objectMapper.readTree(responseString.toString());

            // 输出 ChatGPT 的回应
            String chatGPTResponse = responseJson.get("choices").get(0).get("message").get("content").asText();
            System.out.println("ChatGPT Response: " + chatGPTResponse);

            // 关闭客户端
            httpClient.close();

        } catch (Exception e) {
            e.printStackTrace();
        }
    }
}