universal-mcp-applications 0.1.30rc2__py3-none-any.whl → 0.1.36rc2__py3-none-any.whl

universal_mcp/applications/contentful/app.py

@@ -1,6 +1,5 @@
 from collections.abc import Callable
 from typing import Any
-
 from loguru import logger
 from universal_mcp.applications.application import GraphQLApplication
 from universal_mcp.exceptions import NotAuthorizedError
@@ -8,24 +7,14 @@ from universal_mcp.integrations import Integration
 
 
 class ContentfulApp(GraphQLApplication):
-    def __init__(
-        self,
-        integration: Integration | None = None,
-        **kwargs: Any,
-    ) -> None:
+    def __init__(self, integration: Integration | None = None, **kwargs: Any) -> None:
         self.space_id: str | None = None
         self.environment_id: str = "master"
         self._access_token: str | None = None
         self._is_eu_customer: bool = False
         self._credentials_loaded: bool = False
         default_base_url = "https://graphql.contentful.com"
-
-        super().__init__(
-            name="contentful",
-            base_url=default_base_url,
-            integration=integration,
-            **kwargs,
-        )
+        super().__init__(name="contentful", base_url=default_base_url, integration=integration, **kwargs)
 
     def _load_credentials_and_construct_url(self) -> bool:
         """
@@ -37,70 +26,39 @@ class ContentfulApp(GraphQLApplication):
         """
         if self._credentials_loaded:
             return True
-
         logger.debug("Attempting to load Contentful credentials and construct URL...")
-
         if not self.integration:
-            logger.error(
-                "Contentful integration not configured. Cannot load credentials or construct URL."
-            )
-            # Mark as 'loaded' to prevent retries, even though it failed.
+            logger.error("Contentful integration not configured. Cannot load credentials or construct URL.")
             self._credentials_loaded = True
             return False
-
         try:
             credentials = self.integration.get_credentials()
         except NotAuthorizedError as e:
-            logger.error(
-                f"Authorization required or credentials unavailable for Contentful: {e.message}"
-            )
-            self._credentials_loaded = True  # Prevent retries
+            logger.error(f"Authorization required or credentials unavailable for Contentful: {e.message}")
+            self._credentials_loaded = True
             return False
         except Exception as e:
-            logger.error(
-                f"Failed to get credentials from integration: {e}", exc_info=True
-            )
-            self._credentials_loaded = True  # Prevent retries
+            logger.error(f"Failed to get credentials from integration: {e}", exc_info=True)
+            self._credentials_loaded = True
             return False
-
         self.space_id = credentials.get("space_id")
-        self._access_token = credentials.get("access_token") or credentials.get(
-            "api_key"
-        )
-        self.environment_id = credentials.get(
-            "environment_id", "master"
-        )  # Use default if not specified
-        self._is_eu_customer = credentials.get(
-            "is_eu_customer", False
-        )  # Use default if not specified
-
+        self._access_token = credentials.get("access_token") or credentials.get("api_key")
+        self.environment_id = credentials.get("environment_id", "master")
+        self._is_eu_customer = credentials.get("is_eu_customer", False)
         missing_creds = []
         if not self.space_id:
             missing_creds.append("'space_id'")
         if not self._access_token:
             missing_creds.append("'access_token' or 'api_key'")
-
         if missing_creds:
-            logger.error(
-                f"Missing required Contentful credentials in integration: {', '.join(missing_creds)}. "
-                "API calls will fail."
-            )
-            self._credentials_loaded = True  # Prevent retries
+            logger.error(f"Missing required Contentful credentials in integration: {', '.join(missing_creds)}. API calls will fail.")
+            self._credentials_loaded = True
             return False
-
-        contentful_api_domain = (
-            "graphql.eu.contentful.com"
-            if self._is_eu_customer
-            else "graphql.contentful.com"
-        )
+        contentful_api_domain = "graphql.eu.contentful.com" if self._is_eu_customer else "graphql.contentful.com"
         self.base_url = f"https://{contentful_api_domain}/content/v1/spaces/{self.space_id}/environments/{self.environment_id}"
