mirror of
https://github.com/eigent-ai/eigent.git
synced 2026-04-30 20:50:02 +00:00
chore: fix pre commit format and pipeline issue (#1144)
This commit is contained in:
parent
2256497dff
commit
893f51fc82
970 changed files with 11235 additions and 9533 deletions
|
|
@ -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")
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue