layered-assistant-rails 0.1.0

data/app/controllers/layered/assistant/assistants_controller.rb

@@ -0,0 +1,59 @@
+module Layered
+  module Assistant
+    class AssistantsController < ApplicationController
+      before_action :set_assistant, only: [:edit, :update, :destroy]
+      before_action :set_models, only: [:new, :create, :edit, :update]
+
+      def index
+        @page_title = "Assistants"
+        @pagy, @assistants = pagy(Assistant.by_name)
+      end
+
+      def new
+        @page_title = "New assistant"
+        @assistant = Assistant.new
+      end
+
+      def create
+        @assistant = Assistant.new(assistant_params)
+
+        if @assistant.save
+          redirect_to layered_assistant.assistants_path, notice: "Assistant was successfully created."
+        else
+          render :new, status: :unprocessable_entity
+        end
+      end
+
+      def edit
+        @page_title = "Edit assistant"
+      end
+
+      def update
+        if @assistant.update(assistant_params)
+          redirect_to layered_assistant.assistants_path, notice: "Assistant was successfully updated."
+        else
+          render :edit, status: :unprocessable_entity
+        end
+      end
+
+      def destroy
+        @assistant.destroy
+        redirect_to layered_assistant.assistants_path, notice: "Assistant was successfully deleted."
+      end
+
+      private
+
+      def set_assistant
+        @assistant = Assistant.find(params[:id])
+      end
+
+      def set_models
+        @models = Model.available
+      end
+
+      def assistant_params
+        params.require(:assistant).permit(:name, :description, :system_prompt, :default_model_id, :public)
+      end
+    end
+  end
+end

data/app/controllers/layered/assistant/conversations_controller.rb

@@ -0,0 +1,77 @@
+module Layered
+  module Assistant
+    class ConversationsController < ApplicationController
+      before_action :set_conversation, only: [:show, :edit, :update, :destroy]
+      before_action :set_assistants, only: [:new, :create]
+
+      def index
+        if params[:assistant_id]
+          @assistant = Assistant.find(params[:assistant_id])
+          @page_title = "Conversations - #{@assistant.name}"
+          @pagy, @conversations = pagy(@assistant.conversations.includes(:owner).by_created_at)
+        else
+          @page_title = "Conversations"
+          @pagy, @conversations = pagy(Conversation.includes(:assistant, :owner).by_created_at)
+        end
+      end
+
+      def show
+        @page_title = @conversation.name
+        @messages = @conversation.messages.includes(:model).by_created_at
+        @models = Model.available
+        @selected_model_id = @messages.last&.model_id || @conversation.assistant.default_model_id || @models.first&.id
+      end
+
+      def new
+        @page_title = "New conversation"
+        @conversation = Conversation.new(params.permit(conversation: [:assistant_id])[:conversation])
+      end
+
+      def create
+        @conversation = Conversation.new(conversation_params)
+        @conversation.owner = l_ui_current_user
+        @conversation.name = Conversation.default_name if @conversation.name.blank?
+        if @conversation.save
+          redirect_to layered_assistant.conversation_path(@conversation), notice: "Conversation was successfully created."
+        else
+          render :new, status: :unprocessable_entity
+        end
+      end
+
+      def edit
+        @page_title = "Edit conversation"
+      end
+
+      def update
+        if @conversation.update(conversation_params)
+          redirect_to layered_assistant.conversations_path, notice: "Conversation was successfully updated."
+        else
+          render :edit, status: :unprocessable_entity
+        end
+      end
+
+      def destroy
+        @conversation.destroy
+        redirect_to layered_assistant.conversations_path, notice: "Conversation was successfully deleted."
+      end
+
+      private
+
+      def set_conversation
+        @conversation = Conversation.find(params[:id])
+      end
+
+      def set_assistants
+        @assistants = Assistant.by_name
+      end
+
+      def conversation_params
+        if action_name == "create"
+          params.require(:conversation).permit(:name, :assistant_id)
+        else
+          params.require(:conversation).permit(:name)
+        end
+      end
+    end
+  end
+end

data/app/controllers/layered/assistant/messages_controller.rb

