ruby_llm_community 0.0.6 → 1.1.0

data/lib/ruby_llm/streaming.rb

@@ -1,10 +1,7 @@
 # frozen_string_literal: true
 
 module RubyLLM
-  # Handles streaming responses from AI providers. Provides a unified way to process
-  # chunked responses, accumulate content, and handle provider-specific streaming formats.
-  # Each provider implements provider-specific parsing while sharing common stream handling
-  # patterns.
+  # Handles streaming responses from AI providers.
   module Streaming
     module_function
 
@@ -12,17 +9,14 @@ module RubyLLM
       accumulator = StreamAccumulator.new
 
       response = connection.post stream_url, payload do |req|
-        # Merge additional headers, with existing headers taking precedence
         req.headers = additional_headers.merge(req.headers) unless additional_headers.empty?
-        if req.options.respond_to?(:on_data)
-          # Handle Faraday 2.x streaming with on_data method
-          req.options.on_data = handle_stream do |chunk|
+        if faraday_1?
+          req.options[:on_data] = handle_stream do |chunk|
             accumulator.add chunk
             block.call chunk
           end
         else
-          # Handle Faraday 1.x streaming with :on_data key
-          req.options[:on_data] = handle_stream do |chunk|
+          req.options.on_data = handle_stream do |chunk|
             accumulator.add chunk
             block.call chunk
           end
@@ -30,7 +24,7 @@ module RubyLLM
       end
 
       message = accumulator.to_message(response)
-      RubyLLM.logger.debug "Stream completed: #{message.inspect}"
+      RubyLLM.logger.debug "Stream completed: #{message.content}"
       message
     end
 
@@ -42,6 +36,10 @@ module RubyLLM
 
     private
 
+    def faraday_1?
+      Faraday::VERSION.start_with?('1')
+    end
+
     def to_json_stream(&)
       buffer = +''
       parser = EventStreamParser::Parser.new
@@ -50,11 +48,9 @@ module RubyLLM
     end
 
     def create_stream_processor(parser, buffer, &)
-      if Faraday::VERSION.start_with?('1')
-        # Faraday 1.x: on_data receives (chunk, size)
+      if faraday_1?
         legacy_stream_processor(parser, &)
       else
-        # Faraday 2.x: on_data receives (chunk, bytes, env)
         stream_processor(parser, buffer, &)
       end
     end
@@ -94,12 +90,10 @@ module RubyLLM
       status, _message = parse_streaming_error(error_data)
       parsed_data = JSON.parse(error_data)
 
-      # Create a response-like object that works for both Faraday v1 and v2
-      error_response = if env
-                         env.merge(body: parsed_data, status: status)
-                       else
-                         # For Faraday v1, create a simple object that responds to .status and .body
+      error_response = if faraday_1?
                          Struct.new(:body, :status).new(parsed_data, status)
+                       else
+                         env.merge(body: parsed_data, status: status)
                        end
 
       ErrorMiddleware.parse_error(provider: self, response: error_response)
@@ -137,12 +131,10 @@ module RubyLLM
       status, _message = parse_streaming_error(data)
       parsed_data = JSON.parse(data)
 
-      # Create a response-like object that works for both Faraday v1 and v2
-      error_response = if env
-                         env.merge(body: parsed_data, status: status)
-                       else
-                         # For Faraday v1, create a simple object that responds to .status and .body
+      error_response = if faraday_1?
                          Struct.new(:body, :status).new(parsed_data, status)
+                       else
+                         env.merge(body: parsed_data, status: status)
                        end
 
       ErrorMiddleware.parse_error(provider: self, response: error_response)
@@ -150,7 +142,6 @@ module RubyLLM
       RubyLLM.logger.debug "Failed to parse error event: #{e.message}"
     end
 
-    # Default implementation - providers should override this method
     def parse_streaming_error(data)
       error_data = JSON.parse(data)
       [500, error_data['message'] || 'Unknown streaming error']

data/lib/ruby_llm/tool.rb

@@ -1,8 +1,7 @@
 # frozen_string_literal: true
 
 module RubyLLM
-  # Parameter definition for Tool methods. Specifies type constraints,
-  # descriptions, and whether parameters are required.
+  # Parameter definition for Tool methods.
   class Parameter
     attr_reader :name, :type, :description, :required
 
@@ -14,23 +13,7 @@ module RubyLLM
     end
   end
 
