import org.apache.http.HttpEntity;
|
import org.apache.http.HttpHeaders;
|
import org.apache.http.HttpResponse;
|
import org.apache.http.client.HttpClient;
|
import org.apache.http.client.methods.HttpPost;
|
import org.apache.http.entity.StringEntity;
|
import org.apache.http.impl.client.HttpClients;
|
import org.apache.http.util.EntityUtils;
|
import org.jeecg.JeecgLimsCloudApplication;
|
import org.jeecg.modules.lims.service.IWxService;
|
import org.junit.runner.RunWith;
|
import org.springframework.beans.factory.annotation.Autowired;
|
import org.springframework.boot.test.context.SpringBootTest;
|
import org.springframework.test.context.junit4.SpringRunner;
|
|
import java.io.IOException;
|
|
@RunWith(SpringRunner.class)
|
@SpringBootTest(webEnvironment = SpringBootTest.WebEnvironment.RANDOM_PORT, classes = JeecgLimsCloudApplication.class)
|
public class Test {
|
@Autowired
|
private IWxService wxService;
|
@org.junit.Test
|
public void test(){
|
wxService.getWxAccessToken();
|
}
|
|
|
private static final String API_KEY = "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";
|
private static final String API_URL = "https://api.openai.com/v1/chat/completions";
|
|
public static String chatWithGPT(String prompt) throws IOException {
|
HttpClient httpClient = HttpClients.createDefault();
|
|
HttpPost httpPost = new HttpPost(API_URL);
|
httpPost.setHeader(HttpHeaders.AUTHORIZATION, "Bearer " + API_KEY);
|
httpPost.setHeader(HttpHeaders.CONTENT_TYPE, "application/json");
|
String mode = "gpt-3.5-turbo";
|
StringEntity requestBody = new StringEntity("{\"prompt\": \"" + prompt + "\", \"max_tokens\": 100, \"model\": \"" + mode + "\" }");
|
httpPost.setEntity(requestBody);
|
|
HttpResponse response = httpClient.execute(httpPost);
|
HttpEntity entity = response.getEntity();
|
String responseBody = EntityUtils.toString(entity);
|
|
return responseBody;
|
}
|
|
@org.junit.Test
|
public void t(){
|
try {
|
String userPrompt = "Hello, ChatGPT!";
|
String response = chatWithGPT(userPrompt);
|
System.out.println("ChatGPT response: " + response);
|
} catch (IOException e) {
|
e.printStackTrace();
|
}
|
}
|
}
|
