correct variable name

langraph_agent.py

@@ -127,10 +127,10 @@ class CyberLegalAgent:
                 messages.append(HumanMessage(content=content))
             elif role == "assistant":
                 messages.append(AIMessage(content=content))
-        
+        logger.info(f"📝 Created Messages stack: {messages}")
         if not has_system_message:
             messages.insert(0, SystemMessage(content=SYSTEM_PROMPT))
-        answer_prompt = RESPONSE_FORMATTING_PROMPT.format(context=context, query=state["user_query"])
+        answer_prompt = RESPONSE_FORMATTING_PROMPT.format(content=context, user_query=state["user_query"])
         messages.append(HumanMessage(content=answer_prompt))
         
         response = await self.llm.ainvoke(messages)

prompts.py

@@ -36,8 +36,7 @@ RESPONSE_FORMATTING_PROMPT = """### Task
 We aim at answering the user's question based on your common knowledge as well as the knowledge graph context.
 
 ### Inputs
-**Content to Format:**
-{context}
+**Content to Format:**{context}
 **User Query:** {user_query}
 
 ### Response format