Show AI token usage per saved test

2026-10-16 19:43:15 +02:00
parent 8693720b78
commit 42b5006574
6 changed files with 59 additions and 2 deletions
@@ -9,6 +9,48 @@ use Joomla\CMS\MVC\Model\ListModel;

 class ResultsModel extends ListModel
 {
+    public function getItems()
+    {
+        $items = parent::getItems();
+
+        foreach ($items as $item) {
+            $item->ai_input_tokens = 0;
+            $item->ai_output_tokens = 0;
+            $item->ai_total_tokens = 0;
+            $item->ai_cost_usd = 0.0;
+
+            $answers = json_decode((string) ($item->answers ?? ''), true);
+            if (!is_array($answers)) {
+                continue;
+            }
+
+            foreach ($answers as $answer) {
+                if (!is_array($answer)) {
+                    continue;
+                }
+
+                $usages = $answer['ai_usage_history'] ?? (isset($answer['ai_usage']) ? [$answer['ai_usage']] : []);
+                if (!is_array($usages)) {
+                    continue;
+                }
+
+                foreach ($usages as $usage) {
+                    if (!is_array($usage)) {
+                        continue;
+                    }
+                    $item->ai_input_tokens += (int) ($usage['input_tokens'] ?? 0);
+                    $item->ai_output_tokens += (int) ($usage['output_tokens'] ?? 0);
+                    $item->ai_total_tokens += (int) ($usage['total_tokens'] ?? 0);
+                    if (is_numeric($usage['cost_usd'] ?? null)) {
+                        $item->ai_cost_usd += (float) $usage['cost_usd'];
+                    }
+                }
+            }
+        }
+
+        return $items;
+    }
+
    protected function populateState($ordering = 'r.created', $direction = 'desc')
    {
        parent::populateState($ordering, $direction);