mirror of
https://github.com/coder/coder.git
synced 2026-09-22 21:22:17 +08:00
Cian Johnston
a4c867f11b
Fixes CODAGT-548
Adds two idempotent startup backfills run after `newAPI():
- `BackfillBedrockProviderType`: promotes `ai_providers` rows from
`type=anthropic` with Bedrock settings to `type=bedrock`.
- `BackfillChatModelConfigProviderStrings`: fixes stale
`chat_model_configs.provider = "anthropic"` strings on rows whose linked
provider was just promoted.
- `UpdateAIProvider` query now also writes the `type` column, so the
fix persists on any subsequent PATCH.
> 🤖 Generated by Claude with oversight from a human.
106 lines
2.3 KiB
SQL
106 lines
2.3 KiB
SQL
-- name: GetAIProviderByID :one
|
|
SELECT
|
|
*
|
|
FROM
|
|
ai_providers
|
|
WHERE
|
|
id = @id::uuid AND deleted = FALSE;
|
|
|
|
-- name: GetAIProviderByIDForReferenceLock :one
|
|
SELECT
|
|
*
|
|
FROM
|
|
ai_providers
|
|
WHERE
|
|
id = @id::uuid AND deleted = FALSE
|
|
-- Lock the provider row until the model-config write completes. The
|
|
-- transaction alone does not stop a concurrent soft-delete or disable
|
|
-- between validation and writing the model config reference.
|
|
FOR SHARE;
|
|
|
|
-- name: GetAIProviderByName :one
|
|
SELECT
|
|
*
|
|
FROM
|
|
ai_providers
|
|
WHERE
|
|
name = @name::text AND deleted = FALSE;
|
|
|
|
-- name: GetAIProviders :many
|
|
-- Returns AI provider rows. Soft-deleted and disabled rows are excluded
|
|
-- unless include_deleted or include_disabled is set.
|
|
SELECT
|
|
*
|
|
FROM
|
|
ai_providers
|
|
WHERE
|
|
(@include_deleted::boolean OR NOT deleted)
|
|
AND (@include_disabled::boolean OR enabled)
|
|
ORDER BY
|
|
name ASC;
|
|
|
|
-- name: InsertAIProvider :one
|
|
INSERT INTO ai_providers (
|
|
id,
|
|
type,
|
|
name,
|
|
display_name,
|
|
enabled,
|
|
base_url,
|
|
settings,
|
|
settings_key_id
|
|
) VALUES (
|
|
@id::uuid,
|
|
@type::ai_provider_type,
|
|
@name::text,
|
|
sqlc.narg('display_name')::text,
|
|
@enabled::boolean,
|
|
@base_url::text,
|
|
sqlc.narg('settings')::text,
|
|
sqlc.narg('settings_key_id')::text
|
|
)
|
|
RETURNING
|
|
*;
|
|
|
|
-- name: UpdateAIProvider :one
|
|
UPDATE
|
|
ai_providers
|
|
SET
|
|
type = @type::ai_provider_type,
|
|
display_name = sqlc.narg('display_name')::text,
|
|
enabled = @enabled::boolean,
|
|
base_url = @base_url::text,
|
|
settings = sqlc.narg('settings')::text,
|
|
settings_key_id = sqlc.narg('settings_key_id')::text,
|
|
updated_at = NOW()
|
|
WHERE
|
|
id = @id::uuid AND deleted = FALSE
|
|
RETURNING
|
|
*;
|
|
|
|
-- name: DeleteAIProviderByID :exec
|
|
UPDATE
|
|
ai_providers
|
|
SET
|
|
deleted = TRUE,
|
|
enabled = FALSE,
|
|
updated_at = NOW()
|
|
WHERE
|
|
id = @id::uuid AND deleted = FALSE;
|
|
|
|
-- name: UpdateEncryptedAIProviderSettings :one
|
|
-- Updates only the encrypted columns (settings, settings_key_id) and
|
|
-- the updated_at timestamp on a row, regardless of its deleted flag.
|
|
-- Used by the dbcrypt key rotation utility to re-encrypt or decrypt
|
|
-- rows in place.
|
|
UPDATE
|
|
ai_providers
|
|
SET
|
|
settings = sqlc.narg('settings')::text,
|
|
settings_key_id = sqlc.narg('settings_key_id')::text,
|
|
updated_at = NOW()
|
|
WHERE
|
|
id = @id::uuid
|
|
RETURNING
|
|
*;
|