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

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

diff --git a/ingest.py b/ingest.py
index 2a9a161..79434b3 100644
--- a/ingest.py
+++ b/ingest.py
@@ -1,4 +1,5 @@
 import os
+from dotenv import load_dotenv
 from langchain.document_loaders import TextLoader, PDFMinerLoader, CSVLoader
 from langchain.text_splitter import RecursiveCharacterTextSplitter
 from langchain.vectorstores import Chroma
@@ -6,6 +7,8 @@ from langchain.embeddings import LlamaCppEmbeddings
 from constants import PERSIST_DIRECTORY
 from constants import CHROMA_SETTINGS
 
+load_dotenv()
+
 def main():
     llama_embeddings_model = os.environ.get('LLAMA_EMBEDDINGS_MODEL')
     persist_directory = os.environ.get('PERSIST_DIRECTORY')
@@ -30,4 +33,4 @@ def main():
     db = None
 
 if __name__ == "__main__":
-    main()
\ No newline at end of file
+    main()