o µÇibă@sHdZddlmZddlZddlmZdZdZdZ dZ dd„Z d d „Z dS) zunify provider model architecture Revision ID: a1b2c3d4e5f6 Revises: add_local_models Create Date: 2026-03-28 12:00:00+08:00 é)ÚopN)Ú postgresqlÚ a1b2c3d4e5f6Úadd_local_modelsc CsčtjdtjdtjddŤddŤddŤtjdtjd tjddŤddŤddŤtjdtjd tjddŤddŤddŤtjdtjd tjt ˇd ŤddŤddŤtjdtjd t ˇddŤddŤtjdtjdtjddŤddŤddŤtjdtjdtjt ˇd ŤddŤddŤtjdtjdtjddŤddŤddŤtjdtjdt  ˇddŤddŤtjdtjdt  ˇddŤddŤtjdtjdt ˇddŤddŤtjdtjdt ˇddŤddŤtjdtjdt  ˇddŤddŤtjdtjdtjt ˇd ŤddŤddŤt dtjdtjddŤddŤˇt dtjdtjddŤddŤˇt dtjdtjddŤddŤˇt dtjdt  ˇddŤˇtj ddddgdgdddŤt  d ˇt  d!ˇt  d"ˇt  d#ˇt  d$ˇt  d%ˇt  d&ˇt  d'ˇt  d(ˇt  d)ˇt  d*ˇt  d+ˇtj dddtjddŤd,d-Ťtj dd dtjddŤd,d-Ťtj dd dtjddŤd,d-Ťtj dd dtjt ˇd Ťd,d-Ťtj dd dt ˇd,d-Ťtj dddt ˇd,d-Ťtj dddt ˇd,d-Ťtj dddt  ˇd,d-Ťt  d.ˇt  d/ˇdS)0NÚmodel_providersÚ provider_kindé2)ÚlengthT)ÚnullableÚpublic©ÚschemaÚprotocolÚ auth_typeÚ capabilities)Ú astext_typeÚis_localÚhealthcheck_pathé˙Ú extra_configÚmodelsÚremote_model_idÚcontext_lengthÚmax_output_tokensÚsupports_streamÚsupports_toolsÚpriorityÚ local_modelsÚ asset_kindÚ runtime_kindÚ source_kindÚpublished_model_idÚ)fk_local_models_published_model_id_modelsÚid)Ú source_schemaÚreferent_schemazZUPDATE public.model_providers SET provider_kind = 'public_api' WHERE provider_kind IS NULLzWUPDATE public.model_providers SET protocol = 'openai_compatible' WHERE protocol IS NULLzŤUPDATE public.model_providers SET auth_type = CASE WHEN api_key IS NULL OR api_key = '' THEN 'none' ELSE 'bearer' END WHERE auth_type IS NULLzVUPDATE public.model_providers SET capabilities = '[]'::json WHERE capabilities IS NULLzIUPDATE public.model_providers SET is_local = false WHERE is_local IS NULLzMUPDATE public.models SET remote_model_id = code WHERE remote_model_id IS NULLzMUPDATE public.models SET supports_stream = true WHERE supports_stream IS NULLzLUPDATE public.models SET supports_tools = false WHERE supports_tools IS NULLz>UPDATE public.models SET priority = 100 WHERE priority IS NULLz€UPDATE local_models SET asset_kind = CASE WHEN model_type = 'lora' THEN 'adapter' ELSE 'full_model' END WHERE asset_kind IS NULLzKUPDATE local_models SET runtime_kind = 'generic' WHERE runtime_kind IS NULLzJUPDATE local_models SET source_kind = 'imported' WHERE source_kind IS NULLF)r Ú existing_typer uÁ INSERT INTO public.model_providers ( name, description, configured, default_base_url, api_key, base_url, priority, enabled, provider_kind, protocol, auth_type, capabilities, is_local, healthcheck_path, extra_config, created_at, updated_at, is_deleted ) SELECT '本地 MLX', 'Local MLX provider', true, 'http://127.0.0.1:8010', NULL, 'http://127.0.0.1:8010', 50, true, 'local_mlx', 'openai_compatible', 'none', '["chat", "embedding", "rerank"]'::json, true, '/health', NULL, NOW(), NOW(), false WHERE NOT EXISTS ( SELECT 1 FROM public.model_providers WHERE name = '本地 MLX' ) a© INSERT INTO public.model_providers ( name, description, configured, default_base_url, api_key, base_url, priority, enabled, provider_kind, protocol, auth_type, capabilities, is_local, healthcheck_path, extra_config, created_at, updated_at, is_deleted ) SELECT 'ECS A100', 'Self-hosted ECS model provider', false, NULL, NULL, NULL, 60, true, 'self_hosted', 'openai_compatible', 'bearer', '["chat", "embedding", "rerank"]'::json, false, '/health', NULL, NOW(), NOW(), false WHERE NOT EXISTS ( SELECT 1 FROM public.model_providers WHERE name = 'ECS A100' ) ) rÚ add_columnÚsaÚColumnÚStringrÚJSONÚTextÚBooleanÚIntegerÚcreate_foreign_keyÚexecuteÚ alter_column©r2r2ú[/lsinfo/ai/hellotax_ai/base_platform/alembic/versions/20260328_1200_unify_provider_model.pyÚupgradesj$$$( $($     (   ů            ˙˙r4cCs$tjddddŤt ddˇt ddˇt ddˇt ddˇtjd d d d Ťtjd d d d Ťtjd dd d Ťtjd dd d Ťtjd dd d Ťtjd dd d Ťtjd dd d Ťtjdd d d Ťtjddd d Ťtjddd d Ťtjddd d Ťtjddd d Ťtjddd d Ťtjddd d ŤdS)Nr"rÚ foreignkey)Útype_r!r rrrrr r rrrrrrrrrrrrr)rÚdrop_constraintÚ drop_columnr2r2r2r3Ú downgradels&    r9) Ú__doc__ÚalembicrÚ sqlalchemyr(Úsqlalchemy.dialectsrÚrevisionÚ down_revisionÚ branch_labelsÚ depends_onr4r9r2r2r2r3Ús   Y