traceloop-sdk 0.0.1 → 0.0.4

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 0f9a52a33ec9fa98e0585bd79a9ba94226f3b905b2d99531554dcbcb20636b53
-  data.tar.gz: 4266d880d4f62c937bcb3d2c3d36c294f25a27c6a111aa22d965aa556ed76a23
+  metadata.gz: 50c7d784c78f716b3416a3143d25a8dfaa0c406a16bbcd10342082a63d93f074
+  data.tar.gz: 26543e0a8ad9b85e4746b338fb086aad27963e23d8279e4548beb12d13629c1c
 SHA512:
-  metadata.gz: 9358cc8b3eb112a37330df529d8b8ab3bcffc7d3c6a437f61c8d90c2f4f23c3dc494fb440d79abdaa6162f59f5e5a97ddd3099823187aa26284f3cb5bfb2712b
-  data.tar.gz: b2f6cb79c744ae964c4c3c614be1493234080e7e5ce40266eb5730222d8c22576b49e6b341f95c27a7b8e5498170556bc248865b154f67fa8bab1e18b4fb2917
+  metadata.gz: 3b4bf642d9f3f412637bafd75cb5a56d6f499a0eb8853113a2c3b37ae3ddb61ab0c92275cc170253df689c305a6e56984008a35be0782bfc122df713a7e7ea66
+  data.tar.gz: cda39a259b39a44d895373b820bc269da1711f2f7cd6618176f3508d6b4047b0bfc2998492225111651fdca35b799306747a047259a85468ec3288dab0407771

data/lib/traceloop/sdk.rb

@@ -23,21 +23,43 @@ 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}.0.role" => "user",
-            "#{OpenTelemetry::SemanticConventionsAi::SpanAttributes::LLM_PROMPTS}.0.content" => user_prompt
+            "#{OpenTelemetry::SemanticConventionsAi::SpanAttributes::LLM_PROMPTS}.1.role" => "user",
+            "#{OpenTelemetry::SemanticConventionsAi::SpanAttributes::LLM_PROMPTS}.1.content" => user_prompt
           })
         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.1
+  version: 0.0.4
 platform: ruby
 authors:
 - Traceloop
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2023-11-03 00:00:00.000000000 Z
+date: 2024-06-24 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: opentelemetry-semantic_conventions_ai