raif 1.0.0 → 1.1.0

data/app/models/raif/model_tools/fetch_url.rb

@@ -1,57 +1,48 @@
 # frozen_string_literal: true
 
 class Raif::ModelTools::FetchUrl < Raif::ModelTool
+  tool_arguments_schema do
+    string "url", description: "The URL to fetch content from"
+  end
 
-  def self.example_model_invocation
+  example_model_invocation do
     {
       "name": tool_name,
       "arguments": { "url": "https://en.wikipedia.org/wiki/NASA" }
     }
   end
 
-  def self.tool_arguments_schema
-    {
-      type: "object",
-      additionalProperties: false,
-      required: ["url"],
-      properties: {
-        url: {
-          type: "string",
-          description: "The URL to fetch content from"
-        }
-      }
-    }
-  end
-
-  def self.tool_description
+  tool_description do
     "Fetch a URL and return the page content as markdown"
   end
 
-  def self.observation_for_invocation(tool_invocation)
-    return "No results found" unless tool_invocation.result.present?
+  class << self
+    def observation_for_invocation(tool_invocation)
+      return "No results found" unless tool_invocation.result.present?
 
-    <<~OBSERVATION
-      Result Status: #{tool_invocation.result["status"]}
-      Result Content:
-      #{tool_invocation.result["content"]}
-    OBSERVATION
-  end
+      <<~OBSERVATION
+        Result Status: #{tool_invocation.result["status"]}
+        Result Content:
+        #{tool_invocation.result["content"]}
+      OBSERVATION
+    end
 
-  def self.process_invocation(tool_invocation)
-    url = tool_invocation.tool_arguments["url"]
-    response = Faraday.get(url)
+    def process_invocation(tool_invocation)
+      url = tool_invocation.tool_arguments["url"]
+      response = Faraday.get(url)
 
-    readable_content = Raif::Utils::ReadableContentExtractor.new(response.body).extract_readable_content
-    markdown_content = Raif::Utils::HtmlToMarkdownConverter.convert(readable_content)
+      readable_content = Raif::Utils::ReadableContentExtractor.new(response.body).extract_readable_content
+      markdown_content = Raif::Utils::HtmlToMarkdownConverter.convert(readable_content)
 
-    tool_invocation.update!(
-      result: {
-        status: response.status,
-        content: markdown_content
-      }
-    )
+      tool_invocation.update!(
+        result: {
+          status: response.status,
+          content: markdown_content
+        }
+      )
 
-    tool_invocation.result
+      tool_invocation.result
+    end
   end
 
 end

data/app/models/raif/model_tools/wikipedia_search.rb

@@ -1,78 +1,69 @@
 # frozen_string_literal: true
 
 class Raif::ModelTools::WikipediaSearch < Raif::ModelTool
+  tool_arguments_schema do
+    string "query", description: "The query to search Wikipedia for"
+  end
 
-  def self.example_model_invocation
+  example_model_invocation do
     {
       "name" => tool_name,
       "arguments" => { "query": "Jimmy Buffett" }
     }
   end
 
-  def self.tool_arguments_schema
-    {
-      type: "object",
-      additionalProperties: false,
-      required: ["query"],
-      properties: {
-        query: {
-          type: "string",
-          description: "The query to search Wikipedia for"
-        }
-      }
-    }
-  end
-
-  def self.tool_description
+  tool_description do
     "Search Wikipedia for information"
   end
 
-  def self.observation_for_invocation(tool_invocation)
-    return "No results found" unless tool_invocation.result.present?
+  class << self
+    def observation_for_invocation(tool_invocation)
+      return "No results found" unless tool_invocation.result.present?
 
-    JSON.pretty_generate(tool_invocation.result)
-  end
+      JSON.pretty_generate(tool_invocation.result)
+    end
 
-  def self.process_invocation(tool_invocation)
-    query = tool_invocation.tool_arguments["query"]
+    def process_invocation(tool_invocation)
+      query = tool_invocation.tool_arguments["query"]
 
-    conn = Faraday.new(url: "https://en.wikipedia.org/w/api.php")
+      conn = Faraday.new(url: "https://en.wikipedia.org/w/api.php")
 
