traceloop-sdk 0.0.2 → 0.0.5

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 061af948a12a48ccb1ce3925d9a89890ee963da3bd61ea3b6659c16453bc04d8
-  data.tar.gz: d6d345827dbaa2ccd4dff791c902bafdc614a422008182baa2cad8b2119586f2
+  metadata.gz: 50629c90b71631149726b2b262d4bf67c4d58c199f15a20781f6a01955a00ff0
+  data.tar.gz: 7ff11a15914c924a7dec64df865afef47f0d4fc323bc4b5f3a38c412f413f416
 SHA512:
-  metadata.gz: 4771c2eef21ad54932cfa3888a2421b3f8d568aac96075ccfdac2b261c1e4c6aa5d494829e37508d71492a6467cc347738e4f4b267137245555cca2f0747b871
-  data.tar.gz: 4ca5771c19b5c1ae9d4cdf145047a068c3bfb90b06fa18f45e43f5f863a594708f918e307b7cb1b83f8d753a5349757127b6eb34715ab679b3cb7f814b05cfcc
+  metadata.gz: 72815dd504966fcf8e438b76447eef5a41570503a3c3fb2a2c8248987477bb964945bb454c5e54f0635a25eba5c1f67bcdb6a393435a8b61defe32201c1cf351
+  data.tar.gz: 6dab487b52fcaaebe634cef83ec749a1ffb0c82221cfff7ff6871249fca82332d1c3b47b710825c25fc51b1c2d4c59b154275c447623bfa48e3dc1f6c3df207d

data/lib/traceloop/sdk.rb

@@ -23,21 +23,50 @@ module Traceloop
       end
 
       class Tracer
-        def initialize(span)
+        def initialize(span, provider, model)
           @span = span
+          @provider = provider
+          @model = model
         end
 
-        def log_prompt(model, system_prompt="", user_prompt)
+        def log_prompt(system_prompt="", user_prompt)
+          unless system_prompt.empty?
+            @span.add_attributes({
+              "#{OpenTelemetry::SemanticConventionsAi::SpanAttributes::LLM_PROMPTS}.0.role" => "system",
+              "#{OpenTelemetry::SemanticConventionsAi::SpanAttributes::LLM_PROMPTS}.0.content" => system_prompt,
+              "#{OpenTelemetry::SemanticConventionsAi::SpanAttributes::LLM_PROMPTS}.1.role" => "user",
+              "#{OpenTelemetry::SemanticConventionsAi::SpanAttributes::LLM_PROMPTS}.1.content" => user_prompt
+            })
+          else
+            @span.add_attributes({
+              "#{OpenTelemetry::SemanticConventionsAi::SpanAttributes::LLM_PROMPTS}.0.role" => "user",
+              "#{OpenTelemetry::SemanticConventionsAi::SpanAttributes::LLM_PROMPTS}.0.content" => user_prompt
+            })
+          end
+        end
+
+        def log_response(response)
+          # This is Gemini specific, see -
+          # https://github.com/gbaptista/gemini-ai?tab=readme-ov-file#generate_content
+          if response.has_key?("candidates")
+            log_gemini_response(response)
+          else
+            log_openai_response(response)
+          end
+        end
+
+        def log_gemini_response(response)
           @span.add_attributes({
-            OpenTelemetry::SemanticConventionsAi::SpanAttributes::LLM_REQUEST_MODEL => model,
-            "#{OpenTelemetry::SemanticConventionsAi::SpanAttributes::LLM_PROMPTS}.0.role" => "system",
-            "#{OpenTelemetry::SemanticConventionsAi::SpanAttributes::LLM_PROMPTS}.0.content" => system_prompt,
-            "#{OpenTelemetry::SemanticConventionsAi::SpanAttributes::LLM_PROMPTS}.1.role" => "user",
-            "#{OpenTelemetry::SemanticConventionsAi::SpanAttributes::LLM_PROMPTS}.1.content" => user_prompt
+            OpenTelemetry::SemanticConventionsAi::SpanAttributes::LLM_RESPONSE_MODEL => @model,
           })
+
+          @span.add_attributes({
+            "#{OpenTelemetry::SemanticConventionsAi::SpanAttributes::LLM_COMPLETIONS}.0.role" => "assistant",
+            "#{OpenTelemetry::SemanticConventionsAi::SpanAttributes::LLM_COMPLETIONS}.0.content" => response.dig("candidates", 0, "content", "parts", 0, "text")
+            })
         end
 
-        def log_response(response)
+        def log_openai_response(response)
           @span.add_attributes({
             OpenTelemetry::SemanticConventionsAi::SpanAttributes::LLM_RESPONSE_MODEL => response.dig("model"),
           })
@@ -57,9 +86,12 @@ module Traceloop
         end
       end
 
-      def llm_call
-        @tracer.in_span("openai.chat") do |span|
-          yield Tracer.new(span)
+      def llm_call(provider, model)
+        @tracer.in_span("#{provider}.chat") do |span|
+          span.add_attributes({
+            OpenTelemetry::SemanticConventionsAi::SpanAttributes::LLM_REQUEST_MODEL => model,
+          })
+          yield Tracer.new(span, provider, model)
         end
       end
     end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: traceloop-sdk
 version: !ruby/object:Gem::Version
-  version: 0.0.2
+  version: 0.0.5
 platform: ruby
 authors:
 - Traceloop
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2023-12-29 00:00:00.000000000 Z
+date: 2024-06-24 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: opentelemetry-semantic_conventions_ai