X-Git-Url: http://git.giacomolli.uk/onechat/blobdiff_plain/4a605fa39078354654ae648e489c117c7a445197..f59bb67ae1bc2c899ef4f6be9ce2ff6296c7ffcc:/prompt_test.go

diff --git a/prompt_test.go b/prompt_test.go
index 4c88ad8..c9d3691 100644
--- a/prompt_test.go
+++ b/prompt_test.go
@@ -48,7 +48,12 @@ func TestPromptWithoutInstructionsHasNoClause(t *testing.T) {
 
 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)