paper-qa-rag / script /llm.py
whosouravsharma's picture
added config setup
66c529e
Raw History Blame Contribute Delete
207 Bytes
# model setup
import os
from langchain_openai import ChatOpenAI
from dotenv import load_dotenv
load_dotenv()
def get_llm():
return ChatOpenAI(
model="gpt-4o-mini",
temperature=0
)