Module: LlmCostTracker::Integrations::Openai::ChatCompletionsPatch

Defined in:
lib/llm_cost_tracker/integrations/openai.rb

Instance Method Summary collapse

Instance Method Details

#create(*args, **kwargs) ⇒ Object 



81
82
83
84
85
86
87
88
89
90
91
 
# File 'lib/llm_cost_tracker/integrations/openai.rb', line 81

def create(*args, **kwargs)
  started_at = Process.clock_gettime(Process::CLOCK_MONOTONIC)
  LlmCostTracker::Integrations::Openai.enforce_budget!
  response = super
  LlmCostTracker::Integrations::Openai.record_response(
    response,
    request: LlmCostTracker::Integrations::Openai.request_params(args, kwargs),
    latency_ms: LlmCostTracker::Integrations::Openai.elapsed_ms(started_at)
  )
  response
end