fix: add ut of react agent for testing messages modifier (#57)

fix: add ut of react agent for test messages modifier
cloudwego · Feb 7, 2025 · 8a2f8a8 · 8a2f8a8
1 parent 3b806ca
commit 8a2f8a8
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/flow/agent/react/react_test.go b/flow/agent/react/react_test.go
@@ -82,7 +82,10 @@ func TestReact(t *testing.T) {
 		ToolsConfig: compose.ToolsNodeConfig{
 			Tools: []tool.BaseTool{fakeTool},
 		},
-
+		MessageModifier: func(ctx context.Context, input []*schema.Message) []*schema.Message {
+			assert.Equal(t, len(input), times*2+1)
+			return input
+		},
 		MaxStep: 40,
 	})
 	assert.Nil(t, err)
@@ -106,7 +109,10 @@ func TestReact(t *testing.T) {
 		ToolsConfig: compose.ToolsNodeConfig{
 			Tools: []tool.BaseTool{fakeTool},
 		},
-
+		MessageModifier: func(ctx context.Context, input []*schema.Message) []*schema.Message {
+			assert.Equal(t, len(input), times*2+1)
+			return input
+		},
 		MaxStep:            40,
 		ToolReturnDirectly: map[string]struct{}{info.Name: {}},
 	})