feat(*): 模型生成链路 - 完善 Agnes 结果兼容、价格展示与请求日志追踪

2026-08-28 06:53:57 +08:00
parent 2c02f5ef3d
commit bd0d0fd4fd
20 changed files with 671 additions and 108 deletions
--- a/backend/internal/service/task_api_call_log.go
+++ b/backend/internal/service/task_api_call_log.go
@@ -0,0 +1,93 @@
+package service
+
+import (
+	"context"
+	"errors"
+	"log"
+	"strings"
+	"time"
+
+	"infinite-canvas/backend/internal/model"
+)
+
+// ensureFailedProviderAttemptLogged fills the task-level audit gap only when no
+// provider log exists. This normally represents a preflight failure; the coarse
+// request/state variants also preserve visibility if primary log persistence failed.
+func (s *Service) ensureFailedProviderAttemptLogged(task model.Task, taskErr error) {
+	if s == nil || s.repo == nil || strings.TrimSpace(task.ID) == "" || taskErr == nil {
+		return
+	}
+	hasLog, err := s.repo.HasAPICallLogForTask(task.ID)
+	if err != nil {
+		log.Printf("provider attempt log lookup failed: task_id=%s error=%v", task.ID, err)
+		return
+	}
+	if hasLog {
+		return
+	}
+
+	metadata := providerAnalyticsContext{
+		UserID:            task.UserID,
+		TaskID:            task.ID,
+		TraceID:           task.TraceID,
+		RequestID:         task.RequestID,
+		BillingOrderID:    task.BillingOrderID,
+		Capability:        capabilityFromTaskType(task.Type),
+		Operation:         task.Operation,
+		Model:             task.Model,
+		ProviderRequestID: task.ProviderRequestID,
+	}
+	decodedTask := task
+	if decrypted, decryptErr := s.decryptTaskInputJSON(task.InputJSON); decryptErr == nil {
+		decodedTask.InputJSON = decrypted
+		if value, ok := withProviderAnalytics(context.Background(), s, decodedTask).Value(providerAnalyticsKey{}).(providerAnalyticsContext); ok {
+			metadata = value
+		}
+	}
+
+	now := time.Now()
+	startedAt := now
+	if task.StartedAt != nil && !task.StartedAt.IsZero() {
+		startedAt = *task.StartedAt
+	}
+	path := "/task/provider-preflight"
+	errorCode := "request_not_sent"
+	statusCode := 0
+	var httpErr providerHTTPError
+	if errors.As(taskErr, &httpErr) {
+		path = "/task/provider-request"
+		errorCode = "provider_request_unlogged"
+		statusCode = httpErr.StatusCode
+	} else if strings.TrimSpace(metadata.ProviderRequestID) != "" {
+		path = "/task/provider-state"
+		errorCode = "provider_state_unlogged"
+	}
+	callLog := model.ApiCallLog{
+		UserID:            metadata.UserID,
+		TraceID:           metadata.TraceID,
+		RequestID:         metadata.RequestID,
+		ChannelID:         metadata.ChannelID,
+		TaskID:            metadata.TaskID,
+		BillingOrderID:    metadata.BillingOrderID,
+		Source:            "backend-task",
+		Capability:        metadata.Capability,
+		Operation:         metadata.Operation,
+		RequestKind:       "create",
+		Billable:          false,
+		APIFormat:         "internal",
+		Method:            "INTERNAL",
+		Path:              path,
+		Model:             metadata.Model,
+		Status:            model.ApiCallStatusFailed,
+		StatusCode:        statusCode,
+		DurationMs:        now.Sub(startedAt).Milliseconds(),
+		ProviderRequestID: metadata.ProviderRequestID,
+		ErrorCode:         errorCode,
+		Error:             truncateRunes(s.UserFacingErrorMessage(taskErr), 2_000),
+		StartedAt:         startedAt,
+		CreatedAt:         now,
+	}
+	if err := s.LogAPICall(callLog); err != nil {
+		log.Printf("provider attempt log write failed: task_id=%s error=%v", task.ID, err)
+	}
+}