mirror of
https://github.com/agent0ai/agent-zero.git
synced 2026-04-30 20:50:21 +00:00
browser-use intervention hack
This commit is contained in:
parent
04fdc67746
commit
cd82c448ae
11 changed files with 70 additions and 5 deletions
14
python/extensions/message_loop_start/_10_iteration_no.py
Normal file
14
python/extensions/message_loop_start/_10_iteration_no.py
Normal file
|
|
@ -0,0 +1,14 @@
|
|||
from python.helpers.extension import Extension
|
||||
from agent import Agent, LoopData
|
||||
|
||||
DATA_NAME_ITER_NO = "iteration_no"
|
||||
|
||||
class IterationNo(Extension):
|
||||
async def execute(self, loop_data: LoopData = LoopData(), **kwargs):
|
||||
# total iteration number
|
||||
no = self.agent.get_data(DATA_NAME_ITER_NO) or 0
|
||||
self.agent.set_data(DATA_NAME_ITER_NO, no + 1)
|
||||
|
||||
|
||||
def get_iter_no(agent: Agent) -> int:
|
||||
return agent.get_data(DATA_NAME_ITER_NO) or 0
|
||||
Loading…
Add table
Add a link
Reference in a new issue