fix package registration

This commit is contained in:
Taylor Wilsdon
2025-07-30 10:12:09 -04:00
parent 2aea2325bc
commit 113a37d5c1
4 changed files with 12 additions and 4 deletions

View File

@@ -56,7 +56,7 @@ def main():
parser.add_argument('--single-user', action='store_true',
help='Run in single-user mode - bypass session mapping and use any credentials from the credentials directory')
parser.add_argument('--tools', nargs='*',
choices=['gmail', 'drive', 'calendar', 'docs', 'sheets', 'chat', 'forms', 'slides', 'tasks'],
choices=['gmail', 'drive', 'calendar', 'docs', 'sheets', 'chat', 'forms', 'slides', 'tasks', 'search'],
help='Specify which tools to register. If not provided, all tools are registered.')
parser.add_argument('--transport', choices=['stdio', 'streamable-http'], default='stdio',
help='Transport mode: stdio (default) or streamable-http')
@@ -92,7 +92,8 @@ def main():
'chat': lambda: __import__('gchat.chat_tools'),
'forms': lambda: __import__('gforms.forms_tools'),
'slides': lambda: __import__('gslides.slides_tools'),
'tasks': lambda: __import__('gtasks.tasks_tools')
'tasks': lambda: __import__('gtasks.tasks_tools'),
'search': lambda: __import__('gsearch.search_tools')
}
tool_icons = {
@@ -104,7 +105,8 @@ def main():
'chat': '💬',
'forms': '📝',
'slides': '🖼️',
'tasks': ''
'tasks': '',
'search': '🔍'
}
# Import specified tools or all tools if none specified