raif 1.0.0 → 1.2.0

data/app/views/layouts/raif/admin.html.erb

@@ -32,6 +32,16 @@
       <div class="col-md-3 col-lg-2 d-md-block bg-light sidebar">
         <div class="position-sticky pt-3">
           <ul class="nav flex-column">
+            <li class="nav-item">
+              <a class="nav-link <%= current_page?(raif.admin_stats_path) ? "active" : "" %>" href="<%= raif.admin_stats_path %>">
+                <span class="d-inline-block me-2">
+                  <svg xmlns="http://www.w3.org/2000/svg" width="16" height="16" fill="currentColor" class="bi bi-bar-chart" viewBox="0 0 16 16">
+                    <path d="M4 11H2v3h2v-3zm5-4H7v7h2V7zm5-5v12h-2V2h2zm-2-1a1 1 0 0 0-1 1v12a1 1 0 0 0 1 1h2a1 1 0 0 0 1-1V2a1 1 0 0 0-1-1h-2zM6 7a1 1 0 0 1 1-1h2a1 1 0 0 1 1 1v7a1 1 0 0 1-1 1H7a1 1 0 0 1-1-1V7zm-5 4a1 1 0 0 1 1-1h2a1 1 0 0 1 1 1v3a1 1 0 0 1-1 1H2a1 1 0 0 1-1-1v-3z" />
+                  </svg>
+                </span>
+                <%= t("raif.admin.common.stats") %>
+              </a>
+            </li>
             <li class="nav-item">
               <a class="nav-link <%= current_page?(raif.admin_tasks_path) ? "active" : "" %>" href="<%= raif.admin_tasks_path %>">
                 <span class="d-inline-block me-2">

data/app/views/raif/admin/agents/show.html.erb

@@ -18,7 +18,9 @@
     </div>
     <div class="row mb-3">
       <div class="col-md-3"><strong><%= t("raif.admin.common.creator") %>:</strong></div>
-      <div class="col-md-9"><%= @agent.creator_type %> #<%= @agent.creator_id %></div>
+      <div class="col-md-9">
+        <%= @agent.creator.try(:raif_display_name) || "#{@agent.creator_type} ##{@agent.creator_id}" %>
+      </div>
     </div>
     <div class="row mb-3">
       <div class="col-md-3"><strong><%= t("raif.admin.common.status") %>:</strong></div>

data/app/views/raif/admin/conversations/_conversation.html.erb

@@ -1,7 +1,7 @@
 <tr id="<%= dom_id(conversation) %>" class="raif-conversation">
   <td><%= link_to "##{conversation.id}", raif.admin_conversation_path(conversation) %></td>
   <td><small class="text-muted"><%= conversation.created_at.rfc822 %></small></td>
-  <td><%= conversation.creator_type %> #<%= conversation.creator_id %></td>
+  <td><%= conversation.creator.try(:raif_display_name) || "#{conversation.creator_type} ##{conversation.creator_id}" %></td>
   <td><%= conversation.type %></td>
   <td><%= conversation.conversation_entries_count %></td>
 </tr>

data/app/views/raif/admin/conversations/_conversation_entry.html.erb

@@ -21,6 +21,7 @@
       <%= link_to t("raif.admin.common.model_completion"), raif.admin_model_completion_path(entry.raif_model_completion) %>
     <% end %>
   </div>
+
   <div class="mb-3">
     <strong><%= t("raif.admin.common.user_message") %>:</strong>
     <pre class="mt-2"><%= entry.user_message %></pre>
@@ -31,4 +32,51 @@
       <pre class="mt-2"><%= entry.model_response_message %></pre>
     </div>
   <% end %>
