File tree Expand file tree Collapse file tree 1 file changed +6
-6
lines changed Expand file tree Collapse file tree 1 file changed +6
-6
lines changed Original file line number Diff line number Diff line change 11
22from typing import Tuple
3- from hackingBuddyGPT .usecases .minimal .agent import MinimalLinuxPrivesc , MinimalLinuxPrivescUseCase
4- from hackingBuddyGPT .usecases .minimal .agent_with_state import MinimalLinuxTemplatedPrivesc , MinimalLinuxTemplatedPrivescUseCase
3+ from hackingBuddyGPT .usecases .examples .agent import ExPrivEscLinux , ExPrivEscLinuxUseCase
4+ from hackingBuddyGPT .usecases .examples .agent_with_state import ExPrivEscLinuxTemplated , ExPrivEscLinuxTemplatedUseCase
55from hackingBuddyGPT .usecases .privesc .linux import LinuxPrivesc , LinuxPrivescUseCase
66from hackingBuddyGPT .utils .console .console import Console
77from hackingBuddyGPT .utils .db_storage .db_storage import DbStorage
@@ -101,8 +101,8 @@ def test_minimal_agent():
101101
102102 log_db .init ()
103103
104- priv_esc = MinimalLinuxPrivescUseCase (
105- agent = MinimalLinuxPrivesc (
104+ priv_esc = ExPrivEscLinuxUseCase (
105+ agent = ExPrivEscLinux (
106106 conn = conn ,
107107 llm = llm
108108 ),
@@ -125,8 +125,8 @@ def test_minimal_agent_state():
125125
126126 log_db .init ()
127127
128- priv_esc = MinimalLinuxTemplatedPrivescUseCase (
129- agent = MinimalLinuxTemplatedPrivesc (
128+ priv_esc = ExPrivEscLinuxTemplatedUseCase (
129+ agent = ExPrivEscLinuxTemplated (
130130 conn = conn ,
131131 llm = llm ,
132132 ),
You can’t perform that action at this time.
0 commit comments