ruby-openai-swarm 0.5.1 → 0.5.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 326dc0fd2fb1f748c48c67f3f610e7dd3d14911f712e4016ab8cd13a19d6fd6e
-  data.tar.gz: 4d12edacb8d5daa6fe863daa72a06b1f3cad8ded0c01a2c882995ac8b23e1561
+  metadata.gz: 2d9005515439f2abbfdc06d0817942f46a0f995762c0cdabd278f526d0d8cafa
+  data.tar.gz: 35d99ba8158c440803dc1d4d2579a1f025298ed14e1e44d0524149d7309f56ac
 SHA512:
-  metadata.gz: 584b8772071a78dfdf43eb33f1e3d31c7c0b47618ffe4d766d07bfdd305781975655a8a1c7b9e9dc4f0e6dee8766881ce3e921ac063bff7a2eb6c7f4dbc1e1bf
-  data.tar.gz: 99c9ff59009c4822397299566a19b0ac07d02d81b47ad03c1449dd7700ca3e9ed5ea62ec1d75223fb1f6ef1c27d44efe93d02717d587527c30c7b41541f2e8b5
+  metadata.gz: 8fbed7f9fc741e9ac0ff68aea0379244ddbfed36d9d9b1f42f40749db459a13c7e2fea90928710f243c9070aa1999197d5466caeb0a8a196f33a56b3ea65b9fb
+  data.tar.gz: 51eba945de062849593f0dcd49bbd98551213edab5bc2599dba2b42ceebdd8f29483606b239878fea641f3669888cf5ada02a5d003a049ed01a280f1626eca8b

data/lib/ruby-openai-swarm/core.rb

@@ -28,7 +28,7 @@ module OpenAISwarm
     #   )
     # end
 
-    def get_chat_completion(agent_tracker, history, context_variables, model_override, stream, debug)
+    def get_chat_completion(agent_tracker, history, context_variables, model_override, stream, debug, metadata = nil)
       agent = agent_tracker.current_agent
       context_variables = context_variables.dup
       instructions = agent.instructions.respond_to?(:call) ? agent.instructions.call(context_variables) : agent.instructions
@@ -56,6 +56,14 @@ module OpenAISwarm
         tools: Util.request_tools_excluded(tools, agent_tracker.tracking_agents_tool_name, agent.strategy.prevent_agent_reentry),
       }
 
+      # Add metadata if provided
+      # Add support for LiteLLM observability with Langfuse
+      # 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?
       # create_params[:function_call] = agent.tool_choice if agent.tool_choice
@@ -168,7 +176,7 @@ module OpenAISwarm
       partial_response
     end
 
-    def run(agent:, messages:, context_variables: {}, model_override: nil, stream: false, debug: false, max_turns: Float::INFINITY, execute_tools: true)
+    def run(agent:, messages:, context_variables: {}, model_override: nil, stream: false, debug: false, max_turns: Float::INFINITY, execute_tools: true, metadata: nil)
       agent_tracker = OpenAISwarm::Agents::ChangeTracker.new(agent)
       if stream
         return run_and_stream(
@@ -178,7 +186,8 @@ module OpenAISwarm
           model_override: model_override,
           debug: debug,
           max_turns: max_turns,
-          execute_tools: execute_tools
+          execute_tools: execute_tools,
+          metadata: metadata
         )
       end
 
@@ -197,7 +206,8 @@ module OpenAISwarm
           context_variables,
           model_override,
           stream,
-          debug
+          debug,
+          metadata
         )
 
         message = completion.dig('choices', 0, 'message') || {}
@@ -237,7 +247,7 @@ module OpenAISwarm
     end
 
     # TODO(Grayson): a lot of copied code here that will be refactored
-    def run_and_stream(agent:, messages:, context_variables: {}, model_override: nil, debug: false, max_turns: Float::INFINITY, execute_tools: true)
+    def run_and_stream(agent:, messages:, context_variables: {}, model_override: nil, debug: false, max_turns: Float::INFINITY, execute_tools: true, metadata: nil)
       agent_tracker = OpenAISwarm::Agents::ChangeTracker.new(agent)
       active_agent = agent
       context_variables = context_variables.dup
@@ -255,7 +265,8 @@ module OpenAISwarm
           context_variables,
           model_override,
           true, # stream
-          debug
+          debug,
+          metadata
         )
 
         yield({ delim: "start" }) if block_given?

data/lib/ruby-openai-swarm/version.rb

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

data/lib/ruby-openai-swarm.rb

@@ -15,7 +15,7 @@ require 'ruby-openai-swarm/memory'
 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;

data/ruby-openai-swarm.gemspec

@@ -21,8 +21,9 @@ Gem::Specification.new do |spec|
   end
 
   spec.require_paths = ["lib"]
-  spec.add_dependency "ruby-openai", "~> 7.3"
+  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"

metadata

@@ -1,29 +1,35 @@
 --- !ruby/object:Gem::Specification
 name: ruby-openai-swarm
 version: !ruby/object:Gem::Version
-  version: 0.5.1
+  version: 0.5.3
 platform: ruby
 authors:
 - Grayson Chen
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2025-03-03 00:00:00.000000000 Z
+date: 2025-03-31 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: ruby-openai
   requirement: !ruby/object:Gem::Requirement
     requirements:
-    - - "~>"
+    - - ">="
       - !ruby/object:Gem::Version
         version: '7.3'
+    - - "<"
+      - !ruby/object:Gem::Version
+        version: '9.0'
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
-    - - "~>"
+    - - ">="
       - !ruby/object:Gem::Version
         version: '7.3'
+    - - "<"
+      - !ruby/object:Gem::Version
+        version: '9.0'
 - !ruby/object:Gem::Dependency
   name: ostruct
   requirement: !ruby/object:Gem::Requirement
@@ -38,6 +44,20 @@ dependencies:
     - - ">="
       - !ruby/object:Gem::Version
         version: '0'
+- !ruby/object:Gem::Dependency
+  name: activesupport
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
 - !ruby/object:Gem::Dependency
   name: rspec
   requirement: !ruby/object:Gem::Requirement
@@ -159,7 +179,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.5.3
+rubygems_version: 3.1.4
 signing_key:
 specification_version: 4
 summary: A Ruby implementation of OpenAI function calling swarm