ruby_llm-agents 3.5.4 → 3.5.5

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 5cd2395534d027c7236282e829cf56f26f2ac5dc745796fbd13c66ebc34fd02b
-  data.tar.gz: 67d71f15b1e72ef3d69267167aacb24a579ab04ff8dacff8ab47e19bb6f14148
+  metadata.gz: fcd75502c369dc11b47f538e49ab52b0a04fa742c0df7b889915ee3b6003eebb
+  data.tar.gz: 14d32d454e00c7bf2f0c9d1f61c0d8c16b82fd6d63fd3d72bc5d263c2f143ab0
 SHA512:
-  metadata.gz: 26411eca1c238ec78cc5c17af8717c29786c6414baac95099a36ef4d02f397f842bf471bfcbc5ac5e457be5a475de7ca747503820d0a84e9a82de5d5059ce138
-  data.tar.gz: 227d6f91868013a5dd9b090c731118b027efced0820bd9dc6af24079431ecd15e7385f463af73ab618ee7d0af6bd39b1814d98b0d6a6eb61de2b210c0004c93b
+  metadata.gz: 8d53fe803e960be5b11e92bae6ad6923b3f648359ec476b9f56c46a355e2f71e75c63b71b546ecd170ee8e923300d6c11889ac8573071a891c8b21570094378e
+  data.tar.gz: 674ba879084fdff24c48c02b0374cff6a1a7218420582d88ad76731f53c8fa9e41b97c081a721de8578d69c8a2348a124c283910b48a8b8b144e44d3ea4ce585

data/lib/ruby_llm/agents/core/version.rb

@@ -4,6 +4,6 @@ module RubyLLM
   module Agents
     # Current version of the RubyLLM::Agents gem
     # @return [String] Semantic version string
-    VERSION = "3.5.4"
+    VERSION = "3.5.5"
   end
 end

data/lib/ruby_llm/agents/pipeline/middleware/instrumentation.rb

@@ -202,6 +202,12 @@ module RubyLLM
               data[:tenant_id] = context.tenant_id
             end
 
+            # Include agent-defined metadata so it appears on the dashboard immediately
+            agent_meta = safe_agent_metadata(context)
+            if agent_meta.any?
+              data[:metadata] = agent_meta.transform_keys(&:to_s)
+            end
+
             data
           end
 
@@ -222,12 +228,18 @@ module RubyLLM
               attempts_count: context.attempts_made
             }
 
-            # Store niche cache key in metadata
-            merged_metadata = begin
+            # Merge metadata: agent metadata (base) < middleware metadata (overlay)
+            agent_meta = safe_agent_metadata(context)
+            merged_metadata = agent_meta.transform_keys(&:to_s)
+
+            context_meta = begin
               context.metadata.dup
             rescue
               {}
             end
+            context_meta.transform_keys!(&:to_s)
+            merged_metadata.merge!(context_meta)
+
             if context.cached? && context[:cache_key]
               merged_metadata["response_cache_key"] = context[:cache_key]
             end
@@ -321,11 +333,18 @@ module RubyLLM
           # @param status [String] "success" or "error"
           # @return [Hash] Execution data
           def build_execution_data(context, status)
-            merged_metadata = begin
+            # Merge metadata: agent metadata (base) < middleware metadata (overlay)
+            agent_meta = safe_agent_metadata(context)
+            merged_metadata = agent_meta.transform_keys(&:to_s)
+
+            context_meta = begin
               context.metadata.dup
             rescue
               {}
             end
+            context_meta.transform_keys!(&:to_s)
+            merged_metadata.merge!(context_meta)
+
             if context.cached? && context[:cache_key]
               merged_metadata["response_cache_key"] = context[:cache_key]
             end
@@ -426,6 +445,24 @@ module RubyLLM
             params
           end
 
+          # Safely retrieves custom metadata from the agent instance
+          #
+          # Returns an empty hash if the agent doesn't define metadata,
+          # the method raises, or the result isn't a Hash.
+          #
+          # @param context [Context] The execution context
+          # @return [Hash] Agent-defined metadata, or empty hash
+          def safe_agent_metadata(context)
+            return {} unless context.agent_instance
+            return {} unless context.agent_instance.respond_to?(:metadata)
+
+            result = context.agent_instance.metadata
+            result.is_a?(Hash) ? result : {}
+          rescue => e
+            debug("Failed to retrieve agent metadata: #{e.message}")
+            {}
+          end
+
           # Sensitive parameter keys that should be redacted
           SENSITIVE_KEYS = %w[
             password token api_key secret credential auth key

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: ruby_llm-agents
 version: !ruby/object:Gem::Version
-  version: 3.5.4
+  version: 3.5.5
 platform: ruby
 authors:
 - adham90