+
+  <% if entry.raif_model_tool_invocations.any? %>
+    <div class="mb-3">
+      <strong><%= t("raif.admin.common.model_tool_invocations") %>:</strong>
+      <div class="ms-3 mt-2">
+        <% entry.raif_model_tool_invocations.each do |invocation| %>
+          <div class="card mb-2">
+            <div class="card-body p-3">
+              <div class="d-flex w-100 justify-content-between align-items-start">
+                <div>
+                  <h6 class="mb-1">
+                    <%= link_to invocation.tool_name, raif.admin_model_tool_invocation_path(invocation) %>
+                  </h6>
+                  <% if invocation.completed_at? %>
+                    <span class="badge bg-success"><%= t("raif.admin.common.completed") %></span>
+                  <% elsif invocation.failed_at? %>
+                    <span class="badge bg-danger"><%= t("raif.admin.common.failed") %></span>
+                  <% else %>
+                    <span class="badge bg-secondary"><%= t("raif.admin.common.pending") %></span>
+                  <% end %>
+                </div>
+                <small class="text-muted"><%= invocation.created_at.rfc822 %></small>
+              </div>
+              <div class="mt-2">
+                <strong><%= t("raif.admin.common.arguments") %>:</strong>
+                <pre class="pre-wrap mt-1"><%= begin
+                                                 JSON.pretty_generate(invocation.tool_arguments)
+                                               rescue StandardError
+                                                 invocation.tool_arguments
+                                               end %></pre>
+              </div>
+              <% if invocation.result.present? %>
+                <div class="mt-2">
+                  <strong><%= t("raif.admin.common.result") %>:</strong>
+                  <pre class="pre-wrap mt-1"><%= begin
+                                                   JSON.pretty_generate(invocation.result)
+                                                 rescue StandardError
+                                                   invocation.result
+                                                 end %></pre>
+                </div>
+              <% end %>
+            </div>
+          </div>
+        <% end %>
+      </div>
+    </div>
+  <% end %>
 </div>

data/app/views/raif/admin/conversations/show.html.erb

@@ -14,7 +14,9 @@
     </div>
     <div class="row mb-3">
       <div class="col-md-3"><strong><%= t("raif.admin.common.creator") %>:</strong></div>
-      <div class="col-md-9"><%= @conversation.creator_type %> #<%= @conversation.creator_id %></div>
+      <div class="col-md-9">
+        <%= @conversation.creator.try(:raif_display_name) || "#{@conversation.creator_type} ##{@conversation.creator_id}" %>
+      </div>
     </div>
     <div class="row mb-3">
       <div class="col-md-3"><strong><%= t("raif.admin.common.type") %>:</strong></div>
@@ -47,7 +49,7 @@
     </div>
     <div class="card-body">
       <div class="list-group">
-        <%= render collection: @conversation.entries.order(created_at: :asc).includes(:raif_model_completion),
+        <%= render collection: @conversation.entries.order(created_at: :asc).includes(:raif_model_completion, :raif_model_tool_invocations),
               partial: "raif/admin/conversations/conversation_entry",
               as: :entry %>
       </div>

data/app/views/raif/admin/model_completions/_model_completion.html.erb

@@ -5,5 +5,13 @@
   <td><%= model_completion.llm_model_key %></td>
   <td><%= model_completion.response_format %></td>
   <td><%= model_completion.total_tokens ? number_with_delimiter(model_completion.total_tokens) : "-" %></td>
+  <td><%= model_completion.total_cost ? number_to_currency(model_completion.total_cost, precision: 6) : "-" %></td>
+  <td>
+    <% if model_completion.citations.present? %>
+      <span class="badge bg-info"><%= model_completion.citations.length %></span>
+    <% else %>
+      <span class="text-muted">-</span>
+    <% end %>
+  </td>
   <td><small class="text-muted"><%= truncate(model_completion.raw_response, length: 100) %></small></td>
 </tr>

data/app/views/raif/admin/model_completions/index.html.erb

@@ -13,6 +13,8 @@
               <th><%= t("raif.admin.common.model") %></th>
               <th><%= t("raif.admin.common.response_format") %></th>
               <th><%= t("raif.admin.common.total_tokens") %></th>
+              <th><%= t("raif.admin.common.total_cost") %></th>
+              <th><%= t("raif.admin.common.citations") %></th>
               <th><%= t("raif.admin.common.response") %></th>
             </tr>
           </thead>

data/app/views/raif/admin/model_completions/show.html.erb

@@ -34,15 +34,42 @@
     </div>
     <div class="row mb-3">
       <div class="col-md-3"><strong><%= t("raif.admin.common.prompt_tokens") %>:</strong></div>
-      <div class="col-md-9"><%= @model_completion.prompt_tokens ? number_with_delimiter(@model_completion.prompt_tokens) : "-" %></div>
+      <div class="col-md-9">
+        <% if @model_completion.prompt_tokens %>
+          <%= number_with_delimiter(@model_completion.prompt_tokens) %>
+          <% if @model_completion.prompt_token_cost %>
+            <small>(<%= t("raif.admin.common.est_cost") %>: <%= "$" %><%= number_with_precision(@model_completion.prompt_token_cost, precision: 6) %>)</small>
+          <% end %>
+        <% else %>
+          -
+        <% end %>
+      </div>
     </div>
     <div class="row mb-3">
       <div class="col-md-3"><strong><%= t("raif.admin.common.completion_tokens") %>:</strong></div>
