fix: summary prompting

Author: neagualexa

src/agents/base_agent/base_agent.py

@@ -120,12 +120,12 @@ def summarize_conversation(self, state: State, config: RunnableConfig) -> dict:
             conversationalStyle_message = self.conversation_preference_prompt
 
         # STEP 1: Summarize the conversation
-        messages = [SystemMessage(content=summary_message)] + state["messages"][:-1]
+        messages = state["messages"][:-1] + [HumanMessage(content=summary_message)]
         valid_messages = self.check_for_valid_messages(messages)
         summary_response = self.summarisation_llm.invoke(valid_messages)
 
         # STEP 2: Analyze the conversational style
-        messages = [SystemMessage(content=conversationalStyle_message)] + state["messages"][:-1]
+        messages = state["messages"][:-1] + [HumanMessage(content=conversationalStyle_message)]
         valid_messages = self.check_for_valid_messages(messages)
         conversationalStyle_response = self.summarisation_llm.invoke(valid_messages)
 

src/agents/base_agent/base_prompts.py

@@ -68,11 +68,13 @@
 Structured: Organize the summary into sections such as 'Topics Discussed' and 'Top 3 Key Detailed Ideas'.
 Neutral and Accurate: Avoid adding interpretations or opinions; focus only on the content shared.
 When summarizing: If the conversation is technical, highlight significant concepts, solutions, and terminology. If context involves problem-solving, detail the problem and the steps or solutions provided. If the user asks for creative input, briefly describe the ideas presented.
-Last messages: Include the most recent 4 messages to provide context for the summary.
+Last messages: Include the most recent 5 messages to provide context for the summary.
 
 Provide the summary in a bulleted format for clarity. Avoid redundant details while preserving the core intent of the discussion."""
 
-summary_prompt = f"""Summarize the conversation between a student and a tutor. Your summary should highlight the major topics discussed during the session, followed by a detailed recollection of the last five significant points or ideas. Ensure the summary flows smoothly to maintain the continuity of the discussion."""
+summary_prompt = f"""Summarize the conversation between a student and a tutor. Your summary should highlight the major topics discussed during the session, followed by a detailed recollection of the last five significant points or ideas. Ensure the summary flows smoothly to maintain the continuity of the discussion.
+
+{summary_guidelines}"""
 
 update_summary_prompt = f"""Update the summary by taking into account the new messages above.
 

src/agents/google_learnLM_agent/google_learnLM_agent.py

@@ -124,12 +124,12 @@ def summarize_conversation(self, state: State, config: RunnableConfig) -> dict:
             conversationalStyle_message = self.conversation_preference_prompt
 
         # STEP 1: Summarize the conversation
-        messages = [SystemMessage(content=summary_message)] + state["messages"][:-1]
+        messages = state["messages"][:-1] + [HumanMessage(content=summary_message)]
         valid_messages = self.check_for_valid_messages(messages)
         summary_response = self.summarisation_llm.invoke(valid_messages)
 
         # STEP 2: Analyze the conversational style
-        messages = [SystemMessage(content=conversationalStyle_message)] + state["messages"][:-1]
+        messages = state["messages"][:-1] + [HumanMessage(content=conversationalStyle_message)]
         valid_messages = self.check_for_valid_messages(messages)
         conversationalStyle_response = self.summarisation_llm.invoke(valid_messages)
 

src/agents/google_learnLM_agent/google_learnLM_prompts.py

@@ -68,11 +68,13 @@
 Structured: Organize the summary into sections such as 'Topics Discussed' and 'Top 3 Key Detailed Ideas'.
 Neutral and Accurate: Avoid adding interpretations or opinions; focus only on the content shared.
 When summarizing: If the conversation is technical, highlight significant concepts, solutions, and terminology. If context involves problem-solving, detail the problem and the steps or solutions provided. If the user asks for creative input, briefly describe the ideas presented.
