Add python-dotenv implementation for .env file usage

Author: Barry Weiss

main.py

@@ -1,4 +1,7 @@
 """Main entrypoint for the app."""
+import os
+from dotenv import load_dotenv
+
 import logging
 import pickle
 from pathlib import Path
@@ -12,6 +15,12 @@
 from query_data import get_chain
 from schemas import ChatResponse
 
+# Load .env
+load_dotenv()
+
+# OpenAI Key for OpenAI API
+openai_api_key = os.environ['OPENAI_API_KEY']
+
 app = FastAPI()
 templates = Jinja2Templates(directory="templates")
 vectorstore: Optional[VectorStore] = None

query_data.py

@@ -1,4 +1,7 @@
 """Create a ChatVectorDBChain for question/answering."""
+import os
+from dotenv import load_dotenv
+
 from langchain.callbacks.base import AsyncCallbackManager
 from langchain.callbacks.tracers import LangChainTracer
 from langchain.chains import ChatVectorDBChain
@@ -9,6 +12,11 @@
 from langchain.llms import OpenAI
 from langchain.vectorstores.base import VectorStore
 
+# Load .env
+load_dotenv()
+
+# OpenAI Key for OpenAI API
+openai_api_key = os.environ['OPENAI_API_KEY']
 
 def get_chain(
     vectorstore: VectorStore, question_handler, stream_handler, tracing: bool = False

requirements.txt

@@ -11,4 +11,5 @@ faiss-cpu
 unstructured
 libmagic
 beautifulsoup4
-tiktoken
+tiktoken
+python-dotenv