Josedcape commited on
Commit
5a275bc
·
verified ·
1 Parent(s): 545e280

Update src/utils/agent_state.py

Browse files
Files changed (1) hide show
  1. src/utils/agent_state.py +30 -30
src/utils/agent_state.py CHANGED
@@ -1,30 +1,30 @@
1
- import asyncio
2
-
3
- class AgentState:
4
- _instance = None
5
-
6
- def __init__(self):
7
- if not hasattr(self, '_stop_requested'):
8
- self._stop_requested = asyncio.Event()
9
- self.last_valid_state = None # store the last valid browser state
10
-
11
- def __new__(cls):
12
- if cls._instance is None:
13
- cls._instance = super(AgentState, cls).__new__(cls)
14
- return cls._instance
15
-
16
- def request_stop(self):
17
- self._stop_requested.set()
18
-
19
- def clear_stop(self):
20
- self._stop_requested.clear()
21
- self.last_valid_state = None
22
-
23
- def is_stop_requested(self):
24
- return self._stop_requested.is_set()
25
-
26
- def set_last_valid_state(self, state):
27
- self.last_valid_state = state
28
-
29
- def get_last_valid_state(self):
30
- return self.last_valid_state
 
1
+ import asyncio
2
+
3
+ class AgentState:
4
+ _instance = None
5
+
6
+ def __init__(self):
7
+ if not hasattr(self, '_stop_requested'):
8
+ self._stop_requested = asyncio.Event()
9
+ self.last_valid_state = None
10
+
11
+ def __new__(cls):
12
+ if cls._instance is None:
13
+ cls._instance = super(AgentState, cls).__new__(cls)
14
+ return cls._instance
15
+
16
+ def request_stop(self):
17
+ self._stop_requested.set()
18
+
19
+ def clear_stop(self):
20
+ self._stop_requested.clear()
21
+ self.last_valid_state = None
22
+
23
+ def is_stop_requested(self):
24
+ return self._stop_requested.is_set()
25
+
26
+ def set_last_valid_state(self, state):
27
+ self.last_valid_state = state
28
+
29
+ def get_last_valid_state(self):
30
+ return self.last_valid_state