mirror of
https://github.com/memohai/Memoh.git
synced 2026-04-25 07:00:48 +09:00
Acbox Liu
b88ca96064
* refactor: move client_type to provider, replace model fields with config JSONB - Move `client_type` from `models` to `llm_providers` table - Add `icon` field to `llm_providers` - Replace `dimensions`, `input_modalities`, `supports_reasoning` on `models` with a single `config` JSONB column containing `dimensions`, `compatibilities` (vision, tool-call, image-output, reasoning), and `context_window` - Auto-imported models default to vision + tool-call + reasoning - Update all backend consumers (agent, flow resolver, handlers, memory) - Regenerate sqlc, swagger, and TypeScript SDK - Update frontend forms, display, and i18n for new schema * ui: show provider icon avatar in sidebar and detail header, remove icon input * feat: add built-in provider registry with YAML definitions and enable toggle - Add `enable` column to llm_providers (default true, backward-compatible) - Create internal/registry package to load YAML provider/model definitions on startup and upsert into database (new providers disabled by default) - Add conf/providers/ with OpenAI, Anthropic, Google YAML definitions - Add RegistryConfig to TOML config (providers_dir, default conf/providers) - Model listing APIs and conversation flow now filter by enabled providers - Frontend: enable switch in provider form, green status dot in sidebar, enabled providers sorted to top * fix: make 0041 migration idempotent for fresh databases Guard data migration steps with column-existence checks so the migration succeeds on databases created from the updated init schema.
63 lines
2.7 KiB
SQL
63 lines
2.7 KiB
SQL
-- 0041_provider_model_refactor
|
||
-- Move client_type to llm_providers, add icon, replace model columns with config JSONB.
|
||
|
||
-- 1. Add client_type and icon to llm_providers
|
||
ALTER TABLE llm_providers
|
||
ADD COLUMN IF NOT EXISTS client_type TEXT NOT NULL DEFAULT 'openai-completions',
|
||
ADD COLUMN IF NOT EXISTS icon TEXT;
|
||
|
||
-- 2–6. Backfill and migrate only when old columns exist (idempotent for fresh DBs).
|
||
DO $$ BEGIN
|
||
-- Back-fill provider client_type from models.client_type (old column)
|
||
IF EXISTS (SELECT 1 FROM information_schema.columns WHERE table_name = 'models' AND column_name = 'client_type') THEN
|
||
UPDATE llm_providers p
|
||
SET client_type = sub.client_type
|
||
FROM (
|
||
SELECT DISTINCT ON (llm_provider_id) llm_provider_id, client_type
|
||
FROM models
|
||
WHERE client_type IS NOT NULL AND client_type != ''
|
||
ORDER BY llm_provider_id, created_at ASC
|
||
) sub
|
||
WHERE p.id = sub.llm_provider_id;
|
||
END IF;
|
||
|
||
-- Add CHECK constraint (skip if already present)
|
||
IF NOT EXISTS (SELECT 1 FROM information_schema.table_constraints WHERE constraint_name = 'llm_providers_client_type_check') THEN
|
||
ALTER TABLE llm_providers
|
||
ADD CONSTRAINT llm_providers_client_type_check
|
||
CHECK (client_type IN ('openai-responses', 'openai-completions', 'anthropic-messages', 'google-generative-ai'));
|
||
END IF;
|
||
|
||
-- Add config JSONB to models
|
||
IF NOT EXISTS (SELECT 1 FROM information_schema.columns WHERE table_name = 'models' AND column_name = 'config') THEN
|
||
ALTER TABLE models ADD COLUMN config JSONB NOT NULL DEFAULT '{}'::jsonb;
|
||
END IF;
|
||
|
||
-- Migrate existing columns into config (only when old columns exist)
|
||
IF EXISTS (SELECT 1 FROM information_schema.columns WHERE table_name = 'models' AND column_name = 'dimensions') THEN
|
||
UPDATE models SET config = jsonb_strip_nulls(jsonb_build_object(
|
||
'dimensions', dimensions,
|
||
'compatibilities', (
|
||
SELECT coalesce(jsonb_agg(c), '[]'::jsonb) FROM (
|
||
SELECT 'tool-call' AS c
|
||
UNION ALL
|
||
SELECT 'vision' WHERE 'image' = ANY(input_modalities)
|
||
UNION ALL
|
||
SELECT 'reasoning' WHERE supports_reasoning = true
|
||
) AS caps
|
||
),
|
||
'context_window', NULL
|
||
));
|
||
END IF;
|
||
END $$;
|
||
|
||
-- Drop old columns and constraints (IF EXISTS makes these safe on fresh DBs)
|
||
ALTER TABLE models DROP CONSTRAINT IF EXISTS models_client_type_check;
|
||
ALTER TABLE models DROP CONSTRAINT IF EXISTS models_chat_client_type_check;
|
||
ALTER TABLE models DROP CONSTRAINT IF EXISTS models_dimensions_check;
|
||
|
||
ALTER TABLE models DROP COLUMN IF EXISTS client_type;
|
||
ALTER TABLE models DROP COLUMN IF EXISTS dimensions;
|
||
ALTER TABLE models DROP COLUMN IF EXISTS input_modalities;
|
||
ALTER TABLE models DROP COLUMN IF EXISTS supports_reasoning;
|