Spaces:
Sleeping
Sleeping
Update ragagent.py
Browse files- ragagent.py +9 -3
ragagent.py
CHANGED
@@ -1,8 +1,12 @@
|
|
1 |
|
2 |
### RAG Agent with Langchain and Langgraph, Hallucination and Sanity Checks with Websearch
|
3 |
|
4 |
-
from langchain_chroma import Chroma
|
5 |
-
from langchain_huggingface import HuggingFaceEmbeddings
|
|
|
|
|
|
|
|
|
6 |
from langchain_text_splitters import RecursiveCharacterTextSplitter
|
7 |
from langchain_core.output_parsers import JsonOutputParser
|
8 |
from langchain_core.output_parsers import StrOutputParser
|
@@ -93,7 +97,9 @@ class RAGAgent():
|
|
93 |
chunk_size=512, chunk_overlap=20
|
94 |
)
|
95 |
doc_splits = text_splitter.split_documents(docs_list)
|
96 |
-
|
|
|
|
|
97 |
|
98 |
collection_name = "DEFAULT"
|
99 |
|
|
|
1 |
|
2 |
### RAG Agent with Langchain and Langgraph, Hallucination and Sanity Checks with Websearch
|
3 |
|
4 |
+
# from langchain_chroma import Chroma
|
5 |
+
# from langchain_huggingface import HuggingFaceEmbeddings
|
6 |
+
|
7 |
+
from langchain_community.vectorstores import Chroma
|
8 |
+
from langchain_community.embeddings.sentence_transformer import SentenceTransformerEmbeddings
|
9 |
+
|
10 |
from langchain_text_splitters import RecursiveCharacterTextSplitter
|
11 |
from langchain_core.output_parsers import JsonOutputParser
|
12 |
from langchain_core.output_parsers import StrOutputParser
|
|
|
97 |
chunk_size=512, chunk_overlap=20
|
98 |
)
|
99 |
doc_splits = text_splitter.split_documents(docs_list)
|
100 |
+
|
101 |
+
# embedding_function = HuggingFaceEmbeddings(model_name="all-MiniLM-L6-v2")
|
102 |
+
embedding_function = SentenceTransformerEmbeddings(model_name="all-MiniLM-L6-v2")
|
103 |
|
104 |
collection_name = "DEFAULT"
|
105 |
|