mirror of
https://github.com/weyne85/PentestGPT.git
synced 2025-10-29 16:58:59 +00:00
24 lines
752 B
Python
24 lines
752 B
Python
from chatgpt_wrapper import ChatGPT
|
|
from llm_handle.parser import extract_cmd
|
|
from task_handle.cmd_execution import execute_cmd
|
|
import os
|
|
|
|
|
|
if __name__ == "__main__":
|
|
bot = ChatGPT()
|
|
conversations = bot.get_history()
|
|
print(conversations)
|
|
# structure of conversation:
|
|
# {conversation_id (str): {'id': conversation_id, 'title': conversation_title, 'create_time': conversation_create_time'}}
|
|
|
|
## select a past conversation
|
|
selected_id = list(conversations.keys())[0]
|
|
result = bot.get_conversation(selected_id)
|
|
## Get the conversation history
|
|
# print(result)
|
|
|
|
## Try to ask a question in this conversation
|
|
question = "What is the meaning of life?"
|
|
response = bot.ask("Hello, world!")
|
|
print(response)
|