feat: connect inbox confirmation workflow
This commit is contained in:
@@ -1,17 +1,33 @@
|
||||
package inbox
|
||||
|
||||
import (
|
||||
"errors"
|
||||
"net/http"
|
||||
"strconv"
|
||||
"time"
|
||||
|
||||
"github.com/gin-gonic/gin"
|
||||
"gorm.io/gorm"
|
||||
"senlinai-agent/backend/internal/httpx"
|
||||
"senlinai-agent/backend/internal/logic/auth"
|
||||
"senlinai-agent/backend/internal/models"
|
||||
)
|
||||
|
||||
type Handler struct {
|
||||
service *Service
|
||||
}
|
||||
|
||||
// InboxItemDTO 隔离 Gorm 自增主键,只通过公开 identity 关联项目和条目。
|
||||
type InboxItemDTO struct {
|
||||
ID string `json:"id"`
|
||||
ProjectID string `json:"projectId"`
|
||||
SourceType string `json:"sourceType"`
|
||||
Title string `json:"title"`
|
||||
Body string `json:"body"`
|
||||
Status string `json:"status"`
|
||||
CreatedAt time.Time `json:"createdAt"`
|
||||
UpdatedAt time.Time `json:"updatedAt"`
|
||||
}
|
||||
|
||||
func NewHandler(service *Service) *Handler {
|
||||
return &Handler{service: service}
|
||||
}
|
||||
@@ -23,68 +39,101 @@ func (h *Handler) Register(router gin.IRouter) {
|
||||
}
|
||||
|
||||
func (h *Handler) capture(c *gin.Context) {
|
||||
userID, ok := auth.CurrentUserID(c)
|
||||
userID, ok := currentInboxUser(c)
|
||||
if !ok {
|
||||
c.JSON(http.StatusUnauthorized, gin.H{"error": "missing current user"})
|
||||
return
|
||||
}
|
||||
projectID, err := strconv.ParseUint(c.Param("id"), 10, 64)
|
||||
if err != nil {
|
||||
c.JSON(http.StatusBadRequest, gin.H{"error": "invalid project id"})
|
||||
projectIdentity, ok := httpx.IdentityParam(c, "id")
|
||||
if !ok {
|
||||
return
|
||||
}
|
||||
var input struct {
|
||||
SourceType string `json:"source_type"`
|
||||
SourceType string `json:"sourceType"`
|
||||
Title string `json:"title"`
|
||||
Body string `json:"body"`
|
||||
}
|
||||
if err := c.ShouldBindJSON(&input); err != nil {
|
||||
c.JSON(http.StatusBadRequest, gin.H{"error": "invalid request"})
|
||||
httpx.Error(c, http.StatusBadRequest, "invalid_request", "请求参数无效")
|
||||
return
|
||||
}
|
||||
item, err := h.service.Capture(CaptureInput{ProjectID: uint(projectID), UserID: userID, SourceType: input.SourceType, Title: input.Title, Body: input.Body})
|
||||
item, err := h.service.Capture(CaptureInput{
|
||||
ProjectIdentity: projectIdentity, UserID: userID, SourceType: input.SourceType, Title: input.Title, Body: input.Body,
|
||||
})
|
||||
if err != nil {
|
||||
c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
|
||||
writeInboxError(c, err)
|
||||
return
|
||||
}
|
||||
c.JSON(http.StatusCreated, item)
|
||||
c.JSON(http.StatusCreated, inboxItemDTO(*item))
|
||||
}
|
||||
|
||||
func (h *Handler) analyze(c *gin.Context) {
|
||||
userID, ok := auth.CurrentUserID(c)
|
||||
userID, ok := currentInboxUser(c)
|
||||
if !ok {
|
||||
c.JSON(http.StatusUnauthorized, gin.H{"error": "missing current user"})
|
||||
return
|
||||
}
|
||||
itemID, err := strconv.ParseUint(c.Param("id"), 10, 64)
|
||||
if err != nil {
|
||||
c.JSON(http.StatusBadRequest, gin.H{"error": "invalid inbox id"})
|
||||
itemIdentity, ok := httpx.IdentityParam(c, "id")
|
||||
if !ok {
|
||||
return
|
||||
}
|
||||
suggestions, err := h.service.Analyze(uint(itemID), userID)
|
||||
suggestions, err := h.service.Analyze(itemIdentity, userID)
|
||||
if err != nil {
|
||||
c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
|
||||
writeInboxError(c, err)
|
||||
return
|
||||
}
|
||||
c.JSON(http.StatusOK, gin.H{"suggestions": suggestions})
|
||||
}
|
||||
|
||||
func (h *Handler) confirm(c *gin.Context) {
|
||||
itemID, err := strconv.ParseUint(c.Param("id"), 10, 64)
|
||||
if err != nil {
|
||||
c.JSON(http.StatusBadRequest, gin.H{"error": "invalid inbox id"})
|
||||
userID, ok := currentInboxUser(c)
|
||||
if !ok {
|
||||
return
|
||||
}
|
||||
itemIdentity, ok := httpx.IdentityParam(c, "id")
|
||||
if !ok {
|
||||
return
|
||||
}
|
||||
var input struct {
|
||||
Suggestions []Suggestion `json:"suggestions"`
|
||||
SuggestionIDs []string `json:"suggestionIds"`
|
||||
}
|
||||
if err := c.ShouldBindJSON(&input); err != nil {
|
||||
c.JSON(http.StatusBadRequest, gin.H{"error": "invalid request"})
|
||||
httpx.Error(c, http.StatusBadRequest, "invalid_request", "请求参数无效")
|
||||
return
|
||||
}
|
||||
if err := h.service.Confirm(uint(itemID), input.Suggestions); err != nil {
|
||||
c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
|
||||
result, err := h.service.Confirm(itemIdentity, userID, input.SuggestionIDs)
|
||||
if err != nil {
|
||||
writeInboxError(c, err)
|
||||
return
|
||||
}
|
||||
c.Status(http.StatusNoContent)
|
||||
c.JSON(http.StatusOK, result)
|
||||
}
|
||||
|
||||
func currentInboxUser(c *gin.Context) (uint, bool) {
|
||||
userID, ok := auth.CurrentUserID(c)
|
||||
if !ok {
|
||||
httpx.Error(c, http.StatusUnauthorized, "unauthorized", "未登录或登录已失效")
|
||||
return 0, false
|
||||
}
|
||||
return userID, true
|
||||
}
|
||||
|
||||
func inboxItemDTO(item models.SenlinAgentInboxItem) InboxItemDTO {
|
||||
return InboxItemDTO{
|
||||
ID: item.Identity, ProjectID: item.ProjectIdentity, SourceType: item.SourceType,
|
||||
Title: item.Title, Body: item.Body, Status: item.Status,
|
||||
CreatedAt: item.CreatedAt.UTC(), UpdatedAt: item.UpdatedAt.UTC(),
|
||||
}
|
||||
}
|
||||
|
||||
func writeInboxError(c *gin.Context, err error) {
|
||||
switch {
|
||||
case errors.Is(err, gorm.ErrRecordNotFound):
|
||||
httpx.Error(c, http.StatusNotFound, "not_found", "Inbox 条目或项目不存在")
|
||||
case errors.Is(err, ErrInboxAlreadyConfirmed):
|
||||
httpx.Error(c, http.StatusConflict, "conflict", "该 Inbox 条目已经确认处理")
|
||||
case errors.Is(err, ErrProjectAndUserRequired), errors.Is(err, ErrSourceTypeRequired),
|
||||
errors.Is(err, ErrSuggestionSelectionRequired), errors.Is(err, ErrSuggestionNotFound):
|
||||
httpx.Error(c, http.StatusBadRequest, "invalid_request", "请求参数或建议选择无效")
|
||||
default:
|
||||
httpx.Error(c, http.StatusInternalServerError, "internal_error", "Inbox 操作失败,请稍后重试")
|
||||
}
|
||||
}
|
||||
|
||||
@@ -2,25 +2,43 @@ package inbox
|
||||
|
||||
import (
|
||||
"errors"
|
||||
"strings"
|
||||
|
||||
"github.com/google/uuid"
|
||||
"gorm.io/gorm"
|
||||
"gorm.io/gorm/clause"
|
||||
"senlinai-agent/backend/internal/models"
|
||||
)
|
||||
|
||||
var (
|
||||
ErrProjectAndUserRequired = errors.New("project and user are required")
|
||||
ErrSourceTypeRequired = errors.New("source type is required")
|
||||
ErrInvalidSuggestion = errors.New("invalid inbox suggestion")
|
||||
ErrSuggestionSelectionRequired = errors.New("suggestion selection is required")
|
||||
ErrSuggestionNotFound = errors.New("selected suggestion was not saved for inbox item")
|
||||
ErrInboxAlreadyConfirmed = errors.New("inbox item already confirmed")
|
||||
)
|
||||
|
||||
type CaptureInput struct {
|
||||
ProjectID uint
|
||||
UserID uint
|
||||
SourceType string
|
||||
Title string
|
||||
Body string
|
||||
ProjectIdentity string
|
||||
UserID uint
|
||||
SourceType string
|
||||
Title string
|
||||
Body string
|
||||
}
|
||||
|
||||
// Suggestion 是分析接口返回的候选草稿;ID 由服务端保存后生成,确认接口不再接受 kind/title/body。
|
||||
type Suggestion struct {
|
||||
ID string `json:"id"`
|
||||
Kind string `json:"kind"`
|
||||
Title string `json:"title"`
|
||||
Body string `json:"body"`
|
||||
}
|
||||
|
||||
type ConfirmResult struct {
|
||||
CreatedCount int `json:"createdCount"`
|
||||
}
|
||||
|
||||
type Analyzer interface {
|
||||
Analyze(item models.SenlinAgentInboxItem, userID uint) ([]Suggestion, error)
|
||||
}
|
||||
@@ -30,72 +48,239 @@ type StaticAnalyzer struct {
|
||||
}
|
||||
|
||||
func (a StaticAnalyzer) Analyze(item models.SenlinAgentInboxItem, userID uint) ([]Suggestion, error) {
|
||||
return a.Suggestions, nil
|
||||
if len(a.Suggestions) > 0 {
|
||||
return a.Suggestions, nil
|
||||
}
|
||||
subject := suggestionSubject(item)
|
||||
body := strings.TrimSpace(item.Body)
|
||||
if body == "" {
|
||||
body = strings.TrimSpace(item.Title)
|
||||
}
|
||||
// 默认分析器只把用户主动收集的内容整理成可审阅草稿;它不调用 provider,也不创建正式对象。
|
||||
return []Suggestion{
|
||||
{Kind: "task", Title: "跟进:" + subject, Body: body},
|
||||
{Kind: "note", Title: subject + "整理笔记", Body: body},
|
||||
{Kind: "source", Title: subject + "背景资料", Body: body},
|
||||
}, nil
|
||||
}
|
||||
|
||||
type Service struct {
|
||||
analyzer Analyzer
|
||||
db *gorm.DB
|
||||
}
|
||||
|
||||
func NewService(analyzer Analyzer) *Service {
|
||||
return &Service{analyzer: analyzer}
|
||||
func NewService(analyzer Analyzer, databases ...*gorm.DB) *Service {
|
||||
var database *gorm.DB
|
||||
if len(databases) > 0 {
|
||||
database = databases[0]
|
||||
}
|
||||
return &Service{analyzer: analyzer, db: database}
|
||||
}
|
||||
|
||||
func (s *Service) database() *gorm.DB {
|
||||
if s.db != nil {
|
||||
return s.db
|
||||
}
|
||||
return models.DBService
|
||||
}
|
||||
|
||||
// Capture 先以 owner + project identity 校验项目边界,再创建归属于当前用户的原始 Inbox 条目。
|
||||
func (s *Service) Capture(input CaptureInput) (*models.SenlinAgentInboxItem, error) {
|
||||
if input.ProjectID == 0 || input.UserID == 0 {
|
||||
return nil, errors.New("project and user are required")
|
||||
if strings.TrimSpace(input.ProjectIdentity) == "" || input.UserID == 0 {
|
||||
return nil, ErrProjectAndUserRequired
|
||||
}
|
||||
if input.SourceType == "" {
|
||||
return nil, errors.New("source type is required")
|
||||
sourceType := strings.TrimSpace(input.SourceType)
|
||||
if sourceType == "" {
|
||||
return nil, ErrSourceTypeRequired
|
||||
}
|
||||
item := &models.SenlinAgentInboxItem{
|
||||
ProjectID: input.ProjectID,
|
||||
CreatedBy: input.UserID,
|
||||
SourceType: input.SourceType,
|
||||
Title: input.Title,
|
||||
Body: input.Body,
|
||||
Status: "open",
|
||||
}
|
||||
return item, models.DBService.Create(item).Error
|
||||
}
|
||||
|
||||
func (s *Service) Analyze(itemID uint, userID uint) ([]Suggestion, error) {
|
||||
var item models.SenlinAgentInboxItem
|
||||
if err := models.DBService.First(&item, itemID).Error; err != nil {
|
||||
return nil, err
|
||||
}
|
||||
if s.analyzer == nil {
|
||||
return []Suggestion{}, nil
|
||||
}
|
||||
return s.analyzer.Analyze(item, userID)
|
||||
}
|
||||
|
||||
func (s *Service) Confirm(itemID uint, selected []Suggestion) error {
|
||||
return models.DBService.Transaction(func(tx *gorm.DB) error {
|
||||
var item models.SenlinAgentInboxItem
|
||||
if err := tx.First(&item, itemID).Error; err != nil {
|
||||
err := s.database().Transaction(func(tx *gorm.DB) error {
|
||||
var project models.SenlinAgentProject
|
||||
if err := tx.Where("owner_id = ? AND identity = ?", input.UserID, input.ProjectIdentity).First(&project).Error; err != nil {
|
||||
return err
|
||||
}
|
||||
sourceInboxItemID := item.ID
|
||||
// Inbox 建议只有在用户确认后才创建正式对象,并把来源 ID 写入每个对象以保留可追溯性。
|
||||
for _, suggestion := range selected {
|
||||
switch suggestion.Kind {
|
||||
case "task":
|
||||
if err := tx.Create(&models.SenlinAgentTask{ProjectID: item.ProjectID, CreatedBy: item.CreatedBy, SourceInboxItemID: &sourceInboxItemID, Title: suggestion.Title, Description: suggestion.Body}).Error; err != nil {
|
||||
return err
|
||||
}
|
||||
case "note":
|
||||
if err := tx.Create(&models.SenlinAgentNote{ProjectID: item.ProjectID, CreatedBy: item.CreatedBy, SourceInboxItemID: &sourceInboxItemID, Title: suggestion.Title, Markdown: suggestion.Body}).Error; err != nil {
|
||||
return err
|
||||
}
|
||||
case "source":
|
||||
if err := tx.Create(&models.SenlinAgentSource{ProjectID: item.ProjectID, CreatedBy: item.CreatedBy, SourceInboxItemID: &sourceInboxItemID, Kind: "link", Title: suggestion.Title, ContentText: suggestion.Body}).Error; err != nil {
|
||||
return err
|
||||
}
|
||||
default:
|
||||
return errors.New("unsupported suggestion kind")
|
||||
}
|
||||
item = models.SenlinAgentInboxItem{
|
||||
ProjectID: project.ID, ProjectIdentity: project.Identity, CreatedBy: input.UserID,
|
||||
SourceType: sourceType, Title: strings.TrimSpace(input.Title), Body: strings.TrimSpace(input.Body), Status: "open",
|
||||
}
|
||||
return tx.Model(&item).Update("status", "processed").Error
|
||||
return tx.Create(&item).Error
|
||||
})
|
||||
return &item, err
|
||||
}
|
||||
|
||||
// Analyze 只保存候选草稿,不创建任何任务、笔记或资料;重新分析会原子替换该条目的旧草稿。
|
||||
func (s *Service) Analyze(itemIdentity string, userID uint) ([]Suggestion, error) {
|
||||
item, err := findOwnedInboxItem(s.database(), itemIdentity, userID, false)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
if item.Status != "open" {
|
||||
return nil, ErrInboxAlreadyConfirmed
|
||||
}
|
||||
generated := []Suggestion{}
|
||||
if s.analyzer != nil {
|
||||
generated, err = s.analyzer.Analyze(*item, userID)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
}
|
||||
normalized, err := normalizeSuggestions(generated)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
saved := make([]Suggestion, 0, len(normalized))
|
||||
err = s.database().Transaction(func(tx *gorm.DB) error {
|
||||
locked, err := findOwnedInboxItem(tx, itemIdentity, userID, true)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
if locked.Status != "open" {
|
||||
return ErrInboxAlreadyConfirmed
|
||||
}
|
||||
if err := tx.Where("inbox_item_id = ?", locked.ID).Delete(&models.SenlinAgentInboxSuggestion{}).Error; err != nil {
|
||||
return err
|
||||
}
|
||||
for _, suggestion := range normalized {
|
||||
draft := models.SenlinAgentInboxSuggestion{
|
||||
InboxItemID: locked.ID, InboxItemIdentity: locked.Identity, CreatedBy: userID,
|
||||
Kind: suggestion.Kind, Title: suggestion.Title, Body: suggestion.Body,
|
||||
}
|
||||
if err := tx.Create(&draft).Error; err != nil {
|
||||
return err
|
||||
}
|
||||
saved = append(saved, Suggestion{ID: draft.Identity, Kind: draft.Kind, Title: draft.Title, Body: draft.Body})
|
||||
}
|
||||
return nil
|
||||
})
|
||||
return saved, err
|
||||
}
|
||||
|
||||
// Confirm 只按服务端保存的建议 identity 创建对象;全部写入和 Inbox 状态变更位于同一事务中。
|
||||
func (s *Service) Confirm(itemIdentity string, userID uint, selectedSuggestionIdentities []string) (ConfirmResult, error) {
|
||||
selected, err := normalizeSelectedIdentities(selectedSuggestionIdentities)
|
||||
if err != nil {
|
||||
return ConfirmResult{}, err
|
||||
}
|
||||
result := ConfirmResult{}
|
||||
err = s.database().Transaction(func(tx *gorm.DB) error {
|
||||
item, err := findOwnedInboxItem(tx, itemIdentity, userID, true)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
if item.Status != "open" {
|
||||
return ErrInboxAlreadyConfirmed
|
||||
}
|
||||
|
||||
var saved []models.SenlinAgentInboxSuggestion
|
||||
if err := tx.Where("inbox_item_id = ? AND created_by = ? AND identity IN ?", item.ID, userID, selected).Find(&saved).Error; err != nil {
|
||||
return err
|
||||
}
|
||||
if len(saved) != len(selected) {
|
||||
return ErrSuggestionNotFound
|
||||
}
|
||||
byIdentity := make(map[string]models.SenlinAgentInboxSuggestion, len(saved))
|
||||
for _, suggestion := range saved {
|
||||
byIdentity[suggestion.Identity] = suggestion
|
||||
}
|
||||
for _, identity := range selected {
|
||||
suggestion, ok := byIdentity[identity]
|
||||
if !ok {
|
||||
return ErrSuggestionNotFound
|
||||
}
|
||||
if err := createConfirmedObject(tx, *item, suggestion); err != nil {
|
||||
return err
|
||||
}
|
||||
result.CreatedCount++
|
||||
}
|
||||
return tx.Model(item).Update("status", "processed").Error
|
||||
})
|
||||
return result, err
|
||||
}
|
||||
|
||||
func findOwnedInboxItem(tx *gorm.DB, identity string, userID uint, lock bool) (*models.SenlinAgentInboxItem, error) {
|
||||
var item models.SenlinAgentInboxItem
|
||||
query := tx.Model(&models.SenlinAgentInboxItem{}).
|
||||
Joins("JOIN senlin_agent_projects ON senlin_agent_projects.id = senlin_agent_inbox_items.project_id").
|
||||
Where("senlin_agent_inbox_items.identity = ? AND senlin_agent_inbox_items.created_by = ? AND senlin_agent_projects.owner_id = ?", identity, userID, userID)
|
||||
if lock {
|
||||
query = query.Clauses(clause.Locking{Strength: "UPDATE"})
|
||||
}
|
||||
if err := query.First(&item).Error; err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return &item, nil
|
||||
}
|
||||
|
||||
func normalizeSuggestions(suggestions []Suggestion) ([]Suggestion, error) {
|
||||
result := make([]Suggestion, 0, len(suggestions))
|
||||
for _, suggestion := range suggestions {
|
||||
kind := strings.ToLower(strings.TrimSpace(suggestion.Kind))
|
||||
title := strings.TrimSpace(suggestion.Title)
|
||||
if title == "" || (kind != "task" && kind != "note" && kind != "source") {
|
||||
return nil, ErrInvalidSuggestion
|
||||
}
|
||||
result = append(result, Suggestion{Kind: kind, Title: title, Body: strings.TrimSpace(suggestion.Body)})
|
||||
}
|
||||
return result, nil
|
||||
}
|
||||
|
||||
func suggestionSubject(item models.SenlinAgentInboxItem) string {
|
||||
value := strings.TrimSpace(item.Title)
|
||||
if value == "" {
|
||||
value = strings.TrimSpace(item.Body)
|
||||
}
|
||||
if value == "" {
|
||||
return "Inbox 内容"
|
||||
}
|
||||
runes := []rune(value)
|
||||
if len(runes) > 40 {
|
||||
return string(runes[:40]) + "…"
|
||||
}
|
||||
return value
|
||||
}
|
||||
|
||||
func normalizeSelectedIdentities(values []string) ([]string, error) {
|
||||
if len(values) == 0 {
|
||||
return nil, ErrSuggestionSelectionRequired
|
||||
}
|
||||
seen := make(map[string]struct{}, len(values))
|
||||
result := make([]string, 0, len(values))
|
||||
for _, value := range values {
|
||||
identity, err := uuid.Parse(strings.TrimSpace(value))
|
||||
if err != nil || identity.Version() != 7 || identity.Variant() != uuid.RFC4122 {
|
||||
return nil, ErrSuggestionNotFound
|
||||
}
|
||||
normalized := identity.String()
|
||||
if _, exists := seen[normalized]; exists {
|
||||
continue
|
||||
}
|
||||
seen[normalized] = struct{}{}
|
||||
result = append(result, normalized)
|
||||
}
|
||||
return result, nil
|
||||
}
|
||||
|
||||
func createConfirmedObject(tx *gorm.DB, item models.SenlinAgentInboxItem, suggestion models.SenlinAgentInboxSuggestion) error {
|
||||
sourceInboxItemID := item.ID
|
||||
switch suggestion.Kind {
|
||||
case "task":
|
||||
return tx.Create(&models.SenlinAgentTask{
|
||||
ProjectID: item.ProjectID, ProjectIdentity: item.ProjectIdentity, CreatedBy: item.CreatedBy,
|
||||
SourceInboxItemID: &sourceInboxItemID, Title: suggestion.Title, Description: suggestion.Body, Status: "open",
|
||||
}).Error
|
||||
case "note":
|
||||
return tx.Create(&models.SenlinAgentNote{
|
||||
ProjectID: item.ProjectID, ProjectIdentity: item.ProjectIdentity, CreatedBy: item.CreatedBy,
|
||||
SourceInboxItemID: &sourceInboxItemID, Title: suggestion.Title, Markdown: suggestion.Body,
|
||||
}).Error
|
||||
case "source":
|
||||
// 文本型 Inbox 资料不是上传文件,不构造或伪造任何本地路径。
|
||||
return tx.Create(&models.SenlinAgentSource{
|
||||
ProjectID: item.ProjectID, ProjectIdentity: item.ProjectIdentity, CreatedBy: item.CreatedBy,
|
||||
SourceInboxItemID: &sourceInboxItemID, Kind: "text", Title: suggestion.Title, ContentText: suggestion.Body,
|
||||
}).Error
|
||||
default:
|
||||
return ErrInvalidSuggestion
|
||||
}
|
||||
}
|
||||
|
||||
@@ -1,56 +1,369 @@
|
||||
package inbox
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"encoding/json"
|
||||
"errors"
|
||||
"fmt"
|
||||
"net/http"
|
||||
"net/http/httptest"
|
||||
"testing"
|
||||
|
||||
"github.com/gin-gonic/gin"
|
||||
"github.com/glebarez/sqlite"
|
||||
"github.com/google/uuid"
|
||||
"github.com/stretchr/testify/require"
|
||||
"gorm.io/gorm"
|
||||
"senlinai-agent/backend/internal/config"
|
||||
"senlinai-agent/backend/internal/httpx"
|
||||
"senlinai-agent/backend/internal/models"
|
||||
)
|
||||
|
||||
func TestAnalyzeReturnsSuggestionsWithoutCreatingObjects(t *testing.T) {
|
||||
database := newTestDB(t)
|
||||
service := NewService(StaticAnalyzer{
|
||||
Suggestions: []Suggestion{{Kind: "task", Title: "跟进报价", Body: "联系客户确认报价"}},
|
||||
})
|
||||
item, err := service.Capture(CaptureInput{ProjectID: 1, UserID: 1, SourceType: "text", Body: "需要跟进报价"})
|
||||
require.NoError(t, err)
|
||||
|
||||
suggestions, err := service.Analyze(item.ID, 1)
|
||||
|
||||
require.NoError(t, err)
|
||||
require.Len(t, suggestions, 1)
|
||||
var count int64
|
||||
require.NoError(t, database.Model(&models.SenlinAgentTask{}).Count(&count).Error)
|
||||
require.Equal(t, int64(0), count)
|
||||
type inboxTestFixture struct {
|
||||
database *gorm.DB
|
||||
owner models.SenlinAgentUser
|
||||
other models.SenlinAgentUser
|
||||
project models.SenlinAgentProject
|
||||
}
|
||||
|
||||
func TestConfirmCreatesSelectedObjectsAndKeepsInboxItem(t *testing.T) {
|
||||
database := newTestDB(t)
|
||||
service := NewService(StaticAnalyzer{
|
||||
Suggestions: []Suggestion{{Kind: "task", Title: "跟进报价", Body: "联系客户确认报价"}},
|
||||
type inboxSuggestionResponse struct {
|
||||
ID string `json:"id"`
|
||||
Kind string `json:"kind"`
|
||||
Title string `json:"title"`
|
||||
Body string `json:"body"`
|
||||
}
|
||||
|
||||
type inboxAnalyzeResponse struct {
|
||||
Suggestions []inboxSuggestionResponse `json:"suggestions"`
|
||||
}
|
||||
|
||||
func TestStaticAnalyzerBuildsReviewableDraftsFromInboxContent(t *testing.T) {
|
||||
item := models.SenlinAgentInboxItem{Title: "客户访谈", Body: "客户希望下周确认交付计划。"}
|
||||
|
||||
suggestions, err := (StaticAnalyzer{}).Analyze(item, 7)
|
||||
|
||||
require.NoError(t, err)
|
||||
require.Len(t, suggestions, 3)
|
||||
require.Equal(t, []string{"task", "note", "source"}, []string{suggestions[0].Kind, suggestions[1].Kind, suggestions[2].Kind})
|
||||
for _, suggestion := range suggestions {
|
||||
require.Contains(t, suggestion.Title, "客户访谈")
|
||||
require.Equal(t, item.Body, suggestion.Body)
|
||||
}
|
||||
}
|
||||
|
||||
func TestInboxCaptureUsesOwnedUUIDv7AndCamelCaseDTO(t *testing.T) {
|
||||
fixture := newInboxTestFixture(t)
|
||||
router := fixture.router(fixture.owner.ID, StaticAnalyzer{})
|
||||
|
||||
response := performInboxJSON(t, router, http.MethodPost, "/api/v1/projects/"+fixture.project.Identity+"/inbox", gin.H{
|
||||
"sourceType": "text",
|
||||
"title": "客户回访",
|
||||
"body": "整理会议记录",
|
||||
})
|
||||
item, err := service.Capture(CaptureInput{ProjectID: 1, UserID: 1, SourceType: "text", Body: "需要跟进报价"})
|
||||
require.NoError(t, err)
|
||||
|
||||
suggestions, err := service.Analyze(item.ID, 1)
|
||||
require.NoError(t, err)
|
||||
err = service.Confirm(item.ID, suggestions)
|
||||
require.Equal(t, http.StatusCreated, response.Code)
|
||||
var payload map[string]any
|
||||
require.NoError(t, json.Unmarshal(response.Body.Bytes(), &payload))
|
||||
require.ElementsMatch(t, []string{"id", "projectId", "sourceType", "title", "body", "status", "createdAt", "updatedAt"}, inboxMapKeys(payload))
|
||||
require.Equal(t, fixture.project.Identity, payload["projectId"])
|
||||
require.Equal(t, "text", payload["sourceType"])
|
||||
require.Equal(t, "open", payload["status"])
|
||||
requireUUIDv7(t, payload["id"].(string))
|
||||
|
||||
var item models.SenlinAgentInboxItem
|
||||
require.NoError(t, fixture.database.Where("identity = ?", payload["id"]).First(&item).Error)
|
||||
require.Equal(t, fixture.owner.ID, item.CreatedBy)
|
||||
require.Equal(t, fixture.owner.Identity, item.CreatedByIdentity)
|
||||
require.Equal(t, fixture.project.ID, item.ProjectID)
|
||||
require.Equal(t, fixture.project.Identity, item.ProjectIdentity)
|
||||
}
|
||||
|
||||
func TestInboxCaptureRejectsNumericAndUnownedProjectIdentities(t *testing.T) {
|
||||
fixture := newInboxTestFixture(t)
|
||||
router := fixture.router(fixture.owner.ID, StaticAnalyzer{})
|
||||
|
||||
numeric := performInboxJSON(t, router, http.MethodPost, "/api/v1/projects/1/inbox", gin.H{
|
||||
"sourceType": "text", "body": "numeric path",
|
||||
})
|
||||
require.Equal(t, http.StatusBadRequest, numeric.Code)
|
||||
|
||||
otherProject := models.SenlinAgentProject{OwnerID: fixture.other.ID, Name: "Other", Identifier: "OTHER"}
|
||||
require.NoError(t, fixture.database.Create(&otherProject).Error)
|
||||
unowned := performInboxJSON(t, router, http.MethodPost, "/api/v1/projects/"+otherProject.Identity+"/inbox", gin.H{
|
||||
"sourceType": "text", "body": "not mine",
|
||||
})
|
||||
require.Equal(t, http.StatusNotFound, unowned.Code)
|
||||
|
||||
var count int64
|
||||
require.NoError(t, fixture.database.Model(&models.SenlinAgentInboxItem{}).Count(&count).Error)
|
||||
require.Zero(t, count)
|
||||
}
|
||||
|
||||
func TestAnalyzeUsesInboxIdentityAndCreatesNoFormalObjects(t *testing.T) {
|
||||
fixture := newInboxTestFixture(t)
|
||||
item := fixture.createInbox(t, fixture.owner.ID, fixture.project.ID)
|
||||
router := fixture.router(fixture.owner.ID, StaticAnalyzer{Suggestions: []Suggestion{
|
||||
{Kind: "task", Title: "跟进报价", Body: "联系客户确认报价"},
|
||||
}})
|
||||
|
||||
response := performInboxJSON(t, router, http.MethodPost, "/api/v1/inbox/"+item.Identity+"/analyze", nil)
|
||||
|
||||
require.Equal(t, http.StatusOK, response.Code)
|
||||
analysis := decodeInboxAnalysis(t, response)
|
||||
require.Len(t, analysis.Suggestions, 1)
|
||||
require.Equal(t, "task", analysis.Suggestions[0].Kind)
|
||||
require.Equal(t, "跟进报价", analysis.Suggestions[0].Title)
|
||||
requireUUIDv7(t, analysis.Suggestions[0].ID)
|
||||
requireFormalObjectCounts(t, fixture.database, 0, 0, 0)
|
||||
|
||||
require.NoError(t, err)
|
||||
var tasks []models.SenlinAgentTask
|
||||
require.NoError(t, database.Find(&tasks).Error)
|
||||
require.Len(t, tasks, 1)
|
||||
require.Equal(t, "跟进报价", tasks[0].Title)
|
||||
require.NotNil(t, tasks[0].SourceInboxItemID)
|
||||
require.Equal(t, item.ID, *tasks[0].SourceInboxItemID)
|
||||
var reloaded models.SenlinAgentInboxItem
|
||||
require.NoError(t, database.First(&reloaded, item.ID).Error)
|
||||
require.NoError(t, fixture.database.First(&reloaded, item.ID).Error)
|
||||
require.Equal(t, "open", reloaded.Status)
|
||||
}
|
||||
|
||||
func TestConfirmCreatesOnlySelectedSavedSuggestionsWithInboxIdentity(t *testing.T) {
|
||||
fixture := newInboxTestFixture(t)
|
||||
item := fixture.createInbox(t, fixture.owner.ID, fixture.project.ID)
|
||||
router := fixture.router(fixture.owner.ID, StaticAnalyzer{Suggestions: []Suggestion{
|
||||
{Kind: "task", Title: "跟进报价", Body: "联系客户确认报价"},
|
||||
{Kind: "note", Title: "会议纪要", Body: "保留讨论结论"},
|
||||
{Kind: "source", Title: "背景资料", Body: "这是一段收集内容,不是上传文件"},
|
||||
{Kind: "task", Title: "不创建的任务", Body: "未被勾选"},
|
||||
}})
|
||||
analysisResponse := performInboxJSON(t, router, http.MethodPost, "/api/v1/inbox/"+item.Identity+"/analyze", nil)
|
||||
require.Equal(t, http.StatusOK, analysisResponse.Code)
|
||||
analysis := decodeInboxAnalysis(t, analysisResponse)
|
||||
require.Len(t, analysis.Suggestions, 4)
|
||||
selectedIDs := []string{analysis.Suggestions[0].ID, analysis.Suggestions[1].ID, analysis.Suggestions[2].ID}
|
||||
|
||||
response := performInboxJSON(t, router, http.MethodPost, "/api/v1/inbox/"+item.Identity+"/confirm", gin.H{
|
||||
"suggestionIds": selectedIDs,
|
||||
})
|
||||
|
||||
require.Equal(t, http.StatusOK, response.Code)
|
||||
var result struct {
|
||||
CreatedCount int `json:"createdCount"`
|
||||
}
|
||||
require.NoError(t, json.Unmarshal(response.Body.Bytes(), &result))
|
||||
require.Equal(t, 3, result.CreatedCount)
|
||||
requireFormalObjectCounts(t, fixture.database, 1, 1, 1)
|
||||
|
||||
var task models.SenlinAgentTask
|
||||
require.NoError(t, fixture.database.First(&task).Error)
|
||||
require.Equal(t, "跟进报价", task.Title)
|
||||
require.Equal(t, fixture.project.ID, task.ProjectID)
|
||||
require.Equal(t, fixture.owner.ID, task.CreatedBy)
|
||||
requireSourceInboxIdentity(t, item, task.SourceInboxItemID, task.SourceInboxItemIdentity)
|
||||
|
||||
var note models.SenlinAgentNote
|
||||
require.NoError(t, fixture.database.First(¬e).Error)
|
||||
require.Equal(t, "会议纪要", note.Title)
|
||||
require.Equal(t, fixture.project.ID, note.ProjectID)
|
||||
require.Equal(t, fixture.owner.ID, note.CreatedBy)
|
||||
requireSourceInboxIdentity(t, item, note.SourceInboxItemID, note.SourceInboxItemIdentity)
|
||||
|
||||
var source models.SenlinAgentSource
|
||||
require.NoError(t, fixture.database.First(&source).Error)
|
||||
require.Equal(t, "背景资料", source.Title)
|
||||
require.Equal(t, "text", source.Kind)
|
||||
require.Empty(t, source.FilePath)
|
||||
require.Equal(t, "这是一段收集内容,不是上传文件", source.ContentText)
|
||||
require.Equal(t, fixture.project.ID, source.ProjectID)
|
||||
require.Equal(t, fixture.owner.ID, source.CreatedBy)
|
||||
requireSourceInboxIdentity(t, item, source.SourceInboxItemID, source.SourceInboxItemIdentity)
|
||||
|
||||
var skipped int64
|
||||
require.NoError(t, fixture.database.Model(&models.SenlinAgentTask{}).Where("title = ?", "不创建的任务").Count(&skipped).Error)
|
||||
require.Zero(t, skipped)
|
||||
var reloaded models.SenlinAgentInboxItem
|
||||
require.NoError(t, fixture.database.First(&reloaded, item.ID).Error)
|
||||
require.Equal(t, "processed", reloaded.Status)
|
||||
}
|
||||
|
||||
func TestConfirmRejectsClientForgedSuggestionsAndUnknownSuggestionIdentities(t *testing.T) {
|
||||
fixture := newInboxTestFixture(t)
|
||||
item := fixture.createInbox(t, fixture.owner.ID, fixture.project.ID)
|
||||
router := fixture.router(fixture.owner.ID, StaticAnalyzer{Suggestions: []Suggestion{
|
||||
{Kind: "task", Title: "服务端建议", Body: "只能创建这个建议"},
|
||||
}})
|
||||
analysisResponse := performInboxJSON(t, router, http.MethodPost, "/api/v1/inbox/"+item.Identity+"/analyze", nil)
|
||||
require.Equal(t, http.StatusOK, analysisResponse.Code)
|
||||
|
||||
forgedContent := performInboxJSON(t, router, http.MethodPost, "/api/v1/inbox/"+item.Identity+"/confirm", gin.H{
|
||||
"suggestions": []gin.H{{"kind": "task", "title": "客户端伪造任务", "body": "不可信内容"}},
|
||||
})
|
||||
require.Equal(t, http.StatusBadRequest, forgedContent.Code)
|
||||
|
||||
unknownIdentity, err := uuid.NewV7()
|
||||
require.NoError(t, err)
|
||||
unknownSuggestion := performInboxJSON(t, router, http.MethodPost, "/api/v1/inbox/"+item.Identity+"/confirm", gin.H{
|
||||
"suggestionIds": []string{unknownIdentity.String()},
|
||||
})
|
||||
require.Equal(t, http.StatusBadRequest, unknownSuggestion.Code)
|
||||
requireFormalObjectCounts(t, fixture.database, 0, 0, 0)
|
||||
}
|
||||
|
||||
func TestAnalyzeAndConfirmHideAnotherUsersInboxItem(t *testing.T) {
|
||||
fixture := newInboxTestFixture(t)
|
||||
item := fixture.createInbox(t, fixture.owner.ID, fixture.project.ID)
|
||||
ownerRouter := fixture.router(fixture.owner.ID, StaticAnalyzer{Suggestions: []Suggestion{
|
||||
{Kind: "task", Title: "私有建议", Body: "不得越权确认"},
|
||||
}})
|
||||
analysis := decodeInboxAnalysis(t, performInboxJSON(t, ownerRouter, http.MethodPost, "/api/v1/inbox/"+item.Identity+"/analyze", nil))
|
||||
require.Len(t, analysis.Suggestions, 1)
|
||||
otherRouter := fixture.router(fixture.other.ID, StaticAnalyzer{})
|
||||
|
||||
unauthorizedAnalyze := performInboxJSON(t, otherRouter, http.MethodPost, "/api/v1/inbox/"+item.Identity+"/analyze", nil)
|
||||
require.Equal(t, http.StatusNotFound, unauthorizedAnalyze.Code)
|
||||
unauthorizedConfirm := performInboxJSON(t, otherRouter, http.MethodPost, "/api/v1/inbox/"+item.Identity+"/confirm", gin.H{
|
||||
"suggestionIds": []string{analysis.Suggestions[0].ID},
|
||||
})
|
||||
require.Equal(t, http.StatusNotFound, unauthorizedConfirm.Code)
|
||||
requireFormalObjectCounts(t, fixture.database, 0, 0, 0)
|
||||
}
|
||||
|
||||
func TestConfirmIsTransactionalWhenASelectedWriteFails(t *testing.T) {
|
||||
fixture := newInboxTestFixture(t)
|
||||
item := fixture.createInbox(t, fixture.owner.ID, fixture.project.ID)
|
||||
router := fixture.router(fixture.owner.ID, StaticAnalyzer{Suggestions: []Suggestion{
|
||||
{Kind: "task", Title: "事务任务", Body: "必须随失败回滚"},
|
||||
{Kind: "source", Title: "失败资料", Body: "模拟持久化失败"},
|
||||
}})
|
||||
analysis := decodeInboxAnalysis(t, performInboxJSON(t, router, http.MethodPost, "/api/v1/inbox/"+item.Identity+"/analyze", nil))
|
||||
require.Len(t, analysis.Suggestions, 2)
|
||||
|
||||
const callbackName = "test:fail_inbox_source_create"
|
||||
require.NoError(t, fixture.database.Callback().Create().Before("gorm:create").Register(callbackName, func(tx *gorm.DB) {
|
||||
if tx.Statement.Schema != nil && tx.Statement.Schema.Name == "SenlinAgentSource" {
|
||||
tx.AddError(errors.New("simulated source write failure"))
|
||||
}
|
||||
}))
|
||||
t.Cleanup(func() { _ = fixture.database.Callback().Create().Remove(callbackName) })
|
||||
|
||||
response := performInboxJSON(t, router, http.MethodPost, "/api/v1/inbox/"+item.Identity+"/confirm", gin.H{
|
||||
"suggestionIds": []string{analysis.Suggestions[0].ID, analysis.Suggestions[1].ID},
|
||||
})
|
||||
|
||||
require.Equal(t, http.StatusInternalServerError, response.Code)
|
||||
requireFormalObjectCounts(t, fixture.database, 0, 0, 0)
|
||||
var reloaded models.SenlinAgentInboxItem
|
||||
require.NoError(t, fixture.database.First(&reloaded, item.ID).Error)
|
||||
require.Equal(t, "open", reloaded.Status)
|
||||
}
|
||||
|
||||
func TestRepeatedConfirmReturnsConflictWithoutDuplicateObjects(t *testing.T) {
|
||||
fixture := newInboxTestFixture(t)
|
||||
item := fixture.createInbox(t, fixture.owner.ID, fixture.project.ID)
|
||||
router := fixture.router(fixture.owner.ID, StaticAnalyzer{Suggestions: []Suggestion{
|
||||
{Kind: "task", Title: "只创建一次", Body: "重复确认不能复制"},
|
||||
}})
|
||||
analysis := decodeInboxAnalysis(t, performInboxJSON(t, router, http.MethodPost, "/api/v1/inbox/"+item.Identity+"/analyze", nil))
|
||||
body := gin.H{"suggestionIds": []string{analysis.Suggestions[0].ID}}
|
||||
|
||||
first := performInboxJSON(t, router, http.MethodPost, "/api/v1/inbox/"+item.Identity+"/confirm", body)
|
||||
second := performInboxJSON(t, router, http.MethodPost, "/api/v1/inbox/"+item.Identity+"/confirm", body)
|
||||
|
||||
require.Equal(t, http.StatusOK, first.Code)
|
||||
require.Equal(t, http.StatusConflict, second.Code)
|
||||
var payload httpx.ErrorEnvelope
|
||||
require.NoError(t, json.Unmarshal(second.Body.Bytes(), &payload))
|
||||
require.Equal(t, "conflict", payload.Error.Code)
|
||||
requireFormalObjectCounts(t, fixture.database, 1, 0, 0)
|
||||
}
|
||||
|
||||
func (fixture inboxTestFixture) router(userID uint, analyzer Analyzer) http.Handler {
|
||||
return httpx.NewProtectedRouter(
|
||||
config.Config{Env: "test"},
|
||||
func(string) (uint, error) { return userID, nil },
|
||||
NewHandler(NewService(analyzer)),
|
||||
)
|
||||
}
|
||||
|
||||
func (fixture inboxTestFixture) createInbox(t *testing.T, userID, projectID uint) models.SenlinAgentInboxItem {
|
||||
t.Helper()
|
||||
item := models.SenlinAgentInboxItem{
|
||||
ProjectID: projectID, CreatedBy: userID, SourceType: "text", Title: "待整理", Body: "需要整理的原始内容", Status: "open",
|
||||
}
|
||||
require.NoError(t, fixture.database.Create(&item).Error)
|
||||
return item
|
||||
}
|
||||
|
||||
func newInboxTestFixture(t *testing.T) inboxTestFixture {
|
||||
t.Helper()
|
||||
database := newTestDB(t)
|
||||
owner := models.SenlinAgentUser{Email: t.Name() + "-owner@example.com", DisplayName: "Owner", PasswordHash: "hash"}
|
||||
other := models.SenlinAgentUser{Email: t.Name() + "-other@example.com", DisplayName: "Other", PasswordHash: "hash"}
|
||||
require.NoError(t, database.Create(&owner).Error)
|
||||
require.NoError(t, database.Create(&other).Error)
|
||||
project := models.SenlinAgentProject{OwnerID: owner.ID, Name: "Inbox Project", Identifier: "INBOX"}
|
||||
require.NoError(t, database.Create(&project).Error)
|
||||
return inboxTestFixture{database: database, owner: owner, other: other, project: project}
|
||||
}
|
||||
|
||||
func performInboxJSON(t *testing.T, router http.Handler, method, path string, payload any) *httptest.ResponseRecorder {
|
||||
t.Helper()
|
||||
var body bytes.Buffer
|
||||
if payload != nil {
|
||||
require.NoError(t, json.NewEncoder(&body).Encode(payload))
|
||||
}
|
||||
request := httptest.NewRequest(method, path, &body)
|
||||
request.Header.Set("Authorization", "Bearer test-token")
|
||||
if payload != nil {
|
||||
request.Header.Set("Content-Type", "application/json")
|
||||
}
|
||||
response := httptest.NewRecorder()
|
||||
router.ServeHTTP(response, request)
|
||||
return response
|
||||
}
|
||||
|
||||
func decodeInboxAnalysis(t *testing.T, response *httptest.ResponseRecorder) inboxAnalyzeResponse {
|
||||
t.Helper()
|
||||
require.Equal(t, http.StatusOK, response.Code, response.Body.String())
|
||||
var payload inboxAnalyzeResponse
|
||||
require.NoError(t, json.Unmarshal(response.Body.Bytes(), &payload))
|
||||
return payload
|
||||
}
|
||||
|
||||
func requireFormalObjectCounts(t *testing.T, database *gorm.DB, tasks, notes, sources int64) {
|
||||
t.Helper()
|
||||
for _, check := range []struct {
|
||||
model any
|
||||
want int64
|
||||
}{
|
||||
{model: &models.SenlinAgentTask{}, want: tasks},
|
||||
{model: &models.SenlinAgentNote{}, want: notes},
|
||||
{model: &models.SenlinAgentSource{}, want: sources},
|
||||
} {
|
||||
var count int64
|
||||
require.NoError(t, database.Model(check.model).Count(&count).Error)
|
||||
require.Equal(t, check.want, count)
|
||||
}
|
||||
}
|
||||
|
||||
func requireSourceInboxIdentity(t *testing.T, item models.SenlinAgentInboxItem, sourceID *uint, sourceIdentity *string) {
|
||||
t.Helper()
|
||||
require.NotNil(t, sourceID)
|
||||
require.Equal(t, item.ID, *sourceID)
|
||||
require.NotNil(t, sourceIdentity)
|
||||
require.Equal(t, item.Identity, *sourceIdentity)
|
||||
}
|
||||
|
||||
func requireUUIDv7(t *testing.T, value string) {
|
||||
t.Helper()
|
||||
parsed, err := uuid.Parse(value)
|
||||
require.NoError(t, err)
|
||||
require.Equal(t, uuid.Version(7), parsed.Version())
|
||||
require.Equal(t, uuid.RFC4122, parsed.Variant())
|
||||
}
|
||||
|
||||
func inboxMapKeys(value map[string]any) []string {
|
||||
keys := make([]string, 0, len(value))
|
||||
for key := range value {
|
||||
keys = append(keys, key)
|
||||
}
|
||||
return keys
|
||||
}
|
||||
|
||||
func newTestDB(t *testing.T) *gorm.DB {
|
||||
t.Helper()
|
||||
database, err := gorm.Open(sqlite.Open(fmt.Sprintf("file:%s?mode=memory&cache=shared", t.Name())), &gorm.Config{})
|
||||
|
||||
@@ -28,6 +28,16 @@ func (m *SenlinAgentInboxItem) BeforeCreate(tx *gorm.DB) error {
|
||||
return resolveIdentity(tx, &SenlinAgentUser{}, m.CreatedBy, &m.CreatedByIdentity)
|
||||
}
|
||||
|
||||
func (m *SenlinAgentInboxSuggestion) BeforeCreate(tx *gorm.DB) error {
|
||||
if err := ensureIdentity(&m.Identity); err != nil {
|
||||
return err
|
||||
}
|
||||
if err := resolveIdentity(tx, &SenlinAgentInboxItem{}, m.InboxItemID, &m.InboxItemIdentity); err != nil {
|
||||
return err
|
||||
}
|
||||
return resolveIdentity(tx, &SenlinAgentUser{}, m.CreatedBy, &m.CreatedByIdentity)
|
||||
}
|
||||
|
||||
func (m *SenlinAgentTask) BeforeCreate(tx *gorm.DB) error {
|
||||
if err := ensureIdentity(&m.Identity); err != nil {
|
||||
return err
|
||||
|
||||
21
backend/internal/models/inbox_suggestion.go
Normal file
21
backend/internal/models/inbox_suggestion.go
Normal file
@@ -0,0 +1,21 @@
|
||||
package models
|
||||
|
||||
import "time"
|
||||
|
||||
// SenlinAgentInboxSuggestion 保存分析阶段的候选草稿;它不是任务、笔记或资料,只有用户确认后才会创建正式对象。
|
||||
type SenlinAgentInboxSuggestion struct {
|
||||
ID uint `gorm:"primaryKey"`
|
||||
Identity string `gorm:"type:char(36);uniqueIndex"`
|
||||
InboxItemID uint `gorm:"index;not null"`
|
||||
InboxItemIdentity string `gorm:"type:char(36);index"`
|
||||
CreatedBy uint `gorm:"index;not null"`
|
||||
CreatedByIdentity string `gorm:"type:char(36);index"`
|
||||
Kind string `gorm:"not null"`
|
||||
Title string `gorm:"not null"`
|
||||
Body string `gorm:"type:text"`
|
||||
CreatedAt time.Time
|
||||
}
|
||||
|
||||
func (SenlinAgentInboxSuggestion) TableName() string {
|
||||
return "senlin_agent_inbox_suggestions"
|
||||
}
|
||||
@@ -27,6 +27,7 @@ func AutoMigrate(database *gorm.DB) error {
|
||||
&SenlinAgentUser{},
|
||||
&SenlinAgentProject{},
|
||||
&SenlinAgentInboxItem{},
|
||||
&SenlinAgentInboxSuggestion{},
|
||||
&SenlinAgentTask{},
|
||||
&SenlinAgentNote{},
|
||||
&SenlinAgentSource{},
|
||||
|
||||
Reference in New Issue
Block a user