mirror of
https://github.com/sipeed/picoclaw.git
synced 2026-06-12 18:08:54 +00:00
Harsh Bansal
d692cc0cc6
* Add Find Skills and Install Skills * Improvements * fix file name * Update pkg/skills/clawhub_registry.go Co-authored-by: Copilot <175728472+Copilot@users.noreply.github.com> * fix * Comments addressed * Resolve comments * fix tests * fixes * Comments resolved * Update pkg/skills/search_cache_repro_test.go Co-authored-by: Copilot <175728472+Copilot@users.noreply.github.com> * minor fix * fix test * fixes --------- Co-authored-by: Copilot <175728472+Copilot@users.noreply.github.com>
83 lines
2.5 KiB
Go
83 lines
2.5 KiB
Go
package tools
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
|
|
"github.com/sipeed/picoclaw/pkg/skills"
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestFindSkillsToolName(t *testing.T) {
|
|
tool := NewFindSkillsTool(skills.NewRegistryManager(), nil)
|
|
assert.Equal(t, "find_skills", tool.Name())
|
|
}
|
|
|
|
func TestFindSkillsToolMissingQuery(t *testing.T) {
|
|
tool := NewFindSkillsTool(skills.NewRegistryManager(), nil)
|
|
result := tool.Execute(context.Background(), map[string]interface{}{})
|
|
assert.True(t, result.IsError)
|
|
assert.Contains(t, result.ForLLM, "query is required")
|
|
}
|
|
|
|
func TestFindSkillsToolEmptyQuery(t *testing.T) {
|
|
tool := NewFindSkillsTool(skills.NewRegistryManager(), nil)
|
|
result := tool.Execute(context.Background(), map[string]interface{}{
|
|
"query": " ",
|
|
})
|
|
assert.True(t, result.IsError)
|
|
}
|
|
|
|
func TestFindSkillsToolCacheHit(t *testing.T) {
|
|
cache := skills.NewSearchCache(10, 5*60*1000*1000*1000) // 5 min
|
|
cache.Put("github", []skills.SearchResult{
|
|
{Slug: "github", Score: 0.9, RegistryName: "clawhub"},
|
|
})
|
|
|
|
tool := NewFindSkillsTool(skills.NewRegistryManager(), cache)
|
|
result := tool.Execute(context.Background(), map[string]interface{}{
|
|
"query": "github",
|
|
})
|
|
|
|
assert.False(t, result.IsError)
|
|
assert.Contains(t, result.ForLLM, "github")
|
|
assert.Contains(t, result.ForLLM, "cached")
|
|
}
|
|
|
|
func TestFindSkillsToolParameters(t *testing.T) {
|
|
tool := NewFindSkillsTool(skills.NewRegistryManager(), nil)
|
|
params := tool.Parameters()
|
|
|
|
props, ok := params["properties"].(map[string]interface{})
|
|
assert.True(t, ok)
|
|
assert.Contains(t, props, "query")
|
|
assert.Contains(t, props, "limit")
|
|
|
|
required, ok := params["required"].([]string)
|
|
assert.True(t, ok)
|
|
assert.Contains(t, required, "query")
|
|
}
|
|
|
|
func TestFindSkillsToolDescription(t *testing.T) {
|
|
tool := NewFindSkillsTool(skills.NewRegistryManager(), nil)
|
|
assert.NotEmpty(t, tool.Description())
|
|
assert.Contains(t, tool.Description(), "skill")
|
|
}
|
|
|
|
func TestFormatSearchResultsEmpty(t *testing.T) {
|
|
result := formatSearchResults("test query", nil, false)
|
|
assert.Contains(t, result, "No skills found")
|
|
}
|
|
|
|
func TestFormatSearchResultsWithData(t *testing.T) {
|
|
results := []skills.SearchResult{
|
|
{Slug: "github", Score: 0.95, DisplayName: "GitHub", Summary: "GitHub API integration", Version: "1.0.0", RegistryName: "clawhub"},
|
|
}
|
|
output := formatSearchResults("github", results, false)
|
|
assert.Contains(t, output, "github")
|
|
assert.Contains(t, output, "v1.0.0")
|
|
assert.Contains(t, output, "0.950")
|
|
assert.Contains(t, output, "clawhub")
|
|
assert.Contains(t, output, "install_skill")
|
|
}
|