Add lint/format checks using ruff

Author: justyns

.github/workflows/test.yml

@@ -6,8 +6,29 @@ permissions:
   contents: read
 
 jobs:
+  lint:
+    runs-on: ubuntu-latest
+    steps:
+    - uses: actions/checkout@v4
+    - name: Set up Python
+      uses: actions/setup-python@v5
+      with:
+        python-version: "3.12"
+        cache: pip
+        cache-dependency-path: pyproject.toml
+    - name: Install dependencies
+      run: |
+        pip install -e '.[dev]'
+    - name: Run ruff linter
+      run: |
+        ruff check .
+    - name: Run ruff formatter
+      run: |
+        ruff format --check .
+
   test:
     runs-on: ubuntu-latest
+    needs: lint
     strategy:
       matrix:
         python-version: ["3.9", "3.10", "3.11", "3.12", "3.13"]

llm_tools_searxng.py

@@ -59,9 +59,7 @@ def search(
             # TODO: Set user agent?
             headers = None
 
-            with httpx.Client(
-                follow_redirects=True, timeout=30.0, headers=headers
-            ) as client:
+            with httpx.Client(follow_redirects=True, timeout=30.0, headers=headers) as client:
                 if self.method == "POST":
                     response = client.post(search_url, data=params)
                 else:

pyproject.toml

@@ -18,8 +18,7 @@ build-backend = "setuptools.build_meta"
 
 [dependency-groups]
 dev = [
-    "black>=25.1.0",
-    "isort>=6.0.1",
+    "ruff>=0.11.12",
 ]
 
 [project.urls]
@@ -33,3 +32,16 @@ llm_tools_searxng = "llm_tools_searxng"
 
 [project.optional-dependencies]
 test = ["pytest", "llm-echo>=0.3a1", "pytest-httpx"]
+
+[tool.ruff]
+line-length = 120
+target-version = "py39"
+
+[tool.ruff.lint]
+ignore = []
+
+[tool.ruff.format]
+quote-style = "double"
+indent-style = "space"
+skip-magic-trailing-comma = false
+line-ending = "auto"

tests/test_llm_tools_searxng.py

@@ -1,8 +1,6 @@
 import json
-import os
 
 import llm
-import pytest
 
 from llm_tools_searxng import SearXNG, searxng_search
 
@@ -33,13 +31,7 @@ def test_searxng_search_function_get(httpx_mock, monkeypatch):
 
     model = llm.get_model("echo")
     chain_response = model.chain(
-        json.dumps(
-            {
-                "tool_calls": [
-                    {"name": "searxng_search", "arguments": {"query": "test query"}}
-                ]
-            }
-        ),
+        json.dumps({"tool_calls": [{"name": "searxng_search", "arguments": {"query": "test query"}}]}),
         tools=[searxng_search],
     )
     responses = list(chain_response.responses())
@@ -69,19 +61,11 @@ def test_searxng_search_function_post(httpx_mock, monkeypatch):
             }
         ],
     }
-    httpx_mock.add_response(
-        url="https://searx.be/search", json=mock_response, method="POST"
-    )
+    httpx_mock.add_response(url="https://searx.be/search", json=mock_response, method="POST")
 
     model = llm.get_model("echo")
     chain_response = model.chain(
-        json.dumps(
-            {
-                "tool_calls": [
-                    {"name": "searxng_search", "arguments": {"query": "test query"}}
-                ]
-            }
-        ),
+        json.dumps({"tool_calls": [{"name": "searxng_search", "arguments": {"query": "test query"}}]}),
         tools=[searxng_search],
     )
     responses = list(chain_response.responses())
@@ -139,9 +123,7 @@ def test_searxng_class_direct_post_default(httpx_mock):
             }
         ],
     }
-    httpx_mock.add_response(
-        url="https://custom.searxng.com/search", json=mock_response, method="POST"
-    )
+    httpx_mock.add_response(url="https://custom.searxng.com/search", json=mock_response, method="POST")
 
     # Test the SearXNG class directly without setting method (should default to POST)
     searxng = SearXNG("https://custom.searxng.com")