Import AITURK IDE 1.0.0-beta.1 from Hermes 63279301; preserve MIT license

tests/run_agent/test_gemini_native_reservation.py

@@ -0,0 +1,96 @@
+"""Native Gemini output-token reservation at agent init (#57275 claim 4).
+
+When ``model.max_tokens`` is unset, the native generateContent adapter still
+sends ``maxOutputTokens=65,535`` (GEMINI_DEFAULT_MAX_OUTPUT_TOKENS) — Gemini
+treats an omitted cap as a low internal default, not "unlimited", so the
+adapter always sends an explicit cap.  The compressor's trigger is
+``pct × (window − max_tokens)``; constructing it with ``max_tokens=None``
+reserved 0 while the wire reserved 65,535: on a 128K window the trigger
+landed at ~96K against a real safe input budget of ~65K, and the provider
+400'd before compaction ever fired.
+
+These tests assert the compressor's reservation mirrors the adapter default
+on the native Gemini path, and ONLY there.
+"""
+
+
+
+
+from unittest.mock import patch
+
+
+
+import agent.context_compressor as cc_mod
+from agent.gemini_native_adapter import GEMINI_DEFAULT_MAX_OUTPUT_TOKENS
+
+
+CFG = {"agent": {}}
+
+
+def _build_agent(model, base_url, provider="", max_tokens=None, window=131072):
+    with (
+        patch("run_agent.get_tool_definitions", return_value=[]),
+        patch("run_agent.check_toolset_requirements", return_value={}),
+        patch("run_agent.OpenAI"),
+        patch("hermes_cli.config.load_config", return_value=CFG),
+        patch("hermes_cli.config.load_config_readonly", return_value=CFG),
+        patch(
+            "agent.model_metadata.get_model_context_length", return_value=window,
+        ),
+        patch.object(cc_mod, "get_model_context_length", return_value=window),
+    ):
+        from run_agent import AIAgent
+
+        return AIAgent(
+            model=model,
+            api_key="test-key-1234567890",
+            base_url=base_url,
+            provider=provider,
+            max_tokens=max_tokens,
+            quiet_mode=True,
+            skip_context_files=True,
+            skip_memory=True,
+        )
+
+
+def test_native_gemini_unset_max_tokens_reserves_adapter_default():
+    agent = _build_agent(
+        "gemma-3-27b-it",
+        "https://generativelanguage.googleapis.com/v1beta",
+    )
+    cc = agent.context_compressor
+    assert cc.max_tokens == GEMINI_DEFAULT_MAX_OUTPUT_TOKENS
+    # Trigger must sit at/below the real safe input budget the wire leaves.
+    assert cc.threshold_tokens <= cc.context_length - GEMINI_DEFAULT_MAX_OUTPUT_TOKENS
+
+
+def test_gemini_provider_name_also_reserves_default():
+    agent = _build_agent(
+        "gemini-3.7-flash", "https://example-proxy.invalid/v1", provider="google",
+    )
+    assert agent.context_compressor.max_tokens == GEMINI_DEFAULT_MAX_OUTPUT_TOKENS
+
+
+def test_explicit_max_tokens_wins_over_adapter_default():
+    agent = _build_agent(
+        "gemma-3-27b-it",
+        "https://generativelanguage.googleapis.com/v1beta",
+        max_tokens=8192,
+    )
+    assert agent.context_compressor.max_tokens == 8192
+
+
+def test_non_gemini_paths_keep_no_reservation():
+    agent = _build_agent(
+        "openai/gpt-4.1", "https://openrouter.ai/api/v1",
+    )
+    assert agent.context_compressor.max_tokens is None
+
+
+def test_gemini_openai_compat_endpoint_not_treated_as_native():
+    # The /openai compatibility endpoint does not use the native adapter.
+    agent = _build_agent(
+        "gemma-3-27b-it",
+        "https://generativelanguage.googleapis.com/v1beta/openai",
+    )
+    assert agent.context_compressor.max_tokens is None