enhancements

Author: andreibondarev

lib/langchainrb_rails/generators/langchainrb_rails/assistant_generator.rb

@@ -74,6 +74,7 @@ def add_routes
         EOS
       end
 
+      # TODO: Copy stylesheet into app/assets/stylesheets or whatever the host app uses
       def copy_stylesheets
         template "assistant/stylesheets/chat.css", "app/assets/stylesheets/chat.css"
       end

lib/langchainrb_rails/generators/langchainrb_rails/templates/assistant/controllers/assistants_controller.rb

@@ -42,6 +42,8 @@ def chat
   private
 
   def assistant_params
-    params.require(:assistant).permit(:instructions, :tool_choice)
+    params
+      .require(:assistant)
+      .permit(:name, :instructions, :tool_choice)
   end
 end

lib/langchainrb_rails/generators/langchainrb_rails/templates/assistant/migrations/create_assistants.rb.tt

@@ -1,6 +1,7 @@
 class <%= migration_class_name %> < ActiveRecord::Migration<%= migration_version %>
   def change
     create_table :assistants do |t|
+      t.string :name, null: false
       t.string :instructions
       t.string :tool_choice
       t.json :tools, default: []

lib/langchainrb_rails/generators/langchainrb_rails/templates/assistant/migrations/create_messages.rb.tt

@@ -2,7 +2,7 @@ class <%= migration_class_name %> < ActiveRecord::Migration<%= migration_version
   def change
     create_table :messages do |t|
       t.references :assistant, foreign_key: true
-      t.string :role
+      t.string :role, null: false
       t.text :content
       t.json :tool_calls, default: []
       t.string :tool_call_id

lib/langchainrb_rails/generators/langchainrb_rails/templates/assistant/models/assistant.rb.tt

@@ -3,6 +3,10 @@
 class Assistant < ActiveRecord::Base
   has_many :messages
 
+  validates :name, presence: true
+
+  # TODO: Validate tool_choice
+
   def llm
     <%= llm_class %>.new(api_key: ENV["<%= llm.upcase %>_API_KEY"])
   end

lib/langchainrb_rails/generators/langchainrb_rails/templates/assistant/models/message.rb.tt

@@ -2,4 +2,6 @@
 
 class Message < ActiveRecord::Base
   belongs_to :assistant
+
+  validates :role, presence: true
 end

lib/langchainrb_rails/generators/langchainrb_rails/templates/assistant/stylesheets/chat.css

@@ -249,3 +249,22 @@
 .chat-container {
   font-family: Arial, sans-serif;
 }
+
+.chat-instructions {
+  font-style: italic;
+  color: #666;
+  margin-bottom: 20px;
+}
+
+.form-group input[type="text"] {
+  width: 100%;
+  padding: 10px;
+  border: 1px solid #ccc;
+  border-radius: 4px;
+  font-size: 16px;
+}
+
+.form-group input[type="text"]:focus {
+  outline: none;
+  border-color: #007bff;
+}

lib/langchainrb_rails/generators/langchainrb_rails/templates/assistant/views/index.html.erb

@@ -6,7 +6,7 @@
   <ul class="assistant-list">
     <%% @assistants.each do |assistant| %>
       <li class="assistant-item">
-        <%%= link_to assistant.instructions, assistant_path(assistant), class: "assistant-link" %>
+        <%%= link_to assistant.name, assistant_path(assistant), class: "assistant-link" %>
       </li>
     <%% end %>
   </ul>

lib/langchainrb_rails/generators/langchainrb_rails/templates/assistant/views/new.html.erb

@@ -1,7 +1,7 @@
 <div class="assistant-form-container">
   <h1 class="form-header">Create New Assistant</h1>
 
-  <%%= form_with(model: @assistant, local: true, class: "assistant-form") do |form| %>
+  <%%= form_with(model: @assistant, local: true, class: "assistant-form", data: { turbo: false }) do |form| %>
     <%% if @assistant.errors.any? %>
       <div class="error-messages">
         <h2><%%= pluralize(@assistant.errors.count, "error") %> prohibited this assistant from being saved:</h2>
@@ -13,6 +13,11 @@
       </div>
     <%% end %>
 
+    <div class="form-group">
+      <%%= form.label :name, "Assistant Name" %>
+      <%%= form.text_field :name, placeholder: "Enter assistant name..." %>
+    </div>
+
     <div class="form-group">
       <%%= form.label :instructions, "Instructions for the Assistant" %>
       <%%= form.text_area :instructions, rows: 5, placeholder: "Enter instructions for the assistant..." %>

lib/langchainrb_rails/generators/langchainrb_rails/templates/assistant/views/show.html.erb

@@ -1,5 +1,6 @@
 <div class="chat-container">
-  <h1 class="chat-header">Assistant: <%%= @assistant.id %></h1>
+  <h1 class="chat-header"><%%= @assistant.name %></h1>
+  <p class="chat-instructions"><%%= @assistant.instructions %></p>
 
   <div id="chat-messages" class="chat-messages">
     <%%= render partial: "assistants/message", collection: @messages %>