Merge pull request #207 from TencentCloudADP/feat/support_stop_at_tools

add `stop_at_tool_names` in AgentConfig

Author: Lightblues

tests/agents/test_stop_at_tools.py

@@ -0,0 +1,16 @@
+from utu.agents import SimpleAgent
+from utu.config import ConfigLoader
+
+
+async def main():
+    config = ConfigLoader.load_agent_config("simple/base_search")
+    config.stop_at_tool_names = ["search"]
+    async with SimpleAgent(config=config) as agent:
+        res = await agent.chat_streamed("Search for Gemini 3.")
+        print(f"res.final_output: {res.final_output}")
+
+
+if __name__ == "__main__":
+    import asyncio
+
+    asyncio.run(main())

utu/agents/simple_agent.py

@@ -55,6 +55,8 @@ def __init__(
             self.config.agent.name = name
         if instructions:
             self.config.agent.instructions = instructions
+        if self.config.stop_at_tool_names:
+            tool_use_behavior = StopAtTools(stop_at_tool_names=self.config.stop_at_tool_names)
         self.model = self._get_model(self.config, model)
         self.model_settings = self._get_model_settings(self.config, model_settings)
         self.tools: list[Tool] = tools or []

utu/config/agent_config.py

@@ -68,6 +68,8 @@ class AgentConfig(ConfigBaseModel):
     """Toolkits config"""
     max_turns: int = 50
     """Max turns for simple agent. This param is derived from @openai-agents"""
+    stop_at_tool_names: list[str] | None = None
+    """Stop at tools for simple agent. This param is derived from @openai-agents"""
 
     # orchestra agent config
     planner_model: ModelConfigs = Field(default_factory=ModelConfigs)