✨Support Agent switching at conversation start.

Author: Phinease

backend/apps/agent_app.py

@@ -9,7 +9,8 @@
 from agents.create_agent_info import create_agent_run_info
 from consts.model import AgentRequest, AgentInfoRequest, AgentIDRequest, ConversationResponse, AgentImportRequest
 from services.agent_service import list_main_agent_info_impl, get_agent_info_impl, \
-    get_creating_sub_agent_info_impl, update_agent_info_impl, delete_agent_impl, export_agent_impl, import_agent_impl
+    get_creating_sub_agent_info_impl, update_agent_info_impl, delete_agent_impl, export_agent_impl, import_agent_impl, \
+    list_all_agent_info_impl
 from services.conversation_management_service import save_conversation_user, save_conversation_assistant
 from utils.config_utils import config_manager
 from utils.thread_utils import submit
@@ -84,7 +85,7 @@ async def reload_config():
     return config_manager.force_reload()
 
 
-@router.get("/list")
+@router.get("/list_main_agent_info")
 async def list_main_agent_info_api(authorization: str = Header(None)):
     """
     List all agents, create if the main Agent cannot be found.
@@ -163,4 +164,15 @@ async def import_agent_api(request: AgentImportRequest, authorization: Optional[
         return {}
     except Exception as e:
         raise HTTPException(status_code=500, detail=f"Agent import error: {str(e)}")
+
+@router.get("/list")
+async def list_all_agent_info_api(authorization: Optional[str] = Header(None), request: Request = None):
+    """
+    list all agent info
+    """
+    try:
+        user_id, tenant_id, _ = get_current_user_info(authorization, request)
+        return list_all_agent_info_impl(tenant_id=tenant_id, user_id=user_id)
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=f"Agent list error: {str(e)}")
 

backend/database/agent_db.py

@@ -58,8 +58,8 @@ def query_or_create_main_agent_id(tenant_id: str, user_id: str) -> int:
         main_agent = query.first()
 
         if main_agent is None:
-            main_agent = create_agent(agent_info={"name": "main",
-                                                  "description": "",
+            main_agent = create_agent(agent_info={"name": "manager_agent",
+                                                  "description": "You are a manager agent capable of invoking other agents and tools.",
                                                   "enabled": True}, tenant_id=tenant_id, user_id=user_id)
 
             return main_agent["agent_id"]
@@ -348,4 +348,12 @@ def check_tool_is_available(tool_id_list: List[int]):
         tools = session.query(ToolInfo).filter(ToolInfo.tool_id.in_(tool_id_list), ToolInfo.delete_flag != 'Y').all()
         return [tool.is_available for tool in tools]
 
+def query_all_agent_info_by_tenant_id(tenant_id: str):
+    """
+    Query all agent info by tenant id
+    """
+    with get_db_session() as session:
+        agents = session.query(AgentInfo).filter(AgentInfo.tenant_id == tenant_id, 
+                                                 AgentInfo.delete_flag != 'Y').order_by(AgentInfo.create_time.desc()).all()
+        return [as_dict(agent) for agent in agents]
 

backend/main_service.py

@@ -18,4 +18,4 @@
 if __name__ == "__main__":
     # Scan agents and update to database
     import_default_agents_to_pg()
-    uvicorn.run(app, host="0.0.0.0", port=5010, log_level="warning")
+    uvicorn.run(app, host="0.0.0.0", port=5010, log_level="info")

backend/services/agent_service.py

@@ -8,7 +8,7 @@
 from database.agent_db import create_agent, query_all_enabled_tool_instances, \
     query_or_create_main_agent_id, query_sub_agents, search_blank_sub_agent_by_main_agent_id, \
     search_tools_for_sub_agent, search_agent_info_by_agent_id, update_agent, delete_agent_by_id, query_all_tools, \
-    create_or_update_tool_by_tool_info, check_tool_is_available
+    create_or_update_tool_by_tool_info, check_tool_is_available, query_all_agent_info_by_tenant_id
 
 from utils.auth_utils import get_current_user_id
 from typing import Optional
@@ -231,8 +231,8 @@ def import_agent_impl(parent_agent_id: int, agent_info: ExportAndImportAgentInfo
         raise ValueError(f"Invalid model name: {agent_info.model_name}. model name must be 'main_model' or 'sub_model'.")
     if agent_info.max_steps <= 0 or agent_info.max_steps > 20:
         raise ValueError(f"Invalid max steps: {agent_info.max_steps}. max steps must be greater than 0 and less than 20.")
-    if agent_info.name == "main":
-        raise ValueError(f"Invalid agent name: {agent_info.name}. agent name cannot be 'main'.")
+    if not agent_info.name.isidentifier():
+        raise ValueError(f"Invalid agent name: {agent_info.name}. agent name must be a valid python variable name.")
     # create a new agent
     user_id, tenant_id = get_current_user_id()
     new_agent = create_agent(agent_info={"name": agent_info.name,
@@ -307,3 +307,38 @@ def import_default_agents_to_pg():
     except Exception as e:
         logger.error(f"Failed to import default agents: {str(e)}")
         raise ValueError(f"Failed to import default agents: {str(e)}")
+
+def list_all_agent_info_impl(tenant_id: str, user_id: str) -> list[dict]:
+    """
+    list all agent info
+
+    Args:
+        tenant_id (str): tenant id
+        user_id (str): user id
+
+    Raises:
+        ValueError: failed to query all agent info
+
+    Returns:
+        list: list of agent info
+    """
+    try:
+        agent_list = query_all_agent_info_by_tenant_id(tenant_id=tenant_id)
+        
+        simple_agent_list = []
+        for agent in agent_list:
+            # check agent is available
+            tool_info = search_tools_for_sub_agent(agent_id=agent["agent_id"], tenant_id=tenant_id, user_id=None)
+            tool_id_list = [tool["tool_id"] for tool in tool_info]
+            is_available = all(check_tool_is_available(tool_id_list))
+
+            simple_agent_list.append({
+                "agent_id": agent["agent_id"],
+                "name": agent["name"],
+                "description": agent["description"],
+                "is_available": is_available
+            })
+        return simple_agent_list
+    except Exception as e:
+        logger.error(f"Failed to query all agent info: {str(e)}")
+        raise ValueError(f"Failed to query all agent info: {str(e)}")

frontend/app/[locale]/chat/internal/chatInterface.tsx

@@ -102,6 +102,9 @@ export function ChatInterface() {
   // Add force scroll to bottom state control
   const [shouldScrollToBottom, setShouldScrollToBottom] = useState(false);
 
+  // Add agent selection state
+  const [selectedAgentId, setSelectedAgentId] = useState<number | null>(null);
+
   // Reset scroll to bottom state
   useEffect(() => {
     if (shouldScrollToBottom) {
@@ -514,7 +517,7 @@ export function ChatInterface() {
       }
 
       // Send request to backend API, add signal parameter
-      const reader = await conversationService.runAgent({
+      const runAgentParams: any = {
         query: finalQuery, // Use preprocessed query or original query
         conversation_id: currentConversationId,
         is_set: isSwitchedConversation || currentMessages.length <= 1,
@@ -540,7 +543,14 @@ export function ChatInterface() {
             description: description
           };
         }) : undefined // Use complete attachment object structure
-      }, currentController.signal);
+      };
+
+      // Only add agent_id if it's not null
+      if (selectedAgentId !== null) {
+        runAgentParams.agent_id = selectedAgentId;
+      }
+
+      const reader = await conversationService.runAgent(runAgentParams, currentController.signal);
 
       if (!reader) throw new Error("Response body is null")
 
@@ -1327,6 +1337,8 @@ export function ChatInterface() {
                 onOpinionChange={handleOpinionChange}
                 currentConversationId={conversationId}
                 shouldScrollToBottom={shouldScrollToBottom}
+                selectedAgentId={selectedAgentId}
+                onAgentSelect={setSelectedAgentId}
               />
 
               {/* Footer */}

frontend/app/[locale]/chat/layout/chatInput.tsx

@@ -28,6 +28,7 @@ import {
 } from "react-icons/ai"
 import { extractColorsFromUri } from "@/lib/avatar"
 import { useTranslation } from "react-i18next"
+import { AgentSelector } from "@/components/ui/agentSelector"
 
 // Image viewer component
 function ImageViewer({ src, alt, onClose }: { src: string, alt: string, onClose: () => void }) {
@@ -307,6 +308,8 @@ interface ChatInputProps {
   onImageUpload?: (file: File) => void
   attachments?: FilePreview[]
   onAttachmentsChange?: (attachments: FilePreview[]) => void
+  selectedAgentId?: number | null
+  onAgentSelect?: (agentId: number | null) => void
 }
 
 export function ChatInput({
@@ -322,7 +325,9 @@ export function ChatInput({
   onFileUpload,
   onImageUpload,
   attachments = [],
-  onAttachmentsChange
+  onAttachmentsChange,
+  selectedAgentId = null,
+  onAgentSelect
 }: ChatInputProps) {
   const [isRecording, setIsRecording] = useState(false)
   const [recordingStatus, setRecordingStatus] = useState<"idle" | "recording" | "connecting" | "error">("idle")
@@ -883,6 +888,16 @@ export function ChatInput({
       />
     </div>
     <div className="h-12 bg-slate-100 relative">
+      {/* Agent selector on the left */}
+      <div className="absolute left-3 top-[40%] -translate-y-1/2">
+        <AgentSelector
+          selectedAgentId={selectedAgentId}
+          onAgentSelect={onAgentSelect || (() => {})}
+          disabled={isLoading || isStreaming}
+          isInitialMode={isInitialMode}
+        />
+      </div>
+      
       <div className="absolute right-3 top-[40%] -translate-y-1/2 flex items-center space-x-1">
         {/* Voice to text button */}
         <TooltipProvider>

frontend/app/[locale]/chat/streaming/chatStreamMain.tsx

@@ -39,6 +39,8 @@ interface ChatStreamMainProps {
   onOpinionChange?: (messageId: number, opinion: 'Y' | 'N' | null) => void
   currentConversationId?: number
   shouldScrollToBottom?: boolean
+  selectedAgentId?: number | null
+  onAgentSelect?: (agentId: number | null) => void
 }
 
 export function ChatStreamMain({
@@ -62,6 +64,8 @@ export function ChatStreamMain({
   onOpinionChange,
   currentConversationId,
   shouldScrollToBottom,
+  selectedAgentId,
+  onAgentSelect,
 }: ChatStreamMainProps) {
   const { t } = useTranslation();
   // Animation variants for ChatInput
@@ -424,6 +428,8 @@ export function ChatStreamMain({
                           onAttachmentsChange={onAttachmentsChange}
                           onFileUpload={onFileUpload}
                           onImageUpload={onImageUpload}
+                          selectedAgentId={selectedAgentId}
+                          onAgentSelect={onAgentSelect}
                         />
                         </motion.div>
                       </AnimatePresence>
@@ -505,6 +511,8 @@ export function ChatStreamMain({
               onAttachmentsChange={onAttachmentsChange}
               onFileUpload={onFileUpload}
               onImageUpload={onImageUpload}
+              selectedAgentId={selectedAgentId}
+              onAgentSelect={onAgentSelect}
             />
           </motion.div>
         </AnimatePresence>