Update rag_langchain.py
Browse files- rag_langchain.py +2 -14
rag_langchain.py
CHANGED
@@ -1,4 +1,4 @@
|
|
1 |
-
import
|
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"
|