jyapayne/langflow
1
0
Fork 0
Code Issues Pull requests Projects Releases 1 Packages Wiki Activity Actions

fix: improve composio gmail component stability (#7673)

Browse source 
* tool change check

* optimised logic

* Update composio_base.py

* update to the init logic

* Update gmail_composio.py

* update default tool logic

* Update src/backend/base/langflow/base/composio/composio_base.py

Co-authored-by: Gabriel Luiz Freitas Almeida <gabriel@langflow.org>

* Update test_gmail.py

---------

Co-authored-by: Gabriel Luiz Freitas Almeida <gabriel@langflow.org>
This commit is contained in:
Edwin Jose 2025-04-22 11:14:56 -04:00 • committed by GitHub
commit b48e57a223
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
3 changed files with 13 additions and 14 deletions
Download patch file Download diff file Expand all files Collapse all files

7
src/backend/base/langflow/base/composio/composio_base.py
View file

@ -91,7 +91,7 @@ class ComposioBaseComponent(Component):
def _build_action_maps(self):
"""Build lookup maps for action names."""
if not self._display_to_key_map:
if not self._display_to_key_map or not self._key_to_display_map:
self._display_to_key_map = {data["display_name"]: key for key, data in self._actions_data.items()}
self._key_to_display_map = {key: data["display_name"] for key, data in self._actions_data.items()}
self._sanitized_names = {
@ -285,6 +285,7 @@ class ComposioBaseComponent(Component):
async def _get_tools(self) -> list[Tool]:
"""Get tools with cached results and optimized name sanitization."""
toolset = self._build_wrapper()
self.set_default_tools()
return self.configure_tools(toolset)
@property
@ -296,3 +297,7 @@ class ComposioBaseComponent(Component):
@abstractmethod
def execute_action(self) -> list[dict]:
"""Execute action and return response as Message."""
@abstractmethod
def set_default_tools(self):
"""Set the default tools."""

19
src/backend/base/langflow/components/composio/gmail_composio.py
View file

@ -110,19 +110,6 @@ class ComposioGmailAPIComponent(ComposioBaseComponent):
_all_fields = {field for action_data in _actions_data.values() for field in action_data["action_fields"]}
_bool_variables = {"is_html", "include_spam_trash"}
def __init__(self, *args, **kwargs):
super().__init__(*args, **kwargs)
self._default_tools = {
self.sanitize_action_name("GMAIL_SEND_EMAIL").replace(" ", "-"),
self.sanitize_action_name("GMAIL_FETCH_EMAILS").replace(" ", "-"),
}
# Build the action maps right away
self._display_to_key_map = {data["display_name"]: key for key, data in self._actions_data.items()}
self._key_to_display_map = {key: data["display_name"] for key, data in self._actions_data.items()}
self._sanitized_names = {
action: self._name_sanitizer.sub("-", self.sanitize_action_name(action)) for action in self._actions_data
}
# Combine base inputs with Gmail-specific inputs
inputs = [
*ComposioBaseComponent._base_inputs,
@ -404,3 +391,9 @@ class ComposioGmailAPIComponent(ComposioBaseComponent):
def update_build_config(self, build_config: dict, field_value: Any, field_name: str | None = None) -> dict:
return super().update_build_config(build_config, field_value, field_name)
def set_default_tools(self):
self._default_tools = {
self.sanitize_action_name("GMAIL_SEND_EMAIL").replace(" ", "-"),
self.sanitize_action_name("GMAIL_FETCH_EMAILS").replace(" ", "-"),
}

1
src/backend/tests/unit/components/bundles/composio/test_gmail.py
View file

@ -99,6 +99,7 @@ class TestGmailComponent(ComponentTestBaseWithoutClient):
# For this specific test, we need to customize the action_data to handle results field
component._actions_data = {
"GMAIL_FETCH_EMAILS": {
"display_name": "Fetch Emails",
"action_fields": ["max_results", "query"],
"result_field": "messages",
"get_result_field": True,