diff --git a/src/agents/__init__.py b/src/agents/__init__.py index 9448ff6f..bf565b90 100644 --- a/src/agents/__init__.py +++ b/src/agents/__init__.py @@ -7,20 +7,20 @@ class AgentManager: self._classes = {} self._instances = {} # 存储已创建的 agent 实例 - def register_agent(self, agent_id, agent_class): - self._classes[agent_id] = agent_class + def register_agent(self, agent_class): + self._classes[agent_class.name] = agent_class def init_all_agents(self): - for agent_id, agent_class in self._classes.items(): - self.get_agent(agent_id) + for agent_class in self._classes.values(): + self.get_agent(agent_class.name) - def get_agent(self, agent_id, **kwargs): + def get_agent(self, agent_name, **kwargs): # 检查是否已经创建了该 agent 的实例 - if agent_id not in self._instances: - agent_class = self._classes[agent_id] - self._instances[agent_id] = agent_class() + if agent_name not in self._instances: + agent_class = self._classes[agent_name] + self._instances[agent_name] = agent_class() - return self._instances[agent_id] + return self._instances[agent_name] def get_agents(self): return list(self._instances.values()) @@ -31,8 +31,8 @@ class AgentManager: agent_manager = AgentManager() -agent_manager.register_agent("chatbot", ChatbotAgent) -agent_manager.register_agent("react", ReActAgent) # 暂时屏蔽 ReActAgent +agent_manager.register_agent(ChatbotAgent) +agent_manager.register_agent(ReActAgent) # 暂时屏蔽 ReActAgent agent_manager.init_all_agents() __all__ = ["agent_manager"] diff --git a/src/agents/react/graph.py b/src/agents/react/graph.py index ca4d543d..de14cfe7 100644 --- a/src/agents/react/graph.py +++ b/src/agents/react/graph.py @@ -7,7 +7,7 @@ from src.agents.registry import BaseAgent from src.agents.react.configuration import ReActConfiguration class ReActAgent(BaseAgent): - name = "react" + name = "ReAct" description = "A react agent that can answer questions and help with tasks." config_schema = ReActConfiguration diff --git a/web/src/components/AgentMessageComponent.vue b/web/src/components/AgentMessageComponent.vue index c421af76..f8fda072 100644 --- a/web/src/components/AgentMessageComponent.vue +++ b/web/src/components/AgentMessageComponent.vue @@ -550,6 +550,10 @@ const toggleToolCall = (toolCallId) => { -moz-tab-size: 4; background-color: var(--gray-100); } + + p:last-child { + margin-bottom: 0; + } } .chat-box.font-smaller #preview-only-preview { diff --git a/web/src/components/ChatSidebarComponent.vue b/web/src/components/ChatSidebarComponent.vue index 6e89937f..ba624556 100644 --- a/web/src/components/ChatSidebarComponent.vue +++ b/web/src/components/ChatSidebarComponent.vue @@ -263,7 +263,6 @@ const toggleCollapse = () => { .conversation-title { font-size: 14px; color: var(--gray-900); - margin-bottom: 4px; white-space: nowrap; overflow: hidden; text-overflow: ellipsis; diff --git a/web/src/components/DebugComponent.vue b/web/src/components/DebugComponent.vue index 164fefd9..51e641f3 100644 --- a/web/src/components/DebugComponent.vue +++ b/web/src/components/DebugComponent.vue @@ -2,13 +2,9 @@