ruby_llm_community 1.1.1 → 1.3.0

data/lib/ruby_llm/providers/vertexai.rb

@@ -8,6 +8,7 @@ module RubyLLM
       include VertexAI::Streaming
       include VertexAI::Embeddings
       include VertexAI::Models
+      include VertexAI::Transcription
 
       def initialize(config)
         super
@@ -37,6 +38,8 @@ module RubyLLM
 
         initialize_authorizer unless @authorizer
         @authorizer.fetch_access_token!['access_token']
+      rescue Google::Auth::AuthorizationError => e
+        raise UnauthorizedError.new(nil, "Invalid Google Cloud credentials for Vertex AI: #{e.message}")
       end
 
       def initialize_authorizer

data/lib/ruby_llm/railtie.rb

@@ -5,7 +5,7 @@ module RubyLLM
   class Railtie < Rails::Railtie
     initializer 'ruby_llm.inflections' do
       ActiveSupport::Inflector.inflections(:en) do |inflect|
-        inflect.acronym 'LLM'
+        inflect.acronym 'RubyLLM'
       end
     end
 

data/lib/ruby_llm/stream_accumulator.rb

@@ -8,10 +8,10 @@ module RubyLLM
     def initialize
       @content = nil
       @tool_calls = {}
-      @input_tokens = 0
-      @output_tokens = 0
-      @cached_tokens = 0
-      @cache_creation_tokens = 0
+      @input_tokens = nil
+      @output_tokens = nil
+      @cached_tokens = nil
+      @cache_creation_tokens = nil
       @latest_tool_call_id = nil
       @reasoning_id = nil
     end
@@ -40,10 +40,10 @@ module RubyLLM
         content: content,
         model_id: model_id,
         tool_calls: tool_calls_from_stream,
-        input_tokens: positive_or_nil(@input_tokens),
-        output_tokens: positive_or_nil(@output_tokens),
-        cached_tokens: positive_or_nil(@cached_tokens),
-        cache_creation_tokens: positive_or_nil(@cache_creation_tokens),
+        input_tokens: @input_tokens,
+        output_tokens: @output_tokens,
+        cached_tokens: @cached_tokens,
+        cache_creation_tokens: @cache_creation_tokens,
         raw: response
       )
     end
@@ -58,10 +58,6 @@ module RubyLLM
       end
     end
 
-    def positive_or_nil(value)
-      value.positive? ? value : nil
-    end
-
     def accumulate_content(new_content)
       return unless new_content
 

data/lib/ruby_llm/tool.rb

@@ -1,5 +1,7 @@
 # frozen_string_literal: true
 
+require 'ruby_llm/schema'
+
 module RubyLLM
   # Parameter definition for Tool methods.
   class Parameter
@@ -29,6 +31,8 @@ module RubyLLM
     end
 
     class << self
+      attr_reader :params_schema_definition
+
       def description(text = nil)
         return @description unless text
 
@@ -42,6 +46,20 @@ module RubyLLM
       def parameters
         @parameters ||= {}
       end
+
+      def params(schema = nil, &block)
+        @params_schema_definition = SchemaDefinition.new(schema:, block:)
+        self
+      end
+
+      def with_params(**params)
+        @provider_params = params
+        self
+      end
+
+      def provider_params
+        @provider_params ||= {}
+      end
     end
 
     def name
@@ -63,6 +81,23 @@ module RubyLLM
       self.class.parameters
     end
 
+    def provider_params
+      self.class.provider_params
+    end
+
+    def params_schema
+      return @params_schema if defined?(@params_schema)
+
+      @params_schema = begin
+        definition = self.class.params_schema_definition
+        if definition&.present?
+          definition.json_schema
+        elsif parameters.any?
+          SchemaDefinition.from_parameters(parameters)&.json_schema
+        end
+      end
+    end
+
     def call(args)
       RubyLLM.logger.debug "Tool #{name} called with: #{args.inspect}"
       result = execute(**args.transform_keys(&:to_sym))
@@ -79,5 +114,96 @@ module RubyLLM
     def halt(message)
       Halt.new(message)
     end
