llm_cost_tracker 0.7.3 → 0.9.0

data/lib/llm_cost_tracker/generators/llm_cost_tracker/upgrade_tags_to_jsonb_generator.rb

@@ -1,29 +0,0 @@
-# frozen_string_literal: true
-
-require "rails/generators"
-require "rails/generators/active_record"
-
-module LlmCostTracker
-  module Generators
-    class UpgradeTagsToJsonbGenerator < Rails::Generators::Base
-      include ActiveRecord::Generators::Migration
-
-      source_root File.expand_path("templates", __dir__)
-
-      desc "Creates a migration to upgrade llm_api_calls.tags to PostgreSQL JSONB"
-
-      def create_migration_file
-        migration_template(
-          "upgrade_llm_api_call_tags_to_jsonb.rb.erb",
-          "db/migrate/upgrade_llm_api_call_tags_to_jsonb.rb"
-        )
-      end
-
-      private
-
-      def migration_version
-        "[#{ActiveRecord::VERSION::MAJOR}.#{ActiveRecord::VERSION::MINOR}]"
-      end
-    end
-  end
-end

data/lib/llm_cost_tracker/ledger/rollups/batch.rb

@@ -1,43 +0,0 @@
-# frozen_string_literal: true
-
-require "bigdecimal"
-
-require_relative "../period"
-
-module LlmCostTracker
-  module Ledger
-    class Rollups
-      class Batch
-        def self.rows(events)
-          new(events).rows
-        end
-
-        def initialize(events)
-          @events = events
-        end
-
-        def rows
-          totals.map do |(period, period_start), total_cost|
-            {
-              period: period,
-              period_start: period_start,
-              total_cost: total_cost
-            }
-          end
-        end
-
-        private
-
-        attr_reader :events
-
-        def totals
-          events.each_with_object(Hash.new { |hash, key| hash[key] = BigDecimal("0") }) do |event, rows|
-            Period::PERIODS.each do |period, name|
-              rows[[name, Period.bucket(period, event.tracked_at)]] += BigDecimal(event.total_cost.to_s)
-            end
-          end
-        end
-      end
-    end
-  end
-end

data/lib/llm_cost_tracker/ledger/schema/period_totals.rb

@@ -1,32 +0,0 @@
-# frozen_string_literal: true
-
-module LlmCostTracker
-  module Ledger
-    module Schema
-      module PeriodTotals
-        REQUIRED_COLUMNS = %w[period period_start total_cost].freeze
-        UNIQUE_COLUMNS = %i[period period_start].freeze
-
-        class << self
-          def current_schema_errors
-            connection = Ledger::Call.connection
-            table_name = Ledger::Period::Total.table_name
-            return ["llm_cost_tracker_period_totals table is missing"] unless connection.data_source_exists?(table_name)
-
-            errors = []
-            missing = REQUIRED_COLUMNS - Ledger::Period::Total.columns_hash.keys
-            errors << "missing columns: #{missing.join(', ')}" if missing.any?
-            errors << "missing unique index: period, period_start" unless unique_period_index?(connection, table_name)
-            errors
-          end
-
-          private
-
-          def unique_period_index?(connection, table_name)
-            connection.index_exists?(table_name, UNIQUE_COLUMNS, unique: true)
-          end
-        end
-      end
-    end
-  end
-end

data/lib/llm_cost_tracker/pricing/components.rb

@@ -1,37 +0,0 @@
-# frozen_string_literal: true
-
-module LlmCostTracker
-  module Pricing
-    Component = Data.define(:price_key, :token_key, :cost_key)
-
-    COMPONENTS = [
-      Component.new(
-        price_key: :input,
-        token_key: :input_tokens,
-        cost_key: :input_cost
-      ),
-      Component.new(
-        price_key: :cache_read_input,
-        token_key: :cache_read_input_tokens,
-        cost_key: :cache_read_input_cost
-      ),
-      Component.new(
-        price_key: :cache_write_input,
-        token_key: :cache_write_input_tokens,
-        cost_key: :cache_write_input_cost
-      ),
-      Component.new(
-        price_key: :cache_write_1h_input,
-        token_key: :cache_write_1h_input_tokens,
-        cost_key: :cache_write_1h_input_cost
-      ),
-      Component.new(
-        price_key: :output,
-        token_key: :output_tokens,
-        cost_key: :output_cost
-      )
-    ].freeze
-
-    COST_KEYS = (COMPONENTS.map(&:cost_key) + %i[total_cost]).freeze
-  end
-end

data/lib/llm_cost_tracker/pricing/sync/registry_loader.rb

@@ -1,63 +0,0 @@
-# frozen_string_literal: true
-
-require "json"
-require "yaml"
-
-require_relative "../registry"
-
-module LlmCostTracker
-  module Pricing
-    module Sync
-      class RegistryLoader
-        YAML_EXTENSIONS = %w[.yml .yaml].freeze
-
-        def call(path:, seed_path:)
-          source_path = File.exist?(path.to_s) ? path.to_s : seed_path.to_s
-          normalize_registry(load_registry_file(source_path))
-        rescue Errno::ENOENT, JSON::ParserError, Psych::Exception, ArgumentError, TypeError => e
-          raise Error, "Unable to load pricing registry #{source_path.inspect}: #{e.message}"
-        end
-
-        private
-
-        def load_registry_file(path)
-          if File.size(path) > Registry::MAX_FILE_BYTES
-            raise ArgumentError, "pricing registry exceeds #{Registry::MAX_FILE_BYTES} bytes"
-          end
-
-          contents = File.read(path)
-          registry = yaml_file?(path) ? (YAML.safe_load(contents, aliases: false) || {}) : JSON.parse(contents)
-          raise ArgumentError, "pricing registry must be a hash" unless registry.is_a?(Hash)
-
-          registry
-        end
-
-        def normalize_registry(registry)
-          {
-            "metadata" => normalize_hash(registry.fetch("metadata", {}), label: "pricing metadata"),
-            "models" => normalize_models(registry.fetch("models", {}))
-          }
-        end
-
-        def normalize_models(models)
-          normalize_hash(models, label: "pricing models").each_with_object({}) do |(model, entry), normalized|
-            normalized[model.to_s] = normalize_hash(entry, label: "pricing model entry")
-          end
-        end
-
-        def normalize_hash(hash, label:)
-          return {} if hash.nil?
-          raise ArgumentError, "#{label} must be a hash" unless hash.is_a?(Hash)
-
-          hash.each_with_object({}) do |(key, value), normalized|
-            normalized[key.to_s] = value
-          end
-        end
-
-        def yaml_file?(path)
-          YAML_EXTENSIONS.include?(File.extname(path).downcase)
-        end
-      end
-    end
-  end
-end