OpenAi 호출 시 연결 시간 초과에 대한 해결책

https://api.openai.com/v1/completions에 대한 HTTP 요청을 시작하기 전에 프록시를 위해 java.net 패키지 아래의 프록시를 사용하십시오.

아래와 같이 코드 쇼:

public class OpenAiApi {


    public String OpenAiAnswerer(CompletionRequest request, String openAiApiKey) {

        //代理
        String proxyHost = "代理地址";
        int proxyPort = 代理端口;
        String url = "https://api.openai.com/v1/completions";
        // json为请求体
        String json = JSONUtil.toJsonStr(request);
        String result = "";
        Proxy proxy = new Proxy(Proxy.Type.HTTP, new InetSocketAddress(proxyHost, proxyPort));
        try {
            HttpURLConnection connection = (HttpURLConnection) new URL(url).openConnection(proxy);
            connection.setRequestProperty("Authorization", "Bearer " + openAiApiKey);
            connection.setRequestProperty("Content-Type", "application/json");

            connection.setRequestMethod("POST");

            connection.setDoOutput(true);
            byte[] requestBodyBytes = json.getBytes(StandardCharsets.UTF_8);
            try (OutputStream outputStream = connection.getOutputStream()) {
                outputStream.write(requestBodyBytes, 0, requestBodyBytes.length);
            }
            try (InputStream inputStream = connection.getInputStream()) {
                BufferedReader reader = new BufferedReader(new InputStreamReader(inputStream));
                StringBuilder response = new StringBuilder();
                String line;
                while ((line = reader.readLine()) != null) {
                    response.append(line);
                }
                result = response.toString();

            }
        } catch (IOException e) {
            throw new RuntimeException(e);
        }

        return JSONUtil.toBean(result, CompletionResponse.class);
    }

}

connection.setRequestProperty("권한 부여", "전달자 " + openAiApiKey);
connection.setRequestProperty("콘텐츠 유형", "응용 프로그램/json");

사용하면서 Content-Type을 설정하는 것을 잊어버려서 계속 에러가 뜨는데 문서를 반복해서 읽어보니 설정을 해야 한다는 것을 알게 되었습니다.

다음은 OpenAiAnswerer 메소드의 공식 매개변수 클래스 CompletionRequest입니다. 여기서 Lombok 플러그인의 @Data 주석은 get, set 메소드 및 인수 없는 구성을 자동으로 채우는 데 사용됩니다. (매개변수화된 구성을 추가할 경우 반드시 비매개변수 구성을 추가해야 함)

@Data 
public class CompletionRequest {
    private String model;

    private String prompt;

    private Integer max_tokens;

    private Integer temperature;

    private Integer top_p;

    private Integer n;

    private Boolean stream;

    private Integer logprobs;

    private String stop;

}

추천

출처blog.csdn.net/dantui_/article/details/130072028