Sync public snapshot from freebuff-private

Source: CodebuffAI/freebuff-private@15cb1ec9a6c680d981325868ce8211025697b637

Author: github-actions[bot]

agents/base-chat.ts

@@ -5,10 +5,11 @@ import { publisher } from './constants'
 import type { SecretAgentDefinition } from './types/secret-agent-definition'
 
 /**
- * Conversational agent behind freebuff.com/chat. Runs with no filesystem and
- * no direct tools, but can spawn researcher-web to look things up on the
- * live internet. The chat server may override `model` per request (DeepSeek
- * Flash vs Pro for full-access users).
+ * Conversational agent behind freebuff.com/chat. Runs with no filesystem, but
+ * can spawn researcher-web to look things up on the live internet and call
+ * gravity_index to recommend third-party developer services. The chat server
+ * may override `model` per request (DeepSeek Flash vs Pro for full-access
+ * users).
  */
 const definition: SecretAgentDefinition = {
   id: 'base-chat',
@@ -23,12 +24,14 @@ const definition: SecretAgentDefinition = {
     },
   },
   outputMode: 'last_message',
-  toolNames: ['spawn_agents'],
+  toolNames: ['spawn_agents', 'gravity_index'],
   spawnableAgents: ['researcher-web'],
 
   systemPrompt: `You are Freebuff Chat, a friendly, sharp assistant made by Freebuff (freebuff.com), the home of free AI coding tools. You are chatting with a user in a web interface that renders markdown.`,
   instructionsPrompt: `Be direct and helpful. Use markdown when it improves clarity (code blocks, lists, tables), and keep answers as short as they can be while fully answering the question.
 
+When the user is choosing a third-party developer service (database, auth, payments, hosting, email, monitoring, analytics, AI APIs, storage, CMS, search, etc.) or asks what provider to use for something, use the gravity_index tool instead of answering from memory: \`search\` with a query that includes their stack and constraints when they want a recommendation, or \`browse\`/\`list_categories\`/\`get_service\` to explore options. Ground your answer in the result. When a search result includes a tracked setup link (\`credential_request.setup_url\` or \`click_url\`), present that exact URL prominently as a markdown link like "Get your {service} API key" — never swap in the vendor homepage for it. Since you can't edit the user's files, share the relevant setup steps and env vars in chat instead of trying to install anything.
+
 You can search the live internet by spawning the researcher-web agent. Spawn it whenever the answer depends on current or recent information (news, prices, releases, versions, schedules, scores, docs), whenever the user asks you to look something up, or whenever you are not confident in your knowledge. Give it a focused question; you can spawn several in parallel for independent questions. After it reports back, answer the user in your own words and cite source URLs when useful. Don't spawn it for questions you can already answer well (general knowledge, coding help, writing, math).
 
 You do not have access to the user's files or a filesystem — if asked to do something that requires those, say so briefly and help with what you can instead.`,

packages/agent-runtime/src/__tests__/gravity-index-tool.test.ts

@@ -148,6 +148,60 @@ describe('gravity_index tool', () => {
     )
   })
 
+  test('tags base-chat traffic with the freebuff_chat surface', async () => {
+    const spy = spyOn(webApi, 'callGravityIndexAPI').mockResolvedValue({
+      result: { search_id: 'search-1' },
+    })
+
+    mockAgentStream([
+      createToolCallChunk('gravity_index', {
+        action: 'search',
+        query: 'transactional email for Next.js',
+      }),
+      createToolCallChunk('end_turn', {}),
+    ])
+
+    const fileContext = {
+      ...mockFileContext,
+      agentTemplates: {
+        'base-chat': {
+          ...gravityTestAgent,
+          id: 'base-chat',
+          displayName: 'Freebuff Chat',
+        },
+      },
+    }
+    const sessionState = getInitialSessionState(fileContext)
+    const agentState = {
+      ...sessionState.mainAgentState,
+      agentType: 'base-chat',
+    }
+    const { agentTemplates } = assembleLocalAgentTemplates({
+      ...agentRuntimeImpl,
+      fileContext,
+    })
+
+    await runAgentStep({
+      ...runAgentStepBaseParams,
+      agentType: 'base-chat',
+      fileContext,
+      localAgentTemplates: agentTemplates,
+      agentTemplate: agentTemplates['base-chat'],
+      agentState,
+      prompt: 'Find an email provider',
+    })
+
+    expect(spy).toHaveBeenCalledWith(
+      expect.objectContaining({
+        input: expect.objectContaining({
+          metadata: expect.objectContaining({
+            surface: 'freebuff_chat',
+          }),
+        }),
+      }),
+    )
+  })
+
   test('stores recommendation and setup URL in tool output', async () => {
     spyOn(webApi, 'callGravityIndexAPI').mockResolvedValue({
       result: {

packages/agent-runtime/src/tools/handlers/tool/gravity-index.ts

@@ -7,6 +7,7 @@ import type {
   CodebuffToolCall,
   CodebuffToolOutput,
 } from '@codebuff/common/tools/list'
+import type { AgentTemplate } from '@codebuff/common/types/agent-template'
 import type { ClientEnv, CiEnv } from '@codebuff/common/types/contracts/env'
 import type { JSONObject, JSONValue } from '@codebuff/common/types/json'
 import type { Logger } from '@codebuff/common/types/contracts/logger'
@@ -24,9 +25,15 @@ const omitUndefined = (value: Record<string, JSONValue | undefined>) => {
 const isJSONObject = (value: JSONValue | undefined): value is JSONObject =>
   !!value && typeof value === 'object' && !Array.isArray(value)
 
+/** Gravity attribution surface, so clicks/conversions are attributable to the
+ *  product the request came from rather than all reading as CLI traffic. */
+const gravitySurface = (agentTemplate: { id: string }): string =>
+  agentTemplate.id === 'base-chat' ? 'freebuff_chat' : 'codebuff_cli'
+
 export const handleGravityIndex = (async (params: {
   previousToolCallFinished: Promise<void>
   toolCall: CodebuffToolCall<'gravity_index'>
+  agentTemplate: AgentTemplate
   logger: Logger
   apiKey: string
 
@@ -47,6 +54,7 @@ export const handleGravityIndex = (async (params: {
   const {
     previousToolCallFinished,
     toolCall,
+    agentTemplate,
     agentStepId,
     apiKey,
     clientSessionId,
@@ -84,7 +92,7 @@ export const handleGravityIndex = (async (params: {
     const metadata = {
       ...existingMetadata,
       ...omitUndefined({
-        surface: 'codebuff_cli',
+        surface: gravitySurface(agentTemplate),
         tool_call_id: toolCall.toolCallId,
         agent_step_id: agentStepId,
         fingerprint_id: fingerprintId,