lfx-nightly 0.2.0.dev25__py3-none-any.whl

lfx/components/composio/apollo_composio.py

@@ -0,0 +1,11 @@
+from lfx.base.composio.composio_base import ComposioBaseComponent
+
+
+class ComposioApolloAPIComponent(ComposioBaseComponent):
+    display_name: str = "Apollo"
+    icon = "Apollo"
+    documentation: str = "https://docs.composio.dev"
+    app_name = "apollo"
+
+    def set_default_tools(self):
+        """Set the default tools for Apollo component."""

lfx/components/composio/asana_composio.py

@@ -0,0 +1,11 @@
+from lfx.base.composio.composio_base import ComposioBaseComponent
+
+
+class ComposioAsanaAPIComponent(ComposioBaseComponent):
+    display_name: str = "Asana"
+    icon = "Asana"
+    documentation: str = "https://docs.composio.dev"
+    app_name = "asana"
+
+    def set_default_tools(self):
+        """Set the default tools for Asana component."""

lfx/components/composio/attio_composio.py

@@ -0,0 +1,11 @@
+from lfx.base.composio.composio_base import ComposioBaseComponent
+
+
+class ComposioAttioAPIComponent(ComposioBaseComponent):
+    display_name: str = "Attio"
+    icon = "Attio"
+    documentation: str = "https://docs.composio.dev"
+    app_name = "attio"
+
+    def set_default_tools(self):
+        """Set the default tools for Attio component."""

lfx/components/composio/bitbucket_composio.py

@@ -0,0 +1,11 @@
+from lfx.base.composio.composio_base import ComposioBaseComponent
+
+
+class ComposioBitbucketAPIComponent(ComposioBaseComponent):
+    display_name: str = "Bitbucket"
+    icon = "Bitbucket"
+    documentation: str = "https://docs.composio.dev"
+    app_name = "bitbucket"
+
+    def set_default_tools(self):
+        """Set the default tools for Bitbucket component."""

lfx/components/composio/bolna_composio.py

@@ -0,0 +1,11 @@
+from lfx.base.composio.composio_base import ComposioBaseComponent
+
+
+class ComposioBolnaAPIComponent(ComposioBaseComponent):
+    display_name: str = "Bolna"
+    icon = "Bolna"
+    documentation: str = "https://docs.composio.dev"
+    app_name = "bolna"
+
+    def set_default_tools(self):
+        """Set the default tools for Bolna component."""

lfx/components/composio/brightdata_composio.py

@@ -0,0 +1,11 @@
+from lfx.base.composio.composio_base import ComposioBaseComponent
+
+
+class ComposioBrightdataAPIComponent(ComposioBaseComponent):
+    display_name: str = "Brightdata"
+    icon = "Brightdata"
+    documentation: str = "https://docs.composio.dev"
+    app_name = "brightdata"
+
+    def set_default_tools(self):
+        """Set the default tools for Brightdata component."""

lfx/components/composio/calendly_composio.py

@@ -0,0 +1,11 @@
+from lfx.base.composio.composio_base import ComposioBaseComponent
+
+
+class ComposioCalendlyAPIComponent(ComposioBaseComponent):
+    display_name: str = "Calendly"
+    icon = "Calendly"
+    documentation: str = "https://docs.composio.dev"
+    app_name = "calendly"
+
+    def set_default_tools(self):
+        """Set the default tools for Calendly component."""

lfx/components/composio/canva_composio.py

@@ -0,0 +1,11 @@
+from lfx.base.composio.composio_base import ComposioBaseComponent
+
+
+class ComposioCanvaAPIComponent(ComposioBaseComponent):
+    display_name: str = "Canva"
+    icon = "Canva"
+    documentation: str = "https://docs.composio.dev"
+    app_name = "canva"
+
+    def set_default_tools(self):
+        """Set the default tools for Canva component."""

lfx/components/composio/canvas_composio.py

