Create rag_base.py
Browse files- rag_base.py +14 -0
rag_base.py
ADDED
|
@@ -0,0 +1,14 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
import logging, os, sys
|
| 2 |
+
|
| 3 |
+
logging.basicConfig(stream = sys.stdout, level = logging.INFO)
|
| 4 |
+
logging.getLogger().addHandler(logging.StreamHandler(stream = sys.stdout))
|
| 5 |
+
|
| 6 |
+
class BaseRAG:
|
| 7 |
+
PDF_URL = "https://arxiv.org/pdf/2303.08774.pdf"
|
| 8 |
+
WEB_URL = "https://openai.com/research/gpt-4"
|
| 9 |
+
YOUTUBE_URL_1 = "https://www.youtube.com/watch?v=--khbXchTeE"
|
| 10 |
+
YOUTUBE_URL_2 = "https://www.youtube.com/watch?v=hdhZwyf24mE"
|
| 11 |
+
|
| 12 |
+
MONGODB_ATLAS_CLUSTER_URI = os.environ["MONGODB_ATLAS_CLUSTER_URI"]
|
| 13 |
+
MONGODB_COLLECTION_NAME = "gpt-4"
|
| 14 |
+
MONGODB_INDEX_NAME = "default"
|