pythonize

Author: CLEMENTINATOR

.github/workflows/check.yml

@@ -10,12 +10,17 @@ jobs:
     runs-on: ubuntu-latest
     steps:
       - uses: actions/checkout@v4
-      - uses: actions/setup-python@v5
+      - name: Install uv
+        uses: astral-sh/setup-uv@v4
         with:
-          python-version: "3.12"
-    #   - name: Install
-    #     run: pip install -r requirements.txt
-    #   - name: Lint
-    #     run: pylint edgee
-    #   - name: Test
-    #     run: python -m unittest discover -s tests -p 'test_*.py'
+          enable-cache: true
+      - name: Set up Python
+        run: uv python install 3.12
+      - name: Install dependencies
+        run: uv sync --all-extras
+      - name: Ruff format check
+        run: uv run ruff format --check .
+      - name: Ruff lint
+        run: uv run ruff check .
+      - name: Run tests
+        run: uv run pytest

edgee/__init__.py

@@ -1,18 +1,21 @@
 """Edgee Gateway SDK for Python"""
 
-import os
 import json
-from typing import Optional, Union
+import os
 from dataclasses import dataclass
-from urllib.request import Request, urlopen
 from urllib.error import HTTPError
+from urllib.request import Request, urlopen
+
+# API Configuration
+DEFAULT_BASE_URL = "https://api.edgee.ai"
+API_ENDPOINT = "/v1/chat/completions"
 
 
 @dataclass
 class FunctionDefinition:
     name: str
-    description: Optional[str] = None
-    parameters: Optional[dict] = None
+    description: str | None = None
+    parameters: dict | None = None
 
 
 @dataclass
@@ -31,24 +34,24 @@ class ToolCall:
 @dataclass
 class Message:
     role: str  # "system" | "user" | "assistant" | "tool"
-    content: Optional[str] = None
-    name: Optional[str] = None
-    tool_calls: Optional[list[ToolCall]] = None
-    tool_call_id: Optional[str] = None
+    content: str | None = None
+    name: str | None = None
+    tool_calls: list[ToolCall] | None = None
+    tool_call_id: str | None = None
 
 
 @dataclass
 class InputObject:
     messages: list[dict]
-    tools: Optional[list[dict]] = None
-    tool_choice: Optional[Union[str, dict]] = None
+    tools: list[dict] | None = None
+    tool_choice: str | dict | None = None
 
 
 @dataclass
 class Choice:
     index: int
     message: dict
-    finish_reason: Optional[str]
+    finish_reason: str | None
 
 
 @dataclass
@@ -61,43 +64,44 @@ class Usage:
 @dataclass
 class SendResponse:
     choices: list[Choice]
-    usage: Optional[Usage] = None
+    usage: Usage | None = None
 
 
 @dataclass
 class EdgeeConfig:
-    api_key: Optional[str] = None
-    base_url: Optional[str] = None
+    api_key: str | None = None
+    base_url: str | None = None
 
 
 class Edgee:
     def __init__(
         self,
-        config: Optional[Union[str, EdgeeConfig, dict]] = None,
+        config: str | EdgeeConfig | dict | None = None,
     ):
-        api_key: Optional[str] = None
-        base_url: Optional[str] = None
-
         if isinstance(config, str):
             # Backward compatibility: accept api_key as string
             api_key = config
+            base_url = None
         elif isinstance(config, EdgeeConfig):
             api_key = config.api_key
             base_url = config.base_url
         elif isinstance(config, dict):
             api_key = config.get("api_key")
             base_url = config.get("base_url")
+        else:
+            api_key = None
+            base_url = None
 
         self.api_key = api_key or os.environ.get("EDGEE_API_KEY", "")
         if not self.api_key:
             raise ValueError("EDGEE_API_KEY is not set")
 
-        self.base_url = base_url or os.environ.get("EDGEE_BASE_URL", "https://api.edgee.ai")
+        self.base_url = base_url or os.environ.get("EDGEE_BASE_URL", DEFAULT_BASE_URL)
 
     def send(
         self,
         model: str,
-        input: Union[str, InputObject, dict],
+        input: str | InputObject | dict,
     ) -> SendResponse:
         """Send a completion request to the Edgee AI Gateway."""
 
@@ -121,7 +125,7 @@ def send(
             body["tool_choice"] = tool_choice
 
         request = Request(
-            f"{self.base_url}/v1/chat/completions",
+            f"{self.base_url}{API_ENDPOINT}",
             data=json.dumps(body).encode("utf-8"),
             headers={
                 "Content-Type": "application/json",

example/test.py

@@ -61,4 +61,3 @@
 )
 print(f"Content: {response3.choices[0].message.get('content')}")
 print(f"Tool calls: {response3.choices[0].message.get('tool_calls')}")
-

pyproject.toml

@@ -26,11 +26,39 @@ Homepage = "https://github.com/edgee-cloud/python-sdk"
 Repository = "https://github.com/edgee-cloud/python-sdk"
 
 [project.optional-dependencies]
-dev = ["pytest>=8.0.0"]
+dev = ["pytest>=8.0.0", "ruff>=0.8.0"]
 
 [build-system]
 requires = ["hatchling"]
 build-backend = "hatchling.build"
 
 [tool.pytest.ini_options]
 testpaths = ["tests"]
+
+[tool.ruff]
+target-version = "py310"
+line-length = 100
+
+[tool.ruff.lint]
+select = [
+    "E",   # pycodestyle errors
+    "W",   # pycodestyle warnings
+    "F",   # pyflakes
+    "I",   # isort
+    "B",   # flake8-bugbear
+    "C4",  # flake8-comprehensions
+    "UP",  # pyupgrade
+]
+ignore = [
+    "E501",  # line too long (handled by formatter)
+]
+
+[tool.ruff.format]
+quote-style = "double"
+indent-style = "space"
+
+[dependency-groups]
+dev = [
+    "pytest>=8.0.0",
+    "ruff>=0.8.0",
+]

tests/test_edgee.py

@@ -2,11 +2,11 @@
 
 import json
 import os
+from unittest.mock import MagicMock, patch
+
 import pytest
-from unittest.mock import patch, MagicMock
-from io import BytesIO
 
-from edgee import Edgee, EdgeeConfig, SendResponse, Choice, Usage
+from edgee import Edgee, EdgeeConfig
 
 
 class TestEdgeeConstructor:
@@ -122,7 +122,7 @@ def test_send_with_input_object(self, mock_urlopen):
         mock_urlopen.return_value = self._mock_response(mock_response_data)
 
         client = Edgee("test-api-key")
-        result = client.send(
+        client.send(
             model="gpt-4",
             input={
                 "messages": [
@@ -306,4 +306,3 @@ def test_config_base_url_overrides_env(self, mock_urlopen):
 
         call_args = mock_urlopen.call_args[0][0]
         assert call_args.full_url == f"{config_base_url}/v1/chat/completions"
-