Change proxy usage

terry3041 · Dec 7, 2022 · 78bf50d · 78bf50d
1 parent b3fbf79
commit 78bf50d
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 3 deletions.
diff --git a/README.md b/README.md
@@ -29,6 +29,9 @@ from pyChatGPT import ChatGPT
 
 session_token = 'abc123'  # `__Secure-next-auth.session-token` cookie from https://chat.openai.com/chat
 api = ChatGPT(session_token)
+api2 = ChatGPT(session_token, conversation_id='conversation_id')  # specify a conversation id
+api3 = ChatGPT(session_token, proxy='http://proxy.example.com:8080')  # specify proxy
+
 resp = api.send_message('Hello, world!')
 print(resp['message'])
 

diff --git a/pyproject.toml b/pyproject.toml
@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "pyChatGPT"
-version = "0.0.8"
+version = "0.0.9"
 authors = [
     { name="terry3041", email="[email protected]" },
 ]

diff --git a/src/pyChatGPT/pyChatGPT.py b/src/pyChatGPT/pyChatGPT.py
@@ -9,7 +9,7 @@ class ChatGPT:
     '''
 
     def __init__(
-        self, session_token: str, conversation_id: str = None, proxies: dict = {}
+        self, session_token: str, conversation_id: str = None, proxy: str = None
     ) -> None:
         '''
         Initialize the ChatGPT class\n
@@ -21,7 +21,7 @@ def __init__(
         self.session_token = session_token
         self.conversation_id = conversation_id
         self.parent_id = str(uuid.uuid4())
-        self.proxies = proxies
+        self.proxies = {'http': proxy, 'https': proxy} if proxy else {}
         self.headers = {
             'Cookie': f'__Secure-next-auth.session-token={self.session_token}',
             'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/107.0.0.0 Safari/537.36 Edg/107.0.1418.62',