ruby-openai-swarm 0.3.1 → 0.4.0.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 984c58d324ae34f1a8eec694c3bca07aaf44420ce9ce322ceffbbc8c722e0ddd
-  data.tar.gz: 4ca2dd01658cbf5091aee65e41951fc64d5f16a72ea93c9dafe4a3802d38cad2
+  metadata.gz: 455b72e5872a2f9e11dad1509769c5e3037cef23b3a622337fa419f9c8a2ab20
+  data.tar.gz: 8f51f50b6ee3de651073783610d26f1394f2fbdb130aa8926b5184af921cfce0
 SHA512:
-  metadata.gz: 5a6357efdc1963cfd240fa23aaf3211b89cea7c39e04b4101a4ff2f25e48eb6810cc281a29977fcf5cc2ef33c23ffde4667334cc34374408fff8791c656273b4
-  data.tar.gz: b241abf134b927a950e50bdd4f2bfb06a6217a5ac5b9ea5c80d306bfcce2dd24e7c7b6f1ff85c8fd6700ac3a10080cccd13354f2f15afa6533d466d46014c7a7
+  metadata.gz: fc109240efa4af18db7d9d2d60cf9e22fd55c36a9b9845617aa23a3ea912610001c98e0567cd34abd8dcf5e3f1cb86b85b0d474d356023912add3e2c730ad6cf
+  data.tar.gz: 2d5d5740139edc78c5de1116c41e57397b3fce75f114f0459889c2f49c35846473a38fe35048977a5cecb79cb27d9af7d98c9fce48fa8926666165f6766f1dd3

data/Gemfile.lock

@@ -1,7 +1,7 @@
 PATH
   remote: .
   specs:
-    ruby-openai-swarm (0.3.1)
+    ruby-openai-swarm (0.4.0.1)
       ruby-openai (~> 7.3)
 
 GEM
@@ -15,8 +15,8 @@ GEM
       base64
       faraday-net_http (>= 2.0, < 3.1)
       ruby2_keywords (>= 0.0.4)
-    faraday-multipart (1.0.4)
-      multipart-post (~> 2)
+    faraday-multipart (1.1.0)
+      multipart-post (~> 2.0)
     faraday-net_http (3.0.2)
     method_source (1.1.0)
     multipart-post (2.4.1)

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

@@ -51,9 +51,11 @@ module OpenAISwarm
 
       if stream
         return Enumerator.new do |yielder|
+          yielder << { 'parameters' => create_params }
+
           @client.chat(parameters: create_params.merge(
             stream: proc do |chunk, _bytesize|
-              yielder << chunk
+              yielder << { chunk: chunk }
             end
           ))
         end
@@ -238,7 +240,15 @@ module OpenAISwarm
         )
 
         yield({ delim: "start" }) if block_given?
-        completion.each do |chunk|
+        completion.each do |stream|
+
+          # TODO(Grayson): will refactor it
+          if stream['parameters']
+            yield({ 'parameters' => stream['parameters'], 'agent' => active_agent&.name }) if block_given?
+          end
+          next if stream.key?('parameters')
+
+          chunk = stream[:chunk]
           if chunk['error']
             details = {
               'response' =>
@@ -255,7 +265,7 @@ module OpenAISwarm
             delta['sender'] = active_agent.name
           end
 
-          yield delta if block_given?
+          yield({ 'delta' => delta })  if block_given?
 
           delta.delete('role')
           delta.delete('sender')

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

@@ -4,21 +4,24 @@ module OpenAISwarm
       def process_and_print_streaming_response(response)
         content = []
         last_sender = ""
-        response.each do |chunk|
-          last_sender = chunk['sender'] if chunk.key?('sender')
+        response.each do |stream|
+          delta = stream['delta']
+          if delta
+            last_sender = delta['sender'] if delta.key?('sender')
 
-          if chunk.key?("content") && !chunk["content"].nil?
-            if content.empty? && !last_sender.empty?
-              puts
-              print "\033[94m#{last_sender}:\033[0m "
-              last_sender = ""
+            if delta.key?("content") && !delta["content"].nil?
+              if content.empty? && !last_sender.empty?
+                puts
+                print "\033[94m#{last_sender}:\033[0m "
+                last_sender = ""
+              end
+              print delta["content"]
+              content << delta["content"]
             end
-            print chunk["content"]
-            content << chunk["content"]
           end
 
-          if chunk.key?("tool_calls") && !chunk["tool_calls"].nil?
-            chunk["tool_calls"].each do |tool_call|
+          if stream.key?("tool_calls") && !stream["tool_calls"].nil?
+            stream["tool_calls"].each do |tool_call|
               f = tool_call["function"]
               name = f["name"]
               next if name.nil?
@@ -26,11 +29,11 @@ module OpenAISwarm
             end
           end
 
-          if chunk.key?("delim") && chunk["delim"] == "end" && !content.empty?
+          if stream.key?("delim") && stream["delim"] == "end" && !content.empty?
             puts
             content = ""
           end
-          return chunk["response"] if chunk.key?("response")
+          return stream["response"] if stream.key?("response")
         end
       end
 

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

@@ -1,3 +1,3 @@
 module OpenAISwarm
-  VERSION = "0.3.1"
+  VERSION = "0.4.0.1"
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: ruby-openai-swarm
 version: !ruby/object:Gem::Version
-  version: 0.3.1
+  version: 0.4.0.1
 platform: ruby
 authors:
 - Grayson
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2024-12-18 00:00:00.000000000 Z
+date: 2024-12-26 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: ruby-openai