diff --git a/backend/test_sync_upstream_models.py b/backend/test_sync_upstream_models.py index 40206c9..c410b5b 100644 --- a/backend/test_sync_upstream_models.py +++ b/backend/test_sync_upstream_models.py @@ -82,8 +82,8 @@ def test_sync_upstream_models_original_json_endpoint(db_session, monkeypatch): def list_accounts(self): return [ { - "id": "1", - "name": "Account 1", + "id": "1", + "name": "Account 1", "credentials": { "base_url": "old", "access_token": "secret_access_token", # Sensitive credential @@ -210,8 +210,8 @@ def test_sync_upstream_models_streaming_endpoint(db_session, monkeypatch): def list_accounts(self): return [ { - "id": "1", - "name": "Account 1", + "id": "1", + "name": "Account 1", "credentials": { "base_url": "old", "access_token": "secret_access_token", # Sensitive credential @@ -281,6 +281,7 @@ def test_sync_upstream_models_streaming_endpoint(db_session, monkeypatch): assert "access_token" not in payload1 assert "api_key" not in payload1 assert payload1["openai_capabilities"] == "some_capabilities" + assert payload1["compact_model_mapping"] == "some_mapping" # Second update to write back model_mapping payload2 = acc1_updates[1][1]["credentials"] @@ -288,6 +289,7 @@ def test_sync_upstream_models_streaming_endpoint(db_session, monkeypatch): assert "access_token" not in payload2 assert "api_key" not in payload2 assert payload2["openai_capabilities"] == "some_capabilities" + assert payload2["compact_model_mapping"] == "some_mapping" assert payload2["model_mapping"] == {"gpt-3.5": "gpt-3.5", "gpt-4": "gpt-4"} finally: