Consolidate Gravity Index tool actions

Author: jahooma

agents/base2/base2.ts

@@ -70,7 +70,7 @@ export function createBase2(
       'set_output',
       'list_directory',
       'glob',
-      'gravity_index_search',
+      'gravity_index',
     ),
     spawnableAgents: buildArray(
       !isMax && 'file-picker',
@@ -116,7 +116,7 @@ export function createBase2(
 
 - **Conventions:** Rigorously adhere to existing project conventions when reading or modifying code. Analyze surrounding code, tests, and configuration first.
 - **Libraries/Frameworks:** NEVER assume a library/framework is available or appropriate. Verify its established usage within the project (check imports, configuration files like 'package.json', 'Cargo.toml', 'requirements.txt', 'build.gradle', etc., or observe neighboring files) before employing it.
-- **Third-party services:** When the user needs an external developer service (database, auth, email, payments, AI provider, vector store, observability, hosting, etc.), use the gravity_index_search tool before recommending or integrating a provider. Present its conversion_url prominently as "Get your {service.name} API key" when credentials are needed.
+- **Third-party services:** When the user needs an external developer service (database, auth, email, payments, AI provider, vector store, observability, hosting, etc.), use the gravity_index tool before recommending or integrating a provider. Present its conversion_url prominently as "Get your {service.name} API key" when credentials are needed.
 - **Style & Structure:** Mimic the style (formatting, naming), structure, framework choices, typing, and architectural patterns of existing code in the project.
 - **Idiomatic Changes:** When editing, understand the local context (imports, functions/classes) to ensure your changes integrate naturally and idiomatically.
 - **Simplicity & Minimalism:** You should make as few changes as possible to the codebase to address the user's request. Only do what the user has asked for and no more. When modifying existing code, assume every line of code has a purpose and is there for a reason. Do not change the behavior of code except in the most minimal way to accomplish the user's request.

agents/context-pruner.ts

@@ -291,11 +291,13 @@ const definition: AgentDefinition = {
           const query = input.query as string | undefined
           return query ? `Web search: "${query}"` : 'Web search'
         }
-        case 'gravity_index_search': {
+        case 'gravity_index': {
           const query = input.query as string | undefined
-          return query
-            ? `Gravity Index search: "${query}"`
-            : 'Gravity Index search'
+          const action = input.action as string | undefined
+          if (query) {
+            return `Gravity Index ${action ?? 'search'}: "${query}"`
+          }
+          return action ? `Gravity Index ${action}` : 'Gravity Index'
         }
         case 'read_docs': {
           const libraryTitle = input.libraryTitle as string | undefined

agents/e2e/gravity-index.e2e.test.ts

@@ -12,7 +12,7 @@ import type { PrintModeEvent } from '@codebuff/common/types/print-mode'
 
 describe('Gravity Index SDK E2E', () => {
   it(
-    'base2-free uses gravity_index_search for third-party service selection',
+    'base2-free uses gravity_index for third-party service selection',
     async () => {
       const apiKey = process.env[API_KEY_ENV_VAR]
       if (!apiKey) {
@@ -58,7 +58,7 @@ describe('Gravity Index SDK E2E', () => {
         expect(
           toolCalls.some(
             (event) =>
-              'toolName' in event && event.toolName === 'gravity_index_search',
+              'toolName' in event && event.toolName === 'gravity_index',
           ),
         ).toBe(true)
 

agents/types/tools.ts

@@ -9,7 +9,7 @@ export type ToolName =
   | 'end_turn'
   | 'find_files'
   | 'glob'
-  | 'gravity_index_search'
+  | 'gravity_index'
   | 'list_directory'
   | 'lookup_agent_info'
   | 'propose_str_replace'
@@ -42,7 +42,7 @@ export interface ToolParamsMap {
   end_turn: EndTurnParams
   find_files: FindFilesParams
   glob: GlobParams
-  gravity_index_search: GravityIndexSearchParams
+  gravity_index: GravityIndexParams
   list_directory: ListDirectoryParams
   lookup_agent_info: LookupAgentInfoParams
   propose_str_replace: ProposeStrReplaceParams
@@ -159,12 +159,45 @@ export interface GlobParams {
 }
 
 /**
- * Search the Gravity Index for a recommended developer service.
+ * Search, browse, inspect, or report integrations in the Gravity Index.
  */
-export interface GravityIndexSearchParams {
-  /** A natural-language description of the third-party developer service the user needs, including stack and constraints when known. */
-  query: string
-}
+export type GravityIndexParams =
+  | {
+      /** Search for the best service recommendation. */
+      action: 'search'
+      /** What the user needs, including stack, constraints, and required capabilities when known. */
+      query: string
+      /** Continue a previous Gravity Index search as a follow-up. */
+      search_id?: string
+      /** Optional structured context about the project, stack, or constraints. */
+      context?: Record<string, any>
+    }
+  | {
+      /** Browse catalog services by category and/or keyword. */
+      action: 'browse'
+      /** Optional category filter, e.g. Database, Auth, Payments, Hosting, Email, AI. */
+      category?: string
+      /** Optional keyword filter, e.g. sendgrid or postgres. */
+      q?: string
+    }
+  | {
+      /** List every category with service counts. */
+      action: 'list_categories'
+    }
+  | {
+      /** Fetch full detail for a single service by slug. */
+      action: 'get_service'
+      /** Service slug, e.g. supabase, stripe, sendgrid. */
+      slug: string
+    }
+  | {
+      /** Report that an integration from a prior search was completed. */
+      action: 'report_integration'
+      /** search_id from the earlier search result. */
+      search_id: string
+      /** Slug of the service that was actually integrated. */
+      integrated_slug: string
+    }
 
 /**
  * List files and directories in the specified path. Returns separate arrays of file names and directory names.

common/src/constants/analytics-events.ts

@@ -124,10 +124,10 @@ export enum AnalyticsEvent {
   DOCS_SEARCH_INSUFFICIENT_CREDITS = 'api.docs_search_insufficient_credits',
   DOCS_SEARCH_ERROR = 'api.docs_search_error',
 
-  GRAVITY_INDEX_SEARCH_REQUEST = 'api.gravity_index_search_request',
-  GRAVITY_INDEX_SEARCH_AUTH_ERROR = 'api.gravity_index_search_auth_error',
-  GRAVITY_INDEX_SEARCH_VALIDATION_ERROR = 'api.gravity_index_search_validation_error',
-  GRAVITY_INDEX_SEARCH_ERROR = 'api.gravity_index_search_error',
+  GRAVITY_INDEX_REQUEST = 'api.gravity_index_request',
+  GRAVITY_INDEX_AUTH_ERROR = 'api.gravity_index_auth_error',
+  GRAVITY_INDEX_VALIDATION_ERROR = 'api.gravity_index_validation_error',
+  GRAVITY_INDEX_ERROR = 'api.gravity_index_error',
 
   // Web - Feedback API
   FEEDBACK_SUBMITTED = 'api.feedback_submitted',

common/src/templates/initial-agents-dir/types/tools.ts

@@ -9,7 +9,7 @@ export type ToolName =
   | 'end_turn'
   | 'find_files'
   | 'glob'
-  | 'gravity_index_search'
+  | 'gravity_index'
   | 'list_directory'
   | 'lookup_agent_info'
   | 'propose_str_replace'
@@ -42,7 +42,7 @@ export interface ToolParamsMap {
   end_turn: EndTurnParams
   find_files: FindFilesParams
   glob: GlobParams
-  gravity_index_search: GravityIndexSearchParams
+  gravity_index: GravityIndexParams
   list_directory: ListDirectoryParams
   lookup_agent_info: LookupAgentInfoParams
   propose_str_replace: ProposeStrReplaceParams
@@ -159,12 +159,45 @@ export interface GlobParams {
 }
 
 /**
- * Search the Gravity Index for a recommended developer service.
+ * Search, browse, inspect, or report integrations in the Gravity Index.
  */
-export interface GravityIndexSearchParams {
-  /** A natural-language description of the third-party developer service the user needs, including stack and constraints when known. */
-  query: string
-}
+export type GravityIndexParams =
+  | {
+      /** Search for the best service recommendation. */
+      action: 'search'
+      /** What the user needs, including stack, constraints, and required capabilities when known. */
+      query: string
+      /** Continue a previous Gravity Index search as a follow-up. */
+      search_id?: string
+      /** Optional structured context about the project, stack, or constraints. */
+      context?: Record<string, any>
+    }
+  | {
+      /** Browse catalog services by category and/or keyword. */
+      action: 'browse'
+      /** Optional category filter, e.g. Database, Auth, Payments, Hosting, Email, AI. */
+      category?: string
+      /** Optional keyword filter, e.g. sendgrid or postgres. */
+      q?: string
+    }
+  | {
+      /** List every category with service counts. */
+      action: 'list_categories'
+    }
+  | {
+      /** Fetch full detail for a single service by slug. */
+      action: 'get_service'
+      /** Service slug, e.g. supabase, stripe, sendgrid. */
+      slug: string
+    }
+  | {
+      /** Report that an integration from a prior search was completed. */
+      action: 'report_integration'
+      /** search_id from the earlier search result. */
+      search_id: string
+      /** Slug of the service that was actually integrated. */
+      integrated_slug: string
+    }
 
 /**
  * List files and directories in the specified path. Returns separate arrays of file names and directory names.

common/src/tools/constants.ts

@@ -30,7 +30,7 @@ export const toolNames = [
   'end_turn',
   'find_files',
   'glob',
-  'gravity_index_search',
+  'gravity_index',
   'list_directory',
   'lookup_agent_info',
   'propose_str_replace',
@@ -63,7 +63,7 @@ export const publishedTools = [
   'end_turn',
   'find_files',
   'glob',
-  'gravity_index_search',
+  'gravity_index',
   'list_directory',
   'lookup_agent_info',
   'propose_str_replace',

common/src/tools/list.ts

@@ -11,7 +11,7 @@ import { createPlanParams } from './params/tool/create-plan'
 import { endTurnParams } from './params/tool/end-turn'
 import { findFilesParams } from './params/tool/find-files'
 import { globParams } from './params/tool/glob'
-import { gravityIndexSearchParams } from './params/tool/gravity-index-search'
+import { gravityIndexParams } from './params/tool/gravity-index'
 import { listDirectoryParams } from './params/tool/list-directory'
 import { lookupAgentInfoParams } from './params/tool/lookup-agent-info'
 import { proposeStrReplaceParams } from './params/tool/propose-str-replace'
@@ -50,7 +50,7 @@ export const toolParams = {
   end_turn: endTurnParams,
   find_files: findFilesParams,
   glob: globParams,
-  gravity_index_search: gravityIndexSearchParams,
+  gravity_index: gravityIndexParams,
   list_directory: listDirectoryParams,
   lookup_agent_info: lookupAgentInfoParams,
   propose_str_replace: proposeStrReplaceParams,