import time,json; from common import *
from langchain_openai import ChatOpenAI
from langgraph.prebuilt import create_react_agent
llm=ChatOpenAI(model=MODEL, base_url=BASE, api_key=KEY, temperature=0)
ag=create_react_agent(llm, tools=[], prompt=SYSTEM)
s=time.time(); r=ag.invoke({"messages":[("user",TASK)]}); el=time.time()-s
p=c=0
for msg in r["messages"]:
    um=getattr(msg,"usage_metadata",None)
    if um: p+=um["input_tokens"]; c+=um["output_tokens"]
print("\n=====FINAL=====\n",r["messages"][-1].content); print(f"\nELAPSED {el:.1f}s"); print("USAGE",json.dumps({"prompt_tokens":p,"completion_tokens":c}))