ChatGPTApiController.java
5.5 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
package com.zhonglai.luhui.controller;
import cn.hutool.http.HttpRequest;
import cn.hutool.http.HttpUtil;
import cn.hutool.json.JSONUtil;
import com.alibaba.fastjson.JSONObject;
import com.zhonglai.luhui.dto.*;
import io.swagger.annotations.Api;
import io.swagger.annotations.ApiOperation;
import org.apache.commons.lang3.StringUtils;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.web.bind.annotation.*;
import javax.servlet.http.HttpServletResponse;
import java.io.IOException;
import java.io.PrintWriter;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
@Api(tags = "chatGPT接口")
@RestController
@RequestMapping("/chatGPTApi")
public class ChatGPTApiController {
protected final Logger logger = LoggerFactory.getLogger(this.getClass());
@ApiOperation(value = "发送免费的接口")
@PostMapping("/sendFreeMessage")
public String sendFreeMessage(@RequestBody String jsonStr)
{
logger.info("请求内容：{}", jsonStr);
int timeout = 3000000;
Map<String,String> map = new HashMap<>();
map.put("Authorization","Bearer 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");
String str = HttpRequest.post("https://api.openai.com/v1/chat/completions").setReadTimeout(timeout).timeout(timeout).addHeaders(map).body(jsonStr).execute().body();
logger.info("返回内容：{}", str);
return str;
}
@ApiOperation(value = "发送不免费的接口")
@PostMapping("/sendNotFreeMessage")
public String sendNotFreeMessage(@RequestBody String jsonStr)
{
logger.info("请求内容：{}", jsonStr);
int timeout = 3000000;
Map<String,String> map = new HashMap<>();
map.put("Authorization","Bearer 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");
String str = HttpRequest.post("https://api.openai.com/v1/chat/completions").setReadTimeout(timeout).timeout(timeout).addHeaders(map).body(jsonStr).execute().body();
logger.info("返回内容：{}", str);
return str;
}
@ApiOperation(value = "测试")
@PostMapping("/sendMessage")
public String sendMessage(HttpServletResponse response, @RequestBody ChatGPTApiDto chatGPTApiDto)
{
int timeout = 3000000;
// 设置响应内容类型和编码
response.setContentType("text/html;charset=UTF-8");
JSONObject jsonObject = new JSONObject();
jsonObject.put("model","gpt-3.5-turbo-0301");
jsonObject.put("messages",chatGPTApiDto.getMessageList());
String domainName = "api.openai.com";
String apiDomain = "openai.yu2le.com/api";
// String domainName = "chatgpt.njlaikun.com";
// String apiDomain = "localhost:8082";
Map<String,String> map = new HashMap<>();
if(chatGPTApiDto.getIsfree())
{
map.put("Authorization","Bearer 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");
}else{
map.put("Authorization","Bearer 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");
}
map.put("Content-Type","application/json");
logger.info("开始请求openai接口：{}",JSONObject.toJSON(chatGPTApiDto));
String str = HttpRequest.post("https://"+domainName+"/v1/chat/completions").setReadTimeout(timeout).timeout(timeout).addHeaders(map).body(jsonObject.toString()).execute().body();
logger.info("返回的数据：{}",str);
if(JSONUtil.isTypeJSONObject(str))
{
CompletionResult3_5 completionResult = JSONObject.parseObject(str, CompletionResult3_5.class);
List<CompletionChoice3_5> list = completionResult.getChoices();
if(null != list && list.size() != 0)
{
StringBuffer stringBuffer = new StringBuffer();
for (CompletionChoice3_5 completionChoice3_5:list)
{
stringBuffer.append(completionChoice3_5.getMessage().getContent());
completionChoice3_5.getMessage().setContent(null);
}
completionResult.setContentLength( stringBuffer.length());
completionResult.setRoom_id(chatGPTApiDto.getRoom_id());
completionResult.setGptMessage_id(chatGPTApiDto.getGptMessage_id());
completionResult.setUser_id(chatGPTApiDto.getUser_id());
//通知更新用户信息
String noticestr = JSONObject.toJSONString(completionResult);
String rstr = HttpUtil.post("http://"+apiDomain+"/chatGPTStream/upUserFlowPacketRemain",noticestr);
logger.info("修改返回的数据：{}",rstr);
//返回结果字符串
return stringBuffer.toString();
// writeMessage(stringBuffer,response);
}
}
return str;
}
private void writeMessage(StringBuffer stringBuffer,HttpServletResponse response)
{
PrintWriter out = null;
try {
out = response.getWriter();
//验证验证码
if(StringUtils.isNotEmpty(stringBuffer))
{
for(int i=0;i<stringBuffer.length();i++)
{
char c = stringBuffer.charAt(i);
out.write(c);
out.flush(); // 立即将字符输出到客户端
}
}
} catch (IOException e) {
throw new RuntimeException(e);
}finally {
if(null != out)
{
out.close();
}
}
}
}