Class: Raif::ModelManifest::Dsl::EmbeddingRegistryBuilder

Inherits:
Object
  • Object
show all
Defined in:
lib/raif/model_manifest/dsl.rb

Instance Method Summary collapse

Constructor Details

#initialize(source_path:) ⇒ EmbeddingRegistryBuilder

Returns a new instance of EmbeddingRegistryBuilder.



215
216
217
218
# File 'lib/raif/model_manifest/dsl.rb', line 215

def initialize(source_path:)
  @source_path = source_path
  @providers = []
end

Instance Method Details

#finish ⇒ Object



226
227
228
# File 'lib/raif/model_manifest/dsl.rb', line 226

def finish
  @providers.freeze
end

#provider(name, adapter:, &block) ⇒ Object



220
221
222
223
224
# File 'lib/raif/model_manifest/dsl.rb', line 220

def provider(name, adapter:, &block)
  builder = EmbeddingProviderBuilder.new(name: name, adapter_class_name: adapter, source_path: @source_path)
  block.call(builder)
  @providers << builder.finish
end