@@ -0,0 +1,11 @@
+from lfx.base.composio.composio_base import ComposioBaseComponent
+
+
+class ComposioCanvasAPIComponent(ComposioBaseComponent):
+    display_name: str = "Canvas"
+    icon = "Canvas"
+    documentation: str = "https://docs.composio.dev"
+    app_name = "canvas"
+
+    def set_default_tools(self):
+        """Set the default tools for Canvaas component."""

lfx/components/composio/coda_composio.py

@@ -0,0 +1,11 @@
+from lfx.base.composio.composio_base import ComposioBaseComponent
+
+
+class ComposioCodaAPIComponent(ComposioBaseComponent):
+    display_name: str = "Coda"
+    icon = "Coda"
+    documentation: str = "https://docs.composio.dev"
+    app_name = "coda"
+
+    def set_default_tools(self):
+        """Set the default tools for Coda component."""

lfx/components/composio/composio_api.py

@@ -0,0 +1,278 @@
+# Standard library imports
+from collections.abc import Sequence
+from typing import Any
+
+from composio import Composio
+from composio_langchain import LangchainProvider
+
+# Third-party imports
+from langchain_core.tools import Tool
+
+# Local imports
+from lfx.base.langchain_utilities.model import LCToolComponent
+from lfx.inputs.inputs import (
+    ConnectionInput,
+    MessageTextInput,
+    SecretStrInput,
+    SortableListInput,
+)
+from lfx.io import Output
+from lfx.utils.validate_cloud import raise_error_if_astra_cloud_disable_component
+
+# TODO: We get the list from the API but we need to filter it
+enabled_tools = ["confluence", "discord", "dropbox", "github", "gmail", "linkedin", "notion", "slack", "youtube"]
+
+disable_component_in_astra_cloud_msg = (
+    "Composio tools are not supported in Astra cloud environment. "
+    "Please use local storage mode or cloud-based versions of the tools."
+)
+
+
+class ComposioAPIComponent(LCToolComponent):
+    display_name: str = "Composio Tools"
+    description: str = "Use Composio toolset to run actions with your agent"
+    name = "ComposioAPI"
+    icon = "Composio"
+    documentation: str = "https://docs.composio.dev"
+
+    inputs = [
+        # Basic configuration inputs
+        MessageTextInput(name="entity_id", display_name="Entity ID", value="default", advanced=True),
+        SecretStrInput(
+            name="api_key",
+            display_name="Composio API Key",
+            required=True,
+            info="Refer to https://docs.composio.dev/faq/api_key/api_key",
+            real_time_refresh=True,
+        ),
+        ConnectionInput(
+            name="tool_name",
+            display_name="Tool Name",
+            placeholder="Select a tool...",
+            button_metadata={"icon": "unplug", "variant": "destructive"},
+            options=[],
+            search_category=[],
+            value="",
+            connection_link="",
+            info="The name of the tool to use",
+            real_time_refresh=True,
+        ),
+        SortableListInput(
+            name="actions",
+            display_name="Actions",
+            placeholder="Select action",
+            helper_text="Please connect before selecting actions.",
+            helper_text_metadata={"icon": "OctagonAlert", "variant": "destructive"},
+            options=[],
+            value="",
+            info="The actions to use",
+            limit=1,
+            show=False,
+        ),
+    ]
+
+    outputs = [
+        Output(name="tools", display_name="Tools", method="build_tool"),
+    ]
+
+    def validate_tool(self, build_config: dict, field_value: Any, tool_name: str | None = None) -> dict:
+        # Get the index of the selected tool in the list of options
+        selected_tool_index = next(
+            (
+                ind
+                for ind, tool in enumerate(build_config["tool_name"]["options"])
+                if tool["name"] == field_value
+                or ("validate" in field_value and tool["name"] == field_value["validate"])
+            ),
+            None,
+        )
+
+        # Set the link to be the text 'validated'
+        build_config["tool_name"]["options"][selected_tool_index]["link"] = "validated"
+
+        # Set the helper text and helper text metadata field of the actions now
+        build_config["actions"]["helper_text"] = ""
+        build_config["actions"]["helper_text_metadata"] = {"icon": "Check", "variant": "success"}
+
+        try:
+            composio = self._build_wrapper()
+            current_tool = tool_name or getattr(self, "tool_name", None)
+            if not current_tool:
+                self.log("No tool name available for validate_tool")
+                return build_config
+
+            toolkit_slug = current_tool.lower()
+
+            tools = composio.tools.get(user_id=self.entity_id, toolkits=[toolkit_slug])
+
+            authenticated_actions = []
+            for tool in tools:
+                if hasattr(tool, "name"):
+                    action_name = tool.name
+                    display_name = action_name.replace("_", " ").title()
+                    authenticated_actions.append({"name": action_name, "display_name": display_name})
+        except (ValueError, ConnectionError, AttributeError) as e:
+            self.log(f"Error getting actions for {current_tool or 'unknown tool'}: {e}")
+            authenticated_actions = []
+
+        build_config["actions"]["options"] = [
+            {
+                "name": action["name"],
+            }
+            for action in authenticated_actions
+        ]
+
+        build_config["actions"]["show"] = True
+        return build_config
+
+    def update_build_config(self, build_config: dict, field_value: Any, field_name: str | None = None) -> dict:
+        if field_name == "api_key" or (self.api_key and not build_config["tool_name"]["options"]):
+            if field_name == "api_key" and not field_value:
+                build_config["tool_name"]["options"] = []
+                build_config["tool_name"]["value"] = ""
+
+                # Reset the list of actions
+                build_config["actions"]["show"] = False
+                build_config["actions"]["options"] = []
+                build_config["actions"]["value"] = ""
+
+                return build_config
+
+            # Build the list of available tools
+            build_config["tool_name"]["options"] = [
+                {
+                    "name": app.title(),
+                    "icon": app,
+                    "link": (
+                        build_config["tool_name"]["options"][ind]["link"]
+                        if build_config["tool_name"]["options"]
+                        else ""
+                    ),
+                }
+                for ind, app in enumerate(enabled_tools)
+            ]
+
+            return build_config
+
+        if field_name == "tool_name" and field_value:
+            composio = self._build_wrapper()
+
+            current_tool_name = (
+                field_value
+                if isinstance(field_value, str)
+                else field_value.get("validate")
+                if isinstance(field_value, dict) and "validate" in field_value
+                else getattr(self, "tool_name", None)
+            )
+
+            if not current_tool_name:
+                self.log("No tool name available for connection check")
+                return build_config
+
+            try:
+                toolkit_slug = current_tool_name.lower()
+
+                connection_list = composio.connected_accounts.list(
+                    user_ids=[self.entity_id], toolkit_slugs=[toolkit_slug]
+                )
+
+                # Check for active connections
+                has_active_connections = False
+                if (
+                    connection_list
+                    and hasattr(connection_list, "items")
+                    and connection_list.items
+                    and isinstance(connection_list.items, list)
+                    and len(connection_list.items) > 0
+                ):
+                    for connection in connection_list.items:
+                        if getattr(connection, "status", None) == "ACTIVE":
+                            has_active_connections = True
+                            break
+
+                # Get the index of the selected tool in the list of options
+                selected_tool_index = next(
+                    (
+                        ind
+                        for ind, tool in enumerate(build_config["tool_name"]["options"])
+                        if tool["name"] == current_tool_name.title()
+                    ),
+                    None,
+                )
+
+                if has_active_connections:
+                    # User has active connection
+                    if selected_tool_index is not None:
+                        build_config["tool_name"]["options"][selected_tool_index]["link"] = "validated"
+
+                    # If it's a validation request, validate the tool
+                    if (isinstance(field_value, dict) and "validate" in field_value) or isinstance(field_value, str):
+                        return self.validate_tool(build_config, field_value, current_tool_name)
+                else:
+                    # No active connection - create OAuth connection
+                    try:
+                        connection = composio.toolkits.authorize(user_id=self.entity_id, toolkit=toolkit_slug)
+                        redirect_url = getattr(connection, "redirect_url", None)
+
+                        if redirect_url and redirect_url.startswith(("http://", "https://")):
+                            if selected_tool_index is not None:
+                                build_config["tool_name"]["options"][selected_tool_index]["link"] = redirect_url
+                        elif selected_tool_index is not None:
+                            build_config["tool_name"]["options"][selected_tool_index]["link"] = "error"
+                    except (ValueError, ConnectionError, AttributeError) as e:
+                        self.log(f"Error creating OAuth connection: {e}")
+                        if selected_tool_index is not None:
+                            build_config["tool_name"]["options"][selected_tool_index]["link"] = "error"
+
+            except (ValueError, ConnectionError, AttributeError) as e:
+                self.log(f"Error checking connection status: {e}")
+
+        return build_config
+
+    def build_tool(self) -> Sequence[Tool]:
+        """Build Composio tools based on selected actions.
+
+        Returns:
+            Sequence[Tool]: List of configured Composio tools.
+        """
+        # Check if we're in Astra cloud environment and raise an error if we are.
+        raise_error_if_astra_cloud_disable_component(disable_component_in_astra_cloud_msg)
+        composio = self._build_wrapper()
+        action_names = [action["name"] for action in self.actions]
+
+        # Get toolkits from action names
+        toolkits = set()
+        for action_name in action_names:
+            if "_" in action_name:
+                toolkit = action_name.split("_")[0].lower()
+                toolkits.add(toolkit)
+
+        if not toolkits:
+            return []
+
+        # Get all tools for the relevant toolkits
+        all_tools = composio.tools.get(user_id=self.entity_id, toolkits=list(toolkits))
+
+        # Filter to only the specific actions we want using list comprehension
+        return [tool for tool in all_tools if hasattr(tool, "name") and tool.name in action_names]
+
+    def _build_wrapper(self) -> Composio:
+        """Build the Composio wrapper using new SDK.
+
+        Returns:
+            Composio: The initialized Composio client.
+
+        Raises:
+            ValueError: If the API key is not found or invalid.
+        """
+        # Check if we're in Astra cloud environment and raise an error if we are.
+        raise_error_if_astra_cloud_disable_component(disable_component_in_astra_cloud_msg)
+        try:
+            if not self.api_key:
+                msg = "Composio API Key is required"
+                raise ValueError(msg)
+            return Composio(api_key=self.api_key, provider=LangchainProvider())
+        except ValueError as e:
+            self.log(f"Error building Composio wrapper: {e}")
+            msg = "Please provide a valid Composio API Key in the component settings"
+            raise ValueError(msg) from e

