Эх сурвалжийг харах

refactor(infra): 优化 AI 接口调用和代码生成相关代码

- 修复 AI接口调用中的图片和文档处理逻辑
- 修正代码生成模板查询中的分页参数
- 优化用户信息 AI 处理中的文本信息生成逻辑
zrd 3 сар өмнө
parent
commit
03c3993109

+ 7 - 5
yudao-module-infra/yudao-module-infra-biz/src/main/java/cn/iocoder/yudao/module/infra/api/ai/AiApiImpl.java

@@ -48,8 +48,9 @@ public class AiApiImpl implements AiApi {
     public String dataDual(DifyReqDTO createReqVO) {
         Map<String, Object> inputs = new HashMap<>();
         String apiKey = DictFrameworkUtils.parseDictDataValue("ai_key", "数据提取");
-        List<Map<String, Object>> images = new ArrayList<>();
+        
         if (CollUtil.isNotEmpty(createReqVO.getImages())) {
+            List<Map<String, Object>> images = new ArrayList<>();
             for (String image : createReqVO.getImages()) {
                 Map<String, Object> variableValue = new HashMap<>();
                 variableValue.put("transfer_method", "remote_url");
@@ -61,13 +62,14 @@ public class AiApiImpl implements AiApi {
             inputs.put("images", images);
         }
         if (CollUtil.isNotEmpty(createReqVO.getDocs())) {
-            for (String image : createReqVO.getImages()) {
+            List<Map<String, Object>> docsList = new ArrayList<>();
+            for (String docs : createReqVO.getDocs()) {
                 Map<String, Object> variableValue = new HashMap<>();
                 variableValue.put("transfer_method", "remote_url");
-                variableValue.put("url", image);
+                variableValue.put("url", docs);
                 variableValue.put("type", "document");
-                images.add(variableValue);
-                inputs.put("docs", images);
+                docsList.add(variableValue);
+                inputs.put("docs", docsList);
             }
         }
         

+ 1 - 1
yudao-module-infra/yudao-module-infra-biz/src/main/java/cn/iocoder/yudao/module/infra/service/codegen/CodegenServiceImpl.java

@@ -321,7 +321,7 @@ public class CodegenServiceImpl implements CodegenService {
             }
         }
         CodegenTemplePageReqVO pageReqVO = new CodegenTemplePageReqVO();
-        pageReqVO.setPageNo(-1);
+        pageReqVO.setPageSize(-1);
         pageReqVO.setScene(table.getScene());
         PageResult<CodegenTempleDO> rs = codegenTempleService.getCodegenTemplePage(pageReqVO);
         List<CodegenTempleDO> codegenTempleDOs = rs.getList();

+ 0 - 1
yudao-module-member/yudao-module-member-biz/src/main/java/cn/iocoder/yudao/module/member/service/userinfoai/UserInfoAiServiceImpl.java

@@ -62,7 +62,6 @@ public class UserInfoAiServiceImpl implements UserInfoAiService {
                         .eq(UserInfoAiDO::getInformationType, updateReqVO.getInformationType())
                         .setSql("info_Status =0"));
         updateReqVO.setInfoStatus("1");
-        updateReqVO.setTextInformation(AiFrameworkUtils.dataDual(updateReqVO.getInfoFiles()));
         if (CollUtil.isNotEmpty(updateReqVO.getInfoFiles())) {
             updateReqVO.setTextInformation(AiFrameworkUtils.dataDual(updateReqVO.getInfoFiles()));
         }