core-ai parseDoc api

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

@@ -9,6 +9,7 @@ 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;
@@ -59,17 +60,17 @@ public class AiApiDemoTest {
 
     @Test
     public void testAll() throws Exception {
-        // testOcr();
+        // testOcrImage();
+        // testOrcDoc();
         // autoScore();
         // autoGenerateQuestionWithStream();
         // autoGenerateQuestion();
-        // chatTemplate();
         // chat();
         // getPromptTemplate();
         // getLlmBalance();
     }
 
-    private void testOcr() throws Exception {
+    private void testOcrImage() throws Exception {
         File file = new File("D:\\home\\test.png");
         MultipartBody.Part part = UploadFile.build("image", file.getName(), file);
 
@@ -77,6 +78,20 @@ public class AiApiDemoTest {
         System.out.println(result);
     }
 
+    private void testOrcDoc() throws Exception {
+        File file = new File("D:\\home\\大纲.pdf");
+        MultipartBody.Part part = UploadFile.build("file", file.getName(), file);
+
+        // ParseDocTask parseDocTask = ocrApiClient.parseDocTask(baseUrl, secret, part);
+        // System.out.println(parseDocTask.getTaskId());
+
+        // String taskId = parseDocTask.getTaskId();
+        String taskId = "task-rmebOA853Uk592w9uTOLqV8Wtffb3HhR";
+        ParseDocTaskResult parseDocTaskResult = ocrApiClient.parseDocTaskQuery(baseUrl, secret, taskId);
+        System.out.println(parseDocTaskResult.getStatus());
+        System.out.println(parseDocTaskResult.getContent());
+    }
+
     private void autoScore() throws Exception {
         AutoScoreRequest request = new AutoScoreRequest();
         request.setSubjectName("语文");
@@ -156,18 +171,12 @@ public class AiApiDemoTest {
         System.out.println(new JsonMapper().writeValueAsString(result));
     }
 
-    private void chatTemplate() throws Exception {
-        Map<String, Object> param = this.buildQuestionParam();
-        ChatResult result = llmApiClient.chatTemplate(baseUrl, secret, LlmAppType.AUTO_GENERATE_QUESTION, param);
-        System.out.println(new JsonMapper().writeValueAsString(result));
-    }
-
     private Map<String, Object> buildQuestionParam() {
         return new HashMap<String, Object>() {{
             put("courseName", "语文");
             put("questionTypeName", "多选题");
-            put("syllabusAttachment", true);// 是否包含教学大纲附件
-            put("syllabusNotes", "熟练掌握诗词");// 教学大纲补充说明
+            put("syllabus", "熟练掌握诗词");// 教学大纲内容
+            put("syllabusNotes", "");// 教学大纲补充说明
             put("knowledgePoint", "诗词");// 知识点
             put("knowledgeNotes", "作者出处");// 知识点补充说明
             // SINGLE_ANSWER_QUESTION、MULTIPLE_ANSWER_QUESTION、BOOL_ANSWER_QUESTION、FILL_BLANK_QUESTION、TEXT_ANSWER_QUESTION