gliner 0.2.1 → 0.2.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 749c6a152abe40b2735e56724d01bcb9df57df3ed883f7b7a7a735c7d4dbe613
-  data.tar.gz: dd8add045f98be4876bea92faadd68b2da4a64613502c710c3a6769d224016bd
+  metadata.gz: 81732c1ce09e933b7e2f57291ee6c9c0c9a8442f7a8488667d324a7b25d2981f
+  data.tar.gz: 59cc5210b5915c7e195d99a3c2d80d613f3d9e3aaa9c9ddb58d4311b585bf923
 SHA512:
-  metadata.gz: 996bd63e35f9302f1a01a78d92d39ca47013deb42e58ff3c332ac260c2351f172d5a1897bdb3f673a4a1cf1f63c2ef1b1a7c04c2025ff94be6eb71548c79bf97
-  data.tar.gz: 246d3939daa2ca82df524ae6a89dbd734792d5c6952857d517634f50cc59176f26fd1bef15677b655b1c2f9cb1b35c5cb700bdbbc764d6949b6b941b7c83e77f
+  metadata.gz: 17e88cb80220df06929d6642c92373e88a1628aaee83eb3c8175ea1c5454dc790bb262ea18565d314e533c49082aace8c85b78325aae3ddffa5a5d68188451f7
+  data.tar.gz: 1cafd31a6b5d1c92b1088d3584457f4231853485eb0ba8728155fde333b9f99e5d7b1ec7414ddcf6a9f7547f991ccf98c37564cf9b95a9b1b2ab77f1e11ee68a

data/bin/console

@@ -9,6 +9,7 @@ end
 
 require "gliner"
 require "irb"
+require "debug"
 
 model_dir = ARGV[0] || ENV["GLINER_MODEL_DIR"]
 model_file = ENV["GLINER_MODEL_FILE"]

data/gliner.gemspec

@@ -24,6 +24,7 @@ Gem::Specification.new do |spec|
   spec.add_development_dependency 'irb', '~> 1.16.0'
   spec.add_development_dependency 'rspec', '~> 3.13'
   spec.add_development_dependency 'rubocop', '~> 1.50'
+  spec.add_development_dependency 'debug', '~> 1.11'
 
   spec.metadata['rubygems_mfa_required'] = 'true'
 end

data/lib/gliner/version.rb

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

data/lib/gliner.rb

@@ -134,40 +134,38 @@ module Gliner
       return unless config.auto?
 
       source = config.model
+
       return unless source.nil?
       return if env_model_dir
 
       config.model = download_default_model
     end
 
+    def client
+      @client ||= HTTPX.plugin(:follow_redirects)
+    end
+
     def download_default_model
       model_file = model_file_for_variant(config.variant)
-      root = File.expand_path('..', __dir__)
-      dir = File.join(root, '.cache', 'models', HF_REPO.tr('/', '__'))
+      dir = File.join(Dir.pwd, '.cache', 'gliner', HF_REPO.tr('/', '__'))
 
       FileUtils.mkdir_p(dir)
 
       files = ['tokenizer.json', 'config.json', model_file]
-      client = HTTPX.plugin(:follow_redirects).with(max_redirects: 5)
 
       files.each do |file|
-        dest = File.join(dir, file)
-        next if File.exist?(dest) && File.size?(dest)
-        download_file!(client, "#{DEFAULT_MODEL_BASE}/#{file}", dest)
-      end
+        target = File.join(dir, file)
 
-      dir
-    end
+        next if File.exist?(target) && File.size?(target)
 
-    def download_file!(client, url, dest)
-      response = client.get(url)
-      status = response.status
+        puts "Downloading #{DEFAULT_MODEL_BASE}/#{file}"
 
-      unless status && status.between?(200, 299)
-        raise Error, "Download failed: #{url} (status: #{status || 'unknown'})"
+        client
+          .get("#{DEFAULT_MODEL_BASE}/#{file}")
+          .copy_to(target)
       end
 
-      File.binwrite(dest, response.body.to_s)
+      dir
     end
 
     def env_model_dir

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: gliner
 version: !ruby/object:Gem::Version
-  version: 0.2.1
+  version: 0.2.2
 platform: ruby
 authors:
 - elcuervo
@@ -107,6 +107,20 @@ dependencies:
     - - "~>"
       - !ruby/object:Gem::Version
         version: '1.50'
+- !ruby/object:Gem::Dependency
+  name: debug
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.11'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.11'
 description: Basic Ruby inference wrapper for the GLiNER2 ONNX model.
 executables: []
 extensions: []