From f07581840cf9480147645acb9e8637fef3de8988 Mon Sep 17 00:00:00 2001 From: vizsatiz Date: Mon, 2 Sep 2024 14:45:24 +0530 Subject: [PATCH 1/2] Fix for context in RAG builder --- flo_ai/retrievers/flo_retriever.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/flo_ai/retrievers/flo_retriever.py b/flo_ai/retrievers/flo_retriever.py index eba610f8..7a95c315 100644 --- a/flo_ai/retrievers/flo_retriever.py +++ b/flo_ai/retrievers/flo_retriever.py @@ -21,7 +21,12 @@ def __init__(self, session: FloSession, retriever: VectorStoreRetriever) -> None ("system", """You are an assistant for question-answering tasks. Use the following pieces of retrieved context to answer the question. If you don't know the answer, just say that you don't know. - Use three sentences maximum and keep the answer concise."""), + Use three sentences maximum and keep the answer concise. + + Here is the context: + {context} + + """), MessagesPlaceholder(variable_name="chat_history"), ("human", "{question}"), ] From 0980d9332d11f9e3ed6de1da420e54a1df298179 Mon Sep 17 00:00:00 2001 From: vizsatiz Date: Mon, 2 Sep 2024 14:48:42 +0530 Subject: [PATCH 2/2] Updating version --- pyproject.toml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pyproject.toml b/pyproject.toml index e7456b7d..12614ce7 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -1,6 +1,6 @@ [tool.poetry] name = "flo-ai" -version = "0.0.2" +version = "0.0.2.1" description = "A easy way to create structured AI agents" authors = ["vizsatiz "] license = "MIT"