raif 1.3.0 → 1.5.0

data/app/models/raif/streaming_responses/google.rb

@@ -0,0 +1,71 @@
+# frozen_string_literal: true
+
+class Raif::StreamingResponses::Google
+
+  def initialize
+    @response_json = { "candidates" => [{ "content" => { "parts" => [] } }], "usageMetadata" => {} }
+    @finish_reason = nil
+  end
+
+  def process_streaming_event(event_type, event)
+    delta = nil
+
+    # Google streams complete candidate objects, so we need to extract the new text
+    candidates = event["candidates"]
+    if candidates.present?
+      candidate = candidates[0]
+
+      # Check for finish reason
+      @finish_reason = candidate["finishReason"] if candidate["finishReason"].present?
+
+      # Process content parts
+      parts = candidate.dig("content", "parts")
+      delta = process_content_parts(parts) if parts.present?
+    end
+
+    # Update usage metadata
+    usage_metadata = event["usageMetadata"]
+    @response_json["usageMetadata"] = usage_metadata if usage_metadata.present?
+
+    [delta, @finish_reason]
+  end
+
+  def current_response_json
+    @response_json
+  end
+
+private
+
+  def process_content_parts(parts)
+    delta = nil
+
+    parts.each_with_index do |part, index|
+      if part.key?("text")
+        delta = part["text"]
+        accumulate_text_part(part, index)
+      else
+        # For non-text parts (e.g., functionCall), just store directly.
+        # Note: This works because we don't enable streamFunctionCallArguments in the API request.
+        # Without that opt-in flag, function calls arrive complete in a single chunk.
+        # If streaming function call arguments is enabled in the future (Gemini 3+ models),
+        # this would need to accumulate partialArgs similar to how we accumulate text.
+        @response_json["candidates"][0]["content"]["parts"][index] = part
+      end
+    end
+
+    delta
+  end
+
+  def accumulate_text_part(part, index)
+    existing_part = @response_json.dig("candidates", 0, "content", "parts", index)
+
+    if existing_part.present? && existing_part.key?("text")
+      # Accumulate text from incremental chunks
+      existing_part["text"] += part["text"]
+    else
+      # First text chunk for this index
+      @response_json["candidates"][0]["content"]["parts"][index] = part.dup
+    end
+  end
+
+end

data/app/models/raif/task.rb

@@ -1,19 +1,62 @@
 # frozen_string_literal: true
 
+# == Schema Information
+#
+# Table name: raif_tasks
+#
+#  id                     :bigint           not null, primary key
+#  available_model_tools  :jsonb            not null
+#  completed_at           :datetime
+#  creator_type           :string
+#  failed_at              :datetime
+#  llm_model_key          :string           not null
+#  prompt                 :text
+#  prompt_studio_run      :boolean          default(FALSE), not null
+#  raw_response           :text
+#  requested_language_key :string
+#  response_format        :integer          default("text"), not null
+#  run_with               :jsonb
+#  source_type            :string
+#  started_at             :datetime
+#  system_prompt          :text
+#  type                   :string           not null
+#  created_at             :datetime         not null
+#  updated_at             :datetime         not null
+#  creator_id             :bigint
+#  source_id              :bigint
+#
+# Indexes
+#
+#  index_raif_tasks_on_completed_at           (completed_at)
+#  index_raif_tasks_on_created_at             (created_at)
+#  index_raif_tasks_on_creator                (creator_type,creator_id)
+#  index_raif_tasks_on_failed_at              (failed_at)
+#  index_raif_tasks_on_source                 (source_type,source_id)
+#  index_raif_tasks_on_started_at             (started_at)
+#  index_raif_tasks_on_type                   (type)
+#  index_raif_tasks_on_type_and_completed_at  (type,completed_at)
+#  index_raif_tasks_on_type_and_failed_at     (type,failed_at)
+#  index_raif_tasks_on_type_and_started_at    (type,started_at)
+#
 module Raif
   class Task < Raif::ApplicationRecord
+    prepend Raif::Concerns::HasPromptTemplates
+
     include Raif::Concerns::HasLlm
     include Raif::Concerns::HasRequestedLanguage
     include Raif::Concerns::HasAvailableModelTools
