From d429dc887a9b74f55068e255a7bb6680610b96dd Mon Sep 17 00:00:00 2001
From: sion123 <450702724@qq.com>
Date: Thu, 26 Feb 2026 20:15:24 +0800
Subject: [PATCH] feat: add agent favorite functionality with UI and API
 implementation

- Add `addFavorite` and `removeFavorite` API functions in agent.js
- Implement favorite button UI in Agents.vue with star icons
- Add login check before favorite operations using token manager
- Sort agents with favorites appearing at the top of the list
- Include success/error messages for user feedback
- Add backend API endpoints for creating/deleting agent favorites
- Update agent list response to include favorite status
- Style favorite button with hover and active states
---
 frontend/app/web-gold/src/api/agent.js        | 24 ++++++
 .../app/web-gold/src/views/agents/Agents.vue  | 83 +++++++++++++++++--
 .../muye/aiagent/AppAiAgentController.java    | 44 ++++++++--
 .../muye/aiagent/dal/AiAgentFavoriteDO.java   | 34 ++++++++
 .../aiagent/mapper/AiAgentFavoriteMapper.java | 21 +++++
 .../service/AiAgentFavoriteService.java       | 47 +++++++++++
 .../service/AiAgentFavoriteServiceImpl.java   | 70 ++++++++++++++++
 .../tik/muye/aiagent/vo/AppAiAgentRespVO.java |  3 +
 8 files changed, 315 insertions(+), 11 deletions(-)
 create mode 100644 yudao-module-tik/src/main/java/cn/iocoder/yudao/module/tik/muye/aiagent/dal/AiAgentFavoriteDO.java
 create mode 100644 yudao-module-tik/src/main/java/cn/iocoder/yudao/module/tik/muye/aiagent/mapper/AiAgentFavoriteMapper.java
 create mode 100644 yudao-module-tik/src/main/java/cn/iocoder/yudao/module/tik/muye/aiagent/service/AiAgentFavoriteService.java
 create mode 100644 yudao-module-tik/src/main/java/cn/iocoder/yudao/module/tik/muye/aiagent/service/AiAgentFavoriteServiceImpl.java

diff --git a/frontend/app/web-gold/src/api/agent.js b/frontend/app/web-gold/src/api/agent.js
index 00f3c16742..2ef42581a6 100644
--- a/frontend/app/web-gold/src/api/agent.js
+++ b/frontend/app/web-gold/src/api/agent.js
@@ -116,3 +116,27 @@ export function getMessages(params) {
     params
   })
 }
+
+/**
+ * 添加智能体收藏
+ * @param {number} agentId - 智能体ID
+ */
+export function addFavorite(agentId) {
+  return request({
+    url: `${BASE_URL}/agent/favorite/create`,
+    method: 'post',
+    params: { agentId }
+  })
+}
+
+/**
+ * 取消智能体收藏
+ * @param {number} agentId - 智能体ID
+ */
+export function removeFavorite(agentId) {
+  return request({
+    url: `${BASE_URL}/agent/favorite/delete`,
+    method: 'delete',
+    params: { agentId }
+  })
+}
diff --git a/frontend/app/web-gold/src/views/agents/Agents.vue b/frontend/app/web-gold/src/views/agents/Agents.vue
index b7a93a0a06..597a717439 100644
--- a/frontend/app/web-gold/src/views/agents/Agents.vue
+++ b/frontend/app/web-gold/src/views/agents/Agents.vue
@@ -134,6 +134,15 @@
 
                 <!-- 底部操作栏 -->
                 <div class="card-footer">
+                  <button
+                    class="favorite-btn"
+                    :class="{ 'favorite-btn--active': agent.isFavorite }"
+                    @click.stop="handleFavorite(agent)"
+                  >
+                    <StarFilled v-if="agent.isFavorite" />
+                    <StarOutlined v-else />
+                    <span>{{ agent.isFavorite ? '已收藏' : '收藏' }}</span>
+                  </button>
                   <div class="footer-spacer"></div>
                   <button class="chat-btn" @click.stop="handleChat(agent)">
                     <MessageOutlined class="chat-btn-icon" />
