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"
|