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}"),
             ]
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 <vishnu@rootfor.xyz>"]
 license = "MIT"