fix: delete all states of project

GKcodebase · Apr 25, 2024 · 30e9828 · 30e9828
1 parent 6d8975b
commit 30e9828
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/state.py b/src/state.py
@@ -57,7 +57,8 @@ def delete_state(self, project: str):
         with Session(self.engine) as session:
             agent_state = session.query(AgentStateModel).filter(AgentStateModel.project == project).all()
             if agent_state:
-                session.delete(agent_state)
+                for state in agent_state:
+                    session.delete(state)
                 session.commit()
 
     def add_to_current_state(self, project: str, state: dict):