import time; from common2 import *; import safety as S; SYSTEM,TASK=S.SYSTEM,S.TASK
from pydantic_ai import Agent
from pydantic_ai.models.openai import OpenAIChatModel
from pydantic_ai.providers.openai import OpenAIProvider
ag=Agent(OpenAIChatModel(MODEL, provider=OpenAIProvider(base_url=BASE, api_key=KEY)), system_prompt=SYSTEM, model_settings={"temperature":0}, tools=S.TOOLS)
s=time.time(); OUT=ag.run_sync(TASK).output; print("\n=====FINAL=====\n",OUT); print(f"\nELAPSED {time.time()-s:.1f}s")