chore: import clean Agent Desk snapshot (1/7)
This commit is contained in:
70
cmd/testdata/skill/init.go
vendored
Normal file
70
cmd/testdata/skill/init.go
vendored
Normal file
@@ -0,0 +1,70 @@
|
||||
package skill
|
||||
|
||||
import (
|
||||
"agent-desk/cmd/testdata/seedlang"
|
||||
"agent-desk/cmd/testdata/seeds"
|
||||
"agent-desk/internal/models"
|
||||
"agent-desk/internal/repositories"
|
||||
"fmt"
|
||||
"strings"
|
||||
"time"
|
||||
|
||||
"github.com/mlogclub/simple/sqls"
|
||||
)
|
||||
|
||||
type InitResult struct {
|
||||
Created int
|
||||
Updated int
|
||||
}
|
||||
|
||||
func Init(lang seedlang.Language) (*InitResult, error) {
|
||||
result := &InitResult{}
|
||||
seedItems := buildModels(lang)
|
||||
for _, item := range seedItems {
|
||||
itemCopy := item
|
||||
if err := sqls.WithTransaction(func(ctx *sqls.TxContext) error {
|
||||
existing := repositories.SkillDefinitionRepository.Take(ctx.Tx, "name = ?", strings.TrimSpace(itemCopy.Name))
|
||||
if existing != nil {
|
||||
if err := ctx.Tx.Model(existing).Updates(&itemCopy).Error; err != nil {
|
||||
return err
|
||||
}
|
||||
result.Updated++
|
||||
return nil
|
||||
}
|
||||
if err := ctx.Tx.Create(&itemCopy).Error; err != nil {
|
||||
return err
|
||||
}
|
||||
result.Created++
|
||||
return nil
|
||||
}); err != nil {
|
||||
return nil, fmt.Errorf("upsert skill failed: %w", err)
|
||||
}
|
||||
}
|
||||
return result, nil
|
||||
}
|
||||
|
||||
func buildModels(lang seedlang.Language) []models.SkillDefinition {
|
||||
now := time.Now()
|
||||
seedItems := seeds.SkillDefinitionSeeds(lang)
|
||||
items := make([]models.SkillDefinition, 0, len(seedItems))
|
||||
for _, seed := range seedItems {
|
||||
items = append(items, models.SkillDefinition{
|
||||
Name: seed.Name,
|
||||
Description: seed.Description,
|
||||
Instruction: seed.Instruction,
|
||||
Examples: seed.Examples,
|
||||
ToolWhitelist: seed.ToolWhitelist,
|
||||
Status: seed.Status,
|
||||
Remark: seed.Remark,
|
||||
AuditFields: models.AuditFields{
|
||||
CreatedAt: now,
|
||||
CreateUserID: 0,
|
||||
CreateUserName: "System",
|
||||
UpdatedAt: now,
|
||||
UpdateUserID: 0,
|
||||
UpdateUserName: "System",
|
||||
},
|
||||
})
|
||||
}
|
||||
return items
|
||||
}
|
||||
21
cmd/testdata/skill/init_test.go
vendored
Normal file
21
cmd/testdata/skill/init_test.go
vendored
Normal file
@@ -0,0 +1,21 @@
|
||||
package skill
|
||||
|
||||
import (
|
||||
"agent-desk/cmd/testdata/seedlang"
|
||||
"agent-desk/cmd/testdata/seeds"
|
||||
"regexp"
|
||||
"testing"
|
||||
)
|
||||
|
||||
var hanTextPattern = regexp.MustCompile(`\p{Han}`)
|
||||
|
||||
func TestEnglishSkillSeedDoesNotContainChineseText(t *testing.T) {
|
||||
for _, item := range seeds.SkillDefinitionSeeds(seedlang.English) {
|
||||
values := []string{item.Name, item.Description, item.Instruction, item.Examples, item.ToolWhitelist, item.Remark}
|
||||
for _, value := range values {
|
||||
if hanTextPattern.MatchString(value) {
|
||||
t.Fatalf("english skill seed contains Chinese text: %q", value)
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user