universal-mcp-applications 0.1.21__py3-none-any.whl → 0.1.23__py3-none-any.whl

universal_mcp/applications/google_gemini/app.py

@@ -7,11 +7,8 @@ from typing import Annotated
 
 from google import genai
 from google.genai import types
-from loguru import logger
 from PIL import Image
-
 from universal_mcp.applications.application import APIApplication
-from universal_mcp.applications.file_system.app import FileSystemApp
 from universal_mcp.integrations import Integration
 
 
@@ -216,10 +213,9 @@ class GoogleGeminiApp(APIApplication):
 
 async def test_google_gemini():
     app = GoogleGeminiApp()
-    result = await app.generate_image(
+    await app.generate_image(
         "A beautiful women potrait with red green hair color"
     )
-    print(result)
 
 
 if __name__ == "__main__":

universal_mcp/applications/google_mail/app.py

@@ -4,7 +4,6 @@ from email.message import EmailMessage
 from typing import Any
 
 from loguru import logger
-
 from universal_mcp.applications.application import APIApplication
 from universal_mcp.integrations import Integration
 
@@ -25,22 +24,22 @@ class GoogleMailApp(APIApplication):
     ) -> dict[str, Any]:
         """
         Composes and immediately sends an email message via the Gmail API. It can function as a reply within an existing conversation if a `thread_id` is provided. This action is distinct from `send_draft`, which sends a previously saved draft message, or `create_draft`, which only saves an email.
-        
+
         Args:
             to: The email address of the recipient
             subject: The subject line of the email
             body: The content of the email message
             body_type: The MIME subtype for the body ("plain" or "html"). Defaults to "plain".
             thread_id: Optional thread ID to make this a reply to an existing conversation
-        
+
         Returns:
             A string containing either a success confirmation message or an error description
-        
+
         Raises:
             NotAuthorizedError: When Gmail API authentication is not valid or has expired
             KeyError: When required configuration keys are missing
             Exception: For any other unexpected errors during the email sending process
-        
+
         Tags:
             send, email, api, communication, important, thread, reply, openWorldHint
         """
@@ -80,22 +79,22 @@ class GoogleMailApp(APIApplication):
     ) -> dict[str, Any]:
         """
         Saves a new email draft in Gmail with a specified recipient, subject, and body. An optional thread ID can create the draft as a reply within an existing conversation, distinguishing it from `send_email`, which sends immediately.
-        
+
         Args:
             to: The email address of the recipient
             subject: The subject line of the draft email
             body: The main content/message of the draft email
             body_type: The MIME subtype for the body ("plain" or "html"). Defaults to "plain".
             thread_id: Optional thread ID to make this draft a reply to an existing conversation
-        
+
         Returns:
             A string containing either a success message with the draft ID or an error message describing the failure
-        
+
         Raises:
             NotAuthorizedError: When the user's Gmail API authorization is invalid or expired
             KeyError: When required configuration keys are missing
             Exception: For general API errors, network issues, or other unexpected problems
-        
+
         Tags:
             create, email, draft, gmail, api, important, thread, reply, html
         """
@@ -119,18 +118,18 @@ class GoogleMailApp(APIApplication):
     def send_draft(self, draft_id: str) -> dict[str, Any]:
         """
         Sends a pre-existing Gmail draft identified by its unique ID. It posts to the `/drafts/send` endpoint, converting a saved draft into a sent message. This function acts on drafts from `create_draft` and differs from `send_email`, which composes and sends an email in one step.
-        
+
         Args:
             draft_id: The unique identifier of the Gmail draft to be sent
-        
+
         Returns:
             A string containing either a success message with the sent message ID or an error message detailing the failure reason
-        
+
         Raises:
             NotAuthorizedError: When the user's Gmail API authorization is invalid or expired
             KeyError: When required configuration keys are missing from the API response
             Exception: For other unexpected errors during the API request or response handling
-        
+
         Tags:
             send, email, api, communication, important, draft
         """
