llm_hub 0.2.1 → 0.3.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 353ca2493088bec5034c53bda9389a7277d9b79f9a0af7aa9cd0dea38c7a3d96
-  data.tar.gz: e67417f5a3210cb1a5fbca143df86ed2bb4564ae62efe4dab751e2dd0a1bb1e4
+  metadata.gz: 48d9d052b63d19d6c83b84e5cae7c60ac65ce53d6fe464a557a665d682f55e55
+  data.tar.gz: 276a0ad6024d42b9e6dc0000fb0ed2ecea7339ce8412b1be48c3ab1326c92279
 SHA512:
-  metadata.gz: 3896b924c84d3e05877333187e60c57918d9f968ad347027faa8cac6e37849315a0b2815966c1e1fc506ca898e26456fb054973965d2a535b1a56ff41852a6a2
-  data.tar.gz: 12c26874593c316bde1f841da6161c4395e00c33f63eff6f7e26e10b978f349150ec1a686bbaedb2527432a8d9e726d29194b64f2f7199ba4604977d68ffd60f
+  metadata.gz: 87949cc584f5f8cf6da70de897a1c31c5a0f3e66a312534a90c803ded364fa51cec51b5c855c339fa29c905765e11ea5f315f3de4254870d4fd5d2a50d29c382
+  data.tar.gz: 5d4bdadeec61209087d063ed6e0cb98a490eb5e1f2423e724d0b154259f06416b1b44259a2d9a12fad169f5927fa1c9c6624b023a25e2cacacdeb64b2d3c4346

data/CHANGELOG.md

@@ -1,5 +1,9 @@
 ## [Unreleased]
 
+## [0.3.0] - 2025-07-08
+
+- Added Google as a provider.
+
 ## [0.2.1] - 2025-07-04
 
 - Added new option parameters.

data/README.md

@@ -1,6 +1,6 @@
 # LlmHub
 
-This is a Ruby interface for multiple LLM providers, such as OpenAI, Anthropic, and DeepSeek.
+This is a Ruby interface for multiple LLM providers, such as OpenAI, Anthropic, DeepSeek, and Google Gemini.
 
 It provides easy access to Completion and Embedding functionalities.
 

data/lib/llm_hub/completion/client.rb

@@ -9,12 +9,13 @@ module LlmHub
       PROVIDER_CLASSES = {
         openai: Providers::OpenAI,
         anthropic: Providers::Anthropic,
-        deepseek: Providers::Deepseek
+        deepseek: Providers::Deepseek,
+        google: Providers::Google
       }.freeze
 
       # Initialize a new completion client
       # @param api_key [String] API key for the provider (required)
-      # @param provider [Symbol, String] Provider name (:openai, :anthropic, :deepseek) (required)
+      # @param provider [Symbol, String] Provider name (:openai, :anthropic, :deepseek, :google) (required)
       # @param open_time_out [Integer] HTTP open timeout in seconds (optional, defaults to Config value)
       # @param read_time_out [Integer] HTTP read timeout in seconds (optional, defaults to Config value)
       # @param retry_count [Integer] Number of retries for failed requests (optional, defaults to Config value)
@@ -38,7 +39,7 @@ module LlmHub
         option_params: {}
       )
         with_retry do
-          url = @provider_client.url
+          url = provider_url(model_name)
           request_body = @provider_client.request_body(system_prompt, content, model_name, option_params)
           headers = @provider_client.headers
 
@@ -51,6 +52,17 @@ module LlmHub
 
       private
 
+      # Get the appropriate URL for the provider
+      # @param model_name [String] Model name to use
+      # @return [String] Provider URL
+      def provider_url(model_name)
+        if @provider_client.is_a?(Providers::Google)
+          @provider_client.url(model_name)
+        else
+          @provider_client.url
+        end
+      end
+
       # Format the response from provider
       # @param response_body [Hash] Raw response from provider
       # @return [Hash{Symbol => String, Integer}] Formatted response

data/lib/llm_hub/completion/providers/google.rb

