PentestGPT/test_connection.py
Grey_D 5e957b5b1e fix: 🐛 login issue
Add configurable user agent

 Closes: #26, #20, #13
2023-04-27 11:00:45 +08:00

29 lines
914 B
Python

import loguru
import sys
from utils.chatgpt import ChatGPT
from config.chatgpt_config import ChatGPTConfig
import requests
logger = loguru.logger
if __name__ == "__main__":
chatgpt_config = ChatGPTConfig()
try:
chatgpt = ChatGPT(chatgpt_config)
conversations = chatgpt.get_conversation_history()
print(conversations)
if conversations != None:
# print(text, conversation_id)
print(
"Now you're connected. To start PentestGPT, please use <python3 main.py>"
)
else:
print(
"The cookie is not properly configured. Please follow README to update cookie in config/chatgpt_config.py"
)
except requests.exceptions.JSONDecodeError:
print(
"The cookie is not properly configured. Please follow README to update cookie in config/chatgpt_config.py"
)