-      <div class="col-md-9"><%= @model_completion.completion_tokens ? number_with_delimiter(@model_completion.completion_tokens) : "-" %></div>
+      <div class="col-md-9">
+        <% if @model_completion.completion_tokens %>
+          <%= number_with_delimiter(@model_completion.completion_tokens) %>
+          <% if @model_completion.output_token_cost %>
+            <small>(<%= t("raif.admin.common.est_cost") %>: <%= "$" %><%= number_with_precision(@model_completion.output_token_cost, precision: 6) %>)</small>
+          <% end %>
+        <% else %>
+          -
+        <% end %>
+      </div>
     </div>
     <div class="row mb-3">
       <div class="col-md-3"><strong><%= t("raif.admin.common.total_tokens") %>:</strong></div>
-      <div class="col-md-9"><%= @model_completion.total_tokens ? number_with_delimiter(@model_completion.total_tokens) : "-" %></div>
+      <div class="col-md-9">
+        <% if @model_completion.total_tokens %>
+          <%= number_with_delimiter(@model_completion.total_tokens) %>
+          <% if @model_completion.total_cost %>
+            <small>(<%= t("raif.admin.common.est_cost") %>: <%= "$" %><%= number_with_precision(@model_completion.total_cost, precision: 6) %>)</small>
+          <% end %>
+        <% else %>
+          -
+        <% end %>
+      </div>
     </div>
   </div>
 </div>
@@ -115,3 +142,31 @@
     <% end %>
   </div>
 </div>
+
+<div class="card mb-4">
+  <div class="card-header">
+    <h5 class="mb-0"><%= t("raif.admin.common.citations") %></h5>
+  </div>
+  <div class="card-body">
+    <% if @model_completion.citations.present? %>
+      <% @model_completion.citations.each_with_index do |citation, index| %>
+        <div class="mb-3 p-3 border rounded">
+          <div class="d-flex align-items-start">
+            <span class="badge bg-primary me-3"><%= index + 1 %></span>
+            <div class="flex-grow-1">
+              <h6 class="mb-1">
+                <a href="<%= citation["url"] %>" target="_blank" rel="noopener" class="text-decoration-none">
+                  <%= citation["title"] %>
+                  <i class="bi bi-box-arrow-up-right ms-1" style="font-size: 0.8em;"></i>
+                </a>
+              </h6>
+              <small class="text-muted"><%= citation["url"] %></small>
+            </div>
+          </div>
+        </div>
+      <% end %>
+    <% else %>
+      <p class="text-muted mb-0"><%= t("raif.admin.common.no_citations") %></p>
+    <% end %>
+  </div>
+</div>

data/app/views/raif/admin/stats/index.html.erb

