ruby-openai-swarm 0.2.1 → 0.2.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: b48c60ccd21aaf774b8c7b8a325f42e3dc536d7c8ccea30930fd20f15e5d8903
-  data.tar.gz: 9962aca258023744e3e6cae3ccebb2f1212f8b2136ab7546fb92f9f919917e5a
+  metadata.gz: ccab6a10acd813cb4a174589038d87cf1eb63d27e323f5333acf6ae53a663985
+  data.tar.gz: 78737dfe700a187d8b598ede955598971aeac20a26f0ddd793a4fea18872410d
 SHA512:
-  metadata.gz: d067c62379602b8f0b267eaf4168f14faf96112444f1abe7731145c99f34efe19045d967c2d90e854bb277e64e8659f8a7e7884dfc4f2a5aca990f3d786c2944
-  data.tar.gz: 77eaae12e12a5dd0a3db95737c9221ed867ea0aacb40847020da7de42631c0f51414aeb3b8901658be82b3b5cebc6b76045d2a8f2b20262fb8772ed1812b51b5
+  metadata.gz: '097fefd36ba24bcfa9a566e5cce325b2baf8aa1ae62a7382046752e1c1872ece6a12496ada95db3f089f54b1d7f4a0aff253bd2fc734ba46083f03b04d4c46bc'
+  data.tar.gz: 963693d7321e959d1fd44fea1f70cda22cc789f76e544af2481a4d4134b908d698c32343e002ea59acbf00f495c3e48f9b18fd6bb53fbd72d422048054e822a7

data/Gemfile.lock

@@ -1,7 +1,7 @@
 PATH
   remote: .
   specs:
-    ruby-openai-swarm (0.1.0)
+    ruby-openai-swarm (0.2.3)
       ruby-openai (~> 7.3)
 
 GEM

data/examples/airline/main.rb

@@ -65,4 +65,4 @@ params:
 GUIDE_EXAMPLES
 puts guide_examples
 
-OpenAISwarm::Repl.run_demo_loop(triage_agent, context_variables: context_variables, debug: env_debug)
+OpenAISwarm::Repl.run_demo_loop(triage_agent, context_variables: context_variables, stream: true, debug: env_debug)

data/examples/basic/function_calling.rb

@@ -6,7 +6,20 @@ def get_weather(location:)
   "{'temp':67, 'unit':'F'}"
 end
 
