Class: Raif::EmbeddingModel

Inherits:
Object
  • Object
show all
Includes:
ActiveModel::Model
Defined in:
app/models/raif/embedding_model.rb

Instance Attribute Summary collapse

Instance Method Summary collapse

Instance Attribute Details

#api_name ⇒ Object

Returns the value of attribute api_name.



6
7
8
 
# File 'app/models/raif/embedding_model.rb', line 6

def api_name
  @api_name
end

#default_output_vector_size ⇒ Object

Returns the value of attribute default_output_vector_size.



6
7
8
 
# File 'app/models/raif/embedding_model.rb', line 6

def default_output_vector_size
  @default_output_vector_size
end

#display_name ⇒ Object

Returns the value of attribute display_name.



6
7
8
 
# File 'app/models/raif/embedding_model.rb', line 6

def display_name
  @display_name
end

#input_token_cost ⇒ Object

Returns the value of attribute input_token_cost.



6
7
8
 
# File 'app/models/raif/embedding_model.rb', line 6

def input_token_cost
  @input_token_cost
end

#key ⇒ Object

Returns the value of attribute key.



6
7
8
 
# File 'app/models/raif/embedding_model.rb', line 6

def key
  @key
end

#lifecycle ⇒ Object 



18
 
# File 'app/models/raif/embedding_model.rb', line 18

def lifecycle = @lifecycle || {}

#pricing ⇒ Object 



19
 
# File 'app/models/raif/embedding_model.rb', line 19

def pricing = @pricing || {}

Instance Method Details

#generate_embedding!(input, dimensions: nil) ⇒ Object

Raises:

  • (NotImplementedError) 


29
30
31
 
# File 'app/models/raif/embedding_model.rb', line 29

def generate_embedding!(input, dimensions: nil)
  raise NotImplementedError, "#{self.class.name} must implement #generate_embedding!"
end

#lifecycle_status ⇒ Object 



21
22
23
 
# File 'app/models/raif/embedding_model.rb', line 21

def lifecycle_status
  lifecycle[:status]
end

#name ⇒ Object 



25
26
27
 
# File 'app/models/raif/embedding_model.rb', line 25

def name
  I18n.t("raif.embedding_model_names.#{key}", default: display_name || key.to_s.humanize)
end