feat: implement runtime reply service and refactor related components

2026-04-12 23:37:52 +08:00
parent 6499126f40
commit 93cda6ae82
9 changed files with 502 additions and 382 deletions
@@ -0,0 +1,92 @@
+package runtime
+
+import (
+	"context"
+	"encoding/json"
+	"fmt"
+	"strings"
+	"time"
+
+	"cs-agent/internal/ai/runtime/graphs"
+	"cs-agent/internal/models"
+	svc "cs-agent/internal/services"
+)
+
+type runtimeReplyExecutor struct{}
+
+func newRuntimeReplyExecutor() *runtimeReplyExecutor {
+	return &runtimeReplyExecutor{}
+}
+
+func (e *runtimeReplyExecutor) Run(ctx context.Context, conversation models.Conversation, message models.Message, aiAgent models.AIAgent, trace *aiReplyTraceData) (*Summary, error) {
+	aiConfig := svc.AIConfigService.Get(aiAgent.AIConfigID)
+	if aiConfig == nil {
+		return nil, fmt.Errorf("ai config is nil")
+	}
+	runtimeStartedAt := time.Now()
+	summary, err := Service.Run(ctx, Request{
+		Conversation: &conversation,
+		UserMessage:  &message,
+		AIAgent:      &aiAgent,
+		AIConfig:     aiConfig,
+	})
+	if trace != nil {
+		trace.RuntimeLatencyMs = time.Since(runtimeStartedAt).Milliseconds()
+		e.fillTraceFromSummary(trace, summary, err)
+	}
+	return summary, err
+}
+
+func (e *runtimeReplyExecutor) ResumePendingInterrupt(ctx context.Context, conversation models.Conversation, message models.Message, aiAgent models.AIAgent, pendingInterrupt *models.ConversationInterrupt, trace *aiReplyTraceData) (*Summary, error) {
+	if pendingInterrupt == nil {
+		return nil, nil
+	}
+	aiConfig := svc.AIConfigService.Get(aiAgent.AIConfigID)
+	if aiConfig == nil {
+		return nil, fmt.Errorf("ai config is nil")
+	}
+	runtimeStartedAt := time.Now()
+	if trace != nil {
+		trace.ResumeSource = "pending_interrupt"
+	}
+	summary, err := Service.Resume(ctx, ResumeRequest{
+		Conversation: &conversation,
+		AIAgent:      &aiAgent,
+		AIConfig:     aiConfig,
+		CheckPointID: strings.TrimSpace(pendingInterrupt.CheckPointID),
+		ResumeData: map[string]any{
+			strings.TrimSpace(pendingInterrupt.InterruptID): strings.TrimSpace(message.Content),
+		},
+	})
+	if trace != nil {
+		trace.RuntimeLatencyMs = time.Since(runtimeStartedAt).Milliseconds()
+		e.fillTraceFromSummary(trace, summary, err)
+	}
+	return summary, err
+}
+
+func (e *runtimeReplyExecutor) fillTraceFromSummary(trace *aiReplyTraceData, summary *Summary, runErr error) {
+	if trace == nil {
+		return
+	}
+	if runErr != nil {
+		trace.Status = "runtime_error"
+		trace.FinalAction = "error"
+		if summary != nil {
+			trace.Runtime = json.RawMessage(summary.TraceData)
+		}
+		return
+	}
+	trace.Status = "runtime_prepared"
+	trace.FinalAction = toRunLogFinalAction(summary)
+	if summary != nil && strings.TrimSpace(summary.TraceData) != "" {
+		trace.Runtime = json.RawMessage(summary.TraceData)
+	}
+}
+
+func expiredInterruptSummary() *Summary {
+	return &Summary{
+		Status:    "expired",
+		ReplyText: graphs.ConfirmationExpiredReply,
+	}
+}