import asyncio

from dotenv import load_dotenv

# from pydantic import BaseModel
from pydantic_ai import Agent
from pydantic_ai.settings import ModelSettings

from llm.config import load_config
from llm.factory import LLMModelProvider

load_dotenv()


async def main():


    config = load_config("config.yml")
    model = await LLMModelProvider.create_from_config(config.llm)


    # model = await LLMModelProvider.create_codex(model_name='gpt-5.6-luna')
    # model = LLMModelProvider.create_openai_compatible(model_name="qwen/qwen3-8b",
    #                                                   api_key=os.getenv("API_KEY"),
    #                                                   base_url=os.getenv("BASE_URL"))
    agent = Agent(model=model,
                  system_prompt=config.prompts.system,
                  instructions=config.prompts.instructions)
    settings = ModelSettings(
        **config.generation.model_dump()
    )

    user_prompt = config.prompts.user.format(task="Повтори сообщение")
    result = await agent.run(user_prompt,
                             model_settings=settings)
    print(result.output)


if __name__ == "__main__":
    asyncio.run(main())