llm_hub 0.2.1 → 0.3.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 353ca2493088bec5034c53bda9389a7277d9b79f9a0af7aa9cd0dea38c7a3d96
-  data.tar.gz: e67417f5a3210cb1a5fbca143df86ed2bb4564ae62efe4dab751e2dd0a1bb1e4
+  metadata.gz: 1de72ed920d28f30dbab0a70be8384dfed017795865cfdfb7af895696cf32774
+  data.tar.gz: 0d2388d88588d9b4008294f468ff77813ea9d75d6d663d5d36ead835c5ff42ec
 SHA512:
-  metadata.gz: 3896b924c84d3e05877333187e60c57918d9f968ad347027faa8cac6e37849315a0b2815966c1e1fc506ca898e26456fb054973965d2a535b1a56ff41852a6a2
-  data.tar.gz: 12c26874593c316bde1f841da6161c4395e00c33f63eff6f7e26e10b978f349150ec1a686bbaedb2527432a8d9e726d29194b64f2f7199ba4604977d68ffd60f
+  metadata.gz: be6017b2455e8a02db445cbee8da303143cd5455bc4b1f7d615ed7ddf3ec3df25c95f1a65e2bf0d65d66e7dbfbde5b6e724ebf82c78c007bc1339e23b6704c15
+  data.tar.gz: 66d508f50a7548f5a6be67658bdfee3c04039d0524e7125e091633601dbc9e70b301d13136a6f390211a78ff2cd46303666b6e0aac3c503acf78a4901c1957ea

data/CHANGELOG.md

@@ -1,5 +1,13 @@
 ## [Unreleased]
 
+## [0.3.1] - 2025-07-08
+
+- Updated description
+
+## [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,8 +1,19 @@
 # LlmHub
 
-This is a Ruby interface for multiple LLM providers, such as OpenAI, Anthropic, and DeepSeek.
+A Ruby interface for multiple LLM providers with Completion and Embedding support.
 
-It provides easy access to Completion and Embedding functionalities.
+## Supported Providers
+
+### Completion API
+- **OpenAI**
+- **Anthropic**
+- **DeepSeek**
+- **Google**
+
+### Embedding API
+- **OpenAI**
+
+The gem provides a unified interface to interact with these different providers, making it easy to switch between them or use multiple providers in your application.
 
 ## Installation
 

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.1'
 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.1
 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,8 @@ 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: A Ruby interface for multiple LLM providers.It provides easy access to
+  Completion and Embedding functionalities.
 email:
 - akiran@akiranumakura.com
 executables: []
@@ -60,6 +60,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 +92,5 @@ 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: A Ruby interface for multiple LLM providers.
 test_files: []