Update rag.py

rag.py

@@ -12,7 +12,8 @@ from langchain.retrievers import MergerRetriever
 
 # Load environment variables from .env file
 load_dotenv()
-
+# Retrieve the OpenAI API key from environment variables
+OPENAI_API_KEY = os.getenv("OPENAI_API_KEY")
 # Configure logging
 logging.basicConfig(
     level=logging.INFO,
@@ -89,7 +90,7 @@ def generate_answer(query: str) -> str:
         llm = ChatOpenAI(
             model="gpt-3.5-turbo",
             temperature=0,
-            openai_api_key="#####"
+            openai_api_key=OPENAI_API_KEY
         )
 
         # Define the prompt template