Spaces:
Configuration error
Configuration error
Update agent.py
Browse files
agent.py
CHANGED
|
@@ -15,12 +15,12 @@ MODEL_ID = "o4-mini"
|
|
| 15 |
PHOENIX_CLIENT_HEADERS = "api_key = " + os.environ["DESCRIPTION"]
|
| 16 |
PHOENIX_COLLECTOR_ENDPOINT = "https://app.phoenix.arize.com"
|
| 17 |
|
| 18 |
-
tracer_provider = register(
|
| 19 |
-
project_name = "default",
|
| 20 |
-
endpoint = "https://app.phoenix.arize.com/v1/traces"
|
| 21 |
-
)
|
| 22 |
-
|
| 23 |
def run_agent(question, file_name):
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 24 |
agent = CodeAgent(
|
| 25 |
add_base_tools = True,
|
| 26 |
additional_authorized_imports = ["pandas", "numpy"],
|
|
@@ -31,7 +31,7 @@ def run_agent(question, file_name):
|
|
| 31 |
WikipediaSearchTool()],
|
| 32 |
verbosity_level = 1
|
| 33 |
)
|
| 34 |
-
|
| 35 |
return agent.run(question)
|
| 36 |
|
| 37 |
#answer = agent.run(question)
|
|
|
|
| 15 |
PHOENIX_CLIENT_HEADERS = "api_key = " + os.environ["DESCRIPTION"]
|
| 16 |
PHOENIX_COLLECTOR_ENDPOINT = "https://app.phoenix.arize.com"
|
| 17 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 18 |
def run_agent(question, file_name):
|
| 19 |
+
tracer_provider = register(
|
| 20 |
+
project_name = "gaia",
|
| 21 |
+
endpoint = "https://app.phoenix.arize.com/v1/traces"
|
| 22 |
+
)
|
| 23 |
+
|
| 24 |
agent = CodeAgent(
|
| 25 |
add_base_tools = True,
|
| 26 |
additional_authorized_imports = ["pandas", "numpy"],
|
|
|
|
| 31 |
WikipediaSearchTool()],
|
| 32 |
verbosity_level = 1
|
| 33 |
)
|
| 34 |
+
|
| 35 |
return agent.run(question)
|
| 36 |
|
| 37 |
#answer = agent.run(question)
|