lex-llm 0.1.1

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

@@ -0,0 +1,256 @@
+# 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

@@ -0,0 +1,38 @@
+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

@@ -0,0 +1,21 @@
+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

@@ -0,0 +1,14 @@
+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

@@ -0,0 +1,25 @@
+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

@@ -0,0 +1,12 @@
+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

@@ -0,0 +1,16 @@
+<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

@@ -0,0 +1,31 @@
+<%%= 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 %>

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

@@ -0,0 +1,31 @@
+<%% content_for :title, "<%= chat_model_name.pluralize %>" %>
+
+<div class="w-full">
+  <%% if notice.present? %>
+    <p class="py-2 px-3 bg-green-50 mb-5 text-green-500 font-medium rounded-md inline-block" id="notice"><%%= notice %></p>
+  <%% end %>
+
+  <div class="flex justify-between items-center">
+    <h1 class="font-bold text-4xl"><%= chat_model_name.pluralize %></h1>
+    <div class="flex items-center gap-2">
+      <%%= link_to "<%= model_model_name.pluralize %>", <%= model_table_name %>_path, class: "rounded-md px-3.5 py-2.5 bg-gray-100 hover:bg-gray-50 text-gray-900 block font-medium" %>
+      <%%= link_to "New <%= chat_table_name.singularize.humanize.downcase %>", new_<%= chat_variable_name %>_path, class: "rounded-md px-3.5 py-2.5 bg-blue-600 hover:bg-blue-500 text-white block font-medium" %>
+    </div>
+  </div>
+
+  <div id="<%= chat_table_name %>" class="min-w-full divide-y divide-gray-200 space-y-5">
+    <%% if @<%= chat_table_name %>.any? %>
+      <%% @<%= chat_table_name %>.each do |<%= chat_variable_name %>| %>
+        <div class="flex flex-col sm:flex-row justify-between items-center pb-5 sm:pb-0">
+          <%%= render <%= chat_variable_name %> %>
+          <div class="w-full sm:w-auto flex flex-col sm:flex-row space-x-2 space-y-2">
+            <%%= link_to "Show", <%= chat_variable_name %>, class: "w-full sm:w-auto text-center rounded-md px-3.5 py-2.5 bg-gray-100 hover:bg-gray-50 inline-block font-medium" %>
+            <%%= button_to "Destroy", <%= chat_variable_name %>, method: :delete, class: "w-full sm:w-auto rounded-md px-3.5 py-2.5 bg-red-600 hover:bg-red-500 text-white inline-block font-medium cursor-pointer", data: { turbo_confirm: "Are you sure?" } %>
+          </div>
+        </div>
+      <%% end %>
+    <%% else %>
+      <p class="text-center my-10">No <%= chat_table_name.humanize.downcase %> found.</p>
+    <%% end %>
+  </div>
+</div>

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

@@ -0,0 +1,9 @@
+<%% content_for :title, "New <%= chat_table_name.singularize.humanize.downcase %>" %>
+
+<div class="md:w-2/3 w-full">
+  <h1 class="font-bold text-4xl">New <%= chat_table_name.singularize.humanize.downcase %></h1>
+
+  <%%= render "form", <%= chat_variable_name %>: @<%= chat_variable_name %> %>
+
+  <%%= link_to "Back to <%= chat_table_name.humanize.downcase %>", <%= chat_table_name %>_path, class: "w-full sm:w-auto text-center mt-2 sm:mt-0 sm:ml-2 rounded-md px-3.5 py-2.5 bg-gray-100 hover:bg-gray-50 inline-block font-medium" %>
+</div>

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

