diff --git a/python/composio/tools/toolset.py b/python/composio/tools/toolset.py index 6ed7b1f929d..326f403898e 100644 --- a/python/composio/tools/toolset.py +++ b/python/composio/tools/toolset.py @@ -1360,7 +1360,9 @@ def get_connected_account(self, id: str) -> ConnectedAccountModel: return self.client.connected_accounts.get(connection_id=id) def get_connected_accounts(self) -> t.List[ConnectedAccountModel]: - return self.client.connected_accounts.get() + return self.client.connected_accounts.get( + entity_ids=[self.entity_id] if self.entity_id else None + ) def get_entity(self, id: t.Optional[str] = None) -> Entity: """Get entity object for given ID."""