@@ -172,12 +181,15 @@ import {
   CloseOutlined,
   ArrowRightOutlined,
   MessageOutlined,
-  AppstoreOutlined
+  AppstoreOutlined,
+  StarOutlined,
+  StarFilled
 } from '@ant-design/icons-vue'
 import { message } from 'ant-design-vue'
 import FullWidthLayout from '@/layouts/components/FullWidthLayout.vue'
 import ChatDrawer from '@/components/agents/ChatDrawer.vue'
-import { getAgentList } from '@/api/agent'
+import { getAgentList, addFavorite, removeFavorite } from '@/api/agent'
+import tokenManager from '@gold/utils/token-manager'
 
 // 状态管理
 const loading = ref(false)
@@ -218,7 +230,7 @@ const categories = computed(() => {
   return cats
 })
 
-// 过滤后的列表
+// 过滤后的列表（收藏置顶）
 const filteredAgentList = computed(() => {
   let list = agentList.value
 
@@ -234,7 +246,11 @@ const filteredAgentList = computed(() => {
     )
   }
 
-  return list
+  // 收藏的智能体置顶
+  return list.sort((a, b) => {
+    if (a.isFavorite === b.isFavorite) return 0
+    return a.isFavorite ? -1 : 1
+  })
 })
 
 // 获取智能体列表
