ruby_llm 1.16.0 → 2.0.0.rc2

data/lib/generators/ruby_llm/provider/templates/gem/spec_helper.rb.erb

@@ -0,0 +1,26 @@
+# frozen_string_literal: true
+
+require 'bundler/setup'
+require 'dotenv/load'
+require 'fileutils'
+
+require 'vcr'
+require '<%= require_path %>'
+require 'webmock/rspec'
+
+Dir[File.expand_path('support/**/*.rb', __dir__)].each { |file| require file }
+
+RSpec.configure do |config|
+  config.disable_monkey_patching!
+  config.expect_with(:rspec) { |expectations| expectations.syntax = :expect }
+  config.order = :random
+  Kernel.srand config.seed
+
+  config.around(:each, :live) do |example|
+    cassette_name = example.full_description.downcase.gsub(/[^a-z0-9]+/, '_').delete_prefix('_').delete_suffix('_')
+    cassette_path = File.join(VCR.configuration.cassette_library_dir, "#{cassette_name}.yml")
+
+    VCR.use_cassette(cassette_name) { example.run }
+    FileUtils.rm_f(cassette_path) if example.exception
+  end
+end

data/lib/generators/ruby_llm/provider/templates/gem/speech_spec.rb.erb

@@ -0,0 +1,25 @@
+# frozen_string_literal: true
+
+require 'spec_helper'
+
+RSpec.describe RubyLLM::Speech, :live do
+  include_context 'with configured RubyLLM'
+
+  each_model(SPEECH_MODELS) do |provider, model, model_info|
+    it "#{provider}/#{model} speaks" do
+      speech = RubyLLM.speak(
+        'Ruby is a programming language designed for developer happiness.',
+        model: model,
+        provider: provider,
+        assume_model_exists: true,
+        voice: model_info[:voice],
+        format: model_info[:format]
+      )
+
+      expect(speech.data).to be_a(String)
+      expect(speech.data.bytesize).to be > 1000
+      expect(speech.model).to eq(model)
+      expect(speech.mime_type).to start_with('audio/')
+    end
+  end
+end

data/lib/generators/ruby_llm/provider/templates/gem/vcr_configuration.rb.erb

@@ -0,0 +1,16 @@
+# frozen_string_literal: true
+
+VCR.configure do |config|
+  config.cassette_library_dir = 'spec/fixtures/vcr_cassettes'
+  config.hook_into :webmock
+  config.default_cassette_options = { record: ENV['CI'] ? :none : :once }
+  config.allow_http_connections_when_no_cassette = true
+  config.filter_sensitive_data('<%= "<#{api_key_env}>" %>') { ENV.fetch('<%= api_key_env %>', nil) }
+  config.filter_sensitive_data('<%= "<#{api_base_env}>" %>') { ENV.fetch('<%= api_base_env %>', nil) }
+
+  config.before_record do |interaction|
+    next unless interaction.request.headers['Authorization']
+
+    interaction.request.headers['Authorization'] = ['Bearer <AUTH_TOKEN>']
+  end
+end

data/lib/generators/ruby_llm/provider/templates/gem/video_spec.rb.erb

@@ -0,0 +1,27 @@
+# frozen_string_literal: true
+
+require 'spec_helper'
+
+RSpec.describe RubyLLM::Video, :live do
+  include_context 'with configured RubyLLM'
+
+  before do
+    RubyLLM.config.video_generation_poll_interval = VCR.current_cassette&.recording? ? 5 : 0
+  end
+
+  each_model(VIDEO_GENERATION_MODELS) do |provider, model, model_info|
+    it "#{provider}/#{model} animates a video" do
+      video = RubyLLM.animate(
+        'a calm ocean wave at sunset',
+        model: model,
+        provider: provider,
+        assume_model_exists: true,
+        provider_options: model_info.fetch(:provider_options, {})
+      )
+
+      expect(video.mime_type).to include('video')
+      expect(video.url || video.data).not_to be_nil
+      expect(video.to_blob.bytesize).to be > 10_000
+    end
+  end
+end

data/lib/generators/ruby_llm/schema/schema_generator.rb

@@ -13,7 +13,7 @@ module RubyLLM
       desc 'Creates a RubyLLM schema class'
 
       def create_schema_file
-        template 'schema.rb.tt', File.join('app/schemas', class_path, "#{file_name}.rb")
+        template 'schema.rb.tt', File.join('app/schemas', class_path, "#{schema_file_name}.rb")
       end
 
       private
