From 3db25972405ba894f61f76f82e2f849930bb9c26 Mon Sep 17 00:00:00 2001
From: ire <931903008@qq.com>
Date: Tue, 31 Dec 2024 20:07:20 +0800
Subject: [PATCH] bug

---
 .../prompttemplates/PromptTemplatesServiceImpl.java  | 12 ++++++------
 1 file changed, 6 insertions(+), 6 deletions(-)

diff --git a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/prompttemplates/PromptTemplatesServiceImpl.java b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/prompttemplates/PromptTemplatesServiceImpl.java
index a0954cc88..f970d6864 100644
--- a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/prompttemplates/PromptTemplatesServiceImpl.java
+++ b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/prompttemplates/PromptTemplatesServiceImpl.java
@@ -50,8 +50,8 @@ public class PromptTemplatesServiceImpl implements PromptTemplatesService {
         PromptTemplatesDO promptTemplates = BeanUtils.toBean(createReqVO, PromptTemplatesDO.class);
         // 暂时先去掉这个逻辑，演示用 -- zhangtao
 //        promptTemplates.setTemplateType(2);
-        String categoryName = this.dictDataApi.getDictDataLabel(Long.valueOf(createReqVO
-                .getTemplateCategory().toString()));
+        String categoryName = this.dictDataApi.getDictDataLabel("llm_prompt_template_framework",Integer.valueOf(createReqVO
+                .getTemplateCategory()));
         switch (categoryName) {
             case "简介框架":
                 BriefFramework briefFramework = createReqVO.getBriefFramework();
@@ -102,8 +102,8 @@ public class PromptTemplatesServiceImpl implements PromptTemplatesService {
             this.isNameDuplicate(updateReqVO.getName());
         }
         PromptTemplatesDO updateObj = BeanUtils.toBean(updateReqVO, PromptTemplatesDO.class);
-        String categoryName = this.dictDataApi.getDictDataLabel(Long.valueOf(updateReqVO
-                .getTemplateCategory().toString()));
+        String categoryName = this.dictDataApi.getDictDataLabel("llm_prompt_template_framework",Integer.valueOf(updateReqVO
+                .getTemplateCategory()));
         switch (categoryName) {
             case "简介框架":
                 BriefFramework briefFramework = updateReqVO.getBriefFramework();
@@ -193,7 +193,7 @@ public class PromptTemplatesServiceImpl implements PromptTemplatesService {
                     .collect(Collectors.toList());
             promptTemplates.setApplicationIds(applicationIds);
         }
-        String categoryName = this.dictDataApi.getDictDataLabel(Long.valueOf(templatesDO.getTemplateCategory().toString()));
+        String categoryName = this.dictDataApi.getDictDataLabel("llm_prompt_template_framework",Integer.valueOf(templatesDO.getTemplateCategory()));
         switch (categoryName) {
             case "简介框架":
                 promptTemplates.setBriefFramework(JSON.parseObject(promptTemplates
@@ -278,4 +278,4 @@ public class PromptTemplatesServiceImpl implements PromptTemplatesService {
         ExcelUtils.write(response, "Prompt模板信息.xls", "数据",
                 PromptTemplatesExcelVO.class, excelList);
     }
-}
\ No newline at end of file
+}