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>
70 lines
2.2 KiB
Python
70 lines
2.2 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
|
|
|
|
"""Unit tests for provider model persistence (unslothai/unsloth#7281)."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from pathlib import Path
|
|
|
|
import pytest
|
|
|
|
import storage.providers_db as providers_db
|
|
|
|
|
|
@pytest.fixture()
|
|
def isolated_providers_db(tmp_path: Path, monkeypatch: pytest.MonkeyPatch):
|
|
db_path = tmp_path / "studio.db"
|
|
monkeypatch.setattr(providers_db, "studio_db_path", lambda: db_path)
|
|
monkeypatch.setattr(providers_db, "ensure_dir", lambda _path: None)
|
|
providers_db._schema_ready = False
|
|
yield db_path
|
|
providers_db._schema_ready = False
|
|
|
|
|
|
def test_create_and_list_provider_models(isolated_providers_db: Path):
|
|
providers_db.create_provider(
|
|
id = "ollama1",
|
|
provider_type = "ollama",
|
|
display_name = "Home Ollama",
|
|
base_url = "http://127.0.0.1:11434",
|
|
models = ["llama3.2", "qwen2.5"],
|
|
available_models = ["llama3.2", "qwen2.5", "mistral"],
|
|
)
|
|
|
|
row = providers_db.get_provider("ollama1")
|
|
assert row is not None
|
|
assert row["models"] == ["llama3.2", "qwen2.5"]
|
|
assert row["available_models"] == ["llama3.2", "qwen2.5", "mistral"]
|
|
|
|
listed = providers_db.list_providers()
|
|
assert len(listed) == 1
|
|
assert listed[0]["models"] == ["llama3.2", "qwen2.5"]
|
|
|
|
|
|
def test_update_provider_models(isolated_providers_db: Path):
|
|
providers_db.create_provider(
|
|
id = "vllm1",
|
|
provider_type = "vllm",
|
|
display_name = "Remote vLLM",
|
|
base_url = "http://studio-host:8000/v1",
|
|
models = ["meta-llama/Llama-3.2-1B-Instruct"],
|
|
available_models = ["meta-llama/Llama-3.2-1B-Instruct"],
|
|
)
|
|
|
|
assert providers_db.update_provider(
|
|
id = "vllm1",
|
|
models = ["meta-llama/Llama-3.2-3B-Instruct"],
|
|
available_models = [
|
|
"meta-llama/Llama-3.2-1B-Instruct",
|
|
"meta-llama/Llama-3.2-3B-Instruct",
|
|
],
|
|
)
|
|
|
|
row = providers_db.get_provider("vllm1")
|
|
assert row is not None
|
|
assert row["models"] == ["meta-llama/Llama-3.2-3B-Instruct"]
|
|
assert row["available_models"] == [
|
|
"meta-llama/Llama-3.2-1B-Instruct",
|
|
"meta-llama/Llama-3.2-3B-Instruct",
|
|
]
|