zh21608 commited on
Commit
f2c89e1
·
2 Parent(s): 0f6e550 c9a696a

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
- String url = api_base + "openai/images/generations:submit?api-version=" + api_version;
 
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
  }