ruby-openai-swarm 0.3.1 → 0.4.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 984c58d324ae34f1a8eec694c3bca07aaf44420ce9ce322ceffbbc8c722e0ddd
-  data.tar.gz: 4ca2dd01658cbf5091aee65e41951fc64d5f16a72ea93c9dafe4a3802d38cad2
+  metadata.gz: 49bcf71e27c830e373b9732691667aa4ff8657dc944a8dc113fb4b13e807ccd6
+  data.tar.gz: 38cd26d3670e7a49652b8b52677a2a7bf722c9f5635bfef55bea296d0502c4d1
 SHA512:
-  metadata.gz: 5a6357efdc1963cfd240fa23aaf3211b89cea7c39e04b4101a4ff2f25e48eb6810cc281a29977fcf5cc2ef33c23ffde4667334cc34374408fff8791c656273b4
-  data.tar.gz: b241abf134b927a950e50bdd4f2bfb06a6217a5ac5b9ea5c80d306bfcce2dd24e7c7b6f1ff85c8fd6700ac3a10080cccd13354f2f15afa6533d466d46014c7a7
+  metadata.gz: 0f4b422f084d400ef6eea393f20a650fd02fa8b96370e4781c4228113650763c7eb84c6c1aac21daceb8251cfd83d69ec69b536dff16c3bd7be3e8b0f22a353a
+  data.tar.gz: f24c83ea576a7d305069eda9da308f0d39b257065897377b228c785049c41254d3889d4e82b96d40755151b801b95d3a6f8ae5550d73fb871c902c9b21755793

data/Gemfile.lock

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

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'] }) 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"
 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
 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