Merge pull request #2509 from newrelic/update-summary-attr-datatypes

Update ChatCompletionSummary attribute datatypes
newrelic · Mar 18, 2024 · 6ba9084 · 6ba9084
2 parents b3a6073 + b21f4e0
commit 6ba9084
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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