Arize Phoenix Project Routing

Route traces from a stock agent and a search agent into separate Phoenix projects using dangerously_using_project.

Demonstrates sending traces from different agents to different Phoenix projects.

arize_phoenix_moving_traces_to_different_projects.py

"""
Arize Phoenix Project Routing
=============================

Demonstrates sending traces from different agents to different Phoenix projects.
"""

import asyncio
import os

from agno.agent import Agent
from agno.db.in_memory import InMemoryDb
from agno.models.openai import OpenAIChat
from agno.tools.websearch import WebSearchTools
from agno.tools.yfinance import YFinanceTools
from openinference.instrumentation import dangerously_using_project
from phoenix.otel import register
from pydantic import BaseModel

# ---------------------------------------------------------------------------
# Setup
# ---------------------------------------------------------------------------
os.environ["PHOENIX_API_KEY"] = os.getenv("PHOENIX_API_KEY")
os.environ["PHOENIX_COLLECTOR_ENDPOINT"] = (
    "https://app.phoenix.arize.com/"  # Add the suffix for your organization
)

# Register a single tracer provider (project name here is the default)
tracer_provider = register(
    project_name="default",
    auto_instrument=True,
)


class StockPrice(BaseModel):
    stock_price: float


class SearchResult(BaseModel):
    summary: str
    sources: list[str]


# ---------------------------------------------------------------------------
# Create Agents
# ---------------------------------------------------------------------------
# Agent 1 - Stock Price Agent
stock_agent = Agent(
    name="Stock Price Agent",
    model=OpenAIChat(id="gpt-4o-mini"),
    tools=[YFinanceTools()],
    db=InMemoryDb(),
    instructions="You are a stock price agent. Answer questions in the style of a stock analyst.",
    session_id="stock_session",
    output_schema=StockPrice,
)

# Agent 2 - Search Agent
search_agent = Agent(
    name="Search Agent",
    model=OpenAIChat(id="gpt-4o-mini"),
    tools=[WebSearchTools()],
    db=InMemoryDb(),
    instructions="You are a search agent. Find and summarize information from the web.",
    session_id="search_session",
    output_schema=SearchResult,
)


# ---------------------------------------------------------------------------
# Run Example
# ---------------------------------------------------------------------------
async def main() -> None:
    # Run stock_agent and send traces to "default" project
    with dangerously_using_project("default"):
        await stock_agent.aprint_response(
            "What is the current price of Tesla?", stream=True
        )

    # Run search_agent and send traces to "Testing-agno" project
    with dangerously_using_project("Testing-agno"):
        await search_agent.aprint_response(
            "What is the latest news about AI?", stream=True
        )


if __name__ == "__main__":
    asyncio.run(main())

Run the Example

1. Set up your virtual environment

   Mac

   uv venv --python 3.12
   source .venv/bin/activate
   

   uv venv --python 3.12
   .venv\Scripts\activate
   
2. Install dependencies

   uv pip install -U agno arize-phoenix ddgs openai openinference-instrumentation-agno yfinance
   
3. Export your API keys

   Mac/Linux

   export OPENAI_API_KEY="your_openai_api_key_here"
   export PHOENIX_API_KEY="your_phoenix_api_key_here"
   

   $Env:OPENAI_API_KEY="your_openai_api_key_here"
   $Env:PHOENIX_API_KEY="your_phoenix_api_key_here"
   
4. Run the example

   Save the code above as arize_phoenix_moving_traces_to_different_projects.py, then run:

   python arize_phoenix_moving_traces_to_different_projects.py
   

Full source: cookbook/observability/arize_phoenix_moving_traces_to_different_projects.py