braintrust 0.0.12 → 0.1.0

data/lib/braintrust/trace/tokens.rb

@@ -1,109 +0,0 @@
-# frozen_string_literal: true
-
-module Braintrust
-  module Trace
-    # Parse OpenAI usage tokens into normalized Braintrust metrics.
-    # Handles standard fields and *_tokens_details nested objects.
-    # @param usage [Hash, Object] usage object from OpenAI response
-    # @return [Hash<String, Integer>] normalized metrics
-    def self.parse_openai_usage_tokens(usage)
-      metrics = {}
-      return metrics unless usage
-
-      usage_hash = usage.respond_to?(:to_h) ? usage.to_h : usage
-      return metrics unless usage_hash.is_a?(Hash)
-
-      # Field mappings: OpenAI → Braintrust
-      # Supports both Chat Completions API (prompt_tokens, completion_tokens)
-      # and Responses API (input_tokens, output_tokens)
-      field_map = {
-        "prompt_tokens" => "prompt_tokens",
-        "completion_tokens" => "completion_tokens",
-        "total_tokens" => "tokens",
-        # Responses API uses different field names
-        "input_tokens" => "prompt_tokens",
-        "output_tokens" => "completion_tokens"
-      }
-
-      # Prefix mappings for *_tokens_details
-      prefix_map = {
-        "prompt" => "prompt",
-        "completion" => "completion",
-        # Responses API uses input/output prefixes
-        "input" => "prompt",
-        "output" => "completion"
-      }
-
-      usage_hash.each do |key, value|
-        key_str = key.to_s
-
-        if value.is_a?(Numeric)
-          target = field_map[key_str]
-          metrics[target] = value.to_i if target
-        elsif key_str.end_with?("_tokens_details")
-          # Convert to hash if it's an object (OpenAI SDK returns objects)
-          details_hash = value.respond_to?(:to_h) ? value.to_h : value
-          next unless details_hash.is_a?(Hash)
-
-          raw_prefix = key_str.sub(/_tokens_details$/, "")
-          prefix = prefix_map[raw_prefix] || raw_prefix
-          details_hash.each do |detail_key, detail_value|
-            next unless detail_value.is_a?(Numeric)
-            metrics["#{prefix}_#{detail_key}"] = detail_value.to_i
-          end
-        end
-      end
-
-      # Calculate total if missing
-      if !metrics.key?("tokens") && metrics.key?("prompt_tokens") && metrics.key?("completion_tokens")
-        metrics["tokens"] = metrics["prompt_tokens"] + metrics["completion_tokens"]
-      end
-
-      metrics
-    end
-
-    # Parse Anthropic usage tokens into normalized Braintrust metrics.
-    # Accumulates cache tokens into prompt_tokens and calculates total.
-    # @param usage [Hash, Object] usage object from Anthropic response
-    # @return [Hash<String, Integer>] normalized metrics
-    def self.parse_anthropic_usage_tokens(usage)
-      metrics = {}
-      return metrics unless usage
-
-      usage_hash = usage.respond_to?(:to_h) ? usage.to_h : usage
-      return metrics unless usage_hash.is_a?(Hash)
-
-      # Field mappings: Anthropic → Braintrust
-      # Also handles RubyLLM's simplified cache field names
-      field_map = {
-        "input_tokens" => "prompt_tokens",
-        "output_tokens" => "completion_tokens",
-        "cache_read_input_tokens" => "prompt_cached_tokens",
-        "cache_creation_input_tokens" => "prompt_cache_creation_tokens",
-        # RubyLLM uses simplified names
-        "cached_tokens" => "prompt_cached_tokens",
-        "cache_creation_tokens" => "prompt_cache_creation_tokens"
-      }
-
-      usage_hash.each do |key, value|
-        next unless value.is_a?(Numeric)
-        key_str = key.to_s
-        target = field_map[key_str]
-        metrics[target] = value.to_i if target
-      end
-
-      # Accumulate cache tokens into prompt_tokens (matching TS/Python SDKs)
-      prompt_tokens = (metrics["prompt_tokens"] || 0) +
-        (metrics["prompt_cached_tokens"] || 0) +
-        (metrics["prompt_cache_creation_tokens"] || 0)
-      metrics["prompt_tokens"] = prompt_tokens if prompt_tokens > 0
-
-      # Calculate total
-      if metrics.key?("prompt_tokens") && metrics.key?("completion_tokens")
-        metrics["tokens"] = metrics["prompt_tokens"] + metrics["completion_tokens"]
-      end
-
-      metrics
-    end
-  end
-end