activeagent 1.0.3 → 1.2.0

data/lib/active_agent/telemetry/tracer.rb

@@ -2,18 +2,17 @@
 
 module ActiveAgent
   module Telemetry
-    # Manages trace creation and lifecycle.
+    # Manages trace creation and lifecycle on the activeagents-telemetry
+    # core: builds an ActiveAgents::Telemetry::Trace per unit of work, tracks
+    # the current span in thread-local storage, and hands finished traces to
+    # the Reporter.
     #
-    # The Tracer creates traces, manages the current trace context,
-    # and coordinates with the Reporter for async transmission.
-    #
-    # @example Basic usage
+    # @example
     #   tracer = Tracer.new(configuration)
     #   tracer.trace("MyAgent.greet") do |span|
     #     span.set_attribute("user_id", 123)
     #     span.add_span("llm.generate", span_type: :llm)
     #   end
-    #
     class Tracer
       # @return [Configuration] Telemetry configuration
       attr_reader :configuration
@@ -27,44 +26,47 @@ module ActiveAgent
       def initialize(configuration)
         @configuration = configuration
         @reporter = Reporter.new(configuration)
-        @mutex = Mutex.new
       end
 
       # Creates and executes a new trace.
       #
+      # Callers may supply the trace id (instrumentation passes the
+      # generation's prompt_options[:trace_id]) so external records — e.g.
+      # solid_agent's persisted generations — can correlate with this trace;
+      # otherwise one is generated.
+      #
       # @param name [String] Trace name (typically "AgentClass.action")
-      # @param attributes [Hash] Root span attributes
+      # @param attributes [Hash] Root span attributes; :trace_id and
+      #   :span_type are extracted rather than recorded
       # @yield [span] Yields the root span for adding child spans
       # @return [Object] Result of the block
-      #
-      # @example
-      #   tracer.trace("WeatherAgent.forecast") do |span|
-      #     span.set_attribute("location", "Seattle")
-      #     result = do_llm_call
-      #     span.set_tokens(input: 100, output: 50)
-      #     result
-      #   end
-      def trace(name, **attributes, &block)
+      def trace(name, **attributes)
         return yield(Telemetry::NullSpan.new) unless should_trace?
 
-        trace_id = generate_trace_id
+        trace_id = attributes.delete(:trace_id) || SecureRandom.hex(16)
+        span_type = attributes.delete(:span_type) || :root
+
+        trace = build_trace(trace_id)
         root_span = Span.new(
           name,
           trace_id: trace_id,
-          span_type: :root,
+          span_type: span_type,
           **default_attributes.merge(attributes)
         )
+        trace.add_span(root_span)
 
         with_span(root_span) do
           result = yield(root_span)
           root_span.finish
-          report_trace(root_span)
+          reporter.report(redact_trace!(trace))
           result
         end
       rescue StandardError => e
-        root_span&.record_error(e)
-        root_span&.finish
-        report_trace(root_span) if root_span
+        if root_span && !root_span.finished?
+          root_span.record_error(e)
+          root_span.finish
+          reporter.report(redact_trace!(trace))
+        end
         raise
       end
 
@@ -76,11 +78,12 @@ module ActiveAgent
       def span(name, **attributes)
         return Telemetry::NullSpan.new unless should_trace?
 
+        span_type = attributes.delete(:span_type) || :root
         current = current_span
         if current
-          current.add_span(name, **attributes)
+          current.add_span(name, span_type: span_type, **attributes)
         else
-          Span.new(name, trace_id: generate_trace_id, **default_attributes.merge(attributes))
+          Span.new(name, trace_id: SecureRandom.hex(16), span_type: span_type, **default_attributes.merge(attributes))
         end
       end
 
@@ -91,7 +94,9 @@ module ActiveAgent
         Thread.current[CURRENT_SPAN_KEY]
       end
 
-      # Flushes buffered traces immediately.
+      # Flushes buffered traces and waits for delivery to complete, so
+      # callers (tests, rails runner, job shutdown) can rely on the traces
+      # having been delivered/stored when this returns.
       #
       # @return [void]
       def flush
@@ -107,11 +112,16 @@ module ActiveAgent
 
       private
 
+      def build_trace(trace_id)
+        ActiveAgents::Telemetry::Trace.new(
+          trace_id: trace_id,
+          service_name: configuration.resolved_service_name,
+          environment: configuration.resolved_environment,
+          resource_attributes: configuration.resource_attributes
+        )
+      end
+
       # Executes block with span as current context.
