2024/10/25 ai标注添加ocr传承

middleware-automatic-center-client/src/main/java/com/zhiwei/middleware/automatic/server/pojo/enums/AiInterfaceEnums.java

@@ -10,6 +10,7 @@ public enum AiInterfaceEnums {
 
     MEDIA("MEDIA"),
     REAL_SOURCE("REAL_SOURCE"),
+    OCR("OCR"),
 
     COMMON("COMMON");
 

middleware-automatic-center-son/src/main/java/com/zhiwei/middleware/automatic/son/task/aiTask/AiTaskServiceOcr.java

+package com.zhiwei.middleware.automatic.son.task.aiTask;
+
+import com.alibaba.fastjson.JSONArray;
+import com.alibaba.fastjson.JSONObject;
+import com.zhiwei.middleware.automatic.server.pojo.AiInterfaceParam;
+import com.zhiwei.middleware.automatic.server.pojo.enums.AiInterfaceEnums;
+import com.zhiwei.middleware.automatic.server.util.RedissonUtil;
+import com.zhiwei.middleware.automatic.son.dao.AiApiResultDao;
+import org.springframework.stereotype.Service;
+
+import java.util.Objects;
+
+/**
+ * author：liu-yu
+ * date: 2024/10/15 10:35
+ **/
+@Service
+public class AiTaskServiceOcr extends AbstractAiTaskService {
+
+    public AiTaskServiceOcr(RedissonUtil redissonUtil, AiApiResultDao aiApiResultDao) {
+        super(AiInterfaceEnums.OCR, redissonUtil, aiApiResultDao);
+    }
+
+    @Override
+    public JSONObject httpParamByMetadata(AiInterfaceParam aiParam) {
+        JSONObject metadata = new JSONObject();
+        JSONArray ocr_text = aiParam.getJson().getJSONArray("ocr_text");
+        if (Objects.nonNull(ocr_text)) {
+            metadata.put("ocr", ocr_text.toJavaList(String.class));
+        }
+        return metadata;
+    }
+}