Use ruff

amandeepg · Jan 7, 2025 · 02a8511 · 02a8511
1 parent 7f4c9c0
commit 02a8511
Show file tree

Hide file tree

Showing 5 changed files with 6 additions and 50 deletions.
diff --git a/.github/workflows/python.yml b/.github/workflows/python.yml
@@ -30,8 +30,8 @@ jobs:
     - name: Install the project
       run: uv sync --all-extras --dev --directory serverless/path-summarize-alerts/
 
-    - name: "Compile python code with flake8"
-      run: uv --directory serverless/path-summarize-alerts/ run flake8 src --count --select=E9,F63,F7,F82 --show-source --statistics
+    - name: "Compile python code with ruff"
+      run: uv --directory serverless/path-summarize-alerts/ tool run ruff check
 
-    - name: "Check python formatting with black"
-      run: uv --directory serverless/path-summarize-alerts/ run black . --check
+    - name: "Check python formatting with ruff"
+      run: uv --directory serverless/path-summarize-alerts/ tool run ruff format --check
diff --git a/serverless/path-summarize-alerts/pyproject.toml b/serverless/path-summarize-alerts/pyproject.toml
@@ -54,5 +54,4 @@ dependencies = [
     "langsmith>=0.2.10",
     "aws-xray-sdk>=2.14.0",
     "tool>=0.8.0",
-    "flake8>=7.1.1",
 ]
diff --git a/serverless/path-summarize-alerts/src/lib/cache.py b/serverless/path-summarize-alerts/src/lib/cache.py
@@ -5,7 +5,7 @@
 import boto3
 from aws_lambda_powertools import Logger, Tracer
 
-from .constants import *
+from .constants import SYSTEM_MESSAGE, MODEL_NAME, CACHE_INT
 from .models import CacheResponse
 
 logger = Logger()

diff --git a/serverless/path-summarize-alerts/src/lib/summarizer.py b/serverless/path-summarize-alerts/src/lib/summarizer.py
@@ -10,7 +10,7 @@
 from openai import AsyncOpenAI
 
 from .cache import CacheService
-from .constants import *
+from .constants import BUCKET_NAME, MODEL_NAME, BUCKET_NAME_RATE_LIMIT, SYSTEM_MESSAGE
 from .models import CacheResponse, AlertSummary
 
 logger = Logger()

diff --git a/serverless/path-summarize-alerts/uv.lock b/serverless/path-summarize-alerts/uv.lock