feat: show model pricing in configuration UI (#528)

* feat: show model pricing in configuration UI

* fix: display effective pricing rounding policy
This commit is contained in:
Abhishek 2026-07-13 14:20:24 +05:30 committed by GitHub
parent e7494e9c21
commit c76076fb93
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
13 changed files with 445 additions and 9 deletions

View file

@ -0,0 +1,66 @@
from types import SimpleNamespace
from unittest.mock import AsyncMock
import pytest
from api.routes import organization as organization_routes
@pytest.mark.asyncio
async def test_model_configuration_pricing_returns_empty_in_oss(monkeypatch):
get_pricing = AsyncMock()
monkeypatch.setattr(organization_routes, "DEPLOYMENT_MODE", "oss")
monkeypatch.setattr(
organization_routes.mps_service_key_client,
"get_billing_pricing",
get_pricing,
)
response = await organization_routes.get_model_configuration_pricing(
SimpleNamespace(selected_organization_id=42),
)
assert response.platform_usage is None
assert response.dograh_model is None
get_pricing.assert_not_awaited()
@pytest.mark.asyncio
async def test_model_configuration_pricing_uses_selected_organization(monkeypatch):
get_pricing = AsyncMock(
return_value={
"organization_id": 42,
"platform_usage": {
"metric_code": "platform_usage",
"display_name": "Platform usage",
"unit": "minute",
"price_per_minute": 0.01,
"currency": "USD",
"rounding_policy": "ceil_minute",
},
"dograh_model": {
"metric_code": "voice_minutes",
"display_name": "Dograh model usage",
"unit": "minute",
"price_per_minute": 0.07,
"currency": "USD",
"rounding_policy": "ceil_minute",
},
}
)
monkeypatch.setattr(organization_routes, "DEPLOYMENT_MODE", "saas")
monkeypatch.setattr(
organization_routes.mps_service_key_client,
"get_billing_pricing",
get_pricing,
)
response = await organization_routes.get_model_configuration_pricing(
SimpleNamespace(selected_organization_id=42),
)
get_pricing.assert_awaited_once_with(42)
assert response.platform_usage is not None
assert response.platform_usage.price_per_minute == 0.01
assert response.dograh_model is not None
assert response.dograh_model.price_per_minute == 0.07

View file

@ -260,6 +260,59 @@ async def test_ensure_billing_account_v2_uses_balance_endpoint(monkeypatch):
]
@pytest.mark.asyncio
async def test_get_billing_pricing_uses_hosted_organization_auth(monkeypatch):
calls = []
class FakeAsyncClient:
def __init__(self, timeout):
self.timeout = timeout
async def __aenter__(self):
return self
async def __aexit__(self, exc_type, exc, tb):
return None
async def get(self, url, headers):
calls.append(("GET", url, headers))
return _Response(
200,
{
"organization_id": 42,
"platform_usage": {"price_per_minute": 0.01},
"dograh_model": {"price_per_minute": 0.07},
},
)
monkeypatch.setattr(
"api.services.mps_service_key_client.httpx.AsyncClient", FakeAsyncClient
)
monkeypatch.setattr("api.services.mps_service_key_client.DEPLOYMENT_MODE", "saas")
monkeypatch.setattr(
"api.services.mps_service_key_client.DOGRAH_MPS_SECRET_KEY", "mps-secret"
)
client = MPSServiceKeyClient()
assert await client.get_billing_pricing(42) == {
"organization_id": 42,
"platform_usage": {"price_per_minute": 0.01},
"dograh_model": {"price_per_minute": 0.07},
}
assert calls == [
(
"GET",
f"{client.base_url}/api/v1/billing/accounts/42/pricing",
{
"Content-Type": "application/json",
"X-Secret-Key": "mps-secret",
"X-Organization-Id": "42",
},
)
]
@pytest.mark.asyncio
async def test_get_credit_ledger_sends_page_and_limit(monkeypatch):
calls = []