lex-llm 0.1.2 → 0.1.3

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: lex-llm
 version: !ruby/object:Gem::Version
-  version: 0.1.2
+  version: 0.1.3
 platform: ruby
 authors:
 - LegionIO
@@ -196,130 +196,48 @@ files:
 - LICENSE
 - README.md
 - lex-llm.gemspec
-- lib/generators/lex_llm/agent/agent_generator.rb
-- lib/generators/lex_llm/agent/templates/agent.rb.tt
-- lib/generators/lex_llm/agent/templates/instructions.txt.erb.tt
-- lib/generators/lex_llm/chat_ui/chat_ui_generator.rb
-- lib/generators/lex_llm/chat_ui/templates/controllers/chats_controller.rb.tt
-- lib/generators/lex_llm/chat_ui/templates/controllers/messages_controller.rb.tt
-- lib/generators/lex_llm/chat_ui/templates/controllers/models_controller.rb.tt
-- lib/generators/lex_llm/chat_ui/templates/helpers/messages_helper.rb.tt
-- lib/generators/lex_llm/chat_ui/templates/jobs/chat_response_job.rb.tt
-- lib/generators/lex_llm/chat_ui/templates/tailwind/views/chats/_chat.html.erb.tt
-- lib/generators/lex_llm/chat_ui/templates/tailwind/views/chats/_form.html.erb.tt
-- lib/generators/lex_llm/chat_ui/templates/tailwind/views/chats/index.html.erb.tt
-- lib/generators/lex_llm/chat_ui/templates/tailwind/views/chats/new.html.erb.tt
-- lib/generators/lex_llm/chat_ui/templates/tailwind/views/chats/show.html.erb.tt
-- lib/generators/lex_llm/chat_ui/templates/tailwind/views/messages/_assistant.html.erb.tt
-- lib/generators/lex_llm/chat_ui/templates/tailwind/views/messages/_content.html.erb.tt
-- lib/generators/lex_llm/chat_ui/templates/tailwind/views/messages/_error.html.erb.tt
-- lib/generators/lex_llm/chat_ui/templates/tailwind/views/messages/_form.html.erb.tt
-- lib/generators/lex_llm/chat_ui/templates/tailwind/views/messages/_system.html.erb.tt
-- lib/generators/lex_llm/chat_ui/templates/tailwind/views/messages/_tool.html.erb.tt
-- lib/generators/lex_llm/chat_ui/templates/tailwind/views/messages/_tool_calls.html.erb.tt
-- lib/generators/lex_llm/chat_ui/templates/tailwind/views/messages/_user.html.erb.tt
-- lib/generators/lex_llm/chat_ui/templates/tailwind/views/messages/tool_calls/_default.html.erb.tt
-- lib/generators/lex_llm/chat_ui/templates/tailwind/views/messages/tool_results/_default.html.erb.tt
-- lib/generators/lex_llm/chat_ui/templates/tailwind/views/models/_model.html.erb.tt
-- lib/generators/lex_llm/chat_ui/templates/tailwind/views/models/index.html.erb.tt
-- lib/generators/lex_llm/chat_ui/templates/tailwind/views/models/show.html.erb.tt
-- lib/generators/lex_llm/chat_ui/templates/views/chats/_chat.html.erb.tt
-- lib/generators/lex_llm/chat_ui/templates/views/chats/_form.html.erb.tt
-- lib/generators/lex_llm/chat_ui/templates/views/chats/index.html.erb.tt
-- lib/generators/lex_llm/chat_ui/templates/views/chats/new.html.erb.tt
-- lib/generators/lex_llm/chat_ui/templates/views/chats/show.html.erb.tt
-- lib/generators/lex_llm/chat_ui/templates/views/messages/_assistant.html.erb.tt
-- lib/generators/lex_llm/chat_ui/templates/views/messages/_content.html.erb.tt
-- lib/generators/lex_llm/chat_ui/templates/views/messages/_error.html.erb.tt
-- lib/generators/lex_llm/chat_ui/templates/views/messages/_form.html.erb.tt
-- lib/generators/lex_llm/chat_ui/templates/views/messages/_system.html.erb.tt
-- lib/generators/lex_llm/chat_ui/templates/views/messages/_tool.html.erb.tt
-- lib/generators/lex_llm/chat_ui/templates/views/messages/_tool_calls.html.erb.tt
-- lib/generators/lex_llm/chat_ui/templates/views/messages/_user.html.erb.tt
-- lib/generators/lex_llm/chat_ui/templates/views/messages/create.turbo_stream.erb.tt
-- lib/generators/lex_llm/chat_ui/templates/views/messages/tool_calls/_default.html.erb.tt
-- lib/generators/lex_llm/chat_ui/templates/views/messages/tool_results/_default.html.erb.tt
-- lib/generators/lex_llm/chat_ui/templates/views/models/_model.html.erb.tt
-- lib/generators/lex_llm/chat_ui/templates/views/models/index.html.erb.tt
-- lib/generators/lex_llm/chat_ui/templates/views/models/show.html.erb.tt
-- lib/generators/lex_llm/generator_helpers.rb
-- lib/generators/lex_llm/install/install_generator.rb
-- lib/generators/lex_llm/install/templates/add_references_to_chats_tool_calls_and_messages_migration.rb.tt
-- lib/generators/lex_llm/install/templates/chat_model.rb.tt
-- lib/generators/lex_llm/install/templates/create_chats_migration.rb.tt
-- lib/generators/lex_llm/install/templates/create_messages_migration.rb.tt
-- lib/generators/lex_llm/install/templates/create_models_migration.rb.tt
-- lib/generators/lex_llm/install/templates/create_tool_calls_migration.rb.tt
-- lib/generators/lex_llm/install/templates/initializer.rb.tt
-- lib/generators/lex_llm/install/templates/message_model.rb.tt
-- lib/generators/lex_llm/install/templates/model_model.rb.tt
-- lib/generators/lex_llm/install/templates/tool_call_model.rb.tt
-- lib/generators/lex_llm/schema/schema_generator.rb
-- lib/generators/lex_llm/schema/templates/schema.rb.tt
-- lib/generators/lex_llm/tool/templates/tool.rb.tt
-- lib/generators/lex_llm/tool/templates/tool_call.html.erb.tt
-- lib/generators/lex_llm/tool/templates/tool_result.html.erb.tt
-- lib/generators/lex_llm/tool/tool_generator.rb
-- lib/generators/lex_llm/upgrade_to_v1_10/templates/add_v1_10_message_columns.rb.tt
-- lib/generators/lex_llm/upgrade_to_v1_10/upgrade_to_v1_10_generator.rb
-- lib/generators/lex_llm/upgrade_to_v1_14/templates/add_v1_14_tool_call_columns.rb.tt
-- lib/generators/lex_llm/upgrade_to_v1_14/upgrade_to_v1_14_generator.rb
-- lib/generators/lex_llm/upgrade_to_v1_7/templates/migration.rb.tt
-- lib/generators/lex_llm/upgrade_to_v1_7/upgrade_to_v1_7_generator.rb
-- lib/generators/lex_llm/upgrade_to_v1_9/templates/add_v1_9_message_columns.rb.tt
-- lib/generators/lex_llm/upgrade_to_v1_9/upgrade_to_v1_9_generator.rb
 - lib/legion/extensions/llm.rb
