universal-mcp 0.1.8rc2__py3-none-any.whl → 0.1.8rc4__py3-none-any.whl

universal_mcp/applications/firecrawl/app.py

@@ -15,41 +15,11 @@ class FirecrawlApp(APIApplication):
 
     def __init__(self, integration: Integration | None = None) -> None:
         super().__init__(name="firecrawl", integration=integration)
-        self.api_key: str | None = None
-
-    def _set_api_key(self):
-        """
-        Ensures the API key is loaded from the integration.
-        Raises ValueError if the integration or key is missing/misconfigured.
-        """
-        if self.api_key:
-            return
-
-        if not self.integration:
-            raise ValueError("Integration is None. Cannot retrieve Firecrawl API Key.")
-
-        credentials = self.integration.get_credentials()
-        if not credentials:
-            raise ValueError(
-                f"Failed to retrieve Firecrawl API Key using integration '{self.integration.name}'. "
-                f"Check store configuration (e.g., ensure the correct source like environment variable is set)."
-            )
-        api_key = (
-            credentials.get("api_key")
-            or credentials.get("API_KEY")
-            or credentials.get("apiKey")
-        )
-        if not api_key:
-            raise ValueError(
-                f"Failed to retrieve Firecrawl API Key using integration '{self.integration.name}'. "
-                f"Check store configuration (e.g., ensure the correct environment variable is set)."
-            )
-        self.api_key = api_key
 
     def _get_client(self) -> FirecrawlApiClient:
         """Initializes and returns the Firecrawl client after ensuring API key is set."""
-        self._set_api_key()
-        return FirecrawlApiClient(api_key=self.api_key)
+        api_key = self.integration.get_credentials().get("api_key")
+        return FirecrawlApiClient(api_key=api_key)
 
     def scrape_url(
         self, url: str, params: dict[str, Any] | None = None
@@ -295,4 +265,4 @@ class FirecrawlApp(APIApplication):
             self.check_batch_scrape_status,
             self.start_extract,
             self.check_extract_status,
-        ]
+        ]

universal_mcp/applications/github/app.py

@@ -7,7 +7,6 @@ from universal_mcp.integrations import Integration
 
 
 class GithubApp(APIApplication):
-
     APP_TAGS: ClassVar[list[str]] = ["developers-tools"]
 
     def __init__(self, integration: Integration) -> None:
@@ -28,17 +27,17 @@ class GithubApp(APIApplication):
     def star_repository(self, repo_full_name: str) -> str:
         """
         Stars a GitHub repository using the GitHub API and returns a status message.
-        
+
         Args:
             repo_full_name: The full name of the repository in 'owner/repo' format (e.g., 'octocat/Hello-World')
-        
+
         Returns:
             A string message indicating whether the starring operation was successful, the repository was not found, or an error occurred
-        
+
         Raises:
             RequestException: If there are network connectivity issues or API request failures
             ValueError: If the repository name format is invalid
-        
+
         Tags:
             star, github, api, action, social, repository, important
         """
@@ -55,17 +54,17 @@ class GithubApp(APIApplication):
     def list_commits(self, repo_full_name: str) -> str:
         """
         Retrieves and formats a list of recent commits from a GitHub repository
-        
+
         Args:
             repo_full_name: The full name of the repository in 'owner/repo' format
-        
+
         Returns:
             A formatted string containing the most recent 12 commits, including commit hash, message, and author
-        
+
         Raises:
             requests.exceptions.HTTPError: When the GitHub API request fails (e.g., repository not found, rate limit exceeded)
             requests.exceptions.RequestException: When network issues or other request-related problems occur
-        
+
         Tags:
             list, read, commits, github, history, api, important
         """
