Spaces:
Runtime error
Runtime error
UPDATE
Browse files
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"),
|