@@ -0,0 +1,128 @@
+<div class="d-flex justify-content-between align-items-center my-4">
+  <h1 class="mb-0"><%= t("raif.admin.common.stats") %></h1>
+
+  <div class="period-filter">
+    <%= form_tag raif.admin_stats_path, method: :get, id: "period_filter_form", class: "d-flex align-items-center" do %>
+      <%= select_tag :period,
+            options_for_select(
+              [
+                [t("raif.admin.common.period_day"), "day"],
+                [t("raif.admin.common.period_week"), "week"],
+                [t("raif.admin.common.period_month"), "month"],
+                [t("raif.admin.common.period_all"), "all"]
+              ],
+              @selected_period
+            ),
+            class: "form-select form-select-sm me-2" %>
+      <%= submit_tag t("raif.admin.common.update"), class: "btn btn-sm btn-primary" %>
+    <% end %>
+  </div>
+</div>
+
+<div class="row">
+  <div class="col-12">
+    <div class="card shadow-sm mb-4">
+      <div class="card-body">
+        <div class="row g-4">
+          <!-- Model Completions -->
+          <div class="col-md-4 col-lg-4">
+            <div class="stats-card p-3 border rounded shadow-sm h-100">
+              <div class="d-flex align-items-center mb-2">
+                <span class="stats-icon bg-primary-subtle text-primary rounded p-2 me-3">
+                  <svg xmlns="http://www.w3.org/2000/svg" width="24" height="24" fill="currentColor" class="bi bi-robot" viewBox="0 0 16 16">
+                    <path d="M6 12.5a.5.5 0 0 1 .5-.5h3a.5.5 0 0 1 0 1h-3a.5.5 0 0 1-.5-.5ZM3 8.062C3 6.76 4.235 5.765 5.53 5.886a26.58 26.58 0 0 0 4.94 0C11.765 5.765 13 6.76 13 8.062v1.157a.933.933 0 0 1-.765.935c-.845.147-2.34.346-4.235.346-1.895 0-3.39-.2-4.235-.346A.933.933 0 0 1 3 9.219V8.062Zm4.542-.827a.25.25 0 0 0-.217.068l-.92.9a24.767 24.767 0 0 1-1.871-.183.25.25 0 0 0-.068.495c.55.076 1.232.149 2.02.193a.25.25 0 0 0 .189-.071l.754-.736.847 1.71a.25.25 0 0 0 .404.062l.932-.97a25.286 25.286 0 0 0 1.922-.188.25.25 0 0 0-.068-.495c-.538.074-1.207.145-1.98.189a.25.25 0 0 0-.166.076l-.754.785-.842-1.7a.25.25 0 0 0-.182-.135Z" />
+                    <path d="M8.5 1.866a1 1 0 1 0-1 0V3h-2A4.5 4.5 0 0 0 1 7.5V8a1 1 0 0 0-1 1v2a1 1 0 0 0 1 1v1a2 2 0 0 0 2 2h10a2 2 0 0 0 2-2v-1a1 1 0 0 0 1-1V9a1 1 0 0 0-1-1v-.5A4.5 4.5 0 0 0 10.5 3h-2V1.866ZM14 7.5V13a1 1 0 0 1-1 1H3a1 1 0 0 1-1-1V7.5A3.5 3.5 0 0 1 5.5 4h5A3.5 3.5 0 0 1 14 7.5Z" />
+                  </svg>
+                </span>
+                <div>
+                  <h6 class="mb-0 text-muted"><%= t("raif.admin.common.model_completions") %></h6>
+                </div>
+              </div>
+              <div class="d-flex justify-content-between align-items-center">
+                <h3 class="fw-bold mb-0"><%= number_with_delimiter(@model_completion_count) %></h3>
+                <p class="text-muted mt-2 mb-0"><small><%= t("raif.admin.common.est_cost") %>: <%= number_to_currency(@model_completion_total_cost, precision: 6) %></small></p>
+              </div>
+            </div>
+          </div>
+
+          <!-- Tasks -->
+          <div class="col-md-4 col-lg-4">
+            <div class="stats-card p-3 border rounded shadow-sm h-100">
+              <div class="d-flex align-items-center mb-2">
+                <span class="stats-icon bg-success-subtle text-success rounded p-2 me-3">
+                  <svg xmlns="http://www.w3.org/2000/svg" width="24" height="24" fill="currentColor" class="bi bi-list-check" viewBox="0 0 16 16">
+                    <path fill-rule="evenodd" d="M5 11.5a.5.5 0 0 1 .5-.5h9a.5.5 0 0 1 0 1h-9a.5.5 0 0 1-.5-.5zm0-4a.5.5 0 0 1 .5-.5h9a.5.5 0 0 1 0 1h-9a.5.5 0 0 1-.5-.5zm0-4a.5.5 0 0 1 .5-.5h9a.5.5 0 0 1 0 1h-9a.5.5 0 0 1-.5-.5zM3.854 2.146a.5.5 0 0 1 0 .708l-1.5 1.5a.5.5 0 0 1-.708 0l-.5-.5a.5.5 0 1 1 .708-.708L2 3.293l1.146-1.147a.5.5 0 0 1 .708 0zm0 4a.5.5 0 0 1 0 .708l-1.5 1.5a.5.5 0 0 1-.708 0l-.5-.5a.5.5 0 1 1 .708-.708L2 7.293l1.146-1.147a.5.5 0 0 1 .708 0zm0 4a.5.5 0 0 1 0 .708l-1.5 1.5a.5.5 0 0 1-.708 0l-.5-.5a.5.5 0 0 1 .708-.708l.146.147 1.146-1.147a.5.5 0 0 1 .708 0z" />
+                  </svg>
+                </span>
+                <div>
+                  <h6 class="mb-0 text-muted"><%= t("raif.admin.common.tasks") %></h6>
+                </div>
+              </div>
+              <div class="d-flex justify-content-between align-items-center">
+                <h3 class="fw-bold mb-0"><%= number_with_delimiter(@task_count) %></h3>
+                <div class="mt-2">
+                  <%= link_to raif.admin_stats_tasks_path(period: @selected_period) do %>
+                    <%= t("raif.admin.common.details") %> &raquo;
+                  <% end %>
+                </div>
+              </div>
+            </div>
+          </div>
+
+          <!-- Agents -->
+          <div class="col-md-4 col-lg-4">
+            <div class="stats-card p-3 border rounded shadow-sm h-100">
+              <div class="d-flex align-items-center mb-2">
+                <span class="stats-icon bg-danger-subtle text-danger rounded p-2 me-3">
+                  <svg xmlns="http://www.w3.org/2000/svg" width="24" height="24" fill="currentColor" class="bi bi-cpu" viewBox="0 0 16 16">
+                    <path d="M5 0a.5.5 0 0 1 .5.5V2h1V.5a.5.5 0 0 1 1 0V2h1V.5a.5.5 0 0 1 1 0V2h1V.5a.5.5 0 0 1 1 0V2A2.5 2.5 0 0 1 14 4.5h1.5a.5.5 0 0 1 0 1H14v1h1.5a.5.5 0 0 1 0 1H14v1h1.5a.5.5 0 0 1 0 1H14v1h1.5a.5.5 0 0 1 0 1H14a2.5 2.5 0 0 1-2.5 2.5v1.5a.5.5 0 0 1-1 0V14h-1v1.5a.5.5 0 0 1-1 0V14h-1v1.5a.5.5 0 0 1-1 0V14h-1v1.5a.5.5 0 0 1-1 0V14A2.5 2.5 0 0 1 2 11.5H.5a.5.5 0 0 1 0-1H2v-1H.5a.5.5 0 0 1 0-1H2v-1H.5a.5.5 0 0 1 0-1H2v-1H.5a.5.5 0 0 1 0-1H2A2.5 2.5 0 0 1 4.5 2V.5A.5.5 0 0 1 5 0zm-.5 3A1.5 1.5 0 0 0 3 4.5v7A1.5 1.5 0 0 0 4.5 13h7a1.5 1.5 0 0 0 1.5-1.5v-7A1.5 1.5 0 0 0 11.5 3h-7zM5 6.5A1.5 1.5 0 0 1 6.5 5h3A1.5 1.5 0 0 1 11 6.5v3A1.5 1.5 0 0 1 9.5 11h-3A1.5 1.5 0 0 1 5 9.5v-3zM6.5 6a.5.5 0 0 0-.5.5v3a.5.5 0 0 0 .5.5h3a.5.5 0 0 0 .5-.5v-3a.5.5 0 0 0-.5-.5h-3z" />
+                  </svg>
+                </span>
+                <div>
+                  <h6 class="mb-0 text-muted"><%= t("raif.admin.common.agents") %></h6>
+                </div>
+              </div>
+              <h3 class="fw-bold mb-0"><%= number_with_delimiter(@agent_count) %></h3>
+            </div>
+          </div>
+
+          <!-- Conversations -->
+          <div class="col-md-4 col-lg-4">
+            <div class="stats-card p-3 border rounded shadow-sm h-100">
+              <div class="d-flex align-items-center mb-2">
+                <span class="stats-icon bg-info-subtle text-info rounded p-2 me-3">
+                  <svg xmlns="http://www.w3.org/2000/svg" width="24" height="24" fill="currentColor" class="bi bi-chat-dots" viewBox="0 0 16 16">
+                    <path d="M5 8a1 1 0 1 1-2 0 1 1 0 0 1 2 0zm4 0a1 1 0 1 1-2 0 1 1 0 0 1 2 0zm3 1a1 1 0 1 0 0-2 1 1 0 0 0 0 2z" />
+                    <path d="m2.165 15.803.02-.004c1.83-.363 2.948-.842 3.468-1.105A9.06 9.06 0 0 0 8 15c4.418 0 8-3.134 8-7s-3.582-7-8-7-8 3.134-8 7c0 1.76.743 3.37 1.97 4.6a10.437 10.437 0 0 1-.524 2.318l-.003.011a10.722 10.722 0 0 1-.244.637c-.079.186.074.394.273.362a21.673 21.673 0 0 0 .693-.125zm.8-3.108a1 1 0 0 0-.287-.801C1.618 10.83 1 9.468 1 8c0-3.192 3.004-6 7-6s7 2.808 7 6c0 3.193-3.004 6-7 6a8.06 8.06 0 0 1-2.088-.272a1 1 0 0 0-.711.074c-.387.196-1.24.57-2.634.893a10.97 10.97 0 0 0 .398-2z" />
+                  </svg>
+                </span>
+                <div>
+                  <h6 class="mb-0 text-muted"><%= t("raif.admin.common.conversations") %></h6>
+                </div>
+              </div>
+              <h3 class="fw-bold mb-0"><%= number_with_delimiter(@conversation_count) %></h3>
+            </div>
+          </div>
+
+          <!-- Conversation Entries -->
+          <div class="col-md-4 col-lg-4">
+            <div class="stats-card p-3 border rounded shadow-sm h-100">
+              <div class="d-flex align-items-center mb-2">
+                <span class="stats-icon bg-warning-subtle text-warning rounded p-2 me-3">
+                  <svg xmlns="http://www.w3.org/2000/svg" width="24" height="24" fill="currentColor" class="bi bi-chat-text" viewBox="0 0 16 16">
+                    <path d="M2.678 11.894a1 1 0 0 1 .287.801 10.97 10.97 0 0 1-.398 2c1.395-.323 2.247-.697 2.634-.893a1 1 0 0 1 .71-.074A8.06 8.06 0 0 0 8 14c3.996 0 7-2.807 7-6 0-3.192-3.004-6-7-6S1 4.808 1 8c0 1.468.617 2.83 1.678 3.894zm-.493 3.905a21.682 21.682 0 0 1-.713.129c-.2.032-.352-.176-.273-.362a9.68 9.68 0 0 0 .244-.637l.003-.01c.248-.72.45-1.548.524-2.319C.743 11.37 0 9.76 0 8c0-3.866 3.582-7 8-7s8 3.134 8 7-3.582 7-8 7a9.06 9.06 0 0 1-2.347-.306c-.52.263-1.639.742-3.468 1.105z" />
+                    <path d="M4 5.5a.5.5 0 0 1 .5-.5h7a.5.5 0 0 1 0 1h-7a.5.5 0 0 1-.5-.5zM4 8a.5.5 0 0 1 .5-.5h7a.5.5 0 0 1 0 1h-7A.5.5 0 0 1 4 8zm0 2.5a.5.5 0 0 1 .5-.5h4a.5.5 0 0 1 0 1h-4a.5.5 0 0 1-.5-.5z" />
+                  </svg>
+                </span>
+                <div>
+                  <h6 class="mb-0 text-muted"><%= t("raif.admin.common.conversation_entries") %></h6>
+                </div>
+              </div>
+              <h3 class="fw-bold mb-0"><%= number_with_delimiter(@conversation_entry_count) %></h3>
+            </div>
+          </div>
+        </div>
+      </div>
+    </div>
+  </div>
+</div>