@@ -88,17 +87,17 @@ class GithubApp(APIApplication):
     def list_branches(self, repo_full_name: str) -> str:
         """
         Lists all branches for a specified GitHub repository and returns them in a formatted string representation.
-        
+
         Args:
             repo_full_name: The full name of the repository in 'owner/repo' format (e.g., 'octocat/Hello-World')
-        
+
         Returns:
             A formatted string containing the list of branches, or a message indicating no branches were found
-        
+
         Raises:
             HTTPError: When the GitHub API request fails (e.g., repository not found, authentication error)
             RequestException: When there are network connectivity issues or API communication problems
-        
+
         Tags:
             list, branches, github, read, api, repository, important
         """
@@ -118,17 +117,17 @@ class GithubApp(APIApplication):
     def list_pull_requests(self, repo_full_name: str, state: str = "open") -> str:
         """
         Retrieves and formats a list of pull requests for a specified GitHub repository.
-        
+
         Args:
             repo_full_name: The full name of the repository in the format 'owner/repo' (e.g., 'tensorflow/tensorflow')
             state: Filter for pull request state. Can be 'open', 'closed', or 'all'. Defaults to 'open'
-        
+
         Returns:
             A formatted string containing the list of pull requests, including PR number, title, author, and status. Returns a message if no pull requests are found.
-        
+
         Raises:
             HTTPError: Raised when the GitHub API request fails (e.g., invalid repository name, rate limiting, or authentication issues)
-        
+
         Tags:
             list, pull-request, github, api, read, important, fetch, query
         """
@@ -163,7 +162,7 @@ class GithubApp(APIApplication):
     ) -> list[dict[str, Any]]:
         """
         Retrieves a list of issues from a specified GitHub repository with optional filtering parameters.
-        
+
         Args:
             repo_full_name: The full name of the repository in 'owner/repo' format
             state: Filter issues by state ('open', 'closed', 'all'). Defaults to 'open'
@@ -171,14 +170,14 @@ class GithubApp(APIApplication):
             labels: Comma-separated string of label names to filter by (e.g., 'bug,ui,@high')
             per_page: Number of results per page (max 100). Defaults to 30
             page: Page number for pagination. Defaults to 1
-        
+
         Returns:
             List of dictionaries containing issue data from the GitHub API response
-        
+
         Raises:
             HTTPError: When the GitHub API request fails (e.g., invalid repository name, authentication failure)
             RequestException: When there are network connectivity issues or other request-related problems
-        
+
         Tags:
             list, issues, github, api, read, filter, pagination, important, project-management
         """
@@ -196,18 +195,18 @@ class GithubApp(APIApplication):
     def get_pull_request(self, repo_full_name: str, pull_number: int) -> str:
         """
         Retrieves and formats detailed information about a specific GitHub pull request from a repository
-        
+
         Args:
             repo_full_name: The full repository name in 'owner/repo' format (e.g., 'octocat/Hello-World')
             pull_number: The numeric identifier of the pull request to retrieve
-        
+
         Returns:
             A formatted string containing pull request details including title, creator, status, and description
-        
+
         Raises:
             HTTPError: Raised when the GitHub API request fails (e.g., invalid repository name, non-existent PR number, or authentication issues)
             RequestException: Raised when there are network connectivity issues or other request-related problems
-        
+
         Tags:
             get, read, github, pull-request, api, fetch, format, important
         """
@@ -242,7 +241,7 @@ class GithubApp(APIApplication):
     ) -> dict[str, Any]:
         """
         Creates a new pull request in a GitHub repository, optionally converting an existing issue into a pull request.
-        
+
         Args:
             repo_full_name: The full name of the repository (e.g. 'owner/repo')
             head: The name of the branch where your changes are implemented
@@ -252,14 +251,14 @@ class GithubApp(APIApplication):
             issue: An issue number to convert to a pull request. If specified, the issue's title, body, and comments will be used
             maintainer_can_modify: Indicates whether maintainers can modify the pull request
             draft: Indicates whether the pull request is a draft
-        
+
         Returns:
             A dictionary containing the complete GitHub API response
-        
+
         Raises:
             ValueError: Raised when neither 'title' nor 'issue' parameter is specified
             HTTPError: Raised when the GitHub API request fails
-        
+
         Tags:
             create, pull-request, github, api, write, important
         """
