Add server-side Composio managed auth

Author: jahooma

agents/base2/base2.ts

@@ -1,4 +1,5 @@
 import { buildArray } from '@codebuff/common/util/array'
+import { COMPOSIO_META_TOOL_NAMES } from '@codebuff/common/constants/composio'
 import {
   FREEBUFF_GEMINI_THINKER_AGENT_ID,
   FREEBUFF_GEMINI_THINKER_INSTRUCTIONS_PROMPT,
@@ -113,6 +114,7 @@ export function createBase2(
       'set_output',
       'list_directory',
       'glob',
+      ...COMPOSIO_META_TOOL_NAMES,
     ),
     spawnableAgents: buildArray(
       !isMax && 'file-picker',
@@ -157,6 +159,7 @@ Current date: ${formatCurrentDate(new Date())}.
 - **Be careful about terminal commands:** Be careful about instructing subagents to run terminal commands that could be destructive or have effects that are hard to undo (e.g. git push, git commit, running any scripts -- especially ones that could alter production environments (!), installing packages globally, etc). Don't run any of these effectful commands unless the user explicitly asks you to.
 - **Do what the user asks:** If the user asks you to do something, even running a risky terminal command, do it.
 - **Don't use set_output:** The set_output tool is for spawned subagents to report results. Don't use it yourself.
+- **External apps:** When Composio tools are available and the user asks to work with connected apps or services like Gmail, Google Calendar, GitHub, Slack, Linear, or Notion, use them to search for the right app tools, help the user connect their account, and execute the requested action.
 
 # Code Editing Mandates
 

agents/types/secret-agent-definition.ts

@@ -1,5 +1,6 @@
 import type { AgentDefinition } from './agent-definition'
 import type * as Tools from './tools'
+import type { ComposioMetaToolName } from '@codebuff/common/constants/composio'
 export type { Tools }
 
 export type AllToolNames =
@@ -9,9 +10,12 @@ export type AllToolNames =
   | 'create_plan'
   | 'spawn_agent_inline'
   | 'update_subgoal'
+  | ComposioMetaToolName
 
-export interface SecretAgentDefinition
-  extends Omit<AgentDefinition, 'toolNames'> {
+export interface SecretAgentDefinition extends Omit<
+  AgentDefinition,
+  'toolNames'
+> {
   /** Tools this agent can use. */
   toolNames?: AllToolNames[]
 }

bun.lock

@@ -0,0 +1,12 @@
+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
+
+export type ComposioMetaToolName = (typeof COMPOSIO_META_TOOL_NAMES)[number]

common/src/constants/composio.ts

@@ -0,0 +1,10 @@
+CREATE TABLE "composio_session" (
+	"user_id" text PRIMARY KEY NOT NULL,
+	"session_id" text NOT NULL,
+	"created_at" timestamp with time zone DEFAULT now() NOT NULL,
+	"updated_at" timestamp with time zone DEFAULT now() NOT NULL,
+	CONSTRAINT "composio_session_session_id_unique" UNIQUE("session_id")
+);
+--> statement-breakpoint
+ALTER TABLE "composio_session" ADD CONSTRAINT "composio_session_user_id_user_id_fk" FOREIGN KEY ("user_id") REFERENCES "public"."user"("id") ON DELETE cascade ON UPDATE no action;--> statement-breakpoint
+CREATE INDEX "idx_composio_session_session_id" ON "composio_session" USING btree ("session_id");