ChatAI-v2

2024-09-28 19:10:41 +08:00
parent c042c3cc61
commit 34cb78adf9
12 changed files with 775 additions and 0 deletions
--- a/src/test/java/MyTest.java
+++ b/src/test/java/MyTest.java
@@ -0,0 +1,135 @@
+import com.alibaba.fastjson.JSONObject;
+import lombok.Data;
+import org.apache.http.Header;
+import org.apache.http.HttpEntity;
+import org.apache.http.client.methods.CloseableHttpResponse;
+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.message.BasicHeader;
+import org.apache.http.util.EntityUtils;
+import org.junit.Test;
+
+import java.io.IOException;
+import java.util.regex.Matcher;
+import java.util.regex.Pattern;
+
+public class MyTest {
+    public static void www(String[] args) {
+        String miraiCode = "看下这个题目[mirai:image:https://multimedia.nt.qq.com.cn/download?appid=1407&fileid=CgoyOTk4ODEzODgyEhQW4G7W3v2lT-NKItOqORRIbuK6rxiZzgcg_woo4u2zytCqiANQgL2jAQ&spec=0&rkey=CAMSKMa3OFokB_Tlf6uVSrtr8zFEBtyDtGIdmuztQCnfO9gIDN19LUsLquI]";
+        String url = null;
+        if (miraiCode.matches(".*[mirai:image:(https?://[\\w./?&amp;=]+)].*")) {
+            String regex = "\\[mirai:image:(.*?)]";
+            Pattern pattern = Pattern.compile(regex);
+
+            // 创建Matcher对象
+            Matcher matcher = pattern.matcher(miraiCode);
+
+            // 查找并提取链接
+            if (matcher.find()) {
+                url = matcher.group(1); // 提取第一个括号内的内容
+            }
+        }
+        System.out.println(url);
+    }
+
+    @Test
+    public void sseTest() throws IOException {
+
+        @Data
+        class Content {
+            private String type;
+            private String text;
+
+            public Content() {
+            }
+
+            public Content(String type, String text) {
+                this.type = type;
+                this.text = text;
+            }
+        }
+
+        @Data
+        class Message {
+            private String role;
+            private Content[] content;
+        }
+
+        CloseableHttpClient client = HttpClients.createDefault();
+        HttpPost httpPost = new HttpPost("https://open.bigmodel.cn/api/paas/v4/assistant");
+        httpPost.setHeaders(new Header[]{
+                new BasicHeader("content-type", "application/json"),
+                new BasicHeader("authorization", "abce8567e434a0e4b987c34763c23c73.NCwRwbrkW9Mi6t63")
+        });
+
+        JSONObject jsonObject = new JSONObject();
+        jsonObject.put("assistant_id", "65a265419d72d299a9230616");
+        jsonObject.put("model", "glm-4-assistant");
+        jsonObject.put("stream", "True");
+        Message message = new Message();
+        message.role = "user";
+        Content[] contents = new Content[]{new Content("text", "改写提示词：现在你是一名Java开发工程师...")};
+        message.content = contents;
+        Message[] messages = new Message[]{message};
+        jsonObject.put("messages", messages);
+        HttpEntity httpEntity = new StringEntity(jsonObject.toString(), "UTF-8");
+
+        httpPost.setEntity(httpEntity);
+
+        CloseableHttpResponse response = client.execute(httpPost);
+        System.out.println(response.getStatusLine());
+
+        HttpEntity entity = response.getEntity();
+        String result = EntityUtils.toString(entity, "UTF-8");
+        String[] strs = result.split("\n\n");
+        for (String str : strs) {
+            pojo.Data data = JSONObject.parseObject(str.substring(6), pojo.Data.class);
+            if (data.getStatus().equals("completed")) {
+                break;
+            }
+            //解析返回结果
+//            System.out.println(data.getChoices().get(0).getDelta());
+            String delta = data.getChoices().get(0).getDelta();
+            JSONObject deltaJsonObject = JSONObject.parseObject(delta);
+
+            String role = deltaJsonObject.get("role").toString();
+            if ("assistant".equals(role)) {
+                //输出content
+                System.out.print(deltaJsonObject.get("content"));
+            } else if ("tool".equals(role)) {
+                //解析tool_call内容，根据字段type
+                JSONObject toolCall = deltaJsonObject.getJSONArray("tool_calls").getJSONObject(0);
+                String type = toolCall.get("type").toString();
+                if ("function".equals(type)) {
+                    //如果调用了函数
+                    JSONObject functionJsonObject = toolCall.getJSONObject("function");
+                    if (functionJsonObject.get("outputs") != null) {
+                        String contentUrl = functionJsonObject.getJSONArray("outputs").getJSONObject(0).getJSONObject("content").get("url").toString();
+                        System.out.println("\r\n" + contentUrl);
+                    }
+                } else if ("drawing_tool".equals(type)) {
+                    //如果是绘画工具
+                    JSONObject drawingToolJsonObject = toolCall.getJSONObject("drawing_tool");
+                    if (drawingToolJsonObject.get("outputs") != null) {
+                        String imageUrl = drawingToolJsonObject.getJSONArray("outputs").getJSONObject(0).get("image").toString();
+                        System.out.println(imageUrl);
+                    }
+                } else if ("web_browser".equals(type)) {
+                    JSONObject webBrowserJsonObject = toolCall.getJSONObject("web_browser");
+                    if (webBrowserJsonObject.get("outputs") != null) {
+                        for (Object o : webBrowserJsonObject.getJSONArray("outputs")) {
+                            System.out.println(JSONObject.parseObject(o.toString()).get("title"));
+                            System.out.println(JSONObject.parseObject(o.toString()).get("link"));
+                            System.out.println(JSONObject.parseObject(o.toString()).get("content"));
+                        }
+                    }
+                }
+            }
+
+        }
+        response.close();
+        client.close();
+    }
+}