feat: 完善模型请求适配与输出限制

2026-07-17 13:52:00 +08:00
parent a24eb1aeb0
commit 5ee267ecbd
31 changed files with 3287 additions and 232 deletions
@@ -0,0 +1,216 @@
+UPDATE model_catalog_providers
+SET default_auth_type = 'APIKey',
+    updated_at = now()
+WHERE provider_key = 'keling'
+   OR provider_code = 'keling';
+
+UPDATE integration_platforms
+SET auth_type = CASE
+      WHEN COALESCE(NULLIF(trim(credentials->>'accessKey'), ''), NULLIF(trim(credentials->>'secretKey'), '')) IS NULL
+        THEN 'APIKey'
+      ELSE auth_type
+    END,
+    credentials = CASE
+      WHEN credentials ? 'apiKey' THEN credentials
+      ELSE credentials || '{"apiKey": ""}'::jsonb
+    END,
+    updated_at = now()
+WHERE provider = 'keling'
+  AND deleted_at IS NULL;
+
+WITH keling_30_turbo AS (
+  SELECT
+    'keling:kling-3.0-turbo' AS canonical_model_key,
+    'kling-3.0-turbo' AS provider_model_name,
+    '可灵3.0 Turbo' AS display_name,
+    '["video_generate","image_to_video"]'::jsonb AS model_type,
+    '{
+      "video_generate": {
+        "aspect_ratio_allowed": ["16:9", "1:1", "9:16"],
+        "output_resolutions": ["720p", "1080p"],
+        "duration_range": [3, 15],
+        "output_audio": true,
+        "prompt_length_limit": {
+          "max": 3072,
+          "label": "可灵3.0 Turbo"
+        }
+      },
+      "image_to_video": {
+        "output_resolutions": ["720p", "1080p"],
+        "duration_range": [3, 15],
+        "input_first_frame": true,
+        "input_last_frame": false,
+        "input_first_last_frame": false,
+        "aspect_ratio_allowed": [],
+        "output_audio": true,
+        "input_reference_generate_single": false,
+        "input_reference_generate_multiple": false,
+        "support_video_effect_template": false,
+        "prompt_length_limit": {
+          "max": 2500,
+          "label": "可灵3.0 Turbo"
+        }
+      },
+      "originalTypes": ["video_generate", "image_to_video"]
+    }'::jsonb AS capabilities,
+    COALESCE(
+      (
+        SELECT base_billing_config
+        FROM base_model_catalog
+        WHERE canonical_model_key = 'keling:kling-v3'
+        LIMIT 1
+      ),
+      '{
+        "video": {
+          "basePrice": 100,
+          "baseWeight": 1,
+          "dynamicWeight": {
+            "720p": 1,
+            "1080p": 1.5,
+            "audio-true": 2,
+            "audio-false": 1
+          }
+        },
+        "currency": "resource"
+      }'::jsonb
+    ) AS base_billing_config,
+    COALESCE(
+      (
+        SELECT default_rate_limit_policy
+        FROM base_model_catalog
+        WHERE canonical_model_key = 'keling:kling-v3'
+        LIMIT 1
+      ),
+      '{"platformLimits":{"max_concurrent_requests":5}}'::jsonb
+    ) AS default_rate_limit_policy,
+    '{
+      "source": "server-main.integration-platform",
+      "sourceProviderCode": "keling",
+      "sourceProviderName": "可灵AI",
+      "sourceSpecType": "keling",
+      "originalTypes": ["video_generate", "image_to_video"],
+      "alias": "可灵3.0 Turbo",
+      "description": "可灵新任务 API 的 3.0 Turbo 文生视频与首帧图生视频模型",
+      "iconPath": "https://static.51easyai.com/kling-color.webp",
+      "billingType": "external-api",
+      "billingMode": "",
+      "referenceModel": "",
+      "modelWeight": null,
+      "selectable": true,
+      "rawModel": {
+        "name": "kling-3.0-turbo",
+        "types": ["video_generate", "image_to_video"],
+        "alias": "可灵3.0 Turbo",
+        "icon_path": "https://static.51easyai.com/kling-color.webp"
+      }
+    }'::jsonb AS metadata
+)
+INSERT INTO base_model_catalog (
+  provider_id,
+  provider_key,
+  canonical_model_key,
+  provider_model_name,
+  model_type,
+  display_name,
+  capabilities,
+  base_billing_config,
+  default_rate_limit_policy,
+  metadata,
+  catalog_type,
+  default_snapshot,
+  status
+)
+SELECT
+  (
+    SELECT id
+    FROM model_catalog_providers
+    WHERE provider_key = 'keling' OR provider_code = 'keling'
+    LIMIT 1
+  ),
+  'keling',
+  model.canonical_model_key,
+  model.provider_model_name,
+  model.model_type,
+  model.display_name,
+  model.capabilities,
+  model.base_billing_config,
+  model.default_rate_limit_policy,
+  jsonb_set(model.metadata, '{rawModel,capabilities}', model.capabilities, true),
+  'system',
+  jsonb_build_object(
+    'providerKey', 'keling',
+    'canonicalModelKey', model.canonical_model_key,
+    'providerModelName', model.provider_model_name,
+    'modelType', model.model_type,
+    'modelAlias', model.display_name,
+    'displayName', model.display_name,
+    'capabilities', model.capabilities,
+    'baseBillingConfig', model.base_billing_config,
+    'defaultRateLimitPolicy', model.default_rate_limit_policy,
+    'metadata', jsonb_set(model.metadata, '{rawModel,capabilities}', model.capabilities, true),
+    'status', 'active'
+  ),
+  'active'
+FROM keling_30_turbo model
+ON CONFLICT (canonical_model_key) DO UPDATE
+SET provider_id = EXCLUDED.provider_id,
+    provider_key = EXCLUDED.provider_key,
+    provider_model_name = CASE WHEN base_model_catalog.customized_at IS NULL THEN EXCLUDED.provider_model_name ELSE base_model_catalog.provider_model_name END,
+    model_type = CASE WHEN base_model_catalog.customized_at IS NULL THEN EXCLUDED.model_type ELSE base_model_catalog.model_type END,
+    display_name = CASE WHEN base_model_catalog.customized_at IS NULL THEN EXCLUDED.display_name ELSE base_model_catalog.display_name END,
+    capabilities = CASE WHEN base_model_catalog.customized_at IS NULL THEN EXCLUDED.capabilities ELSE base_model_catalog.capabilities END,
+    base_billing_config = CASE WHEN base_model_catalog.customized_at IS NULL THEN EXCLUDED.base_billing_config ELSE base_model_catalog.base_billing_config END,
+    default_rate_limit_policy = CASE WHEN base_model_catalog.customized_at IS NULL THEN EXCLUDED.default_rate_limit_policy ELSE base_model_catalog.default_rate_limit_policy END,
+    metadata = CASE WHEN base_model_catalog.customized_at IS NULL THEN EXCLUDED.metadata ELSE base_model_catalog.metadata END,
+    catalog_type = CASE WHEN base_model_catalog.customized_at IS NULL THEN 'system' ELSE base_model_catalog.catalog_type END,
+    default_snapshot = CASE WHEN base_model_catalog.customized_at IS NULL THEN EXCLUDED.default_snapshot ELSE base_model_catalog.default_snapshot END,
+    status = CASE WHEN base_model_catalog.customized_at IS NULL THEN 'active' ELSE base_model_catalog.status END,
+    updated_at = now();
+
+INSERT INTO platform_models (
+  platform_id,
+  base_model_id,
+  model_name,
+  provider_model_name,
+  model_alias,
+  model_type,
+  display_name,
+  capabilities,
+  pricing_mode,
+  billing_config,
+  retry_policy,
+  rate_limit_policy,
+  enabled
+)
+SELECT
+  platform.id,
+  base_model.id,
+  base_model.provider_model_name,
+  base_model.provider_model_name,
+  base_model.display_name,
+  base_model.model_type,
+  base_model.display_name,
+  base_model.capabilities,
+  'inherit_discount',
+  base_model.base_billing_config,
+  '{"enabled":true,"maxAttempts":1}'::jsonb,
+  base_model.default_rate_limit_policy,
+  true
+FROM integration_platforms platform
+JOIN base_model_catalog base_model
+  ON base_model.canonical_model_key = 'keling:kling-3.0-turbo'
+WHERE platform.provider = 'keling'
+  AND platform.deleted_at IS NULL
+ON CONFLICT (platform_id, model_name) DO UPDATE
+SET base_model_id = EXCLUDED.base_model_id,
+    provider_model_name = EXCLUDED.provider_model_name,
+    model_alias = EXCLUDED.model_alias,
+    display_name = EXCLUDED.display_name,
+    model_type = EXCLUDED.model_type,
+    capabilities = EXCLUDED.capabilities,
+    pricing_mode = EXCLUDED.pricing_mode,
+    billing_config = EXCLUDED.billing_config,
+    retry_policy = EXCLUDED.retry_policy,
+    rate_limit_policy = EXCLUDED.rate_limit_policy,
+    enabled = EXCLUDED.enabled,
+    updated_at = now();