Spaces:
Build error
Build error
Merge remote-tracking branch 'origin/main'
Browse files
src/main/java/com/bitservice/Aidemo/AIController.java
CHANGED
@@ -25,7 +25,8 @@ public class AIController {
|
|
25 |
String api_key = System.getenv("OPENAI_API_KEY");
|
26 |
String api_base = System.getenv("OPENAI_API_BASE");
|
27 |
String api_version = "2023-06-01-preview";
|
28 |
-
|
|
|
29 |
CloseableHttpClient httpclient = HttpClientBuilder.create().build();
|
30 |
HttpPost httpPost = new HttpPost(url);
|
31 |
ResponseHandler<String> handler = response -> {
|
@@ -53,14 +54,13 @@ public class AIController {
|
|
53 |
response = httpclient.execute(httpget);
|
54 |
HttpEntity he = response.getEntity();
|
55 |
String ret = EntityUtils.toString(response.getEntity());
|
56 |
-
// 方式1
|
57 |
-
// 生成 JSON 对象
|
58 |
JSONObject obj = JSONObject.parseObject(ret);
|
59 |
System.out.println(obj.toString());
|
60 |
status = obj.getString("status");
|
61 |
System.out.println(status);
|
62 |
if (status.equals("succeeded")) {
|
63 |
JSONArray js = obj.getJSONObject("result").getJSONArray("data");
|
|
|
64 |
return js.getJSONObject(0).getString("url");
|
65 |
}
|
66 |
}
|
|
|
25 |
String api_key = System.getenv("OPENAI_API_KEY");
|
26 |
String api_base = System.getenv("OPENAI_API_BASE");
|
27 |
String api_version = "2023-06-01-preview";
|
28 |
+
|
29 |
+
String url = api_base + "/openai/images/generations:submit?api-version=" + api_version;
|
30 |
CloseableHttpClient httpclient = HttpClientBuilder.create().build();
|
31 |
HttpPost httpPost = new HttpPost(url);
|
32 |
ResponseHandler<String> handler = response -> {
|
|
|
54 |
response = httpclient.execute(httpget);
|
55 |
HttpEntity he = response.getEntity();
|
56 |
String ret = EntityUtils.toString(response.getEntity());
|
|
|
|
|
57 |
JSONObject obj = JSONObject.parseObject(ret);
|
58 |
System.out.println(obj.toString());
|
59 |
status = obj.getString("status");
|
60 |
System.out.println(status);
|
61 |
if (status.equals("succeeded")) {
|
62 |
JSONArray js = obj.getJSONObject("result").getJSONArray("data");
|
63 |
+
System.out.println(js);
|
64 |
return js.getJSONObject(0).getString("url");
|
65 |
}
|
66 |
}
|