import asyncio
from langchain_ollama import OllamaLLM

from langchain_prefect.plugins import RecordLLMCalls

llm = OllamaLLM(model="qwen2.5:latest")


async def record_call_using_LLM_agenerate():
    """async func"""
    await llm.agenerate(
        [
            "What would be a good name for a company that makes colorful socks?",
            "What would be a good name for a company that sells carbonated water?",
        ]
    )


with RecordLLMCalls():
    asyncio.run(record_call_using_LLM_agenerate())