forked from agent-lab/smolagents
39 lines
1,023 B
Python
39 lines
1,023 B
Python
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 details for the manager agent:", run_result.timing)
|