@@ -0,0 +1,57 @@
+module Layered
+  module Assistant
+    class MessagesController < ApplicationController
+      include MessageCreation
+
+      before_action :set_conversation
+      before_action :set_message, only: [:destroy]
+
+      def index
+        @page_title = "Messages"
+        @pagy, @messages = pagy(@conversation.messages.by_created_at)
+      end
+
+      def create
+        result = create_messages_for(
+          conversation: @conversation,
+          content: message_params[:content],
+          model_id: message_params[:model_id]
+        )
+        @message = result[:message]
+
+        unless @message.persisted?
+          return head :unprocessable_entity
+        end
+
+        @assistant_message = result[:assistant_message]
+        @models = result[:models]
+        @selected_model_id = result[:selected_model_id]
+        @error = result[:error]
+
+        respond_to do |format|
+          format.turbo_stream
+        end
+      end
+
+      def destroy
+        @message.destroy
+        @conversation.update_token_totals!
+        redirect_to layered_assistant.conversation_messages_path(@conversation), notice: "Message was successfully deleted."
+      end
+
+      private
+
+      def set_conversation
+        @conversation = Conversation.find(params[:conversation_id])
+      end
+
+      def set_message
+        @message = @conversation.messages.find(params[:id])
+      end
+
+      def message_params
+        params.require(:message).permit(:content, :model_id)
+      end
+    end
+  end
+end

data/app/controllers/layered/assistant/models_controller.rb

@@ -0,0 +1,61 @@
+module Layered
+  module Assistant
+    class ModelsController < ApplicationController
+      before_action :set_provider
+      before_action :set_model, only: [:edit, :update, :destroy]
+
+      def index
+        @pagy, @models = pagy(@provider.models.sorted)
+      end
+
+      def new
+        @model = @provider.models.new
+      end
+
+      def create
+        @model = @provider.models.new(model_params)
+
+        if @model.save
+          redirect_to layered_assistant.provider_models_path(@provider), notice: "Model was successfully created."
+        else
+          render :new, status: :unprocessable_entity
+        end
+      end
+
+      def edit
+      end
+
+      def update
+        if @model.update(model_params)
+          redirect_to layered_assistant.provider_models_path(@provider), notice: "Model was successfully updated."
+        else
+          render :edit, status: :unprocessable_entity
+        end
+      end
+
+      def destroy
+        if @model.destroy
+          redirect_to layered_assistant.provider_models_path(@provider), notice: "Model was successfully deleted."
+        else
+          redirect_to layered_assistant.provider_models_path(@provider), alert: @model.errors.full_messages.to_sentence
+        end
+      rescue ActiveRecord::InvalidForeignKey
+        redirect_to layered_assistant.provider_models_path(@provider), alert: "Cannot delete a model that is in use."
+      end
+
+      private
+
+      def set_provider
+        @provider = Provider.find(params[:provider_id])
+      end
+
+      def set_model
+        @model = @provider.models.find(params[:id])
+      end
+
+      def model_params
+        params.require(:model).permit(:name, :identifier, :enabled)
+      end
+    end
+  end
+end

data/app/controllers/layered/assistant/panel/conversations_controller.rb

@@ -0,0 +1,63 @@
+module Layered
+  module Assistant
+    module Panel
+      class ConversationsController < ApplicationController
+        layout false
+
+        before_action :set_conversation, only: [:show, :destroy]
+        before_action :set_conversations, only: [:index, :show]
+        before_action :set_assistants, only: [:index, :show, :new, :create]
+
+        def index
+        end
+
+        def show
+          @messages = @conversation.messages.includes(:model).by_created_at
+          @models = Model.available
+          @selected_model_id = @messages.last&.model_id || @conversation.assistant.default_model_id || @models.first&.id
+        end
+
+        def new
+          @conversation = Conversation.new
+        end
+
+        def create
+          @conversation = Conversation.new(conversation_params)
+          @conversation.owner = l_ui_current_user
+          @conversation.name = Conversation.default_name if @conversation.name.blank?
+
+          if @conversation.save
+            redirect_to layered_assistant.panel_conversation_path(@conversation)
+          else
+            render :new, status: :unprocessable_entity
+          end
+        end
+
+        def destroy
+          @conversation.destroy
+          redirect_to layered_assistant.panel_conversations_path
+        end
+
+        private
+
+        def set_conversation
+          @conversation = Conversation.find(params[:id])
+        end
+
+        def set_assistants
+          @assistants = Assistant.by_name
+        end
+
+        def set_conversations
+          scope = Conversation.by_created_at
+          scope = scope.where(assistant_id: params[:assistant_id]) if params[:assistant_id].present?
+          @conversations = scope.limit(20)
+        end
+
+        def conversation_params
+          params.require(:conversation).permit(:assistant_id, :name)
+        end
+      end
+    end
+  end
+end

data/app/controllers/layered/assistant/panel/messages_controller.rb