data/app/views/raif/admin/stats/tasks/index.html.erb

@@ -0,0 +1,45 @@
+<%= link_to raif.admin_stats_path do %>
+  &laquo; <%= t("raif.admin.stats.tasks.back_to_stats") %>
+<% end %>
+
+<div class="d-flex justify-content-between align-items-center my-4">
+  <h1 class="mb-0"><%= t("raif.admin.stats.tasks.title") %></h1>
+
+  <div class="period-filter">
+    <%= form_tag raif.admin_stats_tasks_path, method: :get, id: "period_filter_form", class: "d-flex align-items-center" do %>
+      <%= select_tag :period,
+            options_for_select(
+              [
+                [t("raif.admin.common.period_day"), "day"],
+                [t("raif.admin.common.period_week"), "week"],
+                [t("raif.admin.common.period_month"), "month"],
+                [t("raif.admin.common.period_all"), "all"]
+              ],
+              @selected_period
+            ),
+            class: "form-select form-select-sm me-2" %>
+      <%= submit_tag t("raif.admin.common.update"), class: "btn btn-sm btn-primary" %>
+    <% end %>
+  </div>
+</div>
+
+<div class="table-responsive">
+  <table class="table table-striped table-hover">
+    <thead class="table-light">
+      <tr>
+        <th><%= t("raif.admin.common.type") %></th>
+        <th><%= t("raif.admin.common.count") %></th>
+        <th><%= t("raif.admin.common.est_cost") %></th>
+      </tr>
+    </thead>
+    <tbody>
+      <% @task_counts_by_type.each do |type, count| %>
+        <tr>
+          <td><%= type %></td>
+          <td><%= number_with_delimiter(count) %></td>
+          <td><%= number_to_currency(@task_costs_by_type[type] || 0, precision: 6) %></td>
+        </tr>
+      <% end %>
+    </tbody>
+  </table>
+</div>