@@ -0,0 +1,75 @@
+# frozen_string_literal: true
+
+module LlmHub
+  module Completion
+    module Providers
+      # Google Gemini completion provider
+      class Google < Base
+        COMPLETIONS_URI = 'https://generativelanguage.googleapis.com/v1beta/models'
+
+        def url(model_name)
+          # Gemini API requires model name in URL
+          "#{COMPLETIONS_URI}/#{model_name}:generateContent"
+        end
+
+        def headers
+          {
+            'Content-Type' => 'application/json',
+            'x-goog-api-key' => @api_key
+          }
+        end
+
+        def request_body(system_prompt, content, _model_name, option_params)
+          {
+            system_instruction: build_system_instruction(system_prompt),
+            contents: build_contents(content),
+            generationConfig: build_generation_config(option_params)
+          }
+        end
+
+        def extract_answer(response_body)
+          response_body&.dig('candidates', 0, 'content', 'parts', 0, 'text')
+        end
+
+        def extract_tokens(response_body)
+          usage_metadata = response_body&.dig('usageMetadata')
+          {
+            total_tokens: usage_metadata&.dig('totalTokenCount'),
+            prompt_tokens: usage_metadata&.dig('promptTokenCount'),
+            completion_tokens: usage_metadata&.dig('candidatesTokenCount')
+          }
+        end
+
+        private
+
+        def build_system_instruction(system_prompt)
+          {
+            parts: [
+              {
+                text: system_prompt
+              }
+            ]
+          }
+        end
+
+        def build_contents(content)
+          [{
+            role: 'user',
+            parts: [{ text: content }]
+          }]
+        end
+
+        def build_generation_config(option_params)
+          base_config = {
+            temperature: 0.2,
+            maxOutputTokens: 1024,
+            topP: 0.8,
+            topK: 40
+          }
+
+          option_params.any? ? base_config.merge(option_params) : base_config
+        end
+      end
+    end
+  end
+end

data/lib/llm_hub/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module LlmHub
-  VERSION = '0.2.1'
+  VERSION = '0.3.0'
 end

data/lib/llm_hub.rb

@@ -19,6 +19,7 @@ require_relative 'llm_hub/completion/providers/base'
 require_relative 'llm_hub/completion/providers/openai'
 require_relative 'llm_hub/completion/providers/anthropic'
 require_relative 'llm_hub/completion/providers/deepseek'
+require_relative 'llm_hub/completion/providers/google'
 require_relative 'llm_hub/completion/client'
 
 # Embedding providers

metadata

@@ -1,13 +1,13 @@
 --- !ruby/object:Gem::Specification
 name: llm_hub
 version: !ruby/object:Gem::Version
-  version: 0.2.1
+  version: 0.3.0
 platform: ruby
 authors:
 - akiraNuma
 bindir: exe
 cert_chain: []
-date: 2025-07-04 00:00:00.000000000 Z
+date: 2025-07-07 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: activesupport
@@ -37,8 +37,9 @@ dependencies:
     - - ">="
       - !ruby/object:Gem::Version
         version: '0'
-description: This is a Ruby interface for multiple LLM providers, such as OpenAI and
-  Anthropic.It provides easy access to Completion and Embedding functionalities.
+description: This is a Ruby interface for multiple LLM providers, such as OpenAI,
+  Anthropic, DeepSeek, and Google Gemini.It provides easy access to Completion and
+  Embedding functionalities.
 email:
 - akiran@akiranumakura.com
 executables: []
@@ -60,6 +61,7 @@ files:
 - lib/llm_hub/completion/providers/anthropic.rb
 - lib/llm_hub/completion/providers/base.rb
 - lib/llm_hub/completion/providers/deepseek.rb
+- lib/llm_hub/completion/providers/google.rb
 - lib/llm_hub/completion/providers/openai.rb
 - lib/llm_hub/config.rb
 - lib/llm_hub/embedding/client.rb
@@ -91,5 +93,6 @@ required_rubygems_version: !ruby/object:Gem::Requirement
 requirements: []
 rubygems_version: 3.6.3
 specification_version: 4
-summary: This is a Ruby interface for multiple LLM providers, such as OpenAI and Anthropic.
+summary: This is a Ruby interface for multiple LLM providers, such as OpenAI, Anthropic,
+  DeepSeek, and Google Gemini.
 test_files: []