lfx/components/composio/contentful_composio.py

@@ -0,0 +1,11 @@
+from lfx.base.composio.composio_base import ComposioBaseComponent
+
+
+class ComposioContentfulAPIComponent(ComposioBaseComponent):
+    display_name: str = "Contentful"
+    icon = "Contentful"
+    documentation: str = "https://docs.composio.dev"
+    app_name = "contentful"
+
+    def set_default_tools(self):
+        """Set the default tools for Contentful component."""

lfx/components/composio/digicert_composio.py

@@ -0,0 +1,11 @@
+from lfx.base.composio.composio_base import ComposioBaseComponent
+
+
+class ComposioDigicertAPIComponent(ComposioBaseComponent):
+    display_name: str = "Digicert"
+    icon = "Digicert"
+    documentation: str = "https://docs.composio.dev"
+    app_name = "digicert"
+
+    def set_default_tools(self):
+        """Set the default tools for Digicert component."""

lfx/components/composio/discord_composio.py

@@ -0,0 +1,11 @@
+from lfx.base.composio.composio_base import ComposioBaseComponent
+
+
+class ComposioDiscordAPIComponent(ComposioBaseComponent):
+    display_name: str = "Discord"
+    icon = "discord"
+    documentation: str = "https://docs.composio.dev"
+    app_name = "discord"
+
+    def set_default_tools(self):
+        """Set the default tools for Discord component."""