+- lib/legion/extensions/llm/agent.rb
+- lib/legion/extensions/llm/aliases.json
+- lib/legion/extensions/llm/aliases.rb
+- lib/legion/extensions/llm/attachment.rb
+- lib/legion/extensions/llm/chat.rb
+- lib/legion/extensions/llm/chunk.rb
+- lib/legion/extensions/llm/configuration.rb
+- lib/legion/extensions/llm/connection.rb
+- lib/legion/extensions/llm/content.rb
+- lib/legion/extensions/llm/context.rb
+- lib/legion/extensions/llm/embedding.rb
+- lib/legion/extensions/llm/error.rb
+- lib/legion/extensions/llm/image.rb
+- lib/legion/extensions/llm/message.rb
+- lib/legion/extensions/llm/mime_type.rb
+- lib/legion/extensions/llm/model.rb
+- lib/legion/extensions/llm/model/info.rb
+- lib/legion/extensions/llm/model/modalities.rb
+- lib/legion/extensions/llm/model/pricing.rb
+- lib/legion/extensions/llm/model/pricing_category.rb
+- lib/legion/extensions/llm/model/pricing_tier.rb
+- lib/legion/extensions/llm/models.json
+- lib/legion/extensions/llm/models.rb
+- lib/legion/extensions/llm/models_schema.json
+- lib/legion/extensions/llm/moderation.rb
+- lib/legion/extensions/llm/provider.rb
+- lib/legion/extensions/llm/provider/open_ai_compatible.rb
 - lib/legion/extensions/llm/provider_settings.rb
