python-open-router 0.1.0__py3-none-any.whl → 0.2.0__py3-none-any.whl

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/models.py

@@ -23,3 +23,36 @@ class KeyData(DataClassORJSONMixin):
     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/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.dist-info → python_open_router-0.2.0.dist-info}/METADATA

@@ -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.2.0.dist-info/RECORD

@@ -0,0 +1,9 @@
+python_open_router/__init__.py,sha256=dmD0nHFaJzzKYD7Q4Ltm9p46uBhAOLpPTnWVyH_KHH8,640
+python_open_router/exceptions.py,sha256=68nskqNdXNMIK4iA1XOE1E5P5rbhAf27DDACkk-ZVwU,312
+python_open_router/models.py,sha256=qIKy3bqx-RnwlbwV4kgTPuDkbII4xPq2PE3gA8mShwo,999
+python_open_router/open_router.py,sha256=WOJtZ40l3dv8_GGWRV1caN--KAgv0g0KfNdl-KzGGLM,3909
+python_open_router/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+python_open_router-0.2.0.dist-info/LICENSE.md,sha256=Xbw7n2Ic30r4j8_MDNND4ss_sdoj9IGeIkEWc2Hsfs8,1077
+python_open_router-0.2.0.dist-info/METADATA,sha256=iYrSXHhOj1w-vHE51V3JZoIdNcjrfkedqw-wuILzYVg,6030
+python_open_router-0.2.0.dist-info/WHEEL,sha256=fGIA9gx4Qxk2KDKeNJCbOEwSrmLtjWCwzBz351GyrPQ,88
+python_open_router-0.2.0.dist-info/RECORD,,

python_open_router-0.1.0.dist-info/RECORD

@@ -1,9 +0,0 @@
-python_open_router/__init__.py,sha256=qWLg-1XHDTgXR7AGOzVjxQ2amINWgwTnMrRe77WFRXw,376
-python_open_router/exceptions.py,sha256=68nskqNdXNMIK4iA1XOE1E5P5rbhAf27DDACkk-ZVwU,312
-python_open_router/models.py,sha256=z-SMl13PAGRyQ_83zXlU7US0teNqxfc4ArTm0ain-zs,478
-python_open_router/open_router.py,sha256=mVcNVZcSEcD5csllZes3RlKa7ReWEQXJf625M-so6Mo,3292
-python_open_router/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-python_open_router-0.1.0.dist-info/LICENSE.md,sha256=Xbw7n2Ic30r4j8_MDNND4ss_sdoj9IGeIkEWc2Hsfs8,1077
-python_open_router-0.1.0.dist-info/METADATA,sha256=KFnz6IFqokjCanKGMC6GmT3hzA53fdjc2JdinQU7OH4,6030
-python_open_router-0.1.0.dist-info/WHEEL,sha256=fGIA9gx4Qxk2KDKeNJCbOEwSrmLtjWCwzBz351GyrPQ,88
-python_open_router-0.1.0.dist-info/RECORD,,