mirror of
https://github.com/csunny/DB-GPT.git
synced 2025-09-05 11:01:09 +00:00
feat: (0.6)New UI (#1855)
Co-authored-by: 夏姜 <wenfengjiang.jwf@digital-engine.com> Co-authored-by: aries_ckt <916701291@qq.com> Co-authored-by: wb-lh513319 <wb-lh513319@alibaba-inc.com> Co-authored-by: csunny <cfqsunny@163.com>
This commit is contained in:
@@ -7,6 +7,8 @@ from typing import Dict, List, Optional, Set, Tuple, Type, cast
|
||||
|
||||
from dbgpt.component import BaseComponent, ComponentType, SystemApp
|
||||
|
||||
from ..expand.Indicator_assistant_agent import IndicatorAssistantAgent
|
||||
from ..expand.simple_assistant_agent import SimpleAssistantAgent
|
||||
from .agent import Agent
|
||||
from .base_agent import ConversableAgent
|
||||
|
||||
@@ -77,6 +79,9 @@ class AgentManager(BaseComponent):
|
||||
core_agents.add(self.register_agent(DataScientistAgent))
|
||||
core_agents.add(self.register_agent(SummaryAssistantAgent))
|
||||
core_agents.add(self.register_agent(ToolAssistantAgent))
|
||||
core_agents.add(self.register_agent(IndicatorAssistantAgent))
|
||||
core_agents.add(self.register_agent(SimpleAssistantAgent))
|
||||
|
||||
self._core_agents = core_agents
|
||||
|
||||
def register_agent(
|
||||
@@ -122,6 +127,9 @@ class AgentManager(BaseComponent):
|
||||
def list_agents(self):
|
||||
"""Return a list of all registered agents and their descriptions."""
|
||||
result = []
|
||||
from datetime import datetime
|
||||
|
||||
logger.info(f"List Agent Begin:{datetime.now()}")
|
||||
for name, value in self._agents.items():
|
||||
result.append(
|
||||
{
|
||||
@@ -129,6 +137,7 @@ class AgentManager(BaseComponent):
|
||||
"desc": value[1].goal,
|
||||
}
|
||||
)
|
||||
logger.info(f"List Agent End:{datetime.now()}")
|
||||
return result
|
||||
|
||||
|
||||
|
Reference in New Issue
Block a user