bstraehle commited on
Commit
676ea08
1 Parent(s): ff3ca13

Create rag_base.py

Browse files
Files changed (1) hide show
  1. 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"