|
@@ -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());
|
|
|
}
|