dify_llm 1.8.0 → 1.8.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 80cc1d17f6a70b28e6f8cfe72ceb5ac79398e5b569f2519a0ba9fe34cba3ae30
-  data.tar.gz: 429561f0d6191e55248ef5610eb42a52dc70acfb97bea116ee70ae53a0919c5f
+  metadata.gz: 43597420591517c392f1551e766fbdddee57f7d28162d792ef3cb3ab52ebfd81
+  data.tar.gz: b8884ccdbff56ff584ac737848f43ae9dd73c4a59ca3f8b20373d3197a77e847
 SHA512:
-  metadata.gz: f4477d96e5572468bb8128105d97eac680e661da7182d7638e789cf8a6908a48ac9719e884995f39eb40093edc10986afee5137f238129a80d5726bcd0881f8f
-  data.tar.gz: ea459447ae6000b3f044da995516c9aeed5683c2c8801ab633601d8c530470d5b35f8076b4592525dc501b36d8a867224be2fcb47cbedbbb3d6ba6bd4dc3b863
+  metadata.gz: 8c587e998fce443c8fa6940144702061bff43d45f54aea0e2d4435d5181aa698ec2fcae0f9d1fac04639fdf3a06df151f3230ae441024abe37b35f0744cdbe96
+  data.tar.gz: 3403bdf1b3dc9317cd0a21d2ea6dd0063b798bb8a7567dc6cdc8ba9f9a49589d691f6441d887039ea55adbfa649140954d9f9d4558f717c5979aea0f9bb71fde

data/lib/dify_llm.rb

@@ -0,0 +1,3 @@
+# frozen_string_literal: true
+
+require 'ruby_llm'

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

@@ -56,6 +56,7 @@ module RubyLLM
         # Message views
         template 'views/messages/_message.html.erb',
                  "app/views/#{message_table_name}/_#{message_model_name.underscore}.html.erb"
+        template 'views/messages/_content.html.erb', "app/views/#{message_table_name}/_content.html.erb"
         template 'views/messages/_form.html.erb', "app/views/#{message_table_name}/_form.html.erb"
         template 'views/messages/create.turbo_stream.erb',
                  "app/views/#{message_table_name}/create.turbo_stream.erb"
@@ -99,12 +100,23 @@ module RubyLLM
         chat_var = chat_model_name.underscore
         broadcasting_code = "broadcasts_to ->(#{msg_var}) { \"#{chat_var}_\#{#{msg_var}.#{chat_var}_id}\" }"
 
-        inject_into_class "app/models/#{msg_var}.rb", message_model_name do
-          "\n  #{broadcasting_code}\n"
+        broadcast_append_chunk_method = <<-RUBY
+
+  def broadcast_append_chunk(content)
+    broadcast_append_to "#{chat_var}_\#{#{chat_var}_id}",
+      target: "#{msg_var}_\#{id}_content",
+      partial: "#{message_table_name}/content",
+      locals: { content: content }
+  end
+        RUBY
+
+        inject_into_file "app/models/#{msg_var}.rb", before: "end\n" do
+          "  #{broadcasting_code}\n#{broadcast_append_chunk_method}"
         end
       rescue Errno::ENOENT
         say "#{message_model_name} model not found. Add broadcasting code to your model.", :yellow
         say "  #{broadcasting_code}", :yellow
+        say broadcast_append_chunk_method, :yellow
       end
 
       def display_post_install_message

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

@@ -5,7 +5,7 @@ class <%= chat_model_name %>ResponseJob < ApplicationJob
     <%= chat_model_name.underscore %>.ask(content) do |chunk|
       if chunk.content && !chunk.content.blank?
         <%= message_model_name.underscore %> = <%= chat_model_name.underscore %>.<%= message_model_name.tableize %>.last
-        <%= message_model_name.underscore %>.update!(content: <%= message_model_name.underscore %>.content + chunk.content)
+        <%= message_model_name.underscore %>.broadcast_append_chunk(chunk.content)
       end
     end
   end

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

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

data/lib/generators/ruby_llm/chat_ui/templates/views/messages/_message.html.erb.tt

@@ -3,7 +3,7 @@
   <div style="font-weight: bold; margin-bottom: 5px;">
     <%%= <%= message_model_name.underscore %>.role&.capitalize %>
   </div>
-  <div style="white-space: pre-wrap;"><%%= <%= message_model_name.underscore %>.content %></div>
+  <div id="<%%= dom_id(<%= message_model_name.underscore %>) %>_content" style="white-space: pre-wrap;"><%%= <%= message_model_name.underscore %>.content %></div>
   <div style="font-size: 0.85em; color: #666; margin-top: 5px;">
     <%%= <%= message_model_name.underscore %>.created_at&.strftime("%I:%M %p") %>
   </div>