ruby-gemini-api 0.1.0 → 0.1.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 1d3fab54be3c9a6830811a1ee33e72aa775d8f313a9aaaa702219b2fbd9393db
-  data.tar.gz: 64a8c2089d56cad74c8847ebb58cf34d810bbb1bc77e7b7500151c268a7f88d5
+  metadata.gz: 36071e1dcc5fb406f9f49318cc2cffdb0e4159a0811c74da0aebd48293d9ef02
+  data.tar.gz: cdccba0ec67eb7fa2cd69ffcb8ddcd79c7056d6a636dcb3ccd0d05af14f5acbd
 SHA512:
-  metadata.gz: 3e6155154922f2b29c55271c35483b1e4a29aa406117d6b880235b501a7af2b10234f997b5149ed6f1a2c708889a652140debc91132854f9bca7139c3130958a
-  data.tar.gz: d2257648e0a703931e16d057b2a62c340ce595e385c160a00b3a43abf9604ded4f9619401c96b3b8765159d4f2b8efcde9bb34c3623d27fa258004252320c0e9
+  metadata.gz: e50e268d2c015f45bf6db61d07e5b124aa07b967a278999aa00c29f760e1c41d1465af8582055764fc64454156f15130f498318cdc264cd78d342dc28cec4030
+  data.tar.gz: 21a68ce8106b7b2d0e5e9468d0464f91a18b62cbfad7c6c6924280d58bf525f9b4ac7e87ffe0e06076bd03b9f2b77fe0b6e35b9ba93bdfaedfbcc65afc6abc40

data/CHANGELOG.md

@@ -1,5 +1,10 @@
 ## [Unreleased]
 
 ## [0.1.0] - 2025-04-05
-
 - Initial release
+
+## [0.1.1] - 2025-05-04
+- Changed generate_contents to accept temperature parameter
+
+## [0.1.2] - 2025-07-10
+- Add function calling

data/README.md

