fehmikaya commited on
Commit
e2f6806
·
verified ·
1 Parent(s): cd1e6c2

Update ragagent.py

Browse files
Files changed (1) hide show
  1. 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
- embedding_function = HuggingFaceEmbeddings(model_name="all-MiniLM-L6-v2")
 
 
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