slhaf/Partner
1
0
Fork 0
mirror of https://github.com/slhaf/Partner.git synced 2026-05-12 08:43:02 +08:00
Code Issues Packages Projects Releases Wiki Activity

refactor(runner): allow object-typed params in BuiltinActionRegistry builtin actions

Browse Source
This commit is contained in:
slhafzjw
2026-03-17 22:51:00 +08:00
parent 4439e5c04b
commit 6474eb8dc6
2 changed files with 7 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
Partner-Core/src/main/java/work/slhaf/partner/module/modules/action/builtin/BuiltinActionRegistry.java
View File

@@ -37,12 +37,12 @@ public class BuiltinActionRegistry extends AbstractAgentModule.Standalone {
return List.of();
}
public void defineBuiltinAction(String name, MetaActionInfo metaActionInfo, Function<Map<String, String>, String> invoker) {
public void defineBuiltinAction(String name, MetaActionInfo metaActionInfo, Function<Map<String, Object>, String> invoker) {
BuiltinActionDefinition definition = new BuiltinActionDefinition(BUILTIN_LOCATION + "::" + name, metaActionInfo, invoker);
definitions.put(definition.actionKey(), definition);
}
public String call(@NonNull String actionKey, @NonNull Map<String, String> params) {
public String call(@NonNull String actionKey, @NonNull Map<String, Object> params) {
BuiltinActionDefinition definition = definitions.get(actionKey);
if (definition == null) {
throw new MetaActionNotFoundException("未找到对应的内置行动程序: " + actionKey);
@@ -63,7 +63,7 @@ public class BuiltinActionRegistry extends AbstractAgentModule.Standalone {
public record BuiltinActionDefinition(
String actionKey,
MetaActionInfo metaActionInfo,
Function<Map<String, String>, String> invoker
Function<Map<String, Object>, String> invoker
) {
}
}

8
Partner-Core/src/test/java/work/slhaf/partner/module/modules/action/builtin/BuiltinActionRegistryTest.java
View File

@@ -42,7 +42,7 @@ class BuiltinActionRegistryTest {
private static BuiltinActionRegistry.BuiltinActionDefinition buildDefinition(
String name,
MetaActionInfo metaActionInfo,
java.util.function.Function<Map<String, Object>, Object> invoker
java.util.function.Function<Map<String, Object>, String> invoker
) {
return new BuiltinActionRegistry.BuiltinActionDefinition(
BUILTIN_LOCATION + "::" + name,
@@ -59,7 +59,7 @@ class BuiltinActionRegistryTest {
BuiltinActionRegistry registry = spy(new BuiltinActionRegistry());
doReturn(List.of(
buildDefinition("echo", buildMetaActionInfo("echo"), params -> params.get("value"))
buildDefinition("echo", buildMetaActionInfo("echo"), params -> params.get("value").toString())
)).when(registry).buildDefaultActionDefinitions();
injectActionCapability(registry, actionCapability);
@@ -75,8 +75,8 @@ class BuiltinActionRegistryTest {
@Test
void testCallReturnsStringifiedResults() {
BuiltinActionRegistry registry = new BuiltinActionRegistry();
registry.defineBuiltinAction("echo", buildMetaActionInfo("echo"), params -> params.get("value"));
registry.defineBuiltinAction("json", buildMetaActionInfo("json"), params -> Map.of("ok", true));
registry.defineBuiltinAction("echo", buildMetaActionInfo("echo"), params -> params.get("value").toString());
registry.defineBuiltinAction("json", buildMetaActionInfo("json"), params -> Map.of("ok", true).toString());
registry.defineBuiltinAction("nil", buildMetaActionInfo("nil"), params -> null);
Assertions.assertEquals("hello", registry.call("builtin::echo", Map.of("value", "hello")));