@@ -0,0 +1,44 @@
+module Layered
+  module Assistant
+    module Panel
+      class MessagesController < ApplicationController
+        layout false
+        include MessageCreation
+
+        before_action :set_conversation
+
+        def create
+          result = create_messages_for(
+            conversation: @conversation,
+            content: message_params[:content],
+            model_id: message_params[:model_id]
+          )
+          @message = result[:message]
+
+          unless @message.persisted?
+            return head :unprocessable_entity
+          end
+
+          @assistant_message = result[:assistant_message]
+          @models = result[:models]
+          @selected_model_id = result[:selected_model_id]
+          @error = result[:error]
+
+          respond_to do |format|
+            format.turbo_stream
+          end
+        end
+
+        private
+
+        def set_conversation
+          @conversation = Conversation.find(params[:conversation_id])
+        end
+
+        def message_params
+          params.require(:message).permit(:content, :model_id)
+        end
+      end
+    end
+  end
+end

data/app/controllers/layered/assistant/providers_controller.rb

@@ -0,0 +1,55 @@
+module Layered
+  module Assistant
+    class ProvidersController < ApplicationController
+      before_action :set_provider, only: [:edit, :update, :destroy]
+
+      def index
+        @page_title = "Providers"
+        @pagy, @providers = pagy(Provider.sorted)
+      end
+
+      def new
+        @page_title = "New Provider"
+        @provider = Provider.new
+      end
+
+      def create
+        @provider = Provider.new(provider_params)
+
+        if @provider.save
+          Models::CreateService.new(@provider).call if params[:provider][:create_models] == "1"
+          redirect_to layered_assistant.providers_path, notice: "Provider was successfully created."
+        else
+          render :new, status: :unprocessable_entity
+        end
+      end
+
+      def edit
+        @page_title = "Edit Provider"
+      end
+
+      def update
+        if @provider.update(provider_params)
+          redirect_to layered_assistant.providers_path, notice: "Provider was successfully updated."
+        else
+          render :edit, status: :unprocessable_entity
+        end
+      end
+
+      def destroy
+        @provider.destroy
+        redirect_to layered_assistant.providers_path, notice: "Provider was successfully deleted."
+      end
+
+      private
+
+      def set_provider
+        @provider = Provider.find(params[:id])
+      end
+
+      def provider_params
+        params.require(:provider).permit(:name, :protocol, :url, :secret, :enabled)
+      end
+    end
+  end
+end

data/app/controllers/layered/assistant/public/application_controller.rb

@@ -0,0 +1,16 @@
+module Layered
+  module Assistant
+    module Public
+      class ApplicationController < Layered::Assistant::ApplicationController
+        skip_before_action :layered_assistant_authorize!
+        include SessionConversations
+
+        private
+
+        def set_public_assistant
+          @assistant = Assistant.publicly_available.find(params[:assistant_id] || params[:id])
+        end
+      end
+    end
+  end
+end

data/app/controllers/layered/assistant/public/assistants_controller.rb

@@ -0,0 +1,16 @@
+module Layered
+  module Assistant
+    module Public
+      class AssistantsController < ApplicationController
+        before_action :set_public_assistant, only: [:show]
+
+        def index
+          @pagy, @assistants = pagy(Assistant.publicly_available.by_name)
+        end
+
+        def show
+        end
+      end
+    end
+  end
+end

data/app/controllers/layered/assistant/public/conversations_controller.rb

@@ -0,0 +1,33 @@
+module Layered
+  module Assistant
+    module Public
+      class ConversationsController < ApplicationController
+        before_action :set_public_assistant, only: [:create]
+        before_action :set_conversation, only: [:show]
+
+        def create
+          @conversation = @assistant.conversations.new(name: Conversation.default_name)
+
+          if @conversation.save
+            add_conversation_to_session(@conversation)
+            redirect_to layered_assistant.public_conversation_path(@conversation)
+          else
+            redirect_to layered_assistant.public_assistant_path(@assistant)
+          end
+        end
+
+        def show
+          @messages = @conversation.messages.includes(:model).by_created_at
+        end
+
+        private
+
+        def set_conversation
+          @conversation = find_session_conversation(params[:id])
+        rescue ActiveRecord::RecordNotFound
+          redirect_to layered_assistant.public_assistants_path
+        end
+      end
+    end
+  end
+end

data/app/controllers/layered/assistant/public/messages_controller.rb

@@ -0,0 +1,42 @@
+module Layered
+  module Assistant
+    module Public
+      class MessagesController < ApplicationController
+        include MessageCreation
+
+        before_action :set_conversation
+
+        def create
+          result = create_messages_for(
+            conversation: @conversation,
+            content: message_params[:content],
+            model_id: @conversation.assistant.default_model_id
+          )
+          @message = result[:message]
+
+          unless @message.persisted?
+            return head :unprocessable_entity
+          end
+
+          @error = result[:error]
+
+          respond_to do |format|
+            format.turbo_stream
+          end
+        end
+
+        private
+
+        def set_conversation
+          @conversation = find_session_conversation(params[:conversation_id])
+        rescue ActiveRecord::RecordNotFound
+          redirect_to layered_assistant.public_assistants_path
+        end
+
+        def message_params
+          params.require(:message).permit(:content)
+        end
+      end
+    end
+  end
+end

