From e947ca1d0f727597e70ff03f6d899f171ea7fc4a Mon Sep 17 00:00:00 2001
From: Sorin Neacsu <sorin@users.noreply.github.com>
Date: Thu, 11 May 2023 15:33:56 -0700
Subject: [PATCH] load .env

---
 privateGPT.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/privateGPT.py b/privateGPT.py
index 38fab48..4c603a2 100644
--- a/privateGPT.py
+++ b/privateGPT.py
@@ -1,3 +1,4 @@
+from dotenv import load_dotenv
 from langchain.chains import RetrievalQA
 from langchain.embeddings import LlamaCppEmbeddings
 from langchain.callbacks.streaming_stdout import StreamingStdOutCallbackHandler
@@ -5,6 +6,8 @@ from langchain.vectorstores import Chroma
 from langchain.llms import GPT4All, LlamaCpp
 import os
 
+load_dotenv()
+
 llama_embeddings_model = os.environ.get("LLAMA_EMBEDDINGS_MODEL")
 persist_directory = os.environ.get('PERSIST_DIRECTORY')
 
@@ -51,4 +54,4 @@ def main():
             print(document.page_content)
 
 if __name__ == "__main__":
-    main()
\ No newline at end of file
+    main()