+    include Raif::Concerns::HasRuntimeDuration
     include Raif::Concerns::InvokesModelTools
     include Raif::Concerns::LlmResponseParsing
     include Raif::Concerns::LlmTemperature
+    include Raif::Concerns::LlmPromptCaching
     include Raif::Concerns::JsonSchemaDefinition
-    include Raif::Concerns::TaskRunArgs
+    include Raif::Concerns::RunWith
 
     llm_temperature 0.7
 
     belongs_to :creator, polymorphic: true, optional: true
+    belongs_to :source, polymorphic: true, optional: true
 
     validates :creator, presence: true, unless: -> { Raif.config.task_creator_optional }
 
@@ -25,8 +68,6 @@ module Raif
 
     normalizes :prompt, :system_prompt, with: ->(text){ text&.strip }
 
-    delegate :json_response_schema, to: :class
-
     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) }
@@ -35,7 +76,7 @@ module Raif
     attr_accessor :files, :images
 
     after_initialize -> { self.available_model_tools ||= [] }
-    after_initialize -> { self.task_run_args ||= {} }
+    after_initialize -> { self.run_with ||= {} }
 
     def status
       if completed_at?
@@ -59,7 +100,7 @@ module Raif
     # @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: nil, available_model_tools: [], llm_model_key: nil, images: [], files: [], **args)
+    def self.run(creator: nil, available_model_tools: [], llm_model_key: nil, images: [], files: [], **args, &block)
       task = new(
         creator: creator,
         llm_model_key: llm_model_key,
@@ -67,12 +108,11 @@ module Raif
         started_at: Time.current,
         images: images,
         files: files,
-        task_run_args: serialize_task_run_args(args),
         **args
       )
 
       task.save!
-      task.run
+      task.run(&block)
       task
     rescue StandardError => e
       task&.failed!
@@ -91,11 +131,17 @@ module Raif
       task
     end
 
-    def run(skip_prompt_population: false)
+    def run(skip_prompt_population: false, &block)
       update_columns(started_at: Time.current) if started_at.nil?
 
       populate_prompts unless skip_prompt_population
-      messages = [{ "role" => "user", "content" => message_content }]
+
+      streaming_block = if block_given?
+        proc do |model_completion, delta, sse_event|
+          update_columns(raw_response: model_completion.raw_response, updated_at: Time.current)
+          block.call(model_completion, delta, sse_event)
+        end
+      end
 
       mc = llm.chat(
         messages: messages,
@@ -103,7 +149,10 @@ module Raif
         system_prompt: system_prompt,
         response_format: response_format.to_sym,
         available_model_tools: available_model_tools,
-        temperature: self.class.temperature
+        temperature: self.class.temperature,
+        anthropic_prompt_caching_enabled: self.class.anthropic_prompt_caching_enabled,
+        bedrock_prompt_caching_enabled: self.class.bedrock_prompt_caching_enabled,
+        &streaming_block
       )
 
       self.raif_model_completion = mc.becomes(Raif::ModelCompletion)
@@ -115,9 +164,13 @@ module Raif
       self
     end
 
-    def re_run
+    def re_run(&block)
       update_columns(started_at: Time.current)
-      run(skip_prompt_population: true)
+      run(skip_prompt_population: true, &block)
+    end
+
+    def messages
+      [{ "role" => "user", "content" => message_content }]
     end
 
     # Returns the LLM prompt for the task.
@@ -146,6 +199,22 @@ module Raif
       end
     end
 
+    # Instance method to get the JSON response schema
+    # For instance-dependent schemas, builds the schema with this instance as context
+    # For class-level schemas, returns the class-level schema
+    def json_response_schema
+      schema_for_instance(:json_response)
+    end
+
+    # Returns additional attributes to assign when creating tasks in Prompt Studio
+    # (reruns, batch runs, and judge tasks). Override in your ApplicationTask or
+    # task subclass to include app-specific attributes.
+    #
+    # @return [Hash] additional attributes to assign to the new task
+    def prompt_studio_task_attributes
+      {}
+    end
+
     def build_prompt
       raise NotImplementedError, "Raif::Task subclasses must implement #build_prompt"
     end
@@ -189,15 +258,13 @@ module Raif
     end
 
     def process_model_tool_invocations
