raif 1.3.0 → 1.5.0

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

@@ -1,18 +1,23 @@
-<h1 class="my-4"><%= t("raif.admin.common.tasks") %></h1>
+<div class="d-flex justify-content-between align-items-center my-4">
+  <h1 class="mb-0"><%= t("raif.admin.common.tasks") %></h1>
+  <%= link_to raif.admin_stats_tasks_path, class: "btn btn-outline-primary" do %>
+    <%= t("raif.admin.common.stats") %> &raquo;
+  <% end %>
+</div>
 
 <div class="row">
   <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-4">
+        <div class="col-md-3">
           <div class="form-group">
             <label for="task_types"><%= t("raif.admin.common.type") %></label>
             <%= select_tag :task_types,
-                  options_for_select([["All", "all"]] + @task_types.map{|type| [type, type] }, @selected_type),
-                  { class: "form-select" } %>
+                  options_for_select([[t("raif.admin.common.all"), "all"]] + @task_types.map{|type| [type, type] }, @selected_type),
+                  { class: "form-select", data: { controller: "raif--tom-select" } } %>
           </div>
         </div>
-        <div class="col-md-4">
+        <div class="col-md-3">
           <div class="form-group">
             <label for="task_statuses"><%= t("raif.admin.common.status") %></label>
             <%= select_tag :task_statuses,
@@ -26,7 +31,18 @@
                     ],
                     @selected_statuses
                   ),
