Import WhatsApp exports before live messages

[onechat] / model_test.go

diff --git a/model_test.go b/model_test.go
index 16946a061dfe25447e34bded0128e079732d0523..11a7d00bcddf7ad8f68373675a8428111fa81fb4 100644 (file)
--- a/model_test.go
+++ b/model_test.go
@@ -85,6 +85,19 @@ func TestClientSendsTheSessionHeaderToChatGPT(t *testing.T) {
                        call := server.Call(0)
                        require.Equal(t, tc.want, call.header.Get("session_id"))
                        require.Equal(t, "agent-key", call.body["prompt_cache_key"])
+
+                       // A subagent has its own key.
+                       request := modelTestRequest()
+                       request.Agent = "GreenOtter"
+                       _, err = client.Create(t.Context(), request)
+                       require.NoError(t, err)
+
+                       call = server.Call(1)
+                       if tc.want != "" {
+                               require.Equal(t, "agent-key-GreenOtter", call.header.Get("session_id"))
+                       }
+
+                       require.Equal(t, "agent-key-GreenOtter", call.body["prompt_cache_key"])
                })
        }
 }
@@ -150,7 +163,7 @@ func TestClientRecordsEachCall(t *testing.T) {
        require.NoError(t, err)
        require.NotContains(t, server.Call(0).body, "node")
        require.NotContains(t, string(server.Call(0).raw), "[2,5]")
-       require.NotContains(t, string(server.Call(0).raw), "GreenOtter")
+       require.NotContains(t, server.Call(0).body, "agent")
 
        _, err = client.Create(t.Context(), modelTestRequest())
        require.ErrorContains(t, err, "HTTP 429")