feat: enrich task record details

apps/api/internal/runner/pricing.go

@@ -16,7 +16,7 @@ type EstimateResult struct {
 }
 
 func (s *Service) Estimate(ctx context.Context, kind string, model string, body map[string]any, user *auth.User) (EstimateResult, error) {
-	candidates, err := s.store.ListModelCandidates(ctx, model, modelTypeFromKind(kind))
+	candidates, err := s.store.ListModelCandidates(ctx, model, modelTypeFromKind(kind), user)
 	if err != nil {
 		return EstimateResult{}, err
 	}
@@ -40,15 +40,15 @@ func (s *Service) Estimate(ctx context.Context, kind string, model string, body
 func (s *Service) billings(ctx context.Context, user *auth.User, kind string, body map[string]any, candidate store.RuntimeModelCandidate, response clients.Response, simulated bool) []any {
 	config := effectiveBillingConfig(candidate)
 	discount := effectiveDiscount(ctx, s.store, user, candidate)
-	if modelTypeFromKind(kind) == "chat" {
+	if isTextGenerationKind(kind) {
 		inputTokens := response.Usage.InputTokens
 		outputTokens := response.Usage.OutputTokens
 		if inputTokens == 0 && outputTokens == 0 {
 			inputTokens = estimateRequestTokens(body)
 			outputTokens = 1
 		}
-		inputAmount := roundPrice(float64(inputTokens) / 1000 * price(config, "textInputPer1k") * discount)
-		outputAmount := roundPrice(float64(outputTokens) / 1000 * price(config, "textOutputPer1k") * discount)
+		inputAmount := roundPrice(float64(inputTokens) / 1000 * resourcePrice(config, "text", "textInputPer1k", "inputTokenPrice", "basePrice") * discount)
+		outputAmount := roundPrice(float64(outputTokens) / 1000 * resourcePrice(config, "text", "textOutputPer1k", "outputTokenPrice", "basePrice") * discount)
 		return []any{
 			billingLine(candidate, "text_input", "1k_tokens", inputTokens, inputAmount, discount, simulated),
 			billingLine(candidate, "text_output", "1k_tokens", outputTokens, outputAmount, discount, simulated),
@@ -59,13 +59,19 @@ func (s *Service) billings(ctx context.Context, user *auth.User, kind string, bo
 		count = 1
 	}
 	resource := "image"
+	unit := "image"
 	baseKey := "imageBase"
 	if kind == "images.edits" {
 		resource = "image_edit"
 		baseKey = "editBase"
 	}
-	amount := float64(count) * price(config, baseKey) * weighted(config, "qualityWeights", stringFromMap(body, "quality")) * weighted(config, "sizeWeights", stringFromMap(body, "size")) * discount
-	return []any{billingLine(candidate, resource, "image", count, roundPrice(amount), discount, simulated)}
+	if kind == "videos.generations" {
+		resource = "video"
+		unit = "video"
+		baseKey = "videoBase"
+	}
+	amount := float64(count) * resourcePrice(config, resource, baseKey, "basePrice") * resourceWeight(config, resource, "qualityWeights", stringFromMap(body, "quality")) * resourceWeight(config, resource, "sizeWeights", stringFromMap(body, "size")) * resourceWeight(config, resource, "resolutionWeights", firstNonEmptyString(stringFromMap(body, "resolution"), stringFromMap(body, "size"))) * discount
+	return []any{billingLine(candidate, resource, unit, count, roundPrice(amount), discount, simulated)}
 }
 
 func effectiveBillingConfig(candidate store.RuntimeModelCandidate) map[string]any {
@@ -121,6 +127,28 @@ func price(config map[string]any, key string) float64 {
 	return 0
 }
 
+func resourcePrice(config map[string]any, resource string, keys ...string) float64 {
+	for _, key := range keys {
+		if value := price(config, key); value > 0 {
+			return value
+		}
+	}
+	if resourceConfig, ok := config[resource].(map[string]any); ok {
+		for _, key := range keys {
+			if value := floatFromAny(resourceConfig[key]); value > 0 {
+				return value
+			}
+		}
+		if value := floatFromAny(resourceConfig["basePrice"]); value > 0 {
+			return value
+		}
+	}
+	if resource == "image_edit" {
+		return resourcePrice(config, "image", keys...)
+	}
+	return 0
+}
+
 func weighted(config map[string]any, key string, name string) float64 {
 	if strings.TrimSpace(name) == "" {
 		return 1
@@ -132,6 +160,39 @@ func weighted(config map[string]any, key string, name string) float64 {
 	return 1
 }
 
+func resourceWeight(config map[string]any, resource string, key string, name string) float64 {
+	if value := weighted(config, key, name); value != 1 {
+		return value
+	}
+	if strings.TrimSpace(name) == "" {
+		return 1
+	}
+	resourceConfig, _ := config[resource].(map[string]any)
+	if len(resourceConfig) == 0 && resource == "image_edit" {
+		resourceConfig, _ = config["image"].(map[string]any)
+	}
+	if weights, ok := resourceConfig["dynamicWeight"].(map[string]any); ok {
+		if value := floatFromAny(weights[name]); value > 0 {
+			return value
+		}
+	}
+	if weights, ok := resourceConfig[key].(map[string]any); ok {
+		if value := floatFromAny(weights[name]); value > 0 {
+			return value
+		}
+	}
+	return 1
+}
+
+func firstNonEmptyString(values ...string) string {
+	for _, value := range values {
+		if strings.TrimSpace(value) != "" {
+			return strings.TrimSpace(value)
+		}
+	}
+	return ""
+}
+
 func roundPrice(value float64) float64 {
 	return math.Round(value*1000000) / 1000000
 }