Disable Composio workbench tools

Author: jahooma

common/src/constants/composio.ts

@@ -3,8 +3,6 @@ export const COMPOSIO_API_KEY_ENV_VAR = 'COMPOSIO_API_KEY'
 export const COMPOSIO_META_TOOL_NAMES = [
   'COMPOSIO_MANAGE_CONNECTIONS',
   'COMPOSIO_MULTI_EXECUTE_TOOL',
-  'COMPOSIO_REMOTE_BASH_TOOL',
-  'COMPOSIO_REMOTE_WORKBENCH',
   'COMPOSIO_SEARCH_TOOLS',
   'COMPOSIO_GET_TOOL_SCHEMAS',
 ] as const

sdk/src/composio.ts

@@ -17,17 +17,6 @@ const sessionIdParam = z
   .optional()
   .describe('Session ID returned by COMPOSIO_SEARCH_TOOLS, when available.')
 
-const workflowStepParams = {
-  current_step: z
-    .string()
-    .optional()
-    .describe('Short enum-style label for the current workflow step.'),
-  current_step_metric: z
-    .string()
-    .optional()
-    .describe('Progress metric such as "3/10 emails" or "0/n messages".'),
-}
-
 const composioMetaToolSchemas = {
   COMPOSIO_SEARCH_TOOLS: z
     .object({
@@ -87,31 +76,8 @@ const composioMetaToolSchemas = {
         .describe('One concise sentence explaining the execution intent.'),
       sync_response_to_workbench: z
         .boolean()
-        .describe('Use true when the response may be large or reused later.'),
-      session_id: sessionIdParam,
-      ...workflowStepParams,
-    })
-    .catchall(z.unknown()),
-  COMPOSIO_REMOTE_WORKBENCH: z
-    .object({
-      code_to_execute: z
-        .string()
-        .describe('Python code to run in the persistent remote workbench.'),
-      thought: z
-        .string()
-        .optional()
-        .describe(
-          'One concise sentence describing why the workbench is needed.',
-        ),
-      session_id: sessionIdParam,
-      ...workflowStepParams,
-    })
-    .catchall(z.unknown()),
-  COMPOSIO_REMOTE_BASH_TOOL: z
-    .object({
-      command: z
-        .string()
-        .describe('Bash command to run in the remote sandbox.'),
+        .default(false)
+        .describe('Always use false. Codebuff disables Composio workbench.'),
       session_id: sessionIdParam,
     })
     .catchall(z.unknown()),
@@ -125,11 +91,7 @@ const composioMetaToolDescriptions = {
   COMPOSIO_MANAGE_CONNECTIONS:
     'Check or initiate user authentication for external app toolkits. Use when search/execution indicates a toolkit is not connected.',
   COMPOSIO_MULTI_EXECUTE_TOOL:
-    'Execute one or more discovered Composio app tools in the current workflow session.',
-  COMPOSIO_REMOTE_WORKBENCH:
-    'Run Python in a persistent Composio workbench for bulk app workflows, large responses, or data transformations.',
-  COMPOSIO_REMOTE_BASH_TOOL:
-    'Run bash commands in the Composio remote sandbox for simple file and data processing.',
+    'Execute one or more discovered Composio app tools in the current workflow session. Do not use workbench offloading.',
 } satisfies Record<ComposioMetaToolName, string>
 
 function toJsonValue(value: unknown): JSONValue {

web/src/server/__tests__/composio.test.ts

@@ -99,7 +99,9 @@ describe('executeComposioTool', () => {
     expect(result).toEqual([{ type: 'json', value: { ok: true } }])
     expect(useSession).toHaveBeenCalledWith('stored-session')
     expect(whereDelete).toHaveBeenCalledTimes(1)
-    expect(createSession).toHaveBeenCalledWith('user-123')
+    expect(createSession).toHaveBeenCalledWith('user-123', {
+      workbench: { enable: false },
+    })
     expect(values).toHaveBeenCalledWith({
       user_id: 'user-123',
       session_id: 'fresh-session',
@@ -124,7 +126,9 @@ describe('executeComposioTool', () => {
     })
 
     expect(result).toEqual([{ type: 'json', value: { ok: true } }])
-    expect(createSession).toHaveBeenCalledWith('user-123')
+    expect(createSession).toHaveBeenCalledWith('user-123', {
+      workbench: { enable: false },
+    })
     expect(values).toHaveBeenCalledWith({
       user_id: 'user-123',
       session_id: 'losing-session',

web/src/server/composio.ts

@@ -80,6 +80,12 @@ function getComposioClient(apiKey: string): ComposioClient {
   })
 }
 
+const COMPOSIO_SESSION_CONFIG = {
+  workbench: {
+    enable: false,
+  },
+} as const
+
 async function insertSessionIfAbsent(params: {
   db: CodebuffPgDatabase
   userId: string
@@ -140,7 +146,7 @@ async function createSessionForUser(params: {
   logger: Logger
 }): Promise<CachedComposioSession> {
   const composio = getComposioClient(params.apiKey)
-  const session = await composio.create(params.userId)
+  const session = await composio.create(params.userId, COMPOSIO_SESSION_CONFIG)
   await insertSessionIfAbsent({
     db: params.db,
     userId: params.userId,