Merge branch 'feat/agent_team' into dev

Author: wangtao2001

docs/tutorials/agent/agent.md

@@ -46,10 +46,10 @@ agent = Agent(llm=llm,
 
 ```python
 from course_graph.agent import Controller
-controller = Controller(max_turns=10)
+controller = Controller()
 ```
 
-`Controller` 负责启动 `Agent` 并为 `Agent` 提供上下文、具体执行外部工具等功能, `max_turns` 参数代表运行最大轮数。
+`Controller` 负责启动 `Agent` 并为 `Agent` 提供上下文、具体执行外部工具等功能。
 
 ## 启动智能体
 
@@ -59,7 +59,7 @@ resp = controller.run_sync(agent=translator, message="请帮我翻译蛋白质
 
 其中 `message` 参数代表用户的具体指令或者是用户与智能体对话的开始。
 
-`controller.run_sync` 方法返回一个 `ControllerResponse` 对象, 其中 `agent` 代表最后响应的 `Agent` 对象 (暂时还用不到), `message` 代表智能体最后的相应内容, `turns` 代表智能体运行的轮数。
+`controller.run_sync` 方法返回一个 `ControllerResponse` 对象, 其中 `agent` 代表最后响应的 `Agent` 对象, `message` 代表智能体最后的相应内容。
 
 ## 使用外部工具
 
@@ -374,24 +374,23 @@ controller = Controller(trace_callback=pprint)
 该回调函数需要接受一个 `TraceEvent` 类型的参数, 其中事件类型包括:
 
 - `USER_MESSAGE`:   用户消息
+- `AGENT_THINK`: 智能体思考过程
 - `AGENT_MESSAGE`:  智能体消息
 - `AGENT_SWITCH`: 智能体切换
 - `TOOL_CALL`: 工具调用
 - `TOOL_RESULT`: 工具调用结果
 - `CONTEXT_UPDATE`: 上下文变量更新
 - `MCP_TOOL_CALL`: MCP 工具调用
 
-## 多智能协作
+## 多智能体协作
 
-除了让智能体使用 <a href='#transfer-function'>转移函数</a> 自行决定接下来被激活的智能体外，也提供了一些固定的协作范式，称之为团队。
+除了手动编写协作流程或让智能体使用 <a href='#transfer-function'>转移函数</a> 自行决定切换外，也提供了一些固定的协作范式，称之为团队。
 
 ### 团队
 
-以 `RoundTeam` 为例，团队中每个 `Agent` 对象会依次被激活，所有 `Agent` 以广播的方式共享相同的上下文。
-
 #### 创建团队
 
-直接使用 `RoundTeam` 类创建一个团队:
+以 `RoundTeam` 为例，直接使用 `RoundTeam` 类创建一个团队:
 
 ```python
 team = RoundTeam([agent1, agent2, agent3])
@@ -403,44 +402,49 @@ team = RoundTeam([agent1, agent2, agent3])
 team: RoundTeam = agent1 | agent2 | agent3
 ```
 
-#### 添加任务
-
-团队中所有 `Agent` 都将围绕这个任务进行协作：
-
-```python
-team.add_task("请创作一首关于春天的七律诗。")
-```
-
 #### 添加终止器
 
 团队中可以设置一个终止器，当满足终止条件时，团队将停止运行：
 
 ```python
-team.terminator = MaxTurnsTerminator(max_turns=10)
+team.termination = TextMentionTermination(text="APPROVED")
 ```
 
 终止器包含以下类型:
 
-- `MaxTurnsTerminator`: 最大轮数终止, 包含每个 `Agent` 调用工具的次数。
-- `MaxActiveTerminator`: 最大激活次数终止, 表示团队中能够激活 `Agent` 的最大次数。
-- `TextTerminator`: 当团队中任意一个 `Agent` 的响应中包含指定的文本时，团队将停止运行。
-- `TimeOutTerminator`: 超时终止，表示整个团队运行的最长时长。
+- `TextMentionTermination`: 响应中包含指定的文本时，团队将停止运行。
 
 所有终止器都支持 `&` 和 `|` 操作符来组合使用。
 
 #### 运行团队
 
-使用 `run_sync` 或 `run` 方法运行团队：
+使用 `run_sync` 或 `run` 方法运行团队，团队中所有成员都将围绕这个任务进行协作：
 
 ```python
-team.run_sync()
+team.run_sync(task="请创作一首关于春天的七律诗。")
 ```
 
-不提供返回值，你可以使用 `team.global_messages` 获取团队中所有 `Agent` 的对话历史或者 `add_trace_callback` 方法添加一个回调函数来获取内部流程。
+该方法提供一个 `TeamResponse` 类型的返回值。如果想知道团队的内部流程, 可以使用 `trace` 属性或通过 `set_trace_callback` 方法设置一个回调函数。
 
 ### 团队类型
 
+- `RoundTeam`: 轮询团队，团队中每个成员会依次被激活并循环该过程，直到满足终止条件。所有成员以广播的方式共享相同的上下文。
+
+```python
+team: RoundTeam = agent1 | agent2 | agent3
+```
+
+- `LinearTeam`: 线性团队，团队中每个成员会依次被激活，每个成员仅接收前一个成员的运行结果。团队在最后一个成员运行结束后自动结束。
 
+```python
+team: LinearTeam = agent1 => agent2 => agent3
+```
 
+- `LeaderTeam`: 领导团队，团队中有一个领导和一个或多个下属，领导负责判断用户意图并切换到相应的下属上执行任务, 每个下属仅接收领导的指令。
 
+```python
+team: LeaderTeam = agent1 <= [agent2, agent3] <= ["负责...", "负责..."]
+```
 
+> [!TIP]
+> 可以通过第二个 `<=` 操作符设置下属 `Agent` 的描述, 也可以省略使用默认描述

src/course_graph/agent/__init__.py

@@ -8,6 +8,5 @@
 from .controller import Controller
 from .types import Result, ContextVariables, Tool
 from .mcp import MCPServer, STDIO, SSE
-from .trace import TraceEvent
-from .utils import trace_callback
-from .teams import Team, RoundTeam, Terminator, TextTerminator, MaxTurnsTerminator
+from .trace import TraceEvent, trace_callback
+from .teams import Team, RoundTeam, Termination, TextMentionTermination, LinearTeam, LeaderTeam

src/course_graph/agent/agent.py

@@ -136,7 +136,7 @@ def add_assistant_message(self, message: str, name: str = None) -> None:
         message = {'content': message, 'role': 'assistant', 'name': name or self.name}
         self.messages.append(message)
 
-    def add_tool_call_message(self, tool_content: str, tool_call_id: str) -> None:
+    def add_tool_call_result_message(self, tool_content: str, tool_call_id: str) -> None:
         """ 添加工具调用记录
 
         Args:
@@ -171,9 +171,7 @@ def add_tools(self, *tools: 'Tool') -> 'Agent':
         for tool in tools:
             self.tools.append(tool["tool"])
             function = tool["function"]
-            function_name = tool.get('function_name', function.__name__)
-            if function_name == '<lambda>':
-                continue
+            function_name = tool['tool']['function']['name']
             self.tool_functions[function_name] = function
             if (r := tool.get('context_variables_parameter_name')) is not None:
                 self.use_context_variables[function_name] = r
@@ -270,7 +268,7 @@ def add_tool_functions(self, *functions: Callable | Awaitable) -> 'Agent':
                             'type': 'object',
                             'properties': properties,
                             'required': required
-                        } if len(properties) != 0 else {}
+                        }
                     },
                 }
             })