Merge pull request #34 from graysonchen/feature/metadata_for_agent_name

Feature/metadata for agent name

Author: graysonchen

lib/ruby-openai-swarm.rb

@@ -15,7 +15,7 @@
 require 'ruby-openai-swarm/memories/entity_store'
 require 'ruby-openai-swarm/memories/core_memory_function'
 require 'ruby-openai-swarm/memories/field'
-
+require 'active_support/core_ext/hash/deep_transform_values'
 
 module OpenAISwarm
   class Error < StandardError;

lib/ruby-openai-swarm/core.rb

@@ -58,8 +58,11 @@ def get_chat_completion(agent_tracker, history, context_variables, model_overrid
 
       # Add metadata if provided
       # Add support for LiteLLM observability with Langfuse
-      # See: https://docs.litellm.ai/docs/observability/langfuse_integration      
-      create_params[:metadata] = metadata if metadata
+      # See: https://docs.litellm.ai/docs/observability/langfuse_integration
+      if metadata && metadata.is_a?(Hash)
+        metadata_hash = metadata.deep_transform_values { |val| val.to_s.to_sym == :agent_name ? agent&.name : val }
+        create_params[:metadata] = metadata_hash
+      end
 
       # TODO: https://platform.openai.com/docs/guides/function-calling/how-do-functions-differ-from-tools
       # create_params[:functions] = tools unless tools.empty?

lib/ruby-openai-swarm/version.rb

@@ -1,3 +1,3 @@
 module OpenAISwarm
-  VERSION = "0.5.2"
+  VERSION = "0.5.3"
 end

ruby-openai-swarm.gemspec

@@ -23,6 +23,7 @@ Gem::Specification.new do |spec|
   spec.require_paths = ["lib"]
   spec.add_dependency "ruby-openai", ">= 7.3", "< 9.0"
   spec.add_dependency "ostruct"
+  spec.add_dependency "activesupport"
   spec.add_development_dependency "rspec", "~> 3.0"
   spec.add_development_dependency "rake", "~> 13.0"
   spec.add_development_dependency "pry"

spec/lib/ruby-openai-swarm/core_spec.rb

@@ -48,6 +48,25 @@
       )
     end
 
+    it 'transforms var_agent_name in metadata to actual agent name' do
+      core = described_class.new(client)
+      metadata_with_var = { user_id: 123, agent_name: :agent_name }
+      
+      expect(client).to receive(:chat) do |params|
+        expect(params[:parameters][:metadata]).to eq({
+          user_id: 123,
+          agent_name: "TestAgent"
+        })
+        chat_response
+      end
+
+      core.run(
+        agent: agent,
+        messages: messages,
+        metadata: metadata_with_var
+      )
+    end
+
     it 'works without metadata' do
       core = described_class.new(client)