Report measured usage costs and link completed runs to the annotation viewer

studio.py

@@ -328,7 +328,12 @@ def launch(plan, output, budget, acknowledged, token, space=None, api_factory=Hf
     mapping = api.model_info(
         MODEL.split(":")[0], expand=["inferenceProviderMapping"]
     ).inference_provider_mapping
-    if not mapping or mapping.get("featherless-ai") is None or mapping["featherless-ai"].status != "live":
+    hosted = (
+        mapping.get("featherless-ai")
+        if isinstance(mapping, dict)
+        else next((item for item in mapping or [] if item.provider == "featherless-ai"), None)
+    )
+    if hosted is None or hosted.status != "live":
         raise ValueError("Qwen3.8 is temporarily unavailable from this provider.")
     space = space or os.environ.get("SPACE_ID")
     if not space:

test_studio.py

@@ -193,8 +193,13 @@ def test_budget_unknown_failure_is_never_refunded():
         budget.reserve(4096)
 
 
-def test_launch_identity_secret_and_immutable_manifest():
+@pytest.mark.parametrize("mapping_format", ["dict", "list"])
+def test_launch_identity_secret_and_immutable_manifest(mapping_format):
     api = api_mock()
+    if mapping_format == "list":
+        api.model_info.return_value.inference_provider_mapping = [
+            SimpleNamespace(provider="featherless-ai", status="live")
+        ]
     configured = plan()
     configured["prompts"] = {"plan_subtasks": "custom instructions preserved verbatim"}
     result = studio.launch(