Package a systemd user service

[onechat] / prompt_test.go

diff --git a/prompt_test.go b/prompt_test.go
index d2f34fa01f1d1d5ce111d18df13744281ad14c05..043cc9908e09c24bfe55c83dd39c8825857d2c7c 100644 (file)
--- a/prompt_test.go
+++ b/prompt_test.go
@@ -77,7 +77,7 @@ func TestPromptWithoutInstructionsHasNoClause(t *testing.T) {
 
 func TestAgentToolsAreStrictAndStable(t *testing.T) {
        tools := promptTools()
 
 func TestAgentToolsAreStrictAndStable(t *testing.T) {
        tools := promptTools()

-       require.Equal(t, []string{"zoom", "date", "shell", "read", "write", "edit", "spawn"}, promptToolNames(tools))
+       require.Equal(t, []string{"zoom", "date", "shell", "read", "write", "edit", "spawn", "tell", "whatsapp_find", "whatsapp_send", "schedule_set", "schedule_list", "schedule_remove"}, promptToolNames(tools))

        require.Equal(t, []string{"zoom", "date", "shell", "read", "write", "edit"}, promptToolNames(promptSubagentTools()))
 
        for _, tool := range tools {
        require.Equal(t, []string{"zoom", "date", "shell", "read", "write", "edit"}, promptToolNames(promptSubagentTools()))
 
        for _, tool := range tools {


@@ -85,7 +85,7 @@ func TestAgentToolsAreStrictAndStable(t *testing.T) {
                properties := schema["properties"].(map[string]any)
                require.True(t, tool.Strict, tool.Name)
                require.Equal(t, false, schema["additionalProperties"], tool.Name)
                properties := schema["properties"].(map[string]any)
                require.True(t, tool.Strict, tool.Name)
                require.Equal(t, false, schema["additionalProperties"], tool.Name)

-               require.Equal(t, slices.Sorted(maps.Keys(properties)), schema["required"], tool.Name)
+               require.ElementsMatch(t, slices.Collect(maps.Keys(properties)), schema["required"], tool.Name)

        }
 
        first, err := json.Marshal(tools)
        }
 
        first, err := json.Marshal(tools)