bstraehle commited on
Commit
98995e1
1 Parent(s): bcd5ded

Update rag_llamaindex.py

Browse files
Files changed (1) hide show
  1. rag_llamaindex.py +1 -14
rag_llamaindex.py CHANGED
@@ -1,4 +1,4 @@
1
- import logging, os, sys
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