chore: move gemini's MAX_RECENT_TURN_WITH_SCREENSHOTS to env vars

Author: Parth220

hud/agents/gemini.py

@@ -24,9 +24,6 @@
 
 logger = logging.getLogger(__name__)
 
-# Maximum number of recent turns to keep screenshots for
-MAX_RECENT_TURN_WITH_SCREENSHOTS = 3
-
 # Predefined Gemini computer use functions
 PREDEFINED_COMPUTER_USE_FUNCTIONS = [
     "open_web_browser",
@@ -110,6 +107,12 @@ def __init__(
         self.excluded_predefined_functions = excluded_predefined_functions or []
         self.hud_console = HUDConsole(logger=logger)
 
+        # Context management: Maximum number of recent turns to keep screenshots for
+        # Configurable via GEMINI_MAX_RECENT_TURN_WITH_SCREENSHOTS environment variable
+        self.max_recent_turn_with_screenshots = (
+            computer_settings.GEMINI_MAX_RECENT_TURN_WITH_SCREENSHOTS
+        )
+
         self.model_name = self.model
 
         # Track mapping from Gemini tool names to MCP tool names
@@ -458,7 +461,7 @@ def _convert_tools_for_gemini(self) -> list[genai_types.Tool]:
     def _remove_old_screenshots(self, messages: list[genai_types.Content]) -> None:
         """
         Remove screenshots from old turns to manage context length.
-        Keeps only the last MAX_RECENT_TURN_WITH_SCREENSHOTS turns with screenshots.
+        Keeps only the last N turns with screenshots (configured via self.max_recent_turn_with_screenshots).
         """
         turn_with_screenshots_found = 0
 
@@ -478,7 +481,7 @@ def _remove_old_screenshots(self, messages: list[genai_types.Content]) -> None:
                 if has_screenshot:
                     turn_with_screenshots_found += 1
                     # Remove the screenshot image if the number of screenshots exceeds the limit
-                    if turn_with_screenshots_found > MAX_RECENT_TURN_WITH_SCREENSHOTS:
+                    if turn_with_screenshots_found > self.max_recent_turn_with_screenshots:
                         for part in content.parts:
                             if (
                                 part.function_response

hud/tools/computer/settings.py

@@ -109,6 +109,11 @@ class ComputerSettings(BaseSettings):
         description="Whether to rescale images to the agent width and height",
         validation_alias="GEMINI_RESCALE_IMAGES",
     )
+    GEMINI_MAX_RECENT_TURN_WITH_SCREENSHOTS: int = Field(
+        default=3,
+        description="Maximum number of recent turns to keep screenshots for in Gemini agent",
+        validation_alias="GEMINI_MAX_RECENT_TURN_WITH_SCREENSHOTS",
+    )
 
 
 computer_settings = ComputerSettings()