-      #
-      # @param span [Span] Span to set as current
-      # @yield Block to execute
-      # @return [Object] Result of block
       def with_span(span)
         previous = Thread.current[CURRENT_SPAN_KEY]
         Thread.current[CURRENT_SPAN_KEY] = span
@@ -120,61 +130,48 @@ module ActiveAgent
         Thread.current[CURRENT_SPAN_KEY] = previous
       end
 
-      # Reports a completed trace to the reporter.
-      #
-      # @param span [Span] Root span of the trace
-      def report_trace(span)
-        reporter.report(build_trace_payload(span))
+      # Redacts span (and span-event) attributes whose keys match any
+      # configured redact_attributes entry, before the trace is handed to
+      # the reporter. Matching is case-insensitive substring — deliberately
+      # over-broad: better to redact a harmless "max_tokens" than to ship
+      # an "api_key". (Ported from the pre-shared-core payload builder,
+      # which no longer exists — the gem's reporter ships the trace as-is.)
+      REDACTED = "[REDACTED]"
+
+      def redact_trace!(trace)
+        patterns = Array(configuration.redact_attributes).map(&:to_s).reject(&:empty?)
+        return trace if patterns.empty?
+
+        matcher = Regexp.union(patterns.map { |pattern| Regexp.new(Regexp.escape(pattern), Regexp::IGNORECASE) })
+        Array(trace.spans).each { |span| redact_span!(span, matcher) }
+        trace
       end
 
-      # Builds the trace payload for transmission.
-      #
-      # @param root_span [Span] Root span
-      # @return [Hash] Trace payload
-      def build_trace_payload(root_span)
-        {
-          trace_id: root_span.trace_id,
-          service_name: configuration.resolved_service_name,
-          environment: configuration.environment,
-          timestamp: Time.current.iso8601(6),
-          resource_attributes: configuration.resource_attributes,
-          spans: flatten_spans(root_span)
-        }
+      def redact_span!(span, matcher)
+        redact_hash!(span.attributes, matcher) if span.attributes.is_a?(Hash)
+        Array(span.events).each do |event|
+          attributes = event.is_a?(Hash) ? (event["attributes"] || event[:attributes]) : nil
+          redact_hash!(attributes, matcher) if attributes.is_a?(Hash)
+        end
+        Array(span.children).each { |child| redact_span!(child, matcher) }
       end
 
-      # Flattens span hierarchy into array.
-      #
-      # @param span [Span] Root span
-      # @return [Array<Hash>] Flattened span data
-      def flatten_spans(span)
-        result = [ span.to_h.except(:children) ]
-        span.children.each do |child|
-          result.concat(flatten_spans(child))
+      def redact_hash!(attributes, matcher)
+        attributes.each_key do |key|
+          attributes[key] = REDACTED if key.to_s.match?(matcher)
         end
-        result
       end
 
-      # Returns whether this trace should be sampled.
-      #
-      # @return [Boolean]
+      # Returns whether this trace should be collected at all.
       def should_trace?
         configuration.enabled? && configuration.configured? && configuration.should_sample?
       end
 
-      # Generates a unique trace ID.
-      #
-      # @return [String] 32-character hex trace ID
-      def generate_trace_id
-        SecureRandom.hex(16)
-      end
-
       # Returns default attributes for all spans.
-      #
-      # @return [Hash] Default attributes
       def default_attributes
         {
           "service.name" => configuration.resolved_service_name,
-          "service.environment" => configuration.environment,
+          "service.environment" => configuration.resolved_environment,
           "telemetry.sdk.name" => "activeagent",
           "telemetry.sdk.version" => ActiveAgent::VERSION
         }

data/lib/active_agent/telemetry.rb

@@ -1,8 +1,7 @@
 # frozen_string_literal: true
 
-require "net/http"
-require "json"
 require "securerandom"
+require "activeagents/telemetry"
 
 module ActiveAgent
   # Telemetry module for collecting and reporting agent traces.
@@ -55,6 +54,7 @@ module ActiveAgent
     autoload :Span
     autoload :Reporter
     autoload :Instrumentation
+    autoload :ToolOrigin
 
     class << self
       # Returns the telemetry configuration instance.

data/lib/active_agent/version.rb

