chore(web): Add telemetry for all api requests (#835)

Author: brendan-kellam

CHANGELOG.md

@@ -12,6 +12,7 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 
 ### Changed
 - Changed `/api/source` api to support fetching source code for any revision, not just revisions that are indexed by zoekt. [#829](https://github.com/sourcebot-dev/sourcebot/pull/829)
+- Added additional telemetry for api requests. [#835](https://github.com/sourcebot-dev/sourcebot/pull/835)
 
 ## [4.10.20] - 2026-01-28
 

packages/mcp/src/client.ts

@@ -35,6 +35,7 @@ export const search = async (request: SearchRequest) => {
         method: 'POST',
         headers: {
             'Content-Type': 'application/json',
+            'X-Sourcebot-Client-Source': 'mcp',
             ...(env.SOURCEBOT_API_KEY ? { 'X-Sourcebot-Api-Key': env.SOURCEBOT_API_KEY } : {})
         },
         body: JSON.stringify(request)
@@ -56,6 +57,7 @@ export const listRepos = async (queryParams: ListReposQueryParams = {}) => {
         method: 'GET',
         headers: {
             'Content-Type': 'application/json',
+            'X-Sourcebot-Client-Source': 'mcp',
             ...(env.SOURCEBOT_API_KEY ? { 'X-Sourcebot-Api-Key': env.SOURCEBOT_API_KEY } : {})
         },
     });
@@ -76,6 +78,7 @@ export const getFileSource = async (request: FileSourceRequest) => {
     const response = await fetch(url, {
         method: 'GET',
         headers: {
+            'X-Sourcebot-Client-Source': 'mcp',
             ...(env.SOURCEBOT_API_KEY ? { 'X-Sourcebot-Api-Key': env.SOURCEBOT_API_KEY } : {})
         },
     });
@@ -95,11 +98,12 @@ export const listCommits = async (queryParams: ListCommitsQueryParamsSchema) =>
         method: 'GET',
         headers: {
             'X-Org-Domain': '~',
+            'X-Sourcebot-Client-Source': 'mcp',
             ...(env.SOURCEBOT_API_KEY ? { 'X-Sourcebot-Api-Key': env.SOURCEBOT_API_KEY } : {})
         },
     });
 
     const commits = await parseResponse(response, listCommitsResponseSchema);
     const totalCount = parseInt(response.headers.get('X-Total-Count') ?? '0', 10);
     return { commits, totalCount };
-}
+}

packages/mcp/src/index.ts

