(self, project: str, is_completed: bool)
| 130 | return None |
| 131 | |
| 132 | def set_agent_completed(self, project: str, is_completed: bool): |
| 133 | with Session(self.engine) as session: |
| 134 | agent_state = session.query(AgentStateModel).filter(AgentStateModel.project == project).first() |
| 135 | if agent_state: |
| 136 | state_stack = json.loads(agent_state.state_stack_json) |
| 137 | state_stack[-1]["internal_monologue"] = "Agent has completed the task." |
| 138 | state_stack[-1]["completed"] = is_completed |
| 139 | agent_state.state_stack_json = json.dumps(state_stack) |
| 140 | session.commit() |
| 141 | else: |
| 142 | state_stack = [self.new_state()] |
| 143 | state_stack[-1]["completed"] = is_completed |
| 144 | agent_state = AgentStateModel(project=project, state_stack_json=json.dumps(state_stack)) |
| 145 | session.add(agent_state) |
| 146 | session.commit() |
| 147 | emit_agent("agent-state", state_stack) |
| 148 | |
| 149 | def is_agent_completed(self, project: str): |
| 150 | with Session(self.engine) as session: |
no test coverage detected