From 338f3b81425ab96d8c856909a775124af5365e3c Mon Sep 17 00:00:00 2001
From: Junjie <fallin.jie@qq.com>
Date: 星期五, 13 三月 2026 08:46:16 +0800
Subject: [PATCH] #
---
src/main/java/com/zy/ai/controller/AiPromptTemplateController.java | 18 ++++++++++++++----
1 files changed, 14 insertions(+), 4 deletions(-)
diff --git a/src/main/java/com/zy/ai/controller/AiPromptTemplateController.java b/src/main/java/com/zy/ai/controller/AiPromptTemplateController.java
index 7e919c5..295f802 100644
--- a/src/main/java/com/zy/ai/controller/AiPromptTemplateController.java
+++ b/src/main/java/com/zy/ai/controller/AiPromptTemplateController.java
@@ -45,7 +45,7 @@
wrapper.eq("status", status);
}
wrapper.orderByAsc("scene_code").orderByDesc("version").orderByDesc("id");
- List<AiPromptTemplate> list = aiPromptTemplateService.list(wrapper);
+ List<AiPromptTemplate> list = aiPromptTemplateService.enrichTemplates(aiPromptTemplateService.list(wrapper));
return R.ok(list);
}
@@ -53,7 +53,7 @@
@ManagerAuth
public R active(@RequestParam("sceneCode") String sceneCode) {
try {
- return R.ok(aiPromptTemplateService.resolvePublished(sceneCode));
+ return R.ok(aiPromptTemplateService.enrichTemplate(aiPromptTemplateService.resolvePublished(sceneCode)));
} catch (IllegalArgumentException | IllegalStateException e) {
return R.error(e.getMessage());
}
@@ -63,7 +63,7 @@
@ManagerAuth
public R save(@RequestBody AiPromptTemplate template) {
try {
- return R.ok(aiPromptTemplateService.savePrompt(template, getUserId()));
+ return R.ok(aiPromptTemplateService.enrichTemplate(aiPromptTemplateService.savePrompt(template, getUserId())));
} catch (IllegalArgumentException e) {
return R.error(e.getMessage());
}
@@ -73,7 +73,17 @@
@ManagerAuth
public R publish(@RequestParam("id") Long id) {
try {
- return R.ok(aiPromptTemplateService.publishPrompt(id, getUserId()));
+ return R.ok(aiPromptTemplateService.enrichTemplate(aiPromptTemplateService.publishPrompt(id, getUserId())));
+ } catch (IllegalArgumentException e) {
+ return R.error(e.getMessage());
+ }
+ }
+
+ @PostMapping("/cancelPublish/auth")
+ @ManagerAuth
+ public R cancelPublish(@RequestParam("id") Long id) {
+ try {
+ return R.ok(aiPromptTemplateService.enrichTemplate(aiPromptTemplateService.cancelPublish(id, getUserId())));
} catch (IllegalArgumentException e) {
return R.error(e.getMessage());
}
--
Gitblit v1.9.1