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;
}