llm_hub 0.2.0 → 0.3.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: c791bf83c85b80125da4cf8a71eab51d804f21f0f29d128f27710c6bd0e9c621
-  data.tar.gz: 244a67dd97fd68b97a345971a7e806520c83f6906f9d50362ad2dc8b29086253
+  metadata.gz: 48d9d052b63d19d6c83b84e5cae7c60ac65ce53d6fe464a557a665d682f55e55
+  data.tar.gz: 276a0ad6024d42b9e6dc0000fb0ed2ecea7339ce8412b1be48c3ab1326c92279
 SHA512:
-  metadata.gz: a64411fb25ba585dffc8251414231871ff173d89a16e06c4c0b0d6c8bf00fcb859bcf5f7ff0bec7c653ca5249fa48b1d70e03ee507dbc82851d1f4b8c02a3e5c
-  data.tar.gz: 89e38e3d90d585b56809a5fb7e18f4e7257fa4f8675e7172a622ef56f222175b2d68b1b60f73a8f6e50a391b121e8e7523f014fe0af79c98e4f57881d7c35eef
+  metadata.gz: 87949cc584f5f8cf6da70de897a1c31c5a0f3e66a312534a90c803ded364fa51cec51b5c855c339fa29c905765e11ea5f315f3de4254870d4fd5d2a50d29c382
+  data.tar.gz: 5d4bdadeec61209087d063ed6e0cb98a490eb5e1f2423e724d0b154259f06416b1b44259a2d9a12fad169f5927fa1c9c6624b023a25e2cacacdeb64b2d3c4346

data/CHANGELOG.md

@@ -1,5 +1,17 @@
 ## [Unreleased]
 
+## [0.3.0] - 2025-07-08
+
+- Added Google as a provider.
+
+## [0.2.1] - 2025-07-04
+
+- Added new option parameters.
+
+## [0.2.0] - 2025-06-17
+
+- Added Deepseek as a provider.
+
 ## [0.1.1] - 2025-05-30
 
 - Updated description in README.md

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.
 
@@ -37,7 +37,7 @@ puts response
 
 ## Development
 
-After checking out the repo, run `bin/setup` to install dependencies. Then, run `rake spec` to run the tests. You can also run `bin/console` for an interactive prompt that will allow you to experiment.
+After checking out the repo, run `bin/setup` to install dependencies. Then, run `bundle exec rake` to run the tests and code quality checks (or `rake spec` for tests only). You can also run `bin/console` for an interactive prompt that will allow you to experiment.
 
 To install this gem onto your local machine, run `bundle exec rake install`. To release a new version, update the version number in `version.rb`, and then run `bundle exec rake release`, which will create a git tag for the version, push git commits and the created tag, and push the `.gem` file to [rubygems.org](https://rubygems.org).
 

data/lib/llm_hub/common/client_base.rb

@@ -8,15 +8,20 @@ module LlmHub
     class ClientBase
       include LlmHub::Common::HttpHelper
 
-      attr_reader :api_key, :provider, :retry_count
+      attr_reader :api_key, :provider, :retry_count, :open_time_out, :read_time_out
 
       # Initialize a new client
       # @param api_key [String] API key for the provider (required)
       # @param provider [Symbol, String] Provider name (required)
-      def initialize(api_key:, provider:)
+      # @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)
+      def initialize(api_key:, provider:, open_time_out: nil, read_time_out: nil, retry_count: nil)
         @api_key = api_key
         @provider = provider
-        @retry_count = LlmHub::Config::RETRY_COUNT
+        @open_time_out = open_time_out || LlmHub::Config::DEFAULT_OPEN_TIME_OUT
+        @read_time_out = read_time_out || LlmHub::Config::DEFAULT_READ_TIME_OUT
+        @retry_count = retry_count || LlmHub::Config::DEFAULT_RETRY_COUNT
       end
 
       protected

data/lib/llm_hub/common/http_helper.rb

@@ -29,8 +29,8 @@ module LlmHub
         )
         http_client.use_ssl = uri.scheme == 'https'
         http_client.verify_mode = OpenSSL::SSL::VERIFY_NONE
-        http_client.open_timeout = LlmHub::Config::OPEN_TIME_OUT
-        http_client.read_timeout = LlmHub::Config::READ_TIME_OUT
+        http_client.open_timeout = @open_time_out
+        http_client.read_timeout = @read_time_out
         http_client
       end
     end

data/lib/llm_hub/completion/client.rb

@@ -9,14 +9,18 @@ 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)
       # @see LlmHub::Common::ClientBase#initialize
-      def initialize(api_key:, provider:)
+      def initialize(api_key:, provider:, open_time_out: nil, read_time_out: nil, retry_count: nil)
         super
         @provider_client = create_provider_client
       end
@@ -35,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
 
@@ -48,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/config.rb

@@ -2,8 +2,8 @@
 
 module LlmHub
   module Config
-    OPEN_TIME_OUT = 5
-    READ_TIME_OUT = 20
-    RETRY_COUNT = 1
+    DEFAULT_OPEN_TIME_OUT = 5
+    DEFAULT_READ_TIME_OUT = 20
+    DEFAULT_RETRY_COUNT = 1
   end
 end

data/lib/llm_hub/embedding/client.rb

@@ -13,8 +13,11 @@ module LlmHub
       # Initialize a new embedding client
       # @param api_key [String] API key for the provider (required)
       # @param provider [Symbol, String] Provider name (:openai) (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)
       # @see LlmHub::Common::ClientBase#initialize
-      def initialize(api_key:, provider:)
+      def initialize(api_key:, provider:, open_time_out: nil, read_time_out: nil, retry_count: nil)
         super
         @provider_client = create_provider_client
       end

data/lib/llm_hub/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module LlmHub
-  VERSION = '0.2.0'
+  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.0
+  version: 0.3.0
 platform: ruby
 authors:
 - akiraNuma
 bindir: exe
 cert_chain: []
-date: 2025-06-17 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: []