composio-gemini 0.7.17__py3-none-any.whl → 1.0.0rc1__py3-none-any.whl

composio_gemini/__init__.py

@@ -1,14 +1,6 @@
-from composio import Action, App, Tag, Trigger, WorkspaceType, action
-
-from .toolset import ComposioToolSet
+from .provider import GeminiProvider
 
 
 __all__ = (
-    "Action",
-    "App",
-    "Tag",
-    "Trigger",
-    "WorkspaceType",
-    "action",
-    "ComposioToolSet",
+    "GeminiProvider",
 )

composio_gemini/provider.py

@@ -0,0 +1,60 @@
+import types
+import typing as t
+from inspect import Signature
+
+from composio.client.types import Tool
+from composio.core.provider import AgenticProvider
+from composio.core.provider.agentic import AgenticProviderExecuteFn
+from composio.utils.openapi import function_signature_from_jsonschema
+
+
+class GeminiProvider(AgenticProvider[t.Callable, list[t.Callable]], name="gemini"):
+    """
+    Composio toolset for Google AI Python Gemini framework.
+    """
+
+    def wrap_tool(
+        self,
+        tool: Tool,
+        execute_tool: AgenticProviderExecuteFn,
+    ) -> t.Callable:
+        """Wraps composio tool as Google Genai SDK compatible function calling object."""
+
+        docstring = tool.description
+        docstring += "\nArgs:"
+        for _param, _schema in tool.input_parameters["properties"].items():  # type: ignore
+            docstring += "\n    "
+            docstring += _param + ": " + _schema.get("description", _param.title())
+
+        docstring += "\nReturns:"
+        docstring += "\n    A dictionary containing response from the action"
+
+        def _execute(**kwargs: t.Any) -> t.Dict:
+            return execute_tool(slug=tool.slug, arguments=kwargs)
+
+        function = types.FunctionType(
+            code=_execute.__code__,
+            name=tool.slug,
+            globals=globals(),
+            closure=_execute.__closure__,
+        )
+        parameters = function_signature_from_jsonschema(
+            schema=tool.input_parameters,
+            skip_default=True,
+        )
+        setattr(function, "__signature__", Signature(parameters=parameters))
+        setattr(
+            function,
+            "__annotations__",
+            {p.name: p.annotation for p in parameters} | {"return": dict},
+        )
+        function.__doc__ = docstring
+        return function
+
+    def wrap_tools(
+        self,
+        tools: t.Sequence[Tool],
+        execute_tool: AgenticProviderExecuteFn,
+    ) -> list[t.Callable]:
+        """Get composio tools wrapped as Google Genai SDK compatible function calling object."""
+        return [self.wrap_tool(tool, execute_tool) for tool in tools]

