bstraehle commited on
Commit
bcd5ded
·
1 Parent(s): 8b439aa

Update rag_langchain.py

Browse files
Files changed (1) hide show
  1. rag_langchain.py +2 -14
rag_langchain.py CHANGED
@@ -1,4 +1,4 @@
1
- import logging, os, sys
2
 
3
  from langchain.callbacks import get_openai_callback
4
  from langchain.chains import LLMChain, RetrievalQA
@@ -14,19 +14,7 @@ from langchain.vectorstores import Chroma
14
  from langchain.vectorstores import MongoDBAtlasVectorSearch
15
 
16
  from pymongo import MongoClient
17
-
18
- #PDF_URL = "https://arxiv.org/pdf/2303.08774.pdf"
19
- #WEB_URL = "https://openai.com/research/gpt-4"
20
- #YOUTUBE_URL_1 = "https://www.youtube.com/watch?v=--khbXchTeE"
21
- #YOUTUBE_URL_2 = "https://www.youtube.com/watch?v=hdhZwyf24mE"
22
-
23
- #MONGODB_ATLAS_CLUSTER_URI = os.environ["MONGODB_ATLAS_CLUSTER_URI"]
24
- #MONGODB_DB_NAME = "langchain_db"
25
- #MONGODB_COLLECTION_NAME = "gpt-4"
26
- #MONGODB_INDEX_NAME = "default"
27
-
28
- #logging.basicConfig(stream = sys.stdout, level = logging.INFO)
29
- #logging.getLogger().addHandler(logging.StreamHandler(stream = sys.stdout))
30
 
31
  class LangChainRAG(BaseRAG):
32
  MONGODB_DB_NAME = "langchain_db"
 
1
+ import os
2
 
3
  from langchain.callbacks import get_openai_callback
4
  from langchain.chains import LLMChain, RetrievalQA
 
14
  from langchain.vectorstores import MongoDBAtlasVectorSearch
15
 
16
  from pymongo import MongoClient
17
+ from rag_base import BaseRAG
 
 
 
 
 
 
 
 
 
 
 
 
18
 
19
  class LangChainRAG(BaseRAG):
20
  MONGODB_DB_NAME = "langchain_db"