payloop 0.0.3 → 0.0.4

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 79647d261089034fcabaf8bc8f02468c22a4dafac253f0e3b387bf169f0fe410
-  data.tar.gz: 5d7796139816c6f880d0acfa0e364065480835def4e1f98fa9fd30004a1cab6b
+  metadata.gz: 0d134effe2b909cc928b4dfc80149d3c9dcde238be7630bb37dba541fa33be0a
+  data.tar.gz: ddf58b4672381d25ab90b349c6bf29391e4fe099f9e304ba20f5169d3ee6bf1c
 SHA512:
-  metadata.gz: e87d2509afeee593b39cd9ab7a519f2fde382aaa939fecaa370cf4817aa38009a5c68a3550444052dbf0867a6162c6a3cd5771cfb039a4f812b280a9e3bb29f6
-  data.tar.gz: d7ac74462b724bce7178e6588d41bb6e91e4bf5ec98ff067bd41164cf17b4fd570892eacdb633b797fc3aab1857503cd4cf88998ad95d675be647980fd0f85d0
+  metadata.gz: 3b943601582c65f45f66209c6c7aa37c0041f3f961130b3fce5109f09369359d05cc2c7264ca85ffea99b067c95b7c1419d165bf11d847fb994f43a29697c4df
+  data.tar.gz: 2d4ff576bccc13c526dc80ae8ab9b2dfb72a9480f3f6c88ed261a723a217271867a3f6f88e2083f8bb5a65813362b5603e8396bd5a6c944885cbf1c4010fe963

data/CHANGELOG.md

@@ -29,4 +29,9 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 ## [0.0.3] - 2025-10-27
 
 ### Fixed
-- Corrected Payloop request payload structure for Google GenAI
+- Correct Payloop request payload structure for Google GenAI
+
+## [0.0.4] - 2025-10-28
+
+### Fixed
+- Correct Payloop request payload structure for OpenAI streaming

data/lib/payloop/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module Payloop
-  VERSION = "0.0.3"
+  VERSION = "0.0.4"
 end

data/lib/payloop/wrappers/base.rb

@@ -84,11 +84,62 @@ module Payloop
         collector.submit_async(payload)
       end
 
+      def payloop_merge_streaming_chunk(accumulated, chunk)
+        chunk.each do |key, value|
+          if accumulated.key?(key)
+            case accumulated[key]
+            when Hash
+              payloop_merge_streaming_chunk(accumulated[key], value) if value.is_a?(Hash)
+            when Array
+              # Concatenate arrays (matches Python SDK behavior: data[key].extend(chunk_value))
+              accumulated[key].concat(value) if value.is_a?(Array)
+            else
+              accumulated[key] = value
+            end
+          else
+            accumulated[key] = value
+          end
+        end
+        accumulated
+      end
+
+      def payloop_normalize_openai_chunk(chunk)
+        # Normalize OpenAI streaming chunks to match Python SDK format
+        # Ensure delta objects have all keys present with null values
+        return chunk unless chunk.is_a?(Hash)
+
+        if chunk["choices"].is_a?(Array)
+          chunk["choices"].each do |choice|
+            next unless choice.is_a?(Hash) && choice.key?("delta")
+
+            delta = choice["delta"]
+            if delta.is_a?(Hash)
+              # Add missing keys with nil values to match Python SDK
+              delta["role"] ||= nil unless delta.key?("role")
+              delta["content"] ||= nil unless delta.key?("content")
+              delta["refusal"] ||= nil unless delta.key?("refusal")
+              delta["tool_calls"] ||= nil unless delta.key?("tool_calls")
+              delta["function_call"] ||= nil unless delta.key?("function_call")
+            end
+          end
+        end
+
+        chunk
+      end
+
       private
 
       def extract_query(_method, _args, kwargs)
         # Deep copy to avoid mutation issues
-        deep_copy(kwargs)
+        query = deep_copy(kwargs)
+
+        # Normalize stream parameter: convert Proc to boolean true
+        # This ensures streaming requests are properly serialized to JSON
+        if query.is_a?(Hash) && query[:stream].is_a?(Proc)
+          query[:stream] = true
+        end
+
+        query
       rescue StandardError
         {}
       end

data/lib/payloop/wrappers/openai.rb

@@ -45,15 +45,38 @@ module Payloop
           define_method(:chat) do |parameters: {}|
             start_time = Time.now
 
+            # Detect streaming and wrap callback to accumulate chunks
+            streaming = parameters[:stream].is_a?(Proc)
+            accumulated_response = {} if streaming
+
+            if streaming
+              # Configure streaming to include usage (matches Python SDK behavior)
+              parameters[:stream_options] ||= {}
+              parameters[:stream_options][:include_usage] = true
+
+              user_callback = parameters[:stream]
+              parameters[:stream] = proc do |chunk, bytesize|
+                # Normalize chunk to match Python SDK format (add missing delta keys with nil)
+                normalized_chunk = payloop_normalize_openai_chunk(chunk)
+                # Accumulate chunk (merge into accumulated_response)
+                payloop_merge_streaming_chunk(accumulated_response, normalized_chunk)
+                # Call user's original callback with original chunk (don't modify user's data)
+                user_callback.call(chunk, bytesize)
+              end
+            end
+
             # Call original method
             response = original_chat(parameters: parameters)
 
+            # Use accumulated response for streaming, otherwise use returned response
+            final_response = streaming ? accumulated_response : response
+
             # Submit analytics
             payloop_submit_analytics(
               method: :chat,
               args: [],
               kwargs: parameters,
-              response: response,
+              response: final_response,
               start_time: start_time,
               end_time: Time.now,
               title: OPENAI_CLIENT_TITLE

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: payloop
 version: !ruby/object:Gem::Version
-  version: 0.0.3
+  version: 0.0.4
 platform: ruby
 authors:
 - Payloop
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2025-10-27 00:00:00.000000000 Z
+date: 2025-10-29 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: concurrent-ruby