@@ -101,7 +101,6 @@ server.tool(
             contextLines: env.DEFAULT_CONTEXT_LINES,
             isRegexEnabled: useRegex,
             isCaseSensitivityEnabled: caseSensitive,
-            source: 'mcp',
         });
 
         if (response.files.length === 0) {

packages/mcp/src/schemas.ts

@@ -31,7 +31,6 @@ export const searchOptionsSchema = z.object({
 
 export const searchRequestSchema = z.object({
     query: z.string(),                                // The zoekt query to execute.
-    source: z.string().optional(),                    // The source of the search request.
     ...searchOptionsSchema.shape,
 });
 

packages/web/src/app/[domain]/components/searchBar/useSuggestionsData.ts

@@ -61,7 +61,6 @@ export const useSuggestionsData = ({
             query: `file:${suggestionQuery}`,
             matches: 15,
             contextLines: 1,
-            source: 'search-bar-file-suggestions'
         }),
         select: (data): Suggestion[] => {
             if (isServiceError(data)) {
@@ -82,7 +81,6 @@ export const useSuggestionsData = ({
             query: `sym:${suggestionQuery.length > 0 ? suggestionQuery : ".*"}`,
             matches: 15,
             contextLines: 1,
-            source: 'search-bar-symbol-suggestions'
         }),
         select: (data): Suggestion[] => {
             if (isServiceError(data)) {

packages/web/src/app/[domain]/search/useStreamedSearch.ts

@@ -121,6 +121,7 @@ export const useStreamedSearch = ({ query, matches, contextLines, whole, isRegex
                     method: 'POST',
                     headers: {
                         'Content-Type': 'application/json',
+                        'X-Sourcebot-Client-Source': 'sourcebot-web-client',
                     },
                     body: JSON.stringify({
                         query,
@@ -129,7 +130,6 @@ export const useStreamedSearch = ({ query, matches, contextLines, whole, isRegex
                         whole,
                         isRegexEnabled,
                         isCaseSensitivityEnabled,
-                        source: 'sourcebot-web-client'
                     } satisfies SearchRequest),
                     signal: abortControllerRef.current.signal,
                 });

packages/web/src/app/api/(client)/client.ts

@@ -25,6 +25,7 @@ export const search = async (body: SearchRequest): Promise<SearchResponse | Serv
         method: "POST",
         headers: {
             "Content-Type": "application/json",
+            "X-Sourcebot-Client-Source": "sourcebot-web-client",
         },
         body: JSON.stringify(body),
     }).then(response => response.json());
@@ -44,6 +45,9 @@ export const getFileSource = async (queryParams: FileSourceRequest): Promise<Fil
 
     const result = await fetch(url, {
         method: "GET",
+        headers: {
+            "X-Sourcebot-Client-Source": "sourcebot-web-client",
+        }
     }).then(response => response.json());
 
     return result as FileSourceResponse | ServiceError;
@@ -59,6 +63,7 @@ export const listRepos = async (queryParams: ListReposQueryParams): Promise<List
         method: "GET",
         headers: {
             "Content-Type": "application/json",
+            "X-Sourcebot-Client-Source": "sourcebot-web-client",
         },
     }).then(response => response.json());
 
@@ -70,6 +75,7 @@ export const getVersion = async (): Promise<GetVersionResponse> => {
         method: "GET",
         headers: {
             "Content-Type": "application/json",
+            "X-Sourcebot-Client-Source": "sourcebot-web-client",
         },
     }).then(response => response.json());
     return result as GetVersionResponse;
@@ -78,6 +84,9 @@ export const getVersion = async (): Promise<GetVersionResponse> => {
 export const findSearchBasedSymbolReferences = async (body: FindRelatedSymbolsRequest): Promise<FindRelatedSymbolsResponse | ServiceError> => {
     const result = await fetch("/api/find_references", {
         method: "POST",
+        headers: {
+            "X-Sourcebot-Client-Source": "sourcebot-web-client",
+        },
         body: JSON.stringify(body),
     }).then(response => response.json());
     return result as FindRelatedSymbolsResponse | ServiceError;
@@ -86,6 +95,9 @@ export const findSearchBasedSymbolReferences = async (body: FindRelatedSymbolsRe
 export const findSearchBasedSymbolDefinitions = async (body: FindRelatedSymbolsRequest): Promise<FindRelatedSymbolsResponse | ServiceError> => {
     const result = await fetch("/api/find_definitions", {
         method: "POST",
+        headers: {
+            "X-Sourcebot-Client-Source": "sourcebot-web-client",
+        },
         body: JSON.stringify(body),
     }).then(response => response.json());
     return result as FindRelatedSymbolsResponse | ServiceError;
@@ -94,6 +106,9 @@ export const findSearchBasedSymbolDefinitions = async (body: FindRelatedSymbolsR
 export const getTree = async (body: GetTreeRequest): Promise<GetTreeResponse | ServiceError> => {
     const result = await fetch("/api/tree", {
         method: "POST",
+        headers: {
+            "X-Sourcebot-Client-Source": "sourcebot-web-client",
+        },
         body: JSON.stringify(body),
     }).then(response => response.json());
     return result as GetTreeResponse | ServiceError;
@@ -102,6 +117,9 @@ export const getTree = async (body: GetTreeRequest): Promise<GetTreeResponse | S
 export const getFiles = async (body: GetFilesRequest): Promise<GetFilesResponse | ServiceError> => {
     const result = await fetch("/api/files", {
         method: "POST",
+        headers: {
+            "X-Sourcebot-Client-Source": "sourcebot-web-client",
+        },
         body: JSON.stringify(body),
     }).then(response => response.json());
     return result as GetFilesResponse | ServiceError;

packages/web/src/app/api/(server)/chat/route.ts

@@ -3,6 +3,7 @@ import { _getConfiguredLanguageModelsFull, _getAISDKLanguageModelAndOptions, upd
 import { createAgentStream } from "@/features/chat/agent";
 import { additionalChatRequestParamsSchema, LanguageModelInfo, SBChatMessage, SearchScope } from "@/features/chat/types";
 import { getAnswerPartFromAssistantMessage, getLanguageModelKey } from "@/features/chat/utils";
+import { apiHandler } from "@/lib/apiHandler";
 import { ErrorCode } from "@/lib/errorCodes";
 import { notFound, requestBodySchemaValidationError, serviceErrorResponse } from "@/lib/serviceError";
 import { isServiceError } from "@/lib/utils";
@@ -22,6 +23,7 @@ import {
 } from "ai";
 import { randomUUID } from "crypto";
 import { StatusCodes } from "http-status-codes";
+import { NextRequest } from "next/server";
 import { z } from "zod";
 
 const logger = createLogger('chat-api');
@@ -33,7 +35,7 @@ const chatRequestSchema = z.object({
     ...additionalChatRequestParamsSchema.shape,
 })
 
-export async function POST(req: Request) {
+export const POST = apiHandler(async (req: NextRequest) => {
     const requestBody = await req.json();
     const parsed = await chatRequestSchema.safeParseAsync(requestBody);
     if (!parsed.success) {
@@ -102,7 +104,7 @@ export async function POST(req: Request) {
     }
 
     return response;
-}
+});
 
 // eslint-disable-next-line @typescript-eslint/no-explicit-any
 const mergeStreamAsync = async (stream: StreamTextResult<any, any>, writer: UIMessageStreamWriter<SBChatMessage>, options: UIMessageStreamOptions<SBChatMessage> = {}) => {

packages/web/src/app/api/(server)/commits/route.ts

@@ -1,4 +1,5 @@
 import { listCommits } from "@/features/git";
+import { apiHandler } from "@/lib/apiHandler";
 import { buildLinkHeader } from "@/lib/pagination";
 import { serviceErrorResponse, queryParamsSchemaValidationError } from "@/lib/serviceError";
 import { isServiceError } from "@/lib/utils";
@@ -16,7 +17,7 @@ const listCommitsQueryParamsSchema = z.object({
     perPage: z.coerce.number().int().positive().max(100).default(50),
 });
 
-export const GET = async (request: NextRequest): Promise<Response> => {
+export const GET = apiHandler(async (request: NextRequest): Promise<Response> => {
     const rawParams = Object.fromEntries(
         Object.keys(listCommitsQueryParamsSchema.shape).map(key => [
             key,
@@ -61,4 +62,4 @@ export const GET = async (request: NextRequest): Promise<Response> => {
     if (linkHeader) headers.set('Link', linkHeader);
 
     return new Response(JSON.stringify(commits), { status: 200, headers });
-}
+});

packages/web/src/app/api/(server)/ee/audit/route.ts

@@ -1,13 +1,14 @@
 'use server';
 
 import { fetchAuditRecords } from "@/ee/features/audit/actions";
+import { apiHandler } from "@/lib/apiHandler";
 import { ErrorCode } from "@/lib/errorCodes";
 import { serviceErrorResponse } from "@/lib/serviceError";
 import { isServiceError } from "@/lib/utils";
 import { getEntitlements } from "@sourcebot/shared";
 import { StatusCodes } from "http-status-codes";
 
-export const GET = async () => {
+export const GET = apiHandler(async () => {
     const entitlements = getEntitlements();
     if (!entitlements.includes('audit')) {
         return serviceErrorResponse({
@@ -22,4 +23,4 @@ export const GET = async () => {
         return serviceErrorResponse(result);
     }
     return Response.json(result);
-}; 
+});