+
+    # Wraps schema handling for tool parameters, supporting JSON Schema hashes,
+    # RubyLLM::Schema instances/classes, and DSL blocks.
+    class SchemaDefinition
+      def self.from_parameters(parameters)
+        return nil if parameters.nil? || parameters.empty?
+
+        properties = parameters.to_h do |name, param|
+          schema = {
+            type: map_type(param.type),
+            description: param.description
+          }.compact
+
+          schema[:items] = default_items_schema if schema[:type] == 'array'
+
+          [name.to_s, schema]
+        end
+
+        required = parameters.select { |_, param| param.required }.keys.map(&:to_s)
+
+        json_schema = {
+          type: 'object',
+          properties: properties,
+          required: required,
+          additionalProperties: false,
+          strict: true
+        }
+
+        new(schema: json_schema)
+      end
+
+      def self.map_type(type)
+        case type.to_s
+        when 'integer', 'int' then 'integer'
+        when 'number', 'float', 'double' then 'number'
+        when 'boolean' then 'boolean'
+        when 'array' then 'array'
+        when 'object' then 'object'
+        else
+          'string'
+        end
+      end
+
+      def self.default_items_schema
+        { type: 'string' }
+      end
+
+      def initialize(schema: nil, block: nil)
+        @schema = schema
+        @block = block
+      end
+
+      def present?
+        @schema || @block
+      end
+
+      def json_schema
+        @json_schema ||= RubyLLM::Utils.deep_stringify_keys(resolve_schema)
+      end
+
+      private
+
+      def resolve_schema
+        return resolve_direct_schema(@schema) if @schema
+        return build_from_block(&@block) if @block
+
+        nil
+      end
+
+      def resolve_direct_schema(schema)
+        return extract_schema(schema.to_json_schema) if schema.respond_to?(:to_json_schema)
+        return RubyLLM::Utils.deep_dup(schema) if schema.is_a?(Hash)
+        if schema.is_a?(Class) && schema.instance_methods.include?(:to_json_schema)
+          return extract_schema(schema.new.to_json_schema)
+        end
+
+        nil
+      end
+
+      def build_from_block(&)
+        schema_class = RubyLLM::Schema.create(&)
+        extract_schema(schema_class.new.to_json_schema)
+      end
+
+      def extract_schema(schema_hash)
+        return nil unless schema_hash.is_a?(Hash)
+
+        schema = schema_hash[:schema] || schema_hash['schema'] || schema_hash
+        RubyLLM::Utils.deep_dup(schema)
+      end
+    end
   end
 end

data/lib/ruby_llm/transcription.rb

@@ -0,0 +1,35 @@
+# frozen_string_literal: true
+
+module RubyLLM
+  # Represents a transcription of audio content.
+  class Transcription
+    attr_reader :text, :model, :language, :duration, :segments, :input_tokens, :output_tokens
+
+    def initialize(text:, model:, **attributes)
+      @text = text
+      @model = model
+      @language = attributes[:language]
+      @duration = attributes[:duration]
+      @segments = attributes[:segments]
+      @input_tokens = attributes[:input_tokens]
+      @output_tokens = attributes[:output_tokens]
+    end
+
+    def self.transcribe(audio_file, **kwargs)
+      model = kwargs.delete(:model)
+      language = kwargs.delete(:language)
+      provider = kwargs.delete(:provider)
+      assume_model_exists = kwargs.delete(:assume_model_exists) { false }
+      context = kwargs.delete(:context)
+      options = kwargs
+
+      config = context&.config || RubyLLM.config
+      model ||= config.default_transcription_model
+      model, provider_instance = Models.resolve(model, provider: provider, assume_exists: assume_model_exists,
+                                                       config: config)
+      model_id = model.id
+
+      provider_instance.transcribe(audio_file, model: model_id, language:, **options)
+    end
+  end
+end

data/lib/ruby_llm/utils.rb

@@ -41,5 +41,51 @@ module RubyLLM
         end
       end
     end
+
+    def deep_dup(value)
+      case value
+      when Hash
+        value.each_with_object({}) do |(key, val), duped|
+          duped[deep_dup(key)] = deep_dup(val)
+        end
+      when Array
+        value.map { |item| deep_dup(item) }
+      else
+        begin
+          value.dup
+        rescue TypeError
+          value
+        end
+      end
+    end
+
+    def deep_stringify_keys(value)
+      case value
+      when Hash
+        value.each_with_object({}) do |(key, val), result|
+          result[key.to_s] = deep_stringify_keys(val)
+        end
+      when Array
+        value.map { |item| deep_stringify_keys(item) }
+      when Symbol
+        value.to_s
+      else
+        value
+      end
+    end
+
+    def deep_symbolize_keys(value)
+      case value
+      when Hash
+        value.each_with_object({}) do |(key, val), result|
+          symbolized_key = key.respond_to?(:to_sym) ? key.to_sym : key
+          result[symbolized_key] = deep_symbolize_keys(val)
+        end
+      when Array
+        value.map { |item| deep_symbolize_keys(item) }
+      else
+        value
+      end
+    end
   end
 end

data/lib/ruby_llm/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module RubyLLM
-  VERSION = '1.1.1'
+  VERSION = '1.3.0'
 end

data/lib/ruby_llm_community.rb

