Update ques_2.py

ques_2.py

@@ -6,7 +6,7 @@ from langchain.chat_models import ChatOpenAI
 from langchain.schema.runnable import RunnablePassthrough
 from langchain.prompts import PromptTemplate
 from langchain import hub
-from env import OPENAI_API_KEY
+# from env import OPENAI_API_KEY
 
 def main():
     loader = CSVLoader(file_path="output.csv")
@@ -15,7 +15,7 @@ def main():
     splits = text_splitter.split_documents(data)
     vectorstore = Chroma.from_documents(
         documents=splits,
-        embedding=OpenAIEmbeddings(openai_api_key=OPENAI_API_KEY),
+        embedding=OpenAIEmbeddings(openai_api_key=os.environ['OPENAI_API_KEY']),
     )
     retriever = vectorstore.as_retriever()
     rag_prompt = hub.pull("rlm/rag-prompt")