python-open-router 0.1.0__tar.gz → 0.2.0__tar.gz

{python_open_router-0.1.0 → python_open_router-0.2.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: python_open_router
-Version: 0.1.0
+Version: 0.2.0
 Summary: Asynchronous Python client for Open Router.
 License: MIT
 Keywords: OpenRouter,api,async,client

{python_open_router-0.1.0 → python_open_router-0.2.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "python_open_router"
-version = "0.1.0"
+version = "0.2.0"
 description = "Asynchronous Python client for Open Router."
 authors = ["Joost Lekkerkerker <joostlek@outlook.com>"]
 maintainers = ["Joost Lekkerkerker <joostlek@outlook.com>"]
@@ -39,14 +39,14 @@ coverage = {version = "7.6.12", extras = ["toml"]}
 mypy = "1.15.0"
 pre-commit = "4.1.0"
 pre-commit-hooks = "5.0.0"
-pylint = "3.3.4"
-pytest = "8.3.4"
+pylint = "3.3.6"
+pytest = "8.3.5"
 pytest-asyncio = "0.25.3"
-pytest-cov = "6.0.0"
+pytest-cov = "6.1.1"
 ruff = "0.9.6"
-safety = "3.3.0"
+safety = "3.3.1"
 yamllint = "1.35.1"
-syrupy = "4.8.1"
+syrupy = "4.9.1"
 aioresponses = "0.7.8"
 
 [tool.poetry.urls]

{python_open_router-0.1.0 → python_open_router-0.2.0}/src/python_open_router/__init__.py

@@ -5,9 +5,23 @@ from python_open_router.exceptions import (
     OpenRouterConnectionError,
     OpenRouterError,
 )
+from python_open_router.models import (
+    CreatedKey,
+    CreateKeyDataWrapper,
+    Key,
+    KeyData,
+    KeyDataWrapper,
+    KeysDataWrapper,
+)
 from python_open_router.open_router import OpenRouterClient
 
 __all__ = [
+    "CreateKeyDataWrapper",
+    "CreatedKey",
+    "Key",
+    "KeyData",
+    "KeyDataWrapper",
+    "KeysDataWrapper",
     "OpenRouterAuthenticationError",
     "OpenRouterClient",
     "OpenRouterConnectionError",

python_open_router-0.2.0/src/python_open_router/models.py

@@ -0,0 +1,58 @@
+"""Models for OpenRouter."""
+
+from __future__ import annotations
+
+from dataclasses import dataclass
+
+from mashumaro.mixins.orjson import DataClassORJSONMixin
+
+
+@dataclass
+class KeyDataWrapper(DataClassORJSONMixin):
+    """Wrapper for OpenRouter key data."""
+
+    data: KeyData
+
+
+@dataclass
+class KeyData(DataClassORJSONMixin):
+    """The OpenRouter key data."""
+
+    label: str
+    usage: int
+    is_provisioning_key: bool
+    limit_remaining: int | None
+    is_free_tier: bool
+
+
+@dataclass
+class KeysDataWrapper(DataClassORJSONMixin):
+    """Wrapper for OpenRouter key data."""
+
+    data: list[Key]
+
+
+@dataclass
+class CreateKeyDataWrapper(DataClassORJSONMixin):
+    """Wrapper for OpenRouter key data."""
+
+    data: CreatedKey
+
+
+@dataclass
+class Key(DataClassORJSONMixin):
+    """The OpenRouter key data."""
+
+    hash: str
+    name: str
+    label: str
+    disabled: bool
+    limit: float
+    usage: float = 0.0
+
+
+@dataclass(kw_only=True)
+class CreatedKey(Key):
+    """Created key."""
+
+    key: str

{python_open_router-0.1.0 → python_open_router-0.2.0}/src/python_open_router/open_router.py

@@ -9,11 +9,17 @@ import socket
 from typing import TYPE_CHECKING, Any
 
 from aiohttp import ClientError, ClientResponseError, ClientSession
-from aiohttp.hdrs import METH_GET
+from aiohttp.hdrs import METH_GET, METH_POST
 from yarl import URL
 
 from python_open_router.exceptions import OpenRouterConnectionError
-from python_open_router.models import KeyData, KeyDataWrapper
+from python_open_router.models import (
+    CreateKeyDataWrapper,
+    Key,
+    KeyData,
+    KeyDataWrapper,
+    KeysDataWrapper,
+)
 
 if TYPE_CHECKING:
     from typing_extensions import Self
@@ -89,6 +95,20 @@ class OpenRouterClient:
         response = await self._request(METH_GET, "key")
         return KeyDataWrapper.from_json(response).data
 
+    async def get_keys(self) -> list[Key]:
+        """Get all keys."""
+        response = await self._request(METH_GET, "keys")
+        return KeysDataWrapper.from_json(response).data
+
+    async def create_key(self, name: str, limit: float | None = None) -> Key:
+        """Create a new key."""
+        data: dict[str, Any] = {"name": name}
+        if limit is not None:
+            data["limit"] = limit
+
+        response = await self._request(METH_POST, "keys", data=data)
+        return CreateKeyDataWrapper.from_json(response).data
+
     async def close(self) -> None:
         """Close open client session."""
         if self.session and self._close_session:

python_open_router-0.1.0/src/python_open_router/models.py

@@ -1,25 +0,0 @@
-"""Models for OpenRouter."""
-
-from __future__ import annotations
-
-from dataclasses import dataclass
-
-from mashumaro.mixins.orjson import DataClassORJSONMixin
-
-
-@dataclass
-class KeyDataWrapper(DataClassORJSONMixin):
-    """Wrapper for OpenRouter key data."""
-
-    data: KeyData
-
-
-@dataclass
-class KeyData(DataClassORJSONMixin):
-    """The OpenRouter key data."""
-
-    label: str
-    usage: int
-    is_provisioning_key: bool
-    limit_remaining: int | None
-    is_free_tier: bool