feat: subagent table structure and crud apis

2026-04-27 07:16:19 +09:00 · 2026-01-31 19:58:17 +08:00
parent 9fd15bfa6b
commit fe50b1d224
13 changed files with 2689 additions and 2 deletions
@@ -0,0 +1,355 @@
+package subagent
+
+import (
+	"context"
+	"encoding/json"
+	"errors"
+	"fmt"
+	"strings"
+
+	"github.com/google/uuid"
+	"github.com/jackc/pgx/v5"
+	"github.com/jackc/pgx/v5/pgtype"
+
+	"github.com/memohai/memoh/internal/db/sqlc"
+)
+
+type Service struct {
+	queries *sqlc.Queries
+}
+
+func NewService(queries *sqlc.Queries) *Service {
+	return &Service{queries: queries}
+}
+
+func (s *Service) Create(ctx context.Context, userID string, req CreateRequest) (Subagent, error) {
+	if s.queries == nil {
+		return Subagent{}, fmt.Errorf("subagent queries not configured")
+	}
+	name := strings.TrimSpace(req.Name)
+	if name == "" {
+		return Subagent{}, fmt.Errorf("name is required")
+	}
+	description := strings.TrimSpace(req.Description)
+	if description == "" {
+		return Subagent{}, fmt.Errorf("description is required")
+	}
+	pgUserID, err := parseUUID(userID)
+	if err != nil {
+		return Subagent{}, err
+	}
+	messagesPayload, err := marshalMessages(req.Messages)
+	if err != nil {
+		return Subagent{}, err
+	}
+	metadataPayload, err := marshalMetadata(req.Metadata)
+	if err != nil {
+		return Subagent{}, err
+	}
+	skillsPayload, err := marshalSkills(req.Skills)
+	if err != nil {
+		return Subagent{}, err
+	}
+	row, err := s.queries.CreateSubagent(ctx, sqlc.CreateSubagentParams{
+		Name:        name,
+		Description: description,
+		UserID:      pgUserID,
+		Messages:    messagesPayload,
+		Metadata:    metadataPayload,
+		Skills:      skillsPayload,
+	})
+	if err != nil {
+		return Subagent{}, err
+	}
+	return toSubagent(row)
+}
+
+func (s *Service) Get(ctx context.Context, id string) (Subagent, error) {
+	pgID, err := parseUUID(id)
+	if err != nil {
+		return Subagent{}, err
+	}
+	row, err := s.queries.GetSubagentByID(ctx, pgID)
+	if err != nil {
+		if errors.Is(err, pgx.ErrNoRows) {
+			return Subagent{}, fmt.Errorf("subagent not found")
+		}
+		return Subagent{}, err
+	}
+	return toSubagent(row)
+}
+
+func (s *Service) List(ctx context.Context, userID string) ([]Subagent, error) {
+	pgUserID, err := parseUUID(userID)
+	if err != nil {
+		return nil, err
+	}
+	rows, err := s.queries.ListSubagentsByUser(ctx, pgUserID)
+	if err != nil {
+		return nil, err
+	}
+	items := make([]Subagent, 0, len(rows))
+	for _, row := range rows {
+		item, err := toSubagent(row)
+		if err != nil {
+			return nil, err
+		}
+		items = append(items, item)
+	}
+	return items, nil
+}
+
+func (s *Service) Update(ctx context.Context, id string, req UpdateRequest) (Subagent, error) {
+	existing, err := s.Get(ctx, id)
+	if err != nil {
+		return Subagent{}, err
+	}
+	name := existing.Name
+	if req.Name != nil {
+		name = strings.TrimSpace(*req.Name)
+		if name == "" {
+			return Subagent{}, fmt.Errorf("name is required")
+		}
+	}
+	description := existing.Description
+	if req.Description != nil {
+		description = strings.TrimSpace(*req.Description)
+		if description == "" {
+			return Subagent{}, fmt.Errorf("description is required")
+		}
+	}
+	metadata := existing.Metadata
+	if req.Metadata != nil {
+		metadata = req.Metadata
+	}
+	metadataPayload, err := marshalMetadata(metadata)
+	if err != nil {
+		return Subagent{}, err
+	}
+	pgID, err := parseUUID(id)
+	if err != nil {
+		return Subagent{}, err
+	}
+	row, err := s.queries.UpdateSubagent(ctx, sqlc.UpdateSubagentParams{
+		ID:          pgID,
+		Name:        name,
+		Description: description,
+		Metadata:    metadataPayload,
+	})
+	if err != nil {
+		return Subagent{}, err
+	}
+	return toSubagent(row)
+}
+
+func (s *Service) UpdateContext(ctx context.Context, id string, req UpdateContextRequest) (Subagent, error) {
+	messagesPayload, err := marshalMessages(req.Messages)
+	if err != nil {
+		return Subagent{}, err
+	}
+	pgID, err := parseUUID(id)
+	if err != nil {
+		return Subagent{}, err
+	}
+	row, err := s.queries.UpdateSubagentMessages(ctx, sqlc.UpdateSubagentMessagesParams{
+		ID:       pgID,
+		Messages: messagesPayload,
+	})
+	if err != nil {
+		return Subagent{}, err
+	}
+	return toSubagent(row)
+}
+
+func (s *Service) UpdateSkills(ctx context.Context, id string, req UpdateSkillsRequest) (Subagent, error) {
+	skillsPayload, err := marshalSkills(req.Skills)
+	if err != nil {
+		return Subagent{}, err
+	}
+	pgID, err := parseUUID(id)
+	if err != nil {
+		return Subagent{}, err
+	}
+	row, err := s.queries.UpdateSubagentSkills(ctx, sqlc.UpdateSubagentSkillsParams{
+		ID:     pgID,
+		Skills: skillsPayload,
+	})
+	if err != nil {
+		return Subagent{}, err
+	}
+	return toSubagent(row)
+}
+
+func (s *Service) AddSkills(ctx context.Context, id string, req AddSkillsRequest) (Subagent, error) {
+	existing, err := s.Get(ctx, id)
+	if err != nil {
+		return Subagent{}, err
+	}
+	merged := mergeSkills(existing.Skills, req.Skills)
+	payload, err := marshalSkills(merged)
+	if err != nil {
+		return Subagent{}, err
+	}
+	pgID, err := parseUUID(id)
+	if err != nil {
+		return Subagent{}, err
+	}
+	row, err := s.queries.UpdateSubagentSkills(ctx, sqlc.UpdateSubagentSkillsParams{
+		ID:     pgID,
+		Skills: payload,
+	})
+	if err != nil {
+		return Subagent{}, err
+	}
+	return toSubagent(row)
+}
+
+func (s *Service) Delete(ctx context.Context, id string) error {
+	pgID, err := parseUUID(id)
+	if err != nil {
+		return err
+	}
+	return s.queries.SoftDeleteSubagent(ctx, pgID)
+}
+
+func toSubagent(row sqlc.Subagent) (Subagent, error) {
+	messages, err := unmarshalMessages(row.Messages)
+	if err != nil {
+		return Subagent{}, err
+	}
+	metadata, err := unmarshalMetadata(row.Metadata)
+	if err != nil {
+		return Subagent{}, err
+	}
+	skills, err := unmarshalSkills(row.Skills)
+	if err != nil {
+		return Subagent{}, err
+	}
+	item := Subagent{
+		ID:          toUUIDString(row.ID),
+		Name:        row.Name,
+		Description: row.Description,
+		UserID:      toUUIDString(row.UserID),
+		Messages:    messages,
+		Metadata:    metadata,
+		Skills:      skills,
+		Deleted:     row.Deleted,
+	}
+	if row.CreatedAt.Valid {
+		item.CreatedAt = row.CreatedAt.Time
+	}
+	if row.UpdatedAt.Valid {
+		item.UpdatedAt = row.UpdatedAt.Time
+	}
+	if row.DeletedAt.Valid {
+		deletedAt := row.DeletedAt.Time
+		item.DeletedAt = &deletedAt
+	}
+	return item, nil
+}
+
+func marshalMessages(messages []map[string]interface{}) ([]byte, error) {
+	if messages == nil {
+		messages = []map[string]interface{}{}
+	}
+	return json.Marshal(messages)
+}
+
+func unmarshalMessages(payload []byte) ([]map[string]interface{}, error) {
+	if len(payload) == 0 {
+		return []map[string]interface{}{}, nil
+	}
+	var messages []map[string]interface{}
+	if err := json.Unmarshal(payload, &messages); err != nil {
+		return nil, err
+	}
+	if messages == nil {
+		messages = []map[string]interface{}{}
+	}
+	return messages, nil
+}
+
+func marshalMetadata(metadata map[string]interface{}) ([]byte, error) {
+	if metadata == nil {
+		metadata = map[string]interface{}{}
+	}
+	return json.Marshal(metadata)
+}
+
+func unmarshalMetadata(payload []byte) (map[string]interface{}, error) {
+	if len(payload) == 0 {
+		return map[string]interface{}{}, nil
+	}
+	var metadata map[string]interface{}
+	if err := json.Unmarshal(payload, &metadata); err != nil {
+		return nil, err
+	}
+	if metadata == nil {
+		metadata = map[string]interface{}{}
+	}
+	return metadata, nil
+}
+
+func marshalSkills(skills []string) ([]byte, error) {
+	return json.Marshal(normalizeSkills(skills))
+}
+
+func unmarshalSkills(payload []byte) ([]string, error) {
+	if len(payload) == 0 {
+		return []string{}, nil
+	}
+	var skills []string
+	if err := json.Unmarshal(payload, &skills); err != nil {
+		return nil, err
+	}
+	if skills == nil {
+		skills = []string{}
+	}
+	return skills, nil
+}
+
+func normalizeSkills(skills []string) []string {
+	seen := map[string]struct{}{}
+	normalized := make([]string, 0, len(skills))
+	for _, skill := range skills {
+		trimmed := strings.TrimSpace(skill)
+		if trimmed == "" {
+			continue
+		}
+		if _, ok := seen[trimmed]; ok {
+			continue
+		}
+		seen[trimmed] = struct{}{}
+		normalized = append(normalized, trimmed)
+	}
+	return normalized
+}
+
+func mergeSkills(existing []string, incoming []string) []string {
+	merged := append([]string{}, existing...)
+	merged = append(merged, incoming...)
+	return normalizeSkills(merged)
+}
+
+func parseUUID(id string) (pgtype.UUID, error) {
+	parsed, err := uuid.Parse(strings.TrimSpace(id))
+	if err != nil {
+		return pgtype.UUID{}, fmt.Errorf("invalid UUID: %w", err)
+	}
+	var pgID pgtype.UUID
+	pgID.Valid = true
+	copy(pgID.Bytes[:], parsed[:])
+	return pgID, nil
+}
+
+func toUUIDString(value pgtype.UUID) string {
+	if !value.Valid {
+		return ""
+	}
+	id, err := uuid.FromBytes(value.Bytes[:])
+	if err != nil {
+		return ""
+	}
+	return id.String()
+}
+
@@ -0,0 +1,56 @@
+package subagent
+
+import "time"
+
+type Subagent struct {
+	ID          string                   `json:"id"`
+	Name        string                   `json:"name"`
+	Description string                   `json:"description"`
+	UserID      string                   `json:"user_id"`
+	Messages    []map[string]interface{} `json:"messages"`
+	Metadata    map[string]interface{}   `json:"metadata"`
+	Skills      []string                 `json:"skills"`
+	CreatedAt   time.Time                `json:"created_at"`
+	UpdatedAt   time.Time                `json:"updated_at"`
+	Deleted     bool                     `json:"deleted"`
+	DeletedAt   *time.Time               `json:"deleted_at,omitempty"`
+}
+
+type CreateRequest struct {
+	Name        string                   `json:"name"`
+	Description string                   `json:"description"`
+	Messages    []map[string]interface{} `json:"messages,omitempty"`
+	Metadata    map[string]interface{}   `json:"metadata,omitempty"`
+	Skills      []string                 `json:"skills,omitempty"`
+}
+
+type UpdateRequest struct {
+	Name        *string                `json:"name,omitempty"`
+	Description *string                `json:"description,omitempty"`
+	Metadata    map[string]interface{} `json:"metadata,omitempty"`
+}
+
+type UpdateContextRequest struct {
+	Messages []map[string]interface{} `json:"messages"`
+}
+
+type UpdateSkillsRequest struct {
+	Skills []string `json:"skills"`
+}
+
+type AddSkillsRequest struct {
+	Skills []string `json:"skills"`
+}
+
+type ListResponse struct {
+	Items []Subagent `json:"items"`
+}
+
+type ContextResponse struct {
+	Messages []map[string]interface{} `json:"messages"`
+}
+
+type SkillsResponse struct {
+	Skills []string `json:"skills"`
+}
+