X-Git-Url: https://git.giacomolli.uk/onechat/blobdiff_plain/3cdbd017da9ae7a7045f4272f16d4b262fd01c79..c781f79f3c8cb9a7577ee1574d196988da2719b3:/prompt_test.go

diff --git a/prompt_test.go b/prompt_test.go
index 06fea49..158784b 100644
--- a/prompt_test.go
+++ b/prompt_test.go
@@ -65,6 +65,7 @@ func TestPromptEndsWithTheInstructions(t *testing.T) {
 	require.Contains(t, text, `Do the user's tasks yourself, with your tools, following the user's instructions
 at the end of this prompt: who they are, how their files are organized and how
 they want work done.`)
+
 	require.True(t, strings.HasSuffix(text, "\n\n# The user's instructions\n\nI am Lena. Keep replies short."))
 }
 
@@ -166,3 +167,14 @@ func TestRootPromptDefinesChildRoutingAndConsent(t *testing.T) {
 	require.NotContains(t, child, "Use agent_create")
 	require.Contains(t, child, "Send nothing to WhatsApp. Draft replies in your report instead")
 }
+
+func TestRootPromptKeepsImageConsentAndChildForwardingRules(t *testing.T) {
+	text := prompt("Ada", "/AGENT/instructions.md", "")
+	require.Contains(t, text, "Use image_show or whatsapp_send with an image only on the user's request.")
+	require.Contains(t, text, "Pass an image to a child with an [image FILE] line in tell.")
+
+	child := promptChild("House", "Ada", "Keep house plans.", "/CHILD/instructions.md", "")
+	require.Contains(t, child, "Send nothing to WhatsApp. Draft replies in your report instead")
+	require.NotContains(t, promptToolNames(promptSubagentTools()), "image_show")
+	require.NotContains(t, promptToolNames(promptSubagentTools()), "whatsapp_send")
+}