@@ -1,3 +1,3 @@
 module ActiveAgent
-  VERSION = "1.0.3"
+  VERSION = "1.2.0"
 end

data/lib/active_agent.rb

@@ -101,13 +101,13 @@ module ActiveAgent
   autoload :Preview, "active_agent/concerns/preview"
   autoload :Previews, "active_agent/concerns/preview"
   autoload :GenerationJob
+  autoload :ModelCapabilities
   autoload :Observers, "active_agent/concerns/observers"
   autoload :Provider, "active_agent/concerns/provider"
   autoload :Rescue, "active_agent/concerns/rescue"
   autoload :Tooling, "active_agent/concerns/tooling"
   autoload :View, "active_agent/concerns/view"
   autoload :Telemetry
-  autoload :Dashboard
 
   class << self
     # Eagerly loads all ActiveAgent components and descendant agent classes.

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: activeagent
 version: !ruby/object:Gem::Version
-  version: 1.0.3
+  version: 1.2.0
 platform: ruby
 authors:
 - Justin Bowen
@@ -109,6 +109,40 @@ dependencies:
     - - "<="
       - !ruby/object:Gem::Version
         version: '9.0'
+- !ruby/object:Gem::Dependency
+  name: railties
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '7.2'
+    - - "<="
+      - !ruby/object:Gem::Version
+        version: '9.0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '7.2'
+    - - "<="
+      - !ruby/object:Gem::Version
+        version: '9.0'
+- !ruby/object:Gem::Dependency
+  name: activeagents-telemetry
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.1'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.1'
 - !ruby/object:Gem::Dependency
   name: jbuilder
   requirement: !ruby/object:Gem::Requirement
@@ -386,38 +420,11 @@ files:
 - lib/active_agent/concerns/tooling.rb
 - lib/active_agent/concerns/view.rb
 - lib/active_agent/configuration.rb
-- lib/active_agent/dashboard.rb
-- lib/active_agent/dashboard/app/controllers/active_agent/dashboard/api/traces_controller.rb
-- lib/active_agent/dashboard/app/controllers/active_agent/dashboard/application_controller.rb
-- lib/active_agent/dashboard/app/controllers/active_agent/dashboard/dashboard_controller.rb
-- lib/active_agent/dashboard/app/controllers/active_agent/dashboard/traces_controller.rb
-- lib/active_agent/dashboard/app/jobs/active_agent/dashboard/agent_execution_job.rb
-- lib/active_agent/dashboard/app/jobs/active_agent/dashboard/application_job.rb
-- lib/active_agent/dashboard/app/jobs/active_agent/dashboard/sandbox_cleanup_job.rb
-- lib/active_agent/dashboard/app/jobs/active_agent/dashboard/sandbox_provision_job.rb
-- lib/active_agent/dashboard/app/jobs/active_agent/process_telemetry_traces_job.rb
-- lib/active_agent/dashboard/app/models/active_agent/dashboard/agent.rb
-- lib/active_agent/dashboard/app/models/active_agent/dashboard/agent_run.rb
-- lib/active_agent/dashboard/app/models/active_agent/dashboard/agent_template.rb
-- lib/active_agent/dashboard/app/models/active_agent/dashboard/agent_version.rb
-- lib/active_agent/dashboard/app/models/active_agent/dashboard/application_record.rb
-- lib/active_agent/dashboard/app/models/active_agent/dashboard/recording_action.rb
-- lib/active_agent/dashboard/app/models/active_agent/dashboard/recording_snapshot.rb
-- lib/active_agent/dashboard/app/models/active_agent/dashboard/sandbox_run.rb
-- lib/active_agent/dashboard/app/models/active_agent/dashboard/sandbox_session.rb
-- lib/active_agent/dashboard/app/models/active_agent/dashboard/session_recording.rb
-- lib/active_agent/dashboard/app/models/active_agent/telemetry_trace.rb
-- lib/active_agent/dashboard/app/views/active_agent/dashboard/traces/_trace_detail.html.erb
-- lib/active_agent/dashboard/app/views/active_agent/dashboard/traces/index.html.erb
-- lib/active_agent/dashboard/app/views/active_agent/dashboard/traces/metrics.html.erb
-- lib/active_agent/dashboard/app/views/active_agent/dashboard/traces/show.html.erb
-- lib/active_agent/dashboard/app/views/layouts/active_agent/dashboard/application.html.erb
-- lib/active_agent/dashboard/config/routes.rb
-- lib/active_agent/dashboard/engine.rb
 - lib/active_agent/deprecator.rb
 - lib/active_agent/generation.rb
 - lib/active_agent/generation_job.rb
 - lib/active_agent/inline_preview_interceptor.rb
