"""Example of observing LLM calls made by via callable OpenAI LLM."""

from langchain.llms import OpenAI

from langchain_prefect.plugins import RecordLLMCalls

llm = OpenAI(temperature=0.9)

with RecordLLMCalls():
    llm("What would be a good name for a company that makes colorful socks?")

langchain.llms.OpenAI langchain_prefect.plugins.RecordLLMCalls
[((166, 189), 'langchain.llms.OpenAI', 'OpenAI', ([], {'temperature': '(0.9)'}), '(temperature=0.9)\n', (172, 189), False, 'from langchain.llms import OpenAI\n'), ((196, 212), 'langchain_prefect.plugins.RecordLLMCalls', 'RecordLLMCalls', ([], {}), '()\n', (210, 212), False, 'from langchain_prefect.plugins import RecordLLMCalls\n')]