Function createAgentStream

packages/web/src/ee/features/chat/agent.ts:239–465

({
    model,
    providerOptions,
    temperature,
    inputMessages,
    inputSources,
    selectedRepos,
    disabledMcpServerIds,
    onWriteSource,
    onMcpServerDiscovered,
    onMcpServerFailed,
    traceId,
    chatId,
    prisma,
    userId,
    orgId,
}: AgentOptions)

}

const createAgentStream = async ({
    model,
    providerOptions,
    temperature,
    inputMessages,
    inputSources,
    selectedRepos,
    disabledMcpServerIds,
    onWriteSource,
    onMcpServerDiscovered,
    onMcpServerFailed,
    traceId,
    chatId,
    prisma,
    userId,
    orgId,
}: AgentOptions) => {
    // For every file source, resolve the source code so that we can include it in the system prompt.
    const fileSources = inputSources.filter((source) => source.type === 'file');
    const resolvedFileSources = (
        await Promise.all(fileSources.map(async (source) => {
            const fileSource = await getFileSource({
                path: source.path,
                repo: source.repo,
                ref: source.revision,
            }, { source: 'sourcebot-ask-agent' });

            if (isServiceError(fileSource)) {
                logger.error("Error fetching file source:", fileSource);
                return undefined;
            }

            return {
                path: fileSource.path,
                source: fileSource.source,
                repo: fileSource.repo,
                language: fileSource.language,
                revision: source.revision,
            };
        }))
    ).filter((source) => source !== undefined);

    let mcpToolSetsObj: McpToolsResult = { tools: {}, failedServers: [], serverFaviconUrls: {}, cleanup: async () => {} };
    if (userId && orgId && await hasEntitlement('ask') && disabledMcpServerIds !== undefined) {
        try {
            const allMcpClients = await getConnectedMcpClients(prisma, userId, orgId);
            const mcpClients = allMcpClients.filter((c) => !disabledMcpServerIds.includes(c.serverId));
            mcpToolSetsObj = await getMcpTools(mcpClients, {
                chatId,
                traceId,
                source: 'sourcebot-ask-agent',
            });

            for (const [sanitizedName, faviconUrl] of Object.entries(mcpToolSetsObj.serverFaviconUrls)) {
                onMcpServerDiscovered(sanitizedName, faviconUrl);
            }

            if (mcpClients.length > 0) {