Record the subagent name in the usage records

[onechat] / prompt_test.go

diff --git a/prompt_test.go b/prompt_test.go
index c9d3691ba038ab6830a9de6925c497663a22dd76..902cb8677e7d3617f901e4b716c8237a229b701a 100644 (file)
--- a/prompt_test.go
+++ b/prompt_test.go
@@ -47,7 +47,7 @@ func TestPromptWithoutInstructionsHasNoClause(t *testing.T) {
 }
 
 func TestAgentToolsAreStrictAndStable(t *testing.T) {
 }
 
 func TestAgentToolsAreStrictAndStable(t *testing.T) {

-       tools := agentTools()
+       tools := promptTools()

        names := make([]string, len(tools))
        for n, tool := range tools {
                names[n] = tool.Name
        names := make([]string, len(tools))
        for n, tool := range tools {
                names[n] = tool.Name


@@ -67,7 +67,7 @@ func TestAgentToolsAreStrictAndStable(t *testing.T) {
        require.NoError(t, err)
 
        for range 10 {
        require.NoError(t, err)
 
        for range 10 {

-               again, err := json.Marshal(agentTools())
+               again, err := json.Marshal(promptTools())

                require.NoError(t, err)
                require.Equal(t, first, again)
        }
                require.NoError(t, err)
                require.Equal(t, first, again)
        }