mirror of
https://github.com/NousResearch/hermes-agent.git
synced 2026-07-29 18:46:59 +00:00
fix(middleware): hide Relay control flag
Signed-off-by: Alex Fournier <afournier@nvidia.com>
This commit is contained in:
parent
2378bd4e4c
commit
0be07b0553
2 changed files with 28 additions and 1 deletions
|
|
@ -132,7 +132,8 @@ def apply_tool_request_middleware(
|
|||
trace: List[Dict[str, Any]] = []
|
||||
|
||||
session_id = str(context.get("session_id") or "")
|
||||
if session_id and not context.pop("skip_relay", False):
|
||||
skip_relay = bool(context.pop("skip_relay", False))
|
||||
if session_id and not skip_relay:
|
||||
from agent import relay_runtime
|
||||
|
||||
relay_args = relay_runtime.apply_tool_request_intercepts(
|
||||
|
|
|
|||
|
|
@ -189,6 +189,32 @@ class TestPluginDiscovery:
|
|||
assert result.changed is True
|
||||
assert result.trace == [{"source": "same-payload"}]
|
||||
|
||||
def test_tool_request_middleware_hides_internal_skip_relay_flag(
|
||||
self,
|
||||
monkeypatch,
|
||||
):
|
||||
observed = []
|
||||
|
||||
def middleware(**kwargs):
|
||||
observed.append(kwargs)
|
||||
return {"args": kwargs["args"]}
|
||||
|
||||
manager = types.SimpleNamespace(
|
||||
_middleware={"tool_request": [middleware]},
|
||||
invoke_middleware=lambda kind, **kwargs: [middleware(**kwargs)],
|
||||
)
|
||||
monkeypatch.setattr("hermes_cli.plugins.get_plugin_manager", lambda: manager)
|
||||
|
||||
apply_tool_request_middleware(
|
||||
"read_file",
|
||||
{"path": "README.md"},
|
||||
session_id="",
|
||||
skip_relay=True,
|
||||
)
|
||||
|
||||
assert len(observed) == 1
|
||||
assert "skip_relay" not in observed[0]
|
||||
|
||||
def test_execution_middleware_post_next_call_error_does_not_retry(self, monkeypatch):
|
||||
calls = []
|
||||
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue