Function askCodebase

packages/web/src/ee/features/mcp/askCodebase.ts:46–241

(params: AskCodebaseParams)

};

export const askCodebase = (params: AskCodebaseParams): Promise<AskCodebaseResult | ServiceError> =>
    sew(() =>
        withOptionalAuth(async ({ org, user, prisma }) => {
            // Ask Sourcebot is a paid feature. askCodebase() is the single choke point
            // for the programmatic ask path (the MCP `ask_codebase` tool and the
            // /api/chat/blocking route both wrap it), so gating here covers both without
            // double-gating at the tool-registration layer.
            const askEntitlementError = await checkAskEntitlement();
            if (askEntitlementError) {
                return askEntitlementError;
            }

            const { query, repos = [], languageModel: requestedLanguageModel, visibility: requestedVisibility, source } = params;

            const configuredModels = await getConfiguredLanguageModels();
            if (configuredModels.length === 0) {
                return {
                    statusCode: StatusCodes.BAD_REQUEST,
                    errorCode: ErrorCode.INVALID_REQUEST_BODY,
                    message: "No language models are configured. Please configure at least one language model. See: https://docs.sourcebot.dev/docs/configuration/language-model-providers",
                } satisfies ServiceError;
            }

            let languageModelConfig = configuredModels[0];
            if (requestedLanguageModel) {
                const matchingModel = configuredModels.find(
                    (m) => getLanguageModelKey(m) === getLanguageModelKey(requestedLanguageModel)
                );
                if (!matchingModel) {
                    return {
                        statusCode: StatusCodes.BAD_REQUEST,
                        errorCode: ErrorCode.INVALID_REQUEST_BODY,
                        message: `Language model '${requestedLanguageModel.provider}/${requestedLanguageModel.model}' is not configured.`,
                    } satisfies ServiceError;
                }
                languageModelConfig = matchingModel;
            }

            const { model, providerOptions, temperature } = await getAISDKLanguageModelAndOptions(languageModelConfig);
            const modelName = languageModelConfig.displayName ?? languageModelConfig.model;

            const chatVisibility = (requestedVisibility && user)
                ? requestedVisibility
                : (user ? ChatVisibility.PRIVATE : ChatVisibility.PUBLIC);

            const chat = await prisma.chat.create({
                data: {
                    orgId: org.id,
                    createdById: user?.id,
                    visibility: chatVisibility,
                    messages: [] as unknown as Prisma.InputJsonValue,
                },
            });

            await captureEvent('ask_thread_created', {
                chatId: chat.id,
                isAnonymous: !user,
                source,