jonny commited on
Commit
bf62ec5
·
1 Parent(s): 6bf0ea4

switched to load api key into OpenAI class instead of setting environment var

Browse files
Files changed (1) hide show
  1. app.py +1 -4
app.py CHANGED
@@ -57,14 +57,11 @@ def loading_pdf():
57
  def loading_database(open_ai_key):
58
  if open_ai_key is not None:
59
  if os.path.exists(PERSIST_DIRECTORY):
60
- os.environ['OPENAI_API_KEY'] = open_ai_key
61
- print("Loading from persisted dir")
62
- embeddings = OpenAIEmbeddings()
63
  docs_retriever = Chroma(persist_directory=PERSIST_DIRECTORY, embedding_function=embeddings)
64
 
65
  global qa_chain
66
  qa_chain = ConversationalRetrievalChain.from_llm(ChatOpenAI(model_name="gpt-3.5-turbo", temperature=0.0),
67
-
68
  retriever=docs_retriever.as_retriever(),
69
  memory=memory,
70
  return_source_documents=False
 
57
  def loading_database(open_ai_key):
58
  if open_ai_key is not None:
59
  if os.path.exists(PERSIST_DIRECTORY):
60
+ embeddings = OpenAIEmbeddings(open_ai_key=open_ai_key)
 
 
61
  docs_retriever = Chroma(persist_directory=PERSIST_DIRECTORY, embedding_function=embeddings)
62
 
63
  global qa_chain
64
  qa_chain = ConversationalRetrievalChain.from_llm(ChatOpenAI(model_name="gpt-3.5-turbo", temperature=0.0),
 
65
  retriever=docs_retriever.as_retriever(),
66
  memory=memory,
67
  return_source_documents=False