]> git.giacomolli.uk - onechat/blobdiff - prompt_test.go
Convert reply Markdown to Telegram entities
[onechat] / prompt_test.go
index 4c88ad8a80d2683573ccc877523488691f6de5ac..c9d3691ba038ab6830a9de6925c497663a22dd76 100644 (file)
@@ -48,7 +48,12 @@ func TestPromptWithoutInstructionsHasNoClause(t *testing.T) {
 
 func TestAgentToolsAreStrictAndStable(t *testing.T) {
        tools := agentTools()
 
 func TestAgentToolsAreStrictAndStable(t *testing.T) {
        tools := agentTools()
-       require.Equal(t, []string{"zoom", "date"}, []string{tools[0].Name, tools[1].Name})
+       names := make([]string, len(tools))
+       for n, tool := range tools {
+               names[n] = tool.Name
+       }
+
+       require.Equal(t, []string{"zoom", "date", "shell", "read", "write", "edit"}, names)
 
        for _, tool := range tools {
                schema := tool.Parameters.(map[string]any)
 
        for _, tool := range tools {
                schema := tool.Parameters.(map[string]any)