Add adaptive learning and model rollback
This commit is contained in:
@@ -60,6 +60,10 @@ class SafetyProfileRequest(BaseModel):
|
||||
safety: SafetyProfile
|
||||
|
||||
|
||||
class ModelRollbackRequest(BaseModel):
|
||||
version_id: str = Field(min_length=1, max_length=120)
|
||||
|
||||
|
||||
class ActuatorSuggestion(BaseModel):
|
||||
entity_id: str
|
||||
domain: str
|
||||
@@ -408,6 +412,20 @@ def set_safety_profile(
|
||||
raise HTTPException(status_code=404, detail=str(exc)) from exc
|
||||
|
||||
|
||||
@router.post("/{actuator_entity_id}/model/rollback", response_model=ActuatorRecord)
|
||||
def rollback_model(
|
||||
actuator_entity_id: str,
|
||||
payload: ModelRollbackRequest,
|
||||
request: Request,
|
||||
) -> ActuatorRecord:
|
||||
try:
|
||||
return _behavior(request).rollback_model(actuator_entity_id, version_id=payload.version_id)
|
||||
except KeyError as exc:
|
||||
raise HTTPException(status_code=404, detail=str(exc)) from exc
|
||||
except ValueError as exc:
|
||||
raise HTTPException(status_code=422, detail=str(exc)) from exc
|
||||
|
||||
|
||||
@router.post("/{actuator_entity_id}/activation", response_model=ActuatorRecord)
|
||||
def set_activation(
|
||||
actuator_entity_id: str,
|
||||
|
||||
Reference in New Issue
Block a user