import { createAction } from '@typebot.io/forge'
import { auth } from '../auth'
import { baseOptions } from '../baseOptions'
import { fetchGPTModels } from '../helpers/fetchModels'
import { isDefined } from '@typebot.io/lib'
import { runGenerateVariables } from '../shared/runGenerateVariables'
import { parseGenerateVariablesOptions } from '../shared/parseGenerateVariablesOptions'
import { createOpenAI } from '@ai-sdk/openai'

export const generateVariables = createAction({
  name: 'Generate variables',
  auth,
  baseOptions,
  options: parseGenerateVariablesOptions({ modelFetch: 'fetchModels' }),
  fetchers: [
    {
      id: 'fetchModels',
      dependencies: ['baseUrl', 'apiVersion'],
      fetch: ({ credentials, options }) =>
        fetchGPTModels({
          apiKey: credentials?.apiKey,
          baseUrl: options.baseUrl,
          apiVersion: options.apiVersion,
        }),
    },
  ],
  turnableInto: [
    {
      blockId: 'mistral',
    },
    {
      blockId: 'anthropic',
      transform: (options) => ({
        ...options,
        model: undefined,
      }),
    },
  ],
  getSetVariableIds: (options) =>
    options.variablesToExtract?.map((v) => v.variableId).filter(isDefined) ??
    [],
  run: {
    server: ({ credentials, options, variables, logs }) => {
      if (credentials?.apiKey === undefined)
        return logs.add('No API key provided')

      if (options.model === undefined) return logs.add('No model provided')

      return runGenerateVariables({
        model: createOpenAI({
          apiKey: credentials.apiKey,
          compatibility: 'strict',
        })(options.model),
        prompt: options.prompt,
        variablesToExtract: options.variablesToExtract,
        variables,
        logs,
      })
    },
  },
})