-    response = conn.get do |req|
-      req.params["action"] = "query"
-      req.params["format"] = "json"
-      req.params["list"] = "search"
-      req.params["srsearch"] = query
-      req.params["srlimit"] = 5 # Limit to 5 results
-      req.params["srprop"] = "snippet"
-    end
+      response = conn.get do |req|
+        req.params["action"] = "query"
+        req.params["format"] = "json"
+        req.params["list"] = "search"
+        req.params["srsearch"] = query
+        req.params["srlimit"] = 5 # Limit to 5 results
+        req.params["srprop"] = "snippet"
+      end
 
-    if response.success?
-      results = JSON.parse(response.body)
-      search_results = results.dig("query", "search") || []
+      if response.success?
+        results = JSON.parse(response.body)
+        search_results = results.dig("query", "search") || []
 
-      # Store the results in the tool_invocation
-      tool_invocation.update!(
-        result: {
-          results: search_results.map do |result|
-            {
-              title: result["title"],
-              snippet: result["snippet"],
-              page_id: result["pageid"],
-              url: "https://en.wikipedia.org/wiki/#{result["title"].gsub(" ", "_")}"
-            }
-          end
-        }
-      )
-    else
-      tool_invocation.update!(
-        result: {
-          error: "Failed to fetch results from Wikipedia API: #{response.status} #{response.reason_phrase}"
-        }
-      )
-    end
+        # Store the results in the tool_invocation
+        tool_invocation.update!(
+          result: {
+            results: search_results.map do |result|
+              {
+                title: result["title"],
+                snippet: result["snippet"],
+                page_id: result["pageid"],
+                url: "https://en.wikipedia.org/wiki/#{result["title"].gsub(" ", "_")}"
+              }
+            end
+          }
+        )
+      else
+        tool_invocation.update!(
+          result: {
+            error: "Failed to fetch results from Wikipedia API: #{response.status} #{response.reason_phrase}"
+          }
+        )
+      end
 
-    tool_invocation.result
+      tool_invocation.result
+    end
   end
 
 end

data/app/models/raif/task.rb

@@ -7,6 +7,10 @@ module Raif
     include Raif::Concerns::HasAvailableModelTools
     include Raif::Concerns::InvokesModelTools
     include Raif::Concerns::LlmResponseParsing
+    include Raif::Concerns::LlmTemperature
+    include Raif::Concerns::JsonSchemaDefinition
+
+    llm_temperature 0.7
 
     belongs_to :creator, polymorphic: true
 
@@ -20,12 +24,25 @@ module Raif
 
     delegate :json_response_schema, to: :class
 
-    after_initialize -> { self.available_model_tools ||= [] }
+    scope :completed, -> { where.not(completed_at: nil) }
+    scope :failed, -> { where.not(failed_at: nil) }
+    scope :in_progress, -> { where.not(started_at: nil).where(completed_at: nil, failed_at: nil) }
+    scope :pending, -> { where(started_at: nil, completed_at: nil, failed_at: nil) }
+
+    attr_accessor :files, :images
 
-    def self.llm_response_format(format)
-      raise ArgumentError, "response_format must be one of: #{response_formats.keys.join(", ")}" unless response_formats.keys.include?(format.to_s)
+    after_initialize -> { self.available_model_tools ||= [] }
 
-      after_initialize -> { self.response_format = format }, if: :new_record?
+    def status
+      if completed_at?
+        :completed
+      elsif failed_at?
+        :failed
+      elsif started_at?
+        :in_progress
+      else
+        :pending
+      end
     end
 
     # The primary interface for running a task. It will hit the LLM with the task's prompt and system prompt and return a Raif::Task object.
@@ -34,10 +51,13 @@ module Raif
     # @param creator [Object] The creator of the task (polymorphic association)
     # @param available_model_tools [Array<Class>] Optional array of model tool classes that will be provided to the LLM for it to invoke.
     # @param llm_model_key [Symbol, String] Optional key for the LLM model to use. If blank, Raif.config.default_llm_model_key will be used.
