@@ -32,6 +32,40 @@ import (
3232 "github.com/cloudwego/eino/schema"
3333)
3434
35+ func TestGenModelInput (t * testing.T ) {
36+ ctx := context .Background ()
37+
38+ t .Run ("WithInstruction" , func (t * testing.T ) {
39+ input := & adk.AgentInput {
40+ Messages : []* schema.Message {
41+ schema .UserMessage ("hello" ),
42+ },
43+ }
44+
45+ msgs , err := genModelInput (ctx , "You are a helpful assistant" , input )
46+ assert .NoError (t , err )
47+ assert .Len (t , msgs , 2 )
48+ assert .Equal (t , schema .System , msgs [0 ].Role )
49+ assert .Equal (t , "You are a helpful assistant" , msgs [0 ].Content )
50+ assert .Equal (t , schema .User , msgs [1 ].Role )
51+ assert .Equal (t , "hello" , msgs [1 ].Content )
52+ })
53+
54+ t .Run ("WithoutInstruction" , func (t * testing.T ) {
55+ input := & adk.AgentInput {
56+ Messages : []* schema.Message {
57+ schema .UserMessage ("hello" ),
58+ },
59+ }
60+
61+ msgs , err := genModelInput (ctx , "" , input )
62+ assert .NoError (t , err )
63+ assert .Len (t , msgs , 1 )
64+ assert .Equal (t , schema .User , msgs [0 ].Role )
65+ assert .Equal (t , "hello" , msgs [0 ].Content )
66+ })
67+ }
68+
3569func TestWriteTodos (t * testing.T ) {
3670 m , err := buildBuiltinAgentMiddlewares (false )
3771 assert .NoError (t , err )
0 commit comments