@@ -250,7 +266,8 @@ const fetchAgentList = async () => {
         description: item.description,
         avatar: item.icon,
         categoryName: item.categoryName || '其他',
-        tagColor: getTagColor(item.categoryName)
+        tagColor: getTagColor(item.categoryName),
+        isFavorite: item.isFavorite || false
       }))
     }
   } catch (error) {
@@ -327,6 +344,30 @@ const handleSendMessage = (data) => {
   console.log('发送消息:', data)
 }
 
+// 收藏切换
+const handleFavorite = async (agent) => {
+  // 检查登录状态
+  if (!tokenManager.isLoggedIn()) {
+    message.warning('请先登录')
+    return
+  }
+
+  try {
+    if (agent.isFavorite) {
+      await removeFavorite(agent.id)
+      agent.isFavorite = false
+      message.success('已取消收藏')
+    } else {
+      await addFavorite(agent.id)
+      agent.isFavorite = true
+      message.success('收藏成功')
+    }
+  } catch (error) {
+    console.error('收藏操作失败:', error)
+    message.error('操作失败，请重试')
+  }
+}
+
 // 初始化
 onMounted(() => {
   fetchAgentList()
@@ -799,6 +840,38 @@ onMounted(() => {
   font-size: 13px;
 }
 
+// 收藏按钮
+.favorite-btn {
+  display: inline-flex;
+  align-items: center;
+  gap: 5px;
+  padding: 6px 12px;
+  background: transparent;
+  border: 1px solid var(--color-gray-300);
+  border-radius: 8px;
+  font-size: 12px;
+  font-weight: 500;
+  color: var(--color-gray-600);
+  cursor: pointer;
+  transition: all 0.2s ease;
+
+  &:hover {
+    border-color: #F59E0B;
+    color: #F59E0B;
+    background: #FFFBEB;
+  }
+
+  &:active {
+    transform: scale(0.95);
+  }
+
+  &--active {
+    border-color: #F59E0B;
+    color: #F59E0B;
+    background: #FFFBEB;
+  }
+}
+
 // ============================================
 // 空状态
 // ============================================
diff --git a/yudao-module-tik/src/main/java/cn/iocoder/yudao/module/tik/muye/aiagent/AppAiAgentController.java b/yudao-module-tik/src/main/java/cn/iocoder/yudao/module/tik/muye/aiagent/AppAiAgentController.java
index 1f979c2cb2..dd93bbddb9 100644
--- a/yudao-module-tik/src/main/java/cn/iocoder/yudao/module/tik/muye/aiagent/AppAiAgentController.java
+++ b/yudao-module-tik/src/main/java/cn/iocoder/yudao/module/tik/muye/aiagent/AppAiAgentController.java
@@ -2,25 +2,25 @@ package cn.iocoder.yudao.module.tik.muye.aiagent;
 
 import cn.iocoder.yudao.framework.common.pojo.CommonResult;
 import cn.iocoder.yudao.framework.common.util.object.BeanUtils;
+import cn.iocoder.yudao.framework.security.core.util.SecurityFrameworkUtils;
 import cn.iocoder.yudao.module.tik.muye.aiagent.dal.AiAgentDO;
+import cn.iocoder.yudao.module.tik.muye.aiagent.service.AiAgentFavoriteService;
 import cn.iocoder.yudao.module.tik.muye.aiagent.service.AiAgentService;
 import cn.iocoder.yudao.module.tik.muye.aiagent.vo.AppAiAgentRespVO;
 import io.swagger.v3.oas.annotations.Operation;
+import io.swagger.v3.oas.annotations.Parameter;
 import io.swagger.v3.oas.annotations.tags.Tag;
 import jakarta.annotation.Resource;
 import org.springframework.validation.annotation.Validated;
-import org.springframework.web.bind.annotation.GetMapping;
-import org.springframework.web.bind.annotation.RequestMapping;
-import org.springframework.web.bind.annotation.RestController;
+import org.springframework.web.bind.annotation.*;
 
 import java.util.List;
+import java.util.Set;
 
 import static cn.iocoder.yudao.framework.common.pojo.CommonResult.success;
 
 /**
  * 用户 App - AI智能体
- *
- * @author 芋道源码
  */
 @Tag(name = "用户 App - AI智能体")
 @RestController
@@ -31,11 +31,43 @@ public class AppAiAgentController {
     @Resource
     private AiAgentService aiAgentService;
 
+    @Resource
+    private AiAgentFavoriteService aiAgentFavoriteService;
+
     @GetMapping("/list")
     @Operation(summary = "获取启用的智能体列表")
     public CommonResult<List<AppAiAgentRespVO>> getEnabledAgentList() {
+        // 获取智能体列表
         List<AiAgentDO> list = aiAgentService.getEnabledAgentList();
-        return success(BeanUtils.toBean(list, AppAiAgentRespVO.class));
+        List<AppAiAgentRespVO> voList = BeanUtils.toBean(list, AppAiAgentRespVO.class);
+
+        // 获取当前用户收藏的智能体ID
+        Long userId = SecurityFrameworkUtils.getLoginUserId();
+        if (userId != null) {
+            Set<Long> favoriteIds = aiAgentFavoriteService.getFavoriteAgentIds(userId);
+            voList.forEach(vo -> vo.setIsFavorite(favoriteIds.contains(vo.getId())));
+        } else {
+            voList.forEach(vo -> vo.setIsFavorite(false));
+        }
+
+        return success(voList);
+    }
+
+    @PostMapping("/favorite/create")
+    @Operation(summary = "添加收藏")
+    @Parameter(name = "agentId", description = "智能体ID", required = true)
+    public CommonResult<Long> createFavorite(@RequestParam("agentId") Long agentId) {
+        Long userId = SecurityFrameworkUtils.getLoginUserId();
+        return success(aiAgentFavoriteService.createFavorite(userId, agentId));
+    }
+
+    @DeleteMapping("/favorite/delete")
+    @Operation(summary = "取消收藏")
+    @Parameter(name = "agentId", description = "智能体ID", required = true)
+    public CommonResult<Boolean> deleteFavorite(@RequestParam("agentId") Long agentId) {
+        Long userId = SecurityFrameworkUtils.getLoginUserId();
+        aiAgentFavoriteService.deleteFavorite(userId, agentId);
+        return success(true);
     }
 
 }
diff --git a/yudao-module-tik/src/main/java/cn/iocoder/yudao/module/tik/muye/aiagent/dal/AiAgentFavoriteDO.java b/yudao-module-tik/src/main/java/cn/iocoder/yudao/module/tik/muye/aiagent/dal/AiAgentFavoriteDO.java
new file mode 100644
index 0000000000..8fc4563bc7
--- /dev/null
+++ b/yudao-module-tik/src/main/java/cn/iocoder/yudao/module/tik/muye/aiagent/dal/AiAgentFavoriteDO.java
@@ -0,0 +1,34 @@
+package cn.iocoder.yudao.module.tik.muye.aiagent.dal;
+
+import lombok.*;
+import com.baomidou.mybatisplus.annotation.*;
+import cn.iocoder.yudao.framework.mybatis.core.dataobject.BaseDO;
+
+/**
+ * AI智能体收藏 DO
+ */
+@TableName("muye_ai_agent_favorite")
+@KeySequence("muye_ai_agent_favorite_seq")
+@Data
+@EqualsAndHashCode(callSuper = true)
+@ToString(callSuper = true)
+@Builder
+@NoArgsConstructor
+@AllArgsConstructor
+public class AiAgentFavoriteDO extends BaseDO {
+
+    /**
+     * 主键
+     */
+    @TableId
+    private Long id;
+    /**
+     * 用户ID
+     */
+    private Long userId;
+    /**
+     * 智能体ID
+     */
+    private Long agentId;
+
+}
diff --git a/yudao-module-tik/src/main/java/cn/iocoder/yudao/module/tik/muye/aiagent/mapper/AiAgentFavoriteMapper.java b/yudao-module-tik/src/main/java/cn/iocoder/yudao/module/tik/muye/aiagent/mapper/AiAgentFavoriteMapper.java
new file mode 100644
index 0000000000..a7ff130ef5
--- /dev/null
+++ b/yudao-module-tik/src/main/java/cn/iocoder/yudao/module/tik/muye/aiagent/mapper/AiAgentFavoriteMapper.java
@@ -0,0 +1,21 @@
+package cn.iocoder.yudao.module.tik.muye.aiagent.mapper;
+
+import cn.iocoder.yudao.framework.mybatis.core.mapper.BaseMapperX;
+import cn.iocoder.yudao.module.tik.muye.aiagent.dal.AiAgentFavoriteDO;
+import org.apache.ibatis.annotations.Mapper;
+
+import java.util.List;
+
+@Mapper
+public interface AiAgentFavoriteMapper extends BaseMapperX<AiAgentFavoriteDO> {
+
+    default AiAgentFavoriteDO selectByUserIdAndAgentId(Long userId, Long agentId) {
+        return selectOne(AiAgentFavoriteDO::getUserId, userId,
+                AiAgentFavoriteDO::getAgentId, agentId);
+    }
+
+    default List<AiAgentFavoriteDO> selectListByUserId(Long userId) {
+        return selectList(AiAgentFavoriteDO::getUserId, userId);
+    }
+
+}
diff --git a/yudao-module-tik/src/main/java/cn/iocoder/yudao/module/tik/muye/aiagent/service/AiAgentFavoriteService.java b/yudao-module-tik/src/main/java/cn/iocoder/yudao/module/tik/muye/aiagent/service/AiAgentFavoriteService.java
new file mode 100644
index 0000000000..8eeb55ac4b
--- /dev/null
+++ b/yudao-module-tik/src/main/java/cn/iocoder/yudao/module/tik/muye/aiagent/service/AiAgentFavoriteService.java
@@ -0,0 +1,47 @@
+package cn.iocoder.yudao.module.tik.muye.aiagent.service;
+
+import cn.iocoder.yudao.module.tik.muye.aiagent.dal.AiAgentFavoriteDO;
+
+import java.util.List;
+import java.util.Set;
+
+/**
+ * AI智能体收藏 Service 接口
+ */
+public interface AiAgentFavoriteService {
+
+    /**
+     * 添加收藏
+     *
+     * @param userId  用户ID
+     * @param agentId 智能体ID
+     * @return 收藏ID
+     */
+    Long createFavorite(Long userId, Long agentId);
+
+    /**
+     * 取消收藏
+     *
+     * @param userId  用户ID
+     * @param agentId 智能体ID
+     */
+    void deleteFavorite(Long userId, Long agentId);
+
+    /**
+     * 获取用户收藏的智能体ID列表
+     *
+     * @param userId 用户ID
+     * @return 智能体ID集合
+     */
+    Set<Long> getFavoriteAgentIds(Long userId);
+
+    /**
+     * 判断是否已收藏
+     *
+     * @param userId  用户ID
+     * @param agentId 智能体ID
+     * @return 是否已收藏
+     */
+    boolean isFavorite(Long userId, Long agentId);
+
+}
diff --git a/yudao-module-tik/src/main/java/cn/iocoder/yudao/module/tik/muye/aiagent/service/AiAgentFavoriteServiceImpl.java b/yudao-module-tik/src/main/java/cn/iocoder/yudao/module/tik/muye/aiagent/service/AiAgentFavoriteServiceImpl.java
new file mode 100644
index 0000000000..2beb87cc66
--- /dev/null
+++ b/yudao-module-tik/src/main/java/cn/iocoder/yudao/module/tik/muye/aiagent/service/AiAgentFavoriteServiceImpl.java
@@ -0,0 +1,70 @@
+package cn.iocoder.yudao.module.tik.muye.aiagent.service;
+
+import cn.iocoder.yudao.framework.common.exception.ErrorCode;
+import cn.iocoder.yudao.module.tik.muye.aiagent.dal.AiAgentFavoriteDO;
+import cn.iocoder.yudao.module.tik.muye.aiagent.mapper.AiAgentFavoriteMapper;
+import jakarta.annotation.Resource;
+import org.springframework.stereotype.Service;
+import org.springframework.validation.annotation.Validated;
+
+import java.util.HashSet;
+import java.util.List;
+import java.util.Set;
+
+import static cn.iocoder.yudao.framework.common.exception.util.ServiceExceptionUtil.exception;
+
+/**
+ * AI智能体收藏 Service 实现类
+ */
+@Service
+@Validated
+public class AiAgentFavoriteServiceImpl implements AiAgentFavoriteService {
+
+    private static final ErrorCode FAVORITE_EXISTS = new ErrorCode(1_013_001_000, "已收藏该智能体");
+    private static final ErrorCode FAVORITE_NOT_EXISTS = new ErrorCode(1_013_001_001, "未收藏该智能体");
+
+    @Resource
+    private AiAgentFavoriteMapper aiAgentFavoriteMapper;
+
+    @Override
+    public Long createFavorite(Long userId, Long agentId) {
+        // 检查是否已收藏
+        AiAgentFavoriteDO existFavorite = aiAgentFavoriteMapper.selectByUserIdAndAgentId(userId, agentId);
+        if (existFavorite != null) {
+            throw exception(FAVORITE_EXISTS);
+        }
+
+        // 创建收藏
+        AiAgentFavoriteDO favorite = AiAgentFavoriteDO.builder()
+                .userId(userId)
+                .agentId(agentId)
+                .build();
+        aiAgentFavoriteMapper.insert(favorite);
+        return favorite.getId();
+    }
+
+    @Override
+    public void deleteFavorite(Long userId, Long agentId) {
+        AiAgentFavoriteDO favorite = aiAgentFavoriteMapper.selectByUserIdAndAgentId(userId, agentId);
+        if (favorite == null) {
+            throw exception(FAVORITE_NOT_EXISTS);
+        }
+        aiAgentFavoriteMapper.deleteById(favorite.getId());
+    }
+
+    @Override
+    public Set<Long> getFavoriteAgentIds(Long userId) {
+        List<AiAgentFavoriteDO> favorites = aiAgentFavoriteMapper.selectListByUserId(userId);
+        Set<Long> agentIds = new HashSet<>();
+        for (AiAgentFavoriteDO favorite : favorites) {
+            agentIds.add(favorite.getAgentId());
+        }
+        return agentIds;
+    }
+
+    @Override
+    public boolean isFavorite(Long userId, Long agentId) {
+        return aiAgentFavoriteMapper.selectByUserIdAndAgentId(userId, agentId) != null;
+    }
+
+}
diff --git a/yudao-module-tik/src/main/java/cn/iocoder/yudao/module/tik/muye/aiagent/vo/AppAiAgentRespVO.java b/yudao-module-tik/src/main/java/cn/iocoder/yudao/module/tik/muye/aiagent/vo/AppAiAgentRespVO.java
index cde2964bb3..d15f39a25d 100644
--- a/yudao-module-tik/src/main/java/cn/iocoder/yudao/module/tik/muye/aiagent/vo/AppAiAgentRespVO.java
+++ b/yudao-module-tik/src/main/java/cn/iocoder/yudao/module/tik/muye/aiagent/vo/AppAiAgentRespVO.java
@@ -29,4 +29,7 @@ public class AppAiAgentRespVO {
     @Schema(description = "分类名称（中文）", example = "文案创作")
     private String categoryName;
 
+    @Schema(description = "是否已收藏", example = "true")
+    private Boolean isFavorite;
+
 }