Spring AI是一个开源的多模态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();
        }
    }
}