Update src/RAG_builder.py
Browse files- src/RAG_builder.py +3 -2
src/RAG_builder.py
CHANGED
|
@@ -53,10 +53,11 @@ class ProjectRAGGraph:
|
|
| 53 |
def generate(self, state: GraphState):
|
| 54 |
print("--- GENERATING ---")
|
| 55 |
prompt = ChatPromptTemplate.from_template("""
|
| 56 |
-
You are a professional Project Analyst.
|
| 57 |
Context: {context}
|
| 58 |
Question: {question}
|
| 59 |
-
Answer strictly using the context.
|
|
|
|
| 60 |
""")
|
| 61 |
|
| 62 |
formatted_context = "\n\n".join(d.page_content for d in state["context"])
|
|
|
|
| 53 |
def generate(self, state: GraphState):
|
| 54 |
print("--- GENERATING ---")
|
| 55 |
prompt = ChatPromptTemplate.from_template("""
|
| 56 |
+
You are a professional Project Analyst.
|
| 57 |
Context: {context}
|
| 58 |
Question: {question}
|
| 59 |
+
Answer strictly using the context.
|
| 60 |
+
If unknown, say you don't know.
|
| 61 |
""")
|
| 62 |
|
| 63 |
formatted_context = "\n\n".join(d.page_content for d in state["context"])
|