lfx/components/composio/dropbox_compnent.py

@@ -0,0 +1,11 @@
+from lfx.base.composio.composio_base import ComposioBaseComponent
+
+
+class ComposioDropboxAPIComponent(ComposioBaseComponent):
+    display_name: str = "Dropbox"
+    icon = "Dropbox"
+    documentation: str = "https://docs.composio.dev"
+    app_name = "dropbox"
+
+    def set_default_tools(self):
+        """Set the default tools for Dropbox component."""

lfx/components/composio/elevenlabs_composio.py

@@ -0,0 +1,11 @@
+from lfx.base.composio.composio_base import ComposioBaseComponent
+
+
+class ComposioElevenLabsAPIComponent(ComposioBaseComponent):
+    display_name: str = "ElevenLabs"
+    icon = "Elevenlabs"
+    documentation: str = "https://docs.composio.dev"
+    app_name = "elevenlabs"
+
+    def set_default_tools(self):
+        """Set the default tools for ElevenLabs component."""

lfx/components/composio/exa_composio.py

@@ -0,0 +1,11 @@
+from lfx.base.composio.composio_base import ComposioBaseComponent
+
+
+class ComposioExaAPIComponent(ComposioBaseComponent):
+    display_name: str = "Exa"
+    icon = "ExaComposio"
+    documentation: str = "https://docs.composio.dev"
+    app_name = "exa"
+
+    def set_default_tools(self):
+        """Set the default tools for Exa component."""

