proxyml 0.1.3__tar.gz → 0.1.4__tar.gz

{proxyml-0.1.3 → proxyml-0.1.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: proxyml
-Version: 0.1.3
+Version: 0.1.4
 Summary: Python SDK for calling the ProxyML API
 Author-email: ProxyML <contact@proxyml.ai>
 License:                                  Apache License

{proxyml-0.1.3 → proxyml-0.1.4}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "proxyml"
-version = "0.1.3"
+version = "0.1.4"
 description = "Python SDK for calling the ProxyML API"
 readme = "README.md"
 license = {file = "LICENSE"}

{proxyml-0.1.3 → proxyml-0.1.4}/src/proxyml/__init__.py

@@ -12,6 +12,8 @@ from proxyml.client import (
     diff_models,
     list_models,
     delete_model,
+    get_usage,
+    rotate_key,
 )
 from proxyml.schema import (
     get_schema,
@@ -34,6 +36,8 @@ __all__ = [
     "diff_models",
     "list_models",
     "delete_model",
+    "get_usage",
+    "rotate_key",
     "get_schema",
     "gen_continuous_schema",
     "gen_categorical_schema",

{proxyml-0.1.3 → proxyml-0.1.4}/src/proxyml/client.py

@@ -316,6 +316,32 @@ def get_feature_importances(version: str | None = None):
     return None
 
 
+def get_usage() -> dict | None:
+    """Return current tier, usage counts, and quota for the authenticated user."""
+    r = get(endpoint='/account/usage', params={})
+    if r.status_code == 200:
+        return r.json()
+    logger.error(
+        "Get usage failed with status %s: %s",
+        r.status_code,
+        r.text,
+    )
+    return None
+
+
+def rotate_key() -> str | None:
+    """Rotate the API key, revoking all old keys. Returns the new key string or None on failure."""
+    r = post(endpoint='/account/keys/rotate', payload={})
+    if r.status_code == 201:
+        return r.json()['api_key']
+    logger.error(
+        "Key rotation failed with status %s: %s",
+        r.status_code,
+        r.text,
+    )
+    return None
+
+
 def list_models() -> list[dict] | None:
     """Return metadata for all trained surrogate models, newest first."""
     r = get(endpoint='/surrogate/models', params={})

{proxyml-0.1.3 → proxyml-0.1.4}/src/proxyml.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: proxyml
-Version: 0.1.3
+Version: 0.1.4
 Summary: Python SDK for calling the ProxyML API
 Author-email: ProxyML <contact@proxyml.ai>
 License:                                  Apache License

{proxyml-0.1.3 → proxyml-0.1.4}/tests/test_client.py

@@ -11,11 +11,13 @@ from proxyml.client import (
     diff_models,
     find_counterfactuals,
     get_model_summary,
+    get_usage,
     interpret_counterfactual,
     list_models,
     predict,
     predict_batch,
     put_schema,
+    rotate_key,
     synthesize_data,
     train_surrogate,
 )
@@ -420,3 +422,50 @@ def test_diff_models_success(mock_get):
 def test_diff_models_failure_returns_none(mock_get):
     mock_get.return_value = _mock_response(422, {"detail": "different tasks"})
     assert diff_models(version_a="aaa-111", version_b="bbb-222") is None
+
+
+# ---------------------------------------------------------------------------
+# get_usage
+# ---------------------------------------------------------------------------
+
+_USAGE_RESPONSE = {
+    "tier": "hobbyist",
+    "period": "2026-04",
+    "calls_this_period": 42,
+    "calls_limit": 1000,
+    "calls_remaining": 958,
+    "surrogates_trained": 2,
+    "surrogate_limit": 3,
+}
+
+
+@patch("proxyml.client.get")
+def test_get_usage_success(mock_get):
+    mock_get.return_value = _mock_response(200, _USAGE_RESPONSE)
+    result = get_usage()
+    assert result == _USAGE_RESPONSE
+    mock_get.assert_called_once_with(endpoint="/account/usage", params={})
+
+
+@patch("proxyml.client.get")
+def test_get_usage_failure_returns_none(mock_get):
+    mock_get.return_value = _mock_response(401, {"detail": "unauthorized"})
+    assert get_usage() is None
+
+
+# ---------------------------------------------------------------------------
+# rotate_key
+# ---------------------------------------------------------------------------
+
+@patch("proxyml.client.post")
+def test_rotate_key_success(mock_post):
+    mock_post.return_value = _mock_response(201, {"api_key": "proxyml_new_secret_key"})
+    result = rotate_key()
+    assert result == "proxyml_new_secret_key"
+    mock_post.assert_called_once_with(endpoint="/account/keys/rotate", payload={})
+
+
+@patch("proxyml.client.post")
+def test_rotate_key_failure_returns_none(mock_post):
+    mock_post.return_value = _mock_response(403, {"detail": "Key rotation is not available for test accounts"})
+    assert rotate_key() is None