@@ -0,0 +1,27 @@
+<%% content_for :title, "Showing <%= chat_table_name.singularize.humanize.downcase %>" %>
+
+<%%= turbo_stream_from "<%= chat_variable_name %>_#{@<%= chat_variable_name %>.id}" %>
+
+<%# Keep layout conventions aligned with tailwindcss-rails scaffold (top-left, md:w-2/3). -%>
+<div class="md:w-2/3 w-full">
+  <%% if notice.present? %>
+    <p class="py-2 px-3 bg-green-50 mb-5 text-green-500 font-medium rounded-md inline-block" id="notice"><%%= notice %></p>
+  <%% end %>
+
+  <h1 class="font-bold text-4xl">Showing <%= chat_table_name.singularize.humanize.downcase %> #<%%= @<%= chat_variable_name %>.id %></h1>
+
+  <div class="my-5">
+    <strong class="block font-medium mb-1">Model:</strong>
+    <%%= @<%= chat_variable_name %>.<%= model_table_name.singularize %>&.label || default_model_display_name %>
+  </div>
+
+  <div id="<%= message_table_name %>" class="min-w-full divide-y divide-gray-200 space-y-5 my-5">
+    <%% @<%= chat_variable_name %>.<%= message_table_name %>.where.not(id: nil).each do |<%= message_variable_name %>| %>
+      <%%= render <%= message_variable_name %> %>
+    <%% end %>
+  </div>
+
+  <%%= render "<%= message_model_name.underscore.pluralize %>/form", <%= chat_variable_name %>: @<%= chat_variable_name %>, <%= message_variable_name %>: @<%= message_variable_name %> %>
+
+  <%%= link_to "Back to <%= chat_table_name.humanize.downcase %>", <%= chat_table_name %>_path, class: "w-full sm:w-auto text-center mt-2 sm:mt-0 sm:ml-2 rounded-md px-3.5 py-2.5 bg-gray-100 hover:bg-gray-50 inline-block font-medium" %>
+</div>

data/lib/generators/lex_llm/chat_ui/templates/tailwind/views/messages/_assistant.html.erb.tt

@@ -0,0 +1,14 @@
+<%% assistant ||= local_assigns[:message] %>
+<div id="<%= message_variable_name %>_<%%= assistant.id %>" class="w-full sm:w-auto my-5 space-y-3 rounded-md px-3 py-2 bg-green-50">
+  <div>
+    <span class="inline-block rounded px-2 py-0.5 text-xs font-medium bg-green-100 text-green-700">
+      Assistant
+    </span>
+  </div>
+
+  <div id="<%= message_variable_name %>_<%%= assistant.id %>_content" class="whitespace-pre-wrap"><%%= assistant.content %></div>
+
+  <div>
+    <span class="text-sm text-gray-600"><%%= assistant.created_at&.strftime("%I:%M %p") %></span>
+  </div>
+</div>

data/lib/generators/lex_llm/chat_ui/templates/tailwind/views/messages/_content.html.erb.tt

@@ -0,0 +1 @@
+<%%= content -%>

data/lib/generators/lex_llm/chat_ui/templates/tailwind/views/messages/_error.html.erb.tt

@@ -0,0 +1,13 @@
+<div id="<%= message_variable_name %>_<%%= <%= message_model_name.demodulize.underscore %>.id %>" class="w-full sm:w-auto my-5 space-y-3 rounded-md px-3 py-2 bg-red-50 border border-red-200">
+  <div>
+    <span class="inline-block rounded px-2 py-0.5 text-xs font-medium bg-red-100 text-red-700">
+      <%%= title.presence || "Error" %>
+    </span>
+  </div>
+
+  <pre class="whitespace-pre-wrap text-red-900 text-sm overflow-x-auto"><%%= error_message %></pre>
+
+  <div>
+    <span class="text-sm text-red-700"><%%= <%= message_model_name.demodulize.underscore %>.created_at&.strftime("%I:%M %p") %></span>
+  </div>
+</div>

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

