Refactor Phoenix tracing setup and integrate with OracleBot

app.py

@@ -9,12 +9,9 @@ from langchain_core.runnables import RunnableConfig
 import asyncio
 from typing import cast
 
-# Phoenix imports
-from phoenix.otel import register
-import logging
-
-from agent.agent import get_agent
+from agent.agent import OracleBot
 from agent.config import create_agent_config
+from logging.phoenix import start_phoenix
 
 load_dotenv()
 
@@ -22,18 +19,8 @@ load_dotenv()
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 APP_NAME = "OracleBot"
 
-# Phoenix tracing setup
-def start_phoenix():
-    """Setup Phoenix tracing for the agent."""
-    register(
-        project_name=APP_NAME,
-        auto_instrument=True,
-    )
-    logging.getLogger("openinference").setLevel(logging.CRITICAL)
-    print("Phoenix tracing enabled.")
-
 # Initialize Phoenix (you can comment this out if you don't want tracing)
-start_phoenix()
+start_phoenix(APP_NAME)
 
 # --- Basic Agent Definition ---
 # ----- THIS IS WERE YOU CAN BUILD WHAT YOU WANT ------
@@ -46,7 +33,7 @@ start_phoenix()
 
 class BasicAgent:
     def __init__(self):
-        self.agent = get_agent()
+        self.agent = OracleBot()
 
     async def __call__(self, question: str) -> str:
         print(f"Agent received question: {question}")

logging/phoenix.py

@@ -0,0 +1,19 @@
+"""Phoenix tracing setup for the OracleBot application."""
+
+import logging
+from phoenix.otel import register
+from app import APP_NAME
+
+
+def start_phoenix(project_name: str = APP_NAME) -> None:
+    """Setup Phoenix tracing for the agent.
+    
+    Args:
+        project_name: Name of the project to use for Phoenix tracing
+    """
+    register(
+        project_name=project_name,
+        auto_instrument=True,
+    )
+    logging.getLogger("openinference").setLevel(logging.CRITICAL)
+    print(f"Phoenix tracing enabled for project: {project_name}")