Carry new images through model request chains

[onechat] / prompt_test.go

diff --git a/prompt_test.go b/prompt_test.go
index c9cadc2167022026836b29e6259992114c22afab..e9e87f347b9f2676f5ba5cc8ba46f7c0dd83352f 100644 (file)
--- a/prompt_test.go
+++ b/prompt_test.go
@@ -18,6 +18,7 @@ func TestPromptNamesTheAgent(t *testing.T) {
        require.Contains(t, text, `- work: an agent's report, starting "[Name]"`)
        require.Contains(t, text, "- zoom(id, 1) gives message id whole\n- zoom(\"Name\") gives an agent's whole chat\n")
        require.NotContains(t, text, promptName)
        require.Contains(t, text, `- work: an agent's report, starting "[Name]"`)
        require.Contains(t, text, "- zoom(id, 1) gives message id whole\n- zoom(\"Name\") gives an agent's whole chat\n")
        require.NotContains(t, text, promptName)

+       require.Contains(t, text, "[image FILE]")

 
        // The old agent name is built from parts, so the source does not spell
        // it.
 
        // The old agent name is built from parts, so the source does not spell
        // it.


@@ -26,7 +27,7 @@ func TestPromptNamesTheAgent(t *testing.T) {
 
 func TestPromptLeavesOutAbsentFeatures(t *testing.T) {
        text := prompt("Ada", "/AGENT/instructions.md", "")
 
 func TestPromptLeavesOutAbsentFeatures(t *testing.T) {
        text := prompt("Ada", "/AGENT/instructions.md", "")

-       for _, absent := range []string{"image", "computer", "device", "instructions at the end"} {
+       for _, absent := range []string{"computer", "device", "instructions at the end"} {

                require.NotContains(t, text, absent)
        }
 }
                require.NotContains(t, text, absent)
        }
 }


@@ -50,7 +51,7 @@ func TestSubagentPromptNamesTheAgentAndEndsWithTheInstructions(t *testing.T) {
        require.True(t, strings.HasSuffix(text, "\n\n# The user's instructions\n\nI am Lena."))
        require.NotContains(t, text, promptName)
 
        require.True(t, strings.HasSuffix(text, "\n\n# The user's instructions\n\nI am Lena."))
        require.NotContains(t, text, promptName)
 

-       for _, absent := range []string{"image", "computer", "device", "Compaction"} {
+       for _, absent := range []string{"computer", "device", "Compaction"} {

                require.NotContains(t, text, absent)
        }
 
                require.NotContains(t, text, absent)
        }