more fixes

This commit is contained in:
Brandon Hancock
2025-03-13 15:47:02 -04:00
parent 358befe2c1
commit 0a6098fb50

View File

@@ -344,12 +344,11 @@ class Agent(BaseAgent):
def get_delegation_tools(self, agents: Sequence[BaseAgent]) -> Sequence[BaseTool]:
# If delegate_to is specified, use those agents instead of all agents
agents_to_use: List[BaseAgent]
if self.delegate_to is not None:
agents_to_use: Sequence[BaseAgent] = cast(
List[BaseAgent], list(self.delegate_to)
)
agents_to_use = cast(List[BaseAgent], list(self.delegate_to))
else:
agents_to_use: Sequence[BaseAgent] = cast(List[BaseAgent], list(agents))
agents_to_use = list(agents) # Convert to list to match expected type
agent_tools = AgentTools(agents=agents_to_use)
delegation_tools = agent_tools.tools()