54607e26b7
- Added support for processing image content in the agent, providing placeholder text when image parsing fails. - Implemented a fallback mechanism to utilize vision tools for image analysis, improving the handling of image content. - Updated tests to validate the new image handling behavior, ensuring correct responses when vision support is unavailable. - Enhanced system prompts to guide users on using the new `image_read` tool for image analysis, improving user experience.
79 lines
2.4 KiB
Go
79 lines
2.4 KiB
Go
package shared
|
|
|
|
import (
|
|
"bytes"
|
|
"errors"
|
|
"io/fs"
|
|
"os"
|
|
"path"
|
|
"strings"
|
|
)
|
|
|
|
const systemToolsMarker = "<!-- Yao System Tools (auto-injected) -->"
|
|
|
|
// writerFS is the minimal filesystem interface needed by the injection helpers.
|
|
// workspace.FS satisfies this interface.
|
|
type writerFS interface {
|
|
ReadFile(name string) ([]byte, error)
|
|
WriteFile(name string, data []byte, perm os.FileMode) error
|
|
MkdirAll(name string, perm os.FileMode) error
|
|
}
|
|
|
|
// InjectSystemSkills copies SKILL files from an embed.FS into the workspace.
|
|
// The skills parameter should be an embed.FS produced by `//go:embed skills`,
|
|
// where each file has a path like "skills/yao-web/SKILL.md". This function
|
|
// strips the "skills/" prefix and writes files into targetDir (e.g. ".claude/skills").
|
|
func InjectSystemSkills(ws writerFS, skills fs.FS, targetDir string) error {
|
|
return fs.WalkDir(skills, "skills", func(p string, d fs.DirEntry, err error) error {
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if d.IsDir() {
|
|
return nil
|
|
}
|
|
|
|
rel := strings.TrimPrefix(p, "skills/")
|
|
dst := path.Join(targetDir, rel)
|
|
|
|
data, err := fs.ReadFile(skills, p)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
dir := path.Dir(dst)
|
|
if err := ws.MkdirAll(dir, 0755); err != nil {
|
|
return err
|
|
}
|
|
return ws.WriteFile(dst, data, 0644)
|
|
})
|
|
}
|
|
|
|
// AppendSystemPrompt injects content into a file in the workspace using an
|
|
// idempotent marker. If the marker already exists, the injected section is
|
|
// replaced with the new content (so updates propagate to existing sandboxes).
|
|
// If the file does not exist it is created with just the marker + content.
|
|
func AppendSystemPrompt(ws writerFS, filename string, content []byte) error {
|
|
existing, err := ws.ReadFile(filename)
|
|
if err != nil {
|
|
if !errors.Is(err, fs.ErrNotExist) {
|
|
return err
|
|
}
|
|
header := []byte(systemToolsMarker + "\n\n")
|
|
return ws.WriteFile(filename, append(header, content...), 0644)
|
|
}
|
|
|
|
idx := bytes.Index(existing, []byte(systemToolsMarker))
|
|
if idx >= 0 {
|
|
injected := append([]byte(systemToolsMarker+"\n\n"), content...)
|
|
prefix := existing[:idx]
|
|
merged := append(bytes.TrimRight(prefix, "\n\r\t "), []byte("\n\n---\n\n")...)
|
|
if idx == 0 {
|
|
merged = nil
|
|
}
|
|
return ws.WriteFile(filename, append(merged, injected...), 0644)
|
|
}
|
|
|
|
separator := []byte("\n\n---\n\n" + systemToolsMarker + "\n\n")
|
|
merged := append(existing, append(separator, content...)...)
|
|
return ws.WriteFile(filename, merged, 0644)
|
|
}
|