|
from openinference.instrumentation.smolagents import SmolagentsInstrumentor |
|
from phoenix.otel import register |
|
|
|
|
|
register() |
|
SmolagentsInstrumentor().instrument(skip_dep_check=True) |
|
|
|
|
|
from smolagents import ( |
|
CodeAgent, |
|
InferenceClientModel, |
|
ToolCallingAgent, |
|
VisitWebpageTool, |
|
WebSearchTool, |
|
) |
|
|
|
|
|
|
|
model = InferenceClientModel(provider="nebius") |
|
|
|
search_agent = ToolCallingAgent( |
|
tools=[WebSearchTool(), VisitWebpageTool()], |
|
model=model, |
|
name="search_agent", |
|
description="This is an agent that can do web search.", |
|
return_full_result=True, |
|
) |
|
|
|
manager_agent = CodeAgent( |
|
tools=[], |
|
model=model, |
|
managed_agents=[search_agent], |
|
return_full_result=True, |
|
) |
|
run_result = manager_agent.run( |
|
"If the US keeps it 2024 growth rate, how many years would it take for the GDP to double?" |
|
) |
|
print("Here is the token usage for the manager agent", run_result.token_usage) |
|
print("Here are the timing informations for the manager agent:", run_result.timing) |
|
|