-  # Base class for creating tools that AI models can use. Provides a simple
-  # interface for defining parameters and implementing tool behavior.
-  #
-  # Example:
-  #    require 'tzinfo'
-  #
-  #    class TimeInfo < RubyLLM::Tool
-  #      description 'Gets the current time in various timezones'
-  #      param :timezone, desc: "Timezone name (e.g., 'UTC', 'America/New_York')"
-  #
-  #      def execute(timezone:)
-  #        time = TZInfo::Timezone.get(timezone).now.strftime('%Y-%m-%d %H:%M:%S')
-  #        "Current time in #{timezone}: #{time}"
-  #       rescue StandardError => e
-  #          { error: e.message }
-  #       end
-  #    end
+  # Base class for creating tools that AI models can use
   class Tool
     # Stops conversation continuation after tool execution
     class Halt

data/lib/ruby_llm/tool_call.rb

@@ -2,15 +2,6 @@
 
 module RubyLLM
   # Represents a function call from an AI model to a Tool.
-  # Encapsulates the function name, arguments, and execution results
-  # in a clean Ruby interface.
-  #
-  # Example:
-  #   tool_call = ToolCall.new(
-  #     id: "call_123",
-  #     name: "calculator",
-  #     arguments: { expression: "2 + 2" }
-  #   )
   class ToolCall
     attr_reader :id, :name, :arguments
 

data/lib/ruby_llm/utils.rb

@@ -5,10 +5,6 @@ module RubyLLM
   module Utils
     module_function
 
-    def format_text_file_for_llm(text_file)
-      "<file name='#{text_file.filename}' mime_type='#{text_file.mime_type}'>#{text_file.content}</file>"
-    end
-
     def hash_get(hash, key)
       hash[key.to_sym] || hash[key.to_s]
     end
@@ -36,12 +32,12 @@ module RubyLLM
       value.is_a?(Date) ? value : Date.parse(value.to_s)
     end
 
-    def deep_merge(params, payload)
-      params.merge(payload) do |_key, params_value, payload_value|
-        if params_value.is_a?(Hash) && payload_value.is_a?(Hash)
-          deep_merge(params_value, payload_value)
+    def deep_merge(original, overrides)
+      original.merge(overrides) do |_key, original_value, overrides_value|
+        if original_value.is_a?(Hash) && overrides_value.is_a?(Hash)
+          deep_merge(original_value, overrides_value)
         else
-          payload_value
+          overrides_value
         end
       end
     end

data/lib/ruby_llm/version.rb

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

data/lib/ruby_llm_community.rb

@@ -4,6 +4,8 @@ require 'base64'
 require 'event_stream_parser'
 require 'faraday'
 require 'faraday/retry'
+require 'faraday/multipart'
+
 require 'json'
 require 'logger'
 require 'securerandom'
@@ -23,12 +25,13 @@ loader.inflector.inflect(
   'gpustack' => 'GPUStack',
   'mistral' => 'Mistral',
   'pdf' => 'PDF',
-  'version' => 'VERSION'
+  'version' => 'VERSION',
+  'vertexai' => 'VertexAI',
+  'xai' => 'XAI',
+  'UI' => 'UI'
 )
 loader.ignore("#{__dir__}/shims")
 loader.ignore("#{__dir__}/tasks")
-loader.ignore("#{__dir__}/ruby_llm/railtie")
-loader.ignore("#{__dir__}/ruby_llm/active_record")
 loader.ignore("#{__dir__}/generators")
 loader.setup
 
@@ -37,8 +40,6 @@ module RubyLlmCommunity
 end
 
 # A delightful Ruby interface to modern AI language models.
-# Provides a unified way to interact with models from OpenAI, Anthropic and others
-# with a focus on developer happiness and convention over configuration.
 module RubyLLM
   class Error < StandardError; end
 
@@ -97,6 +98,8 @@ RubyLLM::Provider.register :ollama, RubyLLM::Providers::Ollama
 RubyLLM::Provider.register :openai, RubyLLM::Providers::OpenAI
 RubyLLM::Provider.register :openrouter, RubyLLM::Providers::OpenRouter
 RubyLLM::Provider.register :perplexity, RubyLLM::Providers::Perplexity
+RubyLLM::Provider.register :vertexai, RubyLLM::Providers::VertexAI
+RubyLLM::Provider.register :xai, RubyLLM::Providers::XAI
 
 if defined?(Rails::Railtie)
   require 'ruby_llm/railtie'