traceloop-sdk 0.0.2 → 0.0.4

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 061af948a12a48ccb1ce3925d9a89890ee963da3bd61ea3b6659c16453bc04d8
-  data.tar.gz: d6d345827dbaa2ccd4dff791c902bafdc614a422008182baa2cad8b2119586f2
+  metadata.gz: 50c7d784c78f716b3416a3143d25a8dfaa0c406a16bbcd10342082a63d93f074
+  data.tar.gz: 26543e0a8ad9b85e4746b338fb086aad27963e23d8279e4548beb12d13629c1c
 SHA512:
-  metadata.gz: 4771c2eef21ad54932cfa3888a2421b3f8d568aac96075ccfdac2b261c1e4c6aa5d494829e37508d71492a6467cc347738e4f4b267137245555cca2f0747b871
-  data.tar.gz: 4ca5771c19b5c1ae9d4cdf145047a068c3bfb90b06fa18f45e43f5f863a594708f918e307b7cb1b83f8d753a5349757127b6eb34715ab679b3cb7f814b05cfcc
+  metadata.gz: 3b4bf642d9f3f412637bafd75cb5a56d6f499a0eb8853113a2c3b37ae3ddb61ab0c92275cc170253df689c305a6e56984008a35be0782bfc122df713a7e7ea66
+  data.tar.gz: cda39a259b39a44d895373b820bc269da1711f2f7cd6618176f3508d6b4047b0bfc2998492225111651fdca35b799306747a047259a85468ec3288dab0407771

data/lib/traceloop/sdk.rb

@@ -23,13 +23,14 @@ 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)
           @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",
@@ -38,6 +39,27 @@ module Traceloop
         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_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_openai_response(response)
           @span.add_attributes({
             OpenTelemetry::SemanticConventionsAi::SpanAttributes::LLM_RESPONSE_MODEL => response.dig("model"),
           })
@@ -57,9 +79,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.4
 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