314574126
003b4c8893
Salvage of #26860 (hunk 2, ported \u2014 the PR's base predates the current gateway layout by ~11.9K commits). Messaging platforms can set gateway.platforms.<key>.skip_context_files: true to skip the filesystem-heavy context-file discovery (SOUL.md, AGENTS.md, .cursorrules walks) during AIAgent construction \u2014 10-100x slower stat()/walk costs on Windows made this a real per-turn tax. Soul identity is still loaded (single small file), so the persona survives. The flag participates in _agent_config_signature so toggling it rebuilds the cached agent instead of silently reusing a prompt built under the other setting (prompt-cache correctness). The PR's hunk 1 (mtime-caching the per-turn dotenv reload) was dropped:df51ad797mtime-cached load_config/read_raw_config andc2eda92fdremoved the per-turn deepcopies, capturing most of that win; the function has since gained a multiplex early-return and managed-scope overlay that the original whole-function skip would have bypassed.
84 lines
3.6 KiB
Python
84 lines
3.6 KiB
Python
"""Per-platform ``skip_context_files`` gateway wiring (#26860).
|
|
|
|
Messaging platforms can opt out of the filesystem-heavy context-file
|
|
discovery (SOUL.md, AGENTS.md, .cursorrules walks) that runs during
|
|
AIAgent construction — especially impactful on Windows where stat() and
|
|
directory walks are 10-100x slower. The agent-side parameters already
|
|
exist (agent/agent_init.py); these tests pin the gateway wiring:
|
|
config -> signature -> AIAgent kwargs.
|
|
"""
|
|
|
|
import pytest
|
|
|
|
from gateway.run import GatewayRunner
|
|
|
|
|
|
class TestSkipContextFilesSignature:
|
|
"""A toggled skip_context_files must invalidate the agent cache."""
|
|
|
|
RUNTIME = {"provider": "openrouter", "base_url": "", "api_mode": ""}
|
|
|
|
def test_signature_differs_when_toggled(self):
|
|
sig_off = GatewayRunner._agent_config_signature(
|
|
"claude-sonnet-4", self.RUNTIME, ["hermes-telegram"], "",
|
|
skip_context_files=False,
|
|
)
|
|
sig_on = GatewayRunner._agent_config_signature(
|
|
"claude-sonnet-4", self.RUNTIME, ["hermes-telegram"], "",
|
|
skip_context_files=True,
|
|
)
|
|
assert sig_off != sig_on, (
|
|
"skip_context_files changes the frozen system prompt (context "
|
|
"files in vs out) — the cache signature must change with it"
|
|
)
|
|
|
|
def test_signature_stable_when_unchanged(self):
|
|
sig_a = GatewayRunner._agent_config_signature(
|
|
"claude-sonnet-4", self.RUNTIME, ["hermes-telegram"], "",
|
|
skip_context_files=True,
|
|
)
|
|
sig_b = GatewayRunner._agent_config_signature(
|
|
"claude-sonnet-4", self.RUNTIME, ["hermes-telegram"], "",
|
|
skip_context_files=True,
|
|
)
|
|
assert sig_a == sig_b
|
|
|
|
def test_default_matches_explicit_false(self):
|
|
"""Back-compat: omitting the param must hash like False so existing
|
|
cached agents aren't all invalidated by this change."""
|
|
sig_default = GatewayRunner._agent_config_signature(
|
|
"claude-sonnet-4", self.RUNTIME, ["hermes-telegram"], "",
|
|
)
|
|
sig_false = GatewayRunner._agent_config_signature(
|
|
"claude-sonnet-4", self.RUNTIME, ["hermes-telegram"], "",
|
|
skip_context_files=False,
|
|
)
|
|
assert sig_default == sig_false
|
|
|
|
|
|
class TestSkipContextFilesConfigResolution:
|
|
"""The gateway resolution path: platform config dict -> bool."""
|
|
|
|
@pytest.mark.parametrize(
|
|
("cfg", "platform_key", "expected"),
|
|
[
|
|
({"gateway": {"platforms": {"telegram": {"skip_context_files": True}}}}, "telegram", True),
|
|
({"gateway": {"platforms": {"telegram": {"skip_context_files": False}}}}, "telegram", False),
|
|
({"gateway": {"platforms": {"telegram": {}}}}, "telegram", False),
|
|
({"gateway": {"platforms": {}}}, "telegram", False),
|
|
({"gateway": {}}, "telegram", False),
|
|
({}, "telegram", False),
|
|
# Set on a DIFFERENT platform — must not leak.
|
|
({"gateway": {"platforms": {"discord": {"skip_context_files": True}}}}, "telegram", False),
|
|
# Truthy non-bool values coerce.
|
|
({"gateway": {"platforms": {"telegram": {"skip_context_files": 1}}}}, "telegram", True),
|
|
],
|
|
)
|
|
def test_resolution(self, cfg, platform_key, expected):
|
|
# Mirror the production resolution in TurnRunner exactly.
|
|
_platforms_gw_cfg = (cfg.get("gateway") or {}).get("platforms") or {}
|
|
_plat_gw_cfg = _platforms_gw_cfg.get(platform_key) or {}
|
|
_skip_context = _plat_gw_cfg.get("skip_context_files")
|
|
skip_context_files = bool(_skip_context) if _skip_context is not None else False
|
|
assert skip_context_files is expected
|