+- lib/legion/extensions/llm/routing.rb
+- lib/legion/extensions/llm/routing/lane_key.rb
+- lib/legion/extensions/llm/routing/model_offering.rb
+- lib/legion/extensions/llm/stream_accumulator.rb
+- lib/legion/extensions/llm/streaming.rb
+- lib/legion/extensions/llm/thinking.rb
+- lib/legion/extensions/llm/tokens.rb
+- lib/legion/extensions/llm/tool.rb
+- lib/legion/extensions/llm/tool_call.rb
+- lib/legion/extensions/llm/transcription.rb
 - lib/legion/extensions/llm/transport/fleet_lane.rb
-- lib/lex_llm.rb
-- lib/lex_llm/active_record/acts_as.rb
-- lib/lex_llm/active_record/acts_as_legacy.rb
-- lib/lex_llm/active_record/chat_methods.rb
-- lib/lex_llm/active_record/message_methods.rb
-- lib/lex_llm/active_record/model_methods.rb
-- lib/lex_llm/active_record/payload_helpers.rb
-- lib/lex_llm/active_record/tool_call_methods.rb
-- lib/lex_llm/agent.rb
-- lib/lex_llm/aliases.json
-- lib/lex_llm/aliases.rb
-- lib/lex_llm/attachment.rb
-- lib/lex_llm/chat.rb
-- lib/lex_llm/chunk.rb
-- lib/lex_llm/configuration.rb
-- lib/lex_llm/connection.rb
-- lib/lex_llm/content.rb
-- lib/lex_llm/context.rb
-- lib/lex_llm/embedding.rb
-- lib/lex_llm/error.rb
-- lib/lex_llm/image.rb
-- lib/lex_llm/message.rb
-- lib/lex_llm/mime_type.rb
-- lib/lex_llm/model.rb
-- lib/lex_llm/model/info.rb
-- lib/lex_llm/model/modalities.rb
-- lib/lex_llm/model/pricing.rb
-- lib/lex_llm/model/pricing_category.rb
-- lib/lex_llm/model/pricing_tier.rb
-- lib/lex_llm/models.json
-- lib/lex_llm/models.rb
-- lib/lex_llm/models_schema.json
-- lib/lex_llm/moderation.rb
-- lib/lex_llm/provider.rb
-- lib/lex_llm/provider/open_ai_compatible.rb
-- lib/lex_llm/railtie.rb
-- lib/lex_llm/routing.rb
-- lib/lex_llm/routing/lane_key.rb
-- lib/lex_llm/routing/model_offering.rb
-- lib/lex_llm/stream_accumulator.rb
-- lib/lex_llm/streaming.rb
-- lib/lex_llm/thinking.rb
-- lib/lex_llm/tokens.rb
-- lib/lex_llm/tool.rb
-- lib/lex_llm/tool_call.rb
-- lib/lex_llm/transcription.rb
-- lib/lex_llm/utils.rb
-- lib/lex_llm/version.rb
-- lib/tasks/lex_llm.rake
+- lib/legion/extensions/llm/utils.rb
+- lib/legion/extensions/llm/version.rb
 homepage: https://github.com/LegionIO/lex-llm
 licenses:
 - MIT

data/lib/generators/lex_llm/agent/agent_generator.rb