@@ -0,0 +1,23 @@
+<%%= form_with(model: <%= message_variable_name %>, url: <%= chat_model_name.include?('::') ? "#{chat_model_name.split('::').first.underscore}_#{chat_model_name.demodulize.underscore}_#{message_model_name.demodulize.underscore.pluralize}_path(@#{chat_variable_name})" : "[@#{chat_variable_name}, #{message_variable_name}]" %>, id: "new_<%= message_variable_name %>", class: "contents") do |form| %>
+  <%% if <%= message_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(<%= message_variable_name %>.errors.count, "error") %> prohibited this <%= message_table_name.singularize.humanize.downcase %> from being saved:</h2>
+
+      <ul class="list-disc ml-6">
+        <%% <%= message_variable_name %>.errors.each do |error| %>
+          <li><%%= error.full_message %></li>
+        <%% end %>
+      </ul>
+    </div>
+  <%% end %>
+
+  <div class="my-5">
+    <%%= form.label :content, "Message" %>
+    <%%= form.text_area :content, rows: 4, placeholder: "Message...", autofocus: true,
+        class: ["block shadow-sm rounded-md border px-3 py-2 mt-2 w-full", {"border-gray-400 focus:outline-blue-600": <%= message_variable_name %>.errors[:content].none?, "border-red-400 focus:outline-red-600": <%= message_variable_name %>.errors[:content].any?}] %>
+  </div>
+
+  <div class="inline">
+    <%%= form.submit "Send <%= message_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 %>

data/lib/generators/lex_llm/chat_ui/templates/tailwind/views/messages/_system.html.erb.tt

@@ -0,0 +1,10 @@
+<%% system ||= local_assigns[:message] %>
+<div id="<%= message_variable_name %>_<%%= system.id %>" class="w-full sm:w-auto my-5 space-y-3 rounded-md px-3 py-2 bg-gray-50 border border-gray-200">
+  <div>
+    <span class="inline-block rounded px-2 py-0.5 text-xs font-medium bg-gray-200 text-gray-700">
+      System
+    </span>
+  </div>
+
+  <div class="whitespace-pre-wrap text-gray-700"><%%= system.content %></div>
+</div>

data/lib/generators/lex_llm/chat_ui/templates/tailwind/views/messages/_tool.html.erb.tt

@@ -0,0 +1,2 @@
+<%% tool ||= local_assigns[:message] %>
+<%%= render tool_result_partial(tool), tool: tool %>

data/lib/generators/lex_llm/chat_ui/templates/tailwind/views/messages/_tool_calls.html.erb.tt

@@ -0,0 +1,4 @@
+<%% tool_calls ||= local_assigns[:message] %>
+<%% tool_calls.<%= tool_call_variable_name.pluralize %>.each do |tool_call| %>
+  <%%= render tool_call_partial(tool_call), tool_calls: tool_calls, tool_call: tool_call %>
+<%% end %>

data/lib/generators/lex_llm/chat_ui/templates/tailwind/views/messages/_user.html.erb.tt

@@ -0,0 +1,14 @@
+<%% user ||= local_assigns[:message] %>
+<div id="<%= message_variable_name %>_<%%= user.id %>" class="w-full sm:w-auto my-5 space-y-3 rounded-md px-3 py-2 bg-blue-50">
+  <div>
+    <span class="inline-block rounded px-2 py-0.5 text-xs font-medium bg-blue-100 text-blue-700">
+      User
+    </span>
+  </div>
+
+  <div id="<%= message_variable_name %>_<%%= user.id %>_content" class="whitespace-pre-wrap"><%%= user.content %></div>
+
+  <div>
+    <span class="text-sm text-gray-600"><%%= user.created_at&.strftime("%I:%M %p") %></span>
+  </div>
+</div>

data/lib/generators/lex_llm/chat_ui/templates/tailwind/views/messages/tool_calls/_default.html.erb.tt

