import time; from common2 import *; import safety as S; SYSTEM,TASK=S.SYSTEM,S.TASK
from langchain_openai import ChatOpenAI
from langchain_core.tools import tool
from langgraph.prebuilt import create_react_agent
ag=create_react_agent(ChatOpenAI(model=MODEL, base_url=BASE, api_key=KEY, temperature=0), tools=[tool(f) for f in S.TOOLS], prompt=SYSTEM)
s=time.time(); OUT=ag.invoke({"messages":[("user",TASK)]},{"recursion_limit":60})["messages"][-1].content; print("\n=====FINAL=====\n",OUT); print(f"\nELAPSED {time.time()-s:.1f}s")