EX MCP Server

from context.context_manager import AdvancedContextManager def test_optimize_context_preserves_system_and_tail(): cm = AdvancedContextManager() # Create a very large middle to exceed the Z.ai 128k-token limit (estimator ~1 token per 4 chars) msgs = ( [{"role": "system", "content": "rules"}] + [{"role": "user", "content": "x" * 3000} for _ in range(250)] # ~750k chars ) out = cm.optimize_context(msgs, platform="zai") assert any(m.get("role") == "system" for m in out) # Tail limited to last 10 non-system messages when over limit assert sum(1 for m in out if m.get("role") != "system") <= 10