@@ -0,0 +1,13 @@
+<div id="<%= message_variable_name %>_<%%= tool_calls.id %>" class="w-full sm:w-auto my-5 space-y-3 rounded-md px-3 py-2 bg-gray-50 border border-gray-200">
+  <div>
+    <span class="inline-block rounded px-2 py-0.5 text-xs font-medium bg-gray-200 text-gray-700">
+      Tool Call
+    </span>
+  </div>
+
+  <pre class="whitespace-pre-wrap text-gray-700 text-sm overflow-x-auto"><%%= tool_call.name %>(<%%= tool_call.arguments.map { |k, v| "#{k}: #{v.inspect}" }.join(", ") %>)</pre>
+
+  <div>
+    <span class="text-sm text-gray-600"><%%= tool_calls.created_at&.strftime("%I:%M %p") %></span>
+  </div>
+</div>

data/lib/generators/lex_llm/chat_ui/templates/tailwind/views/messages/tool_results/_default.html.erb.tt

@@ -0,0 +1,21 @@
+<%% error_message = tool.tool_error_message %>
+<%% if error_message.present? %>
+  <%%= render "<%= message_model_name.underscore.pluralize %>/error",
+    <%= message_model_name.demodulize.underscore %>: tool,
+    title: "Tool Result Error",
+    error_message: error_message %>
+<%% else %>
+  <div id="<%= message_variable_name %>_<%%= tool.id %>" class="w-full sm:w-auto my-5 space-y-3 rounded-md px-3 py-2 bg-gray-50 border border-gray-200">
+    <div>
+      <span class="inline-block rounded px-2 py-0.5 text-xs font-medium bg-gray-200 text-gray-700">
+        Tool
+      </span>
+    </div>
+
+    <pre class="whitespace-pre-wrap text-gray-700 text-sm overflow-x-auto"><%%= tool.content.presence || "(no output)" %></pre>
+
+    <div>
+      <span class="text-sm text-gray-600"><%%= tool.created_at&.strftime("%I:%M %p") %></span>
+    </div>
+  </div>
+<%% end %>

data/lib/generators/lex_llm/chat_ui/templates/tailwind/views/models/_model.html.erb.tt

@@ -0,0 +1,17 @@
+<%% model_id = <%= model_model_name.demodulize.underscore %>.respond_to?(:model_id) ? <%= model_model_name.demodulize.underscore %>.model_id : <%= model_model_name.demodulize.underscore %>.id %>
+<%% row_id = [<%= model_model_name.demodulize.underscore %>.provider, model_id].join("_").parameterize(separator: "_") %>
+<tr id="model_<%%= row_id %>">
+  <td class="px-3 py-2 text-sm text-gray-700"><%%= <%= model_model_name.demodulize.underscore %>.provider_class&.name || <%= model_model_name.demodulize.underscore %>.provider %></td>
+  <td class="px-3 py-2 text-sm text-gray-900"><%%= <%= model_model_name.demodulize.underscore %>.respond_to?(:display_name) ? <%= model_model_name.demodulize.underscore %>.display_name : <%= model_model_name.demodulize.underscore %>.name %></td>
+  <td class="px-3 py-2 text-sm text-gray-700"><%%= number_with_delimiter(<%= model_model_name.demodulize.underscore %>.context_window) if <%= model_model_name.demodulize.underscore %>.context_window %></td>
+  <td class="px-3 py-2 text-sm text-gray-700">
+    <%% input = <%= model_model_name.demodulize.underscore %>.input_price_per_million %>
+    <%% output = <%= model_model_name.demodulize.underscore %>.output_price_per_million %>
+    <%% if input && output %>
+      $<%%= "%.2f" % input %> / $<%%= "%.2f" % output %>
+    <%% end %>
+  </td>
+  <td class="px-3 py-2 text-sm">
+    <%%= link_to "Start <%= chat_table_name.singularize.humanize.downcase %>", new_<%= chat_variable_name %>_path(model: model_id), class: "text-blue-600 hover:text-blue-500" %>
+  </td>
+</tr>