llm_hub 0.1.1 → 0.2.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 2e158630f5cb8b2c10a9727709df6f7f657455e4ff41ffa172c4d0d11107be3b
-  data.tar.gz: d3032679c1d2299a2d6a36544eee64b43ee94a7b0f3b584fc1c469320de0d7ee
+  metadata.gz: 353ca2493088bec5034c53bda9389a7277d9b79f9a0af7aa9cd0dea38c7a3d96
+  data.tar.gz: e67417f5a3210cb1a5fbca143df86ed2bb4564ae62efe4dab751e2dd0a1bb1e4
 SHA512:
-  metadata.gz: 893d3e70031f8bcb191047f6024911d580540352a60354ce2047a20bb3e4bd160715c04aaf685e369fc8ec8b19ca5446a2894fde27f22a79d50c78a3897d9365
-  data.tar.gz: 0f5b42a047a08af58a8db307ece86723684dcf6ad97a6a50b296f4ae540effb0d85559ab72e4273fa4183517e8d577cc6b56490eb0aebf8f20e7927a920c88ef
+  metadata.gz: 3896b924c84d3e05877333187e60c57918d9f968ad347027faa8cac6e37849315a0b2815966c1e1fc506ca898e26456fb054973965d2a535b1a56ff41852a6a2
+  data.tar.gz: 12c26874593c316bde1f841da6161c4395e00c33f63eff6f7e26e10b978f349150ec1a686bbaedb2527432a8d9e726d29194b64f2f7199ba4604977d68ffd60f

data/CHANGELOG.md

@@ -1,5 +1,13 @@
 ## [Unreleased]
 
+## [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 and Anthropic.
+This is a Ruby interface for multiple LLM providers, such as OpenAI, Anthropic, and DeepSeek.
 
 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

@@ -8,14 +8,18 @@ module LlmHub
       # @return [Hash<Symbol, Class>] mapping of provider names to their classes
       PROVIDER_CLASSES = {
         openai: Providers::OpenAI,
-        anthropic: Providers::Anthropic
+        anthropic: Providers::Anthropic,
+        deepseek: Providers::Deepseek
       }.freeze
 
       # Initialize a new completion client
       # @param api_key [String] API key for the provider (required)
-      # @param provider [Symbol, String] Provider name (:openai, :anthropic) (required)
+      # @param provider [Symbol, String] Provider name (:openai, :anthropic, :deepseek) (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/completion/providers/deepseek.rb

@@ -0,0 +1,17 @@
+# frozen_string_literal: true
+
+module LlmHub
+  module Completion
+    module Providers
+      # DeepSeek completion provider
+      # Inherits from OpenAI provider since DeepSeek uses OpenAI-compatible API
+      class Deepseek < OpenAI
+        COMPLETIONS_URI = 'https://api.deepseek.com/v1/chat/completions'
+
+        def url
+          COMPLETIONS_URI
+        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.1.1'
+  VERSION = '0.2.1'
 end

data/lib/llm_hub.rb

@@ -18,6 +18,7 @@ require_relative 'llm_hub/common/client_base'
 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/client'
 
 # Embedding providers

metadata

@@ -1,13 +1,13 @@
 --- !ruby/object:Gem::Specification
 name: llm_hub
 version: !ruby/object:Gem::Version
-  version: 0.1.1
+  version: 0.2.1
 platform: ruby
 authors:
 - akiraNuma
 bindir: exe
 cert_chain: []
-date: 2025-05-29 00:00:00.000000000 Z
+date: 2025-07-04 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: activesupport
@@ -59,6 +59,7 @@ files:
 - lib/llm_hub/completion/client.rb
 - 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/openai.rb
 - lib/llm_hub/config.rb
 - lib/llm_hub/embedding/client.rb