Function createAssistant

packages/server/src/services/assistants/index.ts:24–167

(requestBody: any, orgId: string, workspaceId: string)

import nodesService from '../nodes'

const createAssistant = async (requestBody: any, orgId: string, workspaceId: string): Promise<Assistant> => {
    try {
        const appServer = getRunningExpressApp()
        if (!requestBody.details) {
            throw new InternalFlowiseError(StatusCodes.INTERNAL_SERVER_ERROR, `Invalid request body`)
        }
        const assistantDetails = JSON.parse(requestBody.details)

        if (requestBody.type === 'CUSTOM') {
            // For CUSTOM assistants the credential field is a client-generated UUID used as an
            // internal identifier, not a reference to the Credential entity, so no lookup is needed.
            const newAssistant = new Assistant()
            Object.assign(newAssistant, stripProtectedFields(requestBody))
            newAssistant.workspaceId = workspaceId

            const assistant = appServer.AppDataSource.getRepository(Assistant).create(newAssistant)
            const dbResponse = await appServer.AppDataSource.getRepository(Assistant).save(assistant)

            await appServer.telemetry.sendTelemetry(
                'assistant_created',
                {
                    version: await getAppVersion(),
                    assistantId: dbResponse.id
                },
                orgId
            )
            appServer.metricsProvider?.incrementCounter(FLOWISE_METRIC_COUNTERS.ASSISTANT_CREATED, {
                status: FLOWISE_COUNTER_STATUS.SUCCESS
            })
            return dbResponse
        }

        try {
            const credential = await appServer.AppDataSource.getRepository(Credential).findOneBy({
                id: requestBody.credential,
                workspaceId: workspaceId
            })

            if (!credential) {
                throw new InternalFlowiseError(StatusCodes.NOT_FOUND, `Credential ${requestBody.credential} not found`)
            }

            // Decrpyt credentialData
            const decryptedCredentialData = await decryptCredentialData(credential.encryptedData)
            const openAIApiKey = decryptedCredentialData['openAIApiKey']
            if (!openAIApiKey) {
                throw new InternalFlowiseError(StatusCodes.NOT_FOUND, `OpenAI ApiKey not found`)
            }
            const openai = new OpenAI({ apiKey: openAIApiKey })

            // Prepare tools
            let tools = []
            if (assistantDetails.tools) {
                for (const tool of assistantDetails.tools ?? []) {
                    tools.push({
                        type: tool
                    })
                }