Python SDK Integration

1
import typing as t
2
from typing import List, cast
3
4
# Import from your target framework
5
from targetframework import FrameworkTool, SomeFrameworkClass
6
7
# Import from Composio
8
from composio import ActionType, AppType, TagType
9
from composio.tools import ComposioToolSet as BaseComposioToolSet
10
from composio.tools.toolset import ProcessorsType
11
from composio.utils.pydantic import parse_pydantic_error
12
from composio.utils.shared import get_signature_format_from_schema_params
13
14
class ComposioToolSet(BaseComposioToolSet):
15
    """
16
    Composio toolset for [Your Framework] integration.
17
    
18
    Add usage examples here to help users understand how to use your integration.
19
    """
20
    
21
    def __init__(self, *args, **kwargs):
22
        # Initialize with your framework-specific settings
23
        super().__init__(*args, **kwargs, 
24
                         runtime="your_framework_name",
25
                         description_char_limit=1024,
26
                         action_name_char_limit=64)
27
    
28
    def _wrap_tool(
29
        self,
30
        schema: t.Dict[str, t.Any],
31
        entity_id: t.Optional[str] = None,
32
    ) -> FrameworkTool:
33
        """Convert a Composio tool schema into a framework-specific tool."""
34
        action = schema["name"]
35
        description = schema["description"]
36
        schema_params = schema["parameters"]
37
        
38
        # Implementation specific to your framework
39
        # This is where you adapt the Composio schema to your framework's format
40
        
41
        # Example implementation (modify for your framework):
42
        tool = FrameworkTool(
43
            name=action,
44
            description=description,
45
            # Transform schema_params to match your framework's format
46
            parameters=self._adapt_parameters(schema_params),
47
            # Create a wrapper function that calls Composio's execute_action
48
            function=lambda **kwargs: self.execute_action(
49
                action=action,
50
                params=kwargs,
51
                entity_id=entity_id or self.entity_id,
52
            )
53
        )
54
        
55
        return tool
56
    
57
    def get_tools(
58
        self,
59
        actions: t.Optional[t.Sequence[ActionType]] = None,
60
        apps: t.Optional[t.Sequence[AppType]] = None,
61
        tags: t.Optional[t.List[TagType]] = None,
62
        entity_id: t.Optional[str] = None,
63
        *,
64
        processors: t.Optional[ProcessorsType] = None,
65
        check_connected_accounts: bool = True,
66
    ) -> List[FrameworkTool]:
67
        """
68
        Get Composio tools as your framework's tool objects.
69
        
70
        Args:
71
            actions: List of specific actions to get
72
            apps: List of apps to get tools from
73
            tags: Filter tools by tags
74
            entity_id: Entity ID to use for tool execution
75
            processors: Optional request/response processors
76
            check_connected_accounts: Whether to check for connected accounts
77
            
78
        Returns:
79
            A list of framework-compatible tools
80
        """
81
        # Validate and prepare
82
        self.validate_tools(apps=apps, actions=actions, tags=tags)
83
        if processors is not None:
84
            self._processor_helpers.merge_processors(processors)
85
        
86
        # Get action schemas from Composio
87
        tools = [
88
            self._wrap_tool(
89
                schema=tool.model_dump(exclude_none=True),
90
                entity_id=entity_id or self.entity_id,
91
            )
92
            for tool in self.get_action_schemas(
93
                actions=actions,
94
                apps=apps,
95
                tags=tags,
96
                check_connected_accounts=check_connected_accounts,
97
                _populate_requested=True,
98
            )
99
        ]
100
        
101
        return tools

1
from composio import Action, App, Tag, Trigger, WorkspaceType, action
2
3
from composio_yourframework.toolset import ComposioToolSet
4
5
__all__ = (
6
    "Action",
7
    "App",
8
    "Tag",
9
    "Trigger",
10
    "WorkspaceType",
11
    "action",
12
    "ComposioToolSet",
13
)

1
"""
2
Example demonstrating how to use the [Your Framework] integration with Composio.
3
"""
4
5
import os
6
import dotenv
7
8
# Import from your target framework
9
from targetframework import Agent, Runner
10
11
# Import from your integration
12
from composio_yourframework import App, Action, ComposioToolSet
13
14
# Set up environment
15
dotenv.load_dotenv()
16
17
def main():
18
    # Initialize your toolset
19
    toolset = ComposioToolSet()
20
    
21
    # Get specific tools
22
    tools = toolset.get_tools(actions=[Action.GITHUB_STAR_A_REPOSITORY_FOR_THE_AUTHENTICATED_USER])
23
    
24
    # Create a framework agent with the tools
25
    agent = Agent(
26
        name="Demo Agent",
27
        tools=tools,
28
        # Other framework-specific parameters
29
    )
30
    
31
    # Run the agent
32
    result = Runner.run(
33
        agent, 
34
        "Perform an action using the integrated tools"
35
    )
36
    
37
    print(result)
38
39
if __name__ == "__main__":
40
    main()

1
"""
2
Setup configuration for Composio [Your Framework] plugin
3
"""
4
5
from pathlib import Path
6
from setuptools import setup, find_packages
7
8
setup(
9
    name="composio_yourframework",
10
    version="0.1.0",
11
    author="Your Name",
12
    author_email="your.email@example.com",
13
    description="Use Composio with [Your Framework]",
14
    long_description=(Path(__file__).parent / "README.md").read_text(encoding="utf-8"),
15
    long_description_content_type="text/markdown",
16
    url="https://github.com/ComposioHQ/composio",
17
    classifiers=[
18
        "Programming Language :: Python :: 3",
19
        "License :: OSI Approved :: Apache Software License",
20
        "Operating System :: OS Independent",
21
    ],
22
    python_requires=">=3.9,<4",
23
    packages=find_packages(),
24
    install_requires=[
25
        "composio_core>=0.7.0,<0.8.0",
26
        "your-framework>=X.Y.Z",  # Replace with actual dependency
27
        "pydantic>=2.0.0",
28
        "typing-extensions>=4.0.0",
29
    ],
30
    include_package_data=True,
31
)