add key in ChatOpenAI

Author: ibizabroker

__pycache__/chain.cpython-311.pyc

@@ -19,7 +19,7 @@ def load_chain(openai_api_key, pinecone_api_key, pinecone_env_key):
 
   db = Pinecone.from_existing_index(index_name, embeddings)
 
-  return get_conversation_chain(db)
+  return get_conversation_chain(db, openai_api_key)
 
 def execute_chain(query, openai_api_key, pinecone_api_key, pinecone_env_key):
   chain = load_chain(openai_api_key, pinecone_api_key, pinecone_env_key)
@@ -88,7 +88,7 @@ def main():
       with st.spinner("Processing"):
         vector_store = create_vector_db()
 
-        st.session_state.conversation = get_conversation_chain(vector_store)
+        st.session_state.conversation = get_conversation_chain(vector_store, st.session_state["OPENAI_API_KEY"])
         st.write("Processing Done")
 
   st.title("PDF Bot")

bot.py

@@ -3,8 +3,9 @@
 from langchain.chains import ConversationalRetrievalChain
 from langchain.prompts.prompt import PromptTemplate
 
-def get_conversation_chain(vector_store):
+def get_conversation_chain(vector_store, openai_api_key):
   llm = ChatOpenAI(
+    openai_api_key=openai_api_key,
     model="gpt-3.5-turbo",
     temperature=0.7
   )