Merge pull request #18 from jfelten/issue-17

Issue 17 - add flag to disable danger check

Author: handrew

browserpilot/agents/compilers/instruction_compiler.py

@@ -87,6 +87,7 @@ def __init__(
         base_prompt=BASE_PROMPT,
         model="gpt-4o-mini",
         use_compiled=True,
+        disable_danger_check=False,
     ):
         """Initialize the compiler. The compiler handles the sequencing of
         each set of instructions which are injected into the base prompt.
@@ -125,6 +126,7 @@ def __init__(
         self.functions = {}  # Set in _parse_instructions_into_queue.
         self.finished_instructions = []
         self.history = []  # Keep track of the history of actions.
+        self.disable_danger_check = disable_danger_check
 
         # Set the instructions.
         self.instructions = instructions  # Overriden in set_instructions.
@@ -337,7 +339,8 @@ def get_action_output(self, instructions):
         prompt = self.base_prompt.format(instructions=instructions)
         completion = self.get_completion(prompt).strip()
         action_output = completion.strip()
-        lines = [line for line in action_output.split("\n") if not line.startswith("import ")]
+        print(self.disable_danger_check)
+        lines = [line for line in action_output.split("\n") if (self.disable_danger_check is False or not line.startswith("import "))]
         action_output = "\n".join(lines)
         return {
             "instruction": instructions,

browserpilot/agents/gpt_selenium_agent.py

@@ -56,7 +56,8 @@ def __init__(
         debug_html_folder="",
         instruction_output_file=None,
         close_after_completion=True,
-        remote_url=None, 
+        remote_url=None,
+        disable_danger_check=False,
     ):
         """Initialize the agent.
 
@@ -104,11 +105,13 @@ def __init__(
         self.memory_folder = memory_folder
         self.close_after_completion = close_after_completion
         self.remote_url = remote_url
+        self.disable_danger_check = disable_danger_check
 
         """Fire up the compiler."""
         self.instruction_compiler = InstructionCompiler(
             instructions=instructions,
             model=self.model_for_instructions,
+            disable_danger_check = self.disable_danger_check,
         )
 
         """Set up the memory."""
@@ -148,7 +151,10 @@ def __init__(
 
     def _check_danger(self, action_str):
         """Check that the action is not dangerous. If so, just quit."""
-        if self._is_potentially_dangerous(action_str):
+        if (self.disable_danger_check is True):
+            logger.warning("Code Danger checking is disabled")
+            return
+        if (self._is_potentially_dangerous(action_str)):
             logger.warning("Action is potentially dangerous. Exiting.")
             logger.warning(f"Action: {action_str}")
             sys.exit(1)

examples.py

@@ -27,6 +27,7 @@ def selenium(instructions, chromedriver_path, model, memory_folder, debug, outpu
             memory_folder=memory_folder,
             debug=debug,
             retry=True,
+            disable_danger_check=True
         )
         agent.run()
 

pyproject.toml

@@ -1,21 +1,21 @@
 [tool.poetry]
 name = "browserpilot"
-version = "0.2.52"
+version = "0.2.53"
 description = "Natural language browser automation"
 authors = ["Andrew Han <handrew11@gmail.com>"]
 readme = "README.md"
 
 [tool.poetry.dependencies]
-click = "^8.1.3"
+click = "^8.1.7"
 python = "^3.10"
-openai = "^1.13.3"
-selenium = "^4.8.2"
-tqdm = "^4.66.1"
+openai = "^1.55.0"
+selenium = "^4.26.1"
+tqdm = "^4.67.0"
 beautifulsoup4 = "^4.12.3"
-pyyaml = "^6.0"
-langchain = "^0.1.11"
-llama-index = "^0.10.16"
-langchain_openai = "^0.1.1"
+pyyaml = "^6.0.2"
+langchain = "^0.1.20"
+llama-index = "^0.10.68"
+langchain_openai = "^0.1.7"
 
 
 [build-system]