-      return unless response_format_json?
-      return unless parsed_response.is_a?(Hash)
-      return unless parsed_response["tools"].present? && parsed_response["tools"].is_a?(Array)
+      return unless raif_model_completion&.response_tool_calls.present?
 
-      parsed_response["tools"].each do |t|
-        tool_klass = available_model_tools_map[t["name"]]
+      raif_model_completion.response_tool_calls.each do |tool_call|
+        tool_klass = available_model_tools_map[tool_call["name"]]
         next unless tool_klass
 
-        tool_klass.invoke_tool(tool_arguments: t["arguments"], source: self)
+        tool_klass.invoke_tool(provider_tool_call_id: tool_call["provider_tool_call_id"], tool_arguments: tool_call["arguments"], source: self)
       end
     end
 

data/app/models/raif/user_tool_invocation.rb

@@ -1,5 +1,24 @@
 # frozen_string_literal: true
 
+# == Schema Information
+#
+# Table name: raif_user_tool_invocations
+#
+#  id                         :bigint           not null, primary key
+#  tool_settings              :jsonb            not null
+#  type                       :string           not null
+#  created_at                 :datetime         not null
+#  updated_at                 :datetime         not null
+#  raif_conversation_entry_id :bigint           not null
+#
+# Indexes
+#
+#  index_raif_user_tool_invocations_on_raif_conversation_entry_id  (raif_conversation_entry_id)
+#
+# Foreign Keys
+#
+#  fk_rails_...  (raif_conversation_entry_id => raif_conversation_entries.id)
+#
 class Raif::UserToolInvocation < Raif::ApplicationRecord
   belongs_to :raif_conversation_entry, class_name: "Raif::ConversationEntry"
 

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

@@ -5,14 +5,23 @@
   <meta name="viewport" content="width=device-width,initial-scale=1">
   <%= csrf_meta_tags %>
   <%= csp_meta_tag %>
+  <%= action_cable_meta_tag if respond_to?(:action_cable_meta_tag) %>
 
   <link href="https://cdn.jsdelivr.net/npm/bootstrap@5.3.2/dist/css/bootstrap.min.css" rel="stylesheet" integrity="sha384-T3c6CoIi6uLrA9TneNEoa7RxnatzjcDSCmG1MXxSR1GAsXEV/Dwwykc2MPK8M2HN" crossorigin="anonymous">
   <script src="https://cdn.jsdelivr.net/npm/bootstrap@5.3.2/dist/js/bootstrap.bundle.min.js" integrity="sha384-C6RzsynM9kWDrMNeT87bh95OGNyZPhcTNXj1NW7RuBCsyN/o0jlpcV8Qyq46cDfL" crossorigin="anonymous" nonce="<%= request.content_security_policy_nonce %>"></script>
 
+  <link href="https://cdn.jsdelivr.net/npm/tom-select@2.4.3/dist/css/tom-select.bootstrap5.min.css" rel="stylesheet">
+  <script src="https://cdn.jsdelivr.net/npm/tom-select@2.4.3/dist/js/tom-select.complete.min.js" nonce="<%= request.content_security_policy_nonce %>"></script>
+
+  <% if respond_to?(:javascript_importmap_tags) %>
+    <%= javascript_importmap_tags "raif_admin" %>
+  <% else %>
+    <%= javascript_include_tag "raif_admin_sprockets", nonce: true %>
+  <% end %>
   <%= stylesheet_link_tag "raif_admin" %>
 </head>
 
-<body class="raif-admin">
+<body class="raif-admin" data-controller="raif--copy-to-clipboard">
   <nav class="navbar navbar-expand-md navbar-dark bg-dark">
     <div class="container-fluid">
       <a class="navbar-brand fw-bold" href="<%= raif.admin_tasks_path %>">
@@ -35,7 +44,7 @@
         <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 %>">
+              <a class="nav-link <%= params[:controller].start_with?("raif/admin/stats") ? "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" />
@@ -96,6 +105,38 @@
                 <%= t("raif.admin.common.agents") %>
               </a>
             </li>
