wehub-resource-sync
48 lines
1.2 KiB
Python
48 lines
1.2 KiB
Python
"""
|
|
Tool Router - Session Update Example
|
|
|
|
Demonstrates using session.update() to modify a session's configuration
|
|
after creation — e.g. adding toolkits, changing workbench settings, or
|
|
updating preload config without creating a new session.
|
|
"""
|
|
|
|
import os
|
|
|
|
from composio import Composio
|
|
|
|
composio = Composio(base_url=os.environ.get("COMPOSIO_BASE_URL"))
|
|
|
|
# Create a session with gmail only
|
|
session = composio.create(
|
|
user_id="session-update-demo",
|
|
toolkits=["gmail"],
|
|
manage_connections=False,
|
|
)
|
|
|
|
print(f"Session created: {session.session_id}")
|
|
print(f"Preload: {session.preload}")
|
|
|
|
# Update: add github toolkit, enable workbench, and preload a tool
|
|
session.update(
|
|
toolkits={"enable": ["gmail", "github"]},
|
|
workbench={"enable": True, "sandbox_size": "medium"},
|
|
preload={"tools": ["GITHUB_CREATE_AN_ISSUE"]},
|
|
)
|
|
|
|
print("\nAfter update:")
|
|
print(f"Preload: {session.preload}")
|
|
|
|
# Update: disable workbench entirely
|
|
session.update(
|
|
workbench={"enable": False},
|
|
)
|
|
|
|
print("\nAfter disabling workbench: done")
|
|
|
|
# Update: clear manage_connections by passing None
|
|
session.update(
|
|
manage_connections=None,
|
|
)
|
|
|
|
print("\nAfter clearing manage_connections: done")
|