ruby_llm 0.1.0.pre40 → 0.1.0.pre42

data/lib/ruby_llm/providers/gemini/tools.rb

@@ -0,0 +1,88 @@
+# frozen_string_literal: true
+
+module RubyLLM
+  module Providers
+    module Gemini
+      # Tools methods for the Gemini API implementation
+      module Tools
+        # Format tools for Gemini API
+        def format_tools(tools)
+          return [] if tools.empty?
+
+          [{
+            functionDeclarations: tools.values.map { |tool| function_declaration_for(tool) }
+          }]
+        end
+
+        # Extract tool calls from response data
+        def extract_tool_calls(data) # rubocop:disable Metrics/CyclomaticComplexity,Metrics/MethodLength
+          return nil unless data
+
+          # Get the first candidate
+          candidate = data.is_a?(Hash) ? data.dig('candidates', 0) : nil
+          return nil unless candidate
+
+          # Get the parts array from content
+          parts = candidate.dig('content', 'parts')
+          return nil unless parts.is_a?(Array)
+
+          # Find the function call part
+          function_call_part = parts.find { |p| p['functionCall'] }
+          return nil unless function_call_part
+
+          # Get the function call data
+          function_data = function_call_part['functionCall']
+          return nil unless function_data
+
+          # Create a unique ID for the tool call
+          id = SecureRandom.uuid
+
+          # Return the tool call in the expected format
+          {
+            id => ToolCall.new(
+              id: id,
+              name: function_data['name'],
+              arguments: function_data['args']
+            )
+          }
+        end
+
+        private
+
+        # Format a single tool for Gemini API
+        def function_declaration_for(tool)
+          {
+            name: tool.name,
+            description: tool.description,
+            parameters: {
+              type: 'OBJECT',
+              properties: format_parameters(tool.parameters),
+              required: tool.parameters.select { |_, p| p.required }.keys.map(&:to_s)
+            }
+          }
+        end
+
+        # Format tool parameters for Gemini API
+        def format_parameters(parameters)
+          parameters.transform_values do |param|
+            {
+              type: param_type_for_gemini(param.type),
+              description: param.description
+            }.compact
+          end
+        end
+
+        # Convert RubyLLM param types to Gemini API types
+        def param_type_for_gemini(type)
+          case type.to_s.downcase
+          when 'integer', 'number', 'float' then 'NUMBER'
+          when 'boolean' then 'BOOLEAN'
+          when 'array' then 'ARRAY'
+          when 'object' then 'OBJECT'
+          else 'STRING'
+          end
+        end
+      end
+    end
+  end
+end

data/lib/ruby_llm/providers/gemini.rb

@@ -2,20 +2,26 @@
 
 module RubyLLM
   module Providers
-    # Gemini API integration.
+    # Native Gemini API implementation
     module Gemini
-      extend OpenAI
+      extend Provider
+      extend Gemini::Chat
+      extend Gemini::Embeddings
+      extend Gemini::Images
       extend Gemini::Models
+      extend Gemini::Streaming
+      extend Gemini::Tools
+      extend Gemini::Media
 
       module_function
 
       def api_base
-        'https://generativelanguage.googleapis.com/v1beta/openai'
+        'https://generativelanguage.googleapis.com/v1beta'
       end
 
       def headers
         {
-          'Authorization' => "Bearer #{RubyLLM.config.gemini_api_key}"
+          'x-goog-api-key' => RubyLLM.config.gemini_api_key
         }
       end
 

data/lib/ruby_llm/providers/openai/images.rb

@@ -20,8 +20,6 @@ module RubyLLM
           }
         end
 
-        private
-
         def parse_image_response(response)
           data = response.body
           image_data = data['data'].first

data/lib/ruby_llm/stream_accumulator.rb

@@ -47,7 +47,7 @@ module RubyLLM
         ToolCall.new(
           id: tc.id,
           name: tc.name,
-          arguments: JSON.parse(tc.arguments)
+          arguments: tc.arguments.is_a?(String) ? JSON.parse(tc.arguments) : tc.arguments
         )
       end
     end

data/lib/ruby_llm/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module RubyLLM
-  VERSION = '0.1.0.pre40'
+  VERSION = '0.1.0.pre42'
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: ruby_llm
 version: !ruby/object:Gem::Version
-  version: 0.1.0.pre40
+  version: 0.1.0.pre42
 platform: ruby
 authors:
 - Carmine Paolino
@@ -129,6 +129,7 @@ files:
 - lib/ruby_llm/providers/anthropic/capabilities.rb
 - lib/ruby_llm/providers/anthropic/chat.rb
 - lib/ruby_llm/providers/anthropic/embeddings.rb
+- lib/ruby_llm/providers/anthropic/media.rb
 - lib/ruby_llm/providers/anthropic/models.rb
 - lib/ruby_llm/providers/anthropic/streaming.rb
 - lib/ruby_llm/providers/anthropic/tools.rb
@@ -136,7 +137,13 @@ files:
 - lib/ruby_llm/providers/deepseek/capabilities.rb
 - lib/ruby_llm/providers/gemini.rb
 - lib/ruby_llm/providers/gemini/capabilities.rb
+- lib/ruby_llm/providers/gemini/chat.rb
+- lib/ruby_llm/providers/gemini/embeddings.rb
+- lib/ruby_llm/providers/gemini/images.rb
+- lib/ruby_llm/providers/gemini/media.rb
 - 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/openai.rb
 - lib/ruby_llm/providers/openai/capabilities.rb
 - lib/ruby_llm/providers/openai/chat.rb