{composio_gemini-0.7.17.dist-info → composio_gemini-1.0.0rc1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: composio_gemini
-Version: 0.7.17
+Version: 1.0.0rc1
 Summary: Use Composio to get an array of tools with your Gemini agent.
 Home-page: https://github.com/ComposioHQ/composio
 Author: Composio
@@ -10,7 +10,6 @@ Classifier: License :: OSI Approved :: Apache Software License
 Classifier: Operating System :: OS Independent
 Requires-Python: >=3.9,<4
 Description-Content-Type: text/markdown
-Requires-Dist: composio_core<0.8.0,>=0.7.0
 Requires-Dist: google-genai
 Dynamic: author
 Dynamic: author-email

composio_gemini-1.0.0rc1.dist-info/RECORD

@@ -0,0 +1,6 @@
+composio_gemini/__init__.py,sha256=9S9oUYw-AB66xkTGxivUZkKGMC_gWEb9ylhNeMxT8Ik,75
+composio_gemini/provider.py,sha256=1rKIAzXasJsAuU_OL1Ht-XBAMdsGD0tKnFV0cmSR1D4,2104
+composio_gemini-1.0.0rc1.dist-info/METADATA,sha256=plNx2TxwguYHgpKm5Xvi8WSWYbVE9Pc1BCRGTGq3TeA,2652
+composio_gemini-1.0.0rc1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+composio_gemini-1.0.0rc1.dist-info/top_level.txt,sha256=OXlT1cwZ_H8zLnD-WjZOokKdqRgsq5cqhx0uCnxCikk,16
+composio_gemini-1.0.0rc1.dist-info/RECORD,,

composio_gemini/toolset.py

@@ -1,124 +0,0 @@
-import types
-import typing as t
-from inspect import Signature
-
-from composio import ActionType, AppType, TagType
-from composio.client.collections import ActionModel
-from composio.tools import ComposioToolSet as BaseComposioToolSet
-from composio.utils.openapi import function_signature_from_jsonschema
-
-
-class ComposioToolSet(
-    BaseComposioToolSet,
-    runtime="google_gemini",
-    description_char_limit=1024,
-    action_name_char_limit=64,
-):
-    """
-    Composio toolset for Google AI Python Gemini framework.
-
-    Example:
-    ```python
-        from google import genai
-        from google.genai import types
-
-        from composio_gemini import Action, ComposioToolSet
-
-
-        # Create composio client
-        toolset = ComposioToolSet()
-
-        # Create google client
-        client = genai.Client()
-
-        # Create genai client config
-        config = types.GenerateContentConfig(
-            tools=toolset.get_tools(  # type: ignore
-                actions=[
-                    Action.GITHUB_STAR_A_REPOSITORY_FOR_THE_AUTHENTICATED_USER,
-                ]
-            )
-        )
-        # Use the chat interface.
-        chat = client.chats.create(model="gemini-2.0-flash", config=config)
-        response = chat.send_message(
-            "Can you star composiohq/composio repository on github",
-        )
-        print(response.text)
-    ```
-    """
-
-    def _wrap_tool(
-        self,
-        schema: ActionModel,
-        entity_id: t.Optional[str] = None,
-        skip_default: bool = False,
-    ) -> t.Callable:
-        """Wraps composio tool as Google Genai SDK compatible function calling object."""
-
-        docstring = schema.description
-        docstring += "\nArgs:"
-        for _param, _schema in schema.parameters.properties.items():
-            docstring += "\n    "
-            docstring += _param + ": " + _schema.get("description", _param.title())
-
-        docstring += "\nReturns:"
-        docstring += "\n    A dictionary containing response from the action"
-
-        def _execute(**kwargs: t.Any) -> t.Dict:
-            return self.execute_action(
-                action=schema.name,
-                params=kwargs,
-                entity_id=entity_id,
-            )
-
-        function = types.FunctionType(
-            code=_execute.__code__,
-            name=schema.name,
-            globals=globals(),
-            closure=_execute.__closure__,
-        )
-        parameters = function_signature_from_jsonschema(
-            schema=schema.parameters.model_dump(exclude_none=True),
-            skip_default=skip_default,
-        )
-        setattr(function, "__signature__", Signature(parameters=parameters))
-        setattr(
-            function,
-            "__annotations__",
-            {p.name: p.annotation for p in parameters} | {"return": dict},
-        )
-        function.__doc__ = docstring
-        return function
-
-    def get_tools(
-        self,
-        actions: t.Optional[t.Sequence[ActionType]] = None,
-        apps: t.Optional[t.Sequence[AppType]] = None,
-        tags: t.Optional[t.List[TagType]] = None,
-        *,
-        entity_id: t.Optional[str] = None,
-        skip_default: bool = False,
-        check_connected_accounts: bool = True,
-    ) -> t.List[t.Callable]:
-        """
-        Get composio tools wrapped as Google Genai SDK compatible function calling object.
-
-        :param actions: List of actions to wrap
-        :param apps: List of apps to wrap
-        :param tags: Filter the apps by given tags
-        :param entity_id: Entity ID for the function wrapper
-
-        :return: Composio tools wrapped as python callable
-        """
-        self.validate_tools(apps=apps, actions=actions, tags=tags)
-        return [
-            self._wrap_tool(schema=tool, entity_id=entity_id, skip_default=skip_default)
-            for tool in self.get_action_schemas(
-                actions=actions,
-                apps=apps,
-                tags=tags,
-                _populate_requested=True,
-                check_connected_accounts=check_connected_accounts,
-            )
-        ]

composio_gemini-0.7.17.dist-info/RECORD

@@ -1,6 +0,0 @@
-composio_gemini/__init__.py,sha256=6sKzEl6ZrosTaUQ3SiE0qioM2gD2izRvjU4DerbuAN8,233
-composio_gemini/toolset.py,sha256=QxkIiJtKjcPAZV6siCGhGVhEahzkmjRn88DIav8dPXg,4041
-composio_gemini-0.7.17.dist-info/METADATA,sha256=yqWl74YQqKQ4Nn3CLA9A-GRbne5_JhBAUp5TArXxSSg,2693
-composio_gemini-0.7.17.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-composio_gemini-0.7.17.dist-info/top_level.txt,sha256=OXlT1cwZ_H8zLnD-WjZOokKdqRgsq5cqhx0uCnxCikk,16
-composio_gemini-0.7.17.dist-info/RECORD,,