@@ -148,19 +147,19 @@ class GoogleMailApp(APIApplication):
     def get_draft(self, draft_id: str, format: str = "full") -> dict[str, Any]:
         """
         Retrieves a specific Gmail draft by its unique ID. This function allows specifying the output format (e.g., full, raw) to control the response detail. Unlike `list_drafts`, it fetches a single, known draft rather than a collection of multiple drafts.
-        
+
         Args:
             draft_id: String identifier of the draft email to retrieve
             format: Output format of the draft (options: minimal, full, raw, metadata). Defaults to 'full'
-        
+
         Returns:
             A formatted string containing the draft email details (ID, recipient, subject) or an error message if retrieval fails
-        
+
         Raises:
             NotAuthorizedError: When the user's Gmail authorization is invalid or expired
             KeyError: When required configuration keys or response data fields are missing
             Exception: For any other unexpected errors during draft retrieval
-        
+
         Tags:
             retrieve, email, gmail, draft, api, format, important
         """
@@ -184,20 +183,20 @@ class GoogleMailApp(APIApplication):
     ) -> dict[str, Any]:
         """
         Fetches a list of email drafts, allowing filtering by a search query and limiting results. It can optionally include drafts from spam and trash, returning a collection of draft objects. This is distinct from `get_draft`, which retrieves only a single, specific draft by its ID.
-        
+
         Args:
             max_results: Maximum number of drafts to return (max 500, default 20)
             q: Search query string to filter drafts using Gmail search syntax (default None)
             include_spam_trash: Boolean flag to include drafts from spam and trash folders (default False)
-        
+
         Returns:
             A formatted string containing the list of draft IDs and count information, or an error message if the request fails
-        
+
         Raises:
             NotAuthorizedError: When the Gmail API authentication is missing or invalid
             KeyError: When required configuration keys are missing
             Exception: For general errors during API communication or data processing
-        
+
         Tags:
             list, email, drafts, gmail, api, search, query, pagination, important
         """
@@ -222,13 +221,13 @@ class GoogleMailApp(APIApplication):
     def get_message_details(self, message_id: str) -> dict[str, Any]:
         """
         Retrieves a specific email from Gmail by its ID. It parses the API response to extract and format key details—including sender, subject, body, and attachments—into a structured dictionary. This function provides detailed data for a single message, distinguishing it from `list_messages` which fetches multiple messages.
-        
+
         Args:
             message_id: The unique identifier of the Gmail message to retrieve
-        
+
         Returns:
             A dictionary containing the cleaned message details (serializable as JSON)
-        
+
         Tags:
             retrieve, email, format, api, gmail, message, important, body, content, attachments
         """
@@ -265,8 +264,6 @@ class GoogleMailApp(APIApplication):
             "attachments": attachments,
         }
 
-
-
     def _extract_email_body(self, payload):
         """
         Extracts the email body content from the Gmail API payload.
@@ -332,26 +329,30 @@ class GoogleMailApp(APIApplication):
             list: List of attachment dictionaries with attachment_id, filename, mime_type, and size
         """
         attachments = []
-        
+
         try:
             if payload.get("filename") and payload.get("body", {}).get("attachmentId"):
-                attachments.append({
-                    "attachment_id": payload["body"]["attachmentId"],
-                    "filename": payload["filename"],
-                    "mime_type": payload.get("mimeType", ""),
-                    "size": payload.get("body", {}).get("size", 0)
-                })
+                attachments.append(
+                    {
+                        "attachment_id": payload["body"]["attachmentId"],
+                        "filename": payload["filename"],
+                        "mime_type": payload.get("mimeType", ""),
+                        "size": payload.get("body", {}).get("size", 0),
+                    }
+                )
 
             parts = payload.get("parts", [])
             for part in parts:
                 if part.get("filename") and part.get("body", {}).get("attachmentId"):
-                    attachments.append({
-                        "attachment_id": part["body"]["attachmentId"],
-                        "filename": part["filename"],
-                        "mime_type": part.get("mimeType", ""),
-                        "size": part.get("body", {}).get("size", 0)
-                    })
-                
+                    attachments.append(
+                        {
+                            "attachment_id": part["body"]["attachmentId"],
+                            "filename": part["filename"],
+                            "mime_type": part.get("mimeType", ""),
+                            "size": part.get("body", {}).get("size", 0),
+                        }
+                    )
+
                 elif part.get("parts"):
                     nested_attachments = self._extract_attachments(part)
                     attachments.extend(nested_attachments)