lfx/components/composio/figma_composio.py

@@ -0,0 +1,11 @@
+from lfx.base.composio.composio_base import ComposioBaseComponent
+
+
+class ComposioFigmaAPIComponent(ComposioBaseComponent):
+    display_name: str = "Figma"
+    icon = "Figma"
+    documentation: str = "https://docs.composio.dev"
+    app_name = "figma"
+
+    def set_default_tools(self):
+        """Set the default tools for Figma component."""

lfx/components/composio/finage_composio.py

@@ -0,0 +1,11 @@
+from lfx.base.composio.composio_base import ComposioBaseComponent
+
+
+class ComposioFinageAPIComponent(ComposioBaseComponent):
+    display_name: str = "Finage"
+    icon = "Finage"
+    documentation: str = "https://docs.composio.dev"
+    app_name = "finage"
+
+    def set_default_tools(self):
+        """Set the default tools for Finage component."""

lfx/components/composio/firecrawl_composio.py

@@ -0,0 +1,11 @@
+from lfx.base.composio.composio_base import ComposioBaseComponent
+
+
+class ComposioFirecrawlAPIComponent(ComposioBaseComponent):
+    display_name: str = "Firecrawl"
+    icon = "Firecrawl"
+    documentation: str = "https://docs.composio.dev"
+    app_name = "firecrawl"
+
+    def set_default_tools(self):
+        """Set the default tools for Firecrawl component."""

lfx/components/composio/fireflies_composio.py

@@ -0,0 +1,11 @@
+from lfx.base.composio.composio_base import ComposioBaseComponent
+
+
+class ComposioFirefliesAPIComponent(ComposioBaseComponent):
+    display_name: str = "Fireflies"
+    icon = "Fireflies"
+    documentation: str = "https://docs.composio.dev"
+    app_name = "fireflies"
+
+    def set_default_tools(self):
+        """Set the default tools for Fireflies component."""

lfx/components/composio/fixer_composio.py

@@ -0,0 +1,11 @@
+from lfx.base.composio.composio_base import ComposioBaseComponent
+
+
+class ComposioFixerAPIComponent(ComposioBaseComponent):
+    display_name: str = "Fixer"
+    icon = "Fixer"
+    documentation: str = "https://docs.composio.dev"
+    app_name = "fixer"
+
+    def set_default_tools(self):
+        """Set the default tools for Fixer component."""

lfx/components/composio/flexisign_composio.py

@@ -0,0 +1,11 @@
+from lfx.base.composio.composio_base import ComposioBaseComponent
+
+
+class ComposioFlexisignAPIComponent(ComposioBaseComponent):
+    display_name: str = "Flexisign"
+    icon = "Flexisign"
+    documentation: str = "https://docs.composio.dev"
+    app_name = "flexisign"
+
+    def set_default_tools(self):
+        """Set the default tools for Flexisign component."""

