Skip to content
Prev Previous commit
Merge branch 'main' of github.com:ipa-lab/hackingBuddyGPT into other_…
…llms
  • Loading branch information
andreashappe committed Apr 24, 2024
commit b45bc77ef10d615c15dcfb603650d1ea30eed168
14 changes: 0 additions & 14 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -100,20 +100,6 @@ This work is partially based upon our empiric research into [how hackers work](h
The following would create a new (minimal) linux privilege-escalation agent. Through using our infrastructure, this already uses configurable LLM-connections (e.g., for testing OpenAI or locally run LLMs), logs trace data to a local sqlite database for each run, implements a round limit (after which the agent will stop if root has not been achieved until then) and is able to connect to a linux target over SSH for fully-autonomous command execution (as well as password guessing).

~~~ python
import abc
import pathlib
from dataclasses import dataclass, field
from typing import Dict

from mako.template import Template
from rich.panel import Panel

from capabilities import Capability, SSHRunCommand, SSHTestCredential
from utils import SSHConnection, llm_util, ui
from usecases.usecase import use_case, UseCase
from usecases.usecase.roundbased import RoundBasedUseCase
from utils.cli_history import SlidingCliHistory

template_dir = pathlib.Path(__file__).parent
template_next_cmd = Template(filename=str(template_dir / "next_cmd.txt"))

Expand Down
You are viewing a condensed version of this merge commit. You can view the full changes here.