gliner 0.1.0 → 0.1.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: edbbf72af6499d1823db172793f3dc183f90e445d4ae45587dc1a0cd4005c15e
-  data.tar.gz: 04d0881c074e9e84617591768c1b53767d2f509bbb7497c6bc0920bf3bac96b3
+  metadata.gz: 40c13e552a887bd1d77fc718be8708b228c9e213c9bb49c3591b113738346bd4
+  data.tar.gz: e6dce5cda3fcef66bb9a627ffe23c5324b9395af6bd5eeb3c45a4ea400f52c63
 SHA512:
-  metadata.gz: 0ea1480d83e383534c50d14320f5cbfcc7c7a890101f9e0492cbd486dc36e05fbebaedaeea31ccf4f8b5efb3f553349cee1b25ad890457734443054d7e1dfb91
-  data.tar.gz: da1f798fd89bcaad28a41e67ec2f07e779e8a7d2681ac3d4790f2e27b2adfcfcc02f5c42e7b02c8eb7bc519fea4178f16bad6050576543dde241e8becd18dbec
+  metadata.gz: 723f039e065cd9b86111de3d9478cf711d01efdea5a67ce426cf1bcfb4e84396abe9f9969150b743a8f0e54be5caea0401d25a1dcf191c0f09a932189f19fbed
+  data.tar.gz: 1566d0fc8446ee1463c0dfbeb37d7eb7021ca0f0d04bdf3d83148f1e7b33eb1d2eea4ddea44dcef91bf9b11f7cd2508cdcfd9120ade6fb72ac70aa5510d22896

data/README.md

