Dinesh310 commited on
Commit
82551b1
·
verified ·
1 Parent(s): 9cc8295

Update src/RAG_builder.py

Browse files
Files changed (1) hide show
  1. src/RAG_builder.py +1 -1
src/RAG_builder.py CHANGED
@@ -19,7 +19,7 @@ class GraphState(TypedDict):
19
  class ProjectRAGGraph:
20
  def __init__(self):
21
  self.embeddings = HuggingFaceEmbeddings(
22
- model_name="Octen/Octen-Embedding-0.6B",
23
  encode_kwargs={"normalize_embeddings": True}
24
  )
25
  self.llm = ChatOpenAI(
 
19
  class ProjectRAGGraph:
20
  def __init__(self):
21
  self.embeddings = HuggingFaceEmbeddings(
22
+ model_name="BAAI/bge-small-en-v1.5",
23
  encode_kwargs={"normalize_embeddings": True}
24
  )
25
  self.llm = ChatOpenAI(