apply ruff linter

Author: daixba

.flake8

@@ -0,0 +1,8 @@
+- repo: https://github.com/astral-sh/ruff-pre-commit
+  # Ruff version.
+  rev: v0.9.10
+  hooks:
+    # Run the linter.
+    - id: ruff
+    # Run the formatter.
+    - id: ruff-format

.pre-commit-config.yaml

@@ -0,0 +1,23 @@
+line-length = 120
+indent-width = 4
+target-version = "py312"
+
+exclude = [
+    ".venv",
+    ".vscode",
+    "test/*"
+]
+
+[lint]
+select = ["E", "F"]
+ignore = [
+    "E501",
+    "B008",
+    "C901",
+    "F401",
+    "W191",
+]
+
+[format]
+# use double quotes for strings.
+quote-style = "double"

ruff.toml

@@ -16,21 +16,17 @@
     # For backward compatibility.
     # Please now use secrets manager instead.
     ssm = boto3.client("ssm")
-    api_key = ssm.get_parameter(Name=api_key_param, WithDecryption=True)["Parameter"][
-        "Value"
-    ]
+    api_key = ssm.get_parameter(Name=api_key_param, WithDecryption=True)["Parameter"]["Value"]
 elif api_key_secret_arn:
     sm = boto3.client("secretsmanager")
     try:
         response = sm.get_secret_value(SecretId=api_key_secret_arn)
         if "SecretString" in response:
             secret = json.loads(response["SecretString"])
             api_key = secret["api_key"]
-    except ClientError as e:
-        raise RuntimeError(
-            "Unable to retrieve API KEY, please ensure the secret ARN is correct"
-        )
-    except KeyError as e:
+    except ClientError:
+        raise RuntimeError("Unable to retrieve API KEY, please ensure the secret ARN is correct")
+    except KeyError:
         raise RuntimeError('Please ensure the secret contains a "api_key" field')
 elif api_key_env:
     api_key = api_key_env
@@ -45,6 +41,4 @@ def api_key_auth(
     credentials: Annotated[HTTPAuthorizationCredentials, Depends(security)],
 ):
     if credentials.credentials != api_key:
-        raise HTTPException(
-            status_code=status.HTTP_401_UNAUTHORIZED, detail="Invalid API Key"
-        )
+        raise HTTPException(status_code=status.HTTP_401_UNAUTHORIZED, detail="Invalid API Key")

src/api/auth.py

@@ -43,9 +43,7 @@ def generate_message_id() -> str:
         return "chatcmpl-" + str(uuid.uuid4())[:8]
 
     @staticmethod
-    def stream_response_to_bytes(
-            response: ChatStreamResponse | None = None
-    ) -> bytes:
+    def stream_response_to_bytes(response: ChatStreamResponse | None = None) -> bytes:
         if response:
             # to populate other fields when using exclude_unset=True
             response.system_fingerprint = "fp"