Dinesh310 commited on
Commit
4e52498
·
verified ·
1 Parent(s): dfcca4a

Update src/RAG_builder.py

Browse files
Files changed (1) hide show
  1. 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. Cite sources.
 
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"])