From c0306c7dc242a0e81ebfea8eb0cc9eeadbc10871 Mon Sep 17 00:00:00 2001
From: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com>
Date: Mon, 13 Jul 2026 23:20:55 +0000
Subject: [PATCH 1/2] feat(types): expose USD cost/tier fields on CreditUsage

Co-Authored-By: jeremy <jeremyparkphd@gmail.com>
---
 tests/test_predictions.py | 54 +++++++++++++++++++++++++++++++++++++++
 vlmrun/client/types.py    |  5 ++++
 2 files changed, 59 insertions(+)

diff --git a/tests/test_predictions.py b/tests/test_predictions.py
index 6e54a42..44ff193 100644
--- a/tests/test_predictions.py
+++ b/tests/test_predictions.py
@@ -600,3 +600,57 @@ def test_generation_config_service_tier_invalid():
 
     with pytest.raises(ValidationError):
         GenerationConfig(service_tier="ultra")
+
+
+def test_credit_usage_dollar_fields_parsed_from_api_payload():
+    """CreditUsage exposes the dollar/tier fields returned by the API."""
+    usage = CreditUsage.model_validate(
+        {
+            "elements_processed": 1,
+            "element_type": "page",
+            "credits_used": 50,
+            "duration_seconds": 2,
+            "service_tier": "flex",
+            "mode_multiplier": 0.5,
+            "standard_cost_dollars": 1.0,
+            "cost_dollars": 0.5,
+            "savings_dollars": 0.5,
+        }
+    )
+    assert usage.service_tier == "flex"
+    assert usage.mode_multiplier == 0.5
+    assert usage.standard_cost_dollars == 1.0
+    assert usage.cost_dollars == 0.5
+    assert usage.savings_dollars == 0.5
+
+
+def test_credit_usage_dollar_fields_default_none():
+    """Dollar/tier fields default to None when omitted by the API."""
+    usage = CreditUsage(credits_used=100)
+    assert usage.service_tier is None
+    assert usage.mode_multiplier is None
+    assert usage.standard_cost_dollars is None
+    assert usage.cost_dollars is None
+    assert usage.savings_dollars is None
+
+
+def test_prediction_response_retains_usage_dollar_fields():
+    """PredictionResponse round-trips the dollar/tier usage fields."""
+    response = PredictionResponse(
+        id="pred1",
+        status="completed",
+        created_at="2024-01-01T00:00:00+00:00",
+        completed_at="2024-01-01T00:00:01+00:00",
+        response={"invoice_number": "INV-001", "total_amount": 100.0},
+        usage=CreditUsage(
+            credits_used=50,
+            service_tier="flex",
+            mode_multiplier=0.5,
+            standard_cost_dollars=1.0,
+            cost_dollars=0.5,
+            savings_dollars=0.5,
+        ),
+    )
+    assert response.usage.cost_dollars == 0.5
+    assert response.usage.savings_dollars == 0.5
+    assert response.model_dump()["usage"]["cost_dollars"] == 0.5
diff --git a/vlmrun/client/types.py b/vlmrun/client/types.py
index d6adec7..c651a6f 100644
--- a/vlmrun/client/types.py
+++ b/vlmrun/client/types.py
@@ -76,6 +76,11 @@ class CreditUsage(BaseModel):
     steps: Optional[int] = None
     message: Optional[str] = None
     duration_seconds: int = 0
+    service_tier: Optional[str] = None
+    mode_multiplier: Optional[float] = None
+    standard_cost_dollars: Optional[float] = None
+    cost_dollars: Optional[float] = None
+    savings_dollars: Optional[float] = None
 
 
 class PredictionResponse(BaseModel):

From e608c09979e7a45b0f9df930b6ac6ea80457ab21 Mon Sep 17 00:00:00 2001
From: dineshreddy <dinesh@vlm.run>
Date: Mon, 13 Jul 2026 17:31:39 -0700
Subject: [PATCH 2/2] Update vlmrun/client/types.py

Co-authored-by: devin-ai-integration[bot] <158243242+devin-ai-integration[bot]@users.noreply.github.com>
---
 vlmrun/client/types.py | 10 +++++-----
 1 file changed, 5 insertions(+), 5 deletions(-)

diff --git a/vlmrun/client/types.py b/vlmrun/client/types.py
index c651a6f..cec8ade 100644
--- a/vlmrun/client/types.py
+++ b/vlmrun/client/types.py
@@ -76,11 +76,11 @@ class CreditUsage(BaseModel):
     steps: Optional[int] = None
     message: Optional[str] = None
     duration_seconds: int = 0
-    service_tier: Optional[str] = None
-    mode_multiplier: Optional[float] = None
-    standard_cost_dollars: Optional[float] = None
-    cost_dollars: Optional[float] = None
-    savings_dollars: Optional[float] = None
+    service_tier: str | None = None
+    mode_multiplier: float | None = None
+    standard_cost_dollars: float | None = None
+    cost_dollars: float | None = None
+    savings_dollars: float | None = None
 
 
 class PredictionResponse(BaseModel):