aquin 0.1.0__tar.gz

aquin-0.1.0/PKG-INFO

@@ -0,0 +1,11 @@
+Metadata-Version: 2.4
+Name: aquin
+Version: 0.1.0
+Summary: Python SDK for the Aquin model API
+Project-URL: Homepage, https://aquin.app
+License: MIT
+Requires-Python: >=3.10
+Requires-Dist: httpx>=0.27.0
+Description-Content-Type: text/markdown
+
+Aquin SDK

aquin-0.1.0/README.md

@@ -0,0 +1 @@
+Aquin SDK

aquin-0.1.0/aquin/__init__.py

@@ -0,0 +1,20 @@
+from aquin.client import AquinClient, CompletionResponse
+from aquin.exceptions import (
+    AquinError,
+    InvalidKeyError,
+    InsufficientCreditsError,
+    RateLimitError,
+    ModelNotFoundError,
+    InferenceError,
+)
+
+__all__ = [
+    "AquinClient",
+    "CompletionResponse",
+    "AquinError",
+    "InvalidKeyError",
+    "InsufficientCreditsError",
+    "RateLimitError",
+    "ModelNotFoundError",
+    "InferenceError",
+]

aquin-0.1.0/aquin/client.py

@@ -0,0 +1,106 @@
+import httpx
+from dataclasses import dataclass
+from aquin.exceptions import (
+    AquinError,
+    InvalidKeyError,
+    InsufficientCreditsError,
+    RateLimitError,
+    ModelNotFoundError,
+    InferenceError,
+)
+
+DEFAULT_BASE_URL = "https://www.aquin.app"
+
+
+@dataclass
+class CompletionResponse:
+    text: str
+    tokens_in: int
+    tokens_out: int
+
+
+class AquinClient:
+    def __init__(self, api_key: str, base_url: str = DEFAULT_BASE_URL):
+        if not api_key:
+            raise ValueError("api_key is required")
+        self._api_key = api_key
+        self._base_url = base_url.rstrip("/")
+
+    def complete(
+        self,
+        prompt: str,
+        max_tokens: int = 512,
+        temperature: float = 0.7,
+    ) -> CompletionResponse:
+        try:
+            res = httpx.post(
+                f"{self._base_url}/api/v1/model",
+                headers={
+                    "Authorization": f"Bearer {self._api_key}",
+                    "Content-Type": "application/json",
+                },
+                json={
+                    "prompt": prompt,
+                    "max_tokens": max_tokens,
+                    "temperature": temperature,
+                },
+                timeout=120,
+                follow_redirects=True,
+            )
+        except httpx.ConnectError:
+            raise InferenceError("Could not connect to Aquin API.")
+        except httpx.TimeoutException:
+            raise InferenceError("Request timed out.")
+
+        return self._handle_response(res)
+
+    async def acomplete(
+        self,
+        prompt: str,
+        max_tokens: int = 512,
+        temperature: float = 0.7,
+    ) -> CompletionResponse:
+        async with httpx.AsyncClient() as client:
+            try:
+                res = await client.post(
+                    f"{self._base_url}/api/v1/model",
+                    headers={
+                        "Authorization": f"Bearer {self._api_key}",
+                        "Content-Type": "application/json",
+                    },
+                    json={
+                        "prompt": prompt,
+                        "max_tokens": max_tokens,
+                        "temperature": temperature,
+                    },
+                    timeout=120,
+                    follow_redirects=True,
+                )
+            except httpx.ConnectError:
+                raise InferenceError("Could not connect to Aquin API.")
+            except httpx.TimeoutException:
+                raise InferenceError("Request timed out.")
+
+        return self._handle_response(res)
+
+    def _handle_response(self, res: httpx.Response) -> CompletionResponse:
+        if res.status_code == 200:
+            data = res.json()
+            return CompletionResponse(
+                text=data["text"],
+                tokens_in=data["tokens_in"],
+                tokens_out=data["tokens_out"],
+            )
+
+        try:
+            error_msg = res.json().get("error", "Unknown error")
+        except Exception:
+            error_msg = f"HTTP {res.status_code}: {res.text or 'empty response'}"
+
+        match res.status_code:
+            case 401: raise InvalidKeyError(error_msg, 401)
+            case 402: raise InsufficientCreditsError(error_msg, 402)
+            case 404: raise ModelNotFoundError(error_msg, 404)
+            case 429: raise RateLimitError(error_msg, 429)
+            case 502 | 503: raise InferenceError(error_msg, res.status_code)
+            case _: raise AquinError(error_msg, res.status_code)

aquin-0.1.0/aquin/exceptions.py

@@ -0,0 +1,30 @@
+class AquinError(Exception):
+    """Base exception for all Aquin SDK errors."""
+    def __init__(self, message: str, status_code: int = None):
+        super().__init__(message)
+        self.status_code = status_code
+
+
+class InvalidKeyError(AquinError):
+    """API key is invalid or disabled."""
+    pass
+
+
+class InsufficientCreditsError(AquinError):
+    """Account does not have enough credits."""
+    pass
+
+
+class RateLimitError(AquinError):
+    """Too many requests."""
+    pass
+
+
+class ModelNotFoundError(AquinError):
+    """The model for this API key has not been trained yet."""
+    pass
+
+
+class InferenceError(AquinError):
+    """The inference server failed."""
+    pass

aquin-0.1.0/pyproject.toml

@@ -0,0 +1,17 @@
+[build-system]
+requires = ["hatchling"]
+build-backend = "hatchling.build"
+
+[project]
+name = "aquin"
+version = "0.1.0"
+description = "Python SDK for the Aquin model API"
+readme = "README.md"
+requires-python = ">=3.10"
+license = { text = "MIT" }
+dependencies = [
+    "httpx>=0.27.0",
+]
+
+[project.urls]
+Homepage = "https://aquin.app"

aquin-0.1.0/test.py

@@ -0,0 +1,9 @@
+from aquin import AquinClient
+
+client = AquinClient(
+    api_key="aq-m_R3tlzj46F8FhnJK6XtSMC2x4VYmjCDTNKDY1DkxNLVU",
+    base_url="https://www.aquin.app"
+)
+res = client.complete("What is LoRA?")
+print(res.text)
+print(f"tokens_in={res.tokens_in} tokens_out={res.tokens_out}")