Quellcode durchsuchen

update parseDoc api

deason vor 1 Monat
Ursprung
Commit
fc87038f8b

+ 2 - 0
ai-api-demo/src/main/resources/application.properties

@@ -2,8 +2,10 @@ server.port=8088
 # ai api config
 #com.qmth.llm.server=https://solar.qmth.com.cn
 #com.qmth.ocr.server=https://solar.qmth.com.cn
+#com.qmth.doc.server=https://solar.qmth.com.cn
 com.qmth.llm.server=http://192.168.10.83:8090
 com.qmth.ocr.server=http://192.168.10.83:8090
+com.qmth.doc.server=http://192.168.10.83:8090
 # logger config
 com.qmth.logging.file-path=logs/demo.log
 com.qmth.logging.root-level=info

+ 8 - 3
ai-api-demo/src/test/java/com/qmth/boot/ai/api/demo/test/AiApiDemoTest.java

@@ -1,15 +1,17 @@
 package com.qmth.boot.ai.api.demo.test;
 
 import com.fasterxml.jackson.databind.json.JsonMapper;
+import com.qmth.boot.core.ai.client.DocApiClient;
 import com.qmth.boot.core.ai.client.LlmApiClient;
 import com.qmth.boot.core.ai.client.OcrApiClient;
+import com.qmth.boot.core.ai.model.doc.ParseDocTask;
+import com.qmth.boot.core.ai.model.doc.ParseDocTaskResult;
 import com.qmth.boot.core.ai.model.llm.*;
 import com.qmth.boot.core.ai.model.llm.endpoint.ChatEndpoint;
 import com.qmth.boot.core.ai.model.llm.score.AutoScoreModel;
 import com.qmth.boot.core.ai.model.llm.score.AutoScoreRequest;
 import com.qmth.boot.core.ai.model.llm.score.AutoScoreResult;
 import com.qmth.boot.core.ai.model.ocr.OcrType;
-import com.qmth.boot.core.ai.model.ocr.ParseDocTaskResult;
 import com.qmth.boot.core.ai.service.AiService;
 import com.qmth.boot.core.retrofit.utils.SignatureInfo;
 import com.qmth.boot.core.retrofit.utils.UploadFile;
@@ -43,6 +45,9 @@ public class AiApiDemoTest {
     @Resource
     private OcrApiClient ocrApiClient;
 
+    @Resource
+    private DocApiClient docApiClient;
+
     private SignatureInfo secret;
 
     private String baseUrl;
@@ -82,12 +87,12 @@ public class AiApiDemoTest {
         File file = new File("D:\\home\\大纲.pdf");
         MultipartBody.Part part = UploadFile.build("file", file.getName(), file);
 
-        // ParseDocTask parseDocTask = ocrApiClient.parseDocTask(baseUrl, secret, part);
+        // ParseDocTask parseDocTask = docApiClient.parseDocTask(baseUrl, secret, part);
         // System.out.println(parseDocTask.getTaskId());
 
         // String taskId = parseDocTask.getTaskId();
         String taskId = "task-rmebOA853Uk592w9uTOLqV8Wtffb3HhR";
-        ParseDocTaskResult parseDocTaskResult = ocrApiClient.parseDocTaskQuery(baseUrl, secret, taskId);
+        ParseDocTaskResult parseDocTaskResult = docApiClient.parseDocTaskQuery(baseUrl, secret, taskId);
         System.out.println(parseDocTaskResult.getStatus());
         System.out.println(parseDocTaskResult.getContent());
     }