import time,json; from common import *
from pydantic_ai import Agent
from pydantic_ai.models.openai import OpenAIChatModel
from pydantic_ai.providers.openai import OpenAIProvider
m=OpenAIChatModel(MODEL, provider=OpenAIProvider(base_url=BASE, api_key=KEY))
ag=Agent(m, system_prompt=SYSTEM, model_settings={"temperature":0})
s=time.time(); r=ag.run_sync(TASK); el=time.time()-s
u=r.usage() if callable(r.usage) else r.usage; print("\n=====FINAL=====\n",r.output); print(f"\nELAPSED {el:.1f}s"); print("USAGE",json.dumps({"prompt_tokens":u.input_tokens,"completion_tokens":u.output_tokens}))