composio-gemini 0.7.12__py3-none-any.whl → 0.7.14__py3-none-any.whl

composio_gemini/toolset.py

@@ -5,7 +5,7 @@ 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.shared import get_signature_format_from_schema_params
+from composio.utils.openapi import function_signature_from_jsonschema
 
 
 class ComposioToolSet(
@@ -52,6 +52,7 @@ class ComposioToolSet(
         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."""
 
@@ -77,10 +78,9 @@ class ComposioToolSet(
             globals=globals(),
             closure=_execute.__closure__,
         )
-        parameters = get_signature_format_from_schema_params(
-            schema_params=schema.parameters.model_dump(
-                exclude_none=True,
-            ),
+        parameters = function_signature_from_jsonschema(
+            schema=schema.parameters.model_dump(exclude_none=True),
+            skip_default=skip_default,
         )
         setattr(function, "__signature__", Signature(parameters=parameters))
         setattr(
@@ -96,7 +96,9 @@ class ComposioToolSet(
         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]:
         """
@@ -111,7 +113,7 @@ class ComposioToolSet(
         """
         self.validate_tools(apps=apps, actions=actions, tags=tags)
         return [
-            self._wrap_tool(schema=tool, entity_id=entity_id)
+            self._wrap_tool(schema=tool, entity_id=entity_id, skip_default=skip_default)
             for tool in self.get_action_schemas(
                 actions=actions,
                 apps=apps,

{composio_gemini-0.7.12.dist-info → composio_gemini-0.7.14.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: composio_gemini
-Version: 0.7.12
+Version: 0.7.14
 Summary: Use Composio to get an array of tools with your Gemini agent.
 Home-page: https://github.com/ComposioHQ/composio
 Author: Composio

composio_gemini-0.7.14.dist-info/RECORD

@@ -0,0 +1,6 @@
+composio_gemini/__init__.py,sha256=6sKzEl6ZrosTaUQ3SiE0qioM2gD2izRvjU4DerbuAN8,233
+composio_gemini/toolset.py,sha256=QxkIiJtKjcPAZV6siCGhGVhEahzkmjRn88DIav8dPXg,4041
+composio_gemini-0.7.14.dist-info/METADATA,sha256=VDUH5PY7-LOK5h0_LUAKdMQwZhG5L9Ck2H2EmV-deco,2693
+composio_gemini-0.7.14.dist-info/WHEEL,sha256=CmyFI0kx5cdEMTLiONQRbGQwjIoR1aIYB7eCAQ4KPJ0,91
+composio_gemini-0.7.14.dist-info/top_level.txt,sha256=OXlT1cwZ_H8zLnD-WjZOokKdqRgsq5cqhx0uCnxCikk,16
+composio_gemini-0.7.14.dist-info/RECORD,,

composio_gemini-0.7.12.dist-info/RECORD

@@ -1,6 +0,0 @@
-composio_gemini/__init__.py,sha256=6sKzEl6ZrosTaUQ3SiE0qioM2gD2izRvjU4DerbuAN8,233
-composio_gemini/toolset.py,sha256=GatnCWLKL6b9qEH-lug_DJ6ybHGwJruaqDPRgNDwklA,3939
-composio_gemini-0.7.12.dist-info/METADATA,sha256=WQfPR6HMPY6GmLdKChD8-RCg7bJwOs9wZw9EPAT2PJQ,2693
-composio_gemini-0.7.12.dist-info/WHEEL,sha256=CmyFI0kx5cdEMTLiONQRbGQwjIoR1aIYB7eCAQ4KPJ0,91
-composio_gemini-0.7.12.dist-info/top_level.txt,sha256=OXlT1cwZ_H8zLnD-WjZOokKdqRgsq5cqhx0uCnxCikk,16
-composio_gemini-0.7.12.dist-info/RECORD,,