+            <li class="nav-item">
+              <a class="nav-link <%= current_page?(raif.admin_llms_path) ? "active" : "" %>" href="<%= raif.admin_llms_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-currency-dollar" viewBox="0 0 16 16">
+                    <path d="M4 10.781c.148 1.667 1.513 2.85 3.591 3.003V15h1.043v-1.216c2.27-.179 3.678-1.438 3.678-3.3 0-1.59-.947-2.51-2.956-3.028l-.722-.187V3.467c1.122.11 1.879.714 2.07 1.616h1.47c-.166-1.6-1.54-2.748-3.54-2.875V1H7.591v1.233c-1.939.23-3.27 1.472-3.27 3.156 0 1.454.966 2.483 2.661 2.917l.61.162v4.031c-1.149-.17-1.94-.8-2.131-1.718H4zm3.391-3.836c-1.043-.263-1.6-.825-1.6-1.616 0-.944.704-1.641 1.8-1.828v3.495l-.2-.05zm1.591 1.872c1.287.323 1.852.859 1.852 1.769 0 1.097-.826 1.828-2.2 1.939V8.73l.348.086z" />
+                  </svg>
+                </span>
+                <%= t("raif.admin.common.llms") %>
+              </a>
+            </li>
+            <li class="nav-item">
+              <a class="nav-link <%= params[:controller].start_with?("raif/admin/prompt_studio") ? "active" : "" %>" href="<%= raif.admin_prompt_studio_tasks_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-pencil-square" viewBox="0 0 16 16">
+                    <path d="M15.502 1.94a.5.5 0 0 1 0 .706L14.459 3.69l-2-2L13.502.646a.5.5 0 0 1 .707 0l1.293 1.293zm-1.75 2.456-2-2L4.939 9.21a.5.5 0 0 0-.121.196l-.805 2.414a.25.25 0 0 0 .316.316l2.414-.805a.5.5 0 0 0 .196-.12l6.813-6.814z" />
+                    <path fill-rule="evenodd" d="M1 13.5A1.5 1.5 0 0 0 2.5 15h11a1.5 1.5 0 0 0 1.5-1.5v-6a.5.5 0 0 0-1 0v6a.5.5 0 0 1-.5.5h-11a.5.5 0 0 1-.5-.5v-11a.5.5 0 0 1 .5-.5H9a.5.5 0 0 0 0-1H2.5A1.5 1.5 0 0 0 1 2.5v11z" />
+                  </svg>
+                </span>
+                <%= t("raif.admin.prompt_studio.common.prompt_studio") %>
+              </a>
+            </li>
+            <li class="nav-item">
+              <a class="nav-link <%= current_page?(raif.admin_config_path) ? "active" : "" %>" href="<%= raif.admin_config_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-gear" viewBox="0 0 16 16">
+                    <path d="M8 4.754a3.246 3.246 0 1 0 0 6.492 3.246 3.246 0 0 0 0-6.492zM5.754 8a2.246 2.246 0 1 1 4.492 0 2.246 2.246 0 0 1-4.492 0z" />
+                    <path d="M9.796 1.343c-.527-1.79-3.065-1.79-3.592 0l-.094.319a.873.873 0 0 1-1.255.52l-.292-.16c-1.64-.892-3.433.902-2.54 2.541l.159.292a.873.873 0 0 1-.52 1.255l-.319.094c-1.79.527-1.79 3.065 0 3.592l.319.094a.873.873 0 0 1 .52 1.255l-.16.292c-.892 1.64.901 3.434 2.541 2.54l.292-.159a.873.873 0 0 1 1.255.52l.094.319c.527 1.79 3.065 1.79 3.592 0l.094-.319a.873.873 0 0 1 1.255-.52l.292.16c1.64.893 3.434-.902 2.54-2.541l-.159-.292a.873.873 0 0 1 .52-1.255l.319-.094c1.79-.527 1.79-3.065 0-3.592l-.319-.094a.873.873 0 0 1-.52-1.255l.16-.292c.893-1.64-.902-3.433-2.541-2.54l-.292.159a.873.873 0 0 1-1.255-.52l-.094-.319z" />
+                  </svg>
+                </span>
+                <%= t("raif.admin.common.config") %>
+              </a>
+            </li>
           </ul>
         </div>
       </div>

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