@@ -388,7 +389,7 @@ class GoogleMailApp(APIApplication):
     ) -> dict[str, Any]:
         """
         Fetches a paginated list of detailed email messages using optional search queries. It concurrently retrieves full content (sender, subject, body) for each message, returning the results and a pagination token. This differs from `get_message_details`, which fetches only a single message.
-        
+
         Args:
             max_results: Maximum number of messages to return (max 500, default 20)
             q: Search query string to filter messages using Gmail search syntax.
@@ -409,15 +410,15 @@ class GoogleMailApp(APIApplication):
                     - 'has:attachment' for emails with attachments
                     - 'is:unread' for unread emails
             include_spam_trash: Boolean flag to include messages from spam and trash folders (default False)
-        
+
         Returns:
             A dictionary containing the list of messages and next page token for pagination
-        
+
         Raises:
             NotAuthorizedError: When the Gmail API authentication is invalid or missing
             KeyError: When required configuration keys are missing
             Exception: For general API errors, network issues, or other unexpected problems
-        
+
         Tags:
             list, messages, gmail, search, query, pagination, important
         """
@@ -472,18 +473,18 @@ class GoogleMailApp(APIApplication):
     def get_email_thread(self, thread_id: str) -> dict[str, Any]:
         """
         Fetches a complete email conversation by its unique thread ID. Unlike `get_message_details`, which retrieves a single message, this returns all messages and metadata for the entire thread, providing the full context of the conversation.
-        
+
         Args:
             thread_id: The unique identifier of the Gmail thread to retrieve
-        
+
         Returns:
             A dictionary containing the thread details and all messages in the thread
-        
+
         Raises:
             NotAuthorizedError: When Gmail API authentication is invalid or missing
             KeyError: When required configuration keys are missing
             Exception: For general errors during API communication or data processing
-        
+
         Tags:
             retrieve, email, thread, gmail, api, conversation, important, readOnlyHint, openWorldHint
         """
@@ -495,17 +496,17 @@ class GoogleMailApp(APIApplication):
     def list_labels(self) -> dict[str, Any]:
         """
         Fetches a complete list of all available labels from the user's Gmail account via the API. It retrieves both system-defined (e.g., INBOX) and user-created labels, returning their names and IDs, complementing management functions like `create_label` and `update_label`.
-        
+
         Args:
             None: This method takes no arguments
-        
+
         Returns:
             A formatted string containing a list of Gmail labels categorized by type (system and user), with their IDs, or an error message if the operation fails.
-        
+
         Raises:
             NotAuthorizedError: Raised when the user's Gmail authorization is invalid or missing
             Exception: Raised when any other unexpected error occurs during the API request or data processing
-        
+
         Tags:
             list, gmail, labels, fetch, organize, important, management
         """
@@ -521,17 +522,17 @@ class GoogleMailApp(APIApplication):
     def create_label(self, name: str) -> dict[str, Any]:
         """
         Creates a new Gmail label with a specified name, hardcoding its visibility to ensure it appears in both label and message lists. This function complements `update_label` and `delete_label` by adding new organizational tags to the user's account via the API.
-        
+
         Args:
             name: The display name of the label to create
-        
+
         Returns:
             A formatted string containing the creation status, including the new label's name and ID if successful, or an error message if the creation fails
-        
+
         Raises:
             NotAuthorizedError: Raised when the request lacks proper Gmail API authorization
             Exception: Raised for any other unexpected errors during label creation
-        
+
         Tags:
             create, label, gmail, management, important
         """
@@ -554,17 +555,17 @@ class GoogleMailApp(APIApplication):
     def get_profile(self) -> dict[str, Any]:
         """
         Retrieves the authenticated user's Gmail profile from the API. The profile includes the user's email address, total message and thread counts, and the mailbox's history ID, offering a high-level summary of the account's state.
-        
+
         Args:
             None: This method takes no arguments besides self
-        
+
         Returns:
             A formatted string containing the user's Gmail profile information or an error message if the request fails
-        
+
         Raises:
             NotAuthorizedError: Raised when the user's credentials are invalid or authorization is required
             Exception: Raised for any other unexpected errors during the API request or data processing
-        
+
         Tags:
             fetch, profile, gmail, user-info, api-request, important
         """
