add reset observer cruise functions (#1537)
This commit is contained in:
@@ -1744,6 +1744,19 @@ class AgentDB:
|
|||||||
await session.execute(stmt)
|
await session.execute(stmt)
|
||||||
await session.commit()
|
await session.commit()
|
||||||
|
|
||||||
|
async def delete_observer_cruise_artifacts(
|
||||||
|
self, observer_cruise_id: str, organization_id: str | None = None
|
||||||
|
) -> None:
|
||||||
|
async with self.Session() as session:
|
||||||
|
stmt = delete(ArtifactModel).where(
|
||||||
|
and_(
|
||||||
|
ArtifactModel.observer_cruise_id == observer_cruise_id,
|
||||||
|
ArtifactModel.organization_id == organization_id,
|
||||||
|
)
|
||||||
|
)
|
||||||
|
await session.execute(stmt)
|
||||||
|
await session.commit()
|
||||||
|
|
||||||
async def delete_task_steps(self, organization_id: str, task_id: str) -> None:
|
async def delete_task_steps(self, organization_id: str, task_id: str) -> None:
|
||||||
async with self.Session() as session:
|
async with self.Session() as session:
|
||||||
# delete artifacts by filtering organization_id and task_id
|
# delete artifacts by filtering organization_id and task_id
|
||||||
@@ -1933,6 +1946,19 @@ class AgentDB:
|
|||||||
return ObserverCruise.model_validate(observer_cruise)
|
return ObserverCruise.model_validate(observer_cruise)
|
||||||
return None
|
return None
|
||||||
|
|
||||||
|
async def delete_observer_thoughts_for_cruise(
|
||||||
|
self, observer_cruise_id: str, organization_id: str | None = None
|
||||||
|
) -> None:
|
||||||
|
async with self.Session() as session:
|
||||||
|
stmt = delete(ObserverThoughtModel).where(
|
||||||
|
and_(
|
||||||
|
ObserverThoughtModel.observer_cruise_id == observer_cruise_id,
|
||||||
|
ObserverThoughtModel.organization_id == organization_id,
|
||||||
|
)
|
||||||
|
)
|
||||||
|
await session.execute(stmt)
|
||||||
|
await session.commit()
|
||||||
|
|
||||||
async def get_observer_cruise_by_workflow_run_id(
|
async def get_observer_cruise_by_workflow_run_id(
|
||||||
self,
|
self,
|
||||||
workflow_run_id: str,
|
workflow_run_id: str,
|
||||||
|
|||||||
Reference in New Issue
Block a user