error() {
+ return new ResponseEntity<>("\n" +
+ "\n" +
+ "\n" +
+ " \n" +
+ " \n" +
+ " Document\n" +
+ "\n" +
+ "\n" +
+ " Thanks you use gpt4-copilot-java
\n" +
+ " 详细使用文档
\n" +
+ " 项目地址
\n" +
+ "\n" +
+ "\n", HttpStatus.OK);
+ }
+}
\ No newline at end of file
diff --git a/src/main/java/com/gpt4/copilot/controller/chatController.java b/src/main/java/com/gpt4/copilot/controller/chatController.java
new file mode 100644
index 0000000..a5822a7
--- /dev/null
+++ b/src/main/java/com/gpt4/copilot/controller/chatController.java
@@ -0,0 +1,1003 @@
+package com.gpt4.copilot.controller;
+
+import com.alibaba.fastjson2.JSON;
+import com.fasterxml.jackson.core.JsonProcessingException;
+import com.fasterxml.jackson.databind.ObjectMapper;
+import com.gpt4.copilot.pojo.Result;
+import jakarta.servlet.http.HttpServletRequest;
+import jakarta.servlet.http.HttpServletResponse;
+import lombok.extern.slf4j.Slf4j;
+import okhttp3.*;
+import org.apache.commons.lang.StringUtils;
+import org.json.JSONException;
+import org.json.JSONObject;
+import org.springframework.beans.factory.annotation.Value;
+import org.springframework.http.HttpStatus;
+import org.springframework.http.ResponseEntity;
+import org.springframework.web.bind.annotation.GetMapping;
+import org.springframework.web.bind.annotation.PostMapping;
+import org.springframework.web.bind.annotation.RequestParam;
+import org.springframework.web.bind.annotation.RestController;
+
+import java.io.*;
+import java.nio.charset.StandardCharsets;
+import java.security.MessageDigest;
+import java.security.NoSuchAlgorithmException;
+import java.util.HashMap;
+import java.util.Map;
+import java.util.UUID;
+import java.util.concurrent.*;
+import java.util.concurrent.atomic.AtomicInteger;
+
+/**
+ * @author Yangyang
+ * @create 2023-12-25 18:29
+ */
+
+@Slf4j
+@RestController()
+public class chatController {
+ /**
+ * 缓存cocopilotToken
+ */
+ private static final HashMap copilotTokenList;
+ /**
+ * 缓存copilotToken
+ */
+ private static final HashMap coCopilotTokenList;
+ /**
+ * 缓存selfToken
+ */
+ private static final HashMap selfTokenList;
+ /**
+ * 模型
+ */
+ private static final String models = "{\"data\":[{\"id\":\"text-search-babbage-doc-001\",\"object\":\"model\",\"created\":1651172509,\"owned_by\":\"openai-dev\"},{\"id\":\"gpt-4\",\"object\":\"model\",\"created\":1687882411,\"owned_by\":\"openai\"},{\"id\":\"babbage\",\"object\":\"model\",\"created\":1649358449,\"owned_by\":\"openai\"},{\"id\":\"gpt-3.5-turbo-0613\",\"object\":\"model\",\"created\":1686587434,\"owned_by\":\"openai\"},{\"id\":\"text-babbage-001\",\"object\":\"model\",\"created\":1649364043,\"owned_by\":\"openai\"},{\"id\":\"gpt-3.5-turbo\",\"object\":\"model\",\"created\":1677610602,\"owned_by\":\"openai\"},{\"id\":\"gpt-3.5-turbo-1106\",\"object\":\"model\",\"created\":1698959748,\"owned_by\":\"system\"},{\"id\":\"curie-instruct-beta\",\"object\":\"model\",\"created\":1649364042,\"owned_by\":\"openai\"},{\"id\":\"gpt-3.5-turbo-0301\",\"object\":\"model\",\"created\":1677649963,\"owned_by\":\"openai\"},{\"id\":\"gpt-3.5-turbo-16k-0613\",\"object\":\"model\",\"created\":1685474247,\"owned_by\":\"openai\"},{\"id\":\"text-embedding-ada-002\",\"object\":\"model\",\"created\":1671217299,\"owned_by\":\"openai-internal\"},{\"id\":\"davinci-similarity\",\"object\":\"model\",\"created\":1651172509,\"owned_by\":\"openai-dev\"},{\"id\":\"curie-similarity\",\"object\":\"model\",\"created\":1651172510,\"owned_by\":\"openai-dev\"},{\"id\":\"babbage-search-document\",\"object\":\"model\",\"created\":1651172510,\"owned_by\":\"openai-dev\"},{\"id\":\"curie-search-document\",\"object\":\"model\",\"created\":1651172508,\"owned_by\":\"openai-dev\"},{\"id\":\"babbage-code-search-code\",\"object\":\"model\",\"created\":1651172509,\"owned_by\":\"openai-dev\"},{\"id\":\"ada-code-search-text\",\"object\":\"model\",\"created\":1651172510,\"owned_by\":\"openai-dev\"},{\"id\":\"text-search-curie-query-001\",\"object\":\"model\",\"created\":1651172509,\"owned_by\":\"openai-dev\"},{\"id\":\"text-davinci-002\",\"object\":\"model\",\"created\":1649880484,\"owned_by\":\"openai\"},{\"id\":\"ada\",\"object\":\"model\",\"created\":1649357491,\"owned_by\":\"openai\"},{\"id\":\"text-ada-001\",\"object\":\"model\",\"created\":1649364042,\"owned_by\":\"openai\"},{\"id\":\"ada-similarity\",\"object\":\"model\",\"created\":1651172507,\"owned_by\":\"openai-dev\"},{\"id\":\"code-search-ada-code-001\",\"object\":\"model\",\"created\":1651172507,\"owned_by\":\"openai-dev\"},{\"id\":\"text-similarity-ada-001\",\"object\":\"model\",\"created\":1651172505,\"owned_by\":\"openai-dev\"},{\"id\":\"text-davinci-edit-001\",\"object\":\"model\",\"created\":1649809179,\"owned_by\":\"openai\"},{\"id\":\"code-davinci-edit-001\",\"object\":\"model\",\"created\":1649880484,\"owned_by\":\"openai\"},{\"id\":\"text-search-curie-doc-001\",\"object\":\"model\",\"created\":1651172509,\"owned_by\":\"openai-dev\"},{\"id\":\"text-curie-001\",\"object\":\"model\",\"created\":1649364043,\"owned_by\":\"openai\"},{\"id\":\"curie\",\"object\":\"model\",\"created\":1649359874,\"owned_by\":\"openai\"},{\"id\":\"davinci\",\"object\":\"model\",\"created\":1649359874,\"owned_by\":\"openai\"}]}";
+ private static final String machineId;
+ private final static String get_cocopilotToken_url = "https://api.cocopilot.org/copilot_internal/v2/token";
+
+ static {
+ selfTokenList = new HashMap<>();
+ copilotTokenList = new HashMap<>();
+ coCopilotTokenList = new HashMap<>();
+ machineId = generateMachineId();
+ log.info("初始化接口成功！");
+ }
+
+ private final OkHttpClient client = new OkHttpClient.Builder().connectTimeout(3, TimeUnit.MINUTES).readTimeout(5, TimeUnit.MINUTES).writeTimeout(5, TimeUnit.MINUTES).build();
+
+ private final ThreadFactory threadFactory = new ThreadFactory() {
+ private final AtomicInteger counter = new AtomicInteger(0);
+
+ @Override
+ public Thread newThread(Runnable r) {
+ return new Thread(r, "chatThreadPool-" + counter.getAndIncrement());
+ }
+ };
+
+ private final ExecutorService executor = new ThreadPoolExecutor(0, 300, 60L, TimeUnit.SECONDS, new SynchronousQueue<>(), threadFactory);
+
+ private final String github_get_token_url = "https://api.github.com/copilot_internal/v2/token";
+ private final String github_chat = "https://api.githubcopilot.com/chat/completions";
+ private final String github_embaddings = "https://api.githubcopilot.com/embeddings";
+ private final String vscode_version = "vscode/1.85.2";
+ /**
+ * gpt4单字符睡眠时间
+ */
+ @Value("${gpt4_sleepTime}")
+ private Integer gpt4_sleepTime;
+ /**
+ * gpt3单字符睡眠时间
+ */
+ @Value("${gpt3_sleepTime}")
+ private Integer gpt3_sleepTime;
+ /**
+ * 修改睡眠时间密码
+ */
+ @Value("${password}")
+ private String password;
+ @Value("${get_token_url}")
+ private String get_token_url;
+
+ private static String generateMachineId() {
+ try {
+ UUID uuid = UUID.randomUUID();
+ MessageDigest sha256 = MessageDigest.getInstance("SHA-256");
+ byte[] hash = sha256.digest(uuid.toString().getBytes(StandardCharsets.UTF_8));
+ StringBuilder hexString = new StringBuilder();
+ for (byte b : hash) {
+ String hex = Integer.toHexString(0xff & b);
+ if (hex.length() == 1) {
+ hexString.append('0');
+ }
+ hexString.append(hex);
+ }
+ return hexString.toString();
+ } catch (NoSuchAlgorithmException e) {
+ throw new RuntimeException(e);
+ }
+ }
+
+ public void setGpt4_sleepTime(Integer gpt4_sleepTime) {
+ this.gpt4_sleepTime = gpt4_sleepTime;
+ }
+
+ public void setGpt3_sleepTime(Integer gpt3_sleepTime) {
+ this.gpt3_sleepTime = gpt3_sleepTime;
+ }
+
+ public String getPassword() {
+ return password;
+ }
+
+ /**
+ * 修改sleep时间
+ */
+ @GetMapping(value = "changeSleepTime")
+ private Result changeSleepTime(@RequestParam("gpt3_sleepTime") Integer gpt3_sleepTime,
+ @RequestParam("gpt4_sleepTime") Integer gpt4_sleepTime,
+ @RequestParam("password") String password) {
+ try {
+ if (password.equals(getPassword())) {
+ if (gpt3_sleepTime != null && gpt4_sleepTime >= 0 && gpt4_sleepTime <= 150) {
+ setGpt3_sleepTime(gpt3_sleepTime);
+ }
+ if (gpt4_sleepTime != null && gpt4_sleepTime >= 0 && gpt4_sleepTime <= 150) {
+ setGpt4_sleepTime(gpt4_sleepTime);
+ }
+ return Result.success("修改成功！");
+ } else {
+ return Result.error("管理员密码不对，请重新再试！");
+ }
+ } catch (Exception e) {
+ return Result.error("出错了，修改失败！");
+ }
+ }
+
+ /**
+ * 请求体不是json
+ * Authorization token缺失 会报Authorization header is missing
+ * 无法请求到chat_token 会报github copilot APIKey is wrong
+ * 反代/copilot_internal/v2/token 接口
+ *
+ * @param request
+ * @return
+ * @throws JSONException
+ * @throws IOException
+ */
+ @PostMapping(value = "/copilot_internal/v2/token")
+ public ResponseEntity