feat(sdk): workflows, tasks, agents, tools (#10)

Author: nirga

sample-app/Gemfile.lock

@@ -83,7 +83,7 @@ GEM
       faraday (>= 0.17.5, < 3.a)
       jwt (>= 1.5, < 3.0)
       multi_json (~> 1.10)
-    traceloop-sdk (0.0.8)
+    traceloop-sdk (0.1.2)
       opentelemetry-exporter-otlp (~> 0.26.1)
       opentelemetry-sdk (~> 1.3.1)
       opentelemetry-semantic_conventions_ai (~> 0.0.3)

sample-app/openai.rb

@@ -9,13 +9,15 @@
 
 traceloop = Traceloop::SDK::Traceloop.new
 
-traceloop.llm_call(provider="openai", model="gpt-3.5-turbo") do |tracer|
-  tracer.log_prompt(user_prompt="Tell me a joke about OpenTelemetry")
-  response = client.chat(
-    parameters: {
-      model: "gpt-3.5-turbo",
-      messages: [{ role: "user", content: "Tell me a joke about OpenTelemetry" }]
-    })
-  tracer.log_response(response)
-  puts response.dig("choices", 0, "message", "content")
+traceloop.workflow("joke_generator") do
+  traceloop.llm_call(provider="openai", model="gpt-3.5-turbo") do |tracer|
+    tracer.log_prompt(user_prompt="Tell me a joke about OpenTelemetry")
+    response = client.chat(
+      parameters: {
+        model: "gpt-3.5-turbo",
+        messages: [{ role: "user", content: "Tell me a joke about OpenTelemetry" }]
+      })
+    tracer.log_response(response)
+    puts response.dig("choices", 0, "message", "content")
+  end
 end

traceloop-sdk/lib/traceloop/sdk.rb

@@ -122,6 +122,46 @@ def llm_call(provider, model)
           yield Tracer.new(span, provider, model)
         end
       end
+
+      def workflow(name)
+        @tracer.in_span("#{name}.workflow") do |span|
+          span.add_attributes({
+            OpenTelemetry::SemanticConventionsAi::SpanAttributes::TRACELOOP_SPAN_KIND => "workflow",
+            OpenTelemetry::SemanticConventionsAi::SpanAttributes::TRACELOOP_ENTITY_NAME => name,
+          })
+          yield
+        end
+      end
+
+      def task(name)
+        @tracer.in_span("#{name}.task") do |span|
+          span.add_attributes({
+            OpenTelemetry::SemanticConventionsAi::SpanAttributes::TRACELOOP_SPAN_KIND => "task",
+            OpenTelemetry::SemanticConventionsAi::SpanAttributes::TRACELOOP_ENTITY_NAME => name,
+          })
+          yield
+        end
+      end
+
+      def agent(name)
+        @tracer.in_span("#{name}.agent") do |span|
+          span.add_attributes({
+            OpenTelemetry::SemanticConventionsAi::SpanAttributes::TRACELOOP_SPAN_KIND => "agent",
+            OpenTelemetry::SemanticConventionsAi::SpanAttributes::TRACELOOP_ENTITY_NAME => name,
+          })
+          yield
+        end
+      end
+
+      def tool(name)
+        @tracer.in_span("#{name}.tool") do |span|
+          span.add_attributes({
+            OpenTelemetry::SemanticConventionsAi::SpanAttributes::TRACELOOP_SPAN_KIND => "tool",
+            OpenTelemetry::SemanticConventionsAi::SpanAttributes::TRACELOOP_ENTITY_NAME => name,
+          })
+          yield
+        end
+      end
     end
   end
 end

traceloop-sdk/traceloop-sdk.gemspec

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