@@ -3,11 +3,12 @@
 require 'base64'
 require 'event_stream_parser'
 require 'faraday'
-require 'faraday/retry'
 require 'faraday/multipart'
+require 'faraday/retry'
 
 require 'json'
 require 'logger'
+require 'marcel'
 require 'securerandom'
 require 'zeitwerk'
 
@@ -58,10 +59,18 @@ module RubyLLM
       Embedding.embed(...)
     end
 
+    def moderate(...)
+      Moderation.moderate(...)
+    end
+
     def paint(...)
       Image.paint(...)
     end
 
+    def transcribe(...)
+      Transcription.transcribe(...)
+    end
+
     def models
       Models.instance
     end
@@ -101,6 +110,34 @@ RubyLLM::Provider.register :perplexity, RubyLLM::Providers::Perplexity
 RubyLLM::Provider.register :vertexai, RubyLLM::Providers::VertexAI
 RubyLLM::Provider.register :xai, RubyLLM::Providers::XAI
 
+# Optional Red Candle provider - only available if gem is installed
+begin
+  require 'candle'
+  require 'ruby_llm/providers/red_candle'
+  RubyLLM::Provider.register :red_candle, RubyLLM::Providers::RedCandle
+
+  # Register Red Candle models with the global registry
+  RubyLLM::Providers::RedCandle.models.each do |model|
+    RubyLLM.models.instance_variable_get(:@models) << model
+  end
+rescue LoadError
+  # Red Candle is optional - provider won't be available if gem isn't installed
+end
+
+# Optional Red Candle provider - only available if gem is installed
+begin
+  require 'candle'
+  require 'ruby_llm/providers/red_candle'
+  RubyLLM::Provider.register :red_candle, RubyLLM::Providers::RedCandle
+
+  # Register Red Candle models with the global registry
+  RubyLLM::Providers::RedCandle.models.each do |model|
+    RubyLLM.models.instance_variable_get(:@models) << model
+  end
+rescue LoadError
+  # Red Candle is optional - provider won't be available if gem isn't installed
+end
+
 if defined?(Rails::Railtie)
   require 'ruby_llm/railtie'
   require 'ruby_llm/active_record/acts_as'

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: ruby_llm_community
 version: !ruby/object:Gem::Version
-  version: 1.1.1
+  version: 1.3.0
 platform: ruby
 authors:
 - Paul Shippy
@@ -107,6 +107,20 @@ dependencies:
     - - "~>"
       - !ruby/object:Gem::Version
         version: '1.0'
+- !ruby/object:Gem::Dependency
+  name: ruby_llm-schema
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 0.2.1
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 0.2.1
 - !ruby/object:Gem::Dependency
   name: zeitwerk
   requirement: !ruby/object:Gem::Requirement
@@ -146,14 +160,17 @@ files:
 - lib/generators/ruby_llm/chat_ui/templates/views/chats/index.html.erb.tt
 - lib/generators/ruby_llm/chat_ui/templates/views/chats/new.html.erb.tt
 - lib/generators/ruby_llm/chat_ui/templates/views/chats/show.html.erb.tt
+- lib/generators/ruby_llm/chat_ui/templates/views/messages/_content.html.erb.tt
 - lib/generators/ruby_llm/chat_ui/templates/views/messages/_form.html.erb.tt
 - lib/generators/ruby_llm/chat_ui/templates/views/messages/_message.html.erb.tt
+- lib/generators/ruby_llm/chat_ui/templates/views/messages/_tool_calls.html.erb.tt
 - lib/generators/ruby_llm/chat_ui/templates/views/messages/create.turbo_stream.erb.tt
 - lib/generators/ruby_llm/chat_ui/templates/views/models/_model.html.erb.tt
 - lib/generators/ruby_llm/chat_ui/templates/views/models/index.html.erb.tt
 - lib/generators/ruby_llm/chat_ui/templates/views/models/show.html.erb.tt
 - lib/generators/ruby_llm/generator_helpers.rb
 - lib/generators/ruby_llm/install/install_generator.rb
+- lib/generators/ruby_llm/install/templates/add_references_to_chats_tool_calls_and_messages_migration.rb.tt
 - lib/generators/ruby_llm/install/templates/chat_model.rb.tt
 - lib/generators/ruby_llm/install/templates/create_chats_migration.rb.tt
 - lib/generators/ruby_llm/install/templates/create_messages_migration.rb.tt
@@ -165,6 +182,8 @@ files:
 - lib/generators/ruby_llm/install/templates/tool_call_model.rb.tt
 - lib/generators/ruby_llm/upgrade_to_v1_7/templates/migration.rb.tt
 - lib/generators/ruby_llm/upgrade_to_v1_7/upgrade_to_v1_7_generator.rb
