mirror of
https://github.com/weyne85/PentestGPT.git
synced 2025-10-29 16:58:59 +00:00
feat: 🎸 Update playwright for interface consistency
Now the playwright version has the same interface as the API version.
This commit is contained in:
67
utils/chatgpt_browser.py
Normal file
67
utils/chatgpt_browser.py
Normal file
@@ -0,0 +1,67 @@
|
||||
# -*- coding: utf-8 -*-
|
||||
|
||||
import json
|
||||
import re
|
||||
import time
|
||||
from uuid import uuid1
|
||||
import datetime
|
||||
|
||||
import loguru
|
||||
import requests
|
||||
|
||||
from chatgpt_wrapper import ChatGPT
|
||||
from chatgpt_wrapper.config import Config
|
||||
|
||||
logger = loguru.logger
|
||||
|
||||
|
||||
class ChatGPTBrowser:
|
||||
"""
|
||||
The ChatGPT Wrapper based on browser (playwright).
|
||||
It keeps the same interface as ChatGPT.
|
||||
"""
|
||||
def __init__(self, model=None):
|
||||
config = Config()
|
||||
if model is not None:
|
||||
config.set('chat.model', model)
|
||||
self.bot = ChatGPT(config)
|
||||
|
||||
def get_authorization(self):
|
||||
# TODO: get authorization from browser
|
||||
return
|
||||
|
||||
def get_latest_message_id(self, conversation_id):
|
||||
# TODO: get latest message id from browser
|
||||
return
|
||||
|
||||
def get_conversation_history(self, limit=20, offset=0):
|
||||
# Get the conversation id in the history
|
||||
return self.bot.get_history(limit, offset)
|
||||
|
||||
def send_new_message(self, message):
|
||||
# 发送新会话窗口消息,返回会话id
|
||||
response = self.bot.ask(message)
|
||||
latest_uuid = self.get_conversation_history(limit=1, offset=0).keys()[0]
|
||||
return response, latest_uuid
|
||||
|
||||
def send_message(self, message, conversation_id):
|
||||
# 发送会话窗口消息
|
||||
# TODO: send message from browser
|
||||
return
|
||||
|
||||
def extract_code_fragments(self, text):
|
||||
code_fragments = re.findall(r"```(.*?)```", text, re.DOTALL)
|
||||
return code_fragments
|
||||
|
||||
|
||||
def delete_conversation(self, conversation_id=None):
|
||||
# delete conversation with its uuid
|
||||
if conversation_id is not None:
|
||||
self.bot.delete_conversation(conversation_id)
|
||||
|
||||
if __name__ == "__main__":
|
||||
chatgptBrowser_session = ChatGPTBrowser()
|
||||
text, conversation_id = chatgptBrowser_session.send_new_message(
|
||||
"I am a new tester for RESTful APIs."
|
||||
)
|
||||
|
||||
Reference in New Issue
Block a user