feat: 新增 Agent、Memory、Skill 模块
- handler: agent_handler, memory_handler, skill_handler - model: agent.go, skill.go - repository: agent_repo, skill_repo - service: agent_service, memory_service, skill_service - 新增 migrations 目录 Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
This commit is contained in:
@@ -46,3 +46,108 @@ func (r *AgentRepository) Update(agent *model.Agent) error {
|
||||
func (r *AgentRepository) Delete(id string) error {
|
||||
return r.db.Delete(&model.Agent{}, "id = ?", id).Error
|
||||
}
|
||||
|
||||
// AgentSkill 相关方法
|
||||
|
||||
func (r *AgentRepository) CreateAgentSkill(as *model.AgentSkill) error {
|
||||
return r.db.Create(as).Error
|
||||
}
|
||||
|
||||
func (r *AgentRepository) FindSkillsByAgentID(agentID string) ([]model.AgentSkill, error) {
|
||||
var agentSkills []model.AgentSkill
|
||||
err := r.db.Where("agent_id = ?", agentID).Find(&agentSkills).Error
|
||||
return agentSkills, err
|
||||
}
|
||||
|
||||
func (r *AgentRepository) DeleteAgentSkills(agentID string) error {
|
||||
return r.db.Delete(&model.AgentSkill{}, "agent_id = ?", agentID).Error
|
||||
}
|
||||
|
||||
// AgentKnowledgeBase 相关方法
|
||||
|
||||
func (r *AgentRepository) CreateAgentKnowledgeBase(akb *model.AgentKnowledgeBase) error {
|
||||
return r.db.Create(akb).Error
|
||||
}
|
||||
|
||||
func (r *AgentRepository) FindKnowledgeBasesByAgentID(agentID string) ([]model.AgentKnowledgeBase, error) {
|
||||
var agentKBs []model.AgentKnowledgeBase
|
||||
err := r.db.Where("agent_id = ?", agentID).Find(&agentKBs).Error
|
||||
return agentKBs, err
|
||||
}
|
||||
|
||||
func (r *AgentRepository) DeleteAgentKnowledgeBases(agentID string) error {
|
||||
return r.db.Delete(&model.AgentKnowledgeBase{}, "agent_id = ?", agentID).Error
|
||||
}
|
||||
|
||||
// AgentMemory 相关方法
|
||||
|
||||
func (r *AgentRepository) CreateMemory(memory *model.AgentMemory) error {
|
||||
return r.db.Create(memory).Error
|
||||
}
|
||||
|
||||
func (r *AgentRepository) FindMemoriesByAgentID(agentID string, limit int) ([]model.AgentMemory, error) {
|
||||
var memories []model.AgentMemory
|
||||
err := r.db.Where("agent_id = ?", agentID).Order("importance DESC, created_at DESC").Limit(limit).Find(&memories).Error
|
||||
return memories, err
|
||||
}
|
||||
|
||||
func (r *AgentRepository) FindMemoriesByUserID(agentID, userID string, limit int) ([]model.AgentMemory, error) {
|
||||
var memories []model.AgentMemory
|
||||
err := r.db.Where("agent_id = ? AND user_id = ?", agentID, userID).Order("importance DESC, created_at DESC").Limit(limit).Find(&memories).Error
|
||||
return memories, err
|
||||
}
|
||||
|
||||
func (r *AgentRepository) DeleteMemory(id string) error {
|
||||
return r.db.Delete(&model.AgentMemory{}, "id = ?", id).Error
|
||||
}
|
||||
|
||||
// AgentTeam 相关方法
|
||||
|
||||
func (r *AgentRepository) CreateAgentTeam(team *model.AgentTeam) error {
|
||||
return r.db.Create(team).Error
|
||||
}
|
||||
|
||||
func (r *AgentRepository) FindTeamMembers(supervisorAgentID string) ([]model.AgentTeam, error) {
|
||||
var teams []model.AgentTeam
|
||||
err := r.db.Where("supervisor_agent_id = ?", supervisorAgentID).Find(&teams).Error
|
||||
return teams, err
|
||||
}
|
||||
|
||||
func (r *AgentRepository) FindByMemberAgentID(memberAgentID string) ([]model.AgentTeam, error) {
|
||||
var teams []model.AgentTeam
|
||||
err := r.db.Where("member_agent_id = ?", memberAgentID).Find(&teams).Error
|
||||
return teams, err
|
||||
}
|
||||
|
||||
func (r *AgentRepository) DeleteAgentTeam(id string) error {
|
||||
return r.db.Delete(&model.AgentTeam{}, "id = ?", id).Error
|
||||
}
|
||||
|
||||
func (r *AgentRepository) DeleteTeamMembers(supervisorAgentID string) error {
|
||||
return r.db.Delete(&model.AgentTeam{}, "supervisor_agent_id = ?", supervisorAgentID).Error
|
||||
}
|
||||
|
||||
// AgentTask 相关方法
|
||||
|
||||
func (r *AgentRepository) CreateTask(task *model.AgentTask) error {
|
||||
return r.db.Create(task).Error
|
||||
}
|
||||
|
||||
func (r *AgentRepository) UpdateTask(task *model.AgentTask) error {
|
||||
return r.db.Save(task).Error
|
||||
}
|
||||
|
||||
func (r *AgentRepository) FindTasksByAgentID(agentID string, limit int) ([]model.AgentTask, error) {
|
||||
var tasks []model.AgentTask
|
||||
err := r.db.Where("agent_id = ?", agentID).Order("created_at DESC").Limit(limit).Find(&tasks).Error
|
||||
return tasks, err
|
||||
}
|
||||
|
||||
func (r *AgentRepository) FindTaskByID(id string) (*model.AgentTask, error) {
|
||||
var task model.AgentTask
|
||||
err := r.db.First(&task, "id = ?", id).Error
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return &task, nil
|
||||
}
|
||||
|
||||
90
server/internal/repository/skill_repo.go
Normal file
90
server/internal/repository/skill_repo.go
Normal file
@@ -0,0 +1,90 @@
|
||||
package repository
|
||||
|
||||
import (
|
||||
"x-agents/server/internal/model"
|
||||
|
||||
"gorm.io/gorm"
|
||||
)
|
||||
|
||||
type SkillRepository struct {
|
||||
db *gorm.DB
|
||||
}
|
||||
|
||||
func NewSkillRepository(db *gorm.DB) *SkillRepository {
|
||||
return &SkillRepository{db: db}
|
||||
}
|
||||
|
||||
// DB 获取数据库连接
|
||||
func (r *SkillRepository) DB() *gorm.DB {
|
||||
return r.db
|
||||
}
|
||||
|
||||
func (r *SkillRepository) Create(skill *model.Skill) error {
|
||||
return r.db.Create(skill).Error
|
||||
}
|
||||
|
||||
func (r *SkillRepository) FindAll() ([]model.Skill, error) {
|
||||
var skills []model.Skill
|
||||
err := r.db.Order("skill_type, skill_name").Find(&skills).Error
|
||||
return skills, err
|
||||
}
|
||||
|
||||
func (r *SkillRepository) FindByID(id string) (*model.Skill, error) {
|
||||
var skill model.Skill
|
||||
err := r.db.First(&skill, "id = ?", id).Error
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return &skill, nil
|
||||
}
|
||||
|
||||
func (r *SkillRepository) FindByType(skillType string) ([]model.Skill, error) {
|
||||
var skills []model.Skill
|
||||
err := r.db.Where("skill_type = ?", skillType).Order("skill_name").Find(&skills).Error
|
||||
return skills, err
|
||||
}
|
||||
|
||||
func (r *SkillRepository) FindByName(skillName string) (*model.Skill, error) {
|
||||
var skill model.Skill
|
||||
err := r.db.First(&skill, "skill_name = ?", skillName).Error
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return &skill, nil
|
||||
}
|
||||
|
||||
func (r *SkillRepository) Update(skill *model.Skill) error {
|
||||
return r.db.Save(skill).Error
|
||||
}
|
||||
|
||||
func (r *SkillRepository) Delete(id string) error {
|
||||
return r.db.Delete(&model.Skill{}, "id = ?", id).Error
|
||||
}
|
||||
|
||||
// UpsertBatch 批量upsert skills
|
||||
func (r *SkillRepository) UpsertBatch(skills []model.Skill) error {
|
||||
for _, skill := range skills {
|
||||
var existing model.Skill
|
||||
err := r.db.First(&existing, "skill_name = ? AND skill_type = ?", skill.SkillName, skill.SkillType).Error
|
||||
if err == gorm.ErrRecordNotFound {
|
||||
if err := r.db.Create(&skill).Error; err != nil {
|
||||
return err
|
||||
}
|
||||
} else if err != nil {
|
||||
return err
|
||||
} else {
|
||||
existing.SkillDesc = skill.SkillDesc
|
||||
existing.Path = skill.Path
|
||||
existing.Status = skill.Status
|
||||
if err := r.db.Save(&existing).Error; err != nil {
|
||||
return err
|
||||
}
|
||||
}
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
// DeleteByType 根据类型删除
|
||||
func (r *SkillRepository) DeleteByType(skillType string) error {
|
||||
return r.db.Where("skill_type = ?", skillType).Delete(&model.Skill{}).Error
|
||||
}
|
||||
Reference in New Issue
Block a user