fix(sdk): support message logging in OpenAI format (#12)

Author: nirga

sample-app/Gemfile.lock

@@ -4,54 +4,61 @@ GEM
     addressable (2.8.7)
       public_suffix (>= 2.0.2, < 7.0)
     aws-eventstream (1.3.0)
-    aws-partitions (1.947.0)
-    aws-sdk-bedrockruntime (1.14.0)
-      aws-sdk-core (~> 3, >= 3.199.0)
-      aws-sigv4 (~> 1.1)
-    aws-sdk-core (3.199.0)
+    aws-partitions (1.987.0)
+    aws-sdk-bedrockruntime (1.26.0)
+      aws-sdk-core (~> 3, >= 3.207.0)
+      aws-sigv4 (~> 1.5)
+    aws-sdk-core (3.209.1)
       aws-eventstream (~> 1, >= 1.3.0)
       aws-partitions (~> 1, >= 1.651.0)
-      aws-sigv4 (~> 1.8)
+      aws-sigv4 (~> 1.9)
       jmespath (~> 1, >= 1.6.1)
-    aws-sigv4 (1.8.0)
+    aws-sigv4 (1.10.0)
       aws-eventstream (~> 1, >= 1.0.2)
-    base64 (0.1.1)
+    base64 (0.2.0)
     ethon (0.16.0)
       ffi (>= 1.15.0)
     event_stream_parser (1.0.0)
-    faraday (2.9.2)
-      faraday-net_http (>= 2.0, < 3.2)
+    faraday (2.12.0)
+      faraday-net_http (>= 2.0, < 3.4)
+      json
+      logger
     faraday-multipart (1.0.4)
       multipart-post (~> 2)
-    faraday-net_http (3.0.2)
+    faraday-net_http (3.3.0)
+      net-http
     faraday-typhoeus (1.1.0)
       faraday (~> 2.0)
       typhoeus (~> 1.4)
     ffi (1.17.0-arm64-darwin)
-    gemini-ai (4.1.0)
+    gemini-ai (4.2.0)
       event_stream_parser (~> 1.0)
-      faraday (~> 2.9, >= 2.9.2)
+      faraday (~> 2.10)
       faraday-typhoeus (~> 1.1)
       googleauth (~> 1.8)
       typhoeus (~> 1.4, >= 1.4.1)
-    google-cloud-env (2.1.1)
+    google-cloud-env (2.2.1)
       faraday (>= 1.0, < 3.a)
-    google-protobuf (3.25.3-arm64-darwin)
-    googleapis-common-protos-types (1.15.0)
+    google-protobuf (3.25.5-arm64-darwin)
+    googleapis-common-protos-types (1.16.0)
       google-protobuf (>= 3.18, < 5.a)
-    googleauth (1.11.0)
+    googleauth (1.11.1)
       faraday (>= 1.0, < 3.a)
       google-cloud-env (~> 2.1)
       jwt (>= 1.4, < 3.0)
       multi_json (~> 1.11)
       os (>= 0.9, < 2.0)
       signet (>= 0.16, < 2.a)
     jmespath (1.6.2)
-    jwt (2.8.2)
+    json (2.7.2)
+    jwt (2.9.3)
       base64
+    logger (1.6.1)
     multi_json (1.15.0)
-    multipart-post (2.3.0)
-    opentelemetry-api (1.2.5)
+    multipart-post (2.4.1)
+    net-http (0.4.1)
+      uri
+    opentelemetry-api (1.4.0)
     opentelemetry-common (0.21.0)
       opentelemetry-api (~> 1.0)
     opentelemetry-exporter-otlp (0.26.3)
@@ -68,12 +75,12 @@ GEM
       opentelemetry-common (~> 0.20)
       opentelemetry-registry (~> 0.2)
       opentelemetry-semantic_conventions
-    opentelemetry-semantic_conventions (1.10.0)
+    opentelemetry-semantic_conventions (1.10.1)
       opentelemetry-api (~> 1.0)
     opentelemetry-semantic_conventions_ai (0.0.3)
       opentelemetry-api (~> 1.0)
     os (1.1.4)
-    public_suffix (6.0.0)
+    public_suffix (6.0.1)
     ruby-openai (7.1.0)
       event_stream_parser (>= 0.3.0, < 2.0.0)
       faraday (>= 1)
@@ -83,12 +90,13 @@ GEM
       faraday (>= 0.17.5, < 3.a)
       jwt (>= 1.5, < 3.0)
       multi_json (~> 1.10)
-    traceloop-sdk (0.1.2)
+    traceloop-sdk (0.1.4)
       opentelemetry-exporter-otlp (~> 0.26.1)
       opentelemetry-sdk (~> 1.3.1)
       opentelemetry-semantic_conventions_ai (~> 0.0.3)
     typhoeus (1.4.1)
       ethon (>= 0.9.0)
+    uri (0.13.1)
 
 PLATFORMS
   arm64-darwin-23

traceloop-sdk/lib/traceloop/sdk.rb

@@ -29,6 +29,15 @@ def initialize(span, provider, model)
           @model = model
         end
 
+        def log_messages(messages)
+          messages.each_with_index do |message, index|
+            @span.add_attributes({
+              "#{OpenTelemetry::SemanticConventionsAi::SpanAttributes::LLM_PROMPTS}.#{index}.role" => message[:role] || "user",
+              "#{OpenTelemetry::SemanticConventionsAi::SpanAttributes::LLM_PROMPTS}.#{index}.content" => message[:content] || "",
+            })
+          end
+        end
+
         def log_prompt(system_prompt="", user_prompt)
           unless system_prompt.empty?
             @span.add_attributes({

traceloop-sdk/traceloop-sdk.gemspec

@@ -1,6 +1,6 @@
 Gem::Specification.new do |spec|
   spec.name        = 'traceloop-sdk'
-  spec.version     = '0.1.2'
+  spec.version     = '0.1.4'
   spec.authors     = ["Traceloop"]
   spec.email       = ['dev@traceloop.com']