Update app.py
Browse files
app.py
CHANGED
|
@@ -40,7 +40,7 @@ YOUTUBE_URL_3 = "https://www.youtube.com/watch?v=vw-KWfKwvTQ"
|
|
| 40 |
|
| 41 |
MODEL_NAME = "gpt-4"
|
| 42 |
|
| 43 |
-
def document_retrieval_chroma(llm):
|
| 44 |
vector_db = Chroma(embedding_function = OpenAIEmbeddings(),
|
| 45 |
persist_directory = CHROMA_DIR)
|
| 46 |
rag_chain = RetrievalQA.from_chain_type(llm,
|
|
|
|
| 40 |
|
| 41 |
MODEL_NAME = "gpt-4"
|
| 42 |
|
| 43 |
+
def document_retrieval_chroma(llm, prompt):
|
| 44 |
vector_db = Chroma(embedding_function = OpenAIEmbeddings(),
|
| 45 |
persist_directory = CHROMA_DIR)
|
| 46 |
rag_chain = RetrievalQA.from_chain_type(llm,
|