File size: 1,028 Bytes
dfd6145 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 |
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,
)
# Then we run the agentic part!
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)
|