Class: AI::Engine::Message

Inherits:
ApplicationRecord show all
Includes:
RemoteIdValidatable
Defined in:
app/models/ai/engine/message.rb

Constant Summary

Constants included from RemoteIdValidatable

RemoteIdValidatable::REMOTE_ID_PREFIXES

Instance Method Summary collapse

Instance Method Details

#in_assistant_thread? ⇒ Boolean

Returns:

  • (Boolean) 


40
41
42
 
# File 'app/models/ai/engine/message.rb', line 40

def in_assistant_thread?
  messageable.is_a?(AI::Engine::AssistantThread)
end

#in_chat? ⇒ Boolean

Returns:

  • (Boolean) 


36
37
38
 
# File 'app/models/ai/engine/message.rb', line 36

def in_chat?
  messageable.is_a?(AI::Engine::Chat)
end

#input_cost ⇒ Object 



16
17
18
19
20
21
22
 
# File 'app/models/ai/engine/message.rb', line 16

def input_cost
  return unless assistant?

  return unless prompt_token_usage.present? && model.present?

  (prompt_token_usage.to_i * AI::Engine::DOLLAR_COST_PER_1K_TOKENS[model]["input"] / 1000).round(4)
end

#on_create ⇒ Object 



44
45
46
47
48
49
50
 
# File 'app/models/ai/engine/message.rb', line 44

def on_create
  if in_chat?
    messageable.chattable.ai_engine_on_message_create(message: self)
  else
    messageable.threadable.ai_engine_on_message_create(message: self)
  end
end

#on_update ⇒ Object 



52
53
54
55
56
57
58
 
# File 'app/models/ai/engine/message.rb', line 52

def on_update
  if in_chat?
    messageable.chattable.ai_engine_on_message_update(message: self)
  else
    messageable.threadable.ai_engine_on_message_update(message: self)
  end
end

#output_cost ⇒ Object 



24
25
26
27
28
29
30
 
# File 'app/models/ai/engine/message.rb', line 24

def output_cost
  return unless assistant?

  return unless completion_token_usage.present? && model.present?

  (completion_token_usage.to_i * AI::Engine::DOLLAR_COST_PER_1K_TOKENS[model]["output"] / 1000).round(4)
end

#to_partial_path ⇒ Object 



60
61
62
 
# File 'app/models/ai/engine/message.rb', line 60

def to_partial_path
  "messages/message"
end

#user ⇒ Object 



32
33
34
 
# File 'app/models/ai/engine/message.rb', line 32

def user
  in_chat? ? messageable.chattable : messageable.threadable
end