whisperserve-sdk 0.1.3__tar.gz

whisperserve_sdk-0.1.3/PKG-INFO

@@ -0,0 +1,18 @@
+Metadata-Version: 2.4
+Name: whisperserve-sdk
+Version: 0.1.3
+Summary: Python SDK for the WhisperServe Audio Transcription API
+Author: Pure Tech
+License: MIT
+Project-URL: Homepage, https://github.com/puretechteam/whisperserve
+Classifier: Development Status :: 3 - Alpha
+Classifier: Intended Audience :: Developers
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.8
+Classifier: Programming Language :: Python :: 3.9
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Requires-Python: >=3.8
+Requires-Dist: httpx>=0.25.0
+Dynamic: requires-python

whisperserve_sdk-0.1.3/README.md

@@ -0,0 +1,89 @@
+# WhisperServe SDK
+
+Python SDK for interacting with the WhisperServe Audio Transcription API.
+
+## Installation
+
+```bash
+pip install whisperserve-sdk
+```
+
+## Usage
+
+### Synchronous Client
+
+```python
+from sdk import InferenceClient
+
+client = InferenceClient(api_key="your-api-key")
+
+result = client.transcribe("audio.wav")
+print(result["transcription"])
+
+usage = client.get_usage()
+print(f"Total requests: {usage['total_requests']}")
+
+client.close()
+```
+
+### Asynchronous Client
+
+```python
+import asyncio
+from sdk import AsyncInferenceClient
+
+async def main():
+    async with AsyncInferenceClient(api_key="your-api-key") as client:
+        result = await client.transcribe("audio.wav")
+        print(result["transcription"])
+
+asyncio.run(main())
+```
+
+### Error Handling
+
+```python
+from sdk import InferenceClient, InferenceError
+
+client = InferenceClient(api_key="your-api-key")
+
+try:
+    result = client.transcribe("audio.wav")
+except InferenceError as e:
+    print(f"Error {e.status_code}: {e.message}")
+```
+
+### Context Manager
+
+```python
+from sdk import InferenceClient
+
+with InferenceClient(api_key="your-api-key") as client:
+    result = client.transcribe("audio.wav")
+    print(result["transcription"])
+```
+
+## API Reference
+
+### InferenceClient
+
+| Method | Description |
+|--------|-------------|
+| `transcribe(audio_path)` | Transcribe an audio file |
+| `transcribe_batch(audio_paths)` | Transcribe multiple audio files |
+| `get_usage()` | Get usage statistics |
+| `list_keys()` | List API keys |
+| `revoke_key(key_id)` | Revoke an API key |
+| `close()` | Close the HTTP client |
+
+### AsyncInferenceClient
+
+Same methods as `InferenceClient`, but async versions (prefixed with `async`). Supports `async with` context manager.
+
+### InferenceError
+
+Exception raised when API requests fail. Contains `message` and `status_code` attributes.
+
+## License
+
+MIT License.

whisperserve_sdk-0.1.3/VERSION

@@ -0,0 +1 @@
+0.1.3

whisperserve_sdk-0.1.3/__init__.py

@@ -0,0 +1,4 @@
+from .client import InferenceClient, InferenceError
+from .async_client import AsyncInferenceClient
+
+__all__ = ["InferenceClient", "AsyncInferenceClient", "InferenceError"]

whisperserve_sdk-0.1.3/async_client.py

