WIP: Adding rough views+controller to be used by the assistant generator

Author: andreibondarev

lib/langchainrb_overrides/assistant.rb

@@ -58,7 +58,8 @@ def load(id)
           llm: ar_assistant.llm,
           tools: tools,
           instructions: ar_assistant.instructions,
-          tool_choice: ar_assistant.tool_choice
+          # Default to auto to match the behavior of the original Langchain::Assistant
+          tool_choice: ar_assistant.tool_choice || "auto"
         )
 
         ar_assistant.messages.each do |ar_message|

lib/langchainrb_rails/generators/langchainrb_rails/assistant_generator.rb

@@ -51,6 +51,26 @@ def migration_version
         "[#{::ActiveRecord::VERSION::MAJOR}.#{::ActiveRecord::VERSION::MINOR}]"
       end
 
+      def create_controller_file
+        template "assistant/controllers/assistants_controller.rb", "app/controllers/assistants_controller.rb"
+      end
+
+      def create_view_files
+        template "assistant/views/index.html.erb", "app/views/assistants/index.html.erb"
+        template "assistant/views/new.html.erb", "app/views/assistants/new.html.erb"
+        template "assistant/views/show.html.erb", "app/views/assistants/show.html.erb"
+      end
+
+      def add_routes
+        route <<~EOS
+          resources :assistants do
+            member do
+              post 'chat'
+            end
+          end
+        EOS
+      end
+
       # TODO: Depending on the LLM provider, we may need to add additional gems
       # def add_to_gemfile
       # end

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

@@ -0,0 +1,43 @@
+# frozen_string_literal: true
+
+class AssistantsController < ApplicationController
+  def index
+    @assistants = Assistant.all
+  end
+
+  def new
+    @assistant = Assistant.new
+  end
+
+  def create
+    @assistant = Assistant.new(assistant_params)
+    if @assistant.save
+      redirect_to @assistant, notice: 'Assistant was successfully created.'
+    else
+      render :new
+    end
+  end
+
+  def show
+    @assistant = Assistant.find(params[:id])
+    @messages = @assistant.messages
+  end
+
+  def chat
+    @assistant = Assistant.find(params[:id])
+    message = @assistant.messages.create(role: 'user', content: params[:content])
+    
+    langchain_assistant = Langchain::Assistant.load(@assistant.id)
+    response = langchain_assistant.chat(params[:content])
+    
+    @assistant.messages.create(role: 'assistant', content: response.content)
+    
+    render json: { message: message, response: response.content }
+  end
+
+  private
+
+  def assistant_params
+    params.require(:assistant).permit(:instructions)
+  end
+end

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

@@ -3,7 +3,7 @@ class <%= migration_class_name %> < ActiveRecord::Migration<%= migration_version
     create_table :assistants do |t|
       t.string :instructions
       t.string :tool_choice
-      t.json :tools
+      t.json :tools, default: []
       t.timestamps
     end
   end

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

@@ -4,7 +4,7 @@ class <%= migration_class_name %> < ActiveRecord::Migration<%= migration_version
       t.references :assistant, foreign_key: true
       t.string :role
       t.text :content
-      t.json :tool_calls
+      t.json :tool_calls, default: []
       t.string :tool_call_id
       t.timestamps
     end

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

@@ -0,0 +1,9 @@
+<h1>Assistants</h1>
+
+<%%= link_to 'New Assistant', new_assistant_path %>
+
+<ul>
+  <%% @assistants.each do |assistant| %>
+    <li><%%= link_to assistant.instructions, assistant_path(assistant) %></li>
+  <%% end %>
+</ul>

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

@@ -0,0 +1,10 @@
+<h1>New Assistant</h1>
+
+<%%= form_with(model: @assistant, local: true) do |form| %>
+  <div>
+    <%%= form.label :instructions %>
+    <%%= form.text_area :instructions %>
+  </div>
+
+  <%%= form.submit %>
+<%% end %>

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

@@ -0,0 +1,21 @@
+<h1>Assistant: <%%= @assistant.instructions %></h1>
+
+<div id="chat-messages">
+  <%% @messages.each do |message| %>
+    <p><strong><%%= message.role %>:</strong> <%%= message.content %></p>
+  <%% end %>
+</div>
+
+<%%= form_with(url: chat_assistant_path(@assistant), method: :post, remote: true) do |form| %>
+  <%%= form.text_area :content %>
+  <%%= form.submit 'Send' %>
+<%% end %>
+
+<script>
+  document.addEventListener('ajax:success', function(event) {
+    const [data, _status, _xhr] = event.detail;
+    const chatMessages = document.getElementById('chat-messages');
+    chatMessages.innerHTML += `<p><strong>user:</strong> ${data.message.content}</p>`;
+    chatMessages.innerHTML += `<p><strong>assistant:</strong> ${data.response}</p>`;
+  });
+</script>