lfx/components/composio/freshdesk_composio.py

@@ -0,0 +1,11 @@
+from lfx.base.composio.composio_base import ComposioBaseComponent
+
+
+class ComposioFreshdeskAPIComponent(ComposioBaseComponent):
+    display_name: str = "Freshdesk"
+    icon = "Freshdesk"
+    documentation: str = "https://docs.composio.dev"
+    app_name = "freshdesk"
+
+    def set_default_tools(self):
+        """Set the default tools for Freshdesk component."""

lfx/components/composio/github_composio.py

@@ -0,0 +1,11 @@
+from lfx.base.composio.composio_base import ComposioBaseComponent
+
+
+class ComposioGitHubAPIComponent(ComposioBaseComponent):
+    display_name: str = "GitHub"
+    icon = "Github"
+    documentation: str = "https://docs.composio.dev"
+    app_name = "github"
+
+    def set_default_tools(self):
+        """Set the default tools for GitHub component."""

lfx/components/composio/gmail_composio.py

@@ -0,0 +1,38 @@
+from lfx.base.composio.composio_base import ComposioBaseComponent
+
+
+class ComposioGmailAPIComponent(ComposioBaseComponent):
+    display_name: str = "Gmail"
+    icon = "Gmail"
+    documentation: str = "https://docs.composio.dev"
+    app_name = "gmail"
+
+    def __init__(self, **kwargs):
+        super().__init__(**kwargs)
+        self.post_processors = {
+            "GMAIL_SEND_EMAIL": self._process_send_email_response,
+            "GMAIL_FETCH_EMAILS": self._process_fetch_emails_response,
+        }
+
+    def _process_send_email_response(self, raw_data):
+        """Post-processor for GMAIL_SEND_EMAIL action."""
+        if isinstance(raw_data, dict):
+            response_data = raw_data.get("response_data", raw_data)
+
+            return {
+                "message_id": response_data.get("id"),
+                "thread_id": response_data.get("threadId"),
+                "label_ids": response_data.get("labelIds", []),
+            }
+        return raw_data
+
+    def _process_fetch_emails_response(self, raw_data):
+        """Post-processor for GMAIL_FETCH_EMAILS action."""
+        if isinstance(raw_data, dict):
+            messages = raw_data.get("messages", [])
+            if messages:
+                return messages
+        return raw_data
+
+    def set_default_tools(self):
+        """Set the default tools for Gmail component."""

lfx/components/composio/googlebigquery_composio.py

@@ -0,0 +1,11 @@
+from lfx.base.composio.composio_base import ComposioBaseComponent
+
+
+class ComposioGoogleBigQueryAPIComponent(ComposioBaseComponent):
+    display_name: str = "GoogleBigQuery"
+    icon = "Googlebigquery"
+    documentation: str = "https://docs.composio.dev"
+    app_name = "googlebigquery"
+
+    def set_default_tools(self):
+        """Set the default tools for Google BigQuery component."""

lfx/components/composio/googlecalendar_composio.py

@@ -0,0 +1,11 @@
+from lfx.base.composio.composio_base import ComposioBaseComponent
+
+
+class ComposioGoogleCalendarAPIComponent(ComposioBaseComponent):
+    display_name: str = "GoogleCalendar"
+    icon = "Googlecalendar"
+    documentation: str = "https://docs.composio.dev"
+    app_name = "googlecalendar"
+
+    def set_default_tools(self):
+        """Set the default tools for Google Calendar component."""

lfx/components/composio/googleclassroom_composio.py

@@ -0,0 +1,11 @@
+from lfx.base.composio.composio_base import ComposioBaseComponent
+
+
+class ComposioGoogleclassroomAPIComponent(ComposioBaseComponent):
+    display_name: str = "Google Classroom"
+    icon = "Classroom"
+    documentation: str = "https://docs.composio.dev"
+    app_name = "GOOGLE_CLASSROOM"
+
+    def set_default_tools(self):
+        """Set the default tools for Google Classroom component."""