@@ -0,0 +1,119 @@
+import asyncio
+import httpx
+from typing import Optional
+
+from .client import InferenceError
+
+
+class AsyncInferenceClient:
+    def __init__(
+        self,
+        api_key: str,
+        base_url: str = "http://localhost:8000",
+        timeout: float = 30.0,
+    ):
+        self._api_key = api_key
+        self._base_url = base_url.rstrip("/")
+        self._timeout = timeout
+        self._client: Optional[httpx.AsyncClient] = None
+
+    async def __aenter__(self):
+        self._client = httpx.AsyncClient(
+            headers={"Authorization": f"Bearer {self._api_key}"},
+            timeout=self._timeout,
+        )
+        return self
+
+    async def __aexit__(self, exc_type, exc_val, exc_tb):
+        await self.close()
+
+    async def _get_client(self) -> httpx.AsyncClient:
+        if self._client is None:
+            self._client = httpx.AsyncClient(
+                headers={"Authorization": f"Bearer {self._api_key}"},
+                timeout=self._timeout,
+            )
+        return self._client
+
+    async def _request_with_retry(
+        self,
+        method: str,
+        url: str,
+        *,
+        max_retries: int = 3,
+        **kwargs,
+    ):
+        await self._get_client()
+        delay = 1.0
+        last_exception = None
+        for attempt in range(max_retries + 1):
+            try:
+                response = await self._client.request(method, url, **kwargs)
+                return response
+            except httpx.TransportError as e:
+                last_exception = e
+                if attempt < max_retries:
+                    await asyncio.sleep(delay)
+                    delay *= 2.0
+                    continue
+                raise
+        raise last_exception  # type: ignore[misc]
+
+    async def transcribe(self, audio_path: str) -> dict:
+        with open(audio_path, "rb") as f:
+            response = await self._request_with_retry(
+                "POST",
+                f"{self._base_url}/v1/inference",
+                files={"file": f},
+            )
+        if response.status_code != 200:
+            raise InferenceError(
+                response.json().get("detail", "Inference failed"),
+                status_code=response.status_code,
+            )
+        return response.json()
+
+    async def transcribe_batch(self, audio_paths: list) -> list:
+        tasks = [self.transcribe(path) for path in audio_paths]
+        return await asyncio.gather(*tasks)
+
+    async def get_usage(self) -> dict:
+        response = await self._request_with_retry(
+            "GET",
+            f"{self._base_url}/v1/self-serve/dashboard",
+        )
+        if response.status_code != 200:
+            raise InferenceError(
+                response.json().get("detail", "Failed to fetch usage"),
+                status_code=response.status_code,
+            )
+        return response.json()
+
+    async def list_keys(self) -> list:
+        response = await self._request_with_retry(
+            "GET",
+            f"{self._base_url}/v1/self-serve/api-keys",
+        )
+        if response.status_code != 200:
+            raise InferenceError(
+                response.json().get("detail", "Failed to list API keys"),
+                status_code=response.status_code,
+            )
+        return response.json()
+
+    async def revoke_key(self, key_id: int) -> bool:
+        response = await self._request_with_retry(
+            "DELETE",
+            f"{self._base_url}/v1/self-serve/api-keys/{key_id}",
+        )
+        if response.status_code != 200:
+            raise InferenceError(
+                response.json().get("detail", "Failed to revoke API key"),
+                status_code=response.status_code,
+            )
+        return True
+
+    async def close(self):
+        if self._client is not None:
+            await self._client.aclose()
+            self._client = None

whisperserve_sdk-0.1.3/client.py

@@ -0,0 +1,115 @@
+import concurrent.futures
+import time
+import httpx
+from typing import Optional
+
+
+class InferenceError(Exception):
+    def __init__(self, message: str, status_code: int = 500):
+        self.message = message
+        self.status_code = status_code
+        super().__init__(message)
+
+
+class InferenceClient:
+    def __init__(
+        self,
+        api_key: str,
+        base_url: str = "http://localhost:8000",
+        timeout: float = 30.0,
+    ):
+        self._api_key = api_key
+        self._base_url = base_url.rstrip("/")
+        self._timeout = timeout
+        self._client = httpx.Client(
+            headers={"Authorization": f"Bearer {api_key}"},
+            timeout=timeout,
+        )
+
+    def _request_with_retry(
+        self,
+        method: str,
+        url: str,
+        *,
+        max_retries: int = 3,
+        **kwargs,
+    ):
+        delay = 1.0
+        last_exception = None
+        for attempt in range(max_retries + 1):
+            try:
+                response = self._client.request(method, url, **kwargs)
+                return response
+            except httpx.TransportError as e:
+                last_exception = e
+                if attempt < max_retries:
+                    time.sleep(delay)
+                    delay *= 2.0
+                    continue
+                raise
+        raise last_exception  # type: ignore[misc]
+
+    def transcribe(self, audio_path: str) -> dict:
+        with open(audio_path, "rb") as f:
+            response = self._request_with_retry(
+                "POST",
+                f"{self._base_url}/v1/inference",
+                files={"file": f},
+            )
+        if response.status_code != 200:
+            raise InferenceError(
+                response.json().get("detail", "Inference failed"),
+                status_code=response.status_code,
+            )
+        return response.json()
+
+    def transcribe_batch(self, audio_paths: list) -> list:
+        with concurrent.futures.ThreadPoolExecutor() as executor:
+            futures = [executor.submit(self.transcribe, path) for path in audio_paths]
+            results = [f.result() for f in concurrent.futures.as_completed(futures)]
+        return results
+
+    def get_usage(self) -> dict:
+        response = self._request_with_retry(
+            "GET",
+            f"{self._base_url}/v1/self-serve/dashboard",
+        )
+        if response.status_code != 200:
+            raise InferenceError(
+                response.json().get("detail", "Failed to fetch usage"),
+                status_code=response.status_code,
+            )
+        return response.json()
+
+    def list_keys(self) -> list:
+        response = self._request_with_retry(
+            "GET",
+            f"{self._base_url}/v1/self-serve/api-keys",
+        )
+        if response.status_code != 200:
+            raise InferenceError(
+                response.json().get("detail", "Failed to list API keys"),
+                status_code=response.status_code,
+            )
+        return response.json()
+
+    def revoke_key(self, key_id: int) -> bool:
+        response = self._request_with_retry(
+            "DELETE",
+            f"{self._base_url}/v1/self-serve/api-keys/{key_id}",
+        )
+        if response.status_code != 200:
+            raise InferenceError(
+                response.json().get("detail", "Failed to revoke API key"),
+                status_code=response.status_code,
+            )
+        return True
+
+    def close(self):
+        self._client.close()
+
+    def __enter__(self):
+        return self
+
+    def __exit__(self, exc_type, exc_val, exc_tb):
+        self.close()

