# Generated by SnakeBridge v0.12.0 - DO NOT EDIT MANUALLY
# Regenerate with: mix compile
# Library: dspy 3.1.2
# Python module: dspy.utils.usage_tracker
# Python class: UsageTracker

defmodule Dspy.Utils.UsageTrackerClass3 do
  def __snakebridge_python_name__, do: "dspy.utils.usage_tracker"
  def __snakebridge_python_class__, do: "UsageTracker"
  def __snakebridge_library__, do: "dspy"
  @opaque t :: SnakeBridge.Ref.t()

  @spec new(keyword()) :: {:ok, SnakeBridge.Ref.t()} | {:error, Snakepit.Error.t()}
  def new(opts \\ []) do
    SnakeBridge.Runtime.call_class(__MODULE__, :__init__, [], opts)
  end

  @spec _flatten_usage_entry(SnakeBridge.Ref.t(), %{optional(String.t()) => term()}, keyword()) ::
          {:ok, %{optional(String.t()) => term()}} | {:error, Snakepit.Error.t()}
  def _flatten_usage_entry(ref, usage_entry, opts \\ []) do
    SnakeBridge.Runtime.call_method(ref, :_flatten_usage_entry, [usage_entry], opts)
  end

  @spec _merge_usage_entries(SnakeBridge.Ref.t(), term(), term(), keyword()) ::
          {:ok, %{optional(String.t()) => term()}} | {:error, Snakepit.Error.t()}
  def _merge_usage_entries(ref, usage_entry1, usage_entry2, opts \\ []) do
    SnakeBridge.Runtime.call_method(
      ref,
      :_merge_usage_entries,
      [usage_entry1, usage_entry2],
      opts
    )
  end

  @spec add_usage(SnakeBridge.Ref.t(), String.t(), %{optional(String.t()) => term()}, keyword()) ::
          {:ok, nil} | {:error, Snakepit.Error.t()}
  def add_usage(ref, lm, usage_entry, opts \\ []) do
    SnakeBridge.Runtime.call_method(ref, :add_usage, [lm, usage_entry], opts)
  end

  @spec get_total_tokens(SnakeBridge.Ref.t(), keyword()) ::
          {:ok, %{optional(String.t()) => %{optional(String.t()) => term()}}}
          | {:error, Snakepit.Error.t()}
  def get_total_tokens(ref, opts \\ []) do
    SnakeBridge.Runtime.call_method(ref, :get_total_tokens, [], opts)
  end
end