Feature/add update trace method

lib/llm_eval_ruby/api_clients/langfuse.rb

@@ -35,6 +35,15 @@ def create_trace(params = {})
         create_event(type: "trace-create", body:)
       end
 
+      def update_trace(params = {})
+        body = {
+          id: params[:id],
+          output: params[:output]
+        }
+        # WARNING! trace-create is used for update as well they do upsert if id is provided
+        create_event(type: "trace-create", body:)
+      end
+
       def create_span(params = {})
         body = {
           id: params[:id],

lib/llm_eval_ruby/trace_adapters/langfuse.rb

@@ -17,6 +17,15 @@ def trace(**kwargs)
           trace
         end
 
+        def update_trace(**kwargs)
+          trace = TraceTypes::Trace.new(**kwargs)
+          response = client.update_trace(trace.to_h)
+
+          logger.warn "Failed to update trace" if response["successes"].blank?
+
+          trace
+        end
+
         def span(**kwargs)
           span = TraceTypes::Span.new(id: SecureRandom.uuid, **kwargs)
           response = client.create_span(span.to_h)

lib/llm_eval_ruby/trace_adapters/local.rb

@@ -14,6 +14,14 @@ def trace(**kwargs)
           trace
         end
 
+        def update_trace(**kwargs)
+          trace = TraceTypes::Trace.new(id: SecureRandom.uuid, **kwargs)
+
+          logger.info("Trace updated: #{JSON.pretty_generate(trace.to_h)}")
+
+          trace
+        end
+
         def span(**kwargs)
           span = TraceTypes::Span.new(id: SecureRandom.uuid, **kwargs)
 

lib/llm_eval_ruby/tracer.rb

@@ -11,6 +11,10 @@ def self.trace(...)
       new(adapter: LlmEvalRuby.config.adapter).trace(...)
     end
 
+    def self.update_trace(...)
+      new(adapter: LlmEvalRuby.config.adapter).update_trace(...)
+    end
+
     def self.span(...)
       new(adapter: LlmEvalRuby.config.adapter).span(...)
     end
@@ -38,6 +42,10 @@ def trace(...)
       adapter.trace(...)
     end
 
+    def update_trace(...)
+      adapter.update_trace(...)
+    end
+
     def span(...)
       adapter.span(...)
     end