From 2166c1a14f9629aa49a5f3bb849ce878df4c4892 Mon Sep 17 00:00:00 2001
From: yangbin <yangbin@qq.com>
Date: 星期三, 28 八月 2024 17:23:25 +0800
Subject: [PATCH] 2
---
lxzn-module-system/lxzn-system-start/src/test/java/org/jeecg/modules/ai/AiTest.java | 80 ++++++++++++++++++++++++++++++++++++++++
1 files changed, 80 insertions(+), 0 deletions(-)
diff --git a/lxzn-module-system/lxzn-system-start/src/test/java/org/jeecg/modules/ai/AiTest.java b/lxzn-module-system/lxzn-system-start/src/test/java/org/jeecg/modules/ai/AiTest.java
new file mode 100644
index 0000000..1aec897
--- /dev/null
+++ b/lxzn-module-system/lxzn-system-start/src/test/java/org/jeecg/modules/ai/AiTest.java
@@ -0,0 +1,80 @@
+package org.jeecg.modules.ai;
+
+import org.jeecg.JeecgSystemApplication;
+import org.jeecg.modules.ai.service.IFilePdfInfoService;
+import org.jeecg.modules.ai.service.IWebSocketAiService;
+import org.jeecg.modules.ai.vo.FilePdfSelectVo;
+import org.junit.Test;
+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.File;
+
+/**
+ * @author clown
+ * * @date 2024/7/16
+ */
+
+@RunWith(SpringRunner.class)
+@SpringBootTest(webEnvironment = SpringBootTest.WebEnvironment.RANDOM_PORT,classes = JeecgSystemApplication.class)
+public class AiTest {
+
+ @Autowired
+ private IFilePdfInfoService filePdfInfoService;
+ @Autowired
+ private IWebSocketAiService webSocketAiService;
+
+ @Test
+ public void selectChatIA() {
+ webSocketAiService.selectChatIA();
+ }
+
+ @Test
+ public void filePdf() {
+ // 鎸囧畾鏂囦欢澶硅矾寰�
+ String directoryPath = "F:\\file";
+ // 鍒涘缓File瀵硅薄琛ㄧず鏂囦欢澶�
+ File directory = new File(directoryPath);
+ // 鑾峰彇鏂囦欢澶逛腑鎵�鏈夋枃浠跺拰鏂囦欢澶�
+ File[] files = directory.listFiles();
+ // 妫�鏌ユ枃浠舵暟缁勬槸鍚︿负绌轰互鍙婃槸鍚﹀瓨鍦ㄦ枃浠�
+ if (files != null && files.length > 0) {
+ // 閬嶅巻鏂囦欢鏁扮粍锛屾墦鍗版瘡涓枃浠剁殑鍚嶅瓧
+ for (File file : files) {
+ if (file.isFile()) { // 纭繚鏄枃浠惰�岄潪鏂囦欢澶�
+ System.out.println(file.getName());
+ filePdfInfoService.uploadFilePdf
+ (new File(directoryPath + "\\" + file.getName()));
+ }
+ }
+ } else {
+ System.out.println("鏂囦欢澶逛负绌烘垨涓嶅瓨鍦ㄣ��");
+ }
+
+ }
+
+ @Test
+ public void filePdfImg() {
+ FilePdfSelectVo vo = new FilePdfSelectVo();
+ vo.setPdfContent("鍒�闀�");
+ vo.setPdfFileId("1821103772084142082");
+ filePdfInfoService.findImgPathList(vo);
+ }
+
+ @Test
+ public void sendJsonToHttpsPost() {
+ /* String name = "{\"id\":\"683a65fd-8feb-4446-ad32-714c4785f667\",\"messages\":[{\"role\":\"user\",\"content\":\"缁欐垜璁蹭釜鏁呬簨锛焅"}],\"stream\":true,\"max_tokens\":500}"; // JSON鏁版嵁
+ String url = "https://836u458t54.vicp.fun/chat/test_chat";
+ String aiPost = AiTestUils.sendJsonToHttpsPost(url, name);
+ System.out.println(aiPost);*/
+
+ String s = "\"data\": {\"id\": \"683a65fd-8feb-4446-ad32-714c4785f667\", \"created\": 1723105518, \"model\": \"YxCareer0.7\", \"choices\": [{\"index\": 0, \"delta\": {\"role\": \"assistant\", \"content\": \"\"}, \"finish_reason\": null}]}";
+ System.out.println(s);
+
+
+
+ }
+
+}
--
Gitblit v1.9.3