-
         self._client = None
-
         logger.info(
-            f"Contentful credentials loaded and URL constructed successfully. "
-            f"Space: '{self.space_id}', Environment: '{self.environment_id}'. "
-            f"Base URL: {self.base_url}"
+            f"Contentful credentials loaded and URL constructed successfully. Space: '{self.space_id}', Environment: '{self.environment_id}'. Base URL: {self.base_url}"
         )
         self._credentials_loaded = True
         return True
@@ -114,7 +72,7 @@ class ContentfulApp(GraphQLApplication):
             return ""
         if len(parts) == 1 and parts[0] == s and s:
             return s[0].lower() + s[1:] if len(s) > 1 else s.lower()
-        return parts[0].lower() + "".join(word.capitalize() for word in parts[1:])
+        return parts[0].lower() + "".join((word.capitalize() for word in parts[1:]))
 
     @staticmethod
     def _to_pascal_case(s: str) -> str:
@@ -125,57 +83,36 @@ class ContentfulApp(GraphQLApplication):
             return ""
         if len(parts) == 1 and parts[0] == s and s:
             return s[0].upper() + s[1:] if len(s) > 1 else s.upper()
-        return "".join(word.capitalize() for word in parts)
+        return "".join((word.capitalize() for word in parts))
 
     def _ensure_loaded(self) -> bool:
         """Internal helper to trigger lazy loading and check status."""
         if not self._credentials_loaded:
             return self._load_credentials_and_construct_url()
-        # If already marked as loaded, check if essential parts are actually set (robustness check)
         return bool(self.base_url and self.space_id and self._access_token)
 
