Add short circuit on missing token costs

AllYourBot · Jan 26, 2025 · cf14ea1 · cf14ea1
1 parent 60467b8
commit cf14ea1
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/app/models/message.rb b/app/models/message.rb
@@ -79,10 +79,12 @@ def update_assistant_on_conversation
   end
 
   def update_input_token_cost
+    return if assistant.language_model.input_token_cost_cents.blank?
     self.input_token_cost = assistant.language_model.input_token_cost_cents * input_token_count
   end
 
   def update_output_token_cost
+    return if assistant.language_model.output_token_cost_cents.blank?
     self.output_token_cost = assistant.language_model.output_token_cost_cents * output_token_count
   end
 end