Import AITURK IDE 1.0.0-beta.1 from Hermes 63279301; preserve MIT license
This commit is contained in:
@@ -0,0 +1,89 @@
|
||||
"""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),
|
||||
# ``hermes gateway setup`` writes platforms as a LIST of enabled
|
||||
# platform names — must not crash and must default to False (#83185).
|
||||
({"gateway": {"platforms": ["telegram", "discord"]}}, "telegram", False),
|
||||
({"gateway": {"platforms": []}}, "telegram", False),
|
||||
],
|
||||
)
|
||||
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 {}
|
||||
if not isinstance(_platforms_gw_cfg, dict):
|
||||
_platforms_gw_cfg = {}
|
||||
_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
|
||||
Reference in New Issue
Block a user