Fix type ignore comment for agent check in get_all_edges function

This commit is contained in:
Martín Bravo 2025-03-25 19:17:35 +01:00
parent 351b6074e5
commit 3068e42029

View file

@ -111,7 +111,7 @@ def get_all_edges(agent: Agent, parent: Optional[Agent] = None) -> str:
"{agent.name}" -> "{handoff.name}";""")
parts.append(get_all_edges(handoff, agent))
if not agent.handoffs and not isinstance(agent, Tool):
if not agent.handoffs and not isinstance(agent, Tool): # type: ignore
parts.append(f'"{agent.name}" -> "__end__";')
return "".join(parts)