mirror of
https://github.com/weyne85/PentestGPT.git
synced 2025-10-29 16:58:59 +00:00
37 lines
1.1 KiB
Python
37 lines
1.1 KiB
Python
import loguru
|
|
|
|
from config.chatgpt_config import ChatGPTConfig
|
|
from utils.chatgpt import ChatGPT
|
|
from rich.prompt import Prompt
|
|
from utils.pentest_gpt import pentestGPT
|
|
|
|
logger = loguru.logger
|
|
|
|
if __name__ == "__main__":
|
|
pentestGPTHandler = pentestGPT()
|
|
# initialize
|
|
pentestGPTHandler.initialize()
|
|
|
|
# start the input handler
|
|
text_input = pentestGPTHandler.input_handler()
|
|
|
|
# the previous example
|
|
"""
|
|
chatGPTAgent = ChatGPT(ChatGPTConfig())
|
|
# request user's input to create a new chat.
|
|
question = Prompt.ask("What do you want to ask ChatGPT?")
|
|
# the title of this conversation will be new-chat. We can delete it later.
|
|
text, conversation_id = chatGPTAgent.send_new_message(question)
|
|
print(text, conversation_id)
|
|
# get history id
|
|
history = chatGPTAgent.get_conversation_history()
|
|
print(history)
|
|
for uuid in history:
|
|
print(uuid)
|
|
if history[uuid].lower() == "new chat":
|
|
result = chatGPTAgent.delete_conversation(uuid)
|
|
print(result)
|
|
history = chatGPTAgent.get_conversation_history()
|
|
print(history)
|
|
"""
|