@@ -1,5 +1,5 @@
-[README ‐ 日本語](https://github.com/rira100000000/ruby-gemini/wiki/README-%E2%80%90-%E6%97%A5%E6%9C%AC%E8%AA%9E)
-# Ruby-Gemini
+[README ‐ 日本語](https://github.com/rira100000000/ruby-gemini-api/wiki/README-%E2%80%90-%E6%97%A5%E6%9C%AC%E8%AA%9E)
+# Ruby-Gemini-API
 
 A Ruby client library for Google's Gemini API. This gem provides a simple, intuitive interface for interacting with Gemini's generative AI capabilities, following patterns similar to other AI client libraries.
 
@@ -18,12 +18,61 @@ This project is inspired by and pays homage to [ruby-openai](https://github.com/
 - Document processing (PDFs and other formats)
 - Context caching for efficient processing
 
+### How to use Function Calling
+
+```ruby
+require 'gemini'
+
+# Initialize Gemini client
+client = Gemini::Client.new(ENV['GEMINI_API_KEY'])
+
+# Define function declarations for Function Calling
+# Note: Use camelCase (functionDeclarations) for Gemini API compatibility
+tools = [
+  {
+    functionDeclarations: [
+      {
+        name: "get_current_weather",
+        description: "Get the current weather information",
+        parameters: {
+          type: "object",
+          properties: {
+            location: {
+              type: "string",
+              description: "City name, e.g., Tokyo"
+            }
+          },
+          required: ["location"]
+        }
+      }
+    ]
+  }
+]
+
+# User prompt
+user_prompt = "Tell me the current weather in Tokyo."
+
+# Send request with Function Calling tools
+response = client.generate_content(
+  user_prompt,
+  model: "gemini-2.0-flash",
+  tools: tools
+)
+
+# Parse function call from the response
+unless response.function_calls.empty?
+  function_call = response.function_calls.first
+  puts "Function name to call: #{function_call["name"]}"
+  puts "Function arguments: #{function_call["args"]}"
+end
+```
+
 ## Installation
 
 Add this line to your application's Gemfile:
 
 ```ruby
-gem 'ruby-gemini'
+gem 'ruby-gemini-api'
 ```
 
 And then execute:
@@ -35,7 +84,7 @@ $ bundle install
 Or install it yourself as:
 
 ```bash
-$ gem install ruby-gemini
+$ gem install ruby-gemini-api
 ```
 
 ## Quick Start

data/lib/gemini/client.rb

@@ -116,36 +116,30 @@ module Gemini
     
     # Helper methods for convenience
     
-    # Method with usage similar to OpenAI's chat
+        # Method with usage similar to OpenAI's chat
     def generate_content(prompt, model: "gemini-2.0-flash-lite", system_instruction: nil, 
-                        response_mime_type: nil, response_schema: nil, **parameters, &stream_callback)
-      # For image/text combinations, the prompt is passed as an array
-      # example: [{type: "text", text: "What is this?"}, {type: "image_url", image_url: {url: "https://example.com/image.jpg"}}]
+                        response_mime_type: nil, response_schema: nil, temperature: 0.5, tools: nil, **parameters, &stream_callback)
       content = format_content(prompt)
       params = {
         contents: [content],
         model: model
       }
-      
+
       if system_instruction
         params[:system_instruction] = format_content(system_instruction)
       end
-      
-      if response_mime_type || response_schema
-        params[:generation_config] ||= {}
-        
-        if response_mime_type
-          params[:generation_config]["response_mime_type"] = response_mime_type
-        end
-        
-        if response_schema
-          params[:generation_config]["response_schema"] = response_schema
-        end
+      params[:generation_config] ||= {}
+      params[:generation_config]["temperature"] = temperature
+      if response_mime_type
+        params[:generation_config]["response_mime_type"] = response_mime_type
       end
-      
-      # Merge other parameters
+
+      if response_schema
+        params[:generation_config]["response_schema"] = response_schema
+      end
+      params[:tools] = tools if tools
       params.merge!(parameters)
-      
+
       if block_given?
         chat(parameters: params, &stream_callback)
       else
@@ -155,7 +149,7 @@ module Gemini
 
     # Streaming text generation
     def generate_content_stream(prompt, model: "gemini-2.0-flash-lite", system_instruction: nil,
-                              response_mime_type: nil, response_schema: nil, **parameters, &block)
+                              response_mime_type: nil, response_schema: nil, temperature: 0.5, **parameters, &block)
       raise ArgumentError, "Block is required for streaming" unless block_given?
       
       content = format_content(prompt)
@@ -168,18 +162,16 @@ module Gemini
         params[:system_instruction] = format_content(system_instruction)
       end
       
-      if response_mime_type || response_schema
-        params[:generation_config] ||= {}
-        
-        if response_mime_type
-          params[:generation_config][:response_mime_type] = response_mime_type
-        end
-        
-        if response_schema
-          params[:generation_config][:response_schema] = response_schema
-        end
+      params[:generation_config] ||= {}
+      
+      if response_mime_type
+        params[:generation_config][:response_mime_type] = response_mime_type
       end
       
+      if response_schema
+        params[:generation_config][:response_schema] = response_schema
+      end
+      params[:generation_config]["temperature"] = temperature
       # Merge other parameters
       params.merge!(parameters)
       

data/lib/gemini/http_headers.rb

@@ -13,7 +13,7 @@ module Gemini
     def default_headers
       {
         "Content-Type" => "application/json",
-        "User-Agent" => "ruby-gemini/#{Gemini::VERSION}"
+        "User-Agent" => "ruby-gemini-api/#{Gemini::VERSION}"
       }
     end
 

data/lib/gemini/response.rb

@@ -138,11 +138,8 @@ module Gemini
     
     # Get function call information
     def function_calls
-      return [] unless valid?
-      
-      first_candidate&.dig("content", "parts")
-        &.select { |part| part.key?("functionCall") }
-        &.map { |part| part["functionCall"] } || []
+      parts = first_candidate.dig("content", "parts") || []
+      parts.map { |part| part["functionCall"] }.compact
     end
     
     # Get response role (usually "model")

data/lib/gemini/tool_definition.rb

@@ -0,0 +1,68 @@
+module Gemini
+  class ToolDefinition
+    def initialize(&block)
+      @functions = {}
+      instance_eval(&block) if block_given?
+    end
+
+    def function(name, description:, &block)
+      @functions[name] = {
+        name: name,
+        description: description,
+        parameters: {
+          type: 'object',
+          properties: {},
+          required: []
+        }
+      }
+      @current_function = name
+      instance_eval(&block) if block_given?
+      @current_function = nil
+    end
+    alias add_function function
+
+    def property(name, type:, description:, required: false)
+      raise 'property must be defined within a function block' unless @current_function
+
+      @functions[@current_function][:parameters][:properties][name] = {
+        type: type.to_s,
+        description: description
+      }
+      @functions[@current_function][:parameters][:required] << name if required
+    end
+
+    def +(other)
+      raise ArgumentError, 'can only merge with another ToolDefinition' unless other.is_a?(ToolDefinition)
+
+      new_definition = dup
+      other.instance_variable_get(:@functions).each do |name, definition|
+        new_definition.instance_variable_get(:@functions)[name] = definition
+      end
+      new_definition
+    end
+
+    def delete_function(name)
+      @functions.delete(name)
+    end
+
+    def list_functions
+      @functions.keys
+    end
+
+    def to_h
+      {
+        function_declarations: @functions.values
+      }
+    end
+
+    def to_json(*args)
+      to_h.to_json(*args)
+    end
+
+    def dup
+      new_instance = self.class.new
+      new_instance.instance_variable_set(:@functions, @functions.dup)
+      new_instance
+    end
+  end
+end

data/lib/gemini/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module Gemini
-  VERSION = "0.1.0"
+  VERSION = "0.1.2"
 end

data/lib/gemini.rb

@@ -3,7 +3,8 @@ require "faraday/multipart"
 require "json"
 require 'dotenv/load'
 
-require_relative "gemini/version"
+require_relative 'gemini/version'
+require_relative 'gemini/tool_definition'
 require_relative "gemini/http_headers"
 require_relative "gemini/http"
 require_relative "gemini/client"
@@ -73,7 +74,7 @@ module Gemini
       color = level == :error ? "\033[31m" : "\033[33m"
       logger = Logger.new($stdout)
       logger.formatter = proc do |_severity, _datetime, _progname, msg|
-        "#{color}#{prefix} (spotted in ruby-gemini #{VERSION}): #{msg}\n\033[0m"
+        "#{color}#{prefix} (spotted in ruby-gemini-api #{VERSION}): #{msg}\n\033[0m"
       end
       logger.send(level, message)
     end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: ruby-gemini-api
 version: !ruby/object:Gem::Version
-  version: 0.1.0
+  version: 0.1.2
 platform: ruby
 authors:
 - rira100000000
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2025-04-13 00:00:00.000000000 Z
+date: 2025-07-10 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: faraday
@@ -161,15 +161,16 @@ files:
 - lib/gemini/response.rb
 - lib/gemini/runs.rb
 - lib/gemini/threads.rb
+- lib/gemini/tool_definition.rb
 - lib/gemini/version.rb
 - lib/ruby/gemini.rb
-homepage: https://github.com/rira100000000/ruby-gemini
+homepage: https://github.com/rira100000000/ruby-gemini-api
 licenses:
 - MIT
 metadata:
-  homepage_uri: https://github.com/rira100000000/ruby-gemini
-  source_code_uri: https://github.com/rira100000000/ruby-gemini
-  changelog_uri: https://github.com/rira100000000/ruby-gemini/blob/main/CHANGELOG.md
+  homepage_uri: https://github.com/rira100000000/ruby-gemini-api
+  source_code_uri: https://github.com/rira100000000/ruby-gemini-api
+  changelog_uri: https://github.com/rira100000000/ruby-gemini-api/blob/main/CHANGELOG.md
   rubygems_mfa_required: 'true'
 post_install_message: 
 rdoc_options: []