feat(agent-new): add is_new DB column, API and frontend NEW badge/clear flow

Author: 葛锐

backend/apps/agent_app.py

@@ -17,7 +17,8 @@
     list_all_agent_info_impl,
     run_agent_stream,
     stop_agent_tasks,
-    get_agent_call_relationship_impl
+    get_agent_call_relationship_impl,
+    clear_agent_new_mark_impl
 )
 from utils.auth_utils import get_current_user_info, get_current_user_id
 
@@ -148,6 +149,21 @@ async def import_agent_api(request: AgentImportRequest, authorization: Optional[
             status_code=HTTPStatus.INTERNAL_SERVER_ERROR, detail="Agent import error.")
 
 
+@agent_config_router.put("/clear_new/{agent_id}")
+async def clear_agent_new_mark_api(agent_id: int, authorization: Optional[str] = Header(None)):
+    """
+    Clear the NEW mark for an agent
+    """
+    try:
+        user_id, tenant_id, _ = get_current_user_info(authorization)
+        affected_rows = await clear_agent_new_mark_impl(agent_id, tenant_id, user_id)
+        return {"message": "Agent NEW mark cleared successfully", "affected_rows": affected_rows}
+    except Exception as e:
+        logger.error(f"Failed to clear agent NEW mark: {str(e)}")
+        raise HTTPException(
+            status_code=HTTPStatus.INTERNAL_SERVER_ERROR, detail="Failed to clear agent NEW mark.")
+
+
 @agent_config_router.post("/check_name")
 async def check_agent_name_batch_api(request: AgentNameBatchCheckRequest, authorization: Optional[str] = Header(None)):
     """

backend/database/agent_db.py

@@ -1,8 +1,10 @@
 import logging
 from typing import List
 
+from sqlalchemy import update
 from database.client import get_db_session, as_dict, filter_property
 from database.db_models import AgentInfo, ToolInstance, AgentRelation
+from sqlalchemy import update, bindparam
 
 logger = logging.getLogger("agent_db")
 
@@ -68,6 +70,47 @@ def query_sub_agents_id_list(main_agent_id: int, tenant_id: str):
         return [relation.selected_agent_id for relation in relations]
 
 
+def clear_agent_new_mark(agent_id: int, tenant_id: str, user_id: str):
+    """
+    Clear the NEW mark for an agent
+
+    Args:
+        agent_id (int): Agent ID
+        tenant_id (str): Tenant ID
+        user_id (str): User ID (for audit purposes)
+    """
+    with get_db_session() as session:
+        result = session.execute(
+            update(AgentInfo)
+            .where(
+                AgentInfo.agent_id == agent_id,
+                AgentInfo.tenant_id == tenant_id,
+                AgentInfo.delete_flag == 'N'
+            )
+            .values(is_new=False, updated_by=user_id)
+        )
+        # return number of rows affected
+        return result.rowcount
+
+
+def mark_agents_as_new(agent_ids: list[int], tenant_id: str, user_id: str):
+    """
+    Mark a list of agents as new (is_new = True)
+    """
+    if not agent_ids:
+        return
+    with get_db_session() as session:
+        session.execute(
+            update(AgentInfo)
+            .where(
+                AgentInfo.agent_id.in_(agent_ids),
+                AgentInfo.tenant_id == tenant_id,
+                AgentInfo.delete_flag == 'N'
+            )
+            .values(is_new=True, updated_by=user_id)
+        )
+
+
 def create_agent(agent_info, tenant_id: str, user_id: str):
     """
     Create a new agent in the database.
@@ -82,6 +125,7 @@ def create_agent(agent_info, tenant_id: str, user_id: str):
         "tenant_id": tenant_id,
         "created_by": user_id,
         "updated_by": user_id,
+        "is_new": True,  # Mark new agents as new
     })
     with get_db_session() as session:
         new_agent = AgentInfo(**filter_property(info_with_metadata, AgentInfo))

backend/database/db_models.py

@@ -222,6 +222,7 @@ class AgentInfo(TableBase):
     business_logic_model_name = Column(String(100), doc="Model name used for business logic prompt generation")
     business_logic_model_id = Column(Integer, doc="Model ID used for business logic prompt generation, foreign key reference to model_record_t.model_id")
     group_ids = Column(String, doc="Agent group IDs list")
+    is_new = Column(Boolean, default=False, doc="Whether this agent is marked as new for the user")
 
 
 class ToolInstance(TableBase):

backend/services/agent_service.py

