ChatGPT-用Java写一个调用OpenAI的代码

1,729次阅读

很抱歉，作为AI助手，我无法提供与OpenAI一样强大的AI能力。但我可以给出一个调用OpenAI API的Java示例代码，您需要先到OpenAI官网注册并获取API密钥，然后将其替换到代码中的apiKey变量上：

import java.net.HttpURLConnection;
import java.net.URL;
import java.io.DataOutputStream;
import java.io.BufferedReader;
import java.io.InputStreamReader;

public class OpenAIClient {
    private static final String apiKey = "your_api_key_here";
    private static final String apiEndpoint = "https://api.openai.com/v1/engines/davinci-codex/completions";

    public static String generateCode(String prompt) throws Exception {
        URL url = new URL(apiEndpoint);
        HttpURLConnection connection = (HttpURLConnection) url.openConnection();
        connection.setRequestMethod("POST");
        connection.setRequestProperty("Authorization", "Bearer " + apiKey);
        connection.setRequestProperty("Content-Type", "application/json");

        String data = "{\"prompt\":\"" + prompt + "\",\"max_tokens\":100,\"temperature\":0.5}";
        connection.setDoOutput(true);
        DataOutputStream outputStream = new DataOutputStream(connection.getOutputStream());
        outputStream.writeBytes(data);
        outputStream.flush();
        outputStream.close();

        BufferedReader reader = new BufferedReader(new InputStreamReader(connection.getInputStream()));
        StringBuilder responseBuilder = new StringBuilder();
        String line;
        while ((line = reader.readLine()) != null) {
            responseBuilder.append(line);
        }
        reader.close();
        return responseBuilder.toString();
    }

    public static void main(String[] args) throws Exception {
        String prompt = "Generate a function that counts the number of vowels in a string";
        String response = generateCode(prompt);
        System.out.println(response);
    }
}

这个示例代码使用了Java的HttpURLConnection类来向OpenAI API发送POST请求，并解析返回的JSON格式数据。generateCode()方法接受一个字符串参数prompt，这是一个用于生成代码的提示。这个方法返回一个JSON格式的字符串，其中包含了OpenAI生成的代码。在main()方法中，我们使用示例提示调用generateCode()方法，并输出生成的代码。请注意，此代码仅供参考，您需要根据自己的需要进行修改。