Make create session return type pythonic

python/composio/core/models/tool_router.py

@@ -8,8 +8,8 @@
 from __future__ import annotations
 
 import typing as t
+from types import SimpleNamespace
 
-import typing_extensions as te
 from composio_client.types.tool_router_create_session_params import ConfigToolkit
 
 from composio.client import HttpClient
@@ -19,11 +19,19 @@
 # Data Types
 
 
-class ToolRouterSession(te.TypedDict):
-    """Tool router session response."""
+class ToolRouterSession(SimpleNamespace):
+    """Tool router session response with both dot notation and dict-style access."""
 
-    session_id: str  # Unique session identifier
-    url: str  # Chat session MCP URL
+    def __init__(self, session_id: str, url: str):
+        super().__init__(session_id=session_id, url=url)
+
+    def __getitem__(self, key: str) -> str:
+        """Support dict-style access like session['url']"""
+        return getattr(self, key)
+
+    def __setitem__(self, key: str, value: str) -> None:
+        """Support dict-style assignment like session['url'] = 'new_url'"""
+        setattr(self, key, value)
 
 
 class ToolRouter(Resource):
@@ -73,9 +81,11 @@ def create_session(
             ...     manually_manage_connections=False
             ... )
             >>>
-            >>> # Access session details
-            >>> print(session['session_id'])
-            >>> print(session['url'])
+            >>> # Access session details (both ways work!)
+            >>> print(session.session_id)      # Dot notation (preferred)
+            >>> print(session['session_id'])   # Dict-style access
+            >>> print(session.url)             # Dot notation (preferred)
+            >>> print(session['url'])          # Dict-style access
         """
         try:
             # Normalize toolkits to the format expected by the API

python/composio/core/models/tools.py

@@ -3,8 +3,8 @@
 import functools
 import typing as t
 
-from composio_client import omit
 import typing_extensions as te
+from composio_client import omit
 
 from composio.client import HttpClient
 from composio.client.types import (

python/composio/core/models/triggers.py

@@ -8,9 +8,9 @@
 import uuid
 from concurrent.futures import ThreadPoolExecutor
 from unittest import mock
-from composio_client import omit
 
 import typing_extensions as te
+from composio_client import omit
 from pysher import Pusher
 from pysher.channel import Channel as PusherChannel
 from pysher.connection import Connection as PusherConnection

python/providers/google_adk/google_adk_demo.py

@@ -53,8 +53,10 @@
     new_message=content,
 )
 for event in events:
-    if (event.is_final_response() 
-        and event.content is not None 
-        and event.content.parts 
-        and len(event.content.parts) > 0):
+    if (
+        event.is_final_response()
+        and event.content is not None
+        and event.content.parts
+        and len(event.content.parts) > 0
+    ):
         print("Agent Response: ", event.content.parts[0].text)