diff --git a/agent/assistant/load_test.go b/agent/assistant/load_test.go index 4c9227c2..1c6f26a1 100644 --- a/agent/assistant/load_test.go +++ b/agent/assistant/load_test.go @@ -476,7 +476,7 @@ func TestLoadSystemAgents(t *testing.T) { keywordAst, keywordExists := assistant.GetCache().Get("__yao.keyword") require.True(t, keywordExists, "__yao.keyword should be loaded") assert.Equal(t, "__yao.keyword", keywordAst.ID) - assert.Equal(t, "Keyword Extraction", keywordAst.Name) + assert.Equal(t, "Keyword Extractor", keywordAst.Name) assert.True(t, keywordAst.Readonly) assert.True(t, keywordAst.BuiltIn) assert.Contains(t, keywordAst.Tags, "system") @@ -487,7 +487,7 @@ func TestLoadSystemAgents(t *testing.T) { querydslAst, querydslExists := assistant.GetCache().Get("__yao.querydsl") require.True(t, querydslExists, "__yao.querydsl should be loaded") assert.Equal(t, "__yao.querydsl", querydslAst.ID) - assert.Equal(t, "QueryDSL Generator", querydslAst.Name) + assert.Equal(t, "Query Builder", querydslAst.Name) assert.True(t, querydslAst.Readonly) assert.True(t, querydslAst.BuiltIn) assert.Contains(t, querydslAst.Tags, "system") @@ -514,7 +514,7 @@ func TestLoadSystemAgents(t *testing.T) { needsearchAst, needsearchExists := assistant.GetCache().Get("__yao.needsearch") require.True(t, needsearchExists, "__yao.needsearch should be loaded") assert.Equal(t, "__yao.needsearch", needsearchAst.ID) - assert.Equal(t, "Need Search", needsearchAst.Name) + assert.Equal(t, "Reference Checker", needsearchAst.Name) assert.True(t, needsearchAst.Readonly) assert.True(t, needsearchAst.BuiltIn) })