sapienta 0.1.0__py3-none-any.whl

sapienta/__init__.py

@@ -0,0 +1 @@
+from .client import Sapienta, SapientaError

sapienta/client.py

@@ -0,0 +1,187 @@
+import httpx
+
+
+class SapientaError(Exception):
+    """Base exception for the Sapienta SDK."""
+
+
+class _ChatResponse:
+    """Response returned by the Sapienta chat API."""
+
+    __slots__ = (
+        "__content",
+        "__completion_tokens",
+        "__prompt_tokens",
+        "__total_tokens",
+    )
+
+    def __init__(
+        self,
+        content: str,
+        completion_tokens: int,
+        prompt_tokens: int,
+        total_tokens: int,
+    ):
+        object.__setattr__(self, "_ChatResponse__content", content)
+        object.__setattr__(
+            self, "_ChatResponse__completion_tokens", completion_tokens
+        )
+        object.__setattr__(
+            self, "_ChatResponse__prompt_tokens", prompt_tokens
+        )
+        object.__setattr__(
+            self, "_ChatResponse__total_tokens", total_tokens
+        )
+
+    @property
+    def content(self) -> str:
+        """The assistant's generated response."""
+        return self.__content
+
+    @property
+    def completion_tokens(self) -> int:
+        """Number of tokens generated by the model."""
+        return self.__completion_tokens
+
+    @property
+    def prompt_tokens(self) -> int:
+        """Number of tokens in the input prompt."""
+        return self.__prompt_tokens
+
+    @property
+    def total_tokens(self) -> int:
+        """Total number of tokens used by the request."""
+        return self.__total_tokens
+
+    def __setattr__(self, name, value):
+        raise AttributeError("ChatResponse is read-only.")
+
+    def __dir__(self):
+        return [
+            "content",
+            "completion_tokens",
+            "prompt_tokens",
+            "total_tokens",
+        ]
+
+
+class Sapienta:
+    _BASE_URL = "https://jett-lateenrigged-jessie.ngrok-free.dev"
+    _MODEL = "qwen3-vl:8b"
+
+    def __init__(self, api_key: str):
+        if not isinstance(api_key, str) or not api_key.strip():
+            raise SapientaError("API key is required.")
+
+        self._client = httpx.Client(
+            base_url=self._BASE_URL,
+            headers={
+                "Authorization": f"Bearer {api_key}",
+                "Content-Type": "application/json",
+            },
+            timeout=60.0,
+        )
+
+    def chat(self, message: str) -> _ChatResponse:
+        """
+        Send a message to Sapienta AI.
+
+        Args:
+            message: The message to send to the model.
+
+        Returns:
+            The generated response and token usage.
+
+        Raises:
+            SapientaError: If the request fails or the response is invalid.
+        """
+
+        if not isinstance(message, str) or not message.strip():
+            raise SapientaError("Message is required.")
+
+        try:
+            response = self._client.post(
+                "/v1/chat/completions",
+                json={
+                    "model": self._MODEL,
+                    "messages": [
+                        {
+                            "role": "user",
+                            "content": message,
+                        }
+                    ],
+                    "stream": False,
+                },
+            )
+
+            response.raise_for_status()
+            data = response.json()
+
+        except httpx.TimeoutException as e:
+            raise SapientaError(
+                "Sapienta request timed out. Please try again later."
+            ) from e
+
+        except httpx.HTTPStatusError as e:
+            status = e.response.status_code
+
+            if status == 401:
+                raise SapientaError("Invalid API key.") from e
+
+            if status == 403:
+                raise SapientaError(
+                    "You are not authorized to use Sapienta."
+                ) from e
+
+            if status >= 500:
+                raise SapientaError(
+                    "Sapienta is temporarily unavailable. "
+                    "Please try again later."
+                ) from e
+
+            raise SapientaError(
+                "Sapienta could not process your request. "
+                "Please try again later."
+            ) from e
+
+        except httpx.RequestError as e:
+            raise SapientaError(
+                "Unable to connect to Sapienta. Please try again later."
+            ) from e
+
+        except ValueError as e:
+            raise SapientaError(
+                "Sapienta returned an invalid response."
+            ) from e
+
+        try:
+            usage = data["usage"]
+            message_data = data["choices"][0]["message"]
+
+            content = message_data["content"]
+            completion_tokens = usage["completion_tokens"]
+            prompt_tokens = usage["prompt_tokens"]
+            total_tokens = usage["total_tokens"]
+
+            if (
+                not isinstance(content, str)
+                or not isinstance(completion_tokens, int)
+                or isinstance(completion_tokens, bool)
+                or not isinstance(prompt_tokens, int)
+                or isinstance(prompt_tokens, bool)
+                or not isinstance(total_tokens, int)
+                or isinstance(total_tokens, bool)
+            ):
+                raise TypeError("Unexpected response field types.")
+
+            return _ChatResponse(
+                content=content,
+                completion_tokens=completion_tokens,
+                prompt_tokens=prompt_tokens,
+                total_tokens=total_tokens,
+            )
+
+        except (KeyError, IndexError, TypeError) as e:
+            raise SapientaError(
+                "Sapienta returned an unexpected response."
+            ) from e

sapienta-0.1.0.dist-info/METADATA

@@ -0,0 +1,44 @@
+Metadata-Version: 2.4
+Name: sapienta
+Version: 0.1.0
+Summary: Python SDK for Sapienta AI
+Requires-Python: >=3.9
+Description-Content-Type: text/markdown
+Requires-Dist: httpx>=0.27.0
+
+# Sapienta Python SDK
+
+Official Python SDK for integrating Sapienta AI into your applications.
+
+## Installation
+
+```bash
+pip install sapienta
+```
+
+## Usage
+
+```python
+from sapienta import Sapienta
+
+client = Sapienta("YOUR_API_KEY")
+
+response = client.chat("Hello!")
+
+print(response.content)
+print(response.completion_tokens)
+print(response.prompt_tokens)
+print(response.total_tokens)
+```
+
+## API Key
+
+Contact the Sapienta team to obtain your API key.
+
+## Requirements
+
+Python 3.9 or later.
+
+## License
+
+Proprietary. All rights reserved.

sapienta-0.1.0.dist-info/RECORD

@@ -0,0 +1,6 @@
+sapienta/__init__.py,sha256=0cOI0mF9hlVV2Yk0DqggNE8RwXjCrFk-a5pmH1LpBDU,43
+sapienta/client.py,sha256=Dq4Q0nAtXeGu53bXna9qSOxyooU1X35npn7U6HqExqo,5496
+sapienta-0.1.0.dist-info/METADATA,sha256=8Qs0B94guA2RpelGk8duMiv9dG4Vpdce1odBvInYweA,718
+sapienta-0.1.0.dist-info/WHEEL,sha256=aeYiig01lYGDzBgS8HxWXOg3uV61G9ijOsup-k9o1sk,91
+sapienta-0.1.0.dist-info/top_level.txt,sha256=y0jm4xL71NAmabUZ3eF5fBAdvA232UHo7i6BT-_AVxo,9
+sapienta-0.1.0.dist-info/RECORD,,

sapienta-0.1.0.dist-info/WHEEL

@@ -0,0 +1,5 @@
+Wheel-Version: 1.0
+Generator: setuptools (82.0.1)
+Root-Is-Purelib: true
+Tag: py3-none-any
+

sapienta-0.1.0.dist-info/top_level.txt

@@ -0,0 +1 @@
+sapienta