ChatGPT/unofficial-gpt4
1
0
Fork 0
mirror of https://github.com/YuWanTingbb/unofficial-gpt4.git synced 2025-10-13 21:51:23 +00:00
Code Issues Packages Projects Releases Wiki Activity

对chat接口的模型进行重定向，减少潜在的风险

Browse Source
This commit is contained in:
Yanyutin753
2024-04-13 01:18:12 +08:00
parent 66fe458e9f
commit abaf8437e1
7 changed files with 50 additions and 24 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
.gitignore vendored Normal file
View File

@@ -0,0 +1,2 @@
e -i 66fe458e9ff3b7072252b455b08ac5c3076591bc^

69
src/main/java/com/gpt4/copilot/controller/ChatController.java
View File

@@ -506,7 +506,7 @@ public class ChatController {
Map<String, String> headersMap = new HashMap<>();
//添加头部
addHeader(headersMap, chat_token, apiKey);
String model = (conversation.getModel() != null) ? conversation.getModel() : "gpt-3.5-turbo";
String model = modelAdjust(conversation);
Request streamRequest = getPrompt(conversation, model, headersMap);
try (Response resp = client.newCall(streamRequest).execute()) {
if (!resp.isSuccessful()) {
@@ -535,6 +535,20 @@ public class ChatController {
return getObjectResponseEntity(response, future);
}
private String modelAdjust(Conversation conversation) {
String model = conversation.getModel();
if(model == null){
conversation.setModel("gpt-3.5-turbo");
return "gpt-3.5-turbo";
}
else{
if(model.startsWith("gpt-4")) {
conversation.setModel("gpt-4");
}
return model;
}
}
/**
* 请求体不是json 会报Request body is missing or not in JSON format
* Authorization token缺失 会报Authorization header is missing
@@ -587,7 +601,7 @@ public class ChatController {
Map<String, String> headersMap = new HashMap<>();
//添加头部
addHeader(headersMap, chat_token, apiKey);
String model = (conversation.getModel() != null) ? conversation.getModel() : "gpt-3.5-turbo";
String model = modelAdjust(conversation);
Request streamRequest = getPrompt(conversation, model, headersMap);
try (Response resp = client.newCall(streamRequest).execute()) {
if (!resp.isSuccessful()) {
@@ -645,7 +659,7 @@ public class ChatController {
* @param conversation
* @throws IOException
*/
private String[] extractApiKeyAndRequestUrl(String authorizationHeader, Conversation conversation) throws IllegalArgumentException {
private String[] extractApiKeyAndRequestUrl(String authorizationHeader, Object conversation) throws IllegalArgumentException {
if (conversation == null) {
throw new IllegalArgumentException("Request body is missing or not in JSON format");
}
@@ -712,7 +726,7 @@ public class ChatController {
Map<String, String> headersMap = new HashMap<>();
//添加头部
addHeader(headersMap, chat_token, apiKey);
String model = (conversation.getModel() != null) ? conversation.getModel() : "gpt-3.5-turbo";
String model = modelAdjust(conversation);
Request streamRequest = getPrompt(conversation, model, headersMap);
try (Response resp = client.newCall(streamRequest).execute()) {
if (!resp.isSuccessful()) {
@@ -792,6 +806,22 @@ public class ChatController {
}
}
private Request getEmdPrompt(Object conversation,
Map<String, String> headersMap) {
try {
String json = com.alibaba.fastjson2.JSON.toJSONString(conversation);
RequestBody requestBody = RequestBody.create(json, JSON);
Request.Builder requestBuilder = new Request.Builder().url(github_embaddings).post(requestBody);
headersMap.forEach(requestBuilder::addHeader);
Request streamRequest = requestBuilder.build();
return streamRequest;
} catch (Exception e) {
throw new RuntimeException(e);
}
}
/**
* ghu/gho 请求
* 请求体不是json 会报Request body is missing or not in JSON format
@@ -808,7 +838,7 @@ public class ChatController {
@PostMapping(value = "/v1/embeddings")
public ResponseEntity<Object> coPilotEmbeddings(HttpServletResponse response,
HttpServletRequest request,
@org.springframework.web.bind.annotation.RequestBody Conversation conversation) {
@org.springframework.web.bind.annotation.RequestBody Object conversation) {
String header = request.getHeader("Authorization");
String authorizationHeader = (header != null && !header.trim().isEmpty()) ? header.trim() : null;
// 异步处理
@@ -844,11 +874,7 @@ public class ChatController {
Map<String, String> headersMap = new HashMap<>();
//添加头部
addHeader(headersMap, chat_token, apiKey);
String json = com.alibaba.fastjson2.JSON.toJSONString(conversation);
RequestBody requestBody = RequestBody.create(json, JSON);
Request.Builder requestBuilder = new Request.Builder().url(github_embaddings).post(requestBody);
headersMap.forEach(requestBuilder::addHeader);
Request streamRequest = requestBuilder.build();
Request streamRequest = getEmdPrompt(conversation, headersMap);
try (Response resp = client.newCall(streamRequest).execute()) {
if (!resp.isSuccessful()) {
if (resp.code() == 429) {
@@ -908,7 +934,7 @@ public class ChatController {
@PostMapping(value = "/cocopilot/v1/embeddings")
public ResponseEntity<Object> coCoPilotEmbeddings(HttpServletResponse response,
HttpServletRequest request,
@org.springframework.web.bind.annotation.RequestBody Conversation conversation) {
@org.springframework.web.bind.annotation.RequestBody Object conversation) {
String header = request.getHeader("Authorization");
String authorizationHeader = (header != null && !header.trim().isEmpty()) ? header.trim() : null;
// 异步处理
@@ -944,11 +970,7 @@ public class ChatController {
Map<String, String> headersMap = new HashMap<>();
//添加头部
addHeader(headersMap, chat_token, apiKey);
String json = com.alibaba.fastjson2.JSON.toJSONString(conversation);
RequestBody requestBody = RequestBody.create(json, JSON);
Request.Builder requestBuilder = new Request.Builder().url(github_embaddings).post(requestBody);
headersMap.forEach(requestBuilder::addHeader);
Request streamRequest = requestBuilder.build();
Request streamRequest = getEmdPrompt(conversation, headersMap);
try (Response resp = client.newCall(streamRequest).execute()) {
if (!resp.isSuccessful()) {
if (resp.code() == 429) {
@@ -993,7 +1015,7 @@ public class ChatController {
@PostMapping(value = "/self/v1/embeddings")
public ResponseEntity<Object> selfEmbeddings(HttpServletResponse response,
HttpServletRequest request,
@org.springframework.web.bind.annotation.RequestBody Conversation conversation) {
@org.springframework.web.bind.annotation.RequestBody Object conversation) {
String header = request.getHeader("Authorization");
String authorizationHeader = (header != null && !header.trim().isEmpty()) ? header.trim() : null;
// 异步处理
@@ -1022,11 +1044,7 @@ public class ChatController {
Map<String, String> headersMap = new HashMap<>();
//添加头部
addHeader(headersMap, chat_token, apiKey);
String json = com.alibaba.fastjson2.JSON.toJSONString(conversation);
RequestBody requestBody = RequestBody.create(json, JSON);
Request.Builder requestBuilder = new Request.Builder().url(github_embaddings).post(requestBody);
headersMap.forEach(requestBuilder::addHeader);
Request streamRequest = requestBuilder.build();
Request streamRequest = getEmdPrompt(conversation, headersMap);
try (Response resp = client.newCall(streamRequest).execute()) {
if (!resp.isSuccessful()) {
if (resp.code() == 429) {
@@ -1054,7 +1072,7 @@ public class ChatController {
}
public Object againEmbeddings(HttpServletResponse response,
@org.springframework.web.bind.annotation.RequestBody Conversation conversation,
@org.springframework.web.bind.annotation.RequestBody Object conversation,
String token,
String apiKey) {
try {
@@ -1283,6 +1301,11 @@ public class ChatController {
*/
private String saveMadchineId(String apiKey) {
try {
if(machineIdList.containsKey(apiKey)){
String machineId = machineIdList.get(apiKey);
log.info("机械码读取成功！对应的机械码为：" + machineId);
return machineId;
}
String machineId = generateMachineId();
log.info("机械码生成成功！对应的机械码为：" + machineId);
machineIdList.put(apiKey, machineId);

3
src/main/java/com/gpt4/copilot/copilotApplication.java
View File

@@ -51,7 +51,7 @@ public class copilotApplication {
if (!jsonFile.exists() || jsonFile.length() == 0){
try {
if (!jsonFile.exists()) {
// 创建文件machineIdList.json
// 创建文件config.json
Files.createFile(jsonFilePath);
}
// 往 config.json 文件中添加一个空数组，防止重启报错
@@ -298,6 +298,7 @@ public class copilotApplication {
System.out.println("gpt4-copilot-java 初始化接口成功！");
System.out.println("======================================================");
System.out.println("******原神gpt4-copilot-java-native v0.1.3启动成功******");
System.out.println("* 对chat接口的模型进行重定向，减少潜在的风险");
System.out.println("* 使用ConcurrentHashMap，粗略的对于每个密钥按每分钟进行限速");
System.out.println("* 新增环境变量用于对gpt-4*等模型进行系统prompt提示");
System.out.println("* 新增url|apikey形式传入/self/*接口，用于自定义地址和密钥");

BIN
target/classes/com/gpt4/copilot/controller/ChatController.class
View File

Binary file not shown.

BIN
target/classes/com/gpt4/copilot/copilotApplication.class
View File

Binary file not shown.

BIN
target/gpt-4-copilot-0.1.3.jar
View File

Binary file not shown.

BIN
target/gpt-4-copilot-0.1.3.jar.original
View File

Binary file not shown.