Spaces:
Configuration error
Configuration error
Update agent.py
Browse files
agent.py
CHANGED
|
@@ -13,18 +13,18 @@ from smolagents import (
|
|
| 13 |
|
| 14 |
MODEL_ID = "o4-mini"
|
| 15 |
|
| 16 |
-
PHOENIX_CLIENT_HEADERS = "api_key = " + os.environ["
|
| 17 |
PHOENIX_COLLECTOR_ENDPOINT = "https://app.phoenix.arize.com"
|
| 18 |
|
| 19 |
-
|
| 20 |
-
|
| 21 |
-
|
| 22 |
-
|
| 23 |
-
)
|
| 24 |
|
| 25 |
-
|
| 26 |
-
|
| 27 |
-
|
|
|
|
| 28 |
agent = CodeAgent(
|
| 29 |
add_base_tools = True,
|
| 30 |
additional_authorized_imports = ["pandas", "numpy"],
|
|
|
|
| 13 |
|
| 14 |
MODEL_ID = "o4-mini"
|
| 15 |
|
| 16 |
+
PHOENIX_CLIENT_HEADERS = "api_key = " + os.environ["PHOENIX_API_KEY"]
|
| 17 |
PHOENIX_COLLECTOR_ENDPOINT = "https://app.phoenix.arize.com"
|
| 18 |
|
| 19 |
+
tracer_provider = register(
|
| 20 |
+
project_name = "gaia",
|
| 21 |
+
endpoint = "https://app.phoenix.arize.com/v1/traces"
|
| 22 |
+
)
|
|
|
|
| 23 |
|
| 24 |
+
#OpenAIInstrumentor().instrument(tracer_provider = tracer_provider)
|
| 25 |
+
SmolagentsInstrumentor().instrument(tracer_provider = tracer_provider)
|
| 26 |
+
|
| 27 |
+
def run_agent(question, file_name):
|
| 28 |
agent = CodeAgent(
|
| 29 |
add_base_tools = True,
|
| 30 |
additional_authorized_imports = ["pandas", "numpy"],
|