composio-autogen 0.1.49__tar.gz

composio_autogen-0.1.49/PKG-INFO

@@ -0,0 +1,20 @@
+Metadata-Version: 2.1
+Name: composio_autogen
+Version: 0.1.49
+Summary: Use Composio to get an array of tools with your Autogen agent.
+Home-page: https://github.com/SamparkAI/composio_sdk
+Author: Sawradip
+Author-email: sawradip@composio.dev
+Classifier: Programming Language :: Python :: 3
+Classifier: License :: OSI Approved :: Apache Software License
+Classifier: Operating System :: OS Independent
+Requires-Python: >=3.7
+Description-Content-Type: text/markdown
+Requires-Dist: composio_core===0.1.49
+Requires-Dist: pyautogen===0.2.19
+
+Composio <> Autogen
+Use Composio to enhance your Autogen workflows with a suite of tools.
+
+# Usage
+Within your Autogen project:

composio_autogen-0.1.49/README.md

@@ -0,0 +1,5 @@
+Composio <> Autogen
+Use Composio to enhance your Autogen workflows with a suite of tools.
+
+# Usage
+Within your Autogen project:

composio_autogen-0.1.49/composio_autogen/__init__.py

@@ -0,0 +1,2 @@
+from .autogen_toolspec import ComposioToolset
+from composio import Action, App

composio_autogen-0.1.49/composio_autogen/autogen_toolspec.py

@@ -0,0 +1,169 @@
+import types
+import logging
+from inspect import Parameter, Signature
+from typing import Union, List, Annotated
+
+import autogen
+from composio import ComposioCore, App, Action
+from pydantic import create_model, Field
+from autogen.agentchat.conversable_agent import ConversableAgent
+
+logger = logging.getLogger(__name__)
+
+schema_type_python_type_dict = {
+    'string': str,
+    'number': float,
+    'boolean': bool,
+    'integer': int
+}
+
+fallback_values = {
+    'string': "",
+    'number': 0.0,
+    'integer': 0.0,
+    'boolean': False,
+    'object': {},
+    'array': []
+}
+
+def pydantic_model_from_param_schema(param_schema):
+    fields = {}
+    param_title = param_schema['title'].replace(" ", "")
+    required_props = param_schema.get('required', [])
+    for prop_name, prop_info in param_schema['properties'].items():
+        prop_type = prop_info["type"]
+        prop_title = prop_info['title'].replace(" ", "")
+        prop_default = prop_info.get('default', fallback_values[prop_type])
+        if prop_type in schema_type_python_type_dict:
+            signature_prop_type = schema_type_python_type_dict[prop_type]
+        else:
+            signature_prop_type = pydantic_model_from_param_schema(prop_info)
+
+        if prop_name in required_props:
+            fields[prop_name] = (signature_prop_type, 
+                                Field(..., 
+                                    title=prop_title, 
+                                    description=prop_info.get('description', 
+                                                              prop_info.get('desc', 
+                                                                             prop_title))
+                                    ))
+        else:
+            fields[prop_name] = (signature_prop_type, 
+                                Field(title=prop_title, 
+                                    default=prop_default
+                                    ))
+    fieldModel = create_model(param_title, **fields)
+    return fieldModel
+        
+
+    
+
+def get_signature_format_from_schema_params(
+        schema_params
+):
+    parameters = []
+    required_params = schema_params.get('required', [])
+
+    for param_name, param_schema in schema_params['properties'].items():
+        param_type = param_schema['type']
+        param_title = param_schema['title'].replace(" ", "")
+
+        if param_type in schema_type_python_type_dict:
+            signature_param_type = schema_type_python_type_dict[param_type]
+        else:
+            signature_param_type = pydantic_model_from_param_schema(param_schema)
+
+        param_default = param_schema.get('default', fallback_values[param_type])
+        param_annotation = Annotated[signature_param_type, param_schema.get('description', 
+                                                                            param_schema.get('desc',
+                                                                                             param_title))]
+        param = Parameter(
+            name=param_name,
+            kind=Parameter.POSITIONAL_OR_KEYWORD,
+            annotation=param_annotation,
+            default=Parameter.empty if param_name in required_params else param_default 
+        )
+        parameters.append(param)
+    return parameters
+
+
+class ComposioToolset:
+    def __init__(self, caller = None, executor = None):
+        self.caller = caller
+        self.executor = executor
+        self.client =  ComposioCore()
+
+
+    def register_tools(
+            self,
+            tools: Union[App, List[App]],
+            caller: ConversableAgent = None,
+            executor: ConversableAgent = None
+        ):
+        if isinstance(tools, App):
+            tools = [tools]
+        assert caller or self.caller, "If caller hasn't been specified during initialization, has to be specified during registration"
+        assert executor or self.executor, "If executor hasn't been specified during initialization, has to be specified during registration"
+
+        action_schemas = self.client.sdk.get_list_of_actions(
+                                                apps=tools)
+        
+        for schema in action_schemas:
+            self._register_schema_to_autogen(action_schema=schema,
+                                            caller = caller if caller else self.caller,
+                                            executor = executor if executor else self.executor)
+
+            
+    def register_actions(
+            self,
+            actions: Union[Action, List[Action]],
+            caller: ConversableAgent = None,
+            executor: ConversableAgent = None
+        ):
+        if isinstance(actions, Action):
+            actions = [actions]
+
+        assert caller or self.caller, "If caller hasn't been specified during initialization, has to be specified during registration"
+        assert executor or self.executor, "If executor hasn't been specified during initialization, has to be specified during registration"
+
+        action_schemas = self.client.sdk.get_list_of_actions(
+                                                actions=actions)
+        
+        for schema in action_schemas:
+            self._register_schema_to_autogen(action_schema=schema,
+                                            caller = caller if caller else self.caller,
+                                            executor = executor if executor else self.executor)
+
+
+    def _register_schema_to_autogen(self, 
+                                    action_schema, 
+                                    caller: ConversableAgent,
+                                    executor: ConversableAgent):
+
+        name = action_schema["name"]
+        appName = action_schema["appName"]
+        description = action_schema["description"]
+
+        parameters = get_signature_format_from_schema_params(
+                                            action_schema["parameters"])
+        action_signature = Signature(parameters=parameters)
+        
+        placeholder_function = lambda **kwargs: self.client.execute_action(
+                                                    self.client.get_action_enum(name, appName), 
+                                                    kwargs)
+        action_func = types.FunctionType(
+                                    placeholder_function.__code__, 
+                                    globals=globals(), 
+                                    name=name, 
+                                    closure=placeholder_function.__closure__
+                          )
+        action_func.__signature__ = action_signature
+        action_func.__doc__ = description
+
+        autogen.agentchat.register_function(
+            action_func,
+            caller=caller,
+            executor=executor,
+            name=name,
+            description=description
+        )

