format

DOCUMENTATION.md

@@ -13,11 +13,10 @@ class Chatbot()
 ```
 
 A class to interact with Google Bard.
-
-Parameters:
-
+Parameters
     session_id: str
         The __Secure-1PSID cookie.
+    proxy: str
 
 <a id="Bard.Chatbot.ask"></a>
 
@@ -36,3 +35,4 @@ Send a message to Google Bard and return the response.
 **Returns**:
 
 A dict containing the response from Google Bard.
+

example.py

@@ -1,4 +1,5 @@
 from os import environ
+
 from Bard import Chatbot
 
 token = environ.get("BARD_TOKEN")

src/Bard.py

@@ -3,10 +3,10 @@
 """
 import argparse
 import json
+import os
 import random
 import re
 import string
-import os
 import sys
 
 import requests
@@ -54,6 +54,7 @@ class Chatbot:
     Parameters
         session_id: str
             The __Secure-1PSID cookie.
+        proxy: str
     """
 
     __slots__ = [
@@ -66,7 +67,7 @@ class Chatbot:
         "session",
     ]
 
-    def __init__(self, session_id: str, proxy: str=None):
+    def __init__(self, session_id: str, proxy: str = None):
         headers = {
             "Host": "bard.google.com",
             "X-Same-Domain": "1",
@@ -84,8 +85,8 @@ def __init__(self, session_id: str, proxy: str=None):
             self.session.proxies.update(
                 {
                     "http": proxy,
-                    "https": proxy
-                }
+                    "https": proxy,
+                },
             )
         self.session.headers = headers
         self.session.cookies.set("__Secure-1PSID", session_id)