-Last messages: Include the most recent 4 messages to provide context for the summary.
+Last messages: Include the most recent 5 messages to provide context for the summary.
 
 Provide the summary in a bulleted format for clarity. Avoid redundant details while preserving the core intent of the discussion."""
 
-summary_prompt = f"""Summarize the conversation between a student and a tutor. Your summary should highlight the major topics discussed during the session, followed by a detailed recollection of the last five significant points or ideas. Ensure the summary flows smoothly to maintain the continuity of the discussion."""
+summary_prompt = f"""Summarize the conversation between a student and a tutor. Your summary should highlight the major topics discussed during the session, followed by a detailed recollection of the last five significant points or ideas. Ensure the summary flows smoothly to maintain the continuity of the discussion.
+
+{summary_guidelines}"""
 
 update_summary_prompt = f"""Update the summary by taking into account the new messages above.
 

src/agents/informational_agent/informational_agent.py

@@ -122,13 +122,13 @@ def summarize_conversation(self, state: State, config: RunnableConfig) -> dict:
             conversationalStyle_message = self.conversation_preference_prompt
 
         # STEP 1: Summarize the conversation
-        messages = [SystemMessage(content=summary_message)] + state["messages"][:-1]
+        messages = state["messages"][:-1] + [HumanMessage(content=summary_message)]
         valid_messages = self.check_for_valid_messages(messages)
         print(f"valid_messages SUMMARY:: {valid_messages}")
         summary_response = self.summarisation_llm.invoke(valid_messages)
 
         # STEP 2: Analyze the conversational style
-        messages = [SystemMessage(content=conversationalStyle_message)] + state["messages"][:-1]
+        messages = state["messages"][:-1] + [HumanMessage(content=conversationalStyle_message)]
         valid_messages = self.check_for_valid_messages(messages)
         conversationalStyle_response = self.summarisation_llm.invoke(valid_messages)
 
@@ -194,6 +194,8 @@ def invoke_informational_agent(query: str, conversation_history: list, summary:
     summary = agent.get_summary()
     conversationalStyle = agent.get_conversational_style()
 
+    print(f'SUMMARY: {summary}')
+
     return {
         "input": query,
         "output": pretty_printed_response,

src/agents/socratic_agent/socratic_agent.py

@@ -122,12 +122,12 @@ def summarize_conversation(self, state: State, config: RunnableConfig) -> dict:
         #     conversationalStyle_message = self.conversation_preference_prompt
 
         # STEP 1: Summarize the conversation
-        messages = [SystemMessage(content=summary_message)] + state["messages"][:-1]
+        messages = state["messages"][:-1] + [HumanMessage(content=summary_message)]
         valid_messages = self.check_for_valid_messages(messages)
         summary_response = self.summarisation_llm.invoke(valid_messages)
 
         # STEP 2: Analyze the conversational style
-        # messages = [SystemMessage(content=conversationalStyle_message)] + state["messages"][:-1]
+        # messages = state["messages"][:-1] + [HumanMessage(content=conversationalStyle_message)]
         # valid_messages = self.check_for_valid_messages(messages)
         # conversationalStyle_response = self.summarisation_llm.invoke(valid_messages)
 

src/agents/socratic_agent/socratic_prompts.py

@@ -92,11 +92,13 @@
 Structured: Organize the summary into sections such as 'Topics Discussed' and 'Top 3 Key Detailed Ideas'.
 Neutral and Accurate: Avoid adding interpretations or opinions; focus only on the content shared.
 When summarizing: If the conversation is technical, highlight significant concepts, solutions, and terminology. If context involves problem-solving, detail the problem and the steps or solutions provided. If the user asks for creative input, briefly describe the ideas presented.
-Last messages: Include the most recent 4 messages to provide context for the summary.
+Last messages: Include the most recent 5 messages to provide context for the summary.
 
 Provide the summary in a bulleted format for clarity. Avoid redundant details while preserving the core intent of the discussion."""
 
-summary_prompt = f"""Summarize the conversation between a student and a tutor. Your summary should highlight the major topics discussed during the session, followed by a detailed recollection of the last five significant points or ideas. Ensure the summary flows smoothly to maintain the continuity of the discussion."""
+summary_prompt = f"""Summarize the conversation between a student and a tutor. Your summary should highlight the major topics discussed during the session, followed by a detailed recollection of the last five significant points or ideas. Ensure the summary flows smoothly to maintain the continuity of the discussion.
+
+{summary_guidelines}"""
 
 update_summary_prompt = f"""Update the summary by taking into account the new messages above.