Skip to content

Commit ea56264

Browse files
committed
make lse-based example work
1 parent 38bfbc0 commit ea56264

File tree

1 file changed

+11
-7
lines changed
  • src/hackingBuddyGPT/usecases/privesc

1 file changed

+11
-7
lines changed

src/hackingBuddyGPT/usecases/privesc/linux.py

Lines changed: 11 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -62,7 +62,7 @@ class LinuxPrivescWithLSEUseCase(UseCase):
6262
_got_root: bool = False
6363

6464
# use either an use-case or an agent to perform the privesc
65-
_use_use_case: bool = False
65+
use_use_case: bool = False
6666

6767
def init(self):
6868
super().init()
@@ -104,13 +104,17 @@ def run(self):
104104
def run_using_usecases(self, hint, turns_per_hint):
105105
# TODO: init usecase
106106
linux_privesc = LinuxPrivescUseCase(
107-
conn = self.conn,
108-
llm = self.llm,
109-
hint = hint,
107+
agent = LinuxPrivesc(
108+
conn = self.conn,
109+
enable_explanation = self.enable_explanation,
110+
enable_update_state = self.enable_update_state,
111+
disable_history = self.disable_history,
112+
llm = self.llm,
113+
hint = hint
114+
),
110115
max_turns = turns_per_hint,
111-
enable_explanation = self.enable_explanation,
112-
enable_update_state = self.enable_update_state,
113-
disable_history = self.disable_history
116+
log_db = self.log_db,
117+
console = self.console
114118
)
115119
linux_privesc.init()
116120
return linux_privesc.run()

0 commit comments

Comments
 (0)