@@ -1,36 +0,0 @@
-# frozen_string_literal: true
-
-require 'rails/generators'
-
-module LexLLM
-  module Generators
-    # Generator for LexLLM agent classes and prompt files.
-    class AgentGenerator < Rails::Generators::NamedBase
-      source_root File.expand_path('templates', __dir__)
-
-      namespace 'lex_llm:agent'
-
-      desc 'Creates a LexLLM agent class and default instructions prompt'
-
-      def create_agent_file
-        template 'agent.rb.tt', File.join('app/agents', class_path, "#{agent_file_name}.rb")
-      end
-
-      def create_prompt_file
-        empty_directory File.join('app/prompts', class_path, agent_file_name)
-        template 'instructions.txt.erb.tt',
-                 File.join('app/prompts', class_path, agent_file_name, 'instructions.txt.erb')
-      end
-
-      private
-
-      def agent_class_name
-        class_name.end_with?('Agent') ? class_name : "#{class_name}Agent"
-      end
-
-      def agent_file_name
-        agent_class_name.demodulize.underscore
-      end
-    end
-  end
-end

data/lib/generators/lex_llm/agent/templates/agent.rb.tt

@@ -1,6 +0,0 @@
-class <%= agent_class_name %> < LexLLM::Agent
-  # Change `Chat` to your app's chat model for Rails persistence.
-  # Remove this line to skip persistence and use plain LexLLM chats.
-  chat_model Chat
-  instructions
-end

data/lib/generators/lex_llm/chat_ui/chat_ui_generator.rb

