Class: LlmCostTracker::Providers::Anthropic::Parser

Inherits:
LlmCostTracker::Parsers::Base show all
Defined in:
lib/llm_cost_tracker/providers/anthropic/parser.rb

Constant Summary collapse

HOSTS = 
%w[api.anthropic.com].freeze

Class Method Summary collapse

Instance Method Summary collapse

Methods inherited from LlmCostTracker::Parsers::Base

#auto_enable_stream_usage?, #model_for, #safe_json_parse, #streaming_request?

Methods included from LlmCostTracker::Parsers::UrlMatchers

#match_uri?, #parsed_uri, #path_matches?, #uri_matches?

Class Method Details

.match?(url) ⇒ Boolean

Returns:

  • (Boolean) 


12
13
14
 
# File 'lib/llm_cost_tracker/providers/anthropic/parser.rb', line 12

def match?(url)
  match_uri?(url, hosts: HOSTS, path_includes: "/v1/messages")
end

.provider_names ⇒ Object 



16
17
18
 
# File 'lib/llm_cost_tracker/providers/anthropic/parser.rb', line 16

def provider_names
  %w[anthropic]
end

Instance Method Details

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



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

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

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

  request = symbolize_request(request_body)

  ResponseParser.event_from_usage(
    usage: usage,
    model: response["model"] || request[:model],
    provider_response_id: response["id"],
    usage_source: Usage::Source::RESPONSE,
    request: request
  )
end

#parse_stream(response_status:, request_body: nil, events: []) ⇒ Object 



39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
 
# File 'lib/llm_cost_tracker/providers/anthropic/parser.rb', line 39

def parse_stream(response_status:, request_body: nil, events: [], **)
  return nil unless response_status == 200

  request = symbolize_request(request_body)
  model = find_event_value(events) { |data| data.dig("message", "model") } || request[:model]
  usage = stream_usage(events)&.deep_symbolize_keys
  response_id = find_event_value(events) { |data| data.dig("message", "id") || data["id"] }

  if usage
    ResponseParser.event_from_usage(
      usage: usage,
      model: model,
      provider_response_id: response_id,
      usage_source: Usage::Source::STREAM_FINAL,
      request: request,
      stream: true
    )
  else
    build_unknown_stream_usage(
      provider: "anthropic",
      model: model,
      provider_response_id: response_id,
      pricing_mode: UsageExtractor.pricing_mode(request: request, usage: usage)
    )
  end
end

#provider_for(_request_url) ⇒ Object 



66
67
68
 
# File 'lib/llm_cost_tracker/providers/anthropic/parser.rb', line 66

def provider_for(_request_url)
  "anthropic"
end