data/app/views/raif/admin/tasks/_task.html.erb

@@ -2,14 +2,15 @@
   <td><%= link_to "##{task.id}", raif.admin_task_path(task) %></td>
   <td><%= task.type %></td>
   <td><small class="text-muted"><%= task.created_at.rfc822 %></small></td>
-  <td><%= task.creator_type %> #<%= task.creator_id %></td>
+  <td><%= task.creator.try(:raif_display_name) || "#{task.creator_type} ##{task.creator_id}" %></td>
   <td><%= task.llm_model_key %></td>
   <td>
-    <% if task.completed_at? %>
+    <% case task.status %>
+    <% when :completed %>
       <span class="badge bg-success"><%= t("raif.admin.common.completed") %></span>
-    <% elsif task.failed_at? %>
+    <% when :failed %>
       <span class="badge bg-danger"><%= t("raif.admin.common.failed") %></span>
-    <% elsif task.started_at? %>
+    <% when :in_progress %>
       <span class="badge bg-warning text-dark"><%= t("raif.admin.common.in_progress") %></span>
     <% else %>
       <span class="badge bg-secondary"><%= t("raif.admin.common.pending") %></span>

data/app/views/raif/admin/tasks/index.html.erb

@@ -4,10 +4,28 @@
   <div class="col-12">
     <%= form_tag raif.admin_tasks_path, method: :get, class: "mb-4" do %>
       <div class="row align-items-end">
