import time,json; from common import *
from smolagents import ToolCallingAgent, OpenAIServerModel
m=OpenAIServerModel(model_id=MODEL, api_base=BASE, api_key=KEY, temperature=0)
a=ToolCallingAgent(tools=[], model=m, instructions=SYSTEM, max_steps=4)
s=time.time(); out=a.run(TASK); el=time.time()-s
tu=a.monitor.get_total_token_counts()
print("\n=====FINAL=====\n",out); print(f"\nELAPSED {el:.1f}s"); print("USAGE",json.dumps({"prompt_tokens":tu.input_tokens,"completion_tokens":tu.output_tokens}))