Spring Boot与AI机器学习技术探索方向和实施方

示例代码
package com.example.demo.service;
 
import com.google.gson.JsonObject;
import com.google.gson.JsonParser;
import org.apache.http.client.methods.HttpPost;
import org.apache.http.entity.StringEntity;
import org.apache.http.impl.client.CloseableHttpClient;
import org.apache.http.impl.client.HttpClients;
import org.apache.http.util.EntityUtils;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.stereotype.Service;
 
@Service
public class OpenAIService {
 
   @Value("${openai.api.key}")
   private String apiKey;
 
   public String generateText(String prompt) {
       String apiUrl = "https://api.openai.com/v1/engines/text-davinci-003/completions";
       try (CloseableHttpClient httpClient = HttpClients.createDefault()) {
           HttpPost httpPost = new HttpPost(apiUrl);
           httpPost.setHeader("Content-Type", "application/json");
           httpPost.setHeader("Authorization", "Bearer " + apiKey);
 
           JsonObject json = new JsonObject();
           json.addProperty("prompt", prompt);
           json.addProperty("max_tokens", 150);
 
           StringEntity entity = new StringEntity(json.toString());
           httpPost.setEntity(entity);
 
           String response = EntityUtils.toString(httpClient.execute(httpPost).getEntity());
           JsonObject responseObject = JsonParser.parseString(response).getAsJsonObject();
           return responseObject.getAsJsonArray("choices").get(0).getAsJsonObject().get("text").getAsString().trim();
       } catch (Exception e) {
           e.printStackTrace();
           return "Error: " + e.getMessage();
       }
    }
}