File size: 228 Bytes
67a91b0 |
1 2 3 4 5 6 7 8 9 |
from langchain_openai import ChatOpenAI
class OpenAILLM:
def __init__(self, model_name) -> None:
self.oai_llm = ChatOpenAI(model=model_name)
def run(self, query: str):
return self.oai_llm.invoke(query) |