data/app/controllers/layered/assistant/public/panel/conversations_controller.rb

@@ -0,0 +1,62 @@
+module Layered
+  module Assistant
+    module Public
+      module Panel
+        class ConversationsController < Public::ApplicationController
+          layout false
+
+          before_action :set_public_assistant, only: [:index, :new, :create]
+          before_action :set_conversation, only: [:show]
+          before_action :set_session_conversations, only: [:index, :show]
+
+          def index
+          end
+
+          def new
+          end
+
+          def create
+            @conversation = @assistant.conversations.new(name: Conversation.default_name)
+
+            if @conversation.save
+              add_conversation_to_session(@conversation)
+              redirect_to layered_assistant.public_panel_conversation_path(@conversation)
+            else
+              render :new, status: :unprocessable_entity
+            end
+          end
+
+          def show
+            @messages = @conversation.messages.includes(:model).by_created_at
+          end
+
+          private
+
+          def set_conversation
+            @conversation = find_session_conversation(params[:id])
+          rescue ActiveRecord::RecordNotFound
+            assistant = Conversation.find_by(id: params[:id])&.assistant
+            if assistant&.public?
+              redirect_to layered_assistant.public_panel_conversations_path(assistant_id: assistant.id)
+            else
+              redirect_to layered_assistant.public_assistants_path
+            end
+          end
+
+          def set_session_conversations
+            assistant_id = @conversation&.assistant_id || @assistant&.id
+            @conversations = if session_conversation_ids.any?
+              scope = Conversation.joins(:assistant).merge(Assistant.publicly_available)
+                .where(id: session_conversation_ids)
+                .by_created_at
+              scope = scope.where(assistant_id: assistant_id) if assistant_id
+              scope.limit(20)
+            else
+              Conversation.none
+            end
+          end
+        end
+      end
+    end
+  end
+end

data/app/controllers/layered/assistant/public/panel/messages_controller.rb

@@ -0,0 +1,50 @@
+module Layered
+  module Assistant
+    module Public
+      module Panel
+        class MessagesController < Public::ApplicationController
+          layout false
+          include MessageCreation
+
+          before_action :set_conversation
+
+          def create
+            result = create_messages_for(
+              conversation: @conversation,
+              content: message_params[:content],
+              model_id: @conversation.assistant.default_model_id
+            )
+            @message = result[:message]
+
+            unless @message.persisted?
+              return head :unprocessable_entity
+            end
+
+            @error = result[:error]
+
+            respond_to do |format|
+              format.turbo_stream
+            end
+          end
+
+          private
+
+          def set_conversation
+            @conversation = find_session_conversation(params[:conversation_id])
+          rescue ActiveRecord::RecordNotFound
+            assistant = Conversation.find_by(id: params[:conversation_id])&.assistant
+            if assistant&.public?
+              redirect_to layered_assistant.new_public_panel_conversation_path(assistant_id: assistant.id)
+            else
+              redirect_to layered_assistant.public_assistants_path
+            end
+          end
+
+          def message_params
+            params.require(:message).permit(:content)
+          end
+        end
+      end
+    end
+  end
+end

data/app/controllers/layered/assistant/setup_controller.rb

@@ -0,0 +1,9 @@
+module Layered
+  module Assistant
+    class SetupController < ApplicationController
+      def index
+        @page_title = "Setup"
+      end
+    end
+  end
+end

data/app/helpers/layered/assistant/access_helper.rb

@@ -0,0 +1,45 @@
+module Layered
+  module Assistant
+    module AccessHelper
+      def l_assistant_accessible?
+        block = Layered::Assistant.authorize_block
+        return false unless block
+
+        checker = AccessibilityChecker.new(self)
+        checker.accessible?(&block)
+      end
+
+      class AccessibilityChecker
+        def initialize(context)
+          @context = context
+          @blocked = false
+        end
+
+        def accessible?(&block)
+          instance_exec(&block)
+          !@blocked
+        end
+
+        def head(*)
+          @blocked = true
+        end
+
+        def redirect_to(*)
+          @blocked = true
+        end
+
+        private
+
+        def method_missing(method, ...)
+          @context.send(method, ...)
+        end
+
+        def respond_to_missing?(method, include_private = false)
+          @context.respond_to?(method, include_private)
+        end
+      end
+
+      private_constant :AccessibilityChecker
+    end
+  end
+end