@@ -1,256 +0,0 @@
-# frozen_string_literal: true
-
-require 'rails/generators'
-require_relative '../generator_helpers'
-
-module LexLLM
-  module Generators
-    # Generates a simple chat UI scaffold for LexLLM
-    class ChatUIGenerator < Rails::Generators::Base
-      include LexLLM::Generators::GeneratorHelpers
-
-      source_root File.expand_path('templates', __dir__)
-
-      namespace 'lex_llm:chat_ui'
-
-      argument :model_mappings, type: :array, default: [], banner: 'chat:ChatName message:MessageName ...'
-      class_option :ui, type: :string, default: 'auto', enum: %w[scaffold tailwind auto],
-                        desc: 'UI template style (scaffold, tailwind, auto)'
-
-      desc 'Creates a chat UI scaffold with Turbo streaming\n' \
-           'Usage: bin/rails g lex_llm:chat_ui [chat:ChatName] [message:MessageName] ...'
-
-      def check_model_exists
-        model_path = "app/models/#{message_model_name.underscore}.rb"
-        return if File.exist?(model_path)
-
-        # Build the argument string for the install/upgrade commands
-        args = []
-        args << "chat:#{chat_model_name}" if chat_model_name != 'Chat'
-        args << "message:#{message_model_name}" if message_model_name != 'Message'
-        args << "model:#{model_model_name}" if model_model_name != 'Model'
-        args << "tool_call:#{tool_call_model_name}" if tool_call_model_name != 'ToolCall'
-        arg_string = args.any? ? " #{args.join(' ')}" : ''
-
-        raise Thor::Error, <<~ERROR
-          Model file not found: #{model_path}
-
-          Please run the install generator first:
-            bin/rails generate lex_llm:install#{arg_string}
-
-          Or if upgrading from <= 1.6.x, run the upgrade generator:
-            bin/rails generate lex_llm:upgrade_to_v1_7#{arg_string}
-        ERROR
-      end
-
-      def create_views
-        # Design contract:
-        # - `scaffold` should stay close to Rails scaffold ERB output.
-        # - `tailwind` should stay close to tailwindcss-rails scaffold output.
-        # - Only small chat-specific affordances should be layered on top.
-        # For namespaced models, use the proper Rails convention path
-        chat_view_path = chat_model_name.underscore.pluralize
-        message_view_path = message_model_name.underscore.pluralize
-        model_view_path = model_model_name.underscore.pluralize
-
-        # Chat views
-        template ui_template('views/chats/index.html.erb'), "app/views/#{chat_view_path}/index.html.erb"
-        template ui_template('views/chats/new.html.erb'), "app/views/#{chat_view_path}/new.html.erb"
-        template ui_template('views/chats/show.html.erb'), "app/views/#{chat_view_path}/show.html.erb"
-        template ui_template('views/chats/_chat.html.erb'),
-                 "app/views/#{chat_view_path}/_#{chat_model_name.demodulize.underscore}.html.erb"
-        template ui_template('views/chats/_form.html.erb'), "app/views/#{chat_view_path}/_form.html.erb"
-
-        # Message views
-        template ui_template('views/messages/_assistant.html.erb'), "app/views/#{message_view_path}/_assistant.html.erb"
-        template ui_template('views/messages/_user.html.erb'), "app/views/#{message_view_path}/_user.html.erb"
-        template ui_template('views/messages/_system.html.erb'), "app/views/#{message_view_path}/_system.html.erb"
-        template ui_template('views/messages/_tool.html.erb'), "app/views/#{message_view_path}/_tool.html.erb"
-        template ui_template('views/messages/_error.html.erb'), "app/views/#{message_view_path}/_error.html.erb"
-        template ui_template('views/messages/_tool_calls.html.erb'),
-                 "app/views/#{message_view_path}/_tool_calls.html.erb"
-        empty_directory "app/views/#{message_view_path}/tool_calls"
-        template ui_template('views/messages/tool_calls/_default.html.erb'),
-                 "app/views/#{message_view_path}/tool_calls/_default.html.erb"
-        empty_directory "app/views/#{message_view_path}/tool_results"
-        template ui_template('views/messages/tool_results/_default.html.erb'),
-                 "app/views/#{message_view_path}/tool_results/_default.html.erb"
-        template ui_template('views/messages/create.turbo_stream.erb'),
-                 "app/views/#{message_view_path}/create.turbo_stream.erb"
-        template ui_template('views/messages/_content.html.erb'), "app/views/#{message_view_path}/_content.html.erb"
-        template ui_template('views/messages/_form.html.erb'), "app/views/#{message_view_path}/_form.html.erb"
-
-        # Model views
-        template ui_template('views/models/index.html.erb'), "app/views/#{model_view_path}/index.html.erb"
-        template ui_template('views/models/show.html.erb'), "app/views/#{model_view_path}/show.html.erb"
-        template ui_template('views/models/_model.html.erb'),
-                 "app/views/#{model_view_path}/_#{model_model_name.demodulize.underscore}.html.erb"
-      end
-
-      def create_controllers
-        # For namespaced models, use the proper Rails convention path
-        chat_controller_path = chat_model_name.underscore.pluralize
-        message_controller_path = message_model_name.underscore.pluralize
-        model_controller_path = model_model_name.underscore.pluralize
-
-        template 'controllers/chats_controller.rb', "app/controllers/#{chat_controller_path}_controller.rb"
-        template 'controllers/messages_controller.rb', "app/controllers/#{message_controller_path}_controller.rb"
-        template 'controllers/models_controller.rb', "app/controllers/#{model_controller_path}_controller.rb"
-      end
-
-      def create_jobs
-        template 'jobs/chat_response_job.rb', "app/jobs/#{variable_name_for(chat_model_name)}_response_job.rb"
-      end
-
-      def create_helpers
-        template 'helpers/messages_helper.rb', "app/helpers/#{message_model_name.underscore.pluralize}_helper.rb"
-      end
-
-      def add_available_chat_models_to_application_controller
-        path = 'app/controllers/application_controller.rb'
-        return unless File.exist?(path)
-
-        application_controller = File.read(path)
-        return if application_controller.include?('def available_chat_models')
-
-        inject_into_file path, <<-RUBY, before: /^end\s*\z/
-  private
-
-  def available_chat_models
-    LexLLM.models.chat_models.all
-           .sort_by { |model| [ model.provider.to_s, model.name.to_s ] }
-  end
-        RUBY
-      end
-
-      def add_routes
-        # For namespaced models, use Rails convention with namespace blocks
-        if chat_model_name.include?('::')
-          namespace = chat_model_name.deconstantize.underscore
-          chat_resource = chat_model_name.demodulize.underscore.pluralize
-          message_resource = message_model_name.demodulize.underscore.pluralize
-          model_resource = model_model_name.demodulize.underscore.pluralize
-
-          routes_content = <<~ROUTES.strip
-            namespace :#{namespace} do
-              resources :#{model_resource}, only: [ :index, :show ] do
-                collection do
-                  post :refresh
-                end
-              end
-              resources :#{chat_resource} do
-                resources :#{message_resource}, only: [ :create ]
-              end
-            end
-          ROUTES
-          route routes_content
-        else
-          model_routes = <<~ROUTES.strip
-            resources :#{model_table_name}, only: [ :index, :show ] do
-              collection do
-                post :refresh
-              end
-            end
-          ROUTES
-          route model_routes
-          chat_routes = <<~ROUTES.strip
-            resources :#{chat_table_name} do
-              resources :#{message_table_name}, only: [ :create ]
-            end
-          ROUTES
-          route chat_routes
-        end
-      end
-
-      def add_broadcasting_to_message_model
-        msg_var = variable_name_for(message_model_name)
-        chat_var = variable_name_for(chat_model_name)
-        msg_path = message_model_name.underscore
-
-        # For namespaced models, we need the association name which might be different
-        # e.g., for LLM::Message, the chat association might be :llm_chat
-        chat_association = chat_table_name.singularize
-
-        broadcasting_callbacks = <<-RUBY
-
-  broadcasts_to ->(#{msg_var}) { "#{chat_var}_\#{#{msg_var}.#{chat_association}_id}" }, inserts_by: :append
-
-  def broadcast_append_chunk(content)
-    broadcast_append_to "#{chat_var}_\#{#{chat_association}_id}",
-      target: "#{msg_var}_\#{id}_content",
-      content: ERB::Util.html_escape(content.to_s)
-  end
-        RUBY
-
-        inject_into_file "app/models/#{msg_path}.rb", before: "end\n" do
-          broadcasting_callbacks
-        end
-      rescue Errno::ENOENT
-        say "#{message_model_name} model not found. Add broadcasting code to your model.", :yellow
-        say broadcasting_callbacks, :yellow
-      end
-
-      def display_post_install_message
-        return unless behavior == :invoke
-
-        # Show the correct URL based on whether models are namespaced
-        url_path = if chat_model_name.include?('::')
-                     chat_model_name.underscore.pluralize
-                   else
-                     chat_table_name
-                   end
-
-        say "\n  ✅ Chat UI installed!", :green
-        say "  UI template: #{ui_variant}", :cyan
-        say "\n  Start your server and visit http://localhost:3000/#{url_path}", :cyan
-        say "\n"
-      end
-
-      private
-
-      def ui_variant
-        @ui_variant ||= case options[:ui]
-                        when 'tailwind'
-                          :tailwind
-                        when 'auto'
-                          tailwind_available? ? :tailwind : :scaffold
-                        else
-                          :scaffold
-                        end
-      end
-
-      def ui_template(template_path)
-        return template_path unless ui_variant == :tailwind
-
-        # Keep Tailwind templates as a separate set so we can mirror Rails/Tailwind
-        # scaffold conventions without complicating scaffold templates.
-        tailwind_template = "tailwind/#{template_path}"
-        File.exist?(File.join(self.class.source_root, "#{tailwind_template}.tt")) ? tailwind_template : template_path
-      end
-
-      def message_helper_module_name
-        if message_model_name.include?('::')
-          "#{message_model_name.deconstantize}::#{message_model_name.demodulize.pluralize}Helper"
-        else
-          "#{message_model_name.pluralize}Helper"
-        end
-      end
-
-      def tailwind_available?
-        Rails.root.join('app/assets/tailwind/application.css').exist? ||
-          Rails.root.join('config/tailwind.config.js').exist? ||
-          gem_in_bundle?('tailwindcss-rails') ||
-          gem_in_bundle?('cssbundling-rails')
-      end
-
-      def gem_in_bundle?(gem_name)
-        gemfile_path = Rails.root.join('Gemfile')
-        lockfile_path = Rails.root.join('Gemfile.lock')
-
-        [gemfile_path, lockfile_path].any? do |path|
-          path.exist? && path.read.include?(gem_name)
-        end
-      end
-    end
-  end
-end

data/lib/generators/lex_llm/chat_ui/templates/controllers/chats_controller.rb.tt

@@ -1,38 +0,0 @@
-class <%= chat_controller_class_name %> < ApplicationController
-  before_action :set_<%= chat_variable_name %>, only: [ :show, :destroy ]
-
-  def index
-    @<%= chat_table_name %> = <%= chat_model_name %>.order(created_at: :desc)
-  end
-
-  def new
-    @<%= chat_variable_name %> = <%= chat_model_name %>.new
-    @selected_model = params[:model]
-    @chat_models = available_chat_models
-  end
-
-  def create
-    prompt = params.dig(:<%= chat_variable_name %>, :prompt)
-    if prompt.present?
-      @<%= chat_variable_name %> = <%= chat_model_name %>.create!(model: params.dig(:<%= chat_variable_name %>, :model).presence)
-      <%= chat_job_class_name %>.perform_later(@<%= chat_variable_name %>.id, prompt)
-
-      redirect_to @<%= chat_variable_name %>, notice: "<%= chat_model_name.humanize %> was successfully created."
-    end
-  end
-
-  def show
-    @<%= message_variable_name %> = @<%= chat_variable_name %>.<%= message_table_name %>.build
-  end
-
-  def destroy
-    @<%= chat_variable_name %>.destroy!
-    redirect_to <%= chat_table_name %>_path, notice: "<%= chat_model_name.humanize %> was successfully destroyed.", status: :see_other
-  end
-
-  private
-
-  def set_<%= chat_variable_name %>
-    @<%= chat_variable_name %> = <%= chat_model_name %>.find(params[:id])
-  end
-end

data/lib/generators/lex_llm/chat_ui/templates/controllers/messages_controller.rb.tt

@@ -1,21 +0,0 @@
-class <%= message_controller_class_name %> < ApplicationController
-  before_action :set_<%= chat_variable_name %>
-
-  def create
-    content = params.dig(:<%= message_variable_name %>, :content)
-    if content.present?
-      <%= chat_job_class_name %>.perform_later(@<%= chat_variable_name %>.id, content)
-
-      respond_to do |format|
-        format.turbo_stream
-        format.html { redirect_to @<%= chat_variable_name %> }
-      end
-    end
-  end
-
-  private
-
-  def set_<%= chat_variable_name %>
-    @<%= chat_variable_name %> = <%= chat_model_name %>.find(params[:<%= chat_model_name.include?('::') ? chat_model_name.demodulize.underscore : chat_variable_name %>_id])
-  end
-end

data/lib/generators/lex_llm/chat_ui/templates/controllers/models_controller.rb.tt

@@ -1,14 +0,0 @@
-class <%= model_controller_class_name %> < ApplicationController
-  def index
-    @<%= model_table_name %> = available_chat_models
-  end
-
-  def show
-    @<%= model_variable_name %> = <%= model_model_name %>.find(params[:id])
-  end
-
-  def refresh
-    <%= model_model_name %>.refresh!
-    redirect_to <%= model_table_name %>_path, notice: "<%= model_model_name.pluralize %> refreshed successfully"
-  end
-end

data/lib/generators/lex_llm/chat_ui/templates/helpers/messages_helper.rb.tt

@@ -1,25 +0,0 @@
-module <%= message_helper_module_name %>
-  def default_model_display_name
-    "Default: #{LexLLM.models.find(LexLLM.config.default_model).label}"
-  end
-
-  def tool_result_partial(message)
-    name = message.respond_to?(:parent_tool_call) ? message.parent_tool_call&.name.to_s : ""
-    partial_for(prefix: "<%= message_model_name.underscore.pluralize %>/tool_results", name: name)
-  end
-
-  def tool_call_partial(tool_call)
-    partial_for(prefix: "<%= message_model_name.underscore.pluralize %>/tool_calls", name: tool_call.name.to_s)
-  end
-
-  private
-
-  def partial_for(prefix:, name:)
-    normalized = name.to_s.underscore.tr("-", "_")
-    if normalized.present? && lookup_context.exists?(normalized, [ prefix ], true)
-      "#{prefix}/#{normalized}"
-    else
-      "#{prefix}/default"
-    end
-  end
-end

data/lib/generators/lex_llm/chat_ui/templates/jobs/chat_response_job.rb.tt

@@ -1,12 +0,0 @@
-class <%= chat_job_class_name %> < ApplicationJob
-  def perform(<%= chat_variable_name %>_id, content)
-    <%= chat_variable_name %> = <%= chat_model_name %>.find(<%= chat_variable_name %>_id)
-
-    <%= chat_variable_name %>.ask(content) do |chunk|
-      if chunk.content && !chunk.content.empty?
-        <%= message_variable_name %> = <%= chat_variable_name %>.<%= message_table_name %>.last
-        <%= message_variable_name %>.broadcast_append_chunk(chunk.content)
-      end
-    end
-  end
-end

data/lib/generators/lex_llm/chat_ui/templates/tailwind/views/chats/_chat.html.erb.tt

@@ -1,16 +0,0 @@
-<div id="<%%= dom_id <%= chat_model_name.demodulize.underscore %> %>" class="w-full sm:w-auto my-5 space-y-3">
-  <div>
-    <strong class="block font-medium mb-1">Model:</strong>
-    <%%= <%= chat_model_name.demodulize.underscore %>.<%= model_table_name.singularize %>&.label || default_model_display_name %>
-  </div>
-
-  <div>
-    <strong class="block font-medium mb-1">Messages:</strong>
-    <%%= <%= chat_model_name.demodulize.underscore %>.<%= message_table_name %>.count %>
-  </div>
-
-  <div>
-    <strong class="block font-medium mb-1">Created:</strong>
-    <%%= <%= chat_model_name.demodulize.underscore %>.created_at.strftime("%B %d, %Y at %I:%M %p") %>
-  </div>
-</div>

data/lib/generators/lex_llm/chat_ui/templates/tailwind/views/chats/_form.html.erb.tt

@@ -1,31 +0,0 @@
-<%%= form_with(model: <%= chat_variable_name %>, url: <%= chat_table_name %>_path, class: "contents") do |form| %>
-  <%% if <%= chat_variable_name %>.errors.any? %>
-    <div id="error_explanation" class="bg-red-50 text-red-500 px-3 py-2 font-medium rounded-md mt-3">
-      <h2><%%= pluralize(<%= chat_variable_name %>.errors.count, "error") %> prohibited this <%= chat_table_name.singularize.humanize.downcase %> from being saved:</h2>
-
-      <ul class="list-disc ml-6">
-        <%% <%= chat_variable_name %>.errors.each do |error| %>
-          <li><%%= error.full_message %></li>
-        <%% end %>
-      </ul>
-    </div>
-  <%% end %>
-
-  <div class="my-5">
-    <%%= form.label :model, "Select AI model:" %>
-    <%%= form.select :model,
-        options_for_select(@chat_models.map { |model| [model.label, model.id] }.unshift([default_model_display_name, nil]), @selected_model),
-        {},
-        class: ["block shadow-sm rounded-md border px-3 py-2 mt-2 w-full", {"border-gray-400 focus:outline-blue-600": <%= chat_variable_name %>.errors[:model].none?, "border-red-400 focus:outline-red-600": <%= chat_variable_name %>.errors[:model].any?}] %>
-  </div>
-
-  <div class="my-5">
-    <%%= form.label :prompt, "Prompt" %>
-    <%%= form.text_area :prompt, rows: 4, placeholder: "What would you like to discuss?", autofocus: true,
-        class: ["block shadow-sm rounded-md border px-3 py-2 mt-2 w-full", {"border-gray-400 focus:outline-blue-600": <%= chat_variable_name %>.errors[:prompt].none?, "border-red-400 focus:outline-red-600": <%= chat_variable_name %>.errors[:prompt].any?}] %>
-  </div>
-
-  <div class="inline">
-    <%%= form.submit "Start new <%= chat_table_name.singularize.humanize.downcase %>", class: "w-full sm:w-auto rounded-md px-3.5 py-2.5 bg-blue-600 hover:bg-blue-500 text-white inline-block font-medium cursor-pointer" %>
-  </div>
-<%% end %>