fix: agent_input_token_budget wrongly treated as a secret and unsettable from chat (#1294)

* fix: don't classify agent_input_token_budget as a secret (token must be a suffix)

* test: agent_input_token_budget is settable from chat
This commit is contained in:
Afonso Coutinho
2026-06-02 17:53:47 +01:00
committed by GitHub
parent adde94e430
commit a8395b4e4c
2 changed files with 30 additions and 1 deletions

View File

@@ -1527,7 +1527,14 @@ async def do_manage_settings(content: str, owner: Optional[str] = None) -> Dict:
"tavily_api_key", "serper_api_key", "app_public_url", "tavily_api_key", "serper_api_key", "app_public_url",
} }
def _is_secret(k): def _is_secret(k):
return k in _SECRET_KEYS or any(t in k for t in ("api_key", "_key", "token", "secret", "password")) # `token` must be a suffix, not a substring: otherwise the int
# setting `agent_input_token_budget` (which even has a "token budget"
# alias to set it from chat) is wrongly classified as a credential.
return (
k in _SECRET_KEYS
or k.endswith("token")
or any(t in k for t in ("api_key", "_key", "secret", "password"))
)
# Friendly aliases → real keys, so natural phrasing resolves. # Friendly aliases → real keys, so natural phrasing resolves.
_ALIASES_SET = { _ALIASES_SET = {

View File

@@ -0,0 +1,22 @@
"""Regression: agent_input_token_budget must be settable from chat (not flagged secret)."""
import asyncio
import json
import src.settings as settings_mod
from src.tool_implementations import do_manage_settings
def test_set_token_budget_is_not_refused_as_secret(monkeypatch):
store = {}
monkeypatch.setattr(settings_mod, "load_settings", lambda: dict(store))
monkeypatch.setattr(settings_mod, "save_settings", lambda s: store.update(s))
result = asyncio.run(do_manage_settings(json.dumps({
"action": "set", "key": "agent_input_token_budget", "value": 8000,
})))
# The "token" substring used to flag this int setting as a credential and
# refuse to set it (even though there's a deliberate "token budget" alias).
assert "credential" not in result.get("response", "").lower(), result
assert result.get("exit_code") == 0, result
assert store.get("agent_input_token_budget") == 8000