composio_autogen-0.1.49/composio_autogen.egg-info/PKG-INFO

@@ -0,0 +1,20 @@
+Metadata-Version: 2.1
+Name: composio_autogen
+Version: 0.1.49
+Summary: Use Composio to get an array of tools with your Autogen agent.
+Home-page: https://github.com/SamparkAI/composio_sdk
+Author: Sawradip
+Author-email: sawradip@composio.dev
+Classifier: Programming Language :: Python :: 3
+Classifier: License :: OSI Approved :: Apache Software License
+Classifier: Operating System :: OS Independent
+Requires-Python: >=3.7
+Description-Content-Type: text/markdown
+Requires-Dist: composio_core===0.1.49
+Requires-Dist: pyautogen===0.2.19
+
+Composio <> Autogen
+Use Composio to enhance your Autogen workflows with a suite of tools.
+
+# Usage
+Within your Autogen project:

composio_autogen-0.1.49/composio_autogen.egg-info/SOURCES.txt

@@ -0,0 +1,10 @@
+README.md
+pyproject.toml
+setup.py
+composio_autogen/__init__.py
+composio_autogen/autogen_toolspec.py
+composio_autogen.egg-info/PKG-INFO
+composio_autogen.egg-info/SOURCES.txt
+composio_autogen.egg-info/dependency_links.txt
+composio_autogen.egg-info/requires.txt
+composio_autogen.egg-info/top_level.txt

composio_autogen-0.1.49/composio_autogen.egg-info/dependency_links.txt

@@ -0,0 +1 @@
+

composio_autogen-0.1.49/composio_autogen.egg-info/requires.txt

@@ -0,0 +1,2 @@
+composio_core===0.1.49
+pyautogen===0.2.19

composio_autogen-0.1.49/composio_autogen.egg-info/top_level.txt

@@ -0,0 +1 @@
+composio_autogen

composio_autogen-0.1.49/pyproject.toml

@@ -0,0 +1,11 @@
+[project]
+dynamic = ["classifiers", "version", "readme", "authors", "requires-python", "description"]
+dependencies = [
+    "composio_core===0.1.49",
+    "pyautogen===0.2.19"
+]
+name = "composio_autogen"
+
+[build-system]
+requires = [ "setuptools>=42", "wheel"]
+build-backend = "setuptools.build_meta"

composio_autogen-0.1.49/setup.cfg

@@ -0,0 +1,4 @@
+[egg_info]
+tag_build = 
+tag_date = 0
+

composio_autogen-0.1.49/setup.py

@@ -0,0 +1,31 @@
+from setuptools import setup
+import os
+
+
+def get_current_dir():
+    return os.path.dirname(os.path.realpath(__file__))
+
+
+def resolve_paths(*paths):
+    return os.path.join(*paths)
+
+
+readme_path = resolve_paths(get_current_dir(), "README.md")
+
+setup(
+    name="composio_autogen",
+    version="0.1.49",
+    author="Sawradip",
+    author_email="sawradip@composio.dev",
+    description="Use Composio to get an array of tools with your Autogen agent.",
+    long_description=open(readme_path).read(),
+    long_description_content_type="text/markdown",
+    url="https://github.com/SamparkAI/composio_sdk",
+    classifiers=[
+        "Programming Language :: Python :: 3",
+        "License :: OSI Approved :: Apache Software License",
+        "Operating System :: OS Independent",
+    ],
+    python_requires=">=3.7",
+    include_package_data=True,
+)