feat(chat): ConversationService with title generation hook

2026-05-04 17:13:22 +08:00
parent 8e8edfe3ff
commit d1be9e6eb0
2 changed files with 636 additions and 0 deletions
@@ -0,0 +1,203 @@
+package chat
+
+import (
+	"context"
+	"log/slog"
+	"strings"
+	"time"
+
+	"github.com/google/uuid"
+
+	"github.com/yan1h/agent-coding-workflow/internal/audit"
+	"github.com/yan1h/agent-coding-workflow/internal/infra/errs"
+	"github.com/yan1h/agent-coding-workflow/internal/infra/llm"
+)
+
+// compile-time assertion
+var _ ConversationService = (*conversationService)(nil)
+
+type conversationService struct {
+	repo     Repository
+	registry *llm.Registry
+	auditor  audit.Recorder
+	log      *slog.Logger
+}
+
+// NewConversationService constructs a ConversationService.
+func NewConversationService(repo Repository, reg *llm.Registry, a audit.Recorder, log *slog.Logger) ConversationService {
+	return &conversationService{repo: repo, registry: reg, auditor: a, log: log}
+}
+
+// Create validates the model and optional template, then inserts a new conversation.
+func (s *conversationService) Create(ctx context.Context, userID uuid.UUID, in CreateConversationInput) (*Conversation, error) {
+	// Validate model exists and is enabled.
+	model, err := s.repo.GetLLMModel(ctx, in.ModelID)
+	if err != nil {
+		return nil, err
+	}
+	if !model.Enabled {
+		return nil, errs.New(errs.CodeChatModelNotFound, "model is disabled")
+	}
+
+	// Resolve system prompt: custom override > template content > empty string.
+	var systemPrompt string
+	if in.SystemPrompt != nil {
+		systemPrompt = *in.SystemPrompt
+	} else if in.TemplateID != nil {
+		tpl, err := s.repo.GetPromptTemplate(ctx, *in.TemplateID)
+		if err != nil {
+			return nil, err
+		}
+		// Check user-scoped template ownership.
+		if tpl.Scope == TemplateScopeUser && (tpl.OwnerID == nil || *tpl.OwnerID != userID) {
+			return nil, errs.New(errs.CodeForbidden, "template not accessible")
+		}
+		systemPrompt = tpl.Content
+	}
+
+	id, err := uuid.NewV7()
+	if err != nil {
+		return nil, errs.Wrap(err, errs.CodeInternal, "generate conversation id")
+	}
+
+	c, err := s.repo.InsertConversation(ctx, InsertConversationParams{
+		ID:           id,
+		UserID:       userID,
+		ProjectID:    in.ProjectID,
+		WorkspaceID:  in.WorkspaceID,
+		IssueID:      in.IssueID,
+		ModelID:      in.ModelID,
+		SystemPrompt: systemPrompt,
+		Title:        nil,
+	})
+	if err != nil {
+		return nil, err
+	}
+
+	s.tryAudit(ctx, audit.Entry{
+		UserID:     &userID,
+		Action:     "chat.conversation.created",
+		TargetType: "conversation",
+		TargetID:   c.ID.String(),
+	})
+	return c, nil
+}
+
+// Get returns a conversation and its most recent messages, enforcing ownership.
+func (s *conversationService) Get(ctx context.Context, userID, id uuid.UUID) (*Conversation, []Message, error) {
+	c, err := s.repo.GetConversation(ctx, id)
+	if err != nil {
+		return nil, nil, err
+	}
+	if c.UserID != userID {
+		return nil, nil, errs.New(errs.CodeForbidden, "not conversation owner")
+	}
+	msgs, err := s.repo.ListMessagesByConversation(ctx, id, nil, 50)
+	if err != nil {
+		return nil, nil, err
+	}
+	return c, msgs, nil
+}
+
+// List returns conversations for the given user, defaulting limit to 50.
+func (s *conversationService) List(ctx context.Context, userID uuid.UUID, f ConversationFilter) ([]Conversation, error) {
+	if f.Limit == 0 {
+		f.Limit = 50
+	}
+	return s.repo.ListConversationsByUser(ctx, userID, f)
+}
+
+// UpdateTitle manually sets a conversation title, enforcing ownership.
+func (s *conversationService) UpdateTitle(ctx context.Context, userID, id uuid.UUID, title string) error {
+	c, err := s.repo.GetConversation(ctx, id)
+	if err != nil {
+		return err
+	}
+	if c.UserID != userID {
+		return errs.New(errs.CodeForbidden, "not conversation owner")
+	}
+	return s.repo.SetConversationTitleManual(ctx, id, &title)
+}
+
+// SoftDelete marks a conversation as deleted, enforcing ownership.
+func (s *conversationService) SoftDelete(ctx context.Context, userID, id uuid.UUID) error {
+	c, err := s.repo.GetConversation(ctx, id)
+	if err != nil {
+		return err
+	}
+	if c.UserID != userID {
+		return errs.New(errs.CodeForbidden, "not conversation owner")
+	}
+	if err := s.repo.SoftDeleteConversation(ctx, id); err != nil {
+		return err
+	}
+	s.tryAudit(ctx, audit.Entry{
+		UserID:     &userID,
+		Action:     "chat.conversation.deleted",
+		TargetType: "conversation",
+		TargetID:   id.String(),
+	})
+	return nil
+}
+
+// GenerateTitle is NOT on the ConversationService interface.
+// It is called by the streamer (T14) via type assertion on the concrete type.
+// It runs in a fire-and-forget goroutine; errors are silently swallowed.
+func (s *conversationService) GenerateTitle(ctx context.Context, conversationID uuid.UUID) {
+	tCtx, cancel := context.WithTimeout(ctx, 30*time.Second)
+	defer cancel()
+
+	titleModel, err := s.repo.GetTitleGeneratorModel(tCtx)
+	if err != nil {
+		s.log.Warn("GenerateTitle: no title model", "err", err)
+		return
+	}
+
+	first, err := s.repo.GetFirstUserMessage(tCtx, conversationID)
+	if err != nil {
+		s.log.Warn("GenerateTitle: no first user message", "conversation_id", conversationID, "err", err)
+		return
+	}
+
+	client, err := s.registry.GetClient(tCtx, titleModel.EndpointID)
+	if err != nil {
+		s.log.Warn("GenerateTitle: get client failed", "endpoint_id", titleModel.EndpointID, "err", err)
+		return
+	}
+
+	req := llm.Request{
+		SystemPrompt: "用 ≤20 字概括下面对话主题，仅输出标题：",
+		Messages: []llm.Message{{
+			Role:   llm.RoleUser,
+			Blocks: []llm.ContentBlock{{Type: "text", Text: first.Content}},
+		}},
+		MaxOutputTokens: 64,
+	}
+
+	ch, err := client.Stream(tCtx, titleModel.ModelID, req)
+	if err != nil {
+		s.log.Warn("GenerateTitle: stream failed", "err", err)
+		return
+	}
+
+	var title string
+	for ev := range ch {
+		if ev.Type == llm.EventTextDelta {
+			title += ev.Text
+		}
+	}
+	title = strings.TrimSpace(title)
+	if title == "" {
+		return
+	}
+
+	if err := s.repo.UpdateConversationTitle(tCtx, conversationID, &title); err != nil {
+		s.log.Warn("GenerateTitle: update title failed", "conversation_id", conversationID, "err", err)
+	}
+}
+
+func (s *conversationService) tryAudit(ctx context.Context, e audit.Entry) {
+	if err := s.auditor.Record(ctx, e); err != nil {
+		s.log.Warn("audit record failed", "action", e.Action, "err", err)
+	}
+}