src/llms/openai.py

import os
import streamlit as st
from langchain_openai import ChatOpenAI

class OpenAILLM:
  def __init__(self, user_controls_input):
    self.user_controls_input = user_controls_input
    
  def get_llm_model(self):
    try:
      openai_api_key = self.user_controls_input['API Key']
      openai_model = self.user_controls_input['Selected Model']

      if openai_api_key == "":
        st.error('Please enter your OpenAI API key to proceed.')

      llm = ChatOpenAI(model=openai_model, openai_api_key=openai_api_key)
      return llm

    except KeyError as e:
      st.error(f"Missing configuration: {e}")
      return None