Update rag.py

rag.py

@@ -13,8 +13,6 @@ from langchain.chains import ConversationalRetrievalChain
 import os
 import csv
 
-import api_key
-
 def read_csv_to_dict(filename):
     data_dict = {}
     with open(filename, mode='r', encoding='utf-8') as file:
@@ -44,8 +42,7 @@ def generate_embeddings_and_vectorstore(path):
         #print('docs')
         #print(docs)
 
-        openai_api_key = api_key.OPENAI_KEY
-        fc_embeddings = OpenAIEmbeddings(openai_api_key=openai_api_key)
+        fc_embeddings = OpenAIEmbeddings(openai_api_key=OPENAI_KEY)
         vectorstore = Chroma.from_documents(docs, fc_embeddings)
 
         return vectorstore
@@ -74,7 +71,7 @@ class Rag:
         self.qa = ConversationalRetrievalChain.from_llm(
                     llm=ChatOpenAI(model_name="gpt-3.5-turbo",
                                    temperature=0.3,
-                                   openai_api_key=api_key.OPENAI_KEY,
+                                   openai_api_key=OPENAI_KEY,
                                    max_tokens=int(int(max_words) + (int(max_words) / 2))), #número máximo de tokens para a resposta
                     memory=self.memory,
                     retriever=vectorstore.as_retriever(), #search_kwargs={'k': 3}