apply ruff linter

Author: daixba

.pre-commit-config.yaml

@@ -5,5 +5,6 @@ repos:
     hooks:
       # Run the linter.
       - id: ruff
+        types_or: [python, pyi]
       # Run the formatter.
       - id: ruff-format

ruff.toml

@@ -9,13 +9,11 @@ exclude = [
 ]
 
 [lint]
-select = ["E", "F"]
+select = ["E", "F", "I"]
 ignore = [
     "E501",
-    "B008",
     "C901",
     "F401",
-    "W191",
 ]
 
 [format]

src/api/app.py

@@ -7,8 +7,8 @@
 from fastapi.responses import PlainTextResponse
 from mangum import Mangum
 
-from api.routers import model, chat, embeddings
-from api.setting import API_ROUTE_PREFIX, TITLE, DESCRIPTION, SUMMARY, VERSION
+from api.routers import chat, embeddings, model
+from api.setting import API_ROUTE_PREFIX, DESCRIPTION, SUMMARY, TITLE, VERSION
 
 config = {
     "title": TITLE,
@@ -31,6 +31,7 @@
     allow_headers=["*"],
 )
 
+
 app.include_router(model.router, prefix=API_ROUTE_PREFIX)
 app.include_router(chat.router, prefix=API_ROUTE_PREFIX)
 app.include_router(embeddings.router, prefix=API_ROUTE_PREFIX)

src/api/models/base.py

@@ -5,8 +5,8 @@
 
 from api.schema import (
     # Chat
-    ChatResponse,
     ChatRequest,
+    ChatResponse,
     ChatStreamResponse,
     # Embeddings
     EmbeddingsRequest,

src/api/models/bedrock.py

@@ -15,29 +15,27 @@
 
 from api.models.base import BaseChatModel, BaseEmbeddingsModel
 from api.schema import (
-    # Chat
-    ChatResponse,
+    AssistantMessage,
     ChatRequest,
-    Choice,
+    ChatResponse,
     ChatResponseMessage,
-    Usage,
     ChatStreamResponse,
-    ImageContent,
-    TextContent,
-    ToolCall,
+    Choice,
     ChoiceDelta,
-    UserMessage,
-    AssistantMessage,
-    ToolMessage,
-    Function,
-    ResponseFunction,
-    # Embeddings
+    Embedding,
     EmbeddingsRequest,
     EmbeddingsResponse,
     EmbeddingsUsage,
-    Embedding,
+    Function,
+    ImageContent,
+    ResponseFunction,
+    TextContent,
+    ToolCall,
+    ToolMessage,
+    Usage,
+    UserMessage,
 )
-from api.setting import DEBUG, AWS_REGION, ENABLE_CROSS_REGION_INFERENCE, DEFAULT_MODEL
+from api.setting import AWS_REGION, DEBUG, DEFAULT_MODEL, ENABLE_CROSS_REGION_INFERENCE
 
 logger = logging.getLogger(__name__)
 

src/api/routers/chat.py

@@ -1,6 +1,6 @@
 from typing import Annotated
 
-from fastapi import APIRouter, Depends, Body
+from fastapi import APIRouter, Body, Depends
 from fastapi.responses import StreamingResponse
 
 from api.auth import api_key_auth

src/api/routers/embeddings.py

@@ -1,6 +1,6 @@
 from typing import Annotated
 
-from fastapi import APIRouter, Depends, Body
+from fastapi import APIRouter, Body, Depends
 
 from api.auth import api_key_auth
 from api.models.bedrock import get_embeddings_model

src/api/routers/model.py

@@ -4,7 +4,7 @@
 
 from api.auth import api_key_auth
 from api.models.bedrock import BedrockModel
-from api.schema import Models, Model
+from api.schema import Model, Models
 
 router = APIRouter(
     prefix="/models",

src/api/schema.py

@@ -1,5 +1,5 @@
 import time
-from typing import Literal, Iterable
+from typing import Iterable, Literal
 
 from pydantic import BaseModel, Field