-                  { class: "form-select" } %>
+                  { 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">
@@ -46,6 +62,7 @@
               <th><%= t("raif.admin.common.creator") %></th>
               <th><%= t("raif.admin.common.model") %></th>
               <th><%= t("raif.admin.common.status") %></th>
+              <th><%= t("raif.admin.common.duration") %></th>
               <th><%= t("raif.admin.common.prompt") %></th>
             </tr>
           </thead>

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

@@ -68,12 +68,24 @@
         <% 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"><%= @task.runtime_duration %></div>
+    </div>
     <% if @task.requested_language_key.present? %>
       <div class="row mb-3">
         <div class="col-md-3"><strong><%= t("raif.admin.common.requested_language") %>:</strong></div>
         <div class="col-md-9"><%= @task.requested_language_key %></div>
       </div>
     <% end %>
+    <% if @task.run_with.present? %>
+      <div class="row mb-3">
+        <div class="col-md-3"><strong><%= t("raif.admin.common.run_with") %>:</strong></div>
+        <div class="col-md-9">
+          <pre class="pre-wrap mb-0"><%= JSON.pretty_generate(@task.run_with) %></pre>
+        </div>
+      </div>
+    <% end %>
   </div>
 </div>
 
@@ -93,22 +105,63 @@
       <div class="col-md-3"><strong><%= t("raif.admin.common.created_at") %>:</strong></div>
       <div class="col-md-9"><%= @task.raif_model_completion.created_at.rfc822 %></div>
     </div>
+    <div class="row mb-3">
+      <div class="col-md-3"><strong><%= t("raif.admin.common.retry_count") %>:</strong></div>
+      <div class="col-md-9"><%= @task.raif_model_completion.retry_count %></div>
+    </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"><%= @task.raif_model_completion.prompt_tokens ? number_with_delimiter(@task.raif_model_completion.prompt_tokens) : "-" %></div>
+      <div class="col-md-9">
+        <% if @task.raif_model_completion.prompt_tokens %>
+          <%= number_with_delimiter(@task.raif_model_completion.prompt_tokens) %>
+          <% if @task.raif_model_completion.prompt_token_cost %>
+            <small>(<%= t("raif.admin.common.est_cost") %>: <%= "$" %><%= number_with_precision(@task.raif_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"><%= @task.raif_model_completion.completion_tokens ? number_with_delimiter(@task.raif_model_completion.completion_tokens) : "-" %></div>
+      <div class="col-md-9">
+        <% if @task.raif_model_completion.completion_tokens %>
+          <%= number_with_delimiter(@task.raif_model_completion.completion_tokens) %>
+          <% if @task.raif_model_completion.output_token_cost %>
+            <small>(<%= t("raif.admin.common.est_cost") %>: <%= "$" %><%= number_with_precision(@task.raif_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"><%= @task.raif_model_completion.total_tokens ? number_with_delimiter(@task.raif_model_completion.total_tokens) : "-" %></div>
+      <div class="col-md-9">
+        <% if @task.raif_model_completion.total_tokens %>
+          <%= number_with_delimiter(@task.raif_model_completion.total_tokens) %>
+          <% if @task.raif_model_completion.total_cost %>
+            <small>(<%= t("raif.admin.common.est_cost") %>: <%= "$" %><%= number_with_precision(@task.raif_model_completion.total_cost, precision: 6) %>)</small>
+          <% end %>
+        <% else %>
+          -
+        <% end %>
+      </div>
+    </div>
+    <div class="row mb-3">
+      <div class="col-md-3"><strong><%= t("raif.admin.common.cache_read_input_tokens") %>:</strong></div>
+      <div class="col-md-9"><%= @task.raif_model_completion.cache_read_input_tokens ? number_with_delimiter(@task.raif_model_completion.cache_read_input_tokens) : "-" %></div>
+    </div>
+    <div class="row mb-3">
+      <div class="col-md-3"><strong><%= t("raif.admin.common.cache_creation_input_tokens") %>:</strong></div>
+      <div class="col-md-9"><%= @task.raif_model_completion.cache_creation_input_tokens ? number_with_delimiter(@task.raif_model_completion.cache_creation_input_tokens) : "-" %></div>
     </div>
   </div>
 </div>
 <% end %>
 
+<%= render "raif/admin/model_tools/list", model_tools: @task.available_model_tools_map %>
+
 <div class="card mb-4">
   <div class="card-header">
     <h5 class="mb-0"><%= t("raif.admin.common.system_prompt") %></h5>

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

@@ -1,5 +1,8 @@
+<%# locals: (conversation:, conversation_entry:) %>
+
 <%= form_with model: conversation_entry,
       url: raif.conversation_entries_path(conversation),
+      id: dom_id(conversation, "entry_form"),
       local: true do |f| %>
 
   <% if conversation_entry.raif_user_tool_invocation %>

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

@@ -6,13 +6,17 @@
 
     <% if content.present? %>
       <div class="mb-1 rounded-2 p-3 <%= message_type == :user ? "bg-primary text-white" : "border" %>">
-        <% case local_assigns[:conversation_entry]&.response_format %>
-        <% when "text" %>
-          <%= simple_format content %>
-        <% when "html" %>
-          <%= sanitize content %>
+        <% if message_type == :user %>
+          <%= simple_format h(content), {}, sanitize: false %>
         <% else %>
-          <%= content %>
+          <% case local_assigns[:conversation_entry]&.response_format %>
+          <% when "text" %>
+            <%= simple_format content %>
+          <% when "html" %>
+            <%= sanitize content %>
+          <% else %>
+            <%= content %>
+          <% end %>
         <% end %>
 
         <% if message_type == :model_response && local_assigns[:conversation_entry]&.citations.present? %>

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

@@ -0,0 +1,10 @@
+<%# locals: (conversation:) %>
+
+<tr>
+  <td class="ps-2"><%= conversation.created_at.strftime("%B %d, %Y") %></td>
+  <td>
+    <%= link_to t("raif.conversations.index.table.view"),
+          raif.conversation_path(conversation),
+          class: "btn btn-sm btn-primary" %>
+  </td>
+</tr>

data/app/views/raif/conversations/_entry_processed.turbo_stream.erb

@@ -0,0 +1,12 @@
+<%# locals: (conversation:, conversation_entry:) %>
+
+<%# Update the conversation entry now that it's been processed & has a model response %>
+<%= turbo_stream.replace conversation_entry do %>
+  <%= render conversation_entry %>
+<% end %>
+
+<%= turbo_stream.replace dom_id(conversation, "entry_form") do %>
+  <%= render "raif/conversation_entries/form", conversation: conversation, conversation_entry: Raif::ConversationEntry.new %>
+<% end %>
+
+<%= turbo_stream.action :raif_scroll_to_bottom, ActionView::RecordIdentifier.dom_id(conversation, :entries) %>

data/app/views/raif/conversations/index.html.erb

@@ -0,0 +1,23 @@
+<div class="conversation-history p-4">
+  <h2><%= t("raif.conversations.index.title") %></h2>
+
+  <% if @conversations.any? %>
+    <div class="table-responsive">
+      <table class="table table-striped">
+        <thead>
+          <tr>
+            <th><%= t("raif.conversations.index.table.started") %></th>
+            <th><%= t("raif.conversations.index.table.actions") %></th>
+          </tr>
+        </thead>
+        <tbody>
+          <%= render collection: @conversations, partial: "raif/conversations/conversation" %>
+        </tbody>
+      </table>
+    </div>
+  <% else %>
+    <div class="text-center py-4">
+      <p class="text-muted"><%= t("raif.conversations.index.no_conversations") %></p>
+    </div>
+  <% end %>
+</div>

data/config/importmap.rb

@@ -1,6 +1,14 @@
 # frozen_string_literal: true
 
 pin "raif", to: "raif.js"
+pin "raif_admin", to: "raif_admin.js"
 
 pin "raif/controllers/conversations_controller", to: "raif/controllers/conversations_controller.js"
+pin "raif/admin/judge_config_controller", to: "raif/admin/judge_config_controller.js"
+pin "raif/admin/select_all_checkboxes_controller", to: "raif/admin/select_all_checkboxes_controller.js"
+pin "raif/admin/cost_estimate_controller", to: "raif/admin/cost_estimate_controller.js"
+pin "raif/admin/tom_select_controller", to: "raif/admin/tom_select_controller.js"
+pin "raif/admin/table_search_controller", to: "raif/admin/table_search_controller.js"
+pin "raif/admin/sortable_table_controller", to: "raif/admin/sortable_table_controller.js"
+pin "raif/admin/copy_to_clipboard_controller", to: "raif/admin/copy_to_clipboard_controller.js"
 pin "raif/stream_actions/raif_scroll_to_bottom", to: "raif/stream_actions/raif_scroll_to_bottom.js"

data/config/locales/admin.en.yml

@@ -10,31 +10,46 @@ en:
         agents: Agents
         all: All
         arguments: Arguments
+        arguments_schema: Arguments Schema
         at: at
+        available_tools: Available Tools
+        cache_creation_input_tokens: Cache Creation Tokens
+        cache_read_input_tokens: Cache Read Tokens
         citations: Citations
         completed: Completed
         completed_at: Completed At
         completion_tokens: Completion Tokens
+        config: Config
         conversation_entries: Conversation Entries
         conversation_history: Conversation History
         conversations: Conversations
         count: Count
         created_at: Created At
         creator: Creator
+        default: default
+        description: Description
         details: Details
+        duration: Duration
         entries_count: Entries Count
         entry: Entry
         est_cost: est. cost
+        est_costs: est. costs
         failed: Failed
         failed_at: Failed At
+        failure_error: Failure Error
+        failure_reason: Failure Reason
         filter: Filter
         final_answer: Final Answer
         id: ID
         in_progress: In Progress
+        inferred_from_citations: Inferred from response citations.
         initial_task: Initial Task
+        input: Input
+        input_token_cost: Input Token Cost
         invalid_json: Invalid JSON
         iterations: Iterations
         language: Language
+        llms: LLMs
         messages: Messages
         model: Model
         model_completion: Model Completion
@@ -48,6 +63,9 @@ en:
         no_model_tool_invocations: No model tool invocations found.
         no_tasks: No tasks found.
         no_tool_calls: No tool calls
+        none: None
+        output: Output
+        output_token_cost: Output Token Cost
         pending: Pending
         period_all: All time
         period_day: Past 24 hours
@@ -56,16 +74,21 @@ en:
         prettified: Prettified
         prompt: Prompt
         prompt_tokens: Prompt Tokens
+        provider_managed_tool_calls: Provider-Managed Tool Calls
+        provider_tool_call_id: Provider Tool Call ID
         raw: Raw
         rendered: Rendered
         requested_language: Requested Language
         response: Response
+        response_array: Response Array
         response_format: Response Format
-        response_format_parameter: Response Format Parameter
         result: Result
+        retry_count: Retry Count
+        run_with: Run With
         running: Running
         since: Since
         source: Source
+        sources: Sources
         started_at: Started At
         stats: Stats
         status: Status
@@ -73,15 +96,31 @@ en:
         task: Task
         tasks: Tasks
         tool_arguments: Tool Arguments
+        tool_call: Tool Call
         tool_calls: Tool Calls
         tool_invocations: Tool Invocations
         tool_name: Tool Name
+        tool_result: Tool Result
         tool_type: Tool Type
+        total: Total
         total_cost: Total Cost
         total_tokens: Total Tokens
         type: Type
         update: Update
         user_message: User Message
+      config:
+        show:
+          configuration_settings: Configuration Settings
+          default: default
+          key: Key
+          no_embedding_models_registered: No embedding models registered
+          no_llms_registered: No LLMs registered
+          not_set: Not set
+          registered_embedding_models: Registered Embedding Models
+          registered_llms: Registered LLMs
+          setting: Setting
+          title: Raif Configuration
+          value: Value
       conversations:
         show:
           back_to_conversations: Back to Conversations
@@ -89,6 +128,18 @@ en:
       layouts:
         admin:
           title: Raif Admin
+      llms:
+        index:
+          api_name: API Name
+          cost_note: Costs shown are per 1,000,000 tokens.
+          default_badge: default
+          input_cost: Input Cost (per 1M tokens)
+          name: Name
+          no_llms: No LLMs are currently registered.
+          output_cost: Output Cost (per 1M tokens)
+          provider: Provider
+          search_placeholder: Search by name, provider, or API name...
+          title: Registered LLMs
       model_completions:
         show:
           back_to_model_completions: Back to Model Completions
@@ -96,10 +147,119 @@ en:
       model_tool_invocations:
         show:
           back_to_model_tool_invocations: Back to Model Tool Invocations
+          observation_disclaimer: This observation is regenerated from current data and code. It may differ from what was originally sent to the model.
+          observation_sent_to_model: Observation Sent to Model
+          observation_unavailable: 'Could not generate observation: %{error}'
           title: 'Model Tool Invocation #%{id}'
+      prompt_studio:
+        agents:
+          show:
+            agent_details: Agent Details
+            page_title: 'Prompt Studio - Agent #%{id}'
+        batch_runs:
+          create:
+            confirm: This will create a batch run and send requests to the selected LLM. Continue?
+            error: 'Batch run failed: %{message}'
+            new_batch_run: New Batch Run
+            no_tasks_selected: Please select at least one task.
+            submit: Run Selected Batch
+            tasks_selected: tasks selected
+            title: Batch Run
+          judge:
+            new_response: New Response
+            original_response: Original Response
+            tie: Tie
+            winner: 'Winner: %{name}'
+          judge_config:
+            comparison_criteria_label: Comparison Criteria
+            comparison_criteria_placeholder: Which response better addresses the prompt
+            criteria_label: Criteria
+            criteria_placeholder: Response is accurate and complete
+            description: Optionally evaluate each result with an LLM judge.
+            include_original_prompt_as_context_description: Provides the original task prompt to the judge as additional context for evaluation.
+            include_original_prompt_as_context_label: Include original prompt as context
+            judge_model_label: Judge Model
+            rubric_accuracy: Accuracy
+            rubric_clarity: Clarity
+            rubric_helpfulness: Helpfulness
+            rubric_label: Scoring Rubric
+            strict_mode_label: Strict Mode
+            summarization_description: Uses the original task prompt as the source content and the new response as the summary to evaluate.
+            title: LLM Judge Configuration
+            type_binary: Binary (Pass/Fail)
+            type_comparative: Comparative (A vs B)
+            type_label: Judge Type
+            type_none: None
+            type_scored: Scored (Rubric)
+            type_summarization: Summarization
+          show:
+            average_score: Average Score
+            judge_reasoning: Judge Reasoning
+            judge_result: Judge Result
+            new_response: New Response
+            original_response: Original Response
+            page_title: 'Batch Run #%{id}'
+            pass_rate: Pass Rate
+            progress: Progress
+            results: Results
+            source_task: Source Task
+            summary: Summary
+          status:
+            completed: Completed
+            failed: Failed
+            in_progress: In Progress
+            judging: Judging
+            pending: Pending
+            running: Running
+        common:
+          back: Back
+          created_after: Created After
+          created_before: Created Before
+          current_prompt: Current Prompt
+          current_system_prompt: Current System Prompt
+          estimated_cost: "~%{cost} (estimate)"
+          estimated_tokens: "~%{count} tokens (estimate)"
+          no_instances: No instances found.
+          original_prompt: Original Prompt
+          original_system_prompt: Original System Prompt
+          prompt_changed: Prompt Changed
+          prompt_studio: Prompt Studio
+          prompt_tokens: "%{count} tokens"
+          prompt_unchanged: Prompt Unchanged
+          run_in_prompt_studio: Run in Prompt Studio
+          runs_disabled: Prompt Studio runs are not enabled in this environment.
+          select_type: Select a type to browse instances
+          system_prompt_changed: System Prompt Changed
+          system_prompt_unchanged: System Prompt Unchanged
+          type: Type
+          warning_stale_reference: 'Warning: Some run_with references could not be loaded. The current prompt may differ from what would be generated with the original data.'
+        conversations:
+          show:
+            conversation_details: Conversation Details
+            page_title: 'Prompt Studio - Conversation #%{id}'
+        tasks:
+          rerun:
+            confirm: This will create a new task and send a request to the selected LLM. Continue?
+            description: Re-run this task with the current prompt templates and a selected model. This will create a new task record.
+            error: 'Re-run failed: %{message}'
+            invalid_model: Please select a valid model.
+            running: Running...
+            submit: Re-run Task
+            title: Re-run with Current Prompts
+          show:
+            new_response: New Response
+            original_response: Original Response
+            page_title: 'Prompt Studio - Task #%{id}'
+            response_comparison: Response Comparison
+            task_details: Task Details
+            view_response: View Response
       stats:
+        model_tool_invocations:
+          back_to_stats: Back to Stats
+          title: Model Tool Invocation Stats
         tasks:
           back_to_stats: Back to Stats
+          show_model_breakdown: Show model breakdown
           title: Task Stats
       tasks:
         show:

data/config/locales/en.yml

@@ -6,10 +6,7 @@ en:
         errors:
           available_model_tools:
             too_short: must have at least 1 tool in addition to the agent_final_answer tool
-      re_act_agent:
-        errors:
-          available_model_tools:
-            too_short: must have at least 1 tool
+        final_answer_warning: 'Warning: This is your final iteration. You must provide your final answer using the agent_final_answer tool.'
     common:
       send: Send
       sources: Sources
@@ -18,8 +15,17 @@ en:
       type_your_message: Type your message...
     conversation:
       initial_chat_message: Hello, how can I help you today?
+    conversations:
+      index:
+        no_conversations: No conversations found.
+        table:
+          actions: Actions
+          started: Started
+          view: View
+        title: Past Conversations
     embedding_model_names:
       bedrock_titan_embed_text_v2: AWS Bedrock Titan Text Embeddings v2
+      google_gemini_embedding_2: Google Gemini Embedding 2
       open_ai_text_embedding_3_large: OpenAI Text Embedding 3 Large
       open_ai_text_embedding_3_small: OpenAI Text Embedding 3 Small
       open_ai_text_embedding_ada_002: OpenAI Text Embedding Ada 002
@@ -51,6 +57,13 @@ en:
       anthropic_claude_3_5_sonnet: Anthropic Claude 3.5 Sonnet
       anthropic_claude_3_7_sonnet: Anthropic Claude 3.7 Sonnet
       anthropic_claude_3_opus: Anthropic Claude 3 Opus
+      anthropic_claude_4_1_opus: Anthropic Claude 4.1 Opus
+      anthropic_claude_4_5_haiku: Anthropic Claude 4.5 Haiku
+      anthropic_claude_4_5_opus: Anthropic Claude 4.5 Opus
+      anthropic_claude_4_5_sonnet: Anthropic Claude 4.5 Sonnet
+      anthropic_claude_4_6_opus: Anthropic Claude 4.6 Opus
+      anthropic_claude_4_6_sonnet: Anthropic Claude 4.6 Sonnet
+      anthropic_claude_4_7_opus: Anthropic Claude 4.7 Opus
       anthropic_claude_4_opus: Anthropic Claude 4 Opus
       anthropic_claude_4_sonnet: Anthropic Claude 4 Sonnet
       bedrock_amazon_nova_lite: Amazon Nova Lite (via AWS Bedrock)
@@ -60,8 +73,25 @@ en:
       bedrock_claude_3_5_sonnet: Anthropic Claude 3.5 Sonnet (via AWS Bedrock)
       bedrock_claude_3_7_sonnet: Anthropic Claude 3.7 Sonnet (via AWS Bedrock)
       bedrock_claude_3_opus: Anthropic Claude 3 Opus (via AWS Bedrock)
+      bedrock_claude_4_1_opus: Claude 4.1 Opus (via AWS Bedrock)
+      bedrock_claude_4_5_haiku: Anthropic Claude 4.5 Haiku (via AWS Bedrock)
+      bedrock_claude_4_5_opus: Anthropic Claude 4.5 Opus (via AWS Bedrock)
+      bedrock_claude_4_5_sonnet: Anthropic Claude 4.5 Sonnet (via AWS Bedrock)
+      bedrock_claude_4_6_opus: Anthropic Claude 4.6 Opus (via AWS Bedrock)
+      bedrock_claude_4_6_sonnet: Anthropic Claude 4.6 Sonnet (via AWS Bedrock)
+      bedrock_claude_4_7_opus: Anthropic Claude 4.7 Opus (via AWS Bedrock)
       bedrock_claude_4_opus: Anthropic Claude 4 Opus (via AWS Bedrock)
       bedrock_claude_4_sonnet: Anthropic Claude 4 Sonnet (via AWS Bedrock)
+      bedrock_deepseek_r1: DeepSeek R1 (via AWS Bedrock)
+      bedrock_deepseek_v3_2: DeepSeek v3.2 (via AWS Bedrock)
+      bedrock_gpt_oss_120b: OpenAI GPT-OSS 120B (via AWS Bedrock)
+      bedrock_gpt_oss_20b: OpenAI GPT-OSS 20B (via AWS Bedrock)
+      google_gemini_2_5_flash: Google Gemini 2.5 Flash
+      google_gemini_2_5_pro: Google Gemini 2.5 Pro
+      google_gemini_3_0_flash: Google Gemini 3 Flash
+      google_gemini_3_0_pro: Google Gemini 3 Pro
+      google_gemini_3_1_flash_lite: Google Gemini 3.1 Flash-Lite
+      google_gemini_3_1_pro: Google Gemini 3.1 Pro
       open_ai_gpt_3_5_turbo: OpenAI GPT-3.5 Turbo
       open_ai_gpt_4_1: OpenAI GPT-4.1
       open_ai_gpt_4_1_mini: OpenAI GPT-4.1 Mini
@@ -69,6 +99,12 @@ en:
       open_ai_gpt_4o: OpenAI GPT-4o
       open_ai_gpt_4o_mini: OpenAI GPT-4o Mini
       open_ai_gpt_5: OpenAI GPT-5
+      open_ai_gpt_5_1: OpenAI GPT-5.1
+      open_ai_gpt_5_2: OpenAI GPT-5.2
+      open_ai_gpt_5_3: OpenAI GPT-5.3
+      open_ai_gpt_5_4: OpenAI GPT-5.4
+      open_ai_gpt_5_4_mini: OpenAI GPT-5.4 Mini
+      open_ai_gpt_5_4_nano: OpenAI GPT-5.4 Nano
       open_ai_gpt_5_mini: OpenAI GPT-5 Mini
       open_ai_gpt_5_nano: OpenAI GPT-5 Nano
       open_ai_o1: OpenAI o1
@@ -83,8 +119,17 @@ en:
       open_ai_responses_gpt_4o: OpenAI GPT-4o (Responses API)
       open_ai_responses_gpt_4o_mini: OpenAI GPT-4o Mini (Responses API)
       open_ai_responses_gpt_5: OpenAI GPT-5 (Responses API)
+      open_ai_responses_gpt_5_1: OpenAI GPT-5.1 (Responses API)
+      open_ai_responses_gpt_5_2: OpenAI GPT-5.2 (Responses API)
+      open_ai_responses_gpt_5_2_pro: OpenAI GPT-5.2 Pro (Responses API)
+      open_ai_responses_gpt_5_3: OpenAI GPT-5.3 (Responses API)
+      open_ai_responses_gpt_5_4: OpenAI GPT-5.4 (Responses API)
+      open_ai_responses_gpt_5_4_mini: OpenAI GPT-5.4 Mini (Responses API)
+      open_ai_responses_gpt_5_4_nano: OpenAI GPT-5.4 Nano (Responses API)
+      open_ai_responses_gpt_5_4_pro: OpenAI GPT-5.4 Pro (Responses API)
       open_ai_responses_gpt_5_mini: OpenAI GPT-5 Mini (Responses API)
       open_ai_responses_gpt_5_nano: OpenAI GPT-5 Nano (Responses API)
+      open_ai_responses_gpt_5_pro: OpenAI GPT-5 Pro (Responses API)
       open_ai_responses_o1: OpenAI o1 (Responses API)
       open_ai_responses_o1_mini: OpenAI o1 Mini (Responses API)
       open_ai_responses_o1_pro: OpenAI o1 Pro (Responses API)
@@ -94,11 +139,29 @@ en:
       open_ai_responses_o4_mini: OpenAI o4 Mini (Responses API)
       open_router_claude_3_7_sonnet: Anthropic Claude 3.7 Sonnet (via OpenRouter)
       open_router_deepseek_chat_v3: DeepSeek Chat v3 (via OpenRouter)
+      open_router_deepseek_v3_1: DeepSeek v3.1 (via OpenRouter)
+      open_router_deepseek_v3_2: DeepSeek v3.2 (via OpenRouter)
       open_router_gemini_2_0_flash: Google Gemini 2.0 Flash (via OpenRouter)
+      open_router_gemini_2_5_flash: Gemini 2.5 Flash (via OpenRouter)
+      open_router_gemini_2_5_pro: Gemini 2.5 Pro (via OpenRouter)
+      open_router_gemini_3_1_flash_lite_preview: Gemini 3.1 Flash-Lite Preview (via OpenRouter)
+      open_router_gemini_3_1_pro_preview: Gemini 3.1 Pro Preview (via OpenRouter)
+      open_router_gemini_3_pro_preview: Gemini 3 Pro Preview (via OpenRouter)
+      open_router_google_gemma_4_31b_it: Google Gemma 4 31B IT (via OpenRouter)
+      open_router_grok_4: Grok 4 (via OpenRouter)
+      open_router_grok_4_1_fast: Grok 4.1 Fast (via OpenRouter)
+      open_router_grok_4_20: Grok 4.20 (via OpenRouter)
+      open_router_kimi_k2_5: Kimi K2.5 (via OpenRouter)
+      open_router_kimi_k2_thinking: Kimi K2 Thinking (via OpenRouter)
       open_router_llama_3_1_8b_instruct: Meta Llama 3.1 8B Instruct (via OpenRouter)
       open_router_llama_3_3_70b_instruct: Meta Llama 3.3 70B Instruct (via OpenRouter)
       open_router_llama_4_maverick: Meta Llama 4 Maverick (via OpenRouter)
       open_router_llama_4_scout: Meta Llama 4 Scout (via OpenRouter)
+      open_router_minimax_m2: Minimax M2 (via OpenRouter)
+      open_router_minimax_m2_1: Minimax M2.1 (via OpenRouter)
+      open_router_minimax_m2_5: Minimax M2.5 (via OpenRouter)
+      open_router_mistral_large_3_2512: Mistral Large 3 (via OpenRouter)
+      open_router_mistral_small_3_2_24b: Mistral Small 3.2 24B (via OpenRouter)
       open_router_open_ai_gpt_oss_120b: OpenAI GPT-OSS 120B (via OpenRouter)
       open_router_open_ai_gpt_oss_20b: OpenAI GPT-OSS 20B (via OpenRouter)
       raif_test_llm: Raif Test LLM

data/config/routes.rb

@@ -17,6 +17,7 @@ Raif::Engine.routes.draw do
 
     namespace :stats do
       resources :tasks, only: [:index]
+      resources :model_tool_invocations, only: [:index]
     end
 
     resources :tasks, only: [:index, :show]
@@ -24,5 +25,14 @@ Raif::Engine.routes.draw do
     resources :model_completions, only: [:index, :show]
     resources :agents, only: [:index, :show]
     resources :model_tool_invocations, only: [:index, :show]
+    resources :llms, only: [:index]
+    resource :config, only: [:show]
+
+    namespace :prompt_studio do
+      resources :tasks, only: [:index, :show, :create]
+      resources :conversations, only: [:index, :show]
+      resources :agents, only: [:index, :show]
+      resources :batch_runs, only: [:create, :show]
+    end
   end
 end

data/db/migrate/20250904194456_add_generating_entry_response_to_raif_conversations.rb

@@ -0,0 +1,7 @@
+# frozen_string_literal: true
+
+class AddGeneratingEntryResponseToRaifConversations < ActiveRecord::Migration[7.1]
+  def change
+    add_column :raif_conversations, :generating_entry_response, :boolean, default: false, null: false
+  end
+end

data/db/migrate/20250911125234_add_source_to_raif_tasks.rb

@@ -0,0 +1,7 @@
+# frozen_string_literal: true
+
+class AddSourceToRaifTasks < ActiveRecord::Migration[7.1]
+  def change
+    add_reference :raif_tasks, :source, polymorphic: true, index: true
+  end
+end