-        <div class="col-md-6">
+        <div class="col-md-4">
           <div class="form-group">
+            <label for="task_types"><%= t("raif.admin.common.type") %></label>
             <%= select_tag :task_types,
-                  options_for_select(@task_types.map{|type| [type, type] }, @selected_types),
+                  options_for_select([["All", "all"]] + @task_types.map{|type| [type, type] }, @selected_type),
+                  { class: "form-select" } %>
+          </div>
+        </div>
+        <div class="col-md-4">
+          <div class="form-group">
+            <label for="task_statuses"><%= t("raif.admin.common.status") %></label>
+            <%= select_tag :task_statuses,
+                  options_for_select(
+                    [
+                      [t("raif.admin.common.all"), :all],
+                      [t("raif.admin.common.completed"), :completed],
+                      [t("raif.admin.common.failed"), :failed],
+                      [t("raif.admin.common.in_progress"), :in_progress],
+                      [t("raif.admin.common.pending"), :pending]
+                    ],
+                    @selected_statuses
+                  ),
                   { class: "form-select" } %>
           </div>
         </div>

data/app/views/raif/admin/tasks/show.html.erb

@@ -14,7 +14,9 @@
     </div>
     <div class="row mb-3">
       <div class="col-md-3"><strong><%= t("raif.admin.common.creator") %>:</strong></div>
-      <div class="col-md-9"><%= @task.creator_type %> #<%= @task.creator_id %></div>
+      <div class="col-md-9">
+        <%= @task.creator.try(:raif_display_name) || "#{@task.creator_type} ##{@task.creator_id}" %>
+      </div>
     </div>
     <div class="row mb-3">
       <div class="col-md-3"><strong><%= t("raif.admin.common.model") %>:</strong></div>

data/app/views/raif/conversation_entries/_citations.html.erb

@@ -0,0 +1,9 @@
+<div class="conversation-entry-citations">
+  <h6><%= t("raif.common.sources") %></h6>
+  <% conversation_entry.citations.each do |citation| %>
+    <%= link_to citation["url"], target: "_blank", rel: "noopener noreferrer", class: "d-flex align-items-center" do %>
+      <%= image_tag "https://www.google.com/s2/favicons?sz=16&domain=#{citation["url"]}", class: "me-1" %>
+      <%= citation["title"] %>
+    <% end %>
+  <% end %>
+</div>

data/app/views/raif/conversation_entries/_conversation_entry.html.erb

