mirror of
https://github.com/eigent-ai/eigent.git
synced 2026-04-28 19:50:34 +00:00
update server logs
This commit is contained in:
Saedbhati
parent
6eed287dcb
commit
40779bf1e5
43 changed files with 2354 additions and 1808 deletions
|
|
@ -1,121 +1,172 @@
|
|||
from typing import List, Optional
|
||||
from fastapi import Depends, HTTPException, Query, Response, APIRouter
|
||||
from sqlmodel import Session, select, or_
|
||||
from app.component.database import session
|
||||
from app.component.auth import Auth, auth_must
|
||||
from fastapi_babel import _
|
||||
from app.model.config.config import Config, ConfigCreate, ConfigUpdate, ConfigInfo, ConfigOut
|
||||
|
||||
router = APIRouter(tags=["Config Management"])
|
||||
|
||||
|
||||
@router.get("/configs", name="list configs", response_model=list[ConfigOut])
|
||||
async def list_configs(
|
||||
config_group: Optional[str] = None, session: Session = Depends(session), auth: Auth = Depends(auth_must)
|
||||
):
|
||||
query = select(Config)
|
||||
user_id = auth.user.id
|
||||
if user_id is not None:
|
||||
query = query.where(Config.user_id == user_id)
|
||||
if config_group is not None:
|
||||
query = query.where(Config.config_group == config_group)
|
||||
configs = session.exec(query).all()
|
||||
return configs
|
||||
|
||||
|
||||
@router.get("/configs/{config_id}", name="get config", response_model=ConfigOut)
|
||||
async def get_config(
|
||||
config_id: int,
|
||||
session: Session = Depends(session),
|
||||
auth: Auth = Depends(auth_must),
|
||||
):
|
||||
query = select(Config).where(Config.user_id == auth.user.id)
|
||||
|
||||
if config_id is not None:
|
||||
query = query.where(Config.id == config_id)
|
||||
|
||||
config = session.exec(query).first()
|
||||
|
||||
if not config:
|
||||
raise HTTPException(status_code=404, detail=_("Configuration not found"))
|
||||
return config
|
||||
|
||||
|
||||
@router.post("/configs", name="create config", response_model=ConfigOut)
|
||||
async def create_config(config: ConfigCreate, session: Session = Depends(session), auth: Auth = Depends(auth_must)):
|
||||
if not ConfigInfo.is_valid_env_var(config.config_group, config.config_name):
|
||||
raise HTTPException(status_code=400, detail=_("Config Name is valid"))
|
||||
|
||||
# Check if configuration already exists
|
||||
existing_config = session.exec(
|
||||
select(Config).where(Config.user_id == auth.user.id, Config.config_name == config.config_name)
|
||||
).first()
|
||||
|
||||
if existing_config:
|
||||
raise HTTPException(status_code=400, detail=_("Configuration already exists for this user"))
|
||||
|
||||
db_config = Config(
|
||||
user_id=auth.user.id,
|
||||
config_name=config.config_name,
|
||||
config_value=config.config_value,
|
||||
config_group=config.config_group,
|
||||
)
|
||||
session.add(db_config)
|
||||
session.commit()
|
||||
session.refresh(db_config)
|
||||
return db_config
|
||||
|
||||
|
||||
@router.put("/configs/{config_id}", name="update config", response_model=ConfigOut)
|
||||
async def update_config(
|
||||
config_id: int, config_update: ConfigUpdate, session: Session = Depends(session), auth: Auth = Depends(auth_must)
|
||||
):
|
||||
db_config = session.exec(select(Config).where(Config.id == config_id, Config.user_id == auth.user.id)).first()
|
||||
|
||||
if not db_config:
|
||||
raise HTTPException(status_code=404, detail=_("Configuration not found"))
|
||||
|
||||
# Check if configuration group is valid
|
||||
if not ConfigInfo.is_valid_env_var(config_update.config_group, config_update.config_name):
|
||||
raise HTTPException(status_code=400, detail=_("Invalid configuration group"))
|
||||
|
||||
# Check for conflicts with other configurations
|
||||
existing_config = session.exec(
|
||||
select(Config).where(
|
||||
Config.user_id == auth.user.id,
|
||||
Config.config_name == config_update.config_name,
|
||||
Config.id != config_id,
|
||||
)
|
||||
).first()
|
||||
|
||||
if existing_config:
|
||||
raise HTTPException(status_code=400, detail=_("Configuration already exists for this user"))
|
||||
|
||||
db_config.config_name = config_update.config_name
|
||||
db_config.config_value = config_update.config_value
|
||||
|
||||
session.add(db_config)
|
||||
session.commit()
|
||||
session.refresh(db_config)
|
||||
return db_config
|
||||
|
||||
|
||||
@router.delete("/configs/{config_id}", name="delete config")
|
||||
async def delete_config(config_id: int, session: Session = Depends(session), auth: Auth = Depends(auth_must)):
|
||||
db_config = session.exec(select(Config).where(Config.id == config_id, Config.user_id == auth.user.id)).first()
|
||||
|
||||
if not db_config:
|
||||
raise HTTPException(status_code=404, detail=_("Configuration not found"))
|
||||
session.delete(db_config)
|
||||
session.commit()
|
||||
return Response(status_code=204)
|
||||
|
||||
|
||||
@router.get("/config/info", name="get config info")
|
||||
async def get_config_info(
|
||||
show_all: bool = Query(False, description="Show all config info, including those with empty env_vars"),
|
||||
):
|
||||
configs = ConfigInfo.getinfo()
|
||||
if show_all:
|
||||
return configs
|
||||
return {k: v for k, v in configs.items() if v.get("env_vars") and len(v["env_vars"]) > 0}
|
||||
from typing import List, Optional
|
||||
from fastapi import Depends, HTTPException, Query, Response, APIRouter
|
||||
from sqlmodel import Session, select, or_
|
||||
from app.component.database import session
|
||||
from app.component.auth import Auth, auth_must
|
||||
from fastapi_babel import _
|
||||
from app.model.config.config import Config, ConfigCreate, ConfigUpdate, ConfigInfo, ConfigOut
|
||||
from utils import traceroot_wrapper as traceroot
|
||||
|
||||
logger = traceroot.get_logger("server_config_controller")
|
||||
|
||||
router = APIRouter(tags=["Config Management"])
|
||||
|
||||
|
||||
@router.get("/configs", name="list configs", response_model=list[ConfigOut])
|
||||
@traceroot.trace()
|
||||
async def list_configs(
|
||||
config_group: Optional[str] = None, session: Session = Depends(session), auth: Auth = Depends(auth_must)
|
||||
):
|
||||
"""List user's configurations with optional group filtering."""
|
||||
user_id = auth.user.id
|
||||
query = select(Config).where(Config.user_id == user_id)
|
||||
|
||||
if config_group is not None:
|
||||
query = query.where(Config.config_group == config_group)
|
||||
|
||||
configs = session.exec(query).all()
|
||||
logger.debug("Configs listed", extra={"user_id": user_id, "config_group": config_group, "count": len(configs)})
|
||||
return configs
|
||||
|
||||
|
||||
@router.get("/configs/{config_id}", name="get config", response_model=ConfigOut)
|
||||
@traceroot.trace()
|
||||
async def get_config(
|
||||
config_id: int,
|
||||
session: Session = Depends(session),
|
||||
auth: Auth = Depends(auth_must),
|
||||
):
|
||||
query = select(Config).where(Config.user_id == auth.user.id)
|
||||
|
||||
if config_id is not None:
|
||||
query = query.where(Config.id == config_id)
|
||||
|
||||
config = session.exec(query).first()
|
||||
|
||||
if not config:
|
||||
logger.warning("Config not found")
|
||||
raise HTTPException(status_code=404, detail=_("Configuration not found"))
|
||||
|
||||
logger.debug("Config retrieved")
|
||||
return config
|
||||
|
||||
|
||||
@router.post("/configs", name="create config", response_model=ConfigOut)
|
||||
@traceroot.trace()
|
||||
async def create_config(config: ConfigCreate, session: Session = Depends(session), auth: Auth = Depends(auth_must)):
|
||||
"""Create new configuration."""
|
||||
user_id = auth.user.id
|
||||
|
||||
if not ConfigInfo.is_valid_env_var(config.config_group, config.config_name):
|
||||
logger.warning("Config validation failed", extra={"user_id": user_id, "config_group": config.config_group, "config_name": config.config_name})
|
||||
raise HTTPException(status_code=400, detail=_("Invalid config name or group"))
|
||||
|
||||
# Check if configuration already exists
|
||||
existing_config = session.exec(
|
||||
select(Config).where(Config.user_id == user_id, Config.config_name == config.config_name)
|
||||
).first()
|
||||
|
||||
if existing_config:
|
||||
logger.warning("Config creation failed: already exists", extra={"user_id": user_id, "config_name": config.config_name})
|
||||
raise HTTPException(status_code=400, detail=_("Configuration already exists for this user"))
|
||||
|
||||
try:
|
||||
db_config = Config(
|
||||
user_id=user_id,
|
||||
config_name=config.config_name,
|
||||
config_value=config.config_value,
|
||||
config_group=config.config_group,
|
||||
)
|
||||
session.add(db_config)
|
||||
session.commit()
|
||||
session.refresh(db_config)
|
||||
logger.info("Config created", extra={"user_id": user_id, "config_id": db_config.id, "config_group": config.config_group, "config_name": config.config_name})
|
||||
return db_config
|
||||
except Exception as e:
|
||||
session.rollback()
|
||||
logger.error("Config creation failed", extra={"user_id": user_id, "config_name": config.config_name, "error": str(e)}, exc_info=True)
|
||||
raise
|
||||
|
||||
|
||||
@router.put("/configs/{config_id}", name="update config", response_model=ConfigOut)
|
||||
@traceroot.trace()
|
||||
async def update_config(
|
||||
config_id: int, config_update: ConfigUpdate, session: Session = Depends(session), auth: Auth = Depends(auth_must)
|
||||
):
|
||||
"""Update configuration."""
|
||||
user_id = auth.user.id
|
||||
db_config = session.exec(select(Config).where(Config.id == config_id, Config.user_id == user_id)).first()
|
||||
|
||||
if not db_config:
|
||||
logger.warning("Config not found for update", extra={"user_id": user_id, "config_id": config_id})
|
||||
raise HTTPException(status_code=404, detail=_("Configuration not found"))
|
||||
|
||||
# Check if configuration group is valid
|
||||
if not ConfigInfo.is_valid_env_var(config_update.config_group, config_update.config_name):
|
||||
logger.warning("Config update validation failed", extra={"user_id": user_id, "config_id": config_id, "config_group": config_update.config_group})
|
||||
raise HTTPException(status_code=400, detail=_("Invalid configuration group"))
|
||||
|
||||
# Check for conflicts with other configurations
|
||||
existing_config = session.exec(
|
||||
select(Config).where(
|
||||
Config.user_id == user_id,
|
||||
Config.config_name == config_update.config_name,
|
||||
Config.id != config_id,
|
||||
)
|
||||
).first()
|
||||
|
||||
if existing_config:
|
||||
logger.warning("Config update failed: duplicate name", extra={"user_id": user_id, "config_id": config_id, "config_name": config_update.config_name})
|
||||
raise HTTPException(status_code=400, detail=_("Configuration already exists for this user"))
|
||||
|
||||
try:
|
||||
db_config.config_name = config_update.config_name
|
||||
db_config.config_value = config_update.config_value
|
||||
db_config.config_group = config_update.config_group
|
||||
session.add(db_config)
|
||||
session.commit()
|
||||
session.refresh(db_config)
|
||||
logger.info("Config updated", extra={"user_id": user_id, "config_id": config_id, "config_group": config_update.config_group})
|
||||
return db_config
|
||||
except Exception as e:
|
||||
session.rollback()
|
||||
logger.error("Config update failed", extra={"user_id": user_id, "config_id": config_id, "error": str(e)}, exc_info=True)
|
||||
raise
|
||||
|
||||
|
||||
@router.delete("/configs/{config_id}", name="delete config")
|
||||
@traceroot.trace()
|
||||
async def delete_config(config_id: int, session: Session = Depends(session), auth: Auth = Depends(auth_must)):
|
||||
"""Delete configuration."""
|
||||
user_id = auth.user.id
|
||||
db_config = session.exec(select(Config).where(Config.id == config_id, Config.user_id == user_id)).first()
|
||||
|
||||
if not db_config:
|
||||
logger.warning("Config not found for deletion", extra={"user_id": user_id, "config_id": config_id})
|
||||
raise HTTPException(status_code=404, detail=_("Configuration not found"))
|
||||
|
||||
try:
|
||||
session.delete(db_config)
|
||||
session.commit()
|
||||
logger.info("Config deleted", extra={"user_id": user_id, "config_id": config_id, "config_name": db_config.config_name})
|
||||
return Response(status_code=204)
|
||||
except Exception as e:
|
||||
session.rollback()
|
||||
logger.error("Config deletion failed", extra={"user_id": user_id, "config_id": config_id, "error": str(e)}, exc_info=True)
|
||||
raise
|
||||
|
||||
|
||||
@router.get("/config/info", name="get config info")
|
||||
@traceroot.trace()
|
||||
async def get_config_info(
|
||||
show_all: bool = Query(False, description="Show all config info, including those with empty env_vars"),
|
||||
):
|
||||
"""Get available configuration templates and info."""
|
||||
configs = ConfigInfo.getinfo()
|
||||
if show_all:
|
||||
logger.debug("Config info retrieved", extra={"show_all": True, "count": len(configs)})
|
||||
return configs
|
||||
|
||||
filtered = {k: v for k, v in configs.items() if v.get("env_vars") and len(v["env_vars"]) > 0}
|
||||
logger.debug("Config info retrieved", extra={"show_all": False, "total_count": len(configs), "filtered_count": len(filtered)})
|
||||
return filtered
|
||||
Loading…
Add table
Add a link
Reference in a new issue