Rauhan commited on
Commit
ffc9104
·
1 Parent(s): eebb03e
Files changed (1) hide show
  1. src/components/ragAgent.py +1 -1
src/components/ragAgent.py CHANGED
@@ -41,7 +41,7 @@ class RAGAgent:
41
  retrieval_mode=RetrievalMode.SPARSE,
42
  sparse_embedding=sparseEmbeddings
43
  )
44
- promptTemplate = ChatPromptTemplate.from_template(readYaml(os.path.join(os.getcwd(), "prompts.yaml").get("ragTemplate")))
45
  retriever = vectorStore.as_retriever(search_kwargs = {"k": 5})
46
  llm = ChatCerebras(
47
  model = config.get("RAGAGENT", "modelName"),
 
41
  retrieval_mode=RetrievalMode.SPARSE,
42
  sparse_embedding=sparseEmbeddings
43
  )
44
+ promptTemplate = ChatPromptTemplate.from_template(readYaml(os.path.join(os.getcwd(), "prompts.yaml")).get("ragTemplate"))
45
  retriever = vectorStore.as_retriever(search_kwargs = {"k": 5})
46
  llm = ChatCerebras(
47
  model = config.get("RAGAGENT", "modelName"),