@@ -2,8 +2,6 @@
 
 ![](https://images.unsplash.com/photo-1625768376503-68d2495d78c5?q=80&w=2225&auto=format&fit=crop&ixlib=rb-4.1.0&ixid=M3wxMjA3fDB8MHxwaG90by1wYWdlfHx8fGVufDB8fHx8fA%3D%3D)
 
-Minimal Ruby inference wrapper for the **GLiNER2** ONNX model using:
-
 ## Install
 
 ```ruby
@@ -11,11 +9,18 @@ gem "gliner"
 ```
 
 ## Usage
-### entities
+
+### Entities
 
 ```ruby
 require "gliner"
 
+Gliner.configure do |config|
+  config.threshold = 0.2
+  config.model_dir = "/path/to/gliner2-multi-v1"
+  config.model_file = "model.onnx"
+end
+
 Gliner.load("path/to/gliner2-multi-v1")
 
 text = "Apple CEO Tim Cook announced iPhone 15 in Cupertino yesterday."
@@ -43,7 +48,7 @@ model = Gliner[labels]
 pp model["Email John Doe at john@example.com.", threshold: 0.5]
 ```
 
-### classification
+### Classification
 
 ```ruby
 model = Gliner.classify[
@@ -61,7 +66,7 @@ Expected shape:
 {"sentiment"=>"negative"}
 ```
 
-### structured extraction
+### Structured extraction
 
 ```ruby
 text = "iPhone 15 Pro Max with 256GB storage, A17 Pro chip, priced at $1199."
@@ -101,6 +106,16 @@ This implementation expects a directory containing:
 - (optional) `config.json` with `max_width` and `max_seq_len`
 
 One publicly available ONNX export is `cuerbot/gliner2-multi-v1` on Hugging Face.
+By default, `model_int8.onnx` is used; set `config.model_file` or `GLINER_MODEL_FILE` to override.
+
+You can also configure the model directory in code:
+
+```ruby
+Gliner.configure do |config|
+  config.model_dir = "/path/to/model_dir"
+  config.model_file = "model_int8.onnx"
+end
+```
 
 ## Integration test
 

data/gliner.gemspec

@@ -11,7 +11,7 @@ Gem::Specification.new do |spec|
   spec.description = 'Basic Ruby inference wrapper for the GLiNER2 ONNX model.'
   spec.homepage = 'https://github.com/elcuervo/gliner'
   spec.license = 'MIT'
-  spec.required_ruby_version = '>= 3.2'
+  spec.required_ruby_version = '>= 3.3'
 
   spec.files = Dir.glob('lib/**/*') + Dir.glob('bin/*') + %w[README.md LICENSE gliner.gemspec]
   spec.require_paths = ['lib']

data/lib/gliner/classifier.rb

@@ -5,7 +5,6 @@ require 'gliner/position_iteration'
 module Gliner
   class Classifier
     include PositionIteration
-
     def initialize(inference, max_width:)
       @inference = inference
       @max_width = max_width
@@ -64,5 +63,6 @@ module Gliner
     def format_label(label, score, include_confidence)
       include_confidence ? { 'label' => label, 'confidence' => score } : label
     end
+
   end
 end

data/lib/gliner/configuration.rb

@@ -0,0 +1,15 @@
+# frozen_string_literal: true
+
+module Gliner
+  class Configuration
+    DEFAULT_THRESHOLD = 0.5
+
+    attr_accessor :threshold, :model_dir, :model_file
+
+    def initialize
+      @threshold = DEFAULT_THRESHOLD
+      @model_dir = nil
+      @model_file = nil
+    end
+  end
+end

data/lib/gliner/model.rb

@@ -100,7 +100,7 @@ module Gliner
     end
 
     def extract_entities(text, entity_types, **options)
-      threshold = options.fetch(:threshold, 0.5)
+      threshold = options.fetch(:threshold, Gliner.config.threshold)
       include_confidence = options.fetch(:include_confidence, false)
       include_spans = options.fetch(:include_spans, false)
 
@@ -125,7 +125,7 @@ module Gliner
     end
 
     def extract_json(text, structures, **options)
-      threshold = options.fetch(:threshold, 0.5)
+      threshold = options.fetch(:threshold, Gliner.config.threshold)
       include_confidence = options.fetch(:include_confidence, false)
       include_spans = options.fetch(:include_spans, false)
 

data/lib/gliner/tasks/entity_extraction.rb

@@ -29,7 +29,7 @@ module Gliner
       end
 
       def process_output(logits, parsed, prepared, options)
-        threshold = options.fetch(:threshold, 0.5)
+        threshold = options.fetch(:threshold, Gliner.config.threshold)
         format_opts = FormatOptions.from(options)
         label_positions = options[:label_positions] || inference.label_positions_for(prepared.word_ids, parsed[:labels].length)
 

data/lib/gliner/tasks/json_extraction.rb

@@ -83,7 +83,7 @@ module Gliner
           parsed[:labels],
           label_positions,
           prepared,
-          threshold: options.fetch(:threshold, 0.5)
+          threshold: options.fetch(:threshold, Gliner.config.threshold)
         )
       end
     end

data/lib/gliner/version.rb

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

data/lib/gliner.rb

@@ -1,6 +1,7 @@
 # frozen_string_literal: true
 
 require 'gliner/version'
+require 'gliner/configuration'
 require 'gliner/model'
 require 'gliner/runners/prepared_task'
 require 'gliner/runners/entity_runner'
@@ -41,13 +42,25 @@ module Gliner
 
   class << self
     attr_writer :model
+    attr_writer :config
 
-    def load(dir, file: 'model_int8.onnx')
-      self.model = Model.from_dir(dir, file: file)
+    def configure
+      yield(config)
+    end
+
+    def config
+      @config ||= Configuration.new
+    end
+
+    def load(dir, file: nil)
+      file ||= ENV['GLINER_MODEL_FILE']
+      file ||= config.model_file
+
+      self.model = Model.from_dir(dir, file: file || 'model_int8.onnx')
     end
 
     def model
-      @model ||= model_from_env
+      @model ||= model_from_config || model_from_env
     end
 
     def model!
@@ -64,6 +77,16 @@ module Gliner
 
     private
 
+    def model_from_config
+      dir = config.model_dir
+      return nil if dir.nil? || dir.empty?
+
+      file = config.model_file
+      return Model.from_dir(dir) if file.nil? || file.empty?
+
+      Model.from_dir(dir, file: file)
+    end
+
     def model_from_env
       dir = ENV.fetch('GLINER_MODEL_DIR', nil)
       return nil if dir.nil? || dir.empty?
@@ -76,7 +99,7 @@ module Gliner
       model = self.model
       return model if model
 
-      raise Error, 'No model loaded. Call Gliner.load("/path/to/model") or set GLINER_MODEL_DIR.'
+      raise Error, 'No model loaded. Call Gliner.load("/path/to/model"), set config.model_dir, or set GLINER_MODEL_DIR.'
     end
 
     def runner_for(config)

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: gliner
 version: !ruby/object:Gem::Version
-  version: 0.1.0
+  version: 0.1.1
 platform: ruby
 authors:
 - elcuervo
@@ -108,6 +108,7 @@ files:
 - lib/gliner/config/entity_types.rb
 - lib/gliner/config/field_spec.rb
 - lib/gliner/config_parser.rb
+- lib/gliner/configuration.rb
 - lib/gliner/inference.rb
 - lib/gliner/inference/session_validator.rb
 - lib/gliner/input_builder.rb
@@ -137,7 +138,7 @@ required_ruby_version: !ruby/object:Gem::Requirement
   requirements:
   - - ">="
     - !ruby/object:Gem::Version
-      version: '3.2'
+      version: '3.3'
 required_rubygems_version: !ruby/object:Gem::Requirement
   requirements:
   - - ">="