Class: Roseflow::AI::Model

Inherits:
Object
  • Object
show all
Defined in:
lib/roseflow/ai/model.rb

Instance Attribute Summary collapse

Instance Method Summary collapse

Constructor Details

#initialize(name:, provider:) ⇒ Model

Returns a new instance of Model.



8
9
10
11
12
# File 'lib/roseflow/ai/model.rb', line 8

def initialize(name:, provider:)
  @name = name
  @provider = provider
  @model_ = provider.models.find(name)
end

Instance Attribute Details

#nameObject (readonly)

Returns the value of attribute name.



6
7
8
# File 'lib/roseflow/ai/model.rb', line 6

def name
  @name
end

#providerObject (readonly)

Returns the value of attribute provider.



6
7
8
# File 'lib/roseflow/ai/model.rb', line 6

def provider
  @provider
end

Instance Method Details

#call(operation, input) ⇒ Object



14
15
16
# File 'lib/roseflow/ai/model.rb', line 14

def call(operation, input)
  @model_.call(operation, input)
end