Update app.py
Browse files
app.py
CHANGED
@@ -52,6 +52,7 @@ LLM_CHAIN_PROMPT = PromptTemplate(input_variables = ["question"],
|
|
52 |
RAG_CHAIN_PROMPT = PromptTemplate(input_variables = ["context", "question"],
|
53 |
template = rag_template)
|
54 |
|
|
|
55 |
CHROMA_DIR = "/data/chroma"
|
56 |
YOUTUBE_DIR = "/data/youtube"
|
57 |
|
@@ -95,11 +96,7 @@ def document_storage_mongodb(splits):
|
|
95 |
collection = MONGODB_COLLECTION,
|
96 |
index_name = MONGODB_INDEX_NAME)
|
97 |
|
98 |
-
def document_retrieval_chroma(llm, prompt):
|
99 |
-
# Erstelle das Verzeichnis, wenn es nicht existiert
|
100 |
-
if not os.path.exists(CHROMA_DIR):
|
101 |
-
os.makedirs(CHROMA_DIR)
|
102 |
-
|
103 |
db = Chroma(embedding_function = OpenAIEmbeddings(),
|
104 |
persist_directory = CHROMA_DIR)
|
105 |
return db
|
|
|
52 |
RAG_CHAIN_PROMPT = PromptTemplate(input_variables = ["context", "question"],
|
53 |
template = rag_template)
|
54 |
|
55 |
+
#Pfad, wo Docs abgelegt werden können - lokal, also hier im HF Space (sonst auf eigenem Rechner)
|
56 |
CHROMA_DIR = "/data/chroma"
|
57 |
YOUTUBE_DIR = "/data/youtube"
|
58 |
|
|
|
96 |
collection = MONGODB_COLLECTION,
|
97 |
index_name = MONGODB_INDEX_NAME)
|
98 |
|
99 |
+
def document_retrieval_chroma(llm, prompt):
|
|
|
|
|
|
|
|
|
100 |
db = Chroma(embedding_function = OpenAIEmbeddings(),
|
101 |
persist_directory = CHROMA_DIR)
|
102 |
return db
|