dhisana 0.0.1.dev9__tar.gz → 0.0.1.dev10__tar.gz

{dhisana-0.0.1.dev9 → dhisana-0.0.1.dev10}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: dhisana
-Version: 0.0.1.dev9
+Version: 0.0.1.dev10
 Summary: A Python SDK for Dhisana AI Platform
 Home-page: https://github.com/dhisana-ai/dhisana-python-sdk
 Author: Admin

{dhisana-0.0.1.dev9 → dhisana-0.0.1.dev10}/setup.py

@@ -2,7 +2,7 @@ from setuptools import setup, find_packages
 
 setup(
     name='dhisana',  
-    version='0.0.1-dev9',
+    version='0.0.1-dev10',
     description='A Python SDK for Dhisana AI Platform',
     author='Admin',
     author_email='contact@dhisana.ai',

{dhisana-0.0.1.dev9 → dhisana-0.0.1.dev10}/src/dhisana/utils/google_workspace_tools.py

@@ -16,6 +16,8 @@ from googleapiclient.errors import HttpError
 import datetime
 from dhisana.utils.assistant_tool_tag import assistant_tool
 
+
+
 def convert_base_64_json(base64_string):
     """
     Convert a base64 encoded string to a JSON string.
@@ -445,6 +447,187 @@ async def list_emails_in_time_range_async(start_time: str, end_time: str, unread
 
     return email_details
 
+
+@assistant_tool
+async def fetch_last_n_sent_messages(recipient_email: str, num_messages: int) -> List[Dict[str, Any]]:
+    """
+    Fetch the last n messages sent to a specific recipient using the Gmail API with a service account.
+    The service account must have domain-wide delegation to impersonate the sender.
+
+    :param recipient_email: The recipient's email address.
+    :param num_messages: The number of recent messages to fetch.
+    :return: A list of email message details.
+    """
+    # Retrieve the service account JSON and email for automation from environment variables
+    service_account_base64 = os.getenv('GOOGLE_SERVICE_KEY')
+    automationemails = os.getenv('EMAIL_FOR_AUTOMATION')
+    if not service_account_base64 or not automationemails:
+        raise EnvironmentError("Required environment variables are not set.")
+
+    allowed_emails_for_automation = automationemails.split(",")
+    if not allowed_emails_for_automation:
+        raise EnvironmentError("No allowed emails for automation found in environment variables.")
+
+    mailbox_email = allowed_emails_for_automation[0]
+
+    # Function to convert base64-encoded JSON to a string
+    def convert_base_64_json(base64_json: str) -> str:
+        return base64.b64decode(base64_json).decode('utf-8')
+
+    service_account_json = convert_base_64_json(service_account_base64)
+
+    # Parse the JSON string into a dictionary
+    service_account_info = json.loads(service_account_json)
+
+    # Define the required scope for reading emails via Gmail API
+    SCOPES = ['https://mail.google.com/']
+
+    # Authenticate using the service account info and impersonate the email for automation
+    credentials = service_account.Credentials.from_service_account_info(
+        service_account_info, scopes=SCOPES
+    ).with_subject(mailbox_email)
+
+    # Refresh the token if necessary
+    if not credentials.valid:
+        request = Request()
+        credentials.refresh(request)
+
+    # Get the access token
+    access_token = credentials.token
+
+    # Define the Gmail API endpoint for listing messages
+    gmail_api_url = 'https://gmail.googleapis.com/gmail/v1/users/me/messages'
+
+    # Create the query parameters for the recipient email
+    query = f'to:{recipient_email}'
+
+    # Prepare the headers
+    headers = {
+        'Authorization': f'Bearer {access_token}'
+    }
+
+    # Prepare the parameters
+    params = {
+        'q': query,
+        'maxResults': num_messages
+    }
+
+    async with httpx.AsyncClient() as client:
+        response = await client.get(gmail_api_url, headers=headers, params=params)
+        response.raise_for_status()  # Raise an exception for HTTP errors
+        messages = response.json().get('messages', [])
+
+        email_details = []
+        for message in messages:
+            message_id = message['id']
+            message_url = f'https://gmail.googleapis.com/gmail/v1/users/me/messages/{message_id}'
+            message_response = await client.get(message_url, headers=headers)
+            message_response.raise_for_status()
+            message_data = message_response.json()
+
+            email_details.append({
+                "mailbox_email_id": message_data['id'],
+                "message_id": message_data['threadId'],
+                "email_subject": next((header['value'] for header in message_data['payload']['headers'] if header['name'] == 'Subject'), None),
+                "email_sender": next((header['value'] for header in message_data['payload']['headers'] if header['name'] == 'From'), None),
+                "email_recipients": [header['value'] for header in message_data['payload']['headers'] if header['name'] in ['To', 'Cc', 'Bcc']],
+                "email_date": next((header['value'] for header in message_data['payload']['headers'] if header['name'] == 'Date'), None),
+                "read_email_status": 'UNREAD' if 'UNREAD' in message_data['labelIds'] else 'READ',
+                "email_labels": message_data.get('labelIds', [])
+            })
+
+    return email_details
+
+@assistant_tool
+async def fetch_last_n_received_messages(sender_email: str, num_messages: int) -> List[Dict[str, Any]]:
+    """
+    Fetch the last n messages received from a specific sender using the Gmail API with a service account.
+    The service account must have domain-wide delegation to impersonate the recipient.
+
+    :param sender_email: The sender's email address.
+    :param num_messages: The number of recent messages to fetch.
+    :return: A list of email message details.
+    """
+    # Retrieve the service account JSON and email for automation from environment variables
+    service_account_base64 = os.getenv('GOOGLE_SERVICE_KEY')
+    automationemails = os.getenv('EMAIL_FOR_AUTOMATION')
+    if not service_account_base64 or not automationemails:
+        raise EnvironmentError("Required environment variables are not set.")
+
+    allowed_emails_for_automation = automationemails.split(",")
+    if not allowed_emails_for_automation:
+        raise EnvironmentError("No allowed emails for automation found in environment variables.")
+
+    mailbox_email = allowed_emails_for_automation[0]
+
+    # Function to convert base64-encoded JSON to a string
+    def convert_base_64_json(base64_json: str) -> str:
+        return base64.b64decode(base64_json).decode('utf-8')
+
+    service_account_json = convert_base_64_json(service_account_base64)
+
+    # Parse the JSON string into a dictionary
+    service_account_info = json.loads(service_account_json)
+
+    # Define the required scope for reading emails via Gmail API
+    SCOPES = ['https://mail.google.com/']
+
+    # Authenticate using the service account info and impersonate the email for automation
+    credentials = service_account.Credentials.from_service_account_info(
+        service_account_info, scopes=SCOPES
+    ).with_subject(mailbox_email)
+
+    # Refresh the token if necessary
+    if not credentials.valid:
+        request = Request()
+        credentials.refresh(request)
+
+    # Get the access token
+    access_token = credentials.token
+
+    # Define the Gmail API endpoint for listing messages
+    gmail_api_url = 'https://gmail.googleapis.com/gmail/v1/users/me/messages'
+
+    # Create the query parameters for the sender email
+    query = f'from:{sender_email}'
+
+    # Prepare the headers
+    headers = {
+        'Authorization': f'Bearer {access_token}'
+    }
+
+    # Prepare the parameters
+    params = {
+        'q': query,
+        'maxResults': num_messages
+    }
+
+    async with httpx.AsyncClient() as client:
+        response = await client.get(gmail_api_url, headers=headers, params=params)
+        response.raise_for_status()  # Raise an exception for HTTP errors
+        messages = response.json().get('messages', [])
+
+        email_details = []
+        for message in messages:
+            message_id = message['id']
+            message_url = f'https://gmail.googleapis.com/gmail/v1/users/me/messages/{message_id}'
+            message_response = await client.get(message_url, headers=headers)
+            message_response.raise_for_status()
+            message_data = message_response.json()
+
+            email_details.append({
+                "mailbox_email_id": message_data['id'],
+                "message_id": message_data['threadId'],
+                "email_subject": next((header['value'] for header in message_data['payload']['headers'] if header['name'] == 'Subject'), None),
+                "email_sender": next((header['value'] for header in message_data['payload']['headers'] if header['name'] == 'From'), None),
+                "email_recipients": [header['value'] for header in message_data['payload']['headers'] if header['name'] in ['To', 'Cc', 'Bcc']],
+                "email_date": next((header['value'] for header in message_data['payload']['headers'] if header['name'] == 'Date'), None),
+                "read_email_status": 'UNREAD' if 'UNREAD' in message_data['labelIds'] else 'READ',
+                "email_labels": message_data.get('labelIds', [])
+            })
+
+    return email_details
+
 @assistant_tool
 async def get_email_details_async(message_id: str, mailbox_email: str = "") -> Dict[str, Any]:
     """