withruntime 0.3.0__tar.gz → 0.3.1__tar.gz

{withruntime-0.3.0 → withruntime-0.3.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: withruntime
-Version: 0.3.0
+Version: 0.3.1
 Summary: Runtime Cloud: one client (sync and async) for every Runtime product. Sandboxes first.
 Requires-Python: >=3.10
 Description-Content-Type: text/markdown

{withruntime-0.3.0 → withruntime-0.3.1}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "withruntime"
-version = "0.3.0"
+version = "0.3.1"
 description = "Runtime Cloud: one client (sync and async) for every Runtime product. Sandboxes first."
 readme = "README.md"
 requires-python = ">=3.10"

withruntime-0.3.1/tests/test_limits.py

@@ -0,0 +1,61 @@
+"""runtime.limits.get() against a stub API, sync and async. The route is
+tested in packages/cloud (limits-api.test.ts)."""
+import asyncio
+import json
+import threading
+import unittest
+from http.server import BaseHTTPRequestHandler, ThreadingHTTPServer
+
+from withruntime import AsyncRuntime, Runtime
+
+LIMITS = {
+    "access": "full",
+    "daily": {"limitMicros": "25000000", "usedMicros": "3100000", "remainingMicros": "21900000", "window": "24h"},
+}
+
+
+class Stub(BaseHTTPRequestHandler):
+    seen: list = []
+
+    def log_message(self, *args):
+        pass
+
+    def do_GET(self):
+        Stub.seen.append((self.command, self.path))
+        data = json.dumps(LIMITS).encode()
+        self.send_response(200)
+        self.send_header("content-type", "application/json")
+        self.send_header("content-length", str(len(data)))
+        self.end_headers()
+        self.wfile.write(data)
+
+
+class LimitsTest(unittest.TestCase):
+    @classmethod
+    def setUpClass(cls):
+        cls.server = ThreadingHTTPServer(("127.0.0.1", 0), Stub)
+        threading.Thread(target=cls.server.serve_forever, daemon=True).start()
+        cls.url = f"http://127.0.0.1:{cls.server.server_address[1]}"
+
+    @classmethod
+    def tearDownClass(cls):
+        cls.server.shutdown()
+
+    def setUp(self):
+        Stub.seen = []
+
+    def test_sync_get(self):
+        with Runtime(api_key="rk_test", base_url=self.url, max_retries=0) as runtime:
+            self.assertEqual(runtime.limits.get()["daily"]["remainingMicros"], "21900000")
+        self.assertEqual(Stub.seen, [("GET", "/v1/limits")])
+
+    def test_async_get(self):
+        async def go():
+            async with AsyncRuntime(api_key="rk_test", base_url=self.url, max_retries=0) as runtime:
+                return await runtime.limits.get()
+
+        self.assertEqual(asyncio.run(go())["access"], "full")
+
+
+if __name__ == "__main__":
+    unittest.main()

{withruntime-0.3.0 → withruntime-0.3.1}/withruntime/_async_products/__init__.py

@@ -18,6 +18,7 @@ from typing import Any
 from .desktop import AsyncDesktop
 from .images import AsyncImages
 from .interpreter import AsyncInterpreter
+from .limits import AsyncLimits
 from .network import AsyncNetwork
 from .previews import AsyncPreviews
 from .referrals import AsyncReferrals
@@ -28,6 +29,7 @@ SANDBOX: dict[str, Any] = {}
 CLIENT["images"] = AsyncImages
 CLIENT["volumes"] = AsyncVolumes
 CLIENT["referrals"] = AsyncReferrals
+CLIENT["limits"] = AsyncLimits
 SANDBOX["interpreter"] = AsyncInterpreter
 SANDBOX["previews"] = AsyncPreviews
 SANDBOX["network"] = AsyncNetwork

withruntime-0.3.1/withruntime/_async_products/limits.py

@@ -0,0 +1,20 @@
+from __future__ import annotations
+
+from typing import Any
+
+
+class AsyncLimits:
+    """``runtime.limits``. Whether this key is read-only, and the daily spending
+    limit an owner set on its agent. A key can read the limit, never change it:
+    the owner sets it at https://withruntime.com/account/keys. Past it, a
+    create, wake, extension or renewal fails with ``spending_limit_reached``
+    (HTTP 402). Money is integer microdollars in strings (1,000,000 = $1)."""
+
+    def __init__(self, t: Any) -> None:
+        self._t = t
+
+    async def get(self) -> dict[str, Any]:
+        """``access`` (``full``, ``read`` or ``selected``) and ``daily``:
+        ``limitMicros`` (None for no limit), ``usedMicros`` in the last 24
+        hours, and ``remainingMicros`` (None for no limit)."""
+        return await self._t.json("GET", "/v1/limits")

{withruntime-0.3.0 → withruntime-0.3.1}/withruntime/_sync_products/__init__.py

@@ -4,6 +4,7 @@ from typing import Any
 from .desktop import Desktop
 from .images import Images
 from .interpreter import Interpreter
+from .limits import Limits
 from .network import Network
 from .previews import Previews
 from .referrals import Referrals
@@ -14,6 +15,7 @@ SANDBOX: dict[str, Any] = {}
 CLIENT["images"] = Images
 CLIENT["volumes"] = Volumes
 CLIENT["referrals"] = Referrals
+CLIENT["limits"] = Limits
 SANDBOX["interpreter"] = Interpreter
 SANDBOX["previews"] = Previews
 SANDBOX["network"] = Network

withruntime-0.3.1/withruntime/_sync_products/limits.py

@@ -0,0 +1,21 @@
+"""GENERATED from _async_products/limits.py by scripts/generate_sync.py. Do not edit."""
+from __future__ import annotations
+
+from typing import Any
+
+
+class Limits:
+    """``runtime.limits``. Whether this key is read-only, and the daily spending
+    limit an owner set on its agent. A key can read the limit, never change it:
+    the owner sets it at https://withruntime.com/account/keys. Past it, a
+    create, wake, extension or renewal fails with ``spending_limit_reached``
+    (HTTP 402). Money is integer microdollars in strings (1,000,000 = $1)."""
+
+    def __init__(self, t: Any) -> None:
+        self._t = t
+
+    def get(self) -> dict[str, Any]:
+        """``access`` (``full``, ``read`` or ``selected``) and ``daily``:
+        ``limitMicros`` (None for no limit), ``usedMicros`` in the last 24
+        hours, and ``remainingMicros`` (None for no limit)."""
+        return self._t.json("GET", "/v1/limits")

withruntime-0.3.1/withruntime/_version.py

@@ -0,0 +1 @@
+VERSION = "0.3.1"

{withruntime-0.3.0 → withruntime-0.3.1}/withruntime.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: withruntime
-Version: 0.3.0
+Version: 0.3.1
 Summary: Runtime Cloud: one client (sync and async) for every Runtime product. Sandboxes first.
 Requires-Python: >=3.10
 Description-Content-Type: text/markdown

{withruntime-0.3.0 → withruntime-0.3.1}/withruntime.egg-info/SOURCES.txt

@@ -2,6 +2,7 @@ README.md
 pyproject.toml
 tests/test_client.py
 tests/test_images_volumes_interpreter.py
+tests/test_limits.py
 tests/test_previews_network_desktop.py
 tests/test_referrals.py
 withruntime/__init__.py
@@ -21,6 +22,7 @@ withruntime/_async_products/__init__.py
 withruntime/_async_products/desktop.py
 withruntime/_async_products/images.py
 withruntime/_async_products/interpreter.py
+withruntime/_async_products/limits.py
 withruntime/_async_products/network.py
 withruntime/_async_products/previews.py
 withruntime/_async_products/referrals.py
@@ -29,6 +31,7 @@ withruntime/_sync_products/__init__.py
 withruntime/_sync_products/desktop.py
 withruntime/_sync_products/images.py
 withruntime/_sync_products/interpreter.py
+withruntime/_sync_products/limits.py
 withruntime/_sync_products/network.py
 withruntime/_sync_products/previews.py
 withruntime/_sync_products/referrals.py

withruntime-0.3.0/withruntime/_version.py

@@ -1 +0,0 @@
-VERSION = "0.3.0"