DEV: add spending category for users

Author: keegangeorge

app/serializers/ai_usage_serializer.rb

@@ -55,6 +55,9 @@ def users
         total_cached_tokens: user.total_cached_tokens,
         total_request_tokens: user.total_request_tokens,
         total_response_tokens: user.total_response_tokens,
+        input_spending: user.input_spending,
+        output_spending: user.output_spending,
+        cached_input_spending: user.cached_input_spending,
       }
     end
   end

assets/javascripts/discourse/components/ai-llm-editor-form.gjs

@@ -262,7 +262,6 @@ export default class AiLlmEditorForm extends Component {
       class="ai-llm-editor"
       as |form data|
     >
-      {{log this.formData}}
       {{#if this.modulesUsingModel}}
         <form.Alert @icon="circle-info">
           {{this.inUseWarning}}

assets/javascripts/discourse/components/ai-usage.gjs

@@ -556,6 +556,7 @@ export default class AiUsage extends Component {
                           }}</th>
                         <th>{{i18n "discourse_ai.usage.usage_count"}}</th>
                         <th>{{i18n "discourse_ai.usage.total_tokens"}}</th>
+                        <th>{{i18n "discourse_ai.usage.total_spending"}}</th>
                       </tr>
                     </thead>
                     <tbody>
@@ -580,6 +581,13 @@ export default class AiUsage extends Component {
                             class="ai-usage__users-cell"
                             title={{user.total_tokens}}
                           >{{number user.total_tokens}}</td>
+                          <td>
+                            {{this.totalSpending
+                              user.input_spending
+                              user.cached_input_spending
+                              user.output_spending
+                            }}
+                          </td>
                         </tr>
                       {{/each}}
                     </tbody>

config/locales/client.en.yml

@@ -239,7 +239,7 @@ en:
         net_request_tokens: "Net request tokens"
         cached_tokens: "Cached tokens"
         cached_request_tokens: "Cached request tokens"
-        total_spending: "Total spending"
+        total_spending: "Estimated cost"
         no_users: "No user usage data found"
         no_models: "No model usage data found"
         no_features: "No feature usage data found"

lib/completions/report.rb

@@ -115,7 +115,15 @@ def tokens_by_period(period = nil)
       def user_breakdown
         base_query
           .joins(:user)
-          .group(:user_id, "users.username", "users.uploaded_avatar_id")
+          .joins("LEFT JOIN llm_models ON llm_models.name = language_model")
+          .group(
+            :user_id,
+            "users.username",
+            "users.uploaded_avatar_id",
+            "llm_models.input_cost",
+            "llm_models.output_cost",
+            "llm_models.cached_input_cost",
+          )
           .order("usage_count DESC")
           .limit(USER_LIMIT)
           .select(
@@ -126,6 +134,9 @@ def user_breakdown
             "SUM(COALESCE(cached_tokens,0)) as total_cached_tokens",
             "SUM(COALESCE(request_tokens,0)) as total_request_tokens",
             "SUM(COALESCE(response_tokens,0)) as total_response_tokens",
+            "SUM(COALESCE(request_tokens, 0)) * COALESCE(llm_models.input_cost, 0) / 1000000.0 as input_spending",
+            "SUM(COALESCE(response_tokens, 0)) * COALESCE(llm_models.output_cost, 0) / 1000000.0 as output_spending",
+            "SUM(COALESCE(cached_tokens, 0)) * COALESCE(llm_models.cached_input_cost, 0) / 1000000.0 as cached_input_spending",
           )
       end