GreyDGL / PentestGPT

A GPT-empowered penetration testing tool
MIT License
7.26k stars 880 forks source link

Script Update #151

Open CIPHER0930 opened 1 year ago

CIPHER0930 commented 1 year ago

import loguru import sys import argparse

from pentestgpt.utils.pentest_gpt import pentestGPT

def main(): """main function""" parser = argparse.ArgumentParser(description="PentestGPT")

# parser arguments
# 0. log directory
parser.add_argument("--log_dir", type=str, default="logs")
# 1. Reasoning Model
parser.add_argument("--reasoning_model", type=str, default="gpt-3")
# 2. Parsing Model
parser.add_argument("--parsing_model", type=str, default="gpt-3.5-base")

# Depracted: set to False only for testing if using cookie
parser.add_argument("--useAPI", action="store_true", default=True)

args = parser.parse_args()

pentestGPTHandler = pentestGPT(
    reasoning_model=args.reasoning_model,
    parsing_model=args.parsing_model,
    useAPI=args.useAPI,
    log_dir=args.log_dir,
)

pentestGPTHandler.main()

if name == "main": main()

GreyDGL commented 1 year ago

I don't see the issue here. May you help to elaborate?

CIPHER0930 commented 1 year ago

I don't see the issue here. May you help to elaborate?

I'd love to paticipate on this PentestGPT