payloop 0.0.2 → 0.0.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: f4dd3be8a2eac5f3cda3b04fbc7d73d2fa6c9b4f35d640840a90f6a872e90a58
-  data.tar.gz: e348e027345bc09b1233e976c38871747410d3caccac959fc119c49143f14df4
+  metadata.gz: 79647d261089034fcabaf8bc8f02468c22a4dafac253f0e3b387bf169f0fe410
+  data.tar.gz: 5d7796139816c6f880d0acfa0e364065480835def4e1f98fa9fd30004a1cab6b
 SHA512:
-  metadata.gz: 6d030e46f95833e825651eb0c11e23116eff4f542b10784e96aff2031892c408294115bfae3c5544f37a25b32d4435092d5b17940373f4b0968a2684657611c7
-  data.tar.gz: dc86258b883762641d561b39261bbfc0b66b494f15f5d891446e00b4d786434f5684986d01d7bfb69c989a7db54991e290656fc04b59429c1ff8dad6c034c28d
+  metadata.gz: e87d2509afeee593b39cd9ab7a519f2fde382aaa939fecaa370cf4817aa38009a5c68a3550444052dbf0867a6162c6a3cd5771cfb039a4f812b280a9e3bb29f6
+  data.tar.gz: d7ac74462b724bce7178e6588d41bb6e91e4bf5ec98ff067bd41164cf17b4fd570892eacdb633b797fc3aab1857503cd4cf88998ad95d675be647980fd0f85d0

data/CHANGELOG.md

@@ -25,3 +25,8 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 
 ### Fixed
 - Corrected Payloop request payload structure as needed
+
+## [0.0.3] - 2025-10-27
+
+### Fixed
+- Corrected Payloop request payload structure for Google GenAI

data/lib/payloop/client.rb

@@ -36,12 +36,7 @@ module Payloop
     end
 
     # Set attribution for cost tracking
-    def attribution(parent_id:, parent_name: nil, subsidiary_id: nil, subsidiary_name: nil,
-                    parent_uuid: nil, subsidiary_uuid: nil)
-      # Support deprecated parameters
-      parent_id ||= parent_uuid
-      subsidiary_id ||= subsidiary_uuid
-
+    def attribution(parent_id:, parent_name: nil, subsidiary_id: nil, subsidiary_name: nil)
       attr = Attribution.new(
         parent_id: parent_id,
         parent_name: parent_name,

data/lib/payloop/version.rb

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

data/lib/payloop/wrappers/base.rb

@@ -100,16 +100,49 @@ module Payloop
         when String
           { text: response }
         else
-          if response.respond_to?(:to_h)
+          # Try various serialization methods
+          if response.respond_to?(:as_json)
+            deep_copy(response.as_json)
+          elsif response.respond_to?(:to_h)
             deep_copy(response.to_h)
           elsif response.respond_to?(:to_hash)
             deep_copy(response.to_hash)
           else
-            { raw: response.to_s }
+            # Extract instance variables for objects without serialization methods
+            extract_instance_variables(response)
+          end
+        end
+      rescue StandardError => e
+        # If extraction fails, try to provide useful debug info
+        { raw: response.to_s, error: e.message }
+      end
+
+      def extract_instance_variables(obj)
+        result = {}
+        obj.instance_variables.each do |var|
+          key = var.to_s.delete("@").to_sym
+          value = obj.instance_variable_get(var)
+          result[key] = serialize_value(value)
+        end
+        result
+      end
+
+      def serialize_value(value)
+        case value
+        when Hash
+          value.transform_values { |v| serialize_value(v) }
+        when Array
+          value.map { |v| serialize_value(v) }
+        when String, Numeric, TrueClass, FalseClass, NilClass
+          value
+        else
+          # Recursively extract instance variables for nested objects
+          if value.respond_to?(:instance_variables)
+            extract_instance_variables(value)
+          else
+            value.to_s
           end
         end
-      rescue StandardError
-        { raw: response.to_s }
       end
 
       def deep_copy(obj)

data/lib/payloop/wrappers/google.rb

@@ -4,7 +4,7 @@ require_relative "constants"
 
 module Payloop
   module Wrappers
-    # Wrapper for Google GenerativeAI Ruby client
+    # Wrapper for Google GenerativeAI Ruby client (google-genai gem)
     class Google
       def initialize(config, collector)
         @config = config
@@ -17,6 +17,9 @@ module Payloop
         # Prevent double registration
         return client if client.instance_variable_defined?(:@payloop_registered)
 
+        # Patch response class after client is validated (ensures gem is loaded)
+        patch_response_class!
+
         # Store references in client instance
         client.instance_variable_set(:@payloop_config, @config)
         client.instance_variable_set(:@payloop_collector, @collector)
@@ -37,6 +40,20 @@ module Payloop
               "Client does not appear to be a valid Google GenAI client (missing models method)"
       end
 
+      # Monkey-patch the GenerateContentResponse class to include missing fields
+      def patch_response_class!
+        # Check if the constant exists and hasn't been patched yet
+        return unless defined?(::Google::Genai::Types::GenerateContentResponse)
+        return if ::Google::Genai::Types::GenerateContentResponse.instance_variable_defined?(:@_payloop_patched)
+
+        ::Google::Genai::Types::GenerateContentResponse.class_eval do
+          # Use camelCase to match Google's API response keys
+          attr_accessor :modelVersion, :usageMetadata, :responseId, :createTime
+        end
+
+        ::Google::Genai::Types::GenerateContentResponse.instance_variable_set(:@_payloop_patched, true)
+      end
+
       def wrap_generate_content_method(client)
         # Get the models resource
         models_resource = client.models

metadata

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