639f0c59fc
- Implemented AssistantReloadFunc to enable hot-reloading of assistants after deployment, improving deployment flexibility. - Enhanced gRPC authProvider to include workspace and sandbox IDs from incoming context metadata, enriching the authentication context. - Updated tools to support new agent-related functionalities, including listing, downloading, deploying, and connecting agents. - Expanded system tools documentation to include new agent commands, ensuring comprehensive guidance for users.
88 lines
2 KiB
Go
88 lines
2 KiB
Go
package tools
|
|
|
|
import (
|
|
"io/fs"
|
|
"strings"
|
|
"testing"
|
|
)
|
|
|
|
func TestSkillsFS_ContainsAllSkills(t *testing.T) {
|
|
expected := map[string]bool{
|
|
"skills/yao-web/SKILL.md": false,
|
|
"skills/yao-process/SKILL.md": false,
|
|
"skills/yao-doc/SKILL.md": false,
|
|
"skills/yao-image/SKILL.md": false,
|
|
"skills/yao-agent/SKILL.md": false,
|
|
}
|
|
|
|
err := fs.WalkDir(SkillsFS, "skills", func(path string, d fs.DirEntry, err error) error {
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if _, ok := expected[path]; ok {
|
|
expected[path] = true
|
|
}
|
|
return nil
|
|
})
|
|
if err != nil {
|
|
t.Fatalf("WalkDir failed: %v", err)
|
|
}
|
|
|
|
for path, found := range expected {
|
|
if !found {
|
|
t.Errorf("expected file not found in SkillsFS: %s", path)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestSkillsFS_FrontmatterFields(t *testing.T) {
|
|
skills := []struct {
|
|
path string
|
|
name string
|
|
}{
|
|
{"skills/yao-web/SKILL.md", "yao-web"},
|
|
{"skills/yao-process/SKILL.md", "yao-process"},
|
|
{"skills/yao-doc/SKILL.md", "yao-doc"},
|
|
{"skills/yao-image/SKILL.md", "yao-image"},
|
|
{"skills/yao-agent/SKILL.md", "yao-agent"},
|
|
}
|
|
|
|
for _, s := range skills {
|
|
data, err := fs.ReadFile(SkillsFS, s.path)
|
|
if err != nil {
|
|
t.Fatalf("ReadFile(%s): %v", s.path, err)
|
|
}
|
|
content := string(data)
|
|
|
|
if !strings.Contains(content, "name: "+s.name) {
|
|
t.Errorf("%s: missing 'name: %s' in frontmatter", s.path, s.name)
|
|
}
|
|
if !strings.Contains(content, "description:") {
|
|
t.Errorf("%s: missing 'description:' in frontmatter", s.path)
|
|
}
|
|
if !strings.Contains(content, "ALWAYS invoke this skill") {
|
|
t.Errorf("%s: description missing directive 'ALWAYS invoke this skill'", s.path)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestSystemPrompt_NonEmpty(t *testing.T) {
|
|
if len(SystemPrompt) == 0 {
|
|
t.Fatal("SystemPrompt is empty")
|
|
}
|
|
|
|
content := string(SystemPrompt)
|
|
|
|
markers := []string{
|
|
"Yao Sandbox Environment",
|
|
"Yao System Tools",
|
|
"tai tool",
|
|
"$WORKDIR",
|
|
"$CTX_SKILLS_DIR",
|
|
}
|
|
for _, m := range markers {
|
|
if !strings.Contains(content, m) {
|
|
t.Errorf("SystemPrompt missing expected marker: %q", m)
|
|
}
|
|
}
|
|
}
|