Update agent.py

agent.py

@@ -12,12 +12,14 @@ from smolagents import (
 
 MODEL_ID = "o4-mini"
 
+PHOENIX_PROJECT_NAME = "gaia"
+
 os.environ["OTEL_EXPORTER_OTLP_HEADERS"] = "api_key = " + os.environ["PHOENIX_API_KEY"];
 os.environ["PHOENIX_CLIENT_HEADERS"] = "api_key = " + os.environ["PHOENIX_API_KEY"];
 os.environ["PHOENIX_COLLECTOR_ENDPOINT"] = "https://app.phoenix.arize.com";
 
 tracer_provider = register(
-    project_name = "gaia",
+    project_name = PHOENIX_PROJECT_NAME,
     endpoint = "https://app.phoenix.arize.com/v1/traces"
 )
 
@@ -26,7 +28,7 @@ SmolagentsInstrumentor().instrument(tracer_provider = tracer_provider)
 def run_agent(question, file_name):   
     agent = CodeAgent(
             add_base_tools = True,
-            additional_authorized_imports = ["pandas", "numpy"],
+            additional_authorized_imports = ["json", "numpy", "pandas", "time"],
             max_steps = 25,
             model = OpenAIServerModel(model_id = MODEL_ID),
             #planning_interval=3,