@@ -1,5 +1,6 @@
 <tr id="<%= dom_id(agent) %>" class="raif-agent">
   <td><%= link_to "##{agent.id}", raif.admin_agent_path(agent) %></td>
+  <td><small class="text-muted"><%= agent.type.demodulize %></small></td>
   <td><small class="text-muted"><%= agent.created_at.rfc822 %></small></td>
   <td><small class="text-muted"><%= truncate(agent.task, length: 100) %></small></td>
   <td>
@@ -13,6 +14,14 @@
       <span class="badge bg-secondary"><%= t("raif.admin.common.pending") %></span>
     <% end %>
   </td>
+  <td><small class="text-muted"><%= agent.runtime_duration %></small></td>
   <td><%= agent.iteration_count %> / <%= agent.max_iterations %></td>
+  <td>
+    <% if agent.total_cost && agent.total_cost > 0 %>
+      <small><%= "$" %><%= number_with_precision(agent.total_cost, precision: 6) %></small>
+    <% else %>
+      -
+    <% end %>
+  </td>
   <td><small class="text-muted"><%= truncate(agent.final_answer, length: 100) if agent.final_answer %></small></td>
 </tr>

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

@@ -1,15 +1,37 @@
-<div class="message <%= message["role"] %> mb-3">
+<% message_type = message["type"] || message["role"] %>
+<div class="message <%= message_type %> mb-3">
   <div class="message-header d-flex justify-content-between align-items-center mb-2 px-2">
-    <strong class="<%= message["role"] == "user" ? "text-primary" : "text-success" %>">
-      <%= message["role"].capitalize %>
-      <% if message[:counter] == 0 && message["role"] == 'user' %>
-        (<%= t("raif.admin.common.initial_task") %>)
+    <strong class="<%= conversation_message_header_class(message) %>">
+      <% if message_type == "tool_call" %>
+        <%= t("raif.admin.common.tool_call") %>: <%= message["name"] %>
+      <% elsif message_type == "tool_call_result" %>
+        <%= t("raif.admin.common.tool_result") %>
+      <% else %>
+        <%= message["role"]&.capitalize %>
+        <% if message[:counter] == 0 && message["role"] == 'user' %>
+          (<%= t("raif.admin.common.initial_task") %>)
+        <% end %>
+      <% end %>
+      <% if is_final_answer %>
+        <span class="badge bg-info ms-2"><%= t("raif.admin.common.final_answer") %></span>
       <% end %>
     </strong>
     <span class="badge bg-secondary">#<%= message_count %></span>
   </div>
 
   <div class="message-content px-3 py-2">
-    <code><%= message["content"] %></code>
+    <% if message_type == "tool_call" %>
+      <% if message["assistant_message"].present? %>
+        <div class="mb-2"><code><%= message["assistant_message"] %></code></div>
+      <% end %>
+      <div>
+        <strong><%= t("raif.admin.common.arguments") %>:</strong>
+        <code><%= message["arguments"].to_json %></code>
+      </div>
+    <% elsif message_type == "tool_call_result" %>
+      <code><%= message["result"].is_a?(String) ? message["result"] : JSON.pretty_generate(message["result"]) %></code>
+    <% else %>
+      <code><%= message["content"] %></code>
+    <% end %>
   </div>
 </div>

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

@@ -2,16 +2,66 @@
 
 <div class="row">
   <div class="col-12">
+    <%= form_tag raif.admin_agents_path, method: :get, class: "mb-4" do %>
+      <div class="row align-items-end">
+        <div class="col-md-3">
+          <div class="form-group">
+            <label for="agent_type"><%= t("raif.admin.common.type") %></label>
+            <%= select_tag :agent_type,
+                  options_for_select(
+                    [[t("raif.admin.common.all"), "all"]] + @agent_types.map { |type| [type, type] },
+                    @selected_type
+                  ),
+                  { class: "form-select", data: { controller: "raif--tom-select" } } %>
+          </div>
+        </div>
+        <div class="col-md-3">
+          <div class="form-group">
+            <label for="status"><%= t("raif.admin.common.status") %></label>
+            <%= select_tag :status,
+                  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.running"), :running],
+                      [t("raif.admin.common.pending"), :pending]
+                    ],
+                    @selected_status
+                  ),
+                  { class: "form-select", data: { controller: "raif--tom-select" } } %>
+          </div>
+        </div>
+        <div class="col-md-3">
+          <div class="form-group">
+            <label for="llm_model_key"><%= t("raif.admin.common.model") %></label>
+            <%= select_tag :llm_model_key,
+                  options_for_select(
+                    [[t("raif.admin.common.all"), ""]] + @llm_model_keys.map { |key| [key, key] },
+                    @selected_llm_model_key
+                  ),
+                  { class: "form-select", data: { controller: "raif--tom-select" } } %>
+          </div>
+        </div>
+        <div class="col-md-2">
+          <%= submit_tag t("raif.admin.common.filter"), class: "btn btn-primary" %>
+        </div>
+      </div>
+    <% end %>
+
     <% if @agents.any? %>
       <div class="table-responsive">
         <table class="table table-striped table-hover">
           <thead class="table-light">
             <tr>
               <th><%= t("raif.admin.common.id") %></th>