+- lib/generators/ruby_llm/upgrade_to_v1_9/templates/add_v1_9_message_columns.rb.tt
+- lib/generators/ruby_llm/upgrade_to_v1_9/upgrade_to_v1_9_generator.rb
 - lib/ruby_llm/active_record/acts_as.rb
 - lib/ruby_llm/active_record/acts_as_legacy.rb
 - lib/ruby_llm/active_record/chat_methods.rb
@@ -195,10 +214,12 @@ files:
 - lib/ruby_llm/models.json
 - lib/ruby_llm/models.rb
 - lib/ruby_llm/models_schema.json
+- lib/ruby_llm/moderation.rb
 - lib/ruby_llm/provider.rb
 - lib/ruby_llm/providers/anthropic.rb
 - lib/ruby_llm/providers/anthropic/capabilities.rb
 - lib/ruby_llm/providers/anthropic/chat.rb
+- lib/ruby_llm/providers/anthropic/content.rb
 - lib/ruby_llm/providers/anthropic/embeddings.rb
 - lib/ruby_llm/providers/anthropic/media.rb
 - lib/ruby_llm/providers/anthropic/models.rb
@@ -228,6 +249,7 @@ files:
 - lib/ruby_llm/providers/gemini/models.rb
 - lib/ruby_llm/providers/gemini/streaming.rb
 - lib/ruby_llm/providers/gemini/tools.rb
+- lib/ruby_llm/providers/gemini/transcription.rb
 - lib/ruby_llm/providers/gpustack.rb
 - lib/ruby_llm/providers/gpustack/chat.rb
 - lib/ruby_llm/providers/gpustack/media.rb
@@ -248,10 +270,12 @@ files:
 - lib/ruby_llm/providers/openai/images.rb
 - lib/ruby_llm/providers/openai/media.rb
 - lib/ruby_llm/providers/openai/models.rb
+- lib/ruby_llm/providers/openai/moderation.rb
 - lib/ruby_llm/providers/openai/response.rb
 - lib/ruby_llm/providers/openai/response_media.rb
 - lib/ruby_llm/providers/openai/streaming.rb
 - lib/ruby_llm/providers/openai/tools.rb
+- lib/ruby_llm/providers/openai/transcription.rb
 - lib/ruby_llm/providers/openai_base.rb
 - lib/ruby_llm/providers/openrouter.rb
 - lib/ruby_llm/providers/openrouter/models.rb
@@ -259,11 +283,17 @@ files:
 - lib/ruby_llm/providers/perplexity/capabilities.rb
 - lib/ruby_llm/providers/perplexity/chat.rb
 - lib/ruby_llm/providers/perplexity/models.rb
+- lib/ruby_llm/providers/red_candle.rb
+- lib/ruby_llm/providers/red_candle/capabilities.rb
+- lib/ruby_llm/providers/red_candle/chat.rb
+- lib/ruby_llm/providers/red_candle/models.rb
+- lib/ruby_llm/providers/red_candle/streaming.rb
 - lib/ruby_llm/providers/vertexai.rb
 - lib/ruby_llm/providers/vertexai/chat.rb
 - lib/ruby_llm/providers/vertexai/embeddings.rb
 - lib/ruby_llm/providers/vertexai/models.rb
 - lib/ruby_llm/providers/vertexai/streaming.rb
+- lib/ruby_llm/providers/vertexai/transcription.rb
 - lib/ruby_llm/providers/xai.rb
 - lib/ruby_llm/providers/xai/capabilities.rb
 - lib/ruby_llm/providers/xai/chat.rb
@@ -273,6 +303,7 @@ files:
 - lib/ruby_llm/streaming.rb
 - lib/ruby_llm/tool.rb
 - lib/ruby_llm/tool_call.rb
+- lib/ruby_llm/transcription.rb
 - lib/ruby_llm/utils.rb
 - lib/ruby_llm/version.rb
 - lib/ruby_llm_community.rb
@@ -290,10 +321,11 @@ metadata:
   changelog_uri: https://github.com/tpaulshippy/ruby_llm_community/commits/main
   documentation_uri: https://rubyllm.com
   bug_tracker_uri: https://github.com/tpaulshippy/ruby_llm_community/issues
+  funding_uri: https://github.com/sponsors/crmne
   rubygems_mfa_required: 'true'
 post_install_message: |
-  Upgrading from RubyLLM <= 1.6.x? Check the upgrade guide for new features and migration instructions
-  --> https://rubyllm.com/upgrading-to-1-7/
+  Upgrading from RubyLLM <= 1.8.x? Check the upgrade guide for new features and migration instructions
+  --> https://rubyllm.com/upgrading/
 rdoc_options: []
 require_paths:
 - lib