fix: reset last_code_generated and last_prompt_used on new conversation

pandasai/agent/base.py

@@ -267,6 +267,8 @@ def start_new_conversation(self):
         Clears the previous conversation
         """
         self.clear_memory()
+        self._state.last_code_generated = None
+        self._state.last_prompt_used = None
 
     def _process_query(self, query: str, output_type: Optional[str] = None):
         """Process a user query and return the result."""

tests/unit_tests/agent/test_agent.py

@@ -256,6 +256,14 @@ def test_start_new_conversation(self, sample_df, config):
         memory = agent._state.memory.all()
         assert len(memory) == 0
 
+    def test_start_new_conversation_resets_state(self, sample_df, config):
+        agent = Agent(sample_df, config, memory_size=10)
+        agent._state.last_code_generated = "result = df.head()"
+        agent._state.last_prompt_used = "Show me the first 5 rows"
+        agent.start_new_conversation()
+        assert agent._state.last_code_generated is None
+        assert agent._state.last_prompt_used is None
+
     def test_code_generation_success(self, agent: Agent):
         # Mock the code generator
         agent._code_generator = Mock()