From 6579900c3ef41382b7a7572cd425d5d77cdb2eb6 Mon Sep 17 00:00:00 2001 From: Karan Vaidya Date: Wed, 15 Jan 2025 20:43:25 +0530 Subject: [PATCH 1/2] Fix: filter connections by entity --- python/composio/tools/toolset.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/composio/tools/toolset.py b/python/composio/tools/toolset.py index 6ed7b1f929d..e58ee0db4ee 100644 --- a/python/composio/tools/toolset.py +++ b/python/composio/tools/toolset.py @@ -1360,7 +1360,7 @@ 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.""" From 8583f3a995c838baf259764de6df72bef1990434 Mon Sep 17 00:00:00 2001 From: Karan Vaidya Date: Wed, 15 Jan 2025 20:53:25 +0530 Subject: [PATCH 2/2] Lint --- python/composio/tools/toolset.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/python/composio/tools/toolset.py b/python/composio/tools/toolset.py index e58ee0db4ee..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(entity_ids=[self.entity_id] if self.entity_id else None) + 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."""