@@ -21,6 +21,10 @@ module RubyLLM
       def schema_class_name
         class_name.end_with?('Schema') ? class_name : "#{class_name}Schema"
       end
+
+      def schema_file_name
+        schema_class_name.demodulize.underscore
+      end
     end
   end
 end

data/lib/generators/ruby_llm/schema/templates/schema.rb.tt

@@ -1,2 +1,2 @@
-class <%= schema_class_name %> < RubyLLM::Schema
+class <%= schema_class_name %> < Schematist::Schema
 end

data/lib/generators/ruby_llm/tool/templates/tailwind/tool_call.html.erb.tt

@@ -0,0 +1,13 @@
+<div id="<%= message_variable_name %>_tool_call_<%%= tool_call.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_display_name %> 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"><%%= message.created_at&.strftime("%I:%M %p") %></span>
+  </div>
+</div>

data/lib/generators/ruby_llm/tool/templates/tailwind/tool_result.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_display_name %> Result
+      </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/ruby_llm/tool/templates/tool.rb.tt

@@ -1,9 +1,9 @@
-class <%= class_name %>Tool < RubyLLM::Tool
-  desc "TODO: describe what this tool does"
+class <%= tool_class_name %> < RubyLLM::Tool
+  description "TODO: describe what this tool does"
 
   def execute
     # TODO: return something to the LLM.
     # Anything you return here gets converted to a string.
-    "TODO: implement <%= class_name %>Tool"
+    "TODO: implement <%= tool_class_name %>"
   end
 end

data/lib/generators/ruby_llm/tool/templates/tool_call.html.erb.tt

@@ -1,13 +1,8 @@
-<%% tool_call_error = tool_call.tool_error_message %>
-<%% if tool_call_error.present? %>
-  <%%= render "messages/error", message: tool_calls, title: "Tool Call Error", error_message: tool_call_error %>
-<%% else %>
-  <div id="message_<%%= tool_calls.id %>" class="message"
-       style="margin-bottom: 20px; padding: 10px; border-left: 3px solid #6b7280; background: #f9fafb;">
-    <div style="font-weight: bold; margin-bottom: 5px;"><%= tool_display_name %> Call</div>
-    <pre style="white-space: pre-wrap; margin: 0;"><%%= tool_call.name %>(<%%= tool_call.arguments.map { |k, v| "#{k}: #{v.inspect}" }.join(", ") %>)</pre>
-    <div style="font-size: 0.85em; color: #666; margin-top: 5px;">
-      <%%= tool_calls.created_at&.strftime("%I:%M %p") %>
-    </div>
+<div id="<%= message_variable_name %>_tool_call_<%%= tool_call.id %>" class="<%= message_variable_name %>"
+     style="margin-bottom: 20px; padding: 10px; border-left: 3px solid #6b7280; background: #f9fafb;">
+  <div style="font-weight: bold; margin-bottom: 5px;"><%= tool_display_name %> Call</div>
+  <pre style="white-space: pre-wrap; margin: 0;"><%%= tool_call.name %>(<%%= tool_call.arguments.map { |k, v| "#{k}: #{v.inspect}" }.join(", ") %>)</pre>
+  <div style="font-size: 0.85em; color: #666; margin-top: 5px;">
+    <%%= message.created_at&.strftime("%I:%M %p") %>
   </div>
-<%% end %>
+</div>

data/lib/generators/ruby_llm/tool/templates/tool_result.html.erb.tt

@@ -1,8 +1,11 @@
 <%% error_message = tool.tool_error_message %>
 <%% if error_message.present? %>
-  <%%= render "messages/error", message: tool, title: "Tool Result Error", error_message: error_message %>
+  <%%= 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_<%%= tool.id %>" class="message"
+  <div id="<%= message_variable_name %>_<%%= tool.id %>" class="<%= message_variable_name %>"
        style="margin-bottom: 20px; padding: 10px; border-left: 3px solid #6b7280; background: #f9fafb;">
     <div style="font-weight: bold; margin-bottom: 5px;"><%= tool_display_name %> Result</div>
     <pre style="white-space: pre-wrap; margin: 0;"><%%= tool.content.presence || "(no output)" %></pre>

data/lib/generators/ruby_llm/tool/tool_generator.rb

@@ -1,95 +1,61 @@
 # frozen_string_literal: true
 
 require 'rails/generators'
