updated context management and steps to 12 because errored at 14th step

app.py

@@ -585,6 +585,46 @@ image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_co
 
 with open("prompts.yaml", 'r') as stream:
     prompt_templates = yaml.safe_load(stream)
+
+# Override CodeAgent run to automatically clear context periodically
+original_code_agent_run = CodeAgent.run
+
+def context_managed_run(self, query, max_steps=None, persist_conversation=None):
+    """Override to periodically clear conversation context after steps"""
+    # Initialize step counter if not present
+    if not hasattr(self, '_context_step_counter'):
+        self._context_step_counter = 0
+    
+    # Increment counter
+    self._context_step_counter += 1
+    
+    # Every 5 steps, perform aggressive context management
+    if self._context_step_counter >= 5:
+        print("Performing periodic conversation cleanup")
+        self._context_step_counter = 0
+        
+        # Clear most of the conversation history while keeping essential parts
+        if hasattr(self, 'conversation') and len(self.conversation) > 8:
+            original_length = len(self.conversation)
+            
+            # Find system messages
+            system_messages = [msg for msg in self.conversation if msg.get('role') == 'system']
+            
+            # Get the most recent exchanges (last 4 pairs of messages)
+            recent_messages = self.conversation[-8:]
+            
+            # Rebuild conversation with system messages + note + recent messages
+            self.conversation = system_messages + [
+                {'role': 'system', 'content': '[NOTICE: Previous conversation history has been trimmed to manage context size]'}
+            ] + recent_messages
+            
+            print(f"Conversation history trimmed from {original_length} to {len(self.conversation)} messages")
+    
+    # Call the original run method
+    return original_code_agent_run(self, query, max_steps, persist_conversation)
+
+# Apply the monkey patch to CodeAgent class
+CodeAgent.run = context_managed_run
     
 # Initialize the agent using standard smolagents patterns
 agent = CodeAgent(
@@ -599,10 +639,10 @@ agent = CodeAgent(
         Check_Dataset_Validity,
         visit_webpage_tool,  # This is correctly initialized as VisitWebpageTool()
     ],
-    max_steps=20,
+    max_steps=12,
     verbosity_level=1,
     grammar=None,
-    planning_interval=3,
+    planning_interval=2,
     name="Research Assistant",
     description="""An AI assistant that can search the web, create datasets, and answer questions # Note about working within token limits
                 # When using with queries that might exceed token limits, consider: