add option to fetch actions

Author: OskarZhang

docs/src/tools/composio_example.md

@@ -15,8 +15,8 @@ composio = ComposioAdapter(api_key="YOUR_KEY")
 ### Usage
 
 #### Connecting to Composio apps
-Creating a [Composio connection](https://docs.composio.dev/concepts/auth#connection) which connects your user to a [Contosio intergation](https://docs.composio.dev/concepts/auth#integration).
-The connection_id will be used by Contosio to identify the authenticated user.
+Creating a [Composio connection](https://docs.composio.dev/concepts/auth#connection) which connects your user to a [Composio intergation](https://docs.composio.dev/concepts/auth#integration).
+The connection_id will be used by Composio to identify the authenticated user.
 ```python
 from composio import App
 connection_id, redirect_url = composio.connect(APP.SLACK)
@@ -25,10 +25,15 @@ connection_id, redirect_url = composio.connect(APP.SLACK)
 #### Getting tools
 
 ```python
-slack_tools = composio.get_tools(App.SLACK, connection_id=connection_id)
+slack_tools = composio.get_tools_by_app(App.SLACK, connection_id=connection_id)
 
 # if no connection id is provided, defaut user will be used
-google_doc_tools = composio.get_tools(App.GOOGLEDOCS)
+google_doc_tools = composio.get_tools_by_app(App.GOOGLEDOCS)
+```
+
+You can also fetch only a subset of actions from an app
+```python
+slack_tools = composio.get_tools_by_actions([Action.SLACK_SENDS_A_MESSAGE_TO_A_SLACK_CHANNEL])
 ```
 
 #### Creating agents

examples/python/composio_example.py

@@ -1,4 +1,4 @@
-from composio import App
+from composio import App, Action
 from mainframe_orchestra import Task, Agent, OpenaiModels, Conduct, set_verbosity
 from mainframe_orchestra.adapters.composio_adapter import ComposioAdapter
 
@@ -18,7 +18,7 @@
     goal="Control the Slack app to send and receive messages",
     attributes="You have access to the API provided by Slack. You can help users send and receive messages to a channel and to a single person.",
     llm=OpenaiModels.gpt_4o,
-    tools=composio.get_tools(App.SLACK)
+    tools=composio.get_tools_by_actions([Action.SLACK_SENDS_A_MESSAGE_TO_A_SLACK_CHANNEL])
 )
 
 google_doc = Agent(
@@ -27,7 +27,7 @@
     goal="Control the Google Doc app to create and edit documents",
     attributes="You have access to the API provided by Google Doc. You can help users create and edit documents.",
     llm=OpenaiModels.gpt_4o,
-    tools=composio.get_tools(App.GOOGLEDOCS)
+    tools=composio.get_tools_by_app(App.GOOGLEDOCS)
 )
 
 conductor_agent = Agent(

packages/python/src/mainframe_orchestra/adapters/composio_adapter.py

@@ -1,4 +1,4 @@
-from composio import LogLevel, ComposioToolSet, App
+from composio import LogLevel, ComposioToolSet, App, Action, Tag
 from typing import Optional, Set, Callable, Any, Tuple
 from composio.client.collections import ActionModel
 class ComposioAdapter:
@@ -37,9 +37,9 @@ def connect(self, app: App) -> Tuple[str, str]:
 
         return connection_request.connectedAccountId, connection_request.redirectUrl
 
-    def get_tools(self, app: App, connection_id: Optional[str] = None) -> Set[Callable]:
+    def get_tools_by_app(self, app: App, connection_id: Optional[str] = None) -> Set[Callable]:
         """
-        Get the tools for an app
+        Get the tools for an Composio apps
 
         Args:
             app: The app to get the tools for
@@ -53,6 +53,23 @@ def get_tools(self, app: App, connection_id: Optional[str] = None) -> Set[Callab
         # Convert app names to App enum values if needed
         tool_schemas = self.composio_toolset.get_action_schemas(apps=[app])
         return [self._wrap(tool_schema, connection_id, entity_id) for tool_schema in tool_schemas]
+    
+    def get_tools_by_actions(self, actions: list[Action], connection_id: Optional[str] = None) -> Set[Callable]:
+        """
+        Get the tools for a list of Composio actions
+
+        Args:
+            actions: List of actions to get the tools for
+            connection_id: The connection id to use. If not provided, the default connection id will be used.
+
+        Returns:
+            A set of tools
+        """
+        entity_id = "default" if not connection_id else None
+        
+        # Convert app names to App enum values if needed
+        tool_schemas = self.composio_toolset.get_action_schemas(actions=actions)
+        return [self._wrap(tool_schema, connection_id, entity_id) for tool_schema in tool_schemas]
 
     def _wrap(
             self,