Add price, input_price, price_unit, pricing_model, model_class fields… in FoundationModel

NEXT_CHANGELOG.md

@@ -13,3 +13,5 @@
 ### Internal Changes
 
 ### API Changes
+
+ - extra fields in the FoundationModel class

tests/test_misc.py

@@ -50,3 +50,55 @@ def test_serde_with_empty_dataclass():
     )
     assert inst.as_dict() == {"pipeline_id": "123", "run_continuously": {}}
     assert inst == catalog.OnlineTableSpec.from_dict(inst.as_dict())
+
+
+def test_foundation_model_serde():
+    from databricks.sdk.service.serving import FoundationModel
+
+    model = FoundationModel(
+        name="system.ai.databricks-claude-sonnet-4-5",
+        display_name="Claude Sonnet 4.5",
+        docs="https://docs.databricks.com/machine-learning/foundation-models/supported-models.html#claude-sonnet-4-5",
+        description="Claude Sonnet 4.5 is Anthropic's most advanced hybrid reasoning model. It offers two modes: near-instant responses and extended thinking for deeper reasoning based on the complexity of the task. Claude Sonnet 4.5 specializes in application that require a balance of practical throughput and advanced thinking such as customer-facing agents, production coding workflows, and content generation at scale. This endpoint is hosted by Databricks",
+        price="214.286",
+        input_price="85.714",
+        price_unit="DBUs per 1M tokens",
+        pricing_model="Pay-per-token",
+        model_class="claude",
+    )
+
+    expected = {
+        "name": "system.ai.databricks-claude-sonnet-4-5",
+        "display_name": "Claude Sonnet 4.5",
+        "docs": "https://docs.databricks.com/machine-learning/foundation-models/supported-models.html#claude-sonnet-4-5",
+        "description": "Claude Sonnet 4.5 is Anthropic's most advanced hybrid reasoning model. It offers two modes: near-instant responses and extended thinking for deeper reasoning based on the complexity of the task. Claude Sonnet 4.5 specializes in application that require a balance of practical throughput and advanced thinking such as customer-facing agents, production coding workflows, and content generation at scale. This endpoint is hosted by Databricks",
+        "price": "214.286",
+        "input_price": "85.714",
+        "price_unit": "DBUs per 1M tokens",
+        "pricing_model": "Pay-per-token",
+        "model_class": "claude",
+    }
+
+    assert model.as_dict() == expected
+    assert model.as_shallow_dict() == expected
+    assert model == FoundationModel.from_dict(model.as_dict())
+
+
+def test_foundation_model_serde_partial():
+    from databricks.sdk.service.serving import FoundationModel
+
+    model = FoundationModel(
+        name="system.ai.databricks-claude-sonnet-4-5",
+        price="214.286",
+        model_class="claude",
+    )
+
+    expected = {
+        "name": "system.ai.databricks-claude-sonnet-4-5",
+        "price": "214.286",
+        "model_class": "claude",
+    }
+
+    assert model.as_dict() == expected
+    assert model.as_shallow_dict() == expected
+    assert model == FoundationModel.from_dict(model.as_dict())