@@ -288,19 +287,19 @@ class GithubApp(APIApplication):
     ) -> str:
         """
         Creates a new issue in a specified GitHub repository with a title, body content, and optional labels.
-        
+
         Args:
             repo_full_name: The full name of the repository in 'owner/repo' format
             title: The title of the issue
             body: The contents/description of the issue (defaults to empty string)
             labels: Labels to associate with the issue, as a comma-separated string or list. Only users with push access can set labels
-        
+
         Returns:
             A string containing a confirmation message with the issue number, title, and URL
-        
+
         Raises:
             HTTPError: When the GitHub API request fails (e.g., invalid repository name, authentication issues, or insufficient permissions)
-        
+
         Tags:
             create, issues, github, api, project-management, write, important
         """
@@ -331,19 +330,19 @@ class GithubApp(APIApplication):
     ) -> str:
         """
         Retrieves and formats a list of activities for a specified GitHub repository.
-        
+
         Args:
             repo_full_name: The full name of the repository in 'owner/repo' format
             direction: The sort direction for results ('asc' or 'desc'). Defaults to 'desc'
             per_page: Number of activities to return per page (1-100). Defaults to 30
-        
+
         Returns:
             A formatted string containing a list of repository activities, including timestamps and actor names. Returns a 'No activities' message if no activities are found.
-        
+
         Raises:
             HTTPError: Raised when the GitHub API request fails
             ValueError: May be raised if repo_full_name is invalid or empty after stripping
-        
+
         Tags:
             list, activity, github, read, events, api, query, format
         """
@@ -379,7 +378,7 @@ class GithubApp(APIApplication):
     ) -> dict[str, Any]:
         """
         Updates an existing GitHub issue with specified parameters including title, body, assignee, state, and state reason.
-        
+
         Args:
             repo_full_name: The full name of the repository in 'owner/repo' format
             issue_number: The unique identifier number of the issue to update
@@ -388,14 +387,14 @@ class GithubApp(APIApplication):
             assignee: GitHub username to assign to the issue (optional)
             state: The desired state of the issue ('open' or 'closed') (optional)
             state_reason: The reason for state change ('completed', 'not_planned', 'reopened', or null) (optional)
-        
+
         Returns:
             A dictionary containing the complete GitHub API response with updated issue details
-        
+
         Raises:
             HTTPError: Raised when the GitHub API request fails (e.g., invalid repository, non-existent issue, insufficient permissions)
             RequestException: Raised when there's a network error or API connectivity issue
-        
+
         Tags:
             github, issues, update, api, project-management, write, important
         """
@@ -427,4 +426,4 @@ class GithubApp(APIApplication):
             self.create_issue,
             self.update_issue,
             self.list_repo_activities,
-        ]
+        ]

universal_mcp/applications/google_calendar/app.py

@@ -11,17 +11,6 @@ class GoogleCalendarApp(APIApplication):
         super().__init__(name="google-calendar", integration=integration)
         self.base_api_url = "https://www.googleapis.com/calendar/v3/calendars/primary"
 
-    def _get_headers(self):
-        if not self.integration:
-            raise ValueError("Integration not configured for GoogleCalendarApp")
-        credentials = self.integration.get_credentials()
-        if "headers" in credentials:
-            return credentials["headers"]
-        return {
-            "Authorization": f"Bearer {credentials['access_token']}",
-            "Accept": "application/json",
-        }
-
     def _format_datetime(self, dt_string: str) -> str:
         """Format a datetime string from ISO format to a human-readable format.
 
@@ -61,18 +50,18 @@ class GoogleCalendarApp(APIApplication):
     ) -> str:
         """
         Retrieves and formats events from Google Calendar for today or a specified number of future days, with optional result limiting and timezone specification.