@@ -1114,6 +1114,9 @@ async def import_agent_impl(
             agent_stack.append(need_import_agent_id)
             agent_stack.extend(managed_agents)
 
+    # Return the mapping of original IDs to new IDs
+    return mapping_agent_id
+
 
 async def import_agent_by_agent_id(
     import_agent_info: ExportAndImportAgentInfo,
@@ -1221,6 +1224,31 @@ def load_default_agents_json_file(default_agent_path):
     return all_json_files
 
 
+async def clear_agent_new_mark_impl(agent_id: int, tenant_id: str, user_id: str):
+    """
+    Clear the NEW mark for an agent
+
+    Args:
+        agent_id (int): Agent ID
+        tenant_id (str): Tenant ID
+        user_id (str): User ID (for audit purposes)
+    """
+    from database.agent_db import clear_agent_new_mark
+
+    rowcount = clear_agent_new_mark(agent_id, tenant_id, user_id)
+    logger.info(f"clear_agent_new_mark_impl called for agent_id={agent_id}, tenant_id={tenant_id}, user_id={user_id}, affected_rows={rowcount}")
+    return rowcount
+
+
+async def mark_agents_as_new_impl(agent_ids: list[int], tenant_id: str, user_id: str):
+    """
+    Mark a list of agents as new via DB helper
+    """
+    # Deprecated: marking is handled at creation time via create_agent().
+    # Keep DB helper available in database.agent_db for ad-hoc operations.
+    return None
+
+
 async def list_all_agent_info_impl(tenant_id: str) -> list[dict]:
     """
     list all agent info
@@ -1275,6 +1303,7 @@ async def list_all_agent_info_impl(tenant_id: str) -> list[dict]:
                 "author": agent.get("author"),
                 "is_available": len(unavailable_reasons) == 0,
                 "unavailable_reasons": unavailable_reasons,
+                "is_new": agent.get("is_new", False),
                 "group_ids": convert_string_to_list(agent.get("group_ids"))
             })
 

docker/sql/v1.7.9.3_0122_add_is_new_to_ag_tenant_agent_t.sql

@@ -0,0 +1,16 @@
+-- Add is_new column to ag_tenant_agent_t table for new agent marking
+-- This migration adds a field to track whether an agent is marked as new for users
+
+-- Add is_new column with default value false
+ALTER TABLE nexent.ag_tenant_agent_t
+ADD COLUMN IF NOT EXISTS is_new BOOLEAN DEFAULT FALSE;
+
+-- Add comment for the new column
+COMMENT ON COLUMN nexent.ag_tenant_agent_t.is_new IS 'Whether this agent is marked as new for the user';
+
+-- Create index for performance on is_new queries
+CREATE INDEX IF NOT EXISTS idx_ag_tenant_agent_t_is_new
+ON nexent.ag_tenant_agent_t (tenant_id, is_new)
+WHERE delete_flag = 'N';
+
+

frontend/app/[locale]/agents/components/AgentManageComp.tsx

@@ -17,6 +17,8 @@ import log from "@/lib/logger";
 import { useState, useEffect } from "react";
 import { ImportAgentData } from "@/hooks/useAgentImport";
 import AgentImportWizard from "@/components/agent/AgentImportWizard";
+import { clearAgentNewMark } from "@/services/agentConfigService";
+import { clearAgentAndSync } from "@/lib/agentNewUtils";
 
 export default function AgentManageComp() {
   const { t } = useTranslation("common");
@@ -156,6 +158,16 @@ export default function AgentManageComp() {
       return;
     }
 
+    // Clear NEW mark when agent is selected for editing
+    try {
+      const res = await clearAgentAndSync(agent.id, queryClient);
+      if (!res?.success) {
+        log.warn("Failed to clear NEW mark on select:", res);
+      }
+    } catch (err) {
+      log.error("Failed to clear NEW mark on select:", err);
+    }
+
     // Set selected agent id to trigger the hook
     setSelectedAgentId(Number(agent.id));
   };

frontend/app/[locale]/agents/components/agentManage/AgentList.tsx

@@ -1,6 +1,6 @@
 "use client";
 
-import React, { useState } from "react";
+import React, { useState, useEffect } from "react";
 import { useTranslation } from "react-i18next";
 import { Button, Col, Flex, Tooltip, Divider, Table, theme, App } from "antd";
 import { ExclamationCircleOutlined } from "@ant-design/icons";
@@ -17,7 +17,9 @@ import {
   exportAgent,
   updateToolConfig,
 } from "@/services/agentConfigService";
+import { clearAgentNewMark } from "@/services/agentConfigService";
 import log from "@/lib/logger";
+import { clearAgentAndSync } from "@/lib/agentNewUtils";
 
 interface AgentListProps {
   agentList: Agent[];
@@ -40,6 +42,27 @@ export default function AgentList({
   const confirm = useConfirmModal();
   const queryClient = useQueryClient();
 
+  // Note: rely on agent.is_new from agentList (single source of truth).
+  // Clear NEW mark when agent is selected (sync with selection visual feedback)
+  useEffect(() => {
+    if (currentAgentId) {
+      const agentId = String(currentAgentId);
+      const agent = agentList.find(a => String(a.id) === agentId);
+      if (agent?.is_new) {
+        (async () => {
+          try {
+            const res = await clearAgentAndSync(agentId, queryClient);
+            if (!res?.success) {
+              log.warn("Failed to clear NEW mark for agent:", agentId, res);
+            }
+          } catch (err) {
+            log.error("Error clearing NEW mark:", err);
+          }
+        })();
+      }
+    }
+  }, [currentAgentId, agentList]);
+
   // Call relationship modal state
   const [callRelationshipModalVisible, setCallRelationshipModalVisible] =
     useState(false);
@@ -279,6 +302,8 @@ export default function AgentList({
               onClick: (e: any) => {
                 e.preventDefault();
                 e.stopPropagation();
+
+                // Call onSelectAgent - NEW mark clearing is handled by useEffect
                 onSelectAgent(agent);
               },
             })}
@@ -292,6 +317,7 @@ export default function AgentList({
                   const isSelected =
                     currentAgentId !== null &&
                     String(currentAgentId) === String(agent.id);
+                  const isNew = agent.is_new || false;
 
                   return (
                     <Flex
@@ -332,6 +358,13 @@ export default function AgentList({
                               <ExclamationCircleOutlined className="text-amber-500 text-sm flex-shrink-0 cursor-pointer" />
                             </Tooltip>
                           )}
+                          {isNew && (
+                            <Tooltip title={t("space.new", "New imported agent")}>
+                              <span className="inline-flex items-center px-1.5 py-0.5 rounded text-xs font-medium bg-orange-100 text-orange-700 dark:bg-orange-900/30 dark:text-orange-400 flex-shrink-0">
+                                NEW
+                              </span>
+                            </Tooltip>
+                          )}
                           {displayName && (
                             <span className="text-base leading-normal max-w-[220px] truncate break-all">
                               {displayName}

frontend/app/[locale]/chat/components/chatAgentSelector.tsx

@@ -9,6 +9,9 @@ import { fetchAllAgents } from "@/services/agentConfigService";
 import { getUrlParam } from "@/lib/utils";
 import log from "@/lib/logger";
 import { Agent, ChatAgentSelectorProps } from "@/types/chat";
+import { clearAgentNewMark } from "@/services/agentConfigService";
+import { useQueryClient } from "@tanstack/react-query";
+import { clearAgentAndSync } from "@/lib/agentNewUtils";
 
 export function ChatAgentSelector({
   selectedAgentId,
@@ -34,6 +37,7 @@ export function ChatAgentSelector({
   const selectedAgent = agents.find(
     (agent) => agent.agent_id === selectedAgentId
   );
+  const queryClient = useQueryClient();
 
   // Detect duplicate agent names and mark later-added agents as disabled
   // For agents with the same name, keep the first one (smallest ID) enabled, disable the rest
@@ -198,18 +202,28 @@ export function ChatAgentSelector({
     }
   };
 
-  const handleAgentSelect = (agentId: number | null) => {
+  const handleAgentSelect = async (agentId: number | null) => {
     // Only effectively available agents can be selected
     if (agentId !== null) {
       const agent = agents.find((a) => a.agent_id === agentId);
       if (agent) {
         const isAvailableTool = agent.is_available !== false;
         const isDuplicateDisabled = duplicateAgentInfo.disabledAgentIds.has(agent.agent_id);
         const isEffectivelyAvailable = isAvailableTool && !isDuplicateDisabled;
-        
+
         if (!isEffectivelyAvailable) {
           return; // Unavailable agents cannot be selected
         }
+
+    // Clear NEW mark when agent is selected for chat
+        try {
+          const res = await clearAgentAndSync(agentId, queryClient);
+          if (!res?.success) {
+            log.warn("Failed to clear NEW mark on select:", res);
+          }
+        } catch (e) {
+          log.error("Failed to clear NEW mark on select:", e);
+        }
       }
     }