mirror of
https://github.com/NousResearch/hermes-agent.git
synced 2026-11-06 05:17:15 +00:00
Teknium
5f274827ff
`def h(source, /, **kwargs)` and `def h(*session_id, **kwargs)` cannot take that name by keyword, so the context value belongs in kwargs. (Greptile re-review on #140997.)
33 lines
1.4 KiB
Python
33 lines
1.4 KiB
Python
"""Plugin slash-command handlers receive only the calling-session context they declare."""
|
|
|
|
from hermes_cli.plugin_command_dispatch import invoke_plugin_command
|
|
|
|
CONTEXT = {"source": "src", "session_id": "sid", "session_key": "key", "platform": "telegram"}
|
|
|
|
|
|
def test_handlers_get_only_the_context_they_declare():
|
|
def legacy(raw_args):
|
|
return (raw_args,)
|
|
|
|
def named(raw_args, *, session_id, platform):
|
|
return raw_args, session_id, platform
|
|
|
|
def catch_all(args, **kwargs):
|
|
return args, kwargs
|
|
|
|
def positional_only(raw_args, session_id=None, /): # cannot be passed by keyword: no opt-in
|
|
return raw_args, session_id
|
|
|
|
def first_named_source(source, **kwargs): # raw_args fills `source`; it is never passed twice
|
|
return source, sorted(kwargs)
|
|
|
|
def first_positional_only(source, /, **kwargs): # its name is not a keyword: kwargs keep source
|
|
return source, kwargs["source"]
|
|
|
|
assert invoke_plugin_command(legacy, "a", **CONTEXT) == ("a",)
|
|
assert invoke_plugin_command(named, "a", **CONTEXT) == ("a", "sid", "telegram")
|
|
assert invoke_plugin_command(catch_all, "a", **CONTEXT) == ("a", CONTEXT)
|
|
assert invoke_plugin_command(positional_only, "a", **CONTEXT) == ("a", None)
|
|
assert invoke_plugin_command(first_named_source, "a", **CONTEXT) == ("a", ["platform", "session_id", "session_key"])
|
|
assert invoke_plugin_command(first_positional_only, "a", **CONTEXT) == ("a", "src")
|