Update utils.py
Browse files
utils.py
CHANGED
@@ -100,6 +100,9 @@ RAG_CHAIN_PROMPT = PromptTemplate(input_variables = ["context", "question"],
|
|
100 |
PATH_WORK = "."
|
101 |
#CHROMA_DIR = "/chroma"
|
102 |
CHROMA_DIR = "/chroma/britta"
|
|
|
|
|
|
|
103 |
YOUTUBE_DIR = "/youtube"
|
104 |
HISTORY_PFAD = "/data/history"
|
105 |
|
@@ -189,9 +192,9 @@ def document_loading_splitting():
|
|
189 |
docs = []
|
190 |
|
191 |
# kreiere einen DirectoryLoader für jeden file type
|
192 |
-
pdf_loader = create_directory_loader('.pdf',
|
193 |
-
word_loader = create_directory_loader('.word',
|
194 |
-
|
195 |
|
196 |
# Load the files
|
197 |
pdf_documents = pdf_loader.load()
|
|
|
100 |
PATH_WORK = "."
|
101 |
#CHROMA_DIR = "/chroma"
|
102 |
CHROMA_DIR = "/chroma/britta"
|
103 |
+
#CHROMA_PDF = './chroma/pdf'
|
104 |
+
CHROMA_PDF = './chroma/britta/pdf'
|
105 |
+
CHROMA_WORD = './chroma/word'
|
106 |
YOUTUBE_DIR = "/youtube"
|
107 |
HISTORY_PFAD = "/data/history"
|
108 |
|
|
|
192 |
docs = []
|
193 |
|
194 |
# kreiere einen DirectoryLoader für jeden file type
|
195 |
+
pdf_loader = create_directory_loader('.pdf', CHROMA_PDF)
|
196 |
+
word_loader = create_directory_loader('.word', CHROMA_WORD)
|
197 |
+
print("PDF Loader done............................")
|
198 |
|
199 |
# Load the files
|
200 |
pdf_documents = pdf_loader.load()
|