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 f970d6864..f03742a59 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
@@ -51,7 +51,7 @@ public class PromptTemplatesServiceImpl implements PromptTemplatesService {
         // 暂时先去掉这个逻辑，演示用 -- zhangtao
 //        promptTemplates.setTemplateType(2);
         String categoryName = this.dictDataApi.getDictDataLabel("llm_prompt_template_framework",Integer.valueOf(createReqVO
-                .getTemplateCategory()));
+                .getTemplateFramework()));
         switch (categoryName) {
             case "简介框架":
                 BriefFramework briefFramework = createReqVO.getBriefFramework();
@@ -103,7 +103,7 @@ public class PromptTemplatesServiceImpl implements PromptTemplatesService {
         }
         PromptTemplatesDO updateObj = BeanUtils.toBean(updateReqVO, PromptTemplatesDO.class);
         String categoryName = this.dictDataApi.getDictDataLabel("llm_prompt_template_framework",Integer.valueOf(updateReqVO
-                .getTemplateCategory()));
+                .getTemplateFramework()));
         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("llm_prompt_template_framework",Integer.valueOf(templatesDO.getTemplateCategory()));
+        String categoryName = this.dictDataApi.getDictDataLabel("llm_prompt_template_framework",Integer.valueOf(templatesDO.getTemplateFramework()));
         switch (categoryName) {
             case "简介框架":
                 promptTemplates.setBriefFramework(JSON.parseObject(promptTemplates