weyne/PentestGPT
1
0
Fork 0
mirror of https://github.com/weyne85/PentestGPT.git synced 2025-10-29 16:58:59 +00:00
Code Issues Packages Projects Releases Wiki Activity

docs: ✏️ add HTB pentest record

Browse Source
This commit is contained in:
Grey_D
2023-04-23 15:22:02 +08:00
parent eafa0d5ba8
commit 1e317246db
3 changed files with 4 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
utils/pentest_gpt.py
View File

@@ -42,10 +42,10 @@ class pentestGPT:
"default": "The user did not specify the input source. You need to summarize based on the contents.\n",
}
def __init__(self):
def __init__(self, reasoning_model="gpt-4"):
self.log_dir = "logs"
self.chatGPTAgent = ChatGPT(ChatGPTConfig())
self.chatGPT4Agent = ChatGPT(ChatGPTConfig(model="gpt-4"))
self.chatGPT4Agent = ChatGPT(ChatGPTConfig(model=reasoning_model))
self.prompts = PentestGPTPrompt
self.console = Console()
self.spinner = Spinner("line", "Processing")