Fix Composio CI type surfaces

Author: jahooma

packages/agent-runtime/src/tools/handlers/list.ts

@@ -52,7 +52,9 @@ import type { ToolName } from '@codebuff/common/tools/constants'
  *   - Any additional arguments for the tool
  * - Returns a promise that will be awaited
  */
-export const codebuffToolHandlers = {
+export const codebuffToolHandlers: {
+  [K in ToolName]: CodebuffToolHandlerFunction<K>
+} = {
   add_message: handleAddMessage,
   add_subgoal: handleAddSubgoal,
   apply_patch: handleApplyPatch,
@@ -92,6 +94,4 @@ export const codebuffToolHandlers = {
   web_search: handleWebSearch,
   write_file: handleWriteFile,
   write_todos: handleWriteTodos,
-} satisfies {
-  [K in ToolName]: CodebuffToolHandlerFunction<K>
 }

packages/agent-runtime/src/tools/handlers/tool/composio.ts

@@ -2,8 +2,10 @@ import type { ComposioMetaToolName } from '@codebuff/common/constants/composio'
 import type { CodebuffToolOutput } from '@codebuff/common/tools/list'
 import type { CodebuffToolHandlerFunction } from '../handler-function-type'
 
-function makeComposioHandler<T extends ComposioMetaToolName>() {
-  return (async ({ toolCall, requestClientToolCall }) => {
+function makeComposioHandler<
+  T extends ComposioMetaToolName,
+>(): CodebuffToolHandlerFunction<T> {
+  return async ({ toolCall, requestClientToolCall }) => {
     if (!requestClientToolCall) {
       return {
         output: [
@@ -22,14 +24,14 @@ function makeComposioHandler<T extends ComposioMetaToolName>() {
         toolCall,
       )) as CodebuffToolOutput<T>,
     }
-  }) satisfies CodebuffToolHandlerFunction<T>
+  }
 }
 
-export const handleComposioManageConnections =
+export const handleComposioManageConnections: CodebuffToolHandlerFunction<'COMPOSIO_MANAGE_CONNECTIONS'> =
   makeComposioHandler<'COMPOSIO_MANAGE_CONNECTIONS'>()
-export const handleComposioMultiExecute =
+export const handleComposioMultiExecute: CodebuffToolHandlerFunction<'COMPOSIO_MULTI_EXECUTE_TOOL'> =
   makeComposioHandler<'COMPOSIO_MULTI_EXECUTE_TOOL'>()
-export const handleComposioSearchTools =
+export const handleComposioSearchTools: CodebuffToolHandlerFunction<'COMPOSIO_SEARCH_TOOLS'> =
   makeComposioHandler<'COMPOSIO_SEARCH_TOOLS'>()
-export const handleComposioGetToolSchemas =
+export const handleComposioGetToolSchemas: CodebuffToolHandlerFunction<'COMPOSIO_GET_TOOL_SCHEMAS'> =
   makeComposioHandler<'COMPOSIO_GET_TOOL_SCHEMAS'>()

sdk/src/run.ts

@@ -75,9 +75,7 @@ const wrapContentForUserMessage = (
 }
 
 type OverrideToolHandlers = {
-  [K in PublishedClientToolName]?: (
-    input: Record<string, unknown>,
-  ) => Promise<ToolResultOutput[]>
+  [K in PublishedClientToolName]?: (input: any) => Promise<ToolResultOutput[]>
 } & {
   // Include read_files separately, since it has a different signature.
   read_files?: (input: {