File size: 857 Bytes
974e5e3 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 | import os
import streamlit as st
from langchain_openai import ChatOpenAI
class OpenAILLM:
def __init__(self, user_controls_input=None, model=None, api_key=None):
self.user_controls_input = user_controls_input
self.model = model
self.api_key = api_key
def get_llm_model(self):
try:
if self.user_controls_input:
openai_api_key = self.user_controls_input['OPENAI_API_KEY']
selected_openai_model = self.user_controls_input['selected_openai_model']
llm = ChatOpenAI(api_key=openai_api_key, model= selected_openai_model)
else:
llm = ChatOpenAI(api_key=openai_api_key, model= self.model)
except Exception as e:
raise ValueError(f"Error occured with Exception : {e}")
return llm |