discourse-ai/app/controllers/discourse_ai/admin/ai_personas_controller.rb

# frozen_string_literal: true

module DiscourseAi
  module Admin
    class AiPersonasController < ::Admin::AdminController
      before_action :find_ai_persona, only: %i[show update destroy]

      def index
        ai_personas =
          AiPersona.ordered.map do |persona|
            # we use a special serializer here cause names and descriptions are
            # localized for system personas
            LocalizedAiPersonaSerializer.new(persona, root: false)
          end
        commands =
          DiscourseAi::AiBot::Personas::Persona.all_available_commands.map do |command|
            { id: command.to_s.split("::").last, name: command.name.humanize.titleize }
          end
        render json: { ai_personas: ai_personas, meta: { commands: commands } }
      end

      def show
        render json: LocalizedAiPersonaSerializer.new(@ai_persona)
      end

      def create
        ai_persona = AiPersona.new(ai_persona_params)
        if ai_persona.save
          render json: { ai_persona: ai_persona }, status: :created
        else
          render_json_error ai_persona
        end
      end

      def update
        if @ai_persona.update(ai_persona_params)
          render json: @ai_persona
        else
          render_json_error @ai_persona
        end
      end

      def destroy
        if @ai_persona.destroy
          head :no_content
        else
          render_json_error @ai_persona
        end
      end

      private

      def find_ai_persona
        @ai_persona = AiPersona.find(params[:id])
      end

      def ai_persona_params
        params.require(:ai_persona).permit(
          :name,
          :description,
          :enabled,
          :system_prompt,
          :enabled,
          :priority,
          allowed_group_ids: [],
          commands: [],
        )
      end
    end
  end
end