feature: proxy setup for imap access in hugging face.

Brain/src/common/utils.py

@@ -9,6 +9,10 @@ PINECONE_KEY = os.getenv("PINECONE_KEY")
 PINECONE_ENV = os.getenv("PINECONE_ENV")
 FIREBASE_ENV = os.getenv("FIREBASE_SERVICE_ACCOUNT_TEST3_83FFC")
 
+# proxy for imap access from hugging face.
+PROXY_IP = os.getenv("PROXY_IP")
+PROXY_PORT = os.getenv("PROXY_PORT")
+
 # swagger
 swagger_destination_path = "./src/static/swagger.json"
 SWAGGER_URL = "/api/docs"  # URL for exposing Swagger UI (without trailing '/')

Brain/src/rising_plugin/gmail/email_plugin.py

@@ -9,9 +9,13 @@ import time
 import base64
 from email.header import decode_header
 from email.message import EmailMessage
+from socket import socket
 
+import socks
 from bs4 import BeautifulSoup
 
+from Brain.src.common.utils import PROXY_IP, PROXY_PORT
+
 # email variables
 EMAIL_SMTP_HOST = "smtp.gmail.com"
 EMAIL_SMTP_PORT = 1587
@@ -279,6 +283,11 @@ class EmailPlugin:
     def imap_open(
         self, imap_folder: str, email_sender: str, email_password: str
     ) -> imaplib.IMAP4_SSL:
+        # Create a new socket object for later connections as a proxy
+        socks.set_default_proxy(socks.SOCKS5, PROXY_IP, PROXY_PORT)
+        socket.socket = socks.socksocket
+
+        # IMAP Server Connect
         imap_server = EMAIL_IMAP_SERVER
         conn = imaplib.IMAP4_SSL(imap_server)
         conn.login(user=email_sender, password=email_password)

Brain/src/router/email_router.py

@@ -69,7 +69,11 @@ def construct_blueprint_email_api() -> APIRouter:
         except Exception as e:
             if isinstance(e, BrainException):
                 return e.get_response_exp()
-            return assembler.to_response(400, "Failed to read emails", "")
+            return assembler.to_response(
+                400,
+                "Failed to read emails",
+                "ConnectionResetError occurred for IMAP Server",
+            )
         return assembler.to_response(200, "", result)
 
     """@generator.request_body(

requirements.txt

@@ -75,4 +75,5 @@ google-search-results==2.4.2
 tiktoken==0.4.0
 tiktoken==0.4.0
 gradio-client==0.2.7
-gradio==3.14.0
+gradio==3.14.0
+PySocks==1.7.1