af0d4edd74
- Introduced `image_generate` tool for generating images from text prompts, with options for specifying output file paths and image dimensions. - Updated `image_read` functionality to allow optional provider specification for enhanced image analysis. - Implemented new `GenerateImage` method in the LLM API for seamless integration of image generation capabilities. - Enhanced documentation to include detailed usage examples for both image reading and generation tools. - Updated tests to validate new image generation features and ensure robust functionality across image tools.
86 lines
1.9 KiB
Go
86 lines
1.9 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,
|
|
}
|
|
|
|
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"},
|
|
}
|
|
|
|
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)
|
|
}
|
|
}
|
|
}
|