test: fix type errors and pin image_output in the model serialisation snapshot

tests/integration/test_model_serialization.py

@@ -356,6 +356,7 @@ def _expected_serialised_model(provider_id: int) -> dict:
         "pricing": {
             "completion": 2.0e-7,
             "image": 0.0,
+            "image_output": 0.0,
             "input_cache_read": 0.0,
             "input_cache_write": 0.0,
             "internal_reasoning": 0.0,
@@ -369,6 +370,7 @@ def _expected_serialised_model(provider_id: int) -> dict:
         "sats_pricing": {
             "completion": 2.0e-7 / _SATS,
             "image": 0.0,
+            "image_output": 0.0,
             "input_cache_read": 0.0,
             "input_cache_write": 0.0,
             "internal_reasoning": 0.0,

tests/unit/test_image_billing_units.py

@@ -352,10 +352,10 @@ async def _settle(
             await _drain(response)
 
     async with AsyncSession(engine, expire_on_commit=False) as session:
-        key = await session.get(ApiKey, snapshot.key_hash)
+        settled = await session.get(ApiKey, snapshot.key_hash)
         record = await session.get(ReservationRelease, snapshot.release_id)
-        assert key is not None
-        return key.balance, key.total_spent, record.status if record else None
+        assert settled is not None
+        return settled.balance, settled.total_spent, record.status if record else None
 
 
 @pytest.mark.asyncio

tests/unit/test_image_catalog.py

@@ -296,7 +296,7 @@ def test_openrouter_provider_prices_image_models_from_the_image_api() -> None:
 
 def test_openai_provider_prices_gpt_image_from_its_token_rate() -> None:
     catalog = [
-        dict(entry, id=entry["id"].removeprefix("openai/"))
+        dict(entry, id=str(entry["id"]).removeprefix("openai/"))
         for entry in OPENROUTER_CATALOG[:1]
     ]
     provider = OpenAIUpstreamProvider(api_key="k")