@@ -1,26 +1,34 @@
-<div id="<%= dom_id(conversation_entry) %>" class="my-2">
+<div id="<%= dom_id(conversation_entry) %>" class="my-2 conversation-entry">
   <%= render "raif/conversation_entries/message",
         conversation_entry: conversation_entry,
         content: conversation_entry.user_message,
-        message_type: :user %>
+        message_type: :user if conversation_entry.user_message.present? %>
 
-  <% if conversation_entry.completed? || conversation_entry.generating_response? %>
-    <%= render "raif/conversation_entries/message",
-          conversation_entry: conversation_entry,
-          content: conversation_entry.generating_response? ? content_tag(:span, "", class: "raif-loader") : conversation_entry.model_response_message,
-          message_type: :model_response %>
-
-    <% conversation_entry.raif_model_tool_invocations.select(&:renderable?).each do |ti| %>
-      <div class="mb-4 container">
-        <%= render ti, conversation_entry: conversation_entry %>
-      </div>
-    <% end %>
-  <% elsif conversation_entry.failed? %>
+  <% if conversation_entry.failed? %>
     <div class="mb-4 container">
       <%= render "raif/conversation_entries/message",
             conversation_entry: conversation_entry,
             content: t("raif.common.there_was_an_error_generating_this_response"),
             message_type: :model_response %>
     </div>
+  <% elsif conversation_entry.generating_response? %>
+    <%= render "raif/conversation_entries/message",
+          conversation_entry: conversation_entry,
+          content: if conversation_entry.model_response_message.present?
+                     conversation_entry.model_response_message + content_tag(:span, "", class: "raif-streaming-cursor")
+                   else
+                     content_tag(:span, "", class: "raif-loader")
+                   end,
+          message_type: :model_response %>
+
+  <% elsif conversation_entry.completed? %>
+    <%= render "raif/conversation_entries/message",
+          conversation_entry: conversation_entry,
+          content: conversation_entry.model_response_message,
+          message_type: :model_response if conversation_entry.model_response_message.present? %>
+
+    <% conversation_entry.raif_model_tool_invocations.select(&:renderable?).each do |ti| %>
+      <%= render ti, conversation_entry: conversation_entry %>
+    <% end %>
   <% end %>
 </div>

data/app/views/raif/conversation_entries/_form.html.erb

@@ -10,7 +10,7 @@
     <% end %>
   <% end %>
 
-  <div class="d-flex">
+  <div class="d-flex px-2">
     <%= f.text_field :user_message,
           class: "form-control me-2",
           placeholder: conversation_entry.raif_user_tool_invocation&.message_input_placeholder.presence || t("raif.common.type_your_message"),

data/app/views/raif/conversation_entries/_form_with_available_tools.html.erb

@@ -1,4 +1,4 @@
-<hr class="mb-2 mx-n5">
-
-<%= render "raif/conversations/available_user_tools", conversation: conversation %>
-<%= render "raif/conversation_entries/form", conversation: conversation, conversation_entry: conversation_entry %>
+<div class="border-top pt-2">
+  <%= render "raif/conversations/available_user_tools", conversation: conversation %>
+  <%= render "raif/conversation_entries/form", conversation: conversation, conversation_entry: conversation_entry %>
+</div>

data/app/views/raif/conversation_entries/_message.html.erb

@@ -1,12 +1,23 @@
-<div class="d-flex mb-2 chat-message">
-  <div class="d-flex w-100 <%= "justify-content-end" if message_type == :user %>">
+<div class="d-flex my-2 px-3 <%= "justify-content-end" if message_type == :user %>">
+  <div class="d-flex <%= "chat-message-content" if message_type == :user %> <%= "received-message-content" if message_type == :model_response %>">
     <% if message_type == :model_response %>
       <%= render "raif/conversation_entries/model_response_avatar", conversation_entry: local_assigns[:conversation_entry] %>
     <% end %>
 
     <% if content.present? %>
       <div class="mb-1 rounded-2 p-3 <%= message_type == :user ? "bg-primary text-white" : "border" %>">
-        <%= simple_format content %>
+        <% case local_assigns[:conversation_entry]&.response_format %>
+        <% when "text" %>
+          <%= simple_format content %>
+        <% when "html" %>
+          <%= sanitize content %>
+        <% else %>
+          <%= content %>
+        <% end %>
+
+        <% if message_type == :model_response && local_assigns[:conversation_entry]&.citations.present? %>
+          <%= render "raif/conversation_entries/citations", conversation_entry: conversation_entry %>
+        <% end %>
       </div>
     <% end %>