ruby-gemini-api 0.1.1 → 0.1.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: f5a02e1edca475bc3a7ab45e9e595186a99e2546fb46c1f1f1232a8215d45469
-  data.tar.gz: a688b8e1a0bb724ab24d8cb1113ad689a784bc41915b180e17e9b4552dad4628
+  metadata.gz: 36071e1dcc5fb406f9f49318cc2cffdb0e4159a0811c74da0aebd48293d9ef02
+  data.tar.gz: cdccba0ec67eb7fa2cd69ffcb8ddcd79c7056d6a636dcb3ccd0d05af14f5acbd
 SHA512:
-  metadata.gz: a5d4af0804fb7ded6bbf069b492b31b56979ab6421dafcf671704a8aae62405946ca4efa6cdb1ef13ab69ec16a19fb0b1d19df5045d6840ae843c7500250094d
-  data.tar.gz: e9c83f07e0cfb54827d7c0a4180ab8731a7b8580669073d7d9acfe0a8b45b015d18c42acf6fbd1ffd2da3b9acb89ea7723da7e1abba756761666944a985f63dc
+  metadata.gz: e50e268d2c015f45bf6db61d07e5b124aa07b967a278999aa00c29f760e1c41d1465af8582055764fc64454156f15130f498318cdc264cd78d342dc28cec4030
+  data.tar.gz: 21a68ce8106b7b2d0e5e9468d0464f91a18b62cbfad7c6c6924280d58bf525f9b4ac7e87ffe0e06076bd03b9f2b77fe0b6e35b9ba93bdfaedfbcc65afc6abc40

data/CHANGELOG.md

@@ -5,3 +5,6 @@
 
 ## [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

@@ -18,6 +18,55 @@ 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:

data/lib/gemini/client.rb

@@ -116,34 +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, temperature: 0.5, **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
-      
-        params[:generation_config] ||= {}
-        
+      params[:generation_config] ||= {}
+      params[:generation_config]["temperature"] = temperature
       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[:tools] = tools if tools
       params.merge!(parameters)
-      
+
       if block_given?
         chat(parameters: params, &stream_callback)
       else

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.1"
+  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"

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: ruby-gemini-api
 version: !ruby/object:Gem::Version
-  version: 0.1.1
+  version: 0.1.2
 platform: ruby
 authors:
 - rira100000000
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2025-05-04 00:00:00.000000000 Z
+date: 2025-07-10 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: faraday
@@ -161,6 +161,7 @@ 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-api