Optional: Code Interpreter

def build_specialists(gateway: MCPClient, results: list[str]) -> list:
    """The data and weather specialists as tools; each searches the Gateway for its tools, then records its answer."""

    @tool
    def data_agent(query: str) -> str:
        """Query and analyze the DataStream Corp database (employees, departments, projects).

        Args:
            query: A database or data analysis question.
        """
        agent = Agent(
            model=make_model(),
            system_prompt=f"{DATA_PROMPT}\n{guardrail.prompt_hint()}",
            tools=search_tools(gateway, DATA_TOOLS),
            hooks=[ReadOnlyGuardHook(), *guardrail.hooks()],
            callback_handler=None,
        )
        return consult(agent, query, "data")

    @tool
    def weather_agent(query: str) -> str:
        """Get the weather forecast for a US location (National Weather Service); says so for non-US places.

        Args:
            query: A weather-related question, including the place.
        """
        agent = Agent(
            model=make_model(),
            system_prompt=nws_weather.WEATHER_PROMPT,
            hooks=guardrail.hooks(),
            tools=search_tools(gateway, WEATHER_TOOLS),
            callback_handler=None,
        )
        return consult(agent, query, "weather")

    @tool
    def analysis_agent(question: str) -> str:
        """Analyse DataStream data with Python: statistics, distributions, trends. Use beyond one SQL answer.

        Args:
            question: The analysis to perform.
        """
        from sandbox_tools import run_python

        agent = Agent(
            model=make_model(),
            system_prompt=(
                "You are a data analyst. First fetch the rows you need with the database tool, "
                "then compute with run_python (print every result). Report numbers, not code."
            ),
            tools=[*search_tools(gateway, DATA_TOOLS), run_python],
            hooks=[ReadOnlyGuardHook(), *guardrail.hooks()],
            callback_handler=None,
        )
        return consult(agent, question, "analysis")

    def consult(agent: Agent, query: str, specialist: str) -> str:
        """Ask a specialist; an exception (e.g. the gateway's 502 after its retries) becomes an explicit failure."""
        try:
            answer = str(agent(query))
        except Exception as error:
            answer = SPECIALIST_FAILED.format(specialist=specialist, reason=f"{type(error).__name__}: {error}")
        return remember(answer, specialist)

    def remember(answer: str, specialist: str = "requested") -> str:
        """Record a specialist's answer for `final_answer`; an empty one is recorded as an explicit failure."""
        answer = answer.strip() or SPECIALIST_FAILED.format(specialist=specialist, reason="it returned an empty answer")
        results.append(answer)
        return answer

    return [data_agent, weather_agent, analysis_agent]