+- lib/active_agent/model_capabilities.rb
 - lib/active_agent/providers/_base_provider.rb
 - lib/active_agent/providers/anthropic/_types.rb
 - lib/active_agent/providers/anthropic/options.rb
@@ -453,6 +460,7 @@ files:
 - lib/active_agent/providers/concerns/instrumentation.rb
 - lib/active_agent/providers/concerns/previewable.rb
 - lib/active_agent/providers/concerns/tool_choice_clearing.rb
+- lib/active_agent/providers/errors.rb
 - lib/active_agent/providers/gemini/_types.rb
 - lib/active_agent/providers/gemini/options.rb
 - lib/active_agent/providers/gemini_provider.rb
@@ -535,24 +543,12 @@ files:
 - lib/active_agent/telemetry/instrumentation.rb
 - lib/active_agent/telemetry/reporter.rb
 - lib/active_agent/telemetry/span.rb
+- lib/active_agent/telemetry/tool_origin.rb
 - lib/active_agent/telemetry/tracer.rb
 - lib/active_agent/version.rb
 - lib/activeagent.rb
 - lib/generators/active_agent/agent/USAGE
 - lib/generators/active_agent/agent/agent_generator.rb
-- lib/generators/active_agent/dashboard/install/install_generator.rb
-- lib/generators/active_agent/dashboard/install/templates/initializer.rb
-- lib/generators/active_agent/dashboard/install/templates/migrations/create_active_agent_agent_runs.rb
-- lib/generators/active_agent/dashboard/install/templates/migrations/create_active_agent_agent_templates.rb
-- lib/generators/active_agent/dashboard/install/templates/migrations/create_active_agent_agent_versions.rb
-- lib/generators/active_agent/dashboard/install/templates/migrations/create_active_agent_agents.rb
-- lib/generators/active_agent/dashboard/install/templates/migrations/create_active_agent_sandbox_runs.rb
-- lib/generators/active_agent/dashboard/install/templates/migrations/create_active_agent_sandbox_sessions.rb
-- lib/generators/active_agent/dashboard/install/templates/migrations/create_active_agent_session_recordings.rb
-- lib/generators/active_agent/dashboard/install/templates/migrations/create_active_agent_telemetry_traces.rb
-- lib/generators/active_agent/dashboard/install_generator.rb
-- lib/generators/active_agent/dashboard/templates/active_agent_dashboard.rb.erb
-- lib/generators/active_agent/dashboard/templates/create_active_agent_telemetry_traces.rb.erb
 - lib/generators/active_agent/install/USAGE
 - lib/generators/active_agent/install/install_generator.rb
 - lib/generators/active_agent/templates/active_agent.yml

data/lib/active_agent/dashboard/app/controllers/active_agent/dashboard/api/traces_controller.rb

