Update rag.py
Browse files
rag.py
CHANGED
|
@@ -34,14 +34,6 @@ RAG_CHAIN_PROMPT = PromptTemplate(input_variables = ["context", "question"], tem
|
|
| 34 |
client = MongoClient(MONGODB_ATLAS_CLUSTER_URI)
|
| 35 |
collection = client[MONGODB_DB_NAME][MONGODB_COLLECTION_NAME]
|
| 36 |
|
| 37 |
-
config = {
|
| 38 |
-
"chunk_overlap": 150,
|
| 39 |
-
"chunk_size": 1500,
|
| 40 |
-
"k": 3,
|
| 41 |
-
"model_name": "gpt-4-0613",
|
| 42 |
-
"temperature": 0,
|
| 43 |
-
}
|
| 44 |
-
|
| 45 |
def document_loading_splitting():
|
| 46 |
# Document loading
|
| 47 |
docs = []
|
|
|
|
| 34 |
client = MongoClient(MONGODB_ATLAS_CLUSTER_URI)
|
| 35 |
collection = client[MONGODB_DB_NAME][MONGODB_COLLECTION_NAME]
|
| 36 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 37 |
def document_loading_splitting():
|
| 38 |
# Document loading
|
| 39 |
docs = []
|