From b21f4e0b1a2f418f1d853c122cd670e1896c98bf Mon Sep 17 00:00:00 2001 From: Kayla Reopelle Date: Wed, 13 Mar 2024 16:00:14 -0700 Subject: [PATCH] Update ChatCompletionSummary attribute datatypes request_max_tokens should be an integer request.temperature should be a float This change ensures the correct datatype is assigned. --- .../agent/instrumentation/ruby_openai/instrumentation.rb | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/lib/new_relic/agent/instrumentation/ruby_openai/instrumentation.rb b/lib/new_relic/agent/instrumentation/ruby_openai/instrumentation.rb index f78a61e3e9..58d3bf9b39 100644 --- a/lib/new_relic/agent/instrumentation/ruby_openai/instrumentation.rb +++ b/lib/new_relic/agent/instrumentation/ruby_openai/instrumentation.rb @@ -67,9 +67,9 @@ def chat_completions_instrumentation(parameters) def create_chat_completion_summary(parameters) NewRelic::Agent::Llm::ChatCompletionSummary.new( vendor: VENDOR, - request_max_tokens: parameters[:max_tokens] || parameters['max_tokens'], + request_max_tokens: (parameters[:max_tokens] || parameters['max_tokens'])&.to_i, request_model: parameters[:model] || parameters['model'], - temperature: parameters[:temperature] || parameters['temperature'], + temperature: (parameters[:temperature] || parameters['temperature'])&.to_f, metadata: llm_custom_attributes ) end