@@ -595,7 +596,7 @@ class GoogleMailApp(APIApplication):
     ) -> dict[str, Any]:
         """
         Replaces the entire content of a specific Gmail draft, identified by its ID, with a new message object. This allows complete modification of the recipient, subject, and body, serving as the primary "edit" function for drafts created via `create_draft`.
-        
+
         Args:
             userId (string): userId
             id (string): id
@@ -658,10 +659,10 @@ class GoogleMailApp(APIApplication):
                   }
                 }
                 ```
-        
+
         Returns:
             dict[str, Any]: Successful response
-        
+
         Tags:
             Drafts
         """
@@ -714,7 +715,7 @@ class GoogleMailApp(APIApplication):
     ) -> dict[str, Any]:
         """
         Moves a specific Gmail message to the trash folder by its unique ID. This action performs a soft delete and is the direct counterpart to `untrash_message`, which restores a message. It requires both a user ID and the specific message ID to make the API call.
-        
+
         Args:
             userId (string): userId
             id (string): id
@@ -729,10 +730,10 @@ class GoogleMailApp(APIApplication):
             upload_protocol (string): Upload protocol for media (e.g. "raw", "multipart"). Example: '{{upload_protocol}}'.
             uploadType (string): Legacy upload protocol for media (e.g. "media", "multipart"). Example: '{{uploadType}}'.
             xgafv (string): V1 error format. Example: '{{$.xgafv}}'.
-        
+
         Returns:
             dict[str, Any]: Successful response
-        
+
         Tags:
             Messages, important
         """
@@ -780,7 +781,7 @@ class GoogleMailApp(APIApplication):
     ) -> dict[str, Any]:
         """
         Restores a specific Gmail message from the trash to the user's mailbox, identified by its unique ID. It serves as the direct counterpart to `trash_message`, undoing the deletion action and making the message visible again in the user's account via an API call.
-        
+
         Args:
             userId (string): userId
             id (string): id
@@ -795,10 +796,10 @@ class GoogleMailApp(APIApplication):
             upload_protocol (string): Upload protocol for media (e.g. "raw", "multipart"). Example: '{{upload_protocol}}'.
             uploadType (string): Legacy upload protocol for media (e.g. "media", "multipart"). Example: '{{uploadType}}'.
             xgafv (string): V1 error format. Example: '{{$.xgafv}}'.
-        
+
         Returns:
             dict[str, Any]: Successful response
-        
+
         Tags:
             Messages
         """
@@ -847,7 +848,7 @@ class GoogleMailApp(APIApplication):
     ) -> dict[str, Any]:
         """
         Fetches the raw, base64-encoded content of a specific email attachment using its unique ID. It requires the parent message and user IDs for a targeted API request, returning the file's size and data for download or processing, unlike functions that only list attachment metadata.
-        
+
         Args:
             userId (string): userId
             messageId (string): messageId
@@ -863,10 +864,10 @@ class GoogleMailApp(APIApplication):
             upload_protocol (string): Upload protocol for media (e.g. "raw", "multipart"). Example: '{{upload_protocol}}'.
             uploadType (string): Legacy upload protocol for media (e.g. "media", "multipart"). Example: '{{uploadType}}'.
             xgafv (string): V1 error format. Example: '{{$.xgafv}}'.
-        
+
         Returns:
             dict[str, Any]: Successful response
-        
+
         Tags:
             Messages
         """
@@ -925,7 +926,7 @@ class GoogleMailApp(APIApplication):
     ) -> dict[str, Any]:
         """
         Updates an existing Gmail label's properties, such as its name, color, or visibility, using its unique ID. This modification function is distinct from `create_label` and `delete_label`, returning the full, updated label resource from the API upon successful completion.
-        
+
         Args:
             userId (string): userId
             id (string): id
@@ -967,10 +968,10 @@ class GoogleMailApp(APIApplication):
                   "type": "system"
                 }
                 ```
-        
+
         Returns:
             dict[str, Any]: Successful response
-        
+
         Tags:
             Labels
         """