-function_instance = OpenAISwarm::FunctionDescriptor.new(
+def get_news(category:)
+  [
+    "Tech Company A Acquires Startup B",
+    "New AI Model Revolutionizes Industry",
+    "Breakthrough in Quantum Computing"
+  ].sample
+end
+
+get_news_instance = OpenAISwarm::FunctionDescriptor.new(
+  target_method: :get_news,
+  description: 'Get the latest news headlines. The category of news, e.g., world, business, sports.'
+)
+
+get_weather_instance = OpenAISwarm::FunctionDescriptor.new(
   target_method: :get_weather,
   description: 'Simulate fetching weather data'
 )
@@ -15,16 +28,50 @@ agent = OpenAISwarm::Agent.new(
   name: "Agent",
   instructions: "You are a helpful agent.",
   model: "gpt-4o-mini",
-  functions: [function_instance]
+  functions: [
+    get_weather_instance,
+    get_news_instance
+  ]
 )
-# debugger logger: {:model=>"gpt-4o-mini", :messages=>[{:role=>"system", :content=>"You are a helpful agent."}, {"role"=>"user", "content"=>"What's the weather in NYC?"}], :tools=>[{:type=>"function", :function=>{:name=>"get_weather", :description=>"", :parameters=>{:type=>"object", :properties=>{:location=>{:type=>"string"}}, :required=>["location"]}}}], :stream=>false, :parallel_tool_calls=>true}
-response = client.run(
-  messages: [{"role" => "user", "content" => "What's the weather in NYC?"}],
+
+guide_examples = <<~GUIDE_EXAMPLES
+############# GUIDE_EXAMPLES #####################################
+examples:
+  What's the weather in NYC?
+
+  Tell me the weather in New York and the latest news headlines.
+
+Details:
+	1. Single Function Call
+	     Example: “What’s the weather in NYC?”
+	     Action: Calls get_weather with location “New York City”.
+	     Response: Only provides weather details.
+
+	2. Multiple Function Calls
+	     Example: “Tell me the weather in New York and the latest news headlines.”
+	     Action: Calls get_weather for weather and get_news for news.
+	     Response: Combines weather and news information.
+
+params:
+  `DEBUG=1 ruby examples/basic/function_calling.rb` # turn on debug (default turn off)
+################################################################
+GUIDE_EXAMPLES
+
+puts guide_examples
+
+# OpenAISwarm::Repl.run_demo_loop(agent, stream: true, debug: env_debug)
+
+OpenAISwarm.new.run_and_stream(
   agent: agent,
   debug: true,
-)
-
-pp response.messages.last
+  messages: [{"role" => "user", "content" => "Tell me the weather in New York and the latest news headlines."}]
+) do |chunk|
+  if chunk.key?("content") && !chunk["content"].nil?
+    puts ">>>#{chunk}"
+  end
 
-# print(response.messages[-1]["content"])
-# The current temperature in New York City is 67°F. => nil
+  if chunk.key?('response')
+    binding.pry
+    # log_llm_request(chunk['response'])
+  end
+end

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

@@ -1,6 +1,13 @@
 module OpenAISwarm
   class Agent
-    attr_accessor :name, :model, :instructions, :functions, :tool_choice, :parallel_tool_calls
+    attr_accessor :name, :model, :instructions,
+                  :functions, :tool_choice,
+                  :parallel_tool_calls,
+                  :resource
+    # These attributes can be read and written externally. They include:
+    # - name: The name of the agent.
+    # - model: The model used, e.g., "gpt-4".
+    # - resource: Additional custom parameters or data that the agent might need.
 
     def initialize(
       name: "Agent",
@@ -8,7 +15,8 @@ module OpenAISwarm
       instructions: "You are a helpful agent.",
       functions: [],
       tool_choice: nil,
-      parallel_tool_calls: true
+      parallel_tool_calls: true,
+      resource: nil
     )
       @name = name
       @model = model
@@ -16,6 +24,7 @@ module OpenAISwarm
       @functions = functions
       @tool_choice = tool_choice
       @parallel_tool_calls = parallel_tool_calls
+      @resource = resource
     end
   end
 end

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

@@ -102,10 +102,10 @@ module OpenAISwarm
         unless function_map.key?(name)
           Util.debug_print(debug, "Tool #{name} not found in function map.")
           partial_response.messages << {
-            role: 'tool',
-            tool_call_id: tool_call['id'],
-            tool_name: name,
-            content: "Error: Tool #{name} not found."
+            'role' => 'tool',
+            'tool_call_id' => tool_call['id'],
+            'tool_name' => name,
+            'content' => "Error: Tool #{name} not found."
           }
           next
         end
@@ -123,10 +123,10 @@ module OpenAISwarm
         result = handle_function_result(raw_result, debug)
 
         partial_response.messages << {
-          role: 'tool',
-          tool_call_id: tool_call['id'],
-          tool_name: name,
-          content: result.value
+          'role' => 'tool',
+          'tool_call_id' => tool_call['id'],
+          'tool_name' => name,
+          'content' => result.value
         }
 
         partial_response.context_variables.merge!(result.context_variables)

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

@@ -1,3 +1,3 @@
 module OpenAISwarm
-  VERSION = "0.2.1"
+  VERSION = "0.2.3"
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: ruby-openai-swarm
 version: !ruby/object:Gem::Version
-  version: 0.2.1
+  version: 0.2.3
 platform: ruby
 authors:
 - Grayson
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2024-11-05 00:00:00.000000000 Z
+date: 2024-11-27 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: ruby-openai