Async-first codegen: single source of truth with sync generation

pykalshi/__init__.py

@@ -4,7 +4,7 @@
 A clean, modular interface for the Kalshi trading API.
 """
 
-__version__ = "0.4.0"
+__version__ = "0.5.0"
 
 import logging
 

pykalshi/_async/api_keys.py

@@ -0,0 +1,64 @@
+from __future__ import annotations
+
+from typing import TYPE_CHECKING
+
+from ..models import APIKey, GeneratedAPIKey, APILimits
+
+if TYPE_CHECKING:
+    from .client import AsyncKalshiClient
+
+
+class AsyncAPIKeys:
+    """API key management and account limits."""
+
+    def __init__(self, client: AsyncKalshiClient) -> None:
+        self._client = client
+
+    async def list(self) -> list[APIKey]:
+        """List all API keys for this account."""
+        data = await self._client.get("/api_keys")
+        return [APIKey.model_validate(k) for k in data.get("api_keys", [])]
+
+    async def create(self, public_key: str, name: str | None = None) -> str:
+        """Create an API key with a provided RSA public key.
+
+        Args:
+            public_key: PEM-encoded RSA public key.
+            name: Optional name for the key.
+
+        Returns:
+            The API key ID string.
+        """
+        body: dict = {"public_key": public_key}
+        if name:
+            body["name"] = name
+        data = await self._client.post("/api_keys", body)
+        return data["api_key_id"]
+
+    async def generate(self, name: str | None = None) -> GeneratedAPIKey:
+        """Generate a new API key pair (Kalshi creates both keys).
+
+        Returns a GeneratedAPIKey with the private_key field populated.
+        The private key is only returned ONCE - store it securely.
+
+        Args:
+            name: Optional name for the key.
+        """
+        body: dict = {}
+        if name:
+            body["name"] = name
+        data = await self._client.post("/api_keys/generate", body)
+        return GeneratedAPIKey.model_validate(data)
+
+    async def delete(self, key_id: str) -> None:
+        """Delete an API key.
+
+        Args:
+            key_id: The API key ID to delete.
+        """
+        await self._client.delete(f"/api_keys/{key_id}")
+
+    async def get_limits(self) -> APILimits:
+        """Get API rate limits for this account."""
+        data = await self._client.get("/account/limits")
+        return APILimits.model_validate(data)