-        
+
         Args:
             days: Number of days to retrieve events for (default: 1, which is just today)
             max_results: Maximum number of events to return (optional)
             time_zone: Time zone used in the response (optional, default is calendar's time zone)
-        
+
         Returns:
             A formatted string containing a list of calendar events with their times and IDs, or a message indicating no events were found
-        
+
         Raises:
             HTTPError: Raised when the API request fails or returns an error status code
-        
+
         Tags:
             fetch, list, calendar, events, date-time, important, api, formatting
         """
@@ -139,19 +128,19 @@ class GoogleCalendarApp(APIApplication):
     ) -> str:
         """
         Retrieves and formats detailed information about a specific Google Calendar event by its ID
-        
+
         Args:
             event_id: The unique identifier of the calendar event to retrieve
             max_attendees: Optional. The maximum number of attendees to include in the response. If None, includes all attendees
             time_zone: Optional. The time zone to use for formatting dates in the response. If None, uses the calendar's default time zone
-        
+
         Returns:
             A formatted string containing comprehensive event details including summary, time, location, description, creator, organizer, recurrence status, and attendee information
-        
+
         Raises:
             HTTPError: Raised when the API request fails or returns an error status code
             JSONDecodeError: Raised when the API response cannot be parsed as JSON
-        
+
         Tags:
             retrieve, calendar, event, format, api, important
         """
@@ -219,7 +208,7 @@ class GoogleCalendarApp(APIApplication):
     ) -> str:
         """
         Retrieves and formats a list of events from Google Calendar with customizable filtering, sorting, and pagination options
-        
+
         Args:
             max_results: Maximum number of events to return (default: 10, max: 2500)
             time_min: Start time in ISO format (e.g., '2023-12-01T00:00:00Z'). Defaults to current time if not specified
@@ -229,13 +218,13 @@ class GoogleCalendarApp(APIApplication):
             single_events: Whether to expand recurring events into individual instances (default: True)
             time_zone: Time zone for response formatting (defaults to calendar's time zone)
             page_token: Token for retrieving a specific page of results in paginated responses
-        
+
         Returns:
             A formatted string containing the list of events with details including summary, ID, start time, and location, or a message if no events are found
-        
+
         Raises:
             HTTPError: Raised when the API request fails or returns an error status code
-        
+
         Tags:
             list, calendar, events, search, filter, pagination, format, important
         """
@@ -307,17 +296,17 @@ class GoogleCalendarApp(APIApplication):
     def quick_add_event(self, text: str, send_updates: str = "none") -> str:
         """
         Creates a calendar event using natural language text input and returns a formatted confirmation message with event details.
-        
+
         Args:
             text: Natural language text describing the event (e.g., 'Meeting with John at Coffee Shop tomorrow 3pm-4pm')
             send_updates: Specifies who should receive event notifications: 'all', 'externalOnly', or 'none' (default)
-        
+
         Returns:
             A formatted string containing the confirmation message with event details including summary, time, location, and event ID
-        
+
         Raises:
             HTTPError: Raised when the API request fails or returns an error status code
-        
+
         Tags:
             create, calendar, event, quick-add, natural-language, important
         """
@@ -358,7 +347,7 @@ class GoogleCalendarApp(APIApplication):
     ) -> str:
         """
         Retrieves and formats all instances of a recurring calendar event within a specified time range, showing details like time, status, and modifications for each occurrence.
-        
+
         Args:
             event_id: ID of the recurring event
             max_results: Maximum number of event instances to return (default: 25, max: 2500)
@@ -367,14 +356,14 @@ class GoogleCalendarApp(APIApplication):
             time_zone: Time zone used in the response (defaults to calendar's time zone)
             show_deleted: Whether to include deleted instances (default: False)
             page_token: Token for retrieving a specific page of results
-        
+
         Returns:
             A formatted string containing a list of event instances with details including time, status, instance ID, and modification information, plus pagination token if applicable.
-        
+
         Raises:
             HTTPError: Raised when the API request fails or returns an error status code
             JSONDecodeError: Raised when the API response cannot be parsed as JSON
-        
+
         Tags:
             list, retrieve, calendar, events, recurring, pagination, formatting, important
         """

universal_mcp/applications/google_docs/app.py

@@ -1,9 +1,6 @@
 from typing import Any
 
-from loguru import logger
-
 from universal_mcp.applications.application import APIApplication
-from universal_mcp.exceptions import NotAuthorizedError
 from universal_mcp.integrations import Integration
 
 
@@ -11,36 +8,21 @@ class GoogleDocsApp(APIApplication):
     def __init__(self, integration: Integration) -> None:
         super().__init__(name="google-docs", integration=integration)
         self.base_api_url = "https://docs.googleapis.com/v1/documents"
-    
-    def _get_headers(self):
-        if not self.integration:
-            raise ValueError("Integration not configured for GoogleDocsApp")
-        credentials = self.integration.get_credentials()
-        if not credentials:
-            logger.warning("No Google credentials found via integration.")
-            action = self.integration.authorize()
-            raise NotAuthorizedError(action)
-        if "headers" in credentials:
-            return credentials["headers"]
-        return {
-            "Authorization": f"Bearer {credentials['access_token']}",
-            "Content-Type": "application/json",
-        }
-    
+
     def create_document(self, title: str) -> dict[str, Any]:
         """
         Creates a new blank Google Document with the specified title and returns the API response.
-        
+
         Args:
             title: The title for the new Google Document to be created
-        
+
         Returns:
             A dictionary containing the Google Docs API response with document details and metadata
-        
+
         Raises:
             HTTPError: If the API request fails due to network issues, authentication errors, or invalid parameters
             RequestException: If there are connection errors or timeout issues during the API request
-        
+
         Tags:
             create, document, api, important, google-docs, http
         """
@@ -49,65 +31,58 @@ class GoogleDocsApp(APIApplication):
         response = self._post(url, data=document_data)
         response.raise_for_status()
         return response.json()
-    
+
     def get_document(self, document_id: str) -> dict[str, Any]:
         """
         Retrieves the latest version of a specified document from the Google Docs API.
-        
+
         Args:
             document_id: The unique identifier of the document to retrieve
-        
+
         Returns:
             A dictionary containing the document data from the Google Docs API response
-        
+
         Raises:
             HTTPError: If the API request fails or the document is not found
             JSONDecodeError: If the API response cannot be parsed as JSON
-        
+
         Tags:
             retrieve, read, api, document, google-docs, important
         """
         url = f"{self.base_api_url}/{document_id}"
         response = self._get(url)
         return response.json()
-    
-    def add_content(self, document_id: str, content: str, index: int = 1) -> dict[str, Any]:
+
+    def add_content(
+        self, document_id: str, content: str, index: int = 1
+    ) -> dict[str, Any]:
         """
         Adds text content at a specified position in an existing Google Document via the Google Docs API.
-        
+
         Args:
             document_id: The unique identifier of the Google Document to be updated
             content: The text content to be inserted into the document
             index: The zero-based position in the document where the text should be inserted (default: 1)
-        
+
         Returns:
             A dictionary containing the Google Docs API response after performing the batch update operation
-        
+
         Raises:
             HTTPError: When the API request fails, such as invalid document_id or insufficient permissions
             RequestException: When there are network connectivity issues or API endpoint problems
-        
+
         Tags:
             update, insert, document, api, google-docs, batch, content-management, important
         """
         url = f"{self.base_api_url}/{document_id}:batchUpdate"
         batch_update_data = {
             "requests": [
-                {
-                    "insertText": {
-                        "location": {"index": index},
-                        "text": content
-                    }
-                }
+                {"insertText": {"location": {"index": index}, "text": content}}
             ]
         }
         response = self._post(url, data=batch_update_data)
         response.raise_for_status()
         return response.json()
-    
+
     def list_tools(self):
-        return [
-           self.create_document,
-           self.get_document,
-           self.add_content
-        ]
+        return [self.create_document, self.get_document, self.add_content]