Introduction & Context

Welcome back! In the previous lesson, you mastered building agentic pipelines in which specialized agents work together in a fixed sequence. Today, we're taking a significant architectural leap forward by learning how to build agent orchestration systems, where a central planner agent can dynamically decide which specialist agents to call based on the specific task at hand.

The key insight is that complex agentic systems can be wrapped as simple tools for other agents. This creates a powerful hierarchy in which agents can leverage the full capabilities of other agents as easily as they use basic functions, enabling much more flexible and intelligent problem-solving than fixed pipeline sequences.

The Agent-as-Tool Concept

The fundamental difference between pipelines and orchestration lies in decision-making authority. In a pipeline, you, as the developer, decide the sequence: every problem goes through analysis, then calculation, then presentation. In orchestration, the central agent makes these decisions dynamically based on the nature of each specific request. The key insight is that we can encapsulate an agent call as a tool, allowing an orchestrator agent with access to that tool to seamlessly call another specialist agent.

Here's how the agent orchestration flow works:

  1. User submits a request to the orchestrator agent.
  2. Orchestrator analyzes the request and decides whether to handle it directly or delegate to a specialist.
  3. If delegation is needed, the orchestrator calls the specialist agent as a tool.
  4. Specialist processes the request and returns results to the orchestrator.
  5. Orchestrator provides the final response to the user.

Consider how this changes the system's behavior. If someone asks, "What is the capital of France?", a pipeline system would still run the question through all three agents unnecessarily. An orchestrated system, however, allows the central agent to recognize that this is a straightforward knowledge question requiring no mathematical tools and respond directly. But when someone asks about solving equations, the orchestrator can intelligently delegate this to a calculator specialist. This dynamic delegation creates systems that are both more efficient and more capable.

Building Specialist Agents

Before we can orchestrate agents, we need to create the specialist agents that will serve as tools. Let's build a calculator assistant that specializes in mathematical problem-solving, designed to work as a standalone tool that can handle mathematical questions from start to finish.

This calculator_assistant is designed as a complete mathematical problem-solver with access to all the mathematical tools like sum_numbers, multiply_numbers, subtract_numbers, divide_numbers, power, and square_root. The system_prompt establishes the agent's expertise in mathematics while allowing it the flexibility to handle various types of mathematical problems. This balance is important for specialist agents — they need clear domain focus while maintaining enough flexibility to be useful as tools for different orchestrators.

Creating Agent Tool Wrappers

To enable agents to use other agents as tools, we need to wrap any agent into a callable function that follows the same interface as our other tools. This helper function create_agent_tool will streamline our workflow: we'll call it with our specialist agent to get back both a callable lambda function and its schema, which we can then pass to our orchestrator agent, enabling it to call the specialist agent just like any other tool.

The key part of this function is the lambda that wraps the agent call. When we call agent.run, it returns two things: the complete message history and the final text response. We use _history to ignore the message history because when one agent calls another as a tool, it only needs the final answer — not the entire conversation history with all the intermediate function calls and reasoning steps.

For example, if our orchestrator asks the calculator agent to solve an equation, it just wants the final answer like x = 3 and x = 2, not the detailed trace of every mathematical function that was called along the way. This keeps the tool interface clean and focused on results.

Implementing the Orchestrator Agent

With our specialist agent ready and our wrapper function defined, we can now create the agent tool wrapper and build our orchestrator agent. The orchestrator will be a general-purpose assistant that can handle various types of questions, using the calculator_assistant when mathematical expertise is needed.

The orchestrator agent (helpful_assistant) has a deliberately general system_prompt that doesn't limit it to any specific domain. Instead, it's designed to be helpful across various tasks while knowing it has access to mathematical expertise through the calculator tool. Notice how we pass the calculator tool to the orchestrator just like any other tool — from the orchestrator's perspective, calling another agent is no different from calling a mathematical function.

The tools hash maps the tool name (from the schema) to the actual callable lambda, while tool_schemas provides the array of schemas that describe available tools to the model.

Testing General Knowledge Questions

Let's test our orchestrated system with a general knowledge question to see how the orchestrator handles tasks that don't require specialist agents.

When we run this code, the orchestrator agent will receive the question and analyze whether it needs specialist assistance. Since this is a straightforward factual question about geography, the agent should be able to answer directly from its general knowledge without calling any tools.

Here's what happens when we execute this request:

The orchestrator correctly identified this as a question it could handle directly using its general knowledge without calling any tools. The response was immediate and appropriate, demonstrating the system's efficiency for simple tasks that don't require specialist assistance.

Testing Mathematical Problem Delegation

Now let's test the system with a mathematical question that requires the specialist agent's expertise.

In this example, the orchestrator should recognize that solving a quadratic equation requires mathematical expertise beyond basic knowledge. The agent will likely delegate this task to the calculator_assistant_agent tool, which will then use its mathematical functions to solve the equation step by step.

Here's an example of the execution showing the delegation and calculation process:

For the mathematical equation x² - 5x + 6 = 0, the orchestrator recognized that this required mathematical expertise and delegated the task to the calculator_assistant. The debug output (from our Agent class's puts statements) shows:

  1. The orchestrator calling with the mathematical question.
Summary & Exercises

You've now learned how to build agent orchestration systems in which a central planner can dynamically delegate tasks to specialist agents. This architectural pattern offers significant advantages over fixed pipelines: it's more efficient for simple tasks, more flexible for complex problems, and allows you to build increasingly sophisticated specialist agents without complicating the overall system design.

In your upcoming practice exercises, you'll build your own orchestrated agent systems with multiple specialists for different domains. This foundation will enable you to create much more sophisticated and flexible agentic systems that can adapt their approach based on the specific requirements of each task.

Sign up
Join the 1M+ learners on CodeSignal
Be a part of our community of 1M+ users who develop and demonstrate their skills on CodeSignal