import asyncio

from mcp_agent.app import MCPApp
from mcp_agent.agents.agent import Agent
from mcp_agent.workflows.llm.augmented_llm_openai import OpenAIAugmentedLLM


app = MCPApp(name="weather_agent")


async def main():
    async with app.run():
        agent = Agent(
            name="weather",
            instruction=(
                "Ты ассистент по погоде. "
                "Если нужны актуальные данные — используй доступные инструменты."
            ),
            server_names=["weather"],
        )

        async with agent:
            llm = await agent.attach_llm(OpenAIAugmentedLLM)
            answer = await llm.generate_str("Какая сейчас точная погода в Берлине?")
            print(answer)


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