p
36 lines
1.4 KiB
Python
36 lines
1.4 KiB
Python
from sqlmodel import Session, select
|
|
from app.models.models import ModelConfig
|
|
from app.schemas.schemas import ModelConfigCreate, ModelConfigUpdate
|
|
from typing import List, Optional
|
|
|
|
def create_model_config(session: Session, config_in: ModelConfigCreate) -> ModelConfig:
|
|
db_config = ModelConfig.model_validate(config_in)
|
|
session.add(db_config)
|
|
session.commit()
|
|
session.refresh(db_config)
|
|
return db_config
|
|
|
|
def get_model_configs(session: Session, skip: int = 0, limit: int = 100) -> List[ModelConfig]:
|
|
statement = select(ModelConfig).offset(skip).limit(limit)
|
|
return session.exec(statement).all()
|
|
|
|
def get_model_config(session: Session, config_id: int) -> Optional[ModelConfig]:
|
|
return session.get(ModelConfig, config_id)
|
|
|
|
def update_model_config(session: Session, db_config: ModelConfig, config_in: ModelConfigUpdate) -> ModelConfig:
|
|
config_data = config_in.model_dump(exclude_unset=True)
|
|
for key, value in config_data.items():
|
|
setattr(db_config, key, value)
|
|
session.add(db_config)
|
|
session.commit()
|
|
session.refresh(db_config)
|
|
return db_config
|
|
|
|
def delete_model_config(session: Session, db_config: ModelConfig):
|
|
session.delete(db_config)
|
|
session.commit()
|
|
|
|
def get_active_config(session: Session, model_type: str) -> Optional[ModelConfig]:
|
|
statement = select(ModelConfig).where(ModelConfig.model_type == model_type, ModelConfig.is_active == True)
|
|
return session.exec(statement).first()
|