@@ -1,117 +0,0 @@
-# frozen_string_literal: true
-
-module ActiveAgent
-  module Dashboard
-    module Api
-      # Telemetry ingestion endpoint.
-      #
-      # Receives traces from ActiveAgent::Telemetry::Reporter and stores them
-      # for analysis and visualization in the dashboard.
-      #
-      # Supports two modes:
-      # - Local mode: No authentication, synchronous processing
-      # - Multi-tenant mode: Bearer token auth, async processing via job
-      #
-      # @example Local mode request
-      #   POST /active_agent/api/traces
-      #   Content-Type: application/json
-      #
-      #   {
-      #     "traces": [...],
-      #     "sdk": { "name": "activeagent", "version": "0.5.0" }
-      #   }
-      #
-      # @example Multi-tenant mode request
-      #   POST /active_agent/api/traces
-      #   Authorization: Bearer <api_key>
-      #   Content-Type: application/json
-      #
-      #   {
-      #     "traces": [...],
-      #     "sdk": { "name": "activeagent", "version": "0.5.0" }
-      #   }
-      #
-      class TracesController < ActionController::API
-        before_action :authenticate_api_key!, if: -> { ActiveAgent::Dashboard.multi_tenant? }
-
-        # POST /active_agent/api/traces
-        def create
-          traces = params[:traces] || []
-          sdk_info = params[:sdk] || {}
-
-          return head :accepted if traces.empty?
-
-          if ActiveAgent::Dashboard.multi_tenant?
-            # Multi-tenant mode: process in background
-            ActiveAgent::ProcessTelemetryTracesJob.perform_later(
-              account_id: @account&.id,
-              traces: traces.as_json,
-              sdk_info: sdk_info.as_json,
-              received_at: Time.current.iso8601(6)
-            )
-          else
-            # Local mode: process synchronously
-            process_traces_synchronously(traces, sdk_info)
-          end
-
-          head :accepted
-        rescue ActionController::ParameterMissing => e
-          render json: { error: e.message }, status: :bad_request
-        rescue StandardError => e
-          Rails.logger.error("[ActiveAgent::Dashboard] Trace ingestion error: #{e.message}")
-          render json: { error: "Internal server error" }, status: :internal_server_error
-        end
-
-        private
-
-        # Authenticates the request using Bearer token from Authorization header.
-        # Only used in multi-tenant mode.
-        def authenticate_api_key!
-          token = extract_bearer_token
-
-          if token.blank?
-            render json: { error: "Missing Authorization header" }, status: :unauthorized
-            return
-          end
-
-          account_class = ActiveAgent::Dashboard.account_class.constantize
-          @account = account_class.find_by(telemetry_api_key: token)
-
-          if @account.nil?
-            render json: { error: "Invalid API key" }, status: :unauthorized
-            return
-          end
-
-          # Track usage for rate limiting (if the account responds to it)
-          @account.increment_telemetry_usage! if @account.respond_to?(:increment_telemetry_usage!)
-        end
-
-        # Extracts Bearer token from Authorization header.
-        def extract_bearer_token
-          auth_header = request.headers["Authorization"]
-          return nil if auth_header.blank?
-
-          match = auth_header.match(/^Bearer\s+(.+)$/i)
-          match[1] if match
-        end
-
-        # Process traces synchronously for local development.
-        def process_traces_synchronously(traces, sdk_info)
-          model = ActiveAgent::Dashboard.trace_model
-
-          traces.each do |trace|
-            # Skip if trace already exists (idempotency)
-            next if model.exists?(trace_id: trace["trace_id"])
-
-            model.create_from_payload(trace, sdk_info)
-          rescue StandardError => e
-            Rails.logger.error(
-              "[ActiveAgent::Dashboard] Failed to process trace #{trace['trace_id']}: " \
-              "#{e.class} - #{e.message}"
-            )
-          end
-        end
-      end
-    end
-  end
-end

data/lib/active_agent/dashboard/app/controllers/active_agent/dashboard/application_controller.rb

@@ -1,54 +0,0 @@
-# frozen_string_literal: true
-
-module ActiveAgent
-  module Dashboard
-    # Base controller for the ActiveAgent Dashboard.
-    #
-    # Handles authentication and provides helper methods for multi-tenant mode.
-    class ApplicationController < ActionController::Base
-      protect_from_forgery with: :exception
-
-      before_action :authenticate_dashboard!
-
-      # Add the engine's view path
-      prepend_view_path ActiveAgent::Dashboard::Engine.dashboard_root.join("app", "views").to_s
-
-      # Use custom layout if configured, otherwise use engine layout
-      layout -> { ActiveAgent::Dashboard.layout || "active_agent/dashboard/application" }
-
-      helper_method :current_user, :current_owner
-
-      private
-
-      def authenticate_dashboard!
-        return if ActiveAgent::Dashboard.authentication_method.nil?
-
-        result = ActiveAgent::Dashboard.authentication_method.call(self)
-        head :unauthorized unless result
-      rescue StandardError => e
-        Rails.logger.error("[ActiveAgent::Dashboard] Authentication error: #{e.message}")
-        head :unauthorized
-      end
-
-      # Returns the current user from the host application.
-      def current_user
-        return nil unless ActiveAgent::Dashboard.current_user_method
-
-        send(ActiveAgent::Dashboard.current_user_method)
-      rescue NoMethodError
-        nil
-      end
-
-      # Returns the current owner (account in multi-tenant, user otherwise).
-      def current_owner
-        if ActiveAgent::Dashboard.multi_tenant? && ActiveAgent::Dashboard.current_account_method
-          send(ActiveAgent::Dashboard.current_account_method)
-        else
-          current_user
-        end
-      rescue NoMethodError
-        nil
-      end
-    end
-  end
-end

