Update agent.py

agent.py

@@ -43,12 +43,12 @@ def build_graph():
             "- Never justify or explain"
         )
 
-        # prompt = [
-        #     {"role": "system", "content": system_prompt},
-        #     {"role": "user", "content": query},
-        # ]
+        prompt = [
+            {"role": "system", "content": system_prompt},
+            {"role": "user", "content": query},
+        ]
         
-        prompt = f"{system_prompt}\n\nUser: {query}\nAssistant:"
+        # prompt = f"{system_prompt}\n\nUser: {query}\nAssistant:"
         response = llm.invoke(prompt).strip()
 
         # response = llm.invoke(messages)