jeonghin commited on
Commit
c819284
·
1 Parent(s): 9f791c6

Added try catch for different api keys

Browse files
Files changed (1) hide show
  1. app_function.py +24 -8
app_function.py CHANGED
@@ -39,9 +39,17 @@ def get_vectorstore(text_chunks):
39
  Returns:
40
  - FAISS: A FAISS vector store containing the embeddings of the text chunks.
41
  """
42
- embeddings = OpenAIEmbeddings(
43
- openai_api_base="https://openai.vocareum.com/v1",
44
- )
 
 
 
 
 
 
 
 
45
  vectorstore = FAISS.from_texts(texts=text_chunks, embedding=embeddings)
46
  return vectorstore
47
 
@@ -57,10 +65,18 @@ def get_conversation_chain(vectorstore):
57
  Returns:
58
  - ConversationalRetrievalChain: An initialized conversational chain object.
59
  """
60
- llm = ChatOpenAI(
61
- model_name="gpt-4-1106-preview",
62
- openai_api_base="https://openai.vocareum.com/v1",
63
- )
 
 
 
 
 
 
 
 
64
 
65
  memory = ConversationBufferMemory(memory_key="chat_history", return_messages=True)
66
  conversation_chain = ConversationalRetrievalChain.from_llm(
@@ -144,7 +160,7 @@ def chat_sidebar():
144
  if not st.session_state["song_bool"]:
145
 
146
  if st.sidebar.button("Submit"):
147
- df = choice.process_data('data.json')
148
  recommendations = choice.recommendation(
149
  df,
150
  dance_choice=user_danceability,
 
39
  Returns:
40
  - FAISS: A FAISS vector store containing the embeddings of the text chunks.
41
  """
42
+ try:
43
+ embeddings = OpenAIEmbeddings(
44
+ openai_api_base="https://openai.vocareum.com/v1",
45
+ openai_api_key=OPENAI_API_KEY,
46
+ )
47
+ except:
48
+ embeddings = OpenAIEmbeddings(
49
+ openai_api_base="https://openai.vocareum.com/v1",
50
+ openai_api_key=OPENAI_API_KEY2,
51
+ )
52
+
53
  vectorstore = FAISS.from_texts(texts=text_chunks, embedding=embeddings)
54
  return vectorstore
55
 
 
65
  Returns:
66
  - ConversationalRetrievalChain: An initialized conversational chain object.
67
  """
68
+ try:
69
+ llm = ChatOpenAI(
70
+ model_name="gpt-4-1106-preview",
71
+ openai_api_base="https://openai.vocareum.com/v1",
72
+ openai_api_key=OPENAI_API_KEY,
73
+ )
74
+ except:
75
+ llm = ChatOpenAI(
76
+ model_name="gpt-4-1106-preview",
77
+ openai_api_base="https://openai.vocareum.com/v1",
78
+ openai_api_key=OPENAI_API_KEY2,
79
+ )
80
 
81
  memory = ConversationBufferMemory(memory_key="chat_history", return_messages=True)
82
  conversation_chain = ConversationalRetrievalChain.from_llm(
 
160
  if not st.session_state["song_bool"]:
161
 
162
  if st.sidebar.button("Submit"):
163
+ df = choice.process_data("data.json")
164
  recommendations = choice.recommendation(
165
  df,
166
  dance_choice=user_danceability,