apply ruff linter

This commit is contained in:
Aiden Dai
2025-03-13 14:24:41 +08:00
parent 879b8e2ac7
commit fa14ae8c05
9 changed files with 26 additions and 28 deletions

View File

@@ -7,8 +7,8 @@ from fastapi.middleware.cors import CORSMiddleware
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 @@ app.add_middleware(
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)

View File

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

View File

@@ -15,29 +15,27 @@ from fastapi import HTTPException
from api.models.base import BaseChatModel, BaseEmbeddingsModel
from api.schema import (
# Chat
ChatResponse,
ChatRequest,
Choice,
ChatResponseMessage,
Usage,
ChatStreamResponse,
ImageContent,
TextContent,
ToolCall,
ChoiceDelta,
UserMessage,
AssistantMessage,
ToolMessage,
Function,
ResponseFunction,
# Embeddings
ChatRequest,
ChatResponse,
ChatResponseMessage,
ChatStreamResponse,
Choice,
ChoiceDelta,
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__)

View File

@@ -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

View File

@@ -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

View File

@@ -4,7 +4,7 @@ from fastapi import APIRouter, Depends, HTTPException, Path
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",

View File

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