mlogclub
031855935d
- Implemented QuickReplySeed structure and seeds for both English and Chinese languages. - Created SkillDefinitionSeed structure with detailed instructions and examples for the "After-sales Escalation" skill in both languages. - Added TagSeed structure and corresponding seeds for categorizing inquiries in English and Chinese. - Refactored skill and tag initialization to utilize the new seed structures, ensuring proper data handling and multilingual support. - Updated tests to verify that English seeds do not contain Chinese text, ensuring data integrity.
73 lines
1.8 KiB
Go
73 lines
1.8 KiB
Go
package skill
|
|
|
|
import (
|
|
"agent-desk/cmd/testdata/seedlang"
|
|
"agent-desk/cmd/testdata/seeds"
|
|
"agent-desk/internal/models"
|
|
"agent-desk/internal/repositories"
|
|
"fmt"
|
|
"time"
|
|
|
|
"github.com/mlogclub/simple/sqls"
|
|
)
|
|
|
|
const AfterSalesEscalationSkillCode = seeds.AfterSalesEscalationSkillCode
|
|
|
|
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, "code = ?", itemCopy.Code)
|
|
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{
|
|
Code: seed.Code,
|
|
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
|
|
}
|