feat(web): add sourceOverride to getFileSource and getTree

Extend the sourceOverride pattern to getFileSource and getTree so
internal callers (chat AI agent) can tag audit events with the correct
source instead of relying on the HTTP header.

Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>

Author: msukkari

packages/web/src/features/chat/agent.ts

@@ -171,7 +171,7 @@ const createAgentStream = async ({
                 path: source.path,
                 repo: source.repo,
                 ref: source.revision,
-            });
+            }, { sourceOverride: 'sourcebot-ask-agent' });
 
             if (isServiceError(fileSource)) {
                 logger.error("Error fetching file source:", fileSource);

packages/web/src/features/chat/tools.ts

@@ -115,7 +115,7 @@ export const readFilesTool = tool({
                 path,
                 repo: repository,
                 ref: revision,
-            });
+            }, { sourceOverride: 'sourcebot-ask-agent' });
         }));
 
         if (responses.some(isServiceError)) {
@@ -221,7 +221,8 @@ export const createCodeSearchTool = (selectedRepos: string[]) => tool({
                 contextLines: 3,
                 isCaseSensitivityEnabled: caseSensitive,
                 isRegexEnabled: useRegex,
-            }
+            },
+            sourceOverride: 'sourcebot-ask-agent',
         });
 
         if (isServiceError(response)) {
@@ -253,7 +254,7 @@ export const listReposTool = tool({
     description: 'Lists repositories in the organization with optional filtering and pagination.',
     inputSchema: listReposQueryParamsSchema,
     execute: async (request: ListReposQueryParams) => {
-        const reposResponse = await listRepos(request);
+        const reposResponse = await listRepos({ ...request, sourceOverride: 'sourcebot-ask-agent' });
 
         if (isServiceError(reposResponse)) {
             return reposResponse;

packages/web/src/features/git/getFileSourceApi.ts

@@ -33,9 +33,9 @@ export const fileSourceResponseSchema = z.object({
 });
 export type FileSourceResponse = z.infer<typeof fileSourceResponseSchema>;
 
-export const getFileSource = async ({ path: filePath, repo: repoName, ref }: FileSourceRequest): Promise<FileSourceResponse | ServiceError> => sew(() => withOptionalAuthV2(async ({ org, prisma, user }) => {
+export const getFileSource = async ({ path: filePath, repo: repoName, ref }: FileSourceRequest, { sourceOverride }: { sourceOverride?: string } = {}): Promise<FileSourceResponse | ServiceError> => sew(() => withOptionalAuthV2(async ({ org, prisma, user }) => {
     if (user) {
-        const source = (await headers()).get('X-Sourcebot-Client-Source') ?? undefined;
+        const source = sourceOverride ?? (await headers()).get('X-Sourcebot-Client-Source') ?? undefined;
         getAuditService().createAudit({
             action: 'user.fetched_file_source',
             actor: { id: user.id, type: 'user' },

packages/web/src/features/git/getTreeApi.ts

@@ -26,10 +26,10 @@ export type GetTreeResponse = z.infer<typeof getTreeResponseSchema>;
  * repo/revision, including intermediate directories needed to connect them
  * into a single tree.
  */
-export const getTree = async ({ repoName, revisionName, paths }: GetTreeRequest): Promise<GetTreeResponse | ServiceError> => sew(() =>
+export const getTree = async ({ repoName, revisionName, paths }: GetTreeRequest, { sourceOverride }: { sourceOverride?: string } = {}): Promise<GetTreeResponse | ServiceError> => sew(() =>
     withOptionalAuthV2(async ({ org, prisma, user }) => {
         if (user) {
-            const source = (await headers()).get('X-Sourcebot-Client-Source') ?? undefined;
+            const source = sourceOverride ?? (await headers()).get('X-Sourcebot-Client-Source') ?? undefined;
             getAuditService().createAudit({
                 action: 'user.fetched_file_tree',
                 actor: { id: user.id, type: 'user' },