From 6dd9bc6decc8b6e131afaad74a3f954147a508f4 Mon Sep 17 00:00:00 2001 From: fengsilin Date: Tue, 30 Jun 2026 15:25:17 +0800 Subject: [PATCH 01/11] docs: add model display pricing design --- ...2026-06-30-model-display-pricing-design.md | 243 ++++++++++++++++++ 1 file changed, 243 insertions(+) create mode 100644 docs/superpowers/specs/2026-06-30-model-display-pricing-design.md diff --git a/docs/superpowers/specs/2026-06-30-model-display-pricing-design.md b/docs/superpowers/specs/2026-06-30-model-display-pricing-design.md new file mode 100644 index 0000000..74835ba --- /dev/null +++ b/docs/superpowers/specs/2026-06-30-model-display-pricing-design.md @@ -0,0 +1,243 @@ +# Model Display Pricing Design + +## Background + +The current pricing system mixes two responsibilities in one structure: + +- Real billing reads `PricingConfig.table[].price` through relay pricing lookup, freezes it into `PricingDecision`, and later uses that decision for pre-consumption and final settlement. +- The pricing page also reads `PricingConfig.table[].price` to render model price summaries and matrix details. + +Some video providers need a user-facing estimate such as `768p-6s, price per second`, while the actual charge is settled by token usage, provider-returned deduction, or another backend billing result. Reusing `PricingConfig` for this would make display-only prices affect real billing. + +## Goal + +Add a display-only pricing configuration that lets the frontend show a supplier-style price table for user estimation. + +This configuration must: + +- Use USD as the base amount so the existing frontend currency conversion can support USD, CNY, and custom currencies. +- Never participate in quota pre-consumption, final settlement, token billing, task polling, or billing logs. +- Override only the public pricing page display when configured. +- Preserve current pricing behavior for models without display pricing. + +## Non-Goals + +- Do not change how real billing is calculated. +- Do not add formula-based display pricing in this iteration. +- Do not replace `ModelPricingRules`; real matrix pricing remains unchanged. +- Do not calculate exact task cost from display pricing. It is only an estimate. + +## Data Model + +Add a new option key, separate from `ModelPricingRules`: + +```text +ModelDisplayPricing +``` + +The option value is a JSON map keyed by model name: + +```json +{ + "kling-video": [ + { + "specification": "768p-6s", + "official_supplier_tip": "Text-to-video and image-to-video, 768P 6s", + "price": 0.33333, + "unit": "second", + "sort_order": 1, + "discount_rate": 1 + }, + { + "specification": "768p-10s", + "official_supplier_tip": "Text-to-video and image-to-video, 768P 10s", + "price": 0.4, + "unit": "second", + "sort_order": 2, + "discount_rate": 1 + }, + { + "specification": "1080p", + "official_supplier_tip": "Text-to-video and image-to-video, 1080P", + "price": 0.58333, + "unit": "second", + "sort_order": 3, + "discount_rate": 1 + } + ] +} +``` + +Suggested Go type: + +```go +type ModelDisplayPricingItem struct { + Specification string `json:"specification"` + OfficialSupplierTip string `json:"official_supplier_tip"` + Price float64 `json:"price"` + Unit string `json:"unit"` + SortOrder int `json:"sort_order"` + DiscountRate float64 `json:"discount_rate"` +} +``` + +Field semantics: + +- `specification`: short SKU label shown in compact and detail views. +- `official_supplier_tip`: supplier-facing explanation shown in detail view. +- `price`: USD base display price before discount. This is not quota and not billing input. +- `unit`: display unit, such as `second`, `call`, `image`, or `video`. +- `sort_order`: stable ordering for the frontend and API response. +- `discount_rate`: display-only multiplier. Default is `1` when omitted or invalid. + +Validation rules: + +- Model key must not be empty. +- `specification` must not be empty. +- `price` must be greater than or equal to `0`. +- `unit` must not be empty. +- `discount_rate` defaults to `1`; accepted range is `0 <= discount_rate <= 1`. +- Items should be returned sorted by `sort_order`, then `specification`. + +## Backend Design + +Add a new ratio setting module, parallel to `model_pricing.go`, for display pricing only: + +- `GetModelDisplayPricing(model string) []types.ModelDisplayPricingItem` +- `GetModelDisplayPricingCopy() map[string][]types.ModelDisplayPricingItem` +- `SetModelDisplayPricing(model string, items []types.ModelDisplayPricingItem) error` +- `DeleteModelDisplayPricing(model string) error` +- `UpdateModelDisplayPricingByJSONString(jsonStr string) error` +- `ModelDisplayPricing2JSONString() string` + +Store the JSON in `options`, matching the existing config pattern and avoiding a cross-database migration for a display-only setting. + +Extend `model.Pricing`: + +```go +DisplayPricing []types.ModelDisplayPricingItem `json:"display_pricing,omitempty"` +``` + +During `model.updatePricing()`, after the model metadata and real pricing fields are assembled, attach display pricing for the model if configured. This keeps `/api/pricing` as the single source used by the existing pricing page. + +`/api/pricing/user/:model` should also return `display_pricing` when present, but it must not use it for `original_price`, `user_price`, `original_input`, `original_output`, or savings calculation. + +Admin APIs should mirror the existing model pricing rule APIs: + +- `GET /api/option/model_display_pricing` +- `GET /api/option/model_display_pricing/:model` +- `PUT /api/option/model_display_pricing/:model` +- `DELETE /api/option/model_display_pricing/:model` + +These endpoints manage only display data and call `model.RefreshPricing()` after updates. + +## Frontend Design + +Add display pricing helpers in the model pricing utilities: + +- `hasDisplayPricing(record)` +- `summarizeDisplayPricing({ record, displayPrice })` +- `formatDisplayPricingItem({ item, displayPrice })` + +Display calculation: + +```js +original = displayPrice(item.price) +discounted = displayPrice(item.price * item.discount_rate) +``` + +If `discount_rate` is `1`, show only the regular price. If lower than `1`, show discounted price and optionally the original price with strikethrough. + +Display priority: + +1. If `record.display_pricing` exists and has items, render display pricing. +2. Otherwise, keep the existing matrix pricing, per-token pricing, and per-call pricing behavior. + +List and card compact display: + +```text +CNY 2.41 - CNY 4.22 / second +Display price +``` + +If all visible items share one price: + +```text +CNY 2.41 / second +Display price +``` + +Detail view: + +| Specification | Supplier Tip | Price | Discount | +| --- | --- | --- | --- | +| 768p-6s | Text-to-video and image-to-video, 768P 6s | CNY 2.41 / second | - | +| 768p-10s | Text-to-video and image-to-video, 768P 10s | CNY 2.90 / second | - | +| 1080p | Text-to-video and image-to-video, 1080P | CNY 4.22 / second | - | + +Every display pricing block must include an i18n note with this meaning: + +```text +Display price is only an estimate. Actual consumption is based on the final quota deduction after the task completes. +``` + +When a model has `display_pricing`, the old user discount comparison card should be hidden or replaced with the display pricing table. The current discount card is derived from real billing fields and would be misleading for models whose visible price is only an estimate. + +## Billing Isolation + +The implementation must not reference display pricing in: + +- `relay/helper/price.go` +- `relay/helper/pricing_lookup.go` +- `controller/relay.go` pricing preload logic +- `service/task_polling.go` +- `types.PricingDecision` +- task final deduction calculation +- quota pre-consumption +- consume logs used for settlement + +Only `model.Pricing` and pricing-page APIs should expose it. + +## Data Flow + +```text +Admin saves display pricing + -> options.ModelDisplayPricing + -> ratio_setting display pricing cache + -> model.RefreshPricing() + -> model.updatePricing() attaches display_pricing + -> GET /api/pricing returns display_pricing + -> frontend price page renders display SKU table + +User creates a task + -> relay pricing preload reads ModelPricingRules or legacy price/ratio + -> PricingDecision is frozen + -> task completes + -> final settlement uses real billing data + -> display_pricing is never read +``` + +## Testing + +Backend tests: + +- Parse and validate `ModelDisplayPricing` JSON. +- Default `discount_rate` to `1`. +- Reject invalid model keys, empty specification, negative price, empty unit, and invalid discount rates. +- Confirm `model.GetPricingByModel()` includes sorted `DisplayPricing`. +- Confirm real `PricingConfig` behavior is unchanged when display pricing exists. +- Confirm `/api/pricing/user/:model` returns `display_pricing` but does not use it for computed prices. + +Frontend tests: + +- `hasDisplayPricing()` detects display pricing before matrix pricing. +- Summary renders min-max range with converted currency. +- Detail table renders specification, supplier tip, unit, and discount. +- Existing matrix and legacy pricing rendering remains unchanged when no display pricing exists. +- User discount comparison is hidden or replaced when display pricing exists. + +Manual verification: + +- Configure one model with display pricing and token-based real billing. +- Confirm public price page shows the display table. +- Run a real task and confirm quota deduction follows the existing billing path, not `display_pricing.price`. From 6dd8c19ea849eac3d1faf710595edaecc4569621 Mon Sep 17 00:00:00 2001 From: fengsilin Date: Tue, 30 Jun 2026 16:26:37 +0800 Subject: [PATCH 02/11] docs: clarify model display pricing design --- ...2026-06-30-model-display-pricing-design.md | 31 ++++++++++++------- 1 file changed, 20 insertions(+), 11 deletions(-) diff --git a/docs/superpowers/specs/2026-06-30-model-display-pricing-design.md b/docs/superpowers/specs/2026-06-30-model-display-pricing-design.md index 74835ba..0b70d5a 100644 --- a/docs/superpowers/specs/2026-06-30-model-display-pricing-design.md +++ b/docs/superpowers/specs/2026-06-30-model-display-pricing-design.md @@ -88,7 +88,7 @@ Field semantics: - `price`: USD base display price before discount. This is not quota and not billing input. - `unit`: display unit, such as `second`, `call`, `image`, or `video`. - `sort_order`: stable ordering for the frontend and API response. -- `discount_rate`: display-only multiplier. Default is `1` when omitted or invalid. +- `discount_rate`: display-only multiplier. `0` or omitted means no discount and is normalized to `1`. Validation rules: @@ -96,7 +96,7 @@ Validation rules: - `specification` must not be empty. - `price` must be greater than or equal to `0`. - `unit` must not be empty. -- `discount_rate` defaults to `1`; accepted range is `0 <= discount_rate <= 1`. +- `discount_rate` is normalized before validation: `0` or omitted becomes `1`; after normalization, the accepted range is `0 < discount_rate <= 1`. - Items should be returned sorted by `sort_order`, then `specification`. ## Backend Design @@ -112,6 +112,13 @@ Add a new ratio setting module, parallel to `model_pricing.go`, for display pric Store the JSON in `options`, matching the existing config pattern and avoiding a cross-database migration for a display-only setting. +The option lifecycle must follow the existing `ModelPricingRules` pattern: + +- Add `ModelDisplayPricing` to `model.InitOptionMap()` with an empty JSON object default. +- Add `case "ModelDisplayPricing"` to `model.updateOptionMap()` so DB-loaded values hydrate the in-memory display pricing cache during startup and option sync. +- Persist updates through `model.UpdateOption(ModelDisplayPricingOptionKey, ModelDisplayPricing2JSONString())`. +- Call `model.RefreshPricing()` after successful create, update, or delete so `/api/pricing` reflects the new display data immediately. + Extend `model.Pricing`: ```go @@ -120,16 +127,16 @@ DisplayPricing []types.ModelDisplayPricingItem `json:"display_pricing,omitempty" During `model.updatePricing()`, after the model metadata and real pricing fields are assembled, attach display pricing for the model if configured. This keeps `/api/pricing` as the single source used by the existing pricing page. -`/api/pricing/user/:model` should also return `display_pricing` when present, but it must not use it for `original_price`, `user_price`, `original_input`, `original_output`, or savings calculation. +`/api/pricing/user/*model` should also return `display_pricing` when present, but it must not use it for `original_price`, `user_price`, `original_input`, `original_output`, or savings calculation. Admin APIs should mirror the existing model pricing rule APIs: - `GET /api/option/model_display_pricing` -- `GET /api/option/model_display_pricing/:model` -- `PUT /api/option/model_display_pricing/:model` -- `DELETE /api/option/model_display_pricing/:model` +- `GET /api/option/model_display_pricing/*model` +- `PUT /api/option/model_display_pricing/*model` +- `DELETE /api/option/model_display_pricing/*model` -These endpoints manage only display data and call `model.RefreshPricing()` after updates. +Use wildcard model parameters, matching `/api/pricing/user/*model`, so model names containing `/` can be configured without URL-shape bugs. These endpoints manage only display data and call `model.RefreshPricing()` after updates. ## Frontend Design @@ -181,7 +188,7 @@ Every display pricing block must include an i18n note with this meaning: Display price is only an estimate. Actual consumption is based on the final quota deduction after the task completes. ``` -When a model has `display_pricing`, the old user discount comparison card should be hidden or replaced with the display pricing table. The current discount card is derived from real billing fields and would be misleading for models whose visible price is only an estimate. +When a model has `display_pricing`, hide the old `UserPriceComparison` discount card. The detail side sheet should show the display pricing table and the estimate note instead. The current discount card is derived from real billing fields and would be misleading for models whose visible price is only an estimate. ## Billing Isolation @@ -196,7 +203,7 @@ The implementation must not reference display pricing in: - quota pre-consumption - consume logs used for settlement -Only `model.Pricing` and pricing-page APIs should expose it. +Only admin display-pricing config APIs, `model.Pricing`, and pricing-page APIs should expose it. Relay, billing, task, and consume-log paths must not read it. ## Data Flow @@ -224,9 +231,11 @@ Backend tests: - Parse and validate `ModelDisplayPricing` JSON. - Default `discount_rate` to `1`. - Reject invalid model keys, empty specification, negative price, empty unit, and invalid discount rates. +- Confirm `model.InitOptionMap()` initializes `ModelDisplayPricing` and `model.updateOptionMap()` hydrates the display pricing cache from DB-loaded option values. - Confirm `model.GetPricingByModel()` includes sorted `DisplayPricing`. - Confirm real `PricingConfig` behavior is unchanged when display pricing exists. -- Confirm `/api/pricing/user/:model` returns `display_pricing` but does not use it for computed prices. +- Confirm `/api/pricing/user/*model` returns `display_pricing` but does not use it for computed prices. +- Confirm wildcard admin routes support model names containing `/`. Frontend tests: @@ -234,7 +243,7 @@ Frontend tests: - Summary renders min-max range with converted currency. - Detail table renders specification, supplier tip, unit, and discount. - Existing matrix and legacy pricing rendering remains unchanged when no display pricing exists. -- User discount comparison is hidden or replaced when display pricing exists. +- User discount comparison is hidden when display pricing exists. Manual verification: From 302987989b3220e7e0a1c3b38118d9e08cd5c1b2 Mon Sep 17 00:00:00 2001 From: fengsilin Date: Wed, 1 Jul 2026 10:39:20 +0800 Subject: [PATCH 03/11] docs: expand display pricing frontend spec --- ...2026-06-30-model-display-pricing-design.md | 80 +++++++++++++++++-- 1 file changed, 75 insertions(+), 5 deletions(-) diff --git a/docs/superpowers/specs/2026-06-30-model-display-pricing-design.md b/docs/superpowers/specs/2026-06-30-model-display-pricing-design.md index 0b70d5a..5e97760 100644 --- a/docs/superpowers/specs/2026-06-30-model-display-pricing-design.md +++ b/docs/superpowers/specs/2026-06-30-model-display-pricing-design.md @@ -140,11 +140,23 @@ Use wildcard model parameters, matching `/api/pricing/user/*model`, so model nam ## Frontend Design +The display pricing UI is part of the existing model pricing page. It should be implemented as a display override on top of the current `/api/pricing` response, not as a new standalone public page. + +Affected frontend entry points: + +- `web/src/hooks/model-pricing/useModelPricingData.jsx`: keep loading `/api/pricing`; no extra fetch is needed for public display. +- `web/src/helpers/utils.jsx`: add shared helpers so table, card, and detail views render the same display price. +- `web/src/components/table/model-pricing/view/table/PricingTableColumns.jsx`: compact price column display. +- `web/src/components/table/model-pricing/view/card/PricingCardView.jsx`: compact card display. +- `web/src/components/table/model-pricing/modal/components/ModelPricingTable.jsx`: full detail table display. +- `web/src/components/table/model-pricing/modal/components/UserPriceComparison.jsx`: hide when display pricing exists. + Add display pricing helpers in the model pricing utilities: - `hasDisplayPricing(record)` -- `summarizeDisplayPricing({ record, displayPrice })` -- `formatDisplayPricingItem({ item, displayPrice })` +- `normalizeDisplayPricingItems(record)` +- `formatDisplayPricingItem({ item, displayPrice, t })` +- `summarizeDisplayPricing({ record, displayPrice, t })` Display calculation: @@ -155,26 +167,58 @@ discounted = displayPrice(item.price * item.discount_rate) If `discount_rate` is `1`, show only the regular price. If lower than `1`, show discounted price and optionally the original price with strikethrough. +Unit labels should be translated in the frontend. The API returns stable unit keys such as `second`, `call`, `image`, and `video`; the UI maps them to localized labels such as `/ second` or `/ sec` depending on existing i18n conventions. + Display priority: 1. If `record.display_pricing` exists and has items, render display pricing. 2. Otherwise, keep the existing matrix pricing, per-token pricing, and per-call pricing behavior. -List and card compact display: +### Pricing Table Display + +In the main table price column, display pricing replaces the current matrix/per-token/per-call price block for that model. + +For multiple display pricing items with the same unit, show a converted price range: ```text CNY 2.41 - CNY 4.22 / second Display price ``` -If all visible items share one price: +For one item, or when all items have the same discounted price: ```text CNY 2.41 / second Display price ``` -Detail view: +For multiple units, group by unit and render one short line per unit: + +```text +CNY 2.41 - CNY 4.22 / second +CNY 1.00 / image +Display price +``` + +The compact table view should not render `official_supplier_tip`; that belongs in the detail view. + +### Pricing Card Display + +Cards should use the same summary output as the table column, with at most the first three display items shown below the range: + +```text +CNY 2.41 - CNY 4.22 / second +768p-6s CNY 2.41 / second +768p-10s CNY 2.90 / second +1080p CNY 4.22 / second ++ 2 more +``` + +If the card is too narrow on mobile, keep only the range and the `Display price` label. Do not render a horizontal table inside the card. + +### Detail Side Sheet Display + +When `display_pricing` exists, `ModelPricingTable` should render a dedicated display pricing table instead of the existing group price table or matrix table. | Specification | Supplier Tip | Price | Discount | | --- | --- | --- | --- | @@ -182,14 +226,40 @@ Detail view: | 768p-10s | Text-to-video and image-to-video, 768P 10s | CNY 2.90 / second | - | | 1080p | Text-to-video and image-to-video, 1080P | CNY 4.22 / second | - | +For discounted items, render the discounted price as the primary value and the original converted price as secondary strikethrough text: + +```text +CNY 1.93 / second +Original CNY 2.41, 20% off +``` + Every display pricing block must include an i18n note with this meaning: ```text Display price is only an estimate. Actual consumption is based on the final quota deduction after the task completes. ``` +The note must appear in the detail side sheet and in any compact tooltip/popover if one is added later. The main table and card can show only a short `Display price` label to avoid visual noise. + +### User Price Comparison + When a model has `display_pricing`, hide the old `UserPriceComparison` discount card. The detail side sheet should show the display pricing table and the estimate note instead. The current discount card is derived from real billing fields and would be misleading for models whose visible price is only an estimate. +`/api/pricing/user/*model` may still return `display_pricing` for consistency, but the frontend should not calculate user-specific discounted display pricing from `original_price` or `user_price`. + +### Admin Editing UI + +The first implementation may manage `ModelDisplayPricing` through JSON option editing or a minimal admin form. If a structured form is added, it should expose only display fields: + +- `specification` +- `official_supplier_tip` +- `price` as USD base price +- `unit` +- `sort_order` +- `discount_rate` + +The editor must label `price` as USD base display price and include a warning that it does not affect real billing. + ## Billing Isolation The implementation must not reference display pricing in: From 18a3495968be7443fcae6efb8755cb076e2e613d Mon Sep 17 00:00:00 2001 From: fengsilin Date: Wed, 1 Jul 2026 11:21:00 +0800 Subject: [PATCH 04/11] feat: add model display pricing settings --- .../ratio_setting/model_display_pricing.go | 120 ++++++++++++++++++ .../model_display_pricing_test.go | 95 ++++++++++++++ types/model_display_pricing.go | 19 +++ 3 files changed, 234 insertions(+) create mode 100644 setting/ratio_setting/model_display_pricing.go create mode 100644 setting/ratio_setting/model_display_pricing_test.go create mode 100644 types/model_display_pricing.go diff --git a/setting/ratio_setting/model_display_pricing.go b/setting/ratio_setting/model_display_pricing.go new file mode 100644 index 0000000..b53e2e8 --- /dev/null +++ b/setting/ratio_setting/model_display_pricing.go @@ -0,0 +1,120 @@ +package ratio_setting + +import ( + "fmt" + "sort" + "strings" + + "github.com/QuantumNous/new-api/common" + "github.com/QuantumNous/new-api/types" +) + +const ModelDisplayPricingOptionKey = "ModelDisplayPricing" + +var displayPricingMap = types.NewRWMap[string, []types.ModelDisplayPricingItem]() + +func GetModelDisplayPricing(model string) []types.ModelDisplayPricingItem { + items, ok := displayPricingMap.Get(model) + if !ok { + return nil + } + return types.CloneModelDisplayPricingItems(items) +} + +func GetModelDisplayPricingCopy() map[string][]types.ModelDisplayPricingItem { + src := displayPricingMap.ReadAll() + dst := make(map[string][]types.ModelDisplayPricingItem, len(src)) + for model, items := range src { + dst[model] = types.CloneModelDisplayPricingItems(items) + } + return dst +} + +func SetModelDisplayPricing(model string, items []types.ModelDisplayPricingItem) error { + model = strings.TrimSpace(model) + if model == "" { + return fmt.Errorf("model name is required") + } + normalized, err := normalizeModelDisplayPricingItems(items) + if err != nil { + return err + } + displayPricingMap.Set(model, normalized) + InvalidateExposedDataCache() + return nil +} + +func DeleteModelDisplayPricing(model string) error { + all := displayPricingMap.ReadAll() + delete(all, model) + displayPricingMap.Clear() + displayPricingMap.AddAll(all) + InvalidateExposedDataCache() + return nil +} + +func ModelDisplayPricing2JSONString() string { + bytes, err := common.Marshal(GetModelDisplayPricingCopy()) + if err != nil { + return "{}" + } + return string(bytes) +} + +func UpdateModelDisplayPricingByJSONString(jsonStr string) error { + if strings.TrimSpace(jsonStr) == "" { + jsonStr = "{}" + } + raw := map[string][]types.ModelDisplayPricingItem{} + if err := common.UnmarshalJsonStr(jsonStr, &raw); err != nil { + return err + } + next := make(map[string][]types.ModelDisplayPricingItem, len(raw)) + for model, items := range raw { + model = strings.TrimSpace(model) + if model == "" { + return fmt.Errorf("model name is required") + } + normalized, err := normalizeModelDisplayPricingItems(items) + if err != nil { + return fmt.Errorf("model %s: %w", model, err) + } + next[model] = normalized + } + displayPricingMap.Clear() + displayPricingMap.AddAll(next) + InvalidateExposedDataCache() + return nil +} + +func normalizeModelDisplayPricingItems(items []types.ModelDisplayPricingItem) ([]types.ModelDisplayPricingItem, error) { + normalized := types.CloneModelDisplayPricingItems(items) + for i := range normalized { + item := &normalized[i] + item.Specification = strings.TrimSpace(item.Specification) + item.OfficialSupplierTip = strings.TrimSpace(item.OfficialSupplierTip) + item.Unit = strings.TrimSpace(item.Unit) + if item.Specification == "" { + return nil, fmt.Errorf("row %d: specification is required", i) + } + if item.Price < 0 { + return nil, fmt.Errorf("row %d: price must be >= 0", i) + } + if item.Unit == "" { + return nil, fmt.Errorf("row %d: unit is required", i) + } + if item.DiscountRate == 0 { + item.DiscountRate = 1 + } + if item.DiscountRate <= 0 || item.DiscountRate > 1 { + return nil, fmt.Errorf("row %d: discount_rate must be > 0 and <= 1", i) + } + } + sort.SliceStable(normalized, func(i, j int) bool { + if normalized[i].SortOrder != normalized[j].SortOrder { + return normalized[i].SortOrder < normalized[j].SortOrder + } + return normalized[i].Specification < normalized[j].Specification + }) + return normalized, nil +} diff --git a/setting/ratio_setting/model_display_pricing_test.go b/setting/ratio_setting/model_display_pricing_test.go new file mode 100644 index 0000000..372ce39 --- /dev/null +++ b/setting/ratio_setting/model_display_pricing_test.go @@ -0,0 +1,95 @@ +package ratio_setting + +import ( + "testing" + + "github.com/QuantumNous/new-api/types" + "github.com/stretchr/testify/require" +) + +func TestModelDisplayPricingRoundTripNormalizeAndSort(t *testing.T) { + require.NoError(t, UpdateModelDisplayPricingByJSONString(`{}`)) + + jsonStr := `{ + "kling-video": [ + {"specification":"1080p","official_supplier_tip":"1080P","price":0.58333,"unit":"second","sort_order":3}, + {"specification":"768p-6s","official_supplier_tip":"768P 6s","price":0.33333,"unit":"second","sort_order":1,"discount_rate":0.8}, + {"specification":"768p-10s","official_supplier_tip":"768P 10s","price":0.4,"unit":"second","sort_order":2,"discount_rate":0} + ] + }` + + require.NoError(t, UpdateModelDisplayPricingByJSONString(jsonStr)) + items := GetModelDisplayPricing("kling-video") + require.Len(t, items, 3) + require.Equal(t, "768p-6s", items[0].Specification) + require.Equal(t, 0.8, items[0].DiscountRate) + require.Equal(t, "768p-10s", items[1].Specification) + require.Equal(t, 1.0, items[1].DiscountRate) + require.Equal(t, "1080p", items[2].Specification) + require.Equal(t, 1.0, items[2].DiscountRate) + + items[0].Specification = "mutated" + require.Equal(t, "768p-6s", GetModelDisplayPricing("kling-video")[0].Specification) + + roundTrip := ModelDisplayPricing2JSONString() + require.NoError(t, UpdateModelDisplayPricingByJSONString(roundTrip)) + require.Equal(t, "768p-6s", GetModelDisplayPricing("kling-video")[0].Specification) +} + +func TestSetModelDisplayPricingValidation(t *testing.T) { + tests := []struct { + name string + model string + items []types.ModelDisplayPricingItem + want string + }{ + { + name: "empty_model", + model: "", + items: []types.ModelDisplayPricingItem{{Specification: "768p", Price: 1, Unit: "second"}}, + want: "model name is required", + }, + { + name: "empty_specification", + model: "kling-video", + items: []types.ModelDisplayPricingItem{{Price: 1, Unit: "second"}}, + want: "specification is required", + }, + { + name: "negative_price", + model: "kling-video", + items: []types.ModelDisplayPricingItem{{Specification: "768p", Price: -1, Unit: "second"}}, + want: "price must be >= 0", + }, + { + name: "empty_unit", + model: "kling-video", + items: []types.ModelDisplayPricingItem{{Specification: "768p", Price: 1}}, + want: "unit is required", + }, + { + name: "bad_discount", + model: "kling-video", + items: []types.ModelDisplayPricingItem{{Specification: "768p", Price: 1, Unit: "second", DiscountRate: 1.2}}, + want: "discount_rate must be > 0 and <= 1", + }, + } + + for _, tt := range tests { + t.Run(tt.name, func(t *testing.T) { + err := SetModelDisplayPricing(tt.model, tt.items) + require.Error(t, err) + require.Contains(t, err.Error(), tt.want) + }) + } +} + +func TestDeleteModelDisplayPricing(t *testing.T) { + require.NoError(t, UpdateModelDisplayPricingByJSONString(`{}`)) + require.NoError(t, SetModelDisplayPricing("kling-video", []types.ModelDisplayPricingItem{ + {Specification: "768p", Price: 1, Unit: "second"}, + })) + require.Len(t, GetModelDisplayPricing("kling-video"), 1) + require.NoError(t, DeleteModelDisplayPricing("kling-video")) + require.Empty(t, GetModelDisplayPricing("kling-video")) +} diff --git a/types/model_display_pricing.go b/types/model_display_pricing.go new file mode 100644 index 0000000..3155aea --- /dev/null +++ b/types/model_display_pricing.go @@ -0,0 +1,19 @@ +package types + +type ModelDisplayPricingItem struct { + Specification string `json:"specification"` + OfficialSupplierTip string `json:"official_supplier_tip"` + Price float64 `json:"price"` + Unit string `json:"unit"` + SortOrder int `json:"sort_order"` + DiscountRate float64 `json:"discount_rate"` +} + +func CloneModelDisplayPricingItems(src []ModelDisplayPricingItem) []ModelDisplayPricingItem { + if src == nil { + return nil + } + dst := make([]ModelDisplayPricingItem, len(src)) + copy(dst, src) + return dst +} From de117ac932bdfd6b472a0d50edb353d0ca32366f Mon Sep 17 00:00:00 2001 From: fengsilin Date: Wed, 1 Jul 2026 11:28:26 +0800 Subject: [PATCH 05/11] feat: add model display pricing admin api --- .../model_display_pricing_controller.go | 59 ++++++++++++ .../model_display_pricing_controller_test.go | 96 +++++++++++++++++++ model/option.go | 3 + router/api-router.go | 4 + 4 files changed, 162 insertions(+) create mode 100644 controller/model_display_pricing_controller.go create mode 100644 controller/model_display_pricing_controller_test.go diff --git a/controller/model_display_pricing_controller.go b/controller/model_display_pricing_controller.go new file mode 100644 index 0000000..6f40e1e --- /dev/null +++ b/controller/model_display_pricing_controller.go @@ -0,0 +1,59 @@ +package controller + +import ( + "net/http" + "strings" + + "github.com/QuantumNous/new-api/common" + "github.com/QuantumNous/new-api/model" + "github.com/QuantumNous/new-api/setting/ratio_setting" + "github.com/QuantumNous/new-api/types" + "github.com/gin-gonic/gin" +) + +func GetModelDisplayPricingRules(c *gin.Context) { + common.ApiSuccess(c, ratio_setting.GetModelDisplayPricingCopy()) +} + +func GetModelDisplayPricingRule(c *gin.Context) { + modelName := strings.TrimPrefix(c.Param("model"), "/") + items := ratio_setting.GetModelDisplayPricing(modelName) + if len(items) == 0 { + c.JSON(http.StatusNotFound, gin.H{"success": false, "message": "model display pricing not found"}) + return + } + common.ApiSuccess(c, items) +} + +func UpdateModelDisplayPricingRule(c *gin.Context) { + modelName := strings.TrimPrefix(c.Param("model"), "/") + var items []types.ModelDisplayPricingItem + if err := common.DecodeJson(c.Request.Body, &items); err != nil { + c.JSON(http.StatusBadRequest, gin.H{"success": false, "message": err.Error()}) + return + } + if err := ratio_setting.SetModelDisplayPricing(modelName, items); err != nil { + c.JSON(http.StatusBadRequest, gin.H{"success": false, "message": err.Error()}) + return + } + if err := model.UpdateOption(ratio_setting.ModelDisplayPricingOptionKey, ratio_setting.ModelDisplayPricing2JSONString()); err != nil { + c.JSON(http.StatusInternalServerError, gin.H{"success": false, "message": err.Error()}) + return + } + model.RefreshPricing() + common.ApiSuccess(c, ratio_setting.GetModelDisplayPricing(modelName)) +} + +func DeleteModelDisplayPricingRule(c *gin.Context) { + modelName := strings.TrimPrefix(c.Param("model"), "/") + if err := ratio_setting.DeleteModelDisplayPricing(modelName); err != nil { + c.JSON(http.StatusBadRequest, gin.H{"success": false, "message": err.Error()}) + return + } + if err := model.UpdateOption(ratio_setting.ModelDisplayPricingOptionKey, ratio_setting.ModelDisplayPricing2JSONString()); err != nil { + c.JSON(http.StatusInternalServerError, gin.H{"success": false, "message": err.Error()}) + return + } + model.RefreshPricing() + common.ApiSuccess(c, nil) +} diff --git a/controller/model_display_pricing_controller_test.go b/controller/model_display_pricing_controller_test.go new file mode 100644 index 0000000..bd9da00 --- /dev/null +++ b/controller/model_display_pricing_controller_test.go @@ -0,0 +1,96 @@ +package controller + +import ( + "net/http" + "testing" + + "github.com/QuantumNous/new-api/common" + "github.com/QuantumNous/new-api/model" + "github.com/QuantumNous/new-api/setting/ratio_setting" + "github.com/QuantumNous/new-api/types" + "github.com/gin-gonic/gin" + "github.com/glebarez/sqlite" + "github.com/stretchr/testify/require" + "gorm.io/gorm" +) + +func setupModelDisplayPricingControllerTest(t *testing.T) *gin.Engine { + t.Helper() + gin.SetMode(gin.TestMode) + + db, err := gorm.Open(sqlite.Open(":memory:"), &gorm.Config{}) + require.NoError(t, err) + require.NoError(t, db.AutoMigrate(&model.Option{})) + sqlDB, err := db.DB() + require.NoError(t, err) + sqlDB.SetMaxOpenConns(1) + + origDB := model.DB + origUsingSQLite := common.UsingSQLite + origRedisEnabled := common.RedisEnabled + model.DB = db + common.UsingSQLite = true + common.RedisEnabled = false + model.InitOptionMap() + require.NoError(t, ratio_setting.UpdateModelDisplayPricingByJSONString(`{}`)) + + t.Cleanup(func() { + model.DB = origDB + common.UsingSQLite = origUsingSQLite + common.RedisEnabled = origRedisEnabled + require.NoError(t, ratio_setting.UpdateModelDisplayPricingByJSONString(`{}`)) + require.NoError(t, sqlDB.Close()) + }) + + r := gin.New() + r.GET("/api/option/model_display_pricing", GetModelDisplayPricingRules) + r.GET("/api/option/model_display_pricing/*model", GetModelDisplayPricingRule) + r.PUT("/api/option/model_display_pricing/*model", UpdateModelDisplayPricingRule) + r.DELETE("/api/option/model_display_pricing/*model", DeleteModelDisplayPricingRule) + return r +} + +func TestModelDisplayPricingControllerCRUDAndWildcard(t *testing.T) { + r := setupModelDisplayPricingControllerTest(t) + items := []types.ModelDisplayPricingItem{ + {Specification: "768p-6s", OfficialSupplierTip: "768P 6s", Price: 0.33333, Unit: "second", SortOrder: 1}, + } + + w := performJSONRequest(t, r, http.MethodPut, "/api/option/model_display_pricing/provider/kling-video", items) + require.Equal(t, http.StatusOK, w.Code) + require.Len(t, ratio_setting.GetModelDisplayPricing("provider/kling-video"), 1) + + w = performJSONRequest(t, r, http.MethodGet, "/api/option/model_display_pricing/provider/kling-video", nil) + require.Equal(t, http.StatusOK, w.Code) + require.Contains(t, w.Body.String(), "768p-6s") + + w = performJSONRequest(t, r, http.MethodGet, "/api/option/model_display_pricing", nil) + require.Equal(t, http.StatusOK, w.Code) + require.Contains(t, w.Body.String(), "provider/kling-video") + + w = performJSONRequest(t, r, http.MethodDelete, "/api/option/model_display_pricing/provider/kling-video", nil) + require.Equal(t, http.StatusOK, w.Code) + require.Empty(t, ratio_setting.GetModelDisplayPricing("provider/kling-video")) +} + +func TestModelDisplayPricingControllerValidationError(t *testing.T) { + r := setupModelDisplayPricingControllerTest(t) + items := []types.ModelDisplayPricingItem{{Specification: "", Price: 1, Unit: "second"}} + + w := performJSONRequest(t, r, http.MethodPut, "/api/option/model_display_pricing/kling-video", items) + require.Equal(t, http.StatusBadRequest, w.Code) + require.Contains(t, w.Body.String(), "specification is required") +} + +func TestModelDisplayPricingOptionMapHydratesCache(t *testing.T) { + r := setupModelDisplayPricingControllerTest(t) + require.NotNil(t, r) + + require.NoError(t, model.UpdateOption(ratio_setting.ModelDisplayPricingOptionKey, `{ + "kling-video":[{"specification":"768p","price":1,"unit":"second"}] + }`)) + + items := ratio_setting.GetModelDisplayPricing("kling-video") + require.Len(t, items, 1) + require.Equal(t, "768p", items[0].Specification) +} diff --git a/model/option.go b/model/option.go index 056bfac..a0c2336 100644 --- a/model/option.go +++ b/model/option.go @@ -134,6 +134,7 @@ func InitOptionMap() { common.OptionMap["ModelRatio"] = ratio_setting.ModelRatio2JSONString() common.OptionMap["ModelPrice"] = ratio_setting.ModelPrice2JSONString() common.OptionMap["ModelPricingRules"] = ratio_setting.ModelPricingRules2JSONString() + common.OptionMap["ModelDisplayPricing"] = ratio_setting.ModelDisplayPricing2JSONString() common.OptionMap["CacheRatio"] = ratio_setting.CacheRatio2JSONString() common.OptionMap["CreateCacheRatio"] = ratio_setting.CreateCacheRatio2JSONString() common.OptionMap["GroupRatio"] = ratio_setting.GroupRatio2JSONString() @@ -514,6 +515,8 @@ func updateOptionMap(key string, value string) (err error) { err = ratio_setting.UpdateModelPriceByJSONString(value) case "ModelPricingRules": err = ratio_setting.UpdateModelPricingRulesByJSONString(value) + case "ModelDisplayPricing": + err = ratio_setting.UpdateModelDisplayPricingByJSONString(value) case "CacheRatio": err = ratio_setting.UpdateCacheRatioByJSONString(value) case "CreateCacheRatio": diff --git a/router/api-router.go b/router/api-router.go index edb2a7c..ce023a8 100644 --- a/router/api-router.go +++ b/router/api-router.go @@ -195,6 +195,10 @@ func SetApiRouter(router *gin.Engine) { optionRoute.GET("/model_pricing/:model", controller.GetModelPricingRule) optionRoute.PUT("/model_pricing/:model", controller.UpdateModelPricingRule) optionRoute.DELETE("/model_pricing/:model", controller.DeleteModelPricingRule) + optionRoute.GET("/model_display_pricing", controller.GetModelDisplayPricingRules) + optionRoute.GET("/model_display_pricing/*model", controller.GetModelDisplayPricingRule) + optionRoute.PUT("/model_display_pricing/*model", controller.UpdateModelDisplayPricingRule) + optionRoute.DELETE("/model_display_pricing/*model", controller.DeleteModelDisplayPricingRule) optionRoute.GET("/channel_affinity_cache", controller.GetChannelAffinityCacheStats) optionRoute.DELETE("/channel_affinity_cache", controller.ClearChannelAffinityCache) optionRoute.POST("/rest_model_ratio", controller.ResetModelRatio) From ced4d7f3230df9d8bb878d75cfc9960019acc512 Mon Sep 17 00:00:00 2001 From: fengsilin Date: Wed, 1 Jul 2026 11:34:26 +0800 Subject: [PATCH 06/11] feat: expose model display pricing --- controller/pricing.go | 6 +++++ controller/pricing_user_test.go | 36 ++++++++++++++++++++++++--- model/pricing.go | 43 ++++++++++++++++++++------------- model/pricing_test.go | 35 +++++++++++++++++++++++++++ 4 files changed, 99 insertions(+), 21 deletions(-) diff --git a/controller/pricing.go b/controller/pricing.go index de90e0b..12bcaf6 100644 --- a/controller/pricing.go +++ b/controller/pricing.go @@ -138,6 +138,9 @@ func GetUserPricing(c *gin.Context) { if pricingData.PricingConfig != nil { result["pricing_config"] = pricingData.PricingConfig } + if len(pricingData.DisplayPricing) > 0 { + result["display_pricing"] = pricingData.DisplayPricing + } if pricingData.QuotaType == model.QuotaTypeByTokens { if pricingData.PricingConfig != nil { @@ -184,6 +187,9 @@ func respondOriginalPrice(c *gin.Context, pricingData *model.Pricing) { } result["original_price"] = pricingData.ModelPrice } + if len(pricingData.DisplayPricing) > 0 { + result["display_pricing"] = pricingData.DisplayPricing + } c.JSON(200, result) } diff --git a/controller/pricing_user_test.go b/controller/pricing_user_test.go index 31326b8..33e93a1 100644 --- a/controller/pricing_user_test.go +++ b/controller/pricing_user_test.go @@ -9,8 +9,9 @@ import ( "github.com/QuantumNous/new-api/common" "github.com/QuantumNous/new-api/model" "github.com/QuantumNous/new-api/setting/ratio_setting" - "github.com/glebarez/sqlite" + "github.com/QuantumNous/new-api/types" "github.com/gin-gonic/gin" + "github.com/glebarez/sqlite" "github.com/stretchr/testify/assert" "github.com/stretchr/testify/require" "gorm.io/gorm" @@ -152,6 +153,36 @@ func TestGetUserPricing_NotLoggedIn_PerCall(t *testing.T) { assert.Equal(t, float64(0.04), resp["original_price"]) } +func TestGetUserPricingIncludesDisplayPricingWithoutChangingComputedPrice(t *testing.T) { + setupPricingTestDB(t) + modelName := "display-pricing-user-model" + router := gin.New() + router.GET("/api/pricing/user/*model", GetUserPricing) + + setPricingCache([]model.Pricing{ + { + ModelName: modelName, + QuotaType: model.QuotaTypeByCall, + ModelPrice: 0.5, + EnableGroup: []string{"default"}, + DisplayPricing: []types.ModelDisplayPricingItem{ + {Specification: "768p-6s", Price: 0.33333, Unit: "second", SortOrder: 1, DiscountRate: 1}, + }, + }, + }) + + w := httptest.NewRecorder() + req, _ := http.NewRequest("GET", "/api/pricing/user/"+modelName, nil) + router.ServeHTTP(w, req) + + var resp map[string]interface{} + require.NoError(t, json.Unmarshal(w.Body.Bytes(), &resp)) + require.Equal(t, http.StatusOK, w.Code) + require.Contains(t, resp, "display_pricing") + require.Equal(t, float64(0.5), resp["original_price"]) + require.NotEqual(t, float64(0.33333), resp["original_price"]) +} + // TestGetUserPricing_LoggedIn_NoDiscount 已登录但无折扣(分组倍率=1,无个人倍率) func TestGetUserPricing_LoggedIn_NoDiscount(t *testing.T) { db := setupPricingTestDB(t) @@ -226,8 +257,6 @@ func TestGetUserPricing_LoggedIn_GroupDiscount(t *testing.T) { assert.Equal(t, "8折", resp["discount"]) } - - // TestGetUserPricing_PerCall_WithDiscount 按次计费 + 折扣 func TestGetUserPricing_PerCall_WithDiscount(t *testing.T) { db := setupPricingTestDB(t) @@ -280,4 +309,3 @@ func TestFormatDiscount(t *testing.T) { assert.Equal(t, tt.expected, result, "ratio=%.2f", tt.ratio) } } - diff --git a/model/pricing.go b/model/pricing.go index cf53161..eb5db68 100644 --- a/model/pricing.go +++ b/model/pricing.go @@ -14,23 +14,24 @@ import ( ) type Pricing struct { - ModelName string `json:"model_name"` - Description string `json:"description,omitempty"` - Icon string `json:"icon,omitempty"` - Tags string `json:"tags,omitempty"` - VendorID int `json:"vendor_id,omitempty"` - QuotaType int `json:"quota_type"` - ModelRatio float64 `json:"model_ratio"` - ModelPrice float64 `json:"model_price"` - OwnerBy string `json:"owner_by"` - CompletionRatio float64 `json:"completion_ratio"` - CacheRatio float64 `json:"cache_ratio"` - CacheCreationRatio float64 `json:"cache_creation_ratio"` - EnableGroup []string `json:"enable_groups"` - SupportedEndpointTypes []constant.EndpointType `json:"supported_endpoint_types"` - PricingVersion string `json:"pricing_version,omitempty"` - Type int `json:"type"` - PricingConfig *types.PricingConfig `json:"pricing_config,omitempty"` + ModelName string `json:"model_name"` + Description string `json:"description,omitempty"` + Icon string `json:"icon,omitempty"` + Tags string `json:"tags,omitempty"` + VendorID int `json:"vendor_id,omitempty"` + QuotaType int `json:"quota_type"` + ModelRatio float64 `json:"model_ratio"` + ModelPrice float64 `json:"model_price"` + OwnerBy string `json:"owner_by"` + CompletionRatio float64 `json:"completion_ratio"` + CacheRatio float64 `json:"cache_ratio"` + CacheCreationRatio float64 `json:"cache_creation_ratio"` + EnableGroup []string `json:"enable_groups"` + SupportedEndpointTypes []constant.EndpointType `json:"supported_endpoint_types"` + PricingVersion string `json:"pricing_version,omitempty"` + Type int `json:"type"` + PricingConfig *types.PricingConfig `json:"pricing_config,omitempty"` + DisplayPricing []types.ModelDisplayPricingItem `json:"display_pricing,omitempty"` } type PricingVendor struct { @@ -340,6 +341,7 @@ func updatePricing() { } applyGlobalDefault(&pricing, model) + applyDisplayPricing(&pricing, model) pricingMap = append(pricingMap, pricing) } @@ -426,6 +428,13 @@ func applyMatrixPricingConfig(pricing *Pricing, model string) bool { return true } +func applyDisplayPricing(pricing *Pricing, model string) { + items := ratio_setting.GetModelDisplayPricing(model) + if len(items) > 0 { + pricing.DisplayPricing = items + } +} + func minPricingConfigPrice(cfg *types.PricingConfig) float64 { if cfg == nil || len(cfg.Table) == 0 { return 0 diff --git a/model/pricing_test.go b/model/pricing_test.go index 2070c8b..fbe4b5e 100644 --- a/model/pricing_test.go +++ b/model/pricing_test.go @@ -189,6 +189,41 @@ func TestGetPricingByModel_IncludesMatrixPricingConfig(t *testing.T) { require.Equal(t, []constant.EndpointType{constant.EndpointTypeDoubaoVideo}, pricing.SupportedEndpointTypes) } +func TestGetPricingByModelIncludesDisplayPricing(t *testing.T) { + setupPricingTest(t) + + const displayModel = "display-pricing-model" + require.NoError(t, ratio_setting.UpdateModelRatioByJSONString(`{"`+displayModel+`":10}`)) + require.NoError(t, ratio_setting.SetModelDisplayPricing(displayModel, []types.ModelDisplayPricingItem{ + {Specification: "768p-6s", Price: 0.33333, Unit: "second", SortOrder: 1}, + })) + + require.NoError(t, (&Model{ + ModelName: displayModel, + Status: 1, + }).Insert()) + require.NoError(t, (&Channel{ + Id: 1, + Type: constant.ChannelTypeOpenAI, + Key: "k", + Name: "channel-a", + Status: common.ChannelStatusEnabled, + Models: displayModel, + Group: "default", + }).Insert()) + + RefreshPricing() + pricing := GetPricingByModel(displayModel) + + require.NotNil(t, pricing) + require.Len(t, pricing.DisplayPricing, 1) + require.Equal(t, "768p-6s", pricing.DisplayPricing[0].Specification) + require.Equal(t, QuotaTypeByTokens, pricing.QuotaType) + require.Equal(t, 10.0, pricing.ModelRatio) + require.Equal(t, 0.0, pricing.ModelPrice) + require.Nil(t, pricing.PricingConfig) +} + func TestParseModelEndpointTypes_SupportsObjectAndArray(t *testing.T) { objectEndpoints := parseModelEndpointTypes(`{ "anthropic": {"path": "/v1/messages", "method": "POST"}, From cc97834e5b8da3fe4a273fa0893a9f41eb11bfcc Mon Sep 17 00:00:00 2001 From: fengsilin Date: Wed, 1 Jul 2026 11:58:48 +0800 Subject: [PATCH 07/11] feat: add display pricing frontend helpers --- web/src/helpers/utils.jsx | 101 ++++++++++++++++++++++++++++++++++ web/src/helpers/utils.test.js | 92 +++++++++++++++++++++++++++++++ 2 files changed, 193 insertions(+) diff --git a/web/src/helpers/utils.jsx b/web/src/helpers/utils.jsx index afc0566..8253b1b 100644 --- a/web/src/helpers/utils.jsx +++ b/web/src/helpers/utils.jsx @@ -609,6 +609,93 @@ export const selectFilter = (input, option) => { // ------------------------------- // 模型定价计算工具函数 +export const hasDisplayPricing = (record) => + Array.isArray(record?.display_pricing) && record.display_pricing.length > 0; + +export const normalizeDisplayPricingItems = (record) => { + if (!hasDisplayPricing(record)) return []; + return [...record.display_pricing] + .map((item) => { + const discountRate = + Number(item.discount_rate) > 0 && Number(item.discount_rate) <= 1 + ? Number(item.discount_rate) + : 1; + const price = Number(item.price || 0); + return { + ...item, + price, + discount_rate: discountRate, + effectivePrice: price * discountRate, + unit: item.unit || 'call', + sort_order: Number(item.sort_order || 0), + }; + }) + .sort((a, b) => { + if (a.sort_order !== b.sort_order) return a.sort_order - b.sort_order; + return String(a.specification || '').localeCompare( + String(b.specification || ''), + ); + }); +}; + +export const formatDisplayPricingItem = ({ + item, + displayPrice, + t = (key) => key, +}) => { + const unit = item.unit || 'call'; + const effectivePrice = Number(item.effectivePrice ?? item.price ?? 0); + const hasDiscount = item.discount_rate > 0 && item.discount_rate < 1; + return { + ...item, + formattedPrice: `${displayPrice(effectivePrice)} / ${t(unit)}`, + originalPrice: hasDiscount + ? `${displayPrice(Number(item.price || 0))} / ${t(unit)}` + : '', + discountText: hasDiscount + ? `${Math.round(item.discount_rate * 100) / 10}折` + : '-', + }; +}; + +export const summarizeDisplayPricing = ({ + record, + displayPrice, + t = (key) => key, +}) => { + const items = normalizeDisplayPricingItems(record); + if (items.length === 0) return null; + + const groups = new Map(); + items.forEach((item) => { + const unit = item.unit || 'call'; + const list = groups.get(unit) || []; + list.push(item); + groups.set(unit, list); + }); + + const lines = Array.from(groups.entries()).map(([unit, list]) => { + const prices = list.map((item) => Number(item.effectivePrice || 0)); + const min = Math.min(...prices); + const max = Math.max(...prices); + const priceText = + min === max + ? displayPrice(min) + : `${displayPrice(min)} - ${displayPrice(max)}`; + return `${priceText} / ${t(unit)}`; + }); + + return { + isDisplayPricing: true, + price: lines.join('\n'), + unit: '', + details: items + .slice(0, 3) + .map((item) => formatDisplayPricingItem({ item, displayPrice, t })), + hiddenDetailCount: Math.max(0, items.length - 3), + }; +}; + export const hasMatrixPricing = (record) => !!( record?.pricing_config && @@ -724,6 +811,7 @@ export const calculateModelPrice = ({ displayPrice, currency, precision = 4, + t = (key) => key, }) => { // 1. 选择实际使用的分组 const ratios = groupRatio || {}; @@ -754,6 +842,19 @@ export const calculateModelPrice = ({ } // 2. 根据计费类型计算价格 + if (hasDisplayPricing(record)) { + const summary = summarizeDisplayPricing({ + record, + displayPrice, + t, + }); + return { + ...summary, + usedGroup, + usedGroupRatio, + }; + } + if (hasMatrixPricing(record)) { const summary = summarizeMatrixPricing({ record, diff --git a/web/src/helpers/utils.test.js b/web/src/helpers/utils.test.js index 39bfe77..1b4e0e3 100644 --- a/web/src/helpers/utils.test.js +++ b/web/src/helpers/utils.test.js @@ -18,7 +18,10 @@ globalThis.localStorage = { const { calculateModelPrice, formatMatrixPrice, + hasDisplayPricing, hasMatrixPricing, + normalizeDisplayPricingItems, + summarizeDisplayPricing, summarizeMatrixPricing, } = await import('./utils.jsx'); @@ -119,4 +122,93 @@ describe('model pricing helpers', () => { usedGroupRatio: 1, }); }); + + it('normalizes and sorts display pricing items', () => { + const record = { + display_pricing: [ + { + specification: '1080p', + price: 0.58333, + unit: 'second', + sort_order: 3, + discount_rate: 0, + }, + { + specification: '768p-6s', + price: 0.33333, + unit: 'second', + sort_order: 1, + discount_rate: 0.8, + }, + ], + }; + + expect(hasDisplayPricing(record)).toBe(true); + const items = normalizeDisplayPricingItems(record); + expect(items.map((item) => item.specification)).toEqual([ + '768p-6s', + '1080p', + ]); + expect(items[0].effectivePrice).toBeCloseTo(0.266664); + expect(items[1].discount_rate).toBe(1); + }); + + it('summarizes display pricing by unit', () => { + const summary = summarizeDisplayPricing({ + record: { + display_pricing: [ + { + specification: '768p-6s', + price: 0.33333, + unit: 'second', + sort_order: 1, + discount_rate: 1, + }, + { + specification: '1080p', + price: 0.58333, + unit: 'second', + sort_order: 2, + discount_rate: 1, + }, + ], + }, + displayPrice, + t: (key) => key, + }); + + expect(summary.isDisplayPricing).toBe(true); + expect(summary.price).toBe('$0.333 - $0.583 / second'); + expect(summary.details).toHaveLength(2); + }); + + it('calculateModelPrice prefers display pricing over matrix pricing', () => { + const result = calculateModelPrice({ + record: { + quota_type: 1, + model_price: 99, + pricing_config: { + billing_unit: 'per_call', + table: [{ duration: 5, price: 88, source: 'manual' }], + }, + display_pricing: [ + { + specification: '768p-6s', + price: 0.33333, + unit: 'second', + sort_order: 1, + discount_rate: 1, + }, + ], + }, + selectedGroup: 'default', + groupRatio: { default: 1 }, + tokenUnit: 'M', + displayPrice, + currency: 'USD', + }); + + expect(result.isDisplayPricing).toBe(true); + expect(result.price).toBe('$0.333 / second'); + }); }); From fee95041076902bb128c2b33aa00425ce53a0573 Mon Sep 17 00:00:00 2001 From: fengsilin Date: Wed, 1 Jul 2026 12:16:42 +0800 Subject: [PATCH 08/11] feat: render display pricing on pricing page --- .../modal/components/ModelPricingTable.jsx | 71 ++++++++++++++++++- .../modal/components/UserPriceComparison.jsx | 11 ++- .../view/card/PricingCardView.jsx | 42 ++++++++++- .../view/table/PricingTableColumns.jsx | 26 +++++++ web/src/i18n/locales/en.json | 11 ++- web/src/i18n/locales/zh-CN.json | 11 ++- 6 files changed, 165 insertions(+), 7 deletions(-) diff --git a/web/src/components/table/model-pricing/modal/components/ModelPricingTable.jsx b/web/src/components/table/model-pricing/modal/components/ModelPricingTable.jsx index ca38963..9ec627b 100644 --- a/web/src/components/table/model-pricing/modal/components/ModelPricingTable.jsx +++ b/web/src/components/table/model-pricing/modal/components/ModelPricingTable.jsx @@ -22,9 +22,12 @@ import { Card, Avatar, Typography, Table, Tag } from '@douyinfe/semi-ui'; import { IconCoinMoneyStroked } from '@douyinfe/semi-icons'; import { calculateModelPrice, + formatDisplayPricingItem, formatMatrixPrice, getMatrixPricingUnit, + hasDisplayPricing, hasMatrixPricing, + normalizeDisplayPricingItems, } from '../../../../../helpers'; const { Text } = Typography; @@ -44,6 +47,62 @@ const ModelPricingTable = ({ ? modelData.enable_groups : []; const autoChain = autoGroups.filter((g) => modelEnableGroups.includes(g)); + const renderDisplayPricingTable = () => { + const items = normalizeDisplayPricingItems(modelData).map((item) => + formatDisplayPricingItem({ item, displayPrice, t }), + ); + const columns = [ + { + title: t('规格'), + dataIndex: 'specification', + render: (text) => {text}, + }, + { + title: t('供应商说明'), + dataIndex: 'official_supplier_tip', + render: (text) => {text || '-'}, + }, + { + title: t('价格'), + dataIndex: 'formattedPrice', + render: (text, record) => ( +
+
{text}
+ {record.originalPrice && ( +
+ {t('原价')} {record.originalPrice} +
+ )} +
+ ), + }, + { + title: t('折扣'), + dataIndex: 'discountText', + render: (text) => text || '-', + }, + ]; + + return ( + <> +
+ {t('展示价格仅用于估算,实际消耗以任务完成后的积分扣减为准。')} +
+ ({ + ...item, + key: `${item.specification}-${index}`, + }))} + columns={columns} + pagination={false} + size='small' + bordered={false} + className='!rounded-lg' + /> + + ); + }; + const renderMatrixPriceTable = () => { const cfg = modelData?.pricing_config; const dimensions = cfg?.dimensions || []; @@ -124,6 +183,9 @@ const ModelPricingTable = ({ }; const renderGroupPriceTable = () => { + if (hasDisplayPricing(modelData)) { + return renderDisplayPricingTable(); + } if (hasMatrixPricing(modelData)) { return renderMatrixPriceTable(); } @@ -144,6 +206,7 @@ const ModelPricingTable = ({ tokenUnit, displayPrice, currency, + t, }) : { inputPrice: '-', outputPrice: '-', price: '-' }; @@ -322,9 +385,13 @@ const ModelPricingTable = ({
- {t('分组价格')} + + {hasDisplayPricing(modelData) ? t('展示价格') : t('分组价格')} +
- {t('不同用户分组的价格信息')} + {hasDisplayPricing(modelData) + ? t('展示价格,仅用于估算') + : t('不同用户分组的价格信息')}
diff --git a/web/src/components/table/model-pricing/modal/components/UserPriceComparison.jsx b/web/src/components/table/model-pricing/modal/components/UserPriceComparison.jsx index a103a89..ff35788 100644 --- a/web/src/components/table/model-pricing/modal/components/UserPriceComparison.jsx +++ b/web/src/components/table/model-pricing/modal/components/UserPriceComparison.jsx @@ -7,17 +7,22 @@ const { Text } = Typography; const UserPriceComparison = ({ modelName, + modelData, currency, tokenUnit, displayPrice, t, }) => { + const hasDisplayPricing = + Array.isArray(modelData?.display_pricing) && + modelData.display_pricing.length > 0; + const [loading, setLoading] = useState(false); const [priceData, setPriceData] = useState(null); useEffect(() => { const fetchUserPrice = async () => { - if (!modelName) return; + if (!modelName || hasDisplayPricing) return; setLoading(true); try { const res = await API.get( @@ -37,7 +42,7 @@ const UserPriceComparison = ({ } }; fetchUserPrice(); - }, [modelName]); + }, [modelName, hasDisplayPricing]); const currencySymbol = useMemo(() => { if (currency === 'CNY') return '¥'; @@ -74,6 +79,8 @@ const UserPriceComparison = ({ return `${currencySymbol}${num.toFixed(4)}`; }; + if (hasDisplayPricing) return null; + if (loading) { return ( diff --git a/web/src/components/table/model-pricing/view/card/PricingCardView.jsx b/web/src/components/table/model-pricing/view/card/PricingCardView.jsx index 0418390..5e16842 100644 --- a/web/src/components/table/model-pricing/view/card/PricingCardView.jsx +++ b/web/src/components/table/model-pricing/view/card/PricingCardView.jsx @@ -38,6 +38,7 @@ import { import { stringToColor, getLobeHubIcon, + hasDisplayPricing, hasMatrixPricing, calculateModelPrice, } from '../../../../../helpers'; @@ -163,8 +164,41 @@ const PricingCardView = ({ tokenUnit, displayPrice, currency, + t, }); + if (priceData?.isDisplayPricing) { + return ( +
+ {String(priceData.price || '') + .split('\n') + .map((line) => ( +
+ {line} +
+ ))} +
{t('展示价格')}
+ {(priceData.details || []).map((detail, detailIndex) => ( +
+ {detail.specification} + + {detail.formattedPrice} + +
+ ))} + {priceData.hiddenDetailCount > 0 && ( +
+ + {priceData.hiddenDetailCount} +
+ )} +
+ ); + } + if (!priceData?.isMatrix) { return null; } @@ -206,7 +240,13 @@ const PricingCardView = ({ - ); - if (hasMatrixPricing(record)) { + if (hasDisplayPricing(record)) { + billingTag = ( + + {t('展示价格')} + + ); + } else if (hasMatrixPricing(record)) { billingTag = ( {t('矩阵计费')} diff --git a/web/src/components/table/model-pricing/view/table/PricingTableColumns.jsx b/web/src/components/table/model-pricing/view/table/PricingTableColumns.jsx index 2c34870..e9cff94 100644 --- a/web/src/components/table/model-pricing/view/table/PricingTableColumns.jsx +++ b/web/src/components/table/model-pricing/view/table/PricingTableColumns.jsx @@ -24,6 +24,7 @@ import { renderModelTag, stringToColor, calculateModelPrice, + hasDisplayPricing, hasMatrixPricing, getLobeHubIcon, } from '../../../../../helpers'; @@ -53,6 +54,13 @@ function renderQuotaType(type, t) { } function renderBillingType(record, t) { + if (hasDisplayPricing(record)) { + return ( + + {t('展示价格')} + + ); + } if (hasMatrixPricing(record)) { return ( @@ -137,6 +145,7 @@ export const getPricingTableColumns = ({ tokenUnit, displayPrice, currency, + t, }); priceDataCache.set(record, cache); } @@ -244,6 +253,23 @@ export const getPricingTableColumns = ({ render: (text, record, index) => { const priceData = getPriceData(record); + if (priceData.isDisplayPricing) { + return ( +
+ {String(priceData.price || '') + .split('\n') + .map((line) => ( +
+ {line} +
+ ))} +
+ {t('展示价格,仅用于估算')} +
+
+ ); + } + if (priceData.isMatrix) { return (
diff --git a/web/src/i18n/locales/en.json b/web/src/i18n/locales/en.json index ed3b03e..cbc0e49 100644 --- a/web/src/i18n/locales/en.json +++ b/web/src/i18n/locales/en.json @@ -3295,5 +3295,14 @@ "重试启动失败": "Failed to start retry", "重试已启动": "Retry started", "验证完成": "Verification complete", - "暂无数据": "No data" + "暂无数据": "No data", + "展示价格,仅用于估算": "Display price, estimate only", + "展示价格仅用于估算,实际消耗以任务完成后的积分扣减为准。": "Display price is only an estimate. Actual consumption is based on the final quota deduction after the task completes.", + "规格": "Specification", + "供应商说明": "Supplier tip", + "折扣": "Discount", + "second": "second", + "call": "call", + "image": "image", + "video": "video" } diff --git a/web/src/i18n/locales/zh-CN.json b/web/src/i18n/locales/zh-CN.json index f082d8e..1c8848c 100644 --- a/web/src/i18n/locales/zh-CN.json +++ b/web/src/i18n/locales/zh-CN.json @@ -3249,5 +3249,14 @@ "重试启动失败": "重试启动失败", "重试已启动": "重试已启动", "验证完成": "验证完成", - "暂无数据": "暂无数据" + "暂无数据": "暂无数据", + "展示价格,仅用于估算": "展示价格,仅用于估算", + "展示价格仅用于估算,实际消耗以任务完成后的积分扣减为准。": "展示价格仅用于估算,实际消耗以任务完成后的积分扣减为准。", + "规格": "规格", + "供应商说明": "供应商说明", + "折扣": "折扣", + "second": "秒", + "call": "次", + "image": "张图片", + "video": "个视频" } From a61ca078fc1ee39f7c93303361845735250f07ba Mon Sep 17 00:00:00 2001 From: fengsilin Date: Wed, 1 Jul 2026 16:13:33 +0800 Subject: [PATCH 09/11] fix: localize display pricing units --- web/src/helpers/utils.jsx | 23 ++++++++++++++++++++--- web/src/helpers/utils.test.js | 14 ++++++++++++-- 2 files changed, 32 insertions(+), 5 deletions(-) diff --git a/web/src/helpers/utils.jsx b/web/src/helpers/utils.jsx index 8253b1b..f41adc4 100644 --- a/web/src/helpers/utils.jsx +++ b/web/src/helpers/utils.jsx @@ -644,13 +644,14 @@ export const formatDisplayPricingItem = ({ t = (key) => key, }) => { const unit = item.unit || 'call'; + const unitText = formatDisplayPricingUnit(unit, t); const effectivePrice = Number(item.effectivePrice ?? item.price ?? 0); const hasDiscount = item.discount_rate > 0 && item.discount_rate < 1; return { ...item, - formattedPrice: `${displayPrice(effectivePrice)} / ${t(unit)}`, + formattedPrice: `${displayPrice(effectivePrice)} / ${unitText}`, originalPrice: hasDiscount - ? `${displayPrice(Number(item.price || 0))} / ${t(unit)}` + ? `${displayPrice(Number(item.price || 0))} / ${unitText}` : '', discountText: hasDiscount ? `${Math.round(item.discount_rate * 100) / 10}折` @@ -658,6 +659,22 @@ export const formatDisplayPricingItem = ({ }; }; +const DISPLAY_PRICING_UNIT_FALLBACKS = { + second: '秒', + call: '次', + image: '张图片', + video: '个视频', +}; + +export const formatDisplayPricingUnit = (unit, t = (key) => key) => { + const normalizedUnit = unit || 'call'; + const translated = t(normalizedUnit); + if (translated && translated !== normalizedUnit) { + return translated; + } + return DISPLAY_PRICING_UNIT_FALLBACKS[normalizedUnit] || normalizedUnit; +}; + export const summarizeDisplayPricing = ({ record, displayPrice, @@ -682,7 +699,7 @@ export const summarizeDisplayPricing = ({ min === max ? displayPrice(min) : `${displayPrice(min)} - ${displayPrice(max)}`; - return `${priceText} / ${t(unit)}`; + return `${priceText} / ${formatDisplayPricingUnit(unit, t)}`; }); return { diff --git a/web/src/helpers/utils.test.js b/web/src/helpers/utils.test.js index 1b4e0e3..f831d3d 100644 --- a/web/src/helpers/utils.test.js +++ b/web/src/helpers/utils.test.js @@ -17,6 +17,8 @@ globalThis.localStorage = { const { calculateModelPrice, + formatDisplayPricingItem, + formatDisplayPricingUnit, formatMatrixPrice, hasDisplayPricing, hasMatrixPricing, @@ -151,6 +153,14 @@ describe('model pricing helpers', () => { ]); expect(items[0].effectivePrice).toBeCloseTo(0.266664); expect(items[1].discount_rate).toBe(1); + expect( + formatDisplayPricingItem({ + item: items[0], + displayPrice, + t: (key) => key, + }).discountText, + ).toBe('8折'); + expect(formatDisplayPricingUnit('second', (key) => key)).toBe('秒'); }); it('summarizes display pricing by unit', () => { @@ -178,7 +188,7 @@ describe('model pricing helpers', () => { }); expect(summary.isDisplayPricing).toBe(true); - expect(summary.price).toBe('$0.333 - $0.583 / second'); + expect(summary.price).toBe('$0.333 - $0.583 / 秒'); expect(summary.details).toHaveLength(2); }); @@ -209,6 +219,6 @@ describe('model pricing helpers', () => { }); expect(result.isDisplayPricing).toBe(true); - expect(result.price).toBe('$0.333 / second'); + expect(result.price).toBe('$0.333 / 秒'); }); }); From 56400b6c6123c2b5904bed74502d238d5a829346 Mon Sep 17 00:00:00 2001 From: fengsilin Date: Thu, 2 Jul 2026 10:11:09 +0800 Subject: [PATCH 10/11] =?UTF-8?q?feat:=20KlingAiping=20=E6=B8=A0=E9=81=93?= =?UTF-8?q?=E9=80=82=E9=85=8D=20+=20=E5=B1=95=E7=A4=BA=E4=BB=B7=E6=A0=BC?= =?UTF-8?q?=E7=AE=A1=E7=90=86=E5=90=8E=E5=8F=B0?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit - 新增 KlingAiping 渠道 (ChannelTypeKlingAiping = 59) - 新增 KlingAiping 原生路由 (text2video/image2video/omni-video 等) - 新增 KlingAiping adaptor (relay/channel/task/kling/aiping/) - 前端模型定价页面新增「真实计费 / 展示价格」Tab 切换 - 新增 DisplayPricing 组件,支持模型展示价格的增删改 - 新增 displayPricingConfig 工具函数及测试 - 修复 KlingAiping 路由路径匹配与 task 错误响应 Co-Authored-By: Claude --- constant/channel.go | 3 + controller/kling_aiping_native.go | 452 ++++++++++++++++++ controller/kling_aiping_native_test.go | 235 +++++++++ controller/relay.go | 32 +- relay/channel/task/kling/aiping/adaptor.go | 341 +++++++++++++ .../channel/task/kling/aiping/adaptor_test.go | 168 +++++++ relay/channel/task/kling/aiping/routes.go | 124 +++++ relay/relay_adaptor.go | 3 + router/video-router.go | 46 ++ router/video_router_test.go | 41 ++ web/src/constants/channel.constants.js | 5 + web/src/helpers/render.jsx | 1 + .../Operation/ModelPricing/DisplayPricing.jsx | 277 +++++++++++ .../ModelPricing/displayPricingConfig.js | 61 +++ .../ModelPricing/displayPricingConfig.test.js | 49 ++ .../Setting/Operation/ModelPricing/index.jsx | 121 ++++- 16 files changed, 1934 insertions(+), 25 deletions(-) create mode 100644 controller/kling_aiping_native.go create mode 100644 controller/kling_aiping_native_test.go create mode 100644 relay/channel/task/kling/aiping/adaptor.go create mode 100644 relay/channel/task/kling/aiping/adaptor_test.go create mode 100644 relay/channel/task/kling/aiping/routes.go create mode 100644 router/video_router_test.go create mode 100644 web/src/pages/Setting/Operation/ModelPricing/DisplayPricing.jsx create mode 100644 web/src/pages/Setting/Operation/ModelPricing/displayPricingConfig.js create mode 100644 web/src/pages/Setting/Operation/ModelPricing/displayPricingConfig.test.js diff --git a/constant/channel.go b/constant/channel.go index 4798f99..d54666a 100644 --- a/constant/channel.go +++ b/constant/channel.go @@ -56,6 +56,7 @@ const ( ChannelTypeReplicate = 56 ChannelTypeCodex = 57 ChannelTypeDoubaoVideoCompatibleAiping = 58 + ChannelTypeKlingAiping = 59 ChannelTypeDummy // this one is only for count, do not add any channel after this ) @@ -120,6 +121,7 @@ var ChannelBaseURLs = []string{ "https://api.replicate.com", //56 "https://chatgpt.com", //57 "", //58 + "https://aiping.cn/api", //59 } var ChannelTypeNames = map[int]string{ @@ -178,6 +180,7 @@ var ChannelTypeNames = map[int]string{ ChannelTypeReplicate: "Replicate", ChannelTypeCodex: "Codex", ChannelTypeDoubaoVideoCompatibleAiping: "DoubaoVideoCompatibleAiping", + ChannelTypeKlingAiping: "KlingAiping", } func GetChannelTypeName(channelType int) string { diff --git a/controller/kling_aiping_native.go b/controller/kling_aiping_native.go new file mode 100644 index 0000000..fcc6073 --- /dev/null +++ b/controller/kling_aiping_native.go @@ -0,0 +1,452 @@ +package controller + +import ( + "bytes" + "fmt" + "io" + "net/http" + "strconv" + "strings" + + "github.com/QuantumNous/new-api/common" + "github.com/QuantumNous/new-api/constant" + "github.com/QuantumNous/new-api/dto" + "github.com/QuantumNous/new-api/middleware" + "github.com/QuantumNous/new-api/model" + "github.com/QuantumNous/new-api/relay" + klingaiping "github.com/QuantumNous/new-api/relay/channel/task/kling/aiping" + relaycommon "github.com/QuantumNous/new-api/relay/common" + "github.com/QuantumNous/new-api/service" + "github.com/QuantumNous/new-api/types" + "github.com/gin-gonic/gin" +) + +func KlingAipingNativeTaskSubmit(c *gin.Context) { + route, ok := klingaiping.FindRoute(c.Request.Method, c.FullPath(), klingaiping.RouteKindSubmit) + if !ok { + c.JSON(http.StatusNotFound, gin.H{"code": 404, "message": "route not found"}) + return + } + + relayInfo, err := relaycommon.GenRelayInfo(c, types.RelayFormatTask, nil, nil) + if err != nil { + c.JSON(http.StatusInternalServerError, gin.H{"code": 500, "message": err.Error()}) + return + } + payload, err := readJSONPayload(c) + if err != nil { + c.JSON(http.StatusBadRequest, gin.H{"code": 400, "message": err.Error()}) + return + } + + configureKlingAipingTaskRelayInfo(c, relayInfo, route, payload) + + relayTaskWithInfo(c, relayInfo) +} + +func configureKlingAipingTaskRelayInfo(c *gin.Context, relayInfo *relaycommon.RelayInfo, route klingaiping.Route, payload map[string]any) { + // Native Kling routes bypass Distribute(), so force relayTaskWithInfo to + // select a channel instead of reading a preselected one from context. + if relayInfo.ChannelMeta == nil { + relayInfo.ChannelMeta = &relaycommon.ChannelMeta{} + } + modelName := resolveKlingAipingModel(payload, route) + relayInfo.OriginModelName = modelName + relayInfo.Action = route.Action + relaycommon.StoreTaskRequest(c, relayInfo, route.Action, relaycommon.TaskSubmitReq{ + Model: modelName, + Prompt: stringFromMap(payload, "prompt"), + Seconds: fmt.Sprint(payload["seconds"]), + Metadata: payload, + }) +} + +func KlingAipingNativeTaskFetch(c *gin.Context) { + route, ok := klingaiping.FindRoute(c.Request.Method, c.FullPath(), klingaiping.RouteKindFetch) + if !ok { + c.JSON(http.StatusNotFound, gin.H{"code": 404, "message": "route not found"}) + return + } + task, ok := getKlingAipingUserTask(c, c.Param("task_id"), route.Action) + if !ok { + return + } + refreshKlingAipingTaskIfNeeded(task) + c.JSON(http.StatusOK, buildKlingAipingTaskPayload(task)) +} + +func KlingAipingNativeTaskList(c *gin.Context) { + route, ok := klingaiping.FindRoute(c.Request.Method, c.FullPath(), klingaiping.RouteKindList) + if !ok { + c.JSON(http.StatusNotFound, gin.H{"code": 404, "message": "route not found"}) + return + } + pageNum, pageSize, err := parseKlingAipingPage(c) + if err != nil { + c.JSON(http.StatusUnprocessableEntity, gin.H{"code": 422, "message": err.Error()}) + return + } + tasks := model.TaskGetAllUserTask(c.GetInt("id"), (pageNum-1)*pageSize, pageSize, model.SyncTaskQueryParams{ + Platform: constant.TaskPlatform(strconv.Itoa(constant.ChannelTypeKlingAiping)), + Action: route.Action, + }) + data := make([]any, 0, len(tasks)) + for _, task := range tasks { + data = append(data, taskDataObject(task)) + } + c.JSON(http.StatusOK, gin.H{ + "code": 0, + "message": "success", + "request_id": c.GetString(common.RequestIdKey), + "data": data, + }) +} + +func KlingAipingNativeProxy(c *gin.Context) { + route, ok := klingaiping.FindRoute(c.Request.Method, c.FullPath(), klingaiping.RouteKindProxy) + if !ok { + c.JSON(http.StatusNotFound, gin.H{"code": 404, "message": "route not found"}) + return + } + channel, _, err := service.CacheGetRandomSatisfiedChannel(&service.RetryParam{ + Ctx: c, + TokenGroup: common.GetContextKeyString(c, constant.ContextKeyUsingGroup), + ModelName: route.BillingModel, + Retry: common.GetPointer(0), + RequiredChannelType: constant.ChannelTypeKlingAiping, + }) + if err != nil { + c.JSON(http.StatusServiceUnavailable, gin.H{"code": 503, "message": err.Error()}) + return + } + if setupErr := middleware.SetupContextForSelectedChannel(c, channel, route.BillingModel); setupErr != nil { + c.JSON(setupErr.StatusCode, gin.H{"code": setupErr.GetErrorCode(), "message": setupErr.Error()}) + return + } + + resp, err := doKlingAipingProxyRequest(c, route, channel) + if err != nil { + c.JSON(http.StatusBadGateway, gin.H{"code": 502, "message": err.Error()}) + return + } + defer resp.Body.Close() + copyProxyResponse(c, resp) +} + +func readJSONPayload(c *gin.Context) (map[string]any, error) { + body, err := common.GetBodyStorage(c) + if err != nil { + return nil, err + } + data, err := body.Bytes() + if err != nil { + return nil, err + } + _, _ = body.Seek(0, io.SeekStart) + c.Request.Body = io.NopCloser(body) + payload := map[string]any{} + if strings.TrimSpace(string(data)) == "" { + return payload, nil + } + if err := common.Unmarshal(data, &payload); err != nil { + return nil, err + } + return payload, nil +} + +func resolveKlingAipingModel(payload map[string]any, route klingaiping.Route) string { + if modelName := stringFromMap(payload, "model_name"); modelName != "" { + return modelName + } + if modelName := stringFromMap(payload, "model"); modelName != "" { + return modelName + } + if route.BillingModel != "" { + return route.BillingModel + } + return "kling-v3" +} + +func stringFromMap(payload map[string]any, key string) string { + if value, ok := payload[key].(string); ok { + return strings.TrimSpace(value) + } + return "" +} + +func getKlingAipingUserTask(c *gin.Context, taskID string, action string) (*model.Task, bool) { + task, exist, err := model.GetByTaskId(c.GetInt("id"), taskID) + if err != nil { + c.JSON(http.StatusInternalServerError, gin.H{"code": 500, "message": err.Error()}) + return nil, false + } + if !exist || task.Platform != constant.TaskPlatform(strconv.Itoa(constant.ChannelTypeKlingAiping)) || task.Action != action { + c.JSON(http.StatusBadRequest, gin.H{"code": 400, "message": "task not found"}) + return nil, false + } + return task, true +} + +func refreshKlingAipingTaskIfNeeded(task *model.Task) { + if task.Status == model.TaskStatusSuccess || task.Status == model.TaskStatusFailure { + return + } + channelModel, err := model.GetChannelById(task.ChannelId, true) + if err != nil || channelModel == nil { + return + } + adaptor := relay.GetTaskAdaptor(task.Platform) + if adaptor == nil { + return + } + resp, err := adaptor.FetchTask(channelModel.GetBaseURL(), channelModel.Key, map[string]any{ + "task_id": task.GetUpstreamTaskID(), + "action": task.Action, + }, channelModel.GetSetting().Proxy) + if err != nil || resp == nil { + return + } + defer resp.Body.Close() + body, err := io.ReadAll(resp.Body) + if err != nil || resp.StatusCode < 200 || resp.StatusCode >= 300 { + return + } + taskInfo, err := adaptor.ParseTaskResult(body) + if err != nil || taskInfo == nil { + return + } + snap := task.Snapshot() + task.Data = body + if taskInfo.Status != "" { + task.Status = model.TaskStatus(taskInfo.Status) + } + if taskInfo.Progress != "" { + task.Progress = taskInfo.Progress + } + if taskInfo.Url != "" { + task.PrivateData.ResultURL = taskInfo.Url + } + if !snap.Equal(task.Snapshot()) { + _, _ = task.UpdateWithStatus(snap.Status) + } +} + +func buildKlingAipingTaskPayload(task *model.Task) map[string]any { + payload := map[string]any{ + "code": 0, + "message": "success", + "data": taskDataObject(task), + } + return payload +} + +func taskDataObject(task *model.Task) map[string]any { + payload := map[string]any{} + _ = common.Unmarshal(task.Data, &payload) + delete(payload, "aiping_id") + data, _ := payload["data"].(map[string]any) + if data == nil { + data = map[string]any{} + } + data["task_id"] = task.TaskID + if _, ok := data["task_status"]; !ok { + data["task_status"] = mapKlingAipingTaskStatus(task.Status) + } + if _, ok := data["task_status_msg"]; !ok { + data["task_status_msg"] = task.FailReason + } + if _, ok := data["created_at"]; !ok && task.CreatedAt != 0 { + data["created_at"] = task.CreatedAt + } + if _, ok := data["updated_at"]; !ok && task.UpdatedAt != 0 { + data["updated_at"] = task.UpdatedAt + } + ensureKlingAipingWatermarkURL(data) + return data +} + +func mapKlingAipingTaskStatus(status model.TaskStatus) string { + switch status { + case model.TaskStatusSubmitted, model.TaskStatusQueued: + return "submitted" + case model.TaskStatusInProgress: + return "processing" + case model.TaskStatusSuccess: + return "succeed" + case model.TaskStatusFailure: + return "failed" + default: + return "processing" + } +} + +func ensureKlingAipingWatermarkURL(data map[string]any) { + taskResult, _ := data["task_result"].(map[string]any) + if taskResult == nil { + return + } + videos, _ := taskResult["videos"].([]any) + for _, videoAny := range videos { + video, _ := videoAny.(map[string]any) + if video == nil { + continue + } + if _, ok := video["watermark_url"]; !ok { + video["watermark_url"] = "" + } + } +} + +func parseKlingAipingPage(c *gin.Context) (int, int, error) { + pageNum := parseIntDefault(c.Query("pageNum"), 1) + pageSize := parseIntDefault(c.Query("pageSize"), 30) + if pageNum < 1 || pageNum > 1000 { + return 0, 0, fmt.Errorf("pageNum must be in [1, 1000]") + } + if pageSize < 1 || pageSize > 500 { + return 0, 0, fmt.Errorf("pageSize must be in [1, 500]") + } + return pageNum, pageSize, nil +} + +func parseIntDefault(raw string, fallback int) int { + if strings.TrimSpace(raw) == "" { + return fallback + } + v, err := strconv.Atoi(raw) + if err != nil { + return -1 + } + return v +} + +func doKlingAipingProxyRequest(c *gin.Context, route klingaiping.Route, channelModel *model.Channel) (*http.Response, error) { + baseURL := strings.TrimRight(channelModel.GetBaseURL(), "/") + upstreamPath := strings.Replace(route.UpstreamPath, ":id", c.Param("id"), 1) + url := baseURL + upstreamPath + if c.Request.URL.RawQuery != "" { + url += "?" + c.Request.URL.RawQuery + } + var body io.Reader + if c.Request.Method != http.MethodGet { + data, err := proxyBodyBytes(c, route) + if err != nil { + return nil, err + } + body = bytes.NewReader(data) + } + req, err := http.NewRequest(c.Request.Method, url, body) + if err != nil { + return nil, err + } + req.Header.Set("Accept", "application/json") + req.Header.Set("Content-Type", "application/json") + key := common.GetContextKeyString(c, constant.ContextKeyChannelKey) + if key == "" { + key = channelModel.Key + } + req.Header.Set("Authorization", "Bearer "+key) + client, err := service.GetHttpClientWithProxy(channelModel.GetSetting().Proxy) + if err != nil { + return nil, err + } + return client.Do(req) +} + +func proxyBodyBytes(c *gin.Context, route klingaiping.Route) ([]byte, error) { + storage, err := common.GetBodyStorage(c) + if err != nil { + return nil, err + } + data, err := storage.Bytes() + if err != nil { + return nil, err + } + if route.PathPattern == "/v1/elements/delete" { + payload := map[string]any{} + if err := common.Unmarshal(data, &payload); err == nil { + if _, ok := payload["element_id"]; !ok { + if id := stringFromMap(payload, "id"); id != "" { + payload["element_id"] = id + delete(payload, "id") + data, _ = common.Marshal(payload) + } + } + } + } + return data, nil +} + +func copyProxyResponse(c *gin.Context, resp *http.Response) { + if resp.StatusCode >= http.StatusBadRequest { + copyNormalizedProxyError(c, resp) + return + } + for key, values := range resp.Header { + for _, value := range values { + c.Writer.Header().Add(key, value) + } + } + c.Status(resp.StatusCode) + _, _ = io.Copy(c.Writer, resp.Body) +} + +func copyNormalizedProxyError(c *gin.Context, resp *http.Response) { + body, _ := io.ReadAll(resp.Body) + message := strings.TrimSpace(string(body)) + payload := map[string]any{} + if len(body) > 0 && common.Unmarshal(body, &payload) == nil { + if msg := stringFromMap(payload, "message"); msg != "" { + message = msg + } else if msg := stringFromMap(payload, "msg"); msg != "" { + message = msg + } else if detail, ok := payload["detail"]; ok { + if detailMap, ok := detail.(map[string]any); ok { + if msg := stringFromMap(detailMap, "message"); msg != "" { + message = msg + } else if msg := stringFromMap(detailMap, "msg"); msg != "" { + message = msg + } else { + message = fmt.Sprint(detail) + } + } else { + message = fmt.Sprint(detail) + } + } + delete(payload, "msg") + } else { + payload = map[string]any{} + } + if message == "" { + message = resp.Status + } + if _, ok := payload["code"]; !ok { + payload["code"] = resp.StatusCode + } + payload["message"] = message + payload["request_id"] = c.GetString(common.RequestIdKey) + c.JSON(resp.StatusCode, payload) +} + +func normalizeKlingAipingTaskError(taskErr *dto.TaskError) { + if taskErr == nil || strings.TrimSpace(taskErr.Message) == "" { + return + } + payload := map[string]any{} + if common.Unmarshal([]byte(taskErr.Message), &payload) != nil { + return + } + if msg := stringFromMap(payload, "message"); msg != "" { + taskErr.Message = msg + return + } + if msg := stringFromMap(payload, "msg"); msg != "" { + taskErr.Message = msg + return + } + if detail, ok := payload["detail"].(map[string]any); ok { + if msg := stringFromMap(detail, "message"); msg != "" { + taskErr.Message = msg + } + } +} diff --git a/controller/kling_aiping_native_test.go b/controller/kling_aiping_native_test.go new file mode 100644 index 0000000..70147bf --- /dev/null +++ b/controller/kling_aiping_native_test.go @@ -0,0 +1,235 @@ +package controller + +import ( + "io" + "net/http" + "net/http/httptest" + "strings" + "testing" + + "github.com/QuantumNous/new-api/common" + "github.com/QuantumNous/new-api/constant" + "github.com/QuantumNous/new-api/dto" + "github.com/QuantumNous/new-api/model" + klingaiping "github.com/QuantumNous/new-api/relay/channel/task/kling/aiping" + relaycommon "github.com/QuantumNous/new-api/relay/common" + "github.com/QuantumNous/new-api/service" + "github.com/gin-gonic/gin" + "github.com/stretchr/testify/require" +) + +func TestKlingAipingSubmitPreparationUsesModelNameBeforeModel(t *testing.T) { + c := newControllerJSONContext(t, "/v1/videos/text2video", `{ + "model":"Kling-V1.6", + "model_name":"Kling-V2.6", + "prompt":"prompt" + }`) + payload, err := readJSONPayload(c) + require.NoError(t, err) + route, ok := klingaiping.FindRoute(http.MethodPost, "/v1/videos/text2video", klingaiping.RouteKindSubmit) + require.True(t, ok) + info := &relaycommon.RelayInfo{TaskRelayInfo: &relaycommon.TaskRelayInfo{}} + modelName := resolveKlingAipingModel(payload, route) + relaycommon.StoreTaskRequest(c, info, route.Action, relaycommon.TaskSubmitReq{ + Model: modelName, + Prompt: stringFromMap(payload, "prompt"), + Metadata: payload, + }) + + stored, err := relaycommon.GetTaskRequest(c) + require.NoError(t, err) + require.Equal(t, "Kling-V2.6", stored.Model) + require.Equal(t, "prompt", stored.Prompt) + require.Equal(t, "Kling-V1.6", stored.Metadata["model"]) + require.Equal(t, "Kling-V2.6", stored.Metadata["model_name"]) +} + +func TestKlingAipingSubmitPreparationDoesNotLockChannel(t *testing.T) { + c := newControllerJSONContext(t, "/v1/kling/general/custom-voices", `{ + "voice_url":"https://example.com/voice.mp3", + "voice_name":"voice" + }`) + payload, err := readJSONPayload(c) + require.NoError(t, err) + route, ok := klingaiping.FindRoute(http.MethodPost, "/v1/kling/general/custom-voices", klingaiping.RouteKindSubmit) + require.True(t, ok) + info := &relaycommon.RelayInfo{TaskRelayInfo: &relaycommon.TaskRelayInfo{}} + info.OriginModelName = resolveKlingAipingModel(payload, route) + info.Action = route.Action + relaycommon.StoreTaskRequest(c, info, route.Action, relaycommon.TaskSubmitReq{ + Model: info.OriginModelName, + Metadata: payload, + }) + + require.Equal(t, klingaiping.ModelCustomVoices, info.OriginModelName) + require.Equal(t, klingaiping.ActionVoicesCreate, info.Action) + require.Nil(t, info.LockedChannel) +} + +func TestConfigureKlingAipingTaskRelayInfoForcesChannelSelection(t *testing.T) { + c := newControllerJSONContext(t, "/v1/videos/text2video", `{"model_name":"Kling-V2.6","prompt":"prompt"}`) + payload, err := readJSONPayload(c) + require.NoError(t, err) + route, ok := klingaiping.FindRoute(http.MethodPost, "/v1/videos/text2video", klingaiping.RouteKindSubmit) + require.True(t, ok) + info := &relaycommon.RelayInfo{TaskRelayInfo: &relaycommon.TaskRelayInfo{}} + + configureKlingAipingTaskRelayInfo(c, info, route, payload) + + require.NotNil(t, info.ChannelMeta) + require.Zero(t, info.ChannelMeta.ChannelType) + require.Equal(t, "Kling-V2.6", info.OriginModelName) + require.Equal(t, klingaiping.ActionText2Video, info.Action) + require.Nil(t, info.LockedChannel) +} + +func TestRequiredTaskChannelTypeOnlyMatchesKlingAipingNativePaths(t *testing.T) { + c := newControllerJSONContext(t, "/v1/videos/text2video", `{}`) + c.Request.URL.Path = "/v1/videos/text2video" + require.Equal(t, constant.ChannelTypeKlingAiping, requiredTaskChannelTypeForRequest(c)) + + c = newControllerJSONContext(t, "/v1/videos/video_123/remix", `{}`) + c.Request.URL.Path = "/v1/videos/video_123/remix" + require.Zero(t, requiredTaskChannelTypeForRequest(c)) +} + +func TestKlingAipingTaskDataObjectUsesPublicTaskIDAndWatermarkURL(t *testing.T) { + task := &model.Task{ + TaskID: "task_public", + Status: model.TaskStatusSuccess, + CreatedAt: 100, + UpdatedAt: 200, + Data: []byte(`{ + "code":0, + "aiping_id":"internal", + "data":{ + "task_id":"899333358055493641", + "task_status":"succeed", + "task_result":{"videos":[{"id":"v1","url":"https://example.com/video.mp4","duration":"5.041"}]} + } + }`), + } + + data := taskDataObject(task) + require.Equal(t, "task_public", data["task_id"]) + taskResult := data["task_result"].(map[string]any) + videos := taskResult["videos"].([]any) + require.Equal(t, "", videos[0].(map[string]any)["watermark_url"]) +} + +func TestProxyBodyBytesMapsElementDeleteIDToElementID(t *testing.T) { + c := newControllerJSONContext(t, "/v1/elements/delete", `{"id":"elem_1"}`) + route, ok := klingaiping.FindRoute(http.MethodPost, "/v1/elements/delete", klingaiping.RouteKindProxy) + require.True(t, ok) + + data, err := proxyBodyBytes(c, route) + + require.NoError(t, err) + var payload map[string]any + require.NoError(t, common.Unmarshal(data, &payload)) + require.Equal(t, "elem_1", payload["element_id"]) + require.NotContains(t, payload, "id") +} + +func TestDoKlingAipingProxyRequestUsesSelectedContextKey(t *testing.T) { + service.InitHttpClient() + var gotAuth string + server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) { + gotAuth = r.Header.Get("Authorization") + w.Header().Set("Content-Type", "application/json") + _, _ = w.Write([]byte(`{"code":0,"message":"success","data":[]}`)) + })) + defer server.Close() + + c := newControllerJSONContext(t, "/v1/elements/custom", ``) + common.SetContextKey(c, constant.ContextKeyChannelKey, "selected-key") + route, ok := klingaiping.FindRoute(http.MethodGet, "/v1/elements/custom", klingaiping.RouteKindProxy) + require.True(t, ok) + channel := &model.Channel{ + Key: "raw-channel-key", + BaseURL: common.GetPointer(server.URL), + } + + resp, err := doKlingAipingProxyRequest(c, route, channel) + require.NoError(t, err) + defer resp.Body.Close() + + require.Equal(t, "Bearer selected-key", gotAuth) +} + +func TestCopyProxyResponseNormalizesMsgError(t *testing.T) { + w := httptest.NewRecorder() + c, _ := gin.CreateTestContext(w) + c.Set(common.RequestIdKey, "req-test") + resp := &http.Response{ + StatusCode: http.StatusUnauthorized, + Header: http.Header{"Content-Type": []string{"application/json"}}, + Body: io.NopCloser(strings.NewReader(`{"code":401,"msg":"unauthorized","data":null}`)), + } + + copyProxyResponse(c, resp) + + require.Equal(t, http.StatusUnauthorized, w.Code) + require.Contains(t, w.Body.String(), `"message":"unauthorized"`) + require.NotContains(t, w.Body.String(), `"msg"`) + require.Contains(t, w.Body.String(), `"request_id":"req-test"`) +} + +func TestCopyProxyResponseNormalizesPlainTextError(t *testing.T) { + w := httptest.NewRecorder() + c, _ := gin.CreateTestContext(w) + c.Set(common.RequestIdKey, "req-test") + resp := &http.Response{ + StatusCode: http.StatusMethodNotAllowed, + Header: http.Header{"Content-Type": []string{"text/plain"}}, + Body: io.NopCloser(strings.NewReader("Method Not Allowed")), + } + + copyProxyResponse(c, resp) + + require.Equal(t, http.StatusMethodNotAllowed, w.Code) + require.Contains(t, w.Body.String(), `"message":"Method Not Allowed"`) + require.Contains(t, w.Body.String(), `"request_id":"req-test"`) +} + +func TestCopyProxyResponseNormalizesDetailMessageError(t *testing.T) { + w := httptest.NewRecorder() + c, _ := gin.CreateTestContext(w) + c.Set(common.RequestIdKey, "req-test") + resp := &http.Response{ + StatusCode: http.StatusServiceUnavailable, + Header: http.Header{"Content-Type": []string{"application/json"}}, + Body: io.NopCloser(strings.NewReader(`{"detail":{"message":"not found","error_type":"not_found"},"aiping_id":"internal"}`)), + } + + copyProxyResponse(c, resp) + + require.Equal(t, http.StatusServiceUnavailable, w.Code) + require.Contains(t, w.Body.String(), `"message":"not found"`) + require.NotContains(t, w.Body.String(), `map[`) + require.Contains(t, w.Body.String(), `"request_id":"req-test"`) +} + +func TestNormalizeKlingAipingTaskErrorMessageExtractsUpstreamJSONMessage(t *testing.T) { + taskErr := &dto.TaskError{ + Code: "fail_to_fetch_task", + Message: `{"code":400,"message":"ERROR: image download failed","request_id":"upstream"}`, + StatusCode: http.StatusBadRequest, + } + + normalizeKlingAipingTaskError(taskErr) + + require.Equal(t, "ERROR: image download failed", taskErr.Message) +} + +func TestParseKlingAipingPageBoundaries(t *testing.T) { + c := newControllerJSONContext(t, "/v1/videos/text2video?pageNum=1001&pageSize=30", `{}`) + c.Request.URL.RawQuery = "pageNum=1001&pageSize=30" + _, _, err := parseKlingAipingPage(c) + require.ErrorContains(t, err, "pageNum") + + c = newControllerJSONContext(t, "/v1/videos/text2video?pageNum=1&pageSize=501", `{}`) + c.Request.URL.RawQuery = "pageNum=1&pageSize=501" + _, _, err = parseKlingAipingPage(c) + require.ErrorContains(t, err, "pageSize") +} diff --git a/controller/relay.go b/controller/relay.go index 48734d2..27e0eb7 100644 --- a/controller/relay.go +++ b/controller/relay.go @@ -361,6 +361,9 @@ func requiredTaskChannelTypeForRequest(c *gin.Context) int { if strings.HasPrefix(c.Request.URL.Path, "/api/v3/contents/generations/tasks") { return constant.ChannelTypeDoubaoVideoCompatibleAiping } + if isKlingAipingNativePath(c.Request.URL.Path) { + return constant.ChannelTypeKlingAiping + } return 0 } @@ -554,7 +557,9 @@ func preloadTaskPricingConfig(c *gin.Context, info *relaycommon.RelayInfo) *dto. if err := common.Unmarshal(body, &payload); err != nil { return service.TaskErrorWrapperLocal(err, "invalid_json", http.StatusBadRequest) } - if raw, ok := payload["model"].(string); ok && strings.TrimSpace(raw) != "" { + if raw, ok := payload["model_name"].(string); ok && strings.TrimSpace(raw) != "" { + modelName = strings.TrimSpace(raw) + } else if raw, ok := payload["model"].(string); ok && strings.TrimSpace(raw) != "" { modelName = strings.TrimSpace(raw) } if raw, ok := payload["action"].(string); ok && strings.TrimSpace(raw) != "" { @@ -750,9 +755,34 @@ func respondTaskError(c *gin.Context, taskErr *dto.TaskError) { if taskErr.StatusCode == http.StatusTooManyRequests { taskErr.Message = "当前分组上游负载已饱和,请稍后再试" } + if isKlingAipingNativePath(c.Request.URL.Path) { + normalizeKlingAipingTaskError(taskErr) + } c.JSON(taskErr.StatusCode, taskErr) } +func isKlingAipingNativePath(path string) bool { + return pathMatchesAnyKlingAipingNativePrefix(path, + "/v1/videos/text2video", + "/v1/videos/image2video", + "/v1/videos/motion-control", + "/v1/videos/omni-video", + "/v1/videos/multi-image2video", + "/v1/videos/multi-elements", + "/v1/kling/general/advanced-custom-elements", + "/v1/kling/general/custom-voices", + ) +} + +func pathMatchesAnyKlingAipingNativePrefix(path string, prefixes ...string) bool { + for _, prefix := range prefixes { + if path == prefix || strings.HasPrefix(path, prefix+"/") { + return true + } + } + return false +} + func shouldRetryTaskRelay(c *gin.Context, channelId int, taskErr *dto.TaskError, retryTimes int) bool { if taskErr == nil { return false diff --git a/relay/channel/task/kling/aiping/adaptor.go b/relay/channel/task/kling/aiping/adaptor.go new file mode 100644 index 0000000..69be913 --- /dev/null +++ b/relay/channel/task/kling/aiping/adaptor.go @@ -0,0 +1,341 @@ +package aiping + +import ( + "bytes" + "fmt" + "io" + "net/http" + "strings" + "time" + + "github.com/QuantumNous/new-api/common" + "github.com/QuantumNous/new-api/dto" + "github.com/QuantumNous/new-api/model" + "github.com/QuantumNous/new-api/relay/channel" + taskcommon "github.com/QuantumNous/new-api/relay/channel/task/taskcommon" + relaycommon "github.com/QuantumNous/new-api/relay/common" + "github.com/QuantumNous/new-api/service" + "github.com/gin-gonic/gin" + "github.com/pkg/errors" +) + +type responsePayload struct { + Code int `json:"code"` + Message string `json:"message"` + RequestID string `json:"request_id"` + AipingID string `json:"aiping_id"` + Data struct { + TaskID string `json:"task_id"` + TaskStatus string `json:"task_status"` + TaskStatusMsg string `json:"task_status_msg"` + TaskInfo any `json:"task_info,omitempty"` + TaskResult struct { + Videos []struct { + ID string `json:"id"` + URL string `json:"url"` + WatermarkURL string `json:"watermark_url,omitempty"` + Duration string `json:"duration"` + } `json:"videos"` + } `json:"task_result"` + CreatedAt int64 `json:"created_at"` + UpdatedAt int64 `json:"updated_at"` + } `json:"data"` +} + +type TaskAdaptor struct { + taskcommon.BaseBilling + apiKey string + baseURL string +} + +func (a *TaskAdaptor) Init(info *relaycommon.RelayInfo) { + a.apiKey = info.ApiKey + a.baseURL = strings.TrimRight(info.ChannelBaseUrl, "/") +} + +func (a *TaskAdaptor) ValidateRequestAndSetAction(c *gin.Context, info *relaycommon.RelayInfo) *dto.TaskError { + req, err := relaycommon.GetTaskRequest(c) + if err != nil { + return service.TaskErrorWrapperLocal(fmt.Errorf("task request not prepared"), "invalid_request", http.StatusBadRequest) + } + modelName := resolveRequestModelName(req, info) + if isOmniModel(modelName) && (info.Action == ActionText2Video || info.Action == ActionImage2Video) { + return service.TaskErrorWrapperLocal( + fmt.Errorf("model %s must use /v1/videos/omni-video", modelName), + "invalid_model_route", + http.StatusUnprocessableEntity, + ) + } + return nil +} + +func (a *TaskAdaptor) BuildRequestURL(info *relaycommon.RelayInfo) (string, error) { + route, ok := FindRouteByAction(info.Action) + if !ok { + return "", fmt.Errorf("unsupported kling aiping action: %s", info.Action) + } + return a.baseURL + route.UpstreamPath, nil +} + +func (a *TaskAdaptor) BuildRequestHeader(c *gin.Context, req *http.Request, info *relaycommon.RelayInfo) error { + req.Header.Set("Content-Type", "application/json") + req.Header.Set("Accept", "application/json") + req.Header.Set("Authorization", "Bearer "+a.apiKey) + return nil +} + +func (a *TaskAdaptor) BuildRequestBody(c *gin.Context, info *relaycommon.RelayInfo) (io.Reader, error) { + req, err := relaycommon.GetTaskRequest(c) + if err != nil { + return nil, err + } + body := cloneMap(req.Metadata) + normalizeRequestBody(body, info) + data, err := common.Marshal(body) + if err != nil { + return nil, err + } + return bytes.NewReader(data), nil +} + +func (a *TaskAdaptor) DoRequest(c *gin.Context, info *relaycommon.RelayInfo, requestBody io.Reader) (*http.Response, error) { + return channel.DoTaskApiRequest(a, c, info, requestBody) +} + +func (a *TaskAdaptor) DoResponse(c *gin.Context, resp *http.Response, info *relaycommon.RelayInfo) (string, []byte, *dto.TaskError) { + responseBody, err := io.ReadAll(resp.Body) + if err != nil { + return "", nil, service.TaskErrorWrapper(err, "read_response_body_failed", http.StatusInternalServerError) + } + _ = resp.Body.Close() + + var parsed responsePayload + if err := common.Unmarshal(responseBody, &parsed); err != nil { + return "", nil, service.TaskErrorWrapper(errors.Wrapf(err, "body: %s", responseBody), "unmarshal_response_body_failed", http.StatusInternalServerError) + } + if parsed.Code != 0 { + msg := parsed.Message + if msg == "" { + msg = "upstream task failed" + } + return "", nil, service.TaskErrorWrapper(fmt.Errorf("%s", msg), "task_failed", http.StatusBadRequest) + } + upstreamTaskID := parsed.Data.TaskID + if strings.TrimSpace(upstreamTaskID) == "" { + return "", nil, service.TaskErrorWrapper(fmt.Errorf("task_id is empty"), "invalid_response", http.StatusInternalServerError) + } + + clientPayload := map[string]any{} + if err := common.Unmarshal(responseBody, &clientPayload); err != nil { + return "", nil, service.TaskErrorWrapper(err, "unmarshal_response_body_failed", http.StatusInternalServerError) + } + sanitizeNativeTaskPayload(clientPayload, info.PublicTaskID) + c.JSON(http.StatusOK, clientPayload) + return upstreamTaskID, responseBody, nil +} + +func (a *TaskAdaptor) FetchTask(baseURL, key string, body map[string]any, proxy string) (*http.Response, error) { + taskID, _ := body["task_id"].(string) + if strings.TrimSpace(taskID) == "" { + taskID, _ = body["upstream_task_id"].(string) + } + action, _ := body["action"].(string) + if strings.TrimSpace(taskID) == "" { + return nil, fmt.Errorf("invalid task_id") + } + route, ok := FindFetchRouteByAction(action) + if !ok { + return nil, fmt.Errorf("unsupported kling aiping action: %s", action) + } + url := strings.TrimRight(baseURL, "/") + strings.Replace(route.UpstreamPath, ":task_id", taskID, 1) + req, err := http.NewRequest(http.MethodGet, url, nil) + if err != nil { + return nil, err + } + req.Header.Set("Accept", "application/json") + req.Header.Set("Authorization", "Bearer "+key) + client, err := service.GetHttpClientWithProxy(proxy) + if err != nil { + return nil, fmt.Errorf("new proxy http client failed: %w", err) + } + return client.Do(req) +} + +func (a *TaskAdaptor) ParseTaskResult(respBody []byte) (*relaycommon.TaskInfo, error) { + var parsed responsePayload + if err := common.Unmarshal(respBody, &parsed); err != nil { + return nil, errors.Wrap(err, "unmarshal kling aiping task data failed") + } + info := &relaycommon.TaskInfo{ + Code: parsed.Code, + TaskID: parsed.Data.TaskID, + Reason: parsed.Data.TaskStatusMsg, + } + switch parsed.Data.TaskStatus { + case "submitted": + info.Status = model.TaskStatusSubmitted + info.Progress = taskcommon.ProgressSubmitted + case "processing": + info.Status = model.TaskStatusInProgress + info.Progress = taskcommon.ProgressInProgress + case "succeed": + info.Status = model.TaskStatusSuccess + info.Progress = taskcommon.ProgressComplete + case "failed": + info.Status = model.TaskStatusFailure + info.Progress = taskcommon.ProgressComplete + default: + info.Status = model.TaskStatusInProgress + info.Progress = taskcommon.ProgressInProgress + } + if len(parsed.Data.TaskResult.Videos) > 0 { + info.Url = parsed.Data.TaskResult.Videos[0].URL + } + return info, nil +} + +func (a *TaskAdaptor) GetModelList() []string { + return Models() +} + +func (a *TaskAdaptor) GetChannelName() string { + return "kling-aiping" +} + +func normalizeRequestBody(body map[string]any, info *relaycommon.RelayInfo) { + delete(body, "action_control") + delete(body, "uid") + delete(body, "create_at") + delete(body, "_standard_model") + + modelName := stringValue(body["model_name"]) + if modelName == "" { + modelName = stringValue(body["model"]) + } + if info != nil && info.IsModelMapped && info.UpstreamModelName != "" { + modelName = info.UpstreamModelName + } + if modelName != "" { + body["model_name"] = modelName + } + delete(body, "model") + + duration := body["duration"] + if duration == nil { + duration = body["seconds"] + } + if duration != nil { + body["duration"] = fmt.Sprint(duration) + } + delete(body, "seconds") + + if imageList, ok := body["reference_images"]; ok { + if _, exists := body["image_list"]; !exists { + body["image_list"] = imageList + } + delete(body, "reference_images") + } + if imageList, ok := normalizeImageList(body["image_list"]); ok { + body["image_list"] = imageList + } + + if body["watermark_info"] == nil { + return + } +} + +func normalizeImageList(v any) ([]map[string]any, bool) { + items, ok := v.([]any) + if !ok { + return nil, false + } + out := make([]map[string]any, 0, len(items)) + for _, item := range items { + switch val := item.(type) { + case string: + out = append(out, map[string]any{"image": val}) + case map[string]any: + image := firstString(val, "image", "image_url", "url", "base64") + if image == "" { + out = append(out, val) + continue + } + out = append(out, map[string]any{"image": image}) + default: + out = append(out, map[string]any{"image": fmt.Sprint(val)}) + } + } + return out, true +} + +func sanitizeNativeTaskPayload(payload map[string]any, publicTaskID string) { + delete(payload, "aiping_id") + data, _ := payload["data"].(map[string]any) + if data == nil { + return + } + data["task_id"] = publicTaskID + ensureWatermarkURL(data) + if _, ok := data["created_at"]; !ok { + data["created_at"] = time.Now().UnixMilli() + } + if _, ok := data["updated_at"]; !ok { + data["updated_at"] = data["created_at"] + } +} + +func ensureWatermarkURL(data map[string]any) { + taskResult, _ := data["task_result"].(map[string]any) + if taskResult == nil { + return + } + videos, _ := taskResult["videos"].([]any) + for _, videoAny := range videos { + video, _ := videoAny.(map[string]any) + if video == nil { + continue + } + if _, ok := video["watermark_url"]; !ok { + video["watermark_url"] = "" + } + } +} + +func cloneMap(in map[string]any) map[string]any { + out := make(map[string]any, len(in)) + for k, v := range in { + out[k] = v + } + return out +} + +func stringValue(v any) string { + if s, ok := v.(string); ok { + return strings.TrimSpace(s) + } + return "" +} + +func firstString(m map[string]any, keys ...string) string { + for _, key := range keys { + if s := stringValue(m[key]); s != "" { + return s + } + } + return "" +} + +func resolveRequestModelName(req relaycommon.TaskSubmitReq, info *relaycommon.RelayInfo) string { + if info != nil && info.ChannelMeta != nil && info.IsModelMapped && info.UpstreamModelName != "" { + return strings.TrimSpace(info.UpstreamModelName) + } + if modelName := firstString(req.Metadata, "model_name", "model"); modelName != "" { + return modelName + } + return strings.TrimSpace(req.Model) +} + +func isOmniModel(modelName string) bool { + normalized := strings.ToLower(strings.TrimSpace(modelName)) + return normalized == "kling-video-o1" || normalized == "kling-v3-omni" +} diff --git a/relay/channel/task/kling/aiping/adaptor_test.go b/relay/channel/task/kling/aiping/adaptor_test.go new file mode 100644 index 0000000..34f9941 --- /dev/null +++ b/relay/channel/task/kling/aiping/adaptor_test.go @@ -0,0 +1,168 @@ +package aiping + +import ( + "io" + "net/http" + "net/http/httptest" + "testing" + + "github.com/QuantumNous/new-api/common" + relaycommon "github.com/QuantumNous/new-api/relay/common" + "github.com/QuantumNous/new-api/service" + "github.com/gin-gonic/gin" + "github.com/stretchr/testify/require" +) + +func TestRoutesUseOfficialExternalAndAipingUpstreamPaths(t *testing.T) { + route, ok := FindRoute(http.MethodPost, "/v1/videos/text2video", RouteKindSubmit) + require.True(t, ok) + require.Equal(t, ActionText2Video, route.Action) + require.Equal(t, "/v1/videos/kling/text2video", route.UpstreamPath) + + voice, ok := FindRoute(http.MethodPost, "/v1/kling/general/custom-voices", RouteKindSubmit) + require.True(t, ok) + require.Equal(t, "/v1/kling/general/custom-voices", voice.UpstreamPath) + + presets, ok := FindRoute(http.MethodGet, "/v1/kling/general/presets-voices", RouteKindProxy) + require.True(t, ok) + require.Equal(t, "/v1/kling/general/presets-voices", presets.UpstreamPath) + + deleteVoices, ok := FindRoute(http.MethodPost, "/v1/kling/general/delete-voices", RouteKindProxy) + require.True(t, ok) + require.Equal(t, ActionDeleteVoices, deleteVoices.Action) + + deleteElements, ok := FindRoute(http.MethodPost, "/v1/kling/general/delete-elements", RouteKindProxy) + require.True(t, ok) + require.Equal(t, ActionDeleteElements, deleteElements.Action) +} + +func TestBuildRequestURLUsesAipingKlingPath(t *testing.T) { + adaptor := &TaskAdaptor{} + info := &relaycommon.RelayInfo{ + ChannelMeta: &relaycommon.ChannelMeta{ChannelBaseUrl: "https://aiping.cn/api"}, + TaskRelayInfo: &relaycommon.TaskRelayInfo{ + Action: ActionText2Video, + }, + } + adaptor.Init(info) + + url, err := adaptor.BuildRequestURL(info) + require.NoError(t, err) + require.Equal(t, "https://aiping.cn/api/v1/videos/kling/text2video", url) +} + +func TestFetchTaskUsesSingleTaskPath(t *testing.T) { + service.InitHttpClient() + var gotPath string + server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) { + gotPath = r.URL.Path + w.Header().Set("Content-Type", "application/json") + _, _ = w.Write([]byte(`{"code":0,"data":{"task_id":"upstream-task","task_status":"submitted"}}`)) + })) + defer server.Close() + + resp, err := (&TaskAdaptor{}).FetchTask(server.URL, "key", map[string]any{ + "task_id": "upstream-task", + "action": ActionText2Video, + }, "") + require.NoError(t, err) + defer resp.Body.Close() + + require.Equal(t, "/v1/videos/kling/text2video/upstream-task", gotPath) +} + +func TestBuildRequestBodyNormalizesOfficialCompatibleFields(t *testing.T) { + gin.SetMode(gin.TestMode) + c, _ := gin.CreateTestContext(nil) + info := &relaycommon.RelayInfo{ + ChannelMeta: &relaycommon.ChannelMeta{}, + TaskRelayInfo: &relaycommon.TaskRelayInfo{ + Action: ActionMultiImage2Video, + }, + OriginModelName: "Kling-V2.6", + } + relaycommon.StoreTaskRequest(c, info, ActionMultiImage2Video, relaycommon.TaskSubmitReq{ + Model: "Kling-V2.6", + Metadata: map[string]any{ + "model": "Kling-V1.6", + "model_name": "Kling-V2.6", + "seconds": 3, + "duration": 5, + "action_control": map[string]any{"type": "continuous"}, + "uid": "user-1", + "create_at": float64(1750000000000), + "_standard_model": "Kling-V-2-6", + "image_list": []any{ + "https://example.com/1.png", + map[string]any{"image_url": "https://example.com/2.png"}, + map[string]any{"url": "https://example.com/3.png"}, + map[string]any{"base64": "abc"}, + }, + }, + }) + + body, err := (&TaskAdaptor{}).BuildRequestBody(c, info) + require.NoError(t, err) + data, err := io.ReadAll(body) + require.NoError(t, err) + + var got map[string]any + require.NoError(t, common.Unmarshal(data, &got)) + require.Equal(t, "Kling-V2.6", got["model_name"]) + require.NotContains(t, got, "model") + require.Equal(t, "5", got["duration"]) + require.NotContains(t, got, "seconds") + require.NotContains(t, got, "action_control") + require.NotContains(t, got, "uid") + require.NotContains(t, got, "create_at") + require.NotContains(t, got, "_standard_model") + require.Equal(t, []any{ + map[string]any{"image": "https://example.com/1.png"}, + map[string]any{"image": "https://example.com/2.png"}, + map[string]any{"image": "https://example.com/3.png"}, + map[string]any{"image": "abc"}, + }, got["image_list"]) +} + +func TestValidateRequestRejectsOmniModelOnTextOrImageRoutes(t *testing.T) { + gin.SetMode(gin.TestMode) + c, _ := gin.CreateTestContext(nil) + info := &relaycommon.RelayInfo{ + TaskRelayInfo: &relaycommon.TaskRelayInfo{ + Action: ActionText2Video, + }, + } + relaycommon.StoreTaskRequest(c, info, ActionText2Video, relaycommon.TaskSubmitReq{ + Model: "kling-v3-omni", + Metadata: map[string]any{ + "model_name": "kling-v3-omni", + }, + }) + + taskErr := (&TaskAdaptor{}).ValidateRequestAndSetAction(c, info) + + require.NotNil(t, taskErr) + require.Equal(t, http.StatusUnprocessableEntity, taskErr.StatusCode) + require.Contains(t, taskErr.Message, "/v1/videos/omni-video") +} + +func TestSanitizeNativeTaskPayloadUsesPublicTaskIDAndAddsWatermarkURL(t *testing.T) { + payload := map[string]any{ + "aiping_id": "internal", + "data": map[string]any{ + "task_id": "899333358055493641", + "task_result": map[string]any{ + "videos": []any{ + map[string]any{"id": "v1", "url": "https://example.com/video.mp4"}, + }, + }, + }, + } + + sanitizeNativeTaskPayload(payload, "task_public") + require.NotContains(t, payload, "aiping_id") + data := payload["data"].(map[string]any) + require.Equal(t, "task_public", data["task_id"]) + videos := data["task_result"].(map[string]any)["videos"].([]any) + require.Equal(t, "", videos[0].(map[string]any)["watermark_url"]) +} diff --git a/relay/channel/task/kling/aiping/routes.go b/relay/channel/task/kling/aiping/routes.go new file mode 100644 index 0000000..362c77b --- /dev/null +++ b/relay/channel/task/kling/aiping/routes.go @@ -0,0 +1,124 @@ +package aiping + +import "net/http" + +const ( + ActionText2Video = "kling-text2video" + ActionImage2Video = "kling-image2video" + ActionMotionControl = "kling-motion-control" + ActionOmniVideo = "kling-omni-video" + ActionMultiImage2Video = "kling-multi-image2video" + ActionMultiElements = "kling-multi-elements" + ActionAdvancedElements = "kling-advanced-elements-create" + ActionVoicesCreate = "kling-voices-create" + ActionDeleteElements = "kling-delete-elements" + ActionDeleteVoices = "kling-voices-delete" +) + +const ( + RouteKindSubmit = "submit" + RouteKindFetch = "fetch" + RouteKindList = "list" + RouteKindProxy = "proxy" +) + +const ( + ModelKlingElements = "kling-elements" + ModelKlingAdvancedElements = "kling-advanced-elements" + ModelCustomVoices = "custom-voices" +) + +type Route struct { + Method string + PathPattern string + Action string + UpstreamPath string + Kind string + BillingModel string +} + +var Routes = []Route{ + {http.MethodPost, "/v1/videos/text2video", ActionText2Video, "/v1/videos/kling/text2video", RouteKindSubmit, ""}, + {http.MethodGet, "/v1/videos/text2video/:task_id", ActionText2Video, "/v1/videos/kling/text2video/:task_id", RouteKindFetch, ""}, + {http.MethodGet, "/v1/videos/text2video", ActionText2Video, "/v1/videos/kling/text2video", RouteKindList, ""}, + {http.MethodPost, "/v1/videos/image2video", ActionImage2Video, "/v1/videos/kling/image2video", RouteKindSubmit, ""}, + {http.MethodGet, "/v1/videos/image2video/:task_id", ActionImage2Video, "/v1/videos/kling/image2video/:task_id", RouteKindFetch, ""}, + {http.MethodGet, "/v1/videos/image2video", ActionImage2Video, "/v1/videos/kling/image2video", RouteKindList, ""}, + {http.MethodPost, "/v1/videos/motion-control", ActionMotionControl, "/v1/videos/kling/motion-control", RouteKindSubmit, "kling-v2-6"}, + {http.MethodGet, "/v1/videos/motion-control/:task_id", ActionMotionControl, "/v1/videos/kling/motion-control/:task_id", RouteKindFetch, "kling-v2-6"}, + {http.MethodGet, "/v1/videos/motion-control", ActionMotionControl, "/v1/videos/kling/motion-control", RouteKindList, "kling-v2-6"}, + {http.MethodPost, "/v1/videos/omni-video", ActionOmniVideo, "/v1/videos/kling/omni-video", RouteKindSubmit, ""}, + {http.MethodGet, "/v1/videos/omni-video/:task_id", ActionOmniVideo, "/v1/videos/kling/omni-video/:task_id", RouteKindFetch, ""}, + {http.MethodGet, "/v1/videos/omni-video", ActionOmniVideo, "/v1/videos/kling/omni-video", RouteKindList, ""}, + {http.MethodPost, "/v1/videos/multi-image2video", ActionMultiImage2Video, "/v1/videos/kling/multi-image2video", RouteKindSubmit, ""}, + {http.MethodGet, "/v1/videos/multi-image2video/:task_id", ActionMultiImage2Video, "/v1/videos/kling/multi-image2video/:task_id", RouteKindFetch, ""}, + {http.MethodGet, "/v1/videos/multi-image2video", ActionMultiImage2Video, "/v1/videos/kling/multi-image2video", RouteKindList, ""}, + {http.MethodPost, "/v1/videos/multi-elements", ActionMultiElements, "/v1/videos/kling/multi-elements", RouteKindSubmit, "kling-v1-6"}, + {http.MethodGet, "/v1/videos/multi-elements/:task_id", ActionMultiElements, "/v1/videos/kling/multi-elements/:task_id", RouteKindFetch, "kling-v1-6"}, + {http.MethodGet, "/v1/videos/multi-elements", ActionMultiElements, "/v1/videos/kling/multi-elements", RouteKindList, "kling-v1-6"}, + + {http.MethodPost, "/v1/videos/multi-elements/init-selection", "kling-multi-elements-init-selection", "/v1/videos/kling/multi-elements/init-selection", RouteKindProxy, ModelKlingElements}, + {http.MethodPost, "/v1/videos/multi-elements/add-selection", "kling-multi-elements-add-selection", "/v1/videos/kling/multi-elements/add-selection", RouteKindProxy, ModelKlingElements}, + {http.MethodPost, "/v1/videos/multi-elements/delete-selection", "kling-multi-elements-delete-selection", "/v1/videos/kling/multi-elements/delete-selection", RouteKindProxy, ModelKlingElements}, + {http.MethodPost, "/v1/videos/multi-elements/clear-selection", "kling-multi-elements-clear-selection", "/v1/videos/kling/multi-elements/clear-selection", RouteKindProxy, ModelKlingElements}, + {http.MethodPost, "/v1/videos/multi-elements/preview-selection", "kling-multi-elements-preview-selection", "/v1/videos/kling/multi-elements/preview-selection", RouteKindProxy, ModelKlingElements}, + + {http.MethodPost, "/v1/elements/custom", "kling-elements-custom-create", "/v1/elements/custom", RouteKindProxy, ModelKlingElements}, + {http.MethodGet, "/v1/elements/custom", "kling-elements-custom-list", "/v1/elements/custom", RouteKindProxy, ModelKlingElements}, + {http.MethodGet, "/v1/elements/:id", "kling-elements-get", "/v1/elements/:id", RouteKindProxy, ModelKlingElements}, + {http.MethodGet, "/v1/elements/presets", "kling-elements-presets", "/v1/elements/presets", RouteKindProxy, ModelKlingElements}, + {http.MethodPost, "/v1/elements/delete", "kling-elements-delete", "/v1/elements/delete", RouteKindProxy, ModelKlingElements}, + + {http.MethodPost, "/v1/kling/general/advanced-custom-elements", ActionAdvancedElements, "/v1/kling/general/advanced-custom-elements", RouteKindSubmit, ModelKlingAdvancedElements}, + {http.MethodGet, "/v1/kling/general/advanced-custom-elements/:task_id", ActionAdvancedElements, "/v1/kling/general/advanced-custom-elements/:task_id", RouteKindFetch, ModelKlingAdvancedElements}, + {http.MethodGet, "/v1/kling/general/advanced-custom-elements", "kling-advanced-elements-list", "/v1/kling/general/advanced-custom-elements", RouteKindProxy, ModelKlingAdvancedElements}, + {http.MethodGet, "/v1/kling/general/advanced-presets-elements", "kling-advanced-elements-presets", "/v1/kling/general/advanced-presets-elements", RouteKindProxy, ModelKlingAdvancedElements}, + {http.MethodPost, "/v1/kling/general/delete-elements", ActionDeleteElements, "/v1/kling/general/delete-elements", RouteKindProxy, ModelKlingAdvancedElements}, + + {http.MethodPost, "/v1/kling/general/custom-voices", ActionVoicesCreate, "/v1/kling/general/custom-voices", RouteKindSubmit, ModelCustomVoices}, + {http.MethodGet, "/v1/kling/general/custom-voices/:task_id", ActionVoicesCreate, "/v1/kling/general/custom-voices/:task_id", RouteKindFetch, ModelCustomVoices}, + {http.MethodGet, "/v1/kling/general/custom-voices", "kling-voices-list", "/v1/kling/general/custom-voices", RouteKindProxy, ModelCustomVoices}, + {http.MethodGet, "/v1/kling/general/presets-voices", "kling-voices-presets", "/v1/kling/general/presets-voices", RouteKindProxy, ModelCustomVoices}, + {http.MethodPost, "/v1/kling/general/delete-voices", ActionDeleteVoices, "/v1/kling/general/delete-voices", RouteKindProxy, ModelCustomVoices}, +} + +func FindRoute(method, pathPattern, kind string) (Route, bool) { + for _, route := range Routes { + if route.Method == method && route.PathPattern == pathPattern && route.Kind == kind { + return route, true + } + } + return Route{}, false +} + +func FindRouteByAction(action string) (Route, bool) { + for _, route := range Routes { + if route.Action == action && route.Kind == RouteKindSubmit { + return route, true + } + } + return Route{}, false +} + +func FindFetchRouteByAction(action string) (Route, bool) { + for _, route := range Routes { + if route.Action == action && route.Kind == RouteKindFetch { + return route, true + } + } + return Route{}, false +} + +func Models() []string { + return []string{ + "kling-v1", + "kling-v1-6", + "kling-v2-6", + "kling-v3", + "kling-video-o1", + "kling-v3-omni", + ModelKlingElements, + ModelKlingAdvancedElements, + ModelCustomVoices, + } +} diff --git a/relay/relay_adaptor.go b/relay/relay_adaptor.go index 06d0038..adeb06f 100644 --- a/relay/relay_adaptor.go +++ b/relay/relay_adaptor.go @@ -37,6 +37,7 @@ import ( "github.com/QuantumNous/new-api/relay/channel/task/hailuo" taskjimeng "github.com/QuantumNous/new-api/relay/channel/task/jimeng" "github.com/QuantumNous/new-api/relay/channel/task/kling" + klingaiping "github.com/QuantumNous/new-api/relay/channel/task/kling/aiping" tasksora "github.com/QuantumNous/new-api/relay/channel/task/sora" "github.com/QuantumNous/new-api/relay/channel/task/suno" taskvertex "github.com/QuantumNous/new-api/relay/channel/task/vertex" @@ -156,6 +157,8 @@ func GetTaskAdaptor(platform constant.TaskPlatform) channel.TaskAdaptor { return &taskdoubao.TaskAdaptor{} case constant.ChannelTypeDoubaoVideoCompatibleAiping: return &taskdoubaoaiping.TaskAdaptor{} + case constant.ChannelTypeKlingAiping: + return &klingaiping.TaskAdaptor{} case constant.ChannelTypeSora, constant.ChannelTypeOpenAI: return &tasksora.TaskAdaptor{} case constant.ChannelTypeGemini: diff --git a/router/video-router.go b/router/video-router.go index dd31771..11debfe 100644 --- a/router/video-router.go +++ b/router/video-router.go @@ -22,6 +22,52 @@ func SetVideoRouter(router *gin.Engine) { videoV1Router.GET("/video/generations/:task_id", controller.RelayTaskFetch) videoV1Router.POST("/videos/:video_id/remix", controller.RelayTask) } + klingAipingRouter := router.Group("/v1") + klingAipingRouter.Use(middleware.TokenAuth()) + { + klingAipingRouter.POST("/videos/multi-elements/init-selection", controller.KlingAipingNativeProxy) + klingAipingRouter.POST("/videos/multi-elements/add-selection", controller.KlingAipingNativeProxy) + klingAipingRouter.POST("/videos/multi-elements/delete-selection", controller.KlingAipingNativeProxy) + klingAipingRouter.POST("/videos/multi-elements/clear-selection", controller.KlingAipingNativeProxy) + klingAipingRouter.POST("/videos/multi-elements/preview-selection", controller.KlingAipingNativeProxy) + + klingAipingRouter.POST("/videos/text2video", controller.KlingAipingNativeTaskSubmit) + klingAipingRouter.GET("/videos/text2video/:task_id", controller.KlingAipingNativeTaskFetch) + klingAipingRouter.GET("/videos/text2video", controller.KlingAipingNativeTaskList) + klingAipingRouter.POST("/videos/image2video", controller.KlingAipingNativeTaskSubmit) + klingAipingRouter.GET("/videos/image2video/:task_id", controller.KlingAipingNativeTaskFetch) + klingAipingRouter.GET("/videos/image2video", controller.KlingAipingNativeTaskList) + klingAipingRouter.POST("/videos/motion-control", controller.KlingAipingNativeTaskSubmit) + klingAipingRouter.GET("/videos/motion-control/:task_id", controller.KlingAipingNativeTaskFetch) + klingAipingRouter.GET("/videos/motion-control", controller.KlingAipingNativeTaskList) + klingAipingRouter.POST("/videos/omni-video", controller.KlingAipingNativeTaskSubmit) + klingAipingRouter.GET("/videos/omni-video/:task_id", controller.KlingAipingNativeTaskFetch) + klingAipingRouter.GET("/videos/omni-video", controller.KlingAipingNativeTaskList) + klingAipingRouter.POST("/videos/multi-image2video", controller.KlingAipingNativeTaskSubmit) + klingAipingRouter.GET("/videos/multi-image2video/:task_id", controller.KlingAipingNativeTaskFetch) + klingAipingRouter.GET("/videos/multi-image2video", controller.KlingAipingNativeTaskList) + klingAipingRouter.POST("/videos/multi-elements", controller.KlingAipingNativeTaskSubmit) + klingAipingRouter.GET("/videos/multi-elements/:task_id", controller.KlingAipingNativeTaskFetch) + klingAipingRouter.GET("/videos/multi-elements", controller.KlingAipingNativeTaskList) + + klingAipingRouter.POST("/elements/custom", controller.KlingAipingNativeProxy) + klingAipingRouter.GET("/elements/custom", controller.KlingAipingNativeProxy) + klingAipingRouter.GET("/elements/presets", controller.KlingAipingNativeProxy) + klingAipingRouter.GET("/elements/:id", controller.KlingAipingNativeProxy) + klingAipingRouter.POST("/elements/delete", controller.KlingAipingNativeProxy) + + klingAipingRouter.POST("/kling/general/advanced-custom-elements", controller.KlingAipingNativeTaskSubmit) + klingAipingRouter.GET("/kling/general/advanced-custom-elements/:task_id", controller.KlingAipingNativeTaskFetch) + klingAipingRouter.GET("/kling/general/advanced-custom-elements", controller.KlingAipingNativeProxy) + klingAipingRouter.GET("/kling/general/advanced-presets-elements", controller.KlingAipingNativeProxy) + klingAipingRouter.POST("/kling/general/delete-elements", controller.KlingAipingNativeProxy) + klingAipingRouter.POST("/kling/general/custom-voices", controller.KlingAipingNativeTaskSubmit) + klingAipingRouter.GET("/kling/general/custom-voices/:task_id", controller.KlingAipingNativeTaskFetch) + klingAipingRouter.GET("/kling/general/custom-voices", controller.KlingAipingNativeProxy) + klingAipingRouter.GET("/kling/general/presets-voices", controller.KlingAipingNativeProxy) + klingAipingRouter.POST("/kling/general/delete-voices", controller.KlingAipingNativeProxy) + } + // openai compatible API video routes // docs: https://platform.openai.com/docs/api-reference/videos/create { diff --git a/router/video_router_test.go b/router/video_router_test.go new file mode 100644 index 0000000..20e8c78 --- /dev/null +++ b/router/video_router_test.go @@ -0,0 +1,41 @@ +package router + +import ( + "net/http" + "strings" + "testing" + + "github.com/gin-gonic/gin" + "github.com/stretchr/testify/require" +) + +func TestKlingAipingNativeRoutesDoNotFallThroughToOpenAIVideoRoutes(t *testing.T) { + gin.SetMode(gin.TestMode) + r := gin.New() + SetVideoRouter(r) + + requireHandlerContains(t, r, http.MethodPost, "/v1/videos/text2video", "KlingAipingNativeTaskSubmit") + requireHandlerContains(t, r, http.MethodGet, "/v1/videos/text2video/:task_id", "KlingAipingNativeTaskFetch") + requireHandlerContains(t, r, http.MethodGet, "/v1/videos/text2video", "KlingAipingNativeTaskList") + requireHandlerContains(t, r, http.MethodPost, "/v1/videos/multi-elements/init-selection", "KlingAipingNativeProxy") + requireHandlerContains(t, r, http.MethodPost, "/v1/videos/multi-elements", "KlingAipingNativeTaskSubmit") + requireHandlerContains(t, r, http.MethodGet, "/v1/elements/presets", "KlingAipingNativeProxy") + requireHandlerContains(t, r, http.MethodGet, "/v1/elements/:id", "KlingAipingNativeProxy") + requireHandlerContains(t, r, http.MethodPost, "/v1/kling/general/delete-elements", "KlingAipingNativeProxy") + requireHandlerContains(t, r, http.MethodPost, "/v1/kling/general/delete-voices", "KlingAipingNativeProxy") + + requireHandlerContains(t, r, http.MethodPost, "/v1/videos", "RelayTask") + requireHandlerContains(t, r, http.MethodGet, "/v1/videos/:task_id", "RelayTaskFetch") +} + +func requireHandlerContains(t *testing.T, r *gin.Engine, method, path, want string) { + t.Helper() + for _, route := range r.Routes() { + if route.Method != method || route.Path != path { + continue + } + require.Truef(t, strings.Contains(route.Handler, want), "handler for %s %s = %s, want containing %s", method, path, route.Handler, want) + return + } + t.Fatalf("route not found: %s %s", method, path) +} diff --git a/web/src/constants/channel.constants.js b/web/src/constants/channel.constants.js index 265a94e..065392c 100644 --- a/web/src/constants/channel.constants.js +++ b/web/src/constants/channel.constants.js @@ -194,6 +194,11 @@ export const CHANNEL_OPTIONS = [ color: 'blue', label: 'DoubaoVideoCompatibleAiping', }, + { + value: 59, + color: 'green', + label: 'KlingAiping', + }, ]; export const MODEL_TABLE_PAGE_SIZE = 10; diff --git a/web/src/helpers/render.jsx b/web/src/helpers/render.jsx index 118affa..bdfee36 100644 --- a/web/src/helpers/render.jsx +++ b/web/src/helpers/render.jsx @@ -395,6 +395,7 @@ export function getChannelIcon(channelType) { case 49: // Coze return ; case 50: // 可灵 Kling + case 59: // KlingAiping return ; case 51: // 即梦 Jimeng return ; diff --git a/web/src/pages/Setting/Operation/ModelPricing/DisplayPricing.jsx b/web/src/pages/Setting/Operation/ModelPricing/DisplayPricing.jsx new file mode 100644 index 0000000..f58b226 --- /dev/null +++ b/web/src/pages/Setting/Operation/ModelPricing/DisplayPricing.jsx @@ -0,0 +1,277 @@ +import React, { useEffect, useMemo, useState } from 'react'; +import { + Button, + Input, + Popconfirm, + SideSheet, + Space, + Table, + Tag, + TextArea, + Typography, +} from '@douyinfe/semi-ui'; +import { Edit, Plus, RefreshCw, Trash2 } from 'lucide-react'; +import { useTranslation } from 'react-i18next'; +import { API, showError, showSuccess } from '../../../../helpers'; +import { + createDefaultDisplayPricingItems, + formatDisplayPricingJson, + normalizeDisplayPricingItems, + parseDisplayPricingJson, +} from './displayPricingConfig'; + +export default function DisplayPricing() { + const { t } = useTranslation(); + const [rules, setRules] = useState({}); + const [keyword, setKeyword] = useState(''); + const [editingModel, setEditingModel] = useState(null); + const [editingItems, setEditingItems] = useState([]); + const [loading, setLoading] = useState(false); + + const refresh = async () => { + setLoading(true); + try { + const res = await API.get('/api/option/model_display_pricing'); + if (!res.data?.success) { + showError(res.data?.message || t('加载展示价格失败')); + return; + } + setRules(res.data.data || {}); + } catch (error) { + showError(error.message || t('加载展示价格失败')); + } finally { + setLoading(false); + } + }; + + useEffect(() => { + refresh(); + }, []); + + const data = useMemo( + () => + Object.entries(rules || {}) + .filter(([model]) => + model.toLowerCase().includes(keyword.toLowerCase()), + ) + .sort(([left], [right]) => left.localeCompare(right)) + .map(([model, items]) => { + const normalized = normalizeDisplayPricingItems(items); + return { + model, + count: normalized.length, + preview: normalized.slice(0, 3), + items: normalized, + }; + }), + [rules, keyword], + ); + + const openEditor = (model, items) => { + setEditingModel(model); + setEditingItems(items || createDefaultDisplayPricingItems()); + }; + + const remove = async (model) => { + try { + const res = await API.delete( + `/api/option/model_display_pricing/${encodeURIComponent(model)}`, + ); + if (!res.data?.success) { + showError(res.data?.message || t('删除展示价格失败')); + return; + } + showSuccess(t('展示价格已删除')); + await refresh(); + } catch (error) { + showError(error.message || t('删除展示价格失败')); + } + }; + + const columns = [ + { title: t('模型'), dataIndex: 'model' }, + { title: t('规格数量'), dataIndex: 'count', width: 110 }, + { + title: t('规格预览'), + dataIndex: 'preview', + render: (items) => ( + + {(items || []).map((item) => ( + + {item.specification}: {item.price} / {item.unit} + + ))} + + ), + }, + { + title: t('操作'), + width: 120, + render: (_, row) => ( + + + + + + +
+ setEditingModel(null)} + onSaved={refresh} + /> + + ); +} + +function DisplayPricingEditor({ visible, modelName, items, onClose, onSaved }) { + const { t } = useTranslation(); + const [name, setName] = useState(modelName || ''); + const [jsonText, setJsonText] = useState(formatDisplayPricingJson(items)); + const [saving, setSaving] = useState(false); + + useEffect(() => { + if (!visible) return; + setName(modelName || ''); + setJsonText(formatDisplayPricingJson(items)); + }, [visible, modelName, items]); + + const save = async () => { + if (!name.trim()) { + showError(t('模型名称不能为空')); + return; + } + const parsed = parseDisplayPricingJson(jsonText); + if (!parsed.ok) { + showError(`${t('JSON 格式错误')}: ${parsed.error}`); + return; + } + + setSaving(true); + try { + const res = await API.put( + `/api/option/model_display_pricing/${encodeURIComponent(name.trim())}`, + parsed.items, + ); + if (!res.data?.success) { + showError(res.data?.message || t('保存展示价格失败')); + return; + } + showSuccess(t('展示价格已保存')); + await onSaved?.(); + onClose?.(); + } catch (error) { + showError(error.message || t('保存展示价格失败')); + } finally { + setSaving(false); + } + }; + + return ( + + + + + } + > + + + + {t( + '填写数组 JSON。每一项包含 specification、official_supplier_tip、price、unit、sort_order、discount_rate;price 使用 USD 基准价。', + )} + +