+require_relative '../generator_helpers'
 
 module RubyLLM
   module Generators
     # Generator for RubyLLM tool classes and related message partials.
     class ToolGenerator < Rails::Generators::NamedBase
+      include RubyLLM::Generators::GeneratorHelpers
+
       source_root File.expand_path('templates', __dir__)
 
       namespace 'ruby_llm:tool'
 
-      check_class_collision suffix: 'Tool'
+      argument :model_mappings, type: :array, default: [], banner: 'message:MessageName'
+
+      class_option :ui, type: :string, default: 'auto', enum: %w[scaffold tailwind auto],
+                        desc: 'UI template style (scaffold, tailwind, auto)'
 
       desc 'Creates a RubyLLM tool class and matching tool call/result view partials'
 
+      def check_tool_class_collision
+        class_collisions tool_class_name
+      end
+
       def create_tool_file
-        template 'tool.rb.tt', File.join('app/tools', class_path, "#{file_name}_tool.rb")
+        template 'tool.rb.tt', File.join('app/tools', class_path, "#{tool_file_name}.rb")
       end
 
       def create_tool_view_partials
-        empty_directory 'app/views/messages/tool_calls'
-        empty_directory 'app/views/messages/tool_results'
-
-        create_tool_call_partial
-        create_tool_result_partial
+        template ui_template('tool_call.html.erb'),
+                 File.join(message_view_path, 'tool_calls', "_#{tool_partial_name}.html.erb")
+        template ui_template('tool_result.html.erb'),
+                 File.join(message_view_path, 'tool_results', "_#{tool_partial_name}.html.erb")
       end
 
       private
 
-      def create_tool_call_partial
-        destination_path = File.join('app/views/messages/tool_calls', "_#{tool_partial_name}.html.erb")
-        default_partial_path = File.join(destination_root, 'app/views/messages/tool_calls/_default.html.erb')
-
-        if File.exist?(default_partial_path)
-          default_markup = tool_named_call_markup(File.read(default_partial_path))
-          indented_markup = indent_non_empty_lines(default_markup, 2)
-          create_file destination_path, <<~ERB
-            <% tool_call_error = tool_call.tool_error_message %>
-            <% if tool_call_error.present? %>
-              <%= render "messages/error", message: tool_calls, title: "Tool Call Error", error_message: tool_call_error %>
-            <% else %>
-            #{indented_markup}<% end %>
-          ERB
-        else
-          template 'tool_call.html.erb.tt', destination_path
-        end
-
-        strip_trailing_whitespace(destination_path)
-      end
-
-      def create_tool_result_partial
-        destination_path = File.join('app/views/messages/tool_results', "_#{tool_partial_name}.html.erb")
-        default_partial_path = File.join(destination_root, 'app/views/messages/tool_results/_default.html.erb')
-
-        if File.exist?(default_partial_path)
-          create_file destination_path, tool_named_result_markup(File.read(default_partial_path))
-        else
-          template 'tool_result.html.erb.tt', destination_path
-        end
-
-        strip_trailing_whitespace(destination_path)
+      def message_view_path
+        File.join('app/views', message_model_name.underscore.pluralize)
       end
 
-      def tool_named_call_markup(markup)
-        markup.sub('Tool Call', "#{tool_display_name} Call")
+      def tool_class_name
+        class_name.end_with?('Tool') ? class_name : "#{class_name}Tool"
       end
 
-      def tool_named_result_markup(markup)
-        markup.sub(/\bTool\b(?!\s*Result)/, "#{tool_display_name} Result")
+      def tool_file_name
+        tool_class_name.demodulize.underscore
       end
 
       def tool_display_name
-        class_name.demodulize
+        tool_class_name.demodulize.delete_suffix('Tool')
       end
 
+      # Mirrors RubyLLM::Tool.tool_name, the key the runtime partial lookup uses.
       def tool_partial_name
-        file_name.delete_suffix('_tool')
-      end
-
-      def indent_non_empty_lines(markup, spaces)
-        indentation = ' ' * spaces
-        markup.lines.map { |line| line.strip.empty? ? line : "#{indentation}#{line}" }.join
-      end
-
-      def strip_trailing_whitespace(path)
-        content = File.read(path)
-        stripped_content = content.lines.map(&:rstrip).join("\n")
-        stripped_content = "#{stripped_content}\n" unless stripped_content.end_with?("\n")
-        return if content == stripped_content
-
-        File.write(path, stripped_content)
+        tool_file_name.delete_suffix('_tool')
       end
     end
   end