b448fb5de0* fix(studio): persist connection model selections server-side Remote Studio clients could see saved connections but not their enabled model lists because models lived only in browser localStorage. Store models and available_models in llm_providers and sync them through the providers API so alternate clients inherit the same catalog state. Fixes #7281 * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci * Hydrate external connections on chat startup (#7281) Extract provider sync logic into sync-external-providers.ts and call it from chat-page on mount so persisted model selections appear in the Connected picker without opening Settings → Connections first. * fix(studio): backfill connection models and preserve local options (#7298) Address Codex P2 on remote connection persistence: - Backfill localStorage model selections to /api/providers when backend rows still have empty models_json (legacy upgrades) - Carry promptCacheTtl and openaiContainerTtlMinutes through startup sync - Await hydratePersistedSettings before syncing on ChatPage mount Contract tests: 7 passed; npm run typecheck passed. * Tighten comments * Tighten comments --------- Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com> Co-authored-by: Daniel Han <danielhanchen@gmail.com>
61 lines
2.3 KiB
Python
61 lines
2.3 KiB
Python
# SPDX-License-Identifier: AGPL-3.0-only
|
|
# Copyright 2026-present the Unsloth AI Inc. team. All rights reserved. See /studio/LICENSE.AGPL-3.0
|
|
|
|
"""Static contracts for remote connection model persistence (#7281)."""
|
|
|
|
from pathlib import Path
|
|
|
|
|
|
REPO = Path(__file__).resolve().parents[2]
|
|
FRONTEND = REPO / "studio/frontend/src"
|
|
PROVIDERS_API = FRONTEND / "features/chat/api/providers-api.ts"
|
|
SYNC_PROVIDERS = FRONTEND / "features/chat/sync-external-providers.ts"
|
|
CHAT_PAGE = FRONTEND / "features/chat/chat-page.tsx"
|
|
PROVIDERS_DB = REPO / "studio/backend/storage/providers_db.py"
|
|
PROVIDERS_MODELS = REPO / "studio/backend/models/providers.py"
|
|
|
|
|
|
def test_providers_db_stores_model_json_columns():
|
|
source = PROVIDERS_DB.read_text(encoding = "utf-8")
|
|
assert "models_json" in source
|
|
assert "available_models_json" in source
|
|
assert "ALTER TABLE llm_providers ADD COLUMN models_json" in source
|
|
|
|
|
|
def test_provider_api_schemas_expose_models():
|
|
source = PROVIDERS_MODELS.read_text(encoding = "utf-8")
|
|
assert "models: list[str]" in source
|
|
assert "available_models: list[str]" in source
|
|
|
|
|
|
def test_frontend_sync_prefers_server_models_on_remote_clients():
|
|
source = SYNC_PROVIDERS.read_text(encoding = "utf-8")
|
|
assert "config.models" in source
|
|
assert "config.available_models" in source
|
|
assert "serverModels.length > 0" in source
|
|
|
|
|
|
def test_frontend_sync_backfills_local_models_to_backend():
|
|
source = SYNC_PROVIDERS.read_text(encoding = "utf-8")
|
|
assert "updateProviderConfig" in source
|
|
assert "needsModelBackfill" in source
|
|
assert "Promise.allSettled(backfillTasks)" in source
|
|
|
|
|
|
def test_frontend_sync_preserves_local_provider_options():
|
|
source = SYNC_PROVIDERS.read_text(encoding = "utf-8")
|
|
assert "mergeLocalProviderOptions" in source
|
|
assert "promptCacheTtl" in source
|
|
assert "openaiContainerTtlMinutes" in source
|
|
|
|
|
|
def test_chat_page_hydrates_connections_on_startup():
|
|
source = CHAT_PAGE.read_text(encoding = "utf-8")
|
|
assert "syncExternalProvidersFromBackend" in source
|
|
assert "await hydratePersistedSettings()" in source
|
|
|
|
|
|
def test_providers_api_sends_models_to_backend():
|
|
source = PROVIDERS_API.read_text(encoding = "utf-8")
|
|
assert "available_models: payload.availableModels" in source
|
|
assert "models: payload.models" in source
|