Class: LlmCostTracker::Parsers::Openai

Inherits:
Base
  • Object
show all
Defined in:
lib/llm_cost_tracker/parsers/openai.rb

Direct Known Subclasses

OpenaiCompatible

Constant Summary collapse

HOSTS = 
%w[api.openai.com].freeze
TRACKED_PATHS = 
%w[/v1/chat/completions /v1/completions /v1/embeddings /v1/responses].freeze

Instance Method Summary collapse

Instance Method Details

#match?(url) ⇒ Boolean

Returns:

  • (Boolean) 


14
15
16
17
18
19
 
# File 'lib/llm_cost_tracker/parsers/openai.rb', line 14

def match?(url)
  uri = URI.parse(url.to_s)
  HOSTS.include?(uri.host.to_s.downcase) && TRACKED_PATHS.include?(uri.path)
rescue URI::InvalidURIError
  false
end

#parse(request_url, request_body, response_status, response_body) ⇒ Object 



21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
 
# File 'lib/llm_cost_tracker/parsers/openai.rb', line 21

def parse(request_url, request_body, response_status, response_body)
  return nil unless response_status == 200

  response = safe_json_parse(response_body)
  usage    = response["usage"]
  return nil unless usage

  request = safe_json_parse(request_body)

  {
    provider: provider_for(request_url),
    model: response["model"] || request["model"],
    input_tokens: usage["prompt_tokens"] || usage["input_tokens"] || 0,
    output_tokens: usage["completion_tokens"] || usage["output_tokens"] || 0,
    total_tokens: usage["total_tokens"] || 0,
    cached_input_tokens: cached_input_tokens(usage)
  }.compact
end