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

fix: Composio base to handle dataframe response correctly (#9427)

Browse source 
* fix: df handling

* fix: format

---------

Co-authored-by: Edwin Jose <edwin.jose@datastax.com>
This commit is contained in:
Uday Sidagana 2025-08-22 15:26:47 +05:30 • committed by GitHub
commit 8af190902c
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 21 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -135,7 +135,15 @@ class ComposioBaseComponent(Component):
if isinstance(result, dict): if isinstance(result, dict):
result = [result] result = [result]
return DataFrame(result) # Build DataFrame and avoid exposing a 'data' attribute via column access,
result_dataframe = DataFrame(result)
if hasattr(result_dataframe, "columns"):
try:
if "data" in result_dataframe.columns:
result_dataframe = result_dataframe.rename(columns={"data": "_data"})
except (AttributeError, TypeError, ValueError, KeyError) as e:
logger.debug(f"Failed to rename 'data' column: {e}")
return result_dataframe
def as_data(self) -> Data: def as_data(self) -> Data:
result = self.execute_action() result = self.execute_action()
@ -379,6 +387,8 @@ class ComposioBaseComponent(Component):
# Handle conflicting field names - rename user_id to avoid conflicts with entity_id # Handle conflicting field names - rename user_id to avoid conflicts with entity_id
if clean_field == "user_id": if clean_field == "user_id":
clean_field = f"{self.app_name}_user_id" clean_field = f"{self.app_name}_user_id"
elif clean_field == "status":
clean_field = f"{self.app_name}_status"
action_fields.append(clean_field) action_fields.append(clean_field)
@ -526,11 +536,18 @@ class ComposioBaseComponent(Component):
# Handle conflicting field names - rename user_id to avoid conflicts with entity_id # Handle conflicting field names - rename user_id to avoid conflicts with entity_id
if clean_field_name == "user_id": if clean_field_name == "user_id":
clean_field_name = f"{self.app_name}_user_id" clean_field_name = f"{self.app_name}_user_id"
# Update the field schema description to reflect the name change # Update
field_schema_copy = field_schema.copy() field_schema_copy = field_schema.copy()
field_schema_copy["description"] = ( field_schema_copy["description"] = (
f"User ID for {self.app_name.title()}: " + field_schema["description"] f"User ID for {self.app_name.title()}: " + field_schema["description"]
) )
elif clean_field_name == "status":
clean_field_name = f"{self.app_name}_status"
# Update
field_schema_copy = field_schema.copy()
field_schema_copy["description"] = (
f"Status for {self.app_name.title()}: " + field_schema["description"]
)
else: else:
# Use the original field schema for all other fields # Use the original field schema for all other fields
field_schema_copy = field_schema field_schema_copy = field_schema
@ -1240,6 +1257,8 @@ class ComposioBaseComponent(Component):
final_field_name = field final_field_name = field
if field.endswith("_user_id") and field.startswith(self.app_name): if field.endswith("_user_id") and field.startswith(self.app_name):
final_field_name = "user_id" final_field_name = "user_id"
elif field.endswith("_status") and field.startswith(self.app_name):
final_field_name = "status"
arguments[final_field_name] = value arguments[final_field_name] = value