@@ -1031,7 +1032,7 @@ class GoogleMailApp(APIApplication):
     ) -> Any:
         """
         Permanently removes a specific Gmail label from a user's account, identified by its unique ID. This function performs an irreversible deletion via an API call, requiring both the `userId` and the label `id`. It is the destructive counterpart to `create_label` and `update_label`.
-        
+
         Args:
             userId (string): userId
             id (string): id
@@ -1046,10 +1047,10 @@ class GoogleMailApp(APIApplication):
             upload_protocol (string): Upload protocol for media (e.g. "raw", "multipart"). Example: '{{upload_protocol}}'.
             uploadType (string): Legacy upload protocol for media (e.g. "media", "multipart"). Example: '{{uploadType}}'.
             xgafv (string): V1 error format. Example: '{{$.xgafv}}'.
-        
+
         Returns:
             Any: No Content
-        
+
         Tags:
             Labels
         """
@@ -1097,7 +1098,7 @@ class GoogleMailApp(APIApplication):
     ) -> dict[str, Any]:
         """
         Fetches a single Gmail filter's configuration by its unique ID for a specified user. It returns the filter’s criteria (e.g., sender) and actions (e.g., add label). This differs from `get_all_filters`, which retrieves a complete list of all filters for the user.
-        
+
         Args:
             userId (string): userId
             id (string): id
@@ -1112,10 +1113,10 @@ class GoogleMailApp(APIApplication):
             upload_protocol (string): Upload protocol for media (e.g. "raw", "multipart"). Example: '{{upload_protocol}}'.
             uploadType (string): Legacy upload protocol for media (e.g. "media", "multipart"). Example: '{{uploadType}}'.
             xgafv (string): V1 error format. Example: '{{$.xgafv}}'.
-        
+
         Returns:
             dict[str, Any]: Successful response
-        
+
         Tags:
             settings, Filters
         """
@@ -1163,7 +1164,7 @@ class GoogleMailApp(APIApplication):
     ) -> Any:
         """
         Permanently removes a specific Gmail filter, identified by its unique ID, from a user's account. This action deletes the filter's criteria and all associated automation rules, making it the destructive counterpart to `create_filter` and `get_filter`.
-        
+
         Args:
             userId (string): userId
             id (string): id
@@ -1178,10 +1179,10 @@ class GoogleMailApp(APIApplication):
             upload_protocol (string): Upload protocol for media (e.g. "raw", "multipart"). Example: '{{upload_protocol}}'.
             uploadType (string): Legacy upload protocol for media (e.g. "media", "multipart"). Example: '{{uploadType}}'.
             xgafv (string): V1 error format. Example: '{{$.xgafv}}'.
-        
+
         Returns:
             Any: No Content
-        
+
         Tags:
             settings, Filters
         """
@@ -1228,7 +1229,7 @@ class GoogleMailApp(APIApplication):
     ) -> dict[str, Any]:
         """
         Retrieves all configured email filters for a specified Gmail user ID. It fetches a list of a user's filters, including their matching criteria and automated actions, distinguishing it from `get_filter`, which retrieves only a single filter.
-        
+
         Args:
             userId (string): userId
             access_token (string): OAuth access token. Example: '{{access_token}}'.
@@ -1242,10 +1243,10 @@ class GoogleMailApp(APIApplication):
             upload_protocol (string): Upload protocol for media (e.g. "raw", "multipart"). Example: '{{upload_protocol}}'.
             uploadType (string): Legacy upload protocol for media (e.g. "media", "multipart"). Example: '{{uploadType}}'.
             xgafv (string): V1 error format. Example: '{{$.xgafv}}'.
-        
+
         Returns:
             dict[str, Any]: Successful response
-        
+
         Tags:
             settings, Filters
         """
@@ -1293,7 +1294,7 @@ class GoogleMailApp(APIApplication):
     ) -> dict[str, Any]:
         """
         Creates a new automated email filter in Gmail for a specified user. It requires defining matching criteria (e.g., sender, subject) and an action (e.g., add label). This function adds new rules, distinguishing it from `get_all_filters` and `delete_filter` which manage existing ones.
-        
+
         Args:
             userId (string): userId
             access_token (string): OAuth access token. Example: '{{access_token}}'.
@@ -1338,10 +1339,10 @@ class GoogleMailApp(APIApplication):
                   "id": "in aute anim"
                 }
                 ```
-        
+
         Returns:
             dict[str, Any]: Successful response
-        
+
         Tags:
             settings, Filters
         """