chore: fix pre commit format and pipeline issue (#1144)

This commit is contained in:
Wendong-Fan 2026-02-04 00:06:29 +00:00 committed by GitHub
parent 2256497dff
commit 893f51fc82
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
970 changed files with 11235 additions and 9533 deletions

View file

@ -12,19 +12,24 @@
# limitations under the License.
# ========= Copyright 2025-2026 @ Eigent.ai All Rights Reserved. =========
from typing import List, Optional
from fastapi import Depends, HTTPException, Query, Response, APIRouter
import logging
from fastapi import APIRouter, Depends, HTTPException, Query, Response
from fastapi_babel import _
from fastapi_pagination import Page
from fastapi_pagination.ext.sqlmodel import paginate
from sqlalchemy import update
from sqlmodel import Session, select, col
from sqlalchemy.exc import SQLAlchemyError
from sqlmodel import Session, col, select
from app.component.database import session
from app.component.auth import Auth, auth_must
from app.model.provider.provider import Provider, ProviderIn, ProviderOut, ProviderPreferIn
import logging
from app.component.database import session
from app.model.provider.provider import (
Provider,
ProviderIn,
ProviderOut,
ProviderPreferIn,
)
logger = logging.getLogger("server_provider_controller")
@ -34,7 +39,7 @@ router = APIRouter(tags=["Provider Management"])
@router.get("/providers", name="list providers", response_model=Page[ProviderOut])
async def gets(
keyword: str | None = None,
prefer: Optional[bool] = Query(None, description="Filter by prefer status"),
prefer: bool | None = Query(None, description="Filter by prefer status"),
session: Session = Depends(session),
auth: Auth = Depends(auth_must),
) -> Page[ProviderOut]:
@ -70,7 +75,9 @@ async def post(data: ProviderIn, session: Session = Depends(session), auth: Auth
try:
model = Provider(**data.model_dump(), user_id=user_id)
model.save(session)
logger.info("Provider created", extra={"user_id": user_id, "provider_id": model.id, "provider_name": data.provider_name})
logger.info(
"Provider created", extra={"user_id": user_id, "provider_id": model.id, "provider_name": data.provider_name}
)
return model
except Exception as e:
logger.error("Provider creation failed", extra={"user_id": user_id, "error": str(e)}, exc_info=True)
@ -87,7 +94,7 @@ async def put(id: int, data: ProviderIn, session: Session = Depends(session), au
if not model:
logger.warning("Provider not found for update", extra={"user_id": user_id, "provider_id": id})
raise HTTPException(status_code=404, detail=_("Provider not found"))
try:
model.model_type = data.model_type
model.provider_name = data.provider_name
@ -97,10 +104,14 @@ async def put(id: int, data: ProviderIn, session: Session = Depends(session), au
model.is_vaild = data.is_vaild
model.save(session)
session.refresh(model)
logger.info("Provider updated", extra={"user_id": user_id, "provider_id": id, "provider_name": data.provider_name})
logger.info(
"Provider updated", extra={"user_id": user_id, "provider_id": id, "provider_name": data.provider_name}
)
return model
except Exception as e:
logger.error("Provider update failed", extra={"user_id": user_id, "provider_id": id, "error": str(e)}, exc_info=True)
logger.error(
"Provider update failed", extra={"user_id": user_id, "provider_id": id, "error": str(e)}, exc_info=True
)
raise HTTPException(status_code=500, detail="Internal server error")
@ -114,13 +125,15 @@ async def delete(id: int, session: Session = Depends(session), auth: Auth = Depe
if not model:
logger.warning("Provider not found for deletion", extra={"user_id": user_id, "provider_id": id})
raise HTTPException(status_code=404, detail=_("Provider not found"))
try:
model.delete(session)
logger.info("Provider deleted", extra={"user_id": user_id, "provider_id": id})
return Response(status_code=204)
except Exception as e:
logger.error("Provider deletion failed", extra={"user_id": user_id, "provider_id": id, "error": str(e)}, exc_info=True)
logger.error(
"Provider deletion failed", extra={"user_id": user_id, "provider_id": id, "error": str(e)}, exc_info=True
)
raise HTTPException(status_code=500, detail="Internal server error")
@ -129,7 +142,7 @@ async def set_prefer(data: ProviderPreferIn, session: Session = Depends(session)
"""Set preferred provider for user."""
user_id = auth.user.id
provider_id = data.provider_id
try:
# 1. Set all current user's providers prefer to false
session.exec(update(Provider).where(Provider.user_id == user_id, Provider.no_delete()).values(prefer=False))
@ -144,5 +157,9 @@ async def set_prefer(data: ProviderPreferIn, session: Session = Depends(session)
return {"success": True}
except SQLAlchemyError as e:
session.rollback()
logger.error("Failed to set preferred provider", extra={"user_id": user_id, "provider_id": provider_id, "error": str(e)}, exc_info=True)
raise HTTPException(status_code=500, detail="Internal server error")
logger.error(
"Failed to set preferred provider",
extra={"user_id": user_id, "provider_id": provider_id, "error": str(e)},
exc_info=True,
)
raise HTTPException(status_code=500, detail="Internal server error")