data/lib/active_agent/dashboard/app/controllers/active_agent/dashboard/dashboard_controller.rb

@@ -1,126 +0,0 @@
-# frozen_string_literal: true
-
-module ActiveAgent
-  module Dashboard
-    # Main dashboard controller showing overview metrics and recent activity.
-    #
-    class DashboardController < ApplicationController
-      def index
-        @agents = fetch_agents.limit(10)
-        @recent_runs = fetch_recent_runs.limit(10)
-        @recent_traces = fetch_recent_traces.limit(10)
-        @metrics = calculate_metrics
-
-        if ActiveAgent::Dashboard.use_inertia && defined?(InertiaRails)
-          render inertia: "Dashboard", props: {
-            agents: serialize_agents(@agents),
-            recentRuns: serialize_runs(@recent_runs),
-            recentTraces: serialize_traces(@recent_traces),
-            metrics: @metrics,
-            user: current_user_props,
-            account: current_account_props
-          }
-        else
-          render :index
-        end
-      end
-
-      private
-
-      def fetch_agents
-        agents = Agent.order(updated_at: :desc)
-        agents = agents.for_owner(current_owner) if current_owner
-        agents
-      end
-
-      def fetch_recent_runs
-        runs = AgentRun.includes(:agent).recent
-        if current_owner && ActiveAgent::Dashboard.multi_tenant?
-          runs = runs.joins(:agent).where(agents: { account_id: current_owner.id })
-        end
-        runs
-      end
-
-      def fetch_recent_traces
-        traces = ActiveAgent::Dashboard.trace_model.recent
-        traces = traces.for_account(current_owner) if current_owner && ActiveAgent::Dashboard.multi_tenant?
-        traces
-      end
-
-      def calculate_metrics
-        traces_24h = ActiveAgent::Dashboard.trace_model.where("created_at > ?", 24.hours.ago)
-        runs_24h = AgentRun.where("created_at > ?", 24.hours.ago)
-
-        if current_owner && ActiveAgent::Dashboard.multi_tenant?
-          traces_24h = traces_24h.for_account(current_owner)
-          runs_24h = runs_24h.joins(:agent).where(agents: { account_id: current_owner.id })
-        end
-
-        {
-          total_agents: fetch_agents.count,
-          active_agents: fetch_agents.active_agents.count,
-          total_runs_24h: runs_24h.count,
-          successful_runs_24h: runs_24h.successful.count,
-          failed_runs_24h: runs_24h.failed_runs.count,
-          total_traces_24h: traces_24h.count,
-          total_tokens_24h: traces_24h.sum(:total_input_tokens).to_i + traces_24h.sum(:total_output_tokens).to_i,
-          avg_duration_ms: runs_24h.where.not(duration_ms: nil).average(:duration_ms)&.round || 0
-        }
-      end
-
-      def serialize_agents(agents)
-        agents.map do |agent|
-          {
-            id: agent.id,
-            name: agent.name,
-            slug: agent.slug,
-            description: agent.description,
-            provider: agent.provider,
-            model: agent.model,
-            status: agent.status,
-            presetType: agent.preset_type,
-            versionCount: agent.version_count,
-            updatedAt: agent.updated_at.iso8601
-          }
-        end
-      end
-
-      def serialize_runs(runs)
-        runs.map(&:summary)
-      end
-
-      def serialize_traces(traces)
-        traces.map do |trace|
-          {
-            id: trace.id,
-            traceId: trace.trace_id,
-            displayName: trace.display_name,
-            status: trace.status,
-            duration: trace.formatted_duration,
-            tokens: trace.formatted_tokens,
-            timestamp: trace.timestamp&.iso8601
-          }
-        end
-      end
-
-      def current_user_props
-        return nil unless current_user
-
-        {
-          id: current_user.id,
-          name: current_user.try(:display_name) || current_user.try(:name) || current_user.try(:email),
-          email: current_user.try(:email)
-        }
-      end
-
-      def current_account_props
-        return nil unless current_owner && ActiveAgent::Dashboard.multi_tenant?
-
-        {
-          id: current_owner.id,
-          name: current_owner.try(:name)
-        }
-      end
-    end
-  end
-end