Class: Kreuzberg::EmbeddingModelTypeCustom

Inherits:
Data
  • Object
show all
Extended by:
T::Sig
Includes:
EmbeddingModelType
Defined in:
lib/kreuzberg/native.rb

Overview

Use a custom ONNX model from HuggingFace

Instance Attribute Summary collapse

Class Method Summary collapse

Instance Method Summary collapse

Instance Attribute Details

#dimensions ⇒ Object (readonly)

rubocop:disable Lint/UselessMethodDefinition



231
232
233
 
# File 'lib/kreuzberg/native.rb', line 231

def dimensions
  @dimensions
end

#model_id ⇒ Object (readonly)

rubocop:disable Lint/UselessMethodDefinition



231
232
233
 
# File 'lib/kreuzberg/native.rb', line 231

def model_id
  @model_id
end

Class Method Details

.from_hash(hash) ⇒ Object 



252
253
254
 
# File 'lib/kreuzberg/native.rb', line 252

def self.from_hash(hash)
  new(model_id: hash[:model_id] || hash["model_id"], dimensions: hash[:dimensions] || hash["dimensions"])
end

Instance Method Details

#custom? ⇒ Boolean

Returns:

  • (Boolean) 


244
 
# File 'lib/kreuzberg/native.rb', line 244

def custom? = true

#llm? ⇒ Boolean

Returns:

  • (Boolean) 


246
 
# File 'lib/kreuzberg/native.rb', line 246

def llm? = false

#plugin? ⇒ Boolean

Returns:

  • (Boolean) 


248
249
250
 
# File 'lib/kreuzberg/native.rb', line 248

def plugin? = false
# @param hash [Hash] deserialized from the native extension
# @return [self]

#preset? ⇒ Boolean

Returns:

  • (Boolean) 


242
 
# File 'lib/kreuzberg/native.rb', line 242

def preset? = false