+    # @param images [Array] Optional array of Raif::ModelImageInput objects to include with the prompt.
+    # @param files [Array] Optional array of Raif::ModelFileInput objects to include with the prompt.
     # @param args [Hash] Additional arguments to pass to the instance of the task that is created.
     # @return [Raif::Task, nil] The task instance that was created and run.
-    def self.run(creator:, available_model_tools: [], llm_model_key: nil, **args)
-      task = new(creator:, llm_model_key:, available_model_tools:, started_at: Time.current, **args)
+    def self.run(creator:, available_model_tools: [], llm_model_key: nil, images: [], files: [], **args)
+      task = new(creator:, llm_model_key:, available_model_tools:, started_at: Time.current, images: images, files: files, **args)
+
       task.save!
       task.run
       task
@@ -58,17 +78,19 @@ module Raif
       task
     end
 
-    def run
+    def run(skip_prompt_population: false)
       update_columns(started_at: Time.current) if started_at.nil?
 
-      populate_prompts
-      messages = [{ "role" => "user", "content" => prompt }]
+      populate_prompts unless skip_prompt_population
+      messages = [{ "role" => "user", "content" => message_content }]
+
       mc = llm.chat(
         messages: messages,
         source: self,
         system_prompt: system_prompt,
         response_format: response_format.to_sym,
-        available_model_tools: available_model_tools
+        available_model_tools: available_model_tools,
+        temperature: self.class.temperature
       )
 
       self.raif_model_completion = mc.becomes(Raif::ModelCompletion)
@@ -80,13 +102,18 @@ module Raif
       self
     end
 
+    def re_run
+      update_columns(started_at: Time.current)
+      run(skip_prompt_population: true)
+    end
+
     # Returns the LLM prompt for the task.
     #
     # @param creator [Object] The creator of the task (polymorphic association)
     # @param args [Hash] Additional arguments to pass to the instance of the task that is created.
     # @return [String] The LLM prompt for the task.
     def self.prompt(creator:, **args)
-      new(creator:, **args).prompt
+      new(creator:, **args).build_prompt
     end
 
     # Returns the LLM system prompt for the task.
@@ -95,25 +122,53 @@ module Raif
     # @param args [Hash] Additional arguments to pass to the instance of the task that is created.
     # @return [String] The LLM system prompt for the task.
     def self.system_prompt(creator:, **args)
-      new(creator:, **args).system_prompt
+      new(creator:, **args).build_system_prompt
     end
 
-    def self.json_response_schema
-      nil
+    def self.json_response_schema(&block)
+      if block_given?
+        json_schema_definition(:json_response, &block)
+      elsif schema_defined?(:json_response)
+        schema_for(:json_response)
+      end
     end
 
-  private
-
     def build_prompt
       raise NotImplementedError, "Raif::Task subclasses must implement #build_prompt"
     end
 
     def build_system_prompt
       sp = Raif.config.task_system_prompt_intro
+      sp = sp.call(self) if sp.respond_to?(:call)
       sp += system_prompt_language_preference if requested_language_key.present?
       sp
     end
 
+  private
+
+    def message_content
+      # If there are no images or files, just return the message content can just be a string with the prompt
+      return prompt if images.blank? && files.blank?
+
+      content = [{ "type" => "text", "text" => prompt }]
+
+      images.each do |image|
+        raise Raif::Errors::InvalidModelImageInputError,
+          "Images must be a Raif::ModelImageInput: #{image.inspect}" unless image.is_a?(Raif::ModelImageInput)
+
+        content << image
+      end
+
+      files.each do |file|
+        raise Raif::Errors::InvalidFileInputError,
+          "Files must be a Raif::ModelFileInput: #{file.inspect}" unless file.is_a?(Raif::ModelFileInput)
+
+        content << file
+      end
+
+      content
+    end
+
     def populate_prompts
       self.requested_language_key ||= creator.preferred_language_key if creator.respond_to?(:preferred_language_key)
       self.prompt = build_prompt

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/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>

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

@@ -5,5 +5,6 @@
   <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><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,7 @@
               <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.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>

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>