-    # --- Tool Methods ---
-
-    def get_entry(
-        self,
-        content_type_id: str,
-        entry_id: str,
-        fields_to_select: str,
-        locale: str | None = None,
-        preview: bool = False,
+    async def get_entry(
+        self, content_type_id: str, entry_id: str, fields_to_select: str, locale: str | None = None, preview: bool = False
     ) -> dict[str, Any]:
         """
         Fetches a single entry of a specified content type by its ID.
         (See original docstring for details)
         """
         if not self._ensure_loaded():
-            return {
-                "error": "Failed to initialize ContentfulApp. Check credentials and configuration."
-            }
-
+            return {"error": "Failed to initialize ContentfulApp. Check credentials and configuration."}
         query_field = self._to_camel_case(content_type_id)
-        logger.debug(
-            f"Fetching entry for content_type_id='{content_type_id}' (query field='{query_field}'), entry_id='{entry_id}'"
-        )
-        query_gql = f"""
-            query GetEntryById($id: String!, $locale: String, $preview: Boolean) {{
-                {query_field}(id: $id, locale: $locale, preview: $preview) {{
-                    {fields_to_select}
-                }}
-            }}
-        """
+        logger.debug(f"Fetching entry for content_type_id='{content_type_id}' (query field='{query_field}'), entry_id='{entry_id}'")
+        query_gql = f"\n            query GetEntryById($id: String!, $locale: String, $preview: Boolean) {{\n                {query_field}(id: $id, locale: $locale, preview: $preview) {{\n                    {fields_to_select}\n                }}\n            }}\n        "
         variables: dict[str, Any] = {"id": entry_id, "preview": preview}
         if locale:
             variables["locale"] = locale
-
-        # Call the base class query method which uses the configured client
         try:
             return self.query(query_gql, variables=variables)
         except Exception as e:
             logger.error(f"Error executing get_entry query: {e}", exc_info=True)
             return {"error": f"Failed to get entry: {e}"}
 
-    def get_entries_collection(
+    async def get_entries_collection(
         self,
         content_type_id: str,
         fields_to_select_for_item: str,
@@ -191,28 +128,14 @@ class ContentfulApp(GraphQLApplication):
         (See original docstring for details)
         """
         if not self._ensure_loaded():
-            return {
-                "error": "Failed to initialize ContentfulApp. Check credentials and configuration."
-            }
-
+            return {"error": "Failed to initialize ContentfulApp. Check credentials and configuration."}
         collection_field = self._to_camel_case(content_type_id) + "Collection"
         filter_type = self._to_pascal_case(content_type_id) + "Filter"
         order_enum_type = self._to_pascal_case(content_type_id) + "Order"
         logger.debug(
-            f"Fetching collection for content_type_id='{content_type_id}' "
-            f"(collection field='{collection_field}', filter type='{filter_type}', order enum='{order_enum_type}')"
+            f"Fetching collection for content_type_id='{content_type_id}' (collection field='{collection_field}', filter type='{filter_type}', order enum='{order_enum_type}')"
         )
-        query_gql = f"""
-            query GetEntries(
-                $limit: Int, $skip: Int, $where: {filter_type}, $order: [{order_enum_type}!], $locale: String, $preview: Boolean
-            ) {{
-                {collection_field}(
-                    limit: $limit, skip: $skip, where: $where, order: $order, locale: $locale, preview: $preview
-                ) {{
-                    total skip limit items {{ {fields_to_select_for_item} }}
-                }}
-            }}
-        """
+        query_gql = f"\n            query GetEntries(\n                $limit: Int, $skip: Int, $where: {filter_type}, $order: [{order_enum_type}!], $locale: String, $preview: Boolean\n            ) {{\n                {collection_field}(\n                    limit: $limit, skip: $skip, where: $where, order: $order, locale: $locale, preview: $preview\n                ) {{\n                    total skip limit items {{ {fields_to_select_for_item} }}\n                }}\n            }}\n        "
         variables: dict[str, Any] = {"preview": preview}
         if limit is not None:
             variables["limit"] = limit
@@ -224,16 +147,13 @@ class ContentfulApp(GraphQLApplication):
             variables["order"] = order
         if locale:
             variables["locale"] = locale
-
         try:
             return self.query(query_gql, variables=variables)
         except Exception as e:
-            logger.error(
-                f"Error executing get_entries_collection query: {e}", exc_info=True
-            )
+            logger.error(f"Error executing get_entries_collection query: {e}", exc_info=True)
             return {"error": f"Failed to get entries collection: {e}"}
 
-    def get_asset(
+    async def get_asset(
         self,
         asset_id: str,
         fields_to_select: str = "sys { id } url title description fileName contentType size width height",
@@ -244,25 +164,17 @@ class ContentfulApp(GraphQLApplication):
         (See original docstring for details)
         """
         if not self._ensure_loaded():
-            return {
-                "error": "Failed to initialize ContentfulApp. Check credentials and configuration."
-            }
-
+            return {"error": "Failed to initialize ContentfulApp. Check credentials and configuration."}
         logger.debug(f"Fetching asset_id='{asset_id}'")
-        query_gql = f"""
-            query GetAssetById($id: String!, $preview: Boolean) {{
-                asset(id: $id, preview: $preview) {{ {fields_to_select} }}
-            }}
-        """
+        query_gql = f"\n            query GetAssetById($id: String!, $preview: Boolean) {{\n                asset(id: $id, preview: $preview) {{ {fields_to_select} }}\n            }}\n        "
         variables: dict[str, Any] = {"id": asset_id, "preview": preview}
-
         try:
             return self.query(query_gql, variables=variables)
         except Exception as e:
             logger.error(f"Error executing get_asset query: {e}", exc_info=True)
             return {"error": f"Failed to get asset: {e}"}
 
-    def get_assets_collection(
+    async def get_assets_collection(
         self,
         fields_to_select_for_item: str = "sys { id } url title description fileName contentType size width height",
         limit: int | None = None,
@@ -277,22 +189,9 @@ class ContentfulApp(GraphQLApplication):
         (See original docstring for details)
         """
         if not self._ensure_loaded():
-            return {
-                "error": "Failed to initialize ContentfulApp. Check credentials and configuration."
-            }
-
+            return {"error": "Failed to initialize ContentfulApp. Check credentials and configuration."}
         logger.debug("Fetching assets collection")
-        query_gql = f"""
-            query GetAssets(
-                $limit: Int, $skip: Int, $where: AssetFilter, $order: [AssetOrder!], $locale: String, $preview: Boolean
-            ) {{
-                assetCollection(
-                    limit: $limit, skip: $skip, where: $where, order: $order, locale: $locale, preview: $preview
-                ) {{
-                    total skip limit items {{ {fields_to_select_for_item} }}
-                }}
-            }}
-        """
+        query_gql = f"\n            query GetAssets(\n                $limit: Int, $skip: Int, $where: AssetFilter, $order: [AssetOrder!], $locale: String, $preview: Boolean\n            ) {{\n                assetCollection(\n                    limit: $limit, skip: $skip, where: $where, order: $order, locale: $locale, preview: $preview\n                ) {{\n                    total skip limit items {{ {fields_to_select_for_item} }}\n                }}\n            }}\n        "
         variables: dict[str, Any] = {"preview": preview}
         if limit is not None:
             variables["limit"] = limit
@@ -304,18 +203,13 @@ class ContentfulApp(GraphQLApplication):
             variables["order"] = order
         if locale:
             variables["locale"] = locale
-
         try:
             return self.query(query_gql, variables=variables)
         except Exception as e:
-            logger.error(
-                f"Error executing get_assets_collection query: {e}", exc_info=True
-            )
+            logger.error(f"Error executing get_assets_collection query: {e}", exc_info=True)
             return {"error": f"Failed to get assets collection: {e}"}
 
-    def execute_graphql_query(
-        self, query_string: str, variables: dict[str, Any] | None = None
-    ) -> dict[str, Any]:
+    async def execute_graphql_query(self, query_string: str, variables: dict[str, Any] | None = None) -> dict[str, Any]:
         """
         Executes an arbitrary GraphQL query against the configured Contentful space/environment.
 
@@ -330,10 +224,7 @@ class ContentfulApp(GraphQLApplication):
             important
         """
         if not self._ensure_loaded():
-            return {
-                "error": "Failed to initialize ContentfulApp. Check credentials and configuration."
-            }
-
+            return {"error": "Failed to initialize ContentfulApp. Check credentials and configuration."}
         logger.debug(f"Executing custom GraphQL query with variables: {variables}")
         try:
             return self.query(query_string, variables=variables)
@@ -343,10 +234,4 @@ class ContentfulApp(GraphQLApplication):
 
     def list_tools(self) -> list[Callable]:
         """Returns a list of methods exposed as tools."""
-        return [
-            self.get_entry,
-            self.get_entries_collection,
-            self.get_asset,
-            self.get_assets_collection,
-            self.execute_graphql_query,
-        ]
+        return [self.get_entry, self.get_entries_collection, self.get_asset, self.get_assets_collection, self.execute_graphql_query]

universal_mcp/applications/crustdata/app.py

@@ -1,5 +1,4 @@
 from typing import Any
-
 from universal_mcp.applications.application import APIApplication
 from universal_mcp.integrations import Integration
 
@@ -11,15 +10,9 @@ class CrustdataApp(APIApplication):
 
     def _get_headers(self) -> dict[str, Any]:
         api_key = self.integration.get_credentials().get("api_key")
-        return {
-            "Authorization": f"Token {api_key}",
-            "Content-Type": "application/json",
-            "Accept": "application/json",
-        }
+        return {"Authorization": f"Token {api_key}", "Content-Type": "application/json", "Accept": "application/json"}
 
-    def screen_companies(
-        self, metrics, filters, offset, count, sorts
-    ) -> dict[str, Any]:
+    async def screen_companies(self, metrics, filters, offset, count, sorts) -> dict[str, Any]:
         """
         Screens companies based on specified metrics, filters, sorting, and pagination parameters, and returns the result as a JSON-compatible dictionary.
 
@@ -50,13 +43,7 @@ class CrustdataApp(APIApplication):
             raise ValueError("Missing required parameter 'count'")
         if sorts is None:
             raise ValueError("Missing required parameter 'sorts'")
-        request_body = {
-            "metrics": metrics,
-            "filters": filters,
-            "offset": offset,
-            "count": count,
-            "sorts": sorts,
-        }
+        request_body = {"metrics": metrics, "filters": filters, "offset": offset, "count": count, "sorts": sorts}
         request_body = {k: v for k, v in request_body.items() if v is not None}
         url = f"{self.base_url}/screener/screen/"
         query_params = {}
@@ -64,7 +51,7 @@ class CrustdataApp(APIApplication):
         response.raise_for_status()
         return response.json()
 
-    def get_headcount_timeseries(self, filters, offset, count, sorts) -> dict[str, Any]:
+    async def get_headcount_timeseries(self, filters, offset, count, sorts) -> dict[str, Any]:
         """
         Retrieve headcount timeseries data from the data lab endpoint using the provided filters, pagination, and sorting options.
 
@@ -92,12 +79,7 @@ class CrustdataApp(APIApplication):
             raise ValueError("Missing required parameter 'count'")
         if sorts is None:
             raise ValueError("Missing required parameter 'sorts'")
-        request_body = {
-            "filters": filters,
-            "offset": offset,
-            "count": count,
-            "sorts": sorts,
-        }
+        request_body = {"filters": filters, "offset": offset, "count": count, "sorts": sorts}
         request_body = {k: v for k, v in request_body.items() if v is not None}
         url = f"{self.base_url}/data_lab/headcount_timeseries/"
         query_params = {}
@@ -105,9 +87,7 @@ class CrustdataApp(APIApplication):
         response.raise_for_status()
         return response.json()
 
-    def get_headcount_by_facet_timeseries(
-        self, filters, offset, count, sorts
-    ) -> dict[str, Any]:
+    async def get_headcount_by_facet_timeseries(self, filters, offset, count, sorts) -> dict[str, Any]:
         """
         Retrieves headcount timeseries data aggregated by specified facets using provided filters and sorting options.
 
@@ -135,12 +115,7 @@ class CrustdataApp(APIApplication):
             raise ValueError("Missing required parameter 'count'")
         if sorts is None:
             raise ValueError("Missing required parameter 'sorts'")
-        request_body = {
-            "filters": filters,
-            "offset": offset,
-            "count": count,
-            "sorts": sorts,
-        }
+        request_body = {"filters": filters, "offset": offset, "count": count, "sorts": sorts}
         request_body = {k: v for k, v in request_body.items() if v is not None}
         url = f"{self.base_url}/data_lab/headcount_by_facet_timeseries/"
         query_params = {}
@@ -148,9 +123,7 @@ class CrustdataApp(APIApplication):
         response.raise_for_status()
         return response.json()
 
-    def get_funding_milestone_timeseries(
-        self, filters, offset, count, sorts
-    ) -> dict[str, Any]:
+    async def get_funding_milestone_timeseries(self, filters, offset, count, sorts) -> dict[str, Any]:
         """
         Retrieves a time series of funding milestone data based on specified filters, pagination, and sorting options.
 
@@ -178,12 +151,7 @@ class CrustdataApp(APIApplication):
             raise ValueError("Missing required parameter 'count'")
         if sorts is None:
             raise ValueError("Missing required parameter 'sorts'")
-        request_body = {
-            "filters": filters,
-            "offset": offset,
-            "count": count,
-            "sorts": sorts,
-        }
+        request_body = {"filters": filters, "offset": offset, "count": count, "sorts": sorts}
         request_body = {k: v for k, v in request_body.items() if v is not None}
         url = f"{self.base_url}/data_lab/funding_milestone_timeseries/"
         query_params = {}
@@ -191,9 +159,7 @@ class CrustdataApp(APIApplication):
         response.raise_for_status()
         return response.json()
 
-    def get_decision_makers(
-        self, filters, offset, count, sorts, decision_maker_titles
-    ) -> dict[str, Any]:
+    async def get_decision_makers(self, filters, offset, count, sorts, decision_maker_titles) -> dict[str, Any]:
         """
         Retrieves decision makers based on specified filters and parameters.
 
@@ -238,7 +204,7 @@ class CrustdataApp(APIApplication):
         response.raise_for_status()
         return response.json()
 
-    def get_web_traffic(self, filters, offset, count, sorts) -> dict[str, Any]:
+    async def get_web_traffic(self, filters, offset, count, sorts) -> dict[str, Any]:
         """
         Retrieves web traffic data based on provided filters, pagination, and sorting criteria.
 
@@ -266,12 +232,7 @@ class CrustdataApp(APIApplication):
             raise ValueError("Missing required parameter 'count'")
         if sorts is None:
             raise ValueError("Missing required parameter 'sorts'")
-        request_body = {
-            "filters": filters,
-            "offset": offset,
-            "count": count,
-            "sorts": sorts,
-        }
+        request_body = {"filters": filters, "offset": offset, "count": count, "sorts": sorts}
         request_body = {k: v for k, v in request_body.items() if v is not None}
         url = f"{self.base_url}/data_lab/webtraffic/"
         query_params = {}
@@ -279,7 +240,7 @@ class CrustdataApp(APIApplication):
         response.raise_for_status()
         return response.json()
 
-    def get_investor_portfolio(self, investor_name) -> dict[str, Any]:
+    async def get_investor_portfolio(self, investor_name) -> dict[str, Any]:
         """
         Retrieves the investment portfolio information for a specified investor.
 
@@ -299,16 +260,12 @@ class CrustdataApp(APIApplication):
         if investor_name is None:
             raise ValueError("Missing required parameter 'investor_name'")
         url = f"{self.base_url}/data_lab/investor_portfolio"
-        query_params = {
-            k: v for k, v in [("investor_name", investor_name)] if v is not None
-        }
+        query_params = {k: v for k, v in [("investor_name", investor_name)] if v is not None}
         response = self._get(url, params=query_params)
         response.raise_for_status()
         return response.json()
 
-    def get_job_listings(
-        self, tickers, dataset, filters, offset, count, sorts
-    ) -> dict[str, Any]:
+    async def get_job_listings(self, tickers, dataset, filters, offset, count, sorts) -> dict[str, Any]:
         """
         Retrieves job listings data based on specified parameters.
 
@@ -342,14 +299,7 @@ class CrustdataApp(APIApplication):
             raise ValueError("Missing required parameter 'count'")
         if sorts is None:
             raise ValueError("Missing required parameter 'sorts'")
-        request_body = {
-            "tickers": tickers,
-            "dataset": dataset,
-            "filters": filters,
-            "offset": offset,
-            "count": count,
-            "sorts": sorts,
-        }
+        request_body = {"tickers": tickers, "dataset": dataset, "filters": filters, "offset": offset, "count": count, "sorts": sorts}
         request_body = {k: v for k, v in request_body.items() if v is not None}
         url = f"{self.base_url}/data_lab/job_listings/Table/"
         query_params = {}
@@ -357,7 +307,7 @@ class CrustdataApp(APIApplication):
         response.raise_for_status()
         return response.json()
 
-    def search_persons(self, job_id) -> dict[str, Any]:
+    async def search_persons(self, job_id) -> dict[str, Any]:
         """
         Submits a search request for persons associated with a given asynchronous job and returns the search results as a dictionary.
 
@@ -376,9 +326,7 @@ class CrustdataApp(APIApplication):
         """
         if job_id is None:
             raise ValueError("Missing required parameter 'job_id'")
-        request_body = {
-            "job_id": job_id,
-        }
+        request_body = {"job_id": job_id}
         request_body = {k: v for k, v in request_body.items() if v is not None}
         url = f"{self.base_url}/screener/person/search"
         query_params = {}
@@ -386,7 +334,7 @@ class CrustdataApp(APIApplication):
         response.raise_for_status()
         return response.json()
 
-    def search_companies(self, filters, page) -> dict[str, Any]:
+    async def search_companies(self, filters, page) -> dict[str, Any]:
         """
         Searches for companies using specified filters and pagination parameters.
 
@@ -408,10 +356,7 @@ class CrustdataApp(APIApplication):
             raise ValueError("Missing required parameter 'filters'")
         if page is None:
             raise ValueError("Missing required parameter 'page'")
-        request_body = {
-            "filters": filters,
-            "page": page,
-        }
+        request_body = {"filters": filters, "page": page}
         request_body = {k: v for k, v in request_body.items() if v is not None}
         url = f"{self.base_url}/screener/company/search"
         query_params = {}
@@ -419,9 +364,7 @@ class CrustdataApp(APIApplication):
         response.raise_for_status()
         return response.json()
 
-    def enrich_person(
-        self, linkedin_profile_url, enrich_realtime, fields
-    ) -> dict[str, Any]:
+    async def enrich_person(self, linkedin_profile_url, enrich_realtime, fields) -> dict[str, Any]:
         """
         Retrieves enriched person data from LinkedIn profile using the provided profile URL, enrichment mode, and requested fields.
 
@@ -449,18 +392,14 @@ class CrustdataApp(APIApplication):
         url = f"{self.base_url}/screener/person/enrich"
         query_params = {
             k: v
-            for k, v in [
-                ("linkedin_profile_url", linkedin_profile_url),
-                ("enrich_realtime", enrich_realtime),
-                ("fields", fields),
-            ]
+            for k, v in [("linkedin_profile_url", linkedin_profile_url), ("enrich_realtime", enrich_realtime), ("fields", fields)]
             if v is not None
         }
         response = self._get(url, params=query_params)
         response.raise_for_status()
         return response.json()
 
-    def enrich_company(self, company_domain, enrich_realtime) -> dict[str, Any]:
+    async def enrich_company(self, company_domain, enrich_realtime) -> dict[str, Any]:
         """
         Retrieves enriched company data using the provided company domain and enrichment mode.
 
@@ -483,19 +422,12 @@ class CrustdataApp(APIApplication):
         if enrich_realtime is None:
             raise ValueError("Missing required parameter 'enrich_realtime'")
         url = f"{self.base_url}/screener/company"
-        query_params = {
-            k: v
-            for k, v in [
-                ("company_domain", company_domain),
-                ("enrich_realtime", enrich_realtime),
-            ]
-            if v is not None
-        }
+        query_params = {k: v for k, v in [("company_domain", company_domain), ("enrich_realtime", enrich_realtime)] if v is not None}
         response = self._get(url, params=query_params)
         response.raise_for_status()
         return response.json()
 
-    def get_linked_in_posts(self, company_linkedin_url) -> dict[str, Any]:
+    async def get_linked_in_posts(self, company_linkedin_url) -> dict[str, Any]:
         """
         Fetches LinkedIn posts for a specified company using its LinkedIn URL.
 
@@ -515,18 +447,12 @@ class CrustdataApp(APIApplication):
         if company_linkedin_url is None:
             raise ValueError("Missing required parameter 'company_linkedin_url'")
         url = f"{self.base_url}/screener/linkedin_posts"
-        query_params = {
-            k: v
-            for k, v in [("company_linkedin_url", company_linkedin_url)]
-            if v is not None
-        }
+        query_params = {k: v for k, v in [("company_linkedin_url", company_linkedin_url)] if v is not None}
         response = self._get(url, params=query_params)
         response.raise_for_status()
         return response.json()
 
-    def search_linked_in_posts(
-        self, keyword, page, sort_by, date_posted
-    ) -> dict[str, Any]:
+    async def search_linked_in_posts(self, keyword, page, sort_by, date_posted) -> dict[str, Any]:
         """
         Searches LinkedIn posts using the provided keyword and filters, returning the search results as a dictionary.
 
@@ -554,12 +480,7 @@ class CrustdataApp(APIApplication):
             raise ValueError("Missing required parameter 'sort_by'")
         if date_posted is None:
             raise ValueError("Missing required parameter 'date_posted'")
-        request_body = {
-            "keyword": keyword,
-            "page": page,
-            "sort_by": sort_by,
-            "date_posted": date_posted,
-        }
+        request_body = {"keyword": keyword, "page": page, "sort_by": sort_by, "date_posted": date_posted}
         request_body = {k: v for k, v in request_body.items() if v is not None}
         url = f"{self.base_url}/screener/linkedin_posts/keyword_search/"
         query_params = {}