+              <th><%= t("raif.admin.common.type") %></th>
               <th><%= t("raif.admin.common.created_at") %></th>
               <th><%= t("raif.admin.common.task") %></th>
               <th><%= t("raif.admin.common.status") %></th>
+              <th><%= t("raif.admin.common.duration") %></th>
               <th><%= t("raif.admin.common.iterations") %></th>
+              <th><%= t("raif.admin.common.total_cost") %></th>
               <th><%= t("raif.admin.common.final_answer") %></th>
             </tr>
           </thead>

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

@@ -36,6 +36,10 @@
         <% end %>
       </div>
     </div>
+    <div class="row mb-3">
+      <div class="col-md-3"><strong><%= t("raif.admin.common.duration") %>:</strong></div>
+      <div class="col-md-9"><%= @agent.runtime_duration %></div>
+    </div>
     <div class="row mb-3">
       <div class="col-md-3"><strong><%= t("raif.admin.common.iterations") %>:</strong></div>
       <div class="col-md-9"><%= @agent.iteration_count %> / <%= @agent.max_iterations %></div>
@@ -50,9 +54,50 @@
         <div class="col-md-9"><%= @agent.requested_language_key %></div>
       </div>
     <% end %>
+    <div class="row mb-3">
+      <div class="col-md-3"><strong><%= t("raif.admin.common.prompt_tokens") %>:</strong></div>
+      <div class="col-md-9">
+        <% if @agent.total_prompt_tokens > 0 %>
+          <%= number_with_delimiter(@agent.total_prompt_tokens) %>
+          <% if @agent.total_prompt_token_cost > 0 %>
+            <small>(<%= t("raif.admin.common.est_cost") %>: <%= "$" %><%= number_with_precision(@agent.total_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">
+        <% if @agent.total_completion_tokens > 0 %>
+          <%= number_with_delimiter(@agent.total_completion_tokens) %>
+          <% if @agent.total_output_token_cost > 0 %>
+            <small>(<%= t("raif.admin.common.est_cost") %>: <%= "$" %><%= number_with_precision(@agent.total_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">
+        <% if @agent.total_tokens_sum > 0 %>
+          <%= number_with_delimiter(@agent.total_tokens_sum) %>
+          <% if @agent.total_cost > 0 %>
+            <small>(<%= t("raif.admin.common.est_cost") %>: <%= "$" %><%= number_with_precision(@agent.total_cost, precision: 6) %>)</small>
+          <% end %>
+        <% else %>
+          -
+        <% end %>
+      </div>
+    </div>
   </div>
 </div>
 
+<%= render "raif/admin/model_tools/list", model_tools: @agent.available_model_tools_map %>
+
 <div class="card mb-4">
   <div class="card-header">
     <h5 class="mb-0"><%= t("raif.admin.common.task") %></h5>
@@ -91,7 +136,11 @@
       <div class="conversation-history p-3">
         <% @agent.conversation_history.each_with_index do |message, index| %>
           <%= render partial: "raif/admin/agents/conversation_message",
-                locals: { message: message, message_count: index + 1 } %>
+                locals: {
+                  message: message,
+                  message_count: index + 1,
+                  is_final_answer: message["type"] == "tool_call" && message["name"] == "agent_final_answer"
+                } %>
         <% end %>
       </div>
     </div>