whisperserve_sdk-0.1.3/pyproject.toml

@@ -0,0 +1,38 @@
+[build-system]
+requires = ["setuptools>=68.0", "wheel"]
+build-backend = "setuptools.build_meta"
+
+[project]
+name = "whisperserve-sdk"
+dynamic = ["version"]
+description = "Python SDK for the WhisperServe Audio Transcription API"
+requires-python = ">=3.8"
+authors = [
+    {name = "Pure Tech"},
+]
+license = {text = "MIT"}
+classifiers = [
+    "Development Status :: 3 - Alpha",
+    "Intended Audience :: Developers",
+    "License :: OSI Approved :: MIT License",
+    "Programming Language :: Python :: 3",
+    "Programming Language :: Python :: 3.8",
+    "Programming Language :: Python :: 3.9",
+    "Programming Language :: Python :: 3.10",
+    "Programming Language :: Python :: 3.11",
+]
+dependencies = [
+    "httpx>=0.25.0",
+]
+
+[project.urls]
+Homepage = "https://github.com/puretechteam/whisperserve"
+
+[tool.setuptools.dynamic]
+version = {file = "VERSION"}
+
+[tool.setuptools]
+packages = ["sdk"]
+
+[tool.setuptools.package-dir]
+"sdk" = "."

whisperserve_sdk-0.1.3/setup.cfg

@@ -0,0 +1,4 @@
+[egg_info]
+tag_build = 
+tag_date = 0
+

whisperserve_sdk-0.1.3/setup.py

@@ -0,0 +1,16 @@
+from setuptools import setup
+
+with open("VERSION") as f:
+    version = f.read().strip()
+
+setup(
+    name="whisperserve-sdk",
+    version=version,
+    description="Python SDK for the WhisperServe Audio Transcription API",
+    packages=["sdk"],
+    package_dir={"sdk": "."},
+    install_requires=[
+        "httpx>=0.25.0",
+    ],
+    python_requires=">=3.8",
+)

whisperserve_sdk-0.1.3/whisperserve_sdk.egg-info/PKG-INFO

@@ -0,0 +1,18 @@
+Metadata-Version: 2.4
+Name: whisperserve-sdk
+Version: 0.1.3
+Summary: Python SDK for the WhisperServe Audio Transcription API
+Author: Pure Tech
+License: MIT
+Project-URL: Homepage, https://github.com/puretechteam/whisperserve
+Classifier: Development Status :: 3 - Alpha
+Classifier: Intended Audience :: Developers
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.8
+Classifier: Programming Language :: Python :: 3.9
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Requires-Python: >=3.8
+Requires-Dist: httpx>=0.25.0
+Dynamic: requires-python

whisperserve_sdk-0.1.3/whisperserve_sdk.egg-info/SOURCES.txt

@@ -0,0 +1,16 @@
+README.md
+VERSION
+__init__.py
+async_client.py
+client.py
+pyproject.toml
+setup.py
+./__init__.py
+./async_client.py
+./client.py
+./setup.py
+whisperserve_sdk.egg-info/PKG-INFO
+whisperserve_sdk.egg-info/SOURCES.txt
+whisperserve_sdk.egg-info/dependency_links.txt
+whisperserve_sdk.egg-info/requires.txt
+whisperserve_sdk.egg-info/top_level.txt

whisperserve_sdk-0.1.3/whisperserve_sdk.egg-info/dependency_links.txt

@@ -0,0 +1 @@
+

whisperserve_sdk-0.1.3/whisperserve_sdk.egg-info/requires.txt

@@ -0,0 +1 @@
+httpx>=0.25.0

whisperserve_sdk-0.1.3/whisperserve_sdk.egg-info/top_level.txt

@@ -0,0 +1 @@
+sdk