Update rag_llamaindex.py
Browse files- rag_llamaindex.py +1 -14
rag_llamaindex.py
CHANGED
@@ -1,4 +1,4 @@
|
|
1 |
-
import
|
2 |
|
3 |
from llama_hub.youtube_transcript import YoutubeTranscriptReader
|
4 |
from llama_index import download_loader, PromptTemplate
|
@@ -10,19 +10,6 @@ from pathlib import Path
|
|
10 |
from pymongo import MongoClient
|
11 |
from rag_base import BaseRAG
|
12 |
|
13 |
-
#PDF_URL = "https://arxiv.org/pdf/2303.08774.pdf"
|
14 |
-
#WEB_URL = "https://openai.com/research/gpt-4"
|
15 |
-
#YOUTUBE_URL_1 = "https://www.youtube.com/watch?v=--khbXchTeE"
|
16 |
-
#YOUTUBE_URL_2 = "https://www.youtube.com/watch?v=hdhZwyf24mE"
|
17 |
-
|
18 |
-
#MONGODB_ATLAS_CLUSTER_URI = os.environ["MONGODB_ATLAS_CLUSTER_URI"]
|
19 |
-
#MONGODB_DB_NAME = "llamaindex_db"
|
20 |
-
#MONGODB_COLLECTION_NAME = "gpt-4"
|
21 |
-
#MONGODB_INDEX_NAME = "default"
|
22 |
-
|
23 |
-
#logging.basicConfig(stream = sys.stdout, level = logging.INFO)
|
24 |
-
#logging.getLogger().addHandler(logging.StreamHandler(stream = sys.stdout))
|
25 |
-
|
26 |
class LlamaIndexRAG(BaseRAG):
|
27 |
MONGODB_DB_NAME = "llamaindex_db"
|
28 |
|
|
|
1 |
+
import os
|
2 |
|
3 |
from llama_hub.youtube_transcript import YoutubeTranscriptReader
|
4 |
from llama_index import download_loader, PromptTemplate
|
|
|
10 |
from pymongo import MongoClient
|
11 |
from rag_base import BaseRAG
|
12 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
13 |
class LlamaIndexRAG(BaseRAG):
|
14 |
MONGODB_DB_NAME = "llamaindex_db"
|
15 |
|