paid_ruby 0.1.1 → 0.1.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: e08b12d8087ea612ca1b405c5d6cfa6fab4374f4f89ad81df1a7042b31206c72
-  data.tar.gz: 7dd1f1ef4c71fe4dd19d3c34e6eacd051a95087b74ec5be2f610efe9608e2a55
+  metadata.gz: 1b2054795e06be38ede1daa5465842367cfef452cd91ecc6b1122a66a8942d99
+  data.tar.gz: ba513d072353748c62740300308ce64102e215441123d037891dae820f000231
 SHA512:
-  metadata.gz: 6eed37d4f15ac6939c644a4c7db14fbcaeb3d5e9e67d16b983d3060790f0daeb6054b1e5beb0241470bdaeca393bdc119503dc2ffd893272e9d522c62031059c
-  data.tar.gz: 931419aa3d4767c1df1f7c0ca332db5e629b73f7d38b3f8cdc61463626c726476ac89f327434ecd08bebdc631c8dbcc726484b925a6d4f3a821695b5220a4684
+  metadata.gz: '0792683c410f927576c39efef7549ce4dc88b2730f681142f4556620af55bd283e391ce501adc85ba453555cafb7a1bbabf11c1e2c45613b74adbff4700d0828'
+  data.tar.gz: 6f6f145c2f60fa277a67bf8ff983de8f0ae5877fd2dcf3519afb8756805e5323d8ee373ddcbdcfaafc511121622b7f978ca5a281c3e13797de5196f71dd89a48

data/lib/gemconfig.rb

@@ -9,5 +9,6 @@ module Paid
     HOMEPAGE = "https://github.com/paid-ai/paid-ruby"
     SOURCE_CODE_URI = "https://github.com/paid-ai/paid-ruby"
     CHANGELOG_URI = "https://github.com/paid-ai/paid-ruby/blob/master/CHANGELOG.md"
+    RUBY_VERSION = ">= 3.4.4"
   end
 end

data/lib/paid_ruby/agents/client.rb

@@ -60,6 +60,7 @@ end
     # @param description [String] 
     # @param agent_code [String] 
     # @param external_id [String] 
+    # @param active [Boolean] 
     # @param request_options [Paid::RequestOptions] 
     # @return [Paid::Agent]
     # @example
@@ -68,8 +69,12 @@ end
 #    environment: Paid::Environment::PRODUCTION,
 #    token: "YOUR_AUTH_TOKEN"
 #  )
-#  api.agents.create(name: "name", description: "description")
-    def create(name:, description:, agent_code: nil, external_id: nil, request_options: nil)
+#  api.agents.create(
+#    name: "Acme Agent",
+#    description: "Acme Agent is an AI agent that does things.",
+#    external_id: "acme-agent"
+#  )
+    def create(name:, description:, agent_code: nil, external_id: nil, active: nil, request_options: nil)
       response = @request_client.conn.post do | req |
   unless request_options&.timeout_in_seconds.nil?
     req.options.timeout = request_options.timeout_in_seconds
@@ -81,7 +86,7 @@ end
   unless request_options.nil? || request_options&.additional_query_parameters.nil?
     req.params = { **(request_options&.additional_query_parameters || {}) }.compact
   end
-  req.body = { **(request_options&.additional_body_parameters || {}), name: name, description: description, agentCode: agent_code, externalId: external_id }.compact
+  req.body = { **(request_options&.additional_body_parameters || {}), name: name, description: description, agentCode: agent_code, externalId: external_id, active: active }.compact
   req.url "#{@request_client.get_url(request_options: request_options)}/agents"
 end
       Paid::Agent.from_json(json_object: response.body)
@@ -131,7 +136,7 @@ end
 #    environment: Paid::Environment::PRODUCTION,
 #    token: "YOUR_AUTH_TOKEN"
 #  )
-#  api.agents.update(agent_id: "agentId", request: {  })
+#  api.agents.update(agent_id: "agentId", request: { name: "Acme Agent (Updated)", agent_attributes: [{ name: "Emails sent signal", active: true, pricing: { event_name: "emails_sent", taxable: true, charge_type: USAGE, pricing_model: PER_UNIT, billing_frequency: MONTHLY, price_points: { "USD": { tiers: [{ min_quantity: 0, max_quantity: 10, unit_price: 100 }, { min_quantity: 11, max_quantity: 100, unit_price: 90 }, { min_quantity: 101, unit_price: 80 }] } } } }] })
     def update(agent_id:, request:, request_options: nil)
       response = @request_client.conn.put do | req |
   unless request_options&.timeout_in_seconds.nil?
@@ -222,7 +227,7 @@ end
 #    environment: Paid::Environment::PRODUCTION,
 #    token: "YOUR_AUTH_TOKEN"
 #  )
-#  api.agents.update_by_external_id(external_id: "externalId", request: {  })
+#  api.agents.update_by_external_id(external_id: "externalId", request: { name: "Acme Agent (Updated)", agent_attributes: [{ name: "Emails sent signal", active: true, pricing: { event_name: "emails_sent", taxable: true, charge_type: USAGE, pricing_model: PER_UNIT, billing_frequency: MONTHLY, price_points: { "USD": { unit_price: 150 } } } }] })
     def update_by_external_id(external_id:, request:, request_options: nil)
       response = @request_client.conn.put do | req |
   unless request_options&.timeout_in_seconds.nil?
@@ -317,6 +322,7 @@ end
     # @param description [String] 
     # @param agent_code [String] 
     # @param external_id [String] 
+    # @param active [Boolean] 
     # @param request_options [Paid::RequestOptions] 
     # @return [Paid::Agent]
     # @example
@@ -325,8 +331,12 @@ end
 #    environment: Paid::Environment::PRODUCTION,
 #    token: "YOUR_AUTH_TOKEN"
 #  )
-#  api.agents.create(name: "name", description: "description")
-    def create(name:, description:, agent_code: nil, external_id: nil, request_options: nil)
+#  api.agents.create(
+#    name: "Acme Agent",
+#    description: "Acme Agent is an AI agent that does things.",
+#    external_id: "acme-agent"
+#  )
+    def create(name:, description:, agent_code: nil, external_id: nil, active: nil, request_options: nil)
       Async do
         response = @request_client.conn.post do | req |
   unless request_options&.timeout_in_seconds.nil?
@@ -339,7 +349,7 @@ end
   unless request_options.nil? || request_options&.additional_query_parameters.nil?
     req.params = { **(request_options&.additional_query_parameters || {}) }.compact
   end
-  req.body = { **(request_options&.additional_body_parameters || {}), name: name, description: description, agentCode: agent_code, externalId: external_id }.compact
+  req.body = { **(request_options&.additional_body_parameters || {}), name: name, description: description, agentCode: agent_code, externalId: external_id, active: active }.compact
   req.url "#{@request_client.get_url(request_options: request_options)}/agents"
 end
         Paid::Agent.from_json(json_object: response.body)
@@ -392,7 +402,7 @@ end
 #    environment: Paid::Environment::PRODUCTION,
 #    token: "YOUR_AUTH_TOKEN"
 #  )
-#  api.agents.update(agent_id: "agentId", request: {  })
+#  api.agents.update(agent_id: "agentId", request: { name: "Acme Agent (Updated)", agent_attributes: [{ name: "Emails sent signal", active: true, pricing: { event_name: "emails_sent", taxable: true, charge_type: USAGE, pricing_model: PER_UNIT, billing_frequency: MONTHLY, price_points: { "USD": { tiers: [{ min_quantity: 0, max_quantity: 10, unit_price: 100 }, { min_quantity: 11, max_quantity: 100, unit_price: 90 }, { min_quantity: 101, unit_price: 80 }] } } } }] })
     def update(agent_id:, request:, request_options: nil)
       Async do
         response = @request_client.conn.put do | req |
@@ -489,7 +499,7 @@ end
 #    environment: Paid::Environment::PRODUCTION,
 #    token: "YOUR_AUTH_TOKEN"
 #  )
-#  api.agents.update_by_external_id(external_id: "externalId", request: {  })
+#  api.agents.update_by_external_id(external_id: "externalId", request: { name: "Acme Agent (Updated)", agent_attributes: [{ name: "Emails sent signal", active: true, pricing: { event_name: "emails_sent", taxable: true, charge_type: USAGE, pricing_model: PER_UNIT, billing_frequency: MONTHLY, price_points: { "USD": { unit_price: 150 } } } }] })
     def update_by_external_id(external_id:, request:, request_options: nil)
       Async do
         response = @request_client.conn.put do | req |

data/lib/paid_ruby/contacts/client.rb

@@ -76,16 +76,13 @@ end
 #    token: "YOUR_AUTH_TOKEN"
 #  )
 #  api.contacts.create(
+#    customer_external_id: "acme-inc",
 #    salutation: MR,
-#    first_name: "firstName",
-#    last_name: "lastName",
-#    email: "email",
-#    billing_street: "billingStreet",
-#    billing_city: "billingCity",
-#    billing_country: "billingCountry",
-#    billing_postal_code: "billingPostalCode"
+#    first_name: "John",
+#    last_name: "Doe",
+#    email: "john.doe@example.com"
 #  )
-    def create(external_id: nil, customer_id: nil, customer_external_id: nil, salutation:, first_name:, last_name:, email:, phone: nil, billing_street:, billing_city:, billing_state_province: nil, billing_country:, billing_postal_code:, request_options: nil)
+    def create(external_id: nil, customer_id: nil, customer_external_id: nil, salutation:, first_name:, last_name:, email:, phone: nil, billing_street: nil, billing_city: nil, billing_state_province: nil, billing_country: nil, billing_postal_code: nil, request_options: nil)
       response = @request_client.conn.post do | req |
   unless request_options&.timeout_in_seconds.nil?
     req.options.timeout = request_options.timeout_in_seconds
@@ -283,16 +280,13 @@ end
 #    token: "YOUR_AUTH_TOKEN"
 #  )
 #  api.contacts.create(
+#    customer_external_id: "acme-inc",
 #    salutation: MR,
-#    first_name: "firstName",
-#    last_name: "lastName",
-#    email: "email",
-#    billing_street: "billingStreet",
-#    billing_city: "billingCity",
-#    billing_country: "billingCountry",
-#    billing_postal_code: "billingPostalCode"
+#    first_name: "John",
+#    last_name: "Doe",
+#    email: "john.doe@example.com"
 #  )
-    def create(external_id: nil, customer_id: nil, customer_external_id: nil, salutation:, first_name:, last_name:, email:, phone: nil, billing_street:, billing_city:, billing_state_province: nil, billing_country:, billing_postal_code:, request_options: nil)
+    def create(external_id: nil, customer_id: nil, customer_external_id: nil, salutation:, first_name:, last_name:, email:, phone: nil, billing_street: nil, billing_city: nil, billing_state_province: nil, billing_country: nil, billing_postal_code: nil, request_options: nil)
       Async do
         response = @request_client.conn.post do | req |
   unless request_options&.timeout_in_seconds.nil?

data/lib/paid_ruby/customers/client.rb

@@ -82,7 +82,7 @@ end
 #    environment: Paid::Environment::PRODUCTION,
 #    token: "YOUR_AUTH_TOKEN"
 #  )
-#  api.customers.create(name: "name")
+#  api.customers.create(name: "Acme, Inc.", external_id: "acme-inc")
     def create(name:, external_id: nil, phone: nil, employee_count: nil, annual_revenue: nil, tax_exempt_status: nil, creation_source: nil, website: nil, billing_address: nil, request_options: nil)
       response = @request_client.conn.post do | req |
   unless request_options&.timeout_in_seconds.nil?
@@ -154,7 +154,7 @@ end
 #    environment: Paid::Environment::PRODUCTION,
 #    token: "YOUR_AUTH_TOKEN"
 #  )
-#  api.customers.update(customer_id: "customerId", request: {  })
+#  api.customers.update(customer_id: "customerId", request: { name: "Acme, Inc. (Updated)", phone: "123-456-7890", employee_count: 101, annual_revenue: 1000001 })
     def update(customer_id:, request:, request_options: nil)
       response = @request_client.conn.put do | req |
   unless request_options&.timeout_in_seconds.nil?
@@ -368,7 +368,7 @@ end
 #    environment: Paid::Environment::PRODUCTION,
 #    token: "YOUR_AUTH_TOKEN"
 #  )
-#  api.customers.create(name: "name")
+#  api.customers.create(name: "Acme, Inc.", external_id: "acme-inc")
     def create(name:, external_id: nil, phone: nil, employee_count: nil, annual_revenue: nil, tax_exempt_status: nil, creation_source: nil, website: nil, billing_address: nil, request_options: nil)
       Async do
         response = @request_client.conn.post do | req |
@@ -444,7 +444,7 @@ end
 #    environment: Paid::Environment::PRODUCTION,
 #    token: "YOUR_AUTH_TOKEN"
 #  )
-#  api.customers.update(customer_id: "customerId", request: {  })
+#  api.customers.update(customer_id: "customerId", request: { name: "Acme, Inc. (Updated)", phone: "123-456-7890", employee_count: 101, annual_revenue: 1000001 })
     def update(customer_id:, request:, request_options: nil)
       Async do
         response = @request_client.conn.put do | req |

data/lib/paid_ruby/orders/client.rb

@@ -79,13 +79,14 @@ end
 #    token: "YOUR_AUTH_TOKEN"
 #  )
 #  api.orders.create(
-#    customer_id: "customerId",
-#    billing_contact_id: "billingContactId",
-#    name: "name",
-#    start_date: "startDate",
-#    currency: "currency"
+#    customer_external_id: "acme-inc",
+#    name: "Acme Order",
+#    description: "Acme Order is an order for Acme, Inc.",
+#    start_date: "2025-01-01",
+#    end_date: "2026-01-01",
+#    currency: "USD"
 #  )
-    def create(customer_id:, customer_external_id: nil, billing_contact_id:, name:, description: nil, start_date:, end_date: nil, currency:, order_lines: nil, request_options: nil)
+    def create(customer_id: nil, customer_external_id: nil, billing_contact_id: nil, name:, description: nil, start_date:, end_date: nil, currency:, order_lines: nil, request_options: nil)
       response = @request_client.conn.post do | req |
   unless request_options&.timeout_in_seconds.nil?
     req.options.timeout = request_options.timeout_in_seconds
@@ -258,13 +259,14 @@ end
 #    token: "YOUR_AUTH_TOKEN"
 #  )
 #  api.orders.create(
-#    customer_id: "customerId",
-#    billing_contact_id: "billingContactId",
-#    name: "name",
-#    start_date: "startDate",
-#    currency: "currency"
+#    customer_external_id: "acme-inc",
+#    name: "Acme Order",
+#    description: "Acme Order is an order for Acme, Inc.",
+#    start_date: "2025-01-01",
+#    end_date: "2026-01-01",
+#    currency: "USD"
 #  )
-    def create(customer_id:, customer_external_id: nil, billing_contact_id:, name:, description: nil, start_date:, end_date: nil, currency:, order_lines: nil, request_options: nil)
+    def create(customer_id: nil, customer_external_id: nil, billing_contact_id: nil, name:, description: nil, start_date:, end_date: nil, currency:, order_lines: nil, request_options: nil)
       Async do
         response = @request_client.conn.post do | req |
   unless request_options&.timeout_in_seconds.nil?

data/lib/paid_ruby/orders/lines/client.rb

@@ -31,7 +31,7 @@ module Paid
 #    environment: Paid::Environment::PRODUCTION,
 #    token: "YOUR_AUTH_TOKEN"
 #  )
-#  api.orders.lines.update(order_id: "orderId")
+#  api.orders.lines.update(order_id: "orderId", lines: [{ agent_external_id: "acme-agent", name: "Order Line One", description: "Order Line One is an order line for Acme, Inc." }, { agent_external_id: "acme-agent-2", name: "Order Line Two", description: "Order Line Two is an order line for Acme, Inc." }])
       def update(order_id:, lines: nil, request_options: nil)
         response = @request_client.conn.put do | req |
   unless request_options&.timeout_in_seconds.nil?
@@ -74,7 +74,7 @@ end
 #    environment: Paid::Environment::PRODUCTION,
 #    token: "YOUR_AUTH_TOKEN"
 #  )
-#  api.orders.lines.update(order_id: "orderId")
+#  api.orders.lines.update(order_id: "orderId", lines: [{ agent_external_id: "acme-agent", name: "Order Line One", description: "Order Line One is an order line for Acme, Inc." }, { agent_external_id: "acme-agent-2", name: "Order Line Two", description: "Order Line Two is an order line for Acme, Inc." }])
       def update(order_id:, lines: nil, request_options: nil)
         Async do
           response = @request_client.conn.put do | req |

data/lib/paid_ruby/tracing/logging.rb

@@ -0,0 +1,32 @@
+# frozen_string_literal: true
+
+require "logger"
+
+module Paid
+  module Tracing
+    # Provides a central logger for the gem.
+    # The log level can be configured via the PAID_LOG_LEVEL environment variable.
+    # Supported levels are DEBUG, INFO, WARN, ERROR, FATAL.
+    # If the variable is not set, the level defaults to FATAL to suppress output.
+    module Logging
+      def self.logger
+        @logger ||= begin
+          log_level_str = ENV["PAID_LOG_LEVEL"]&.upcase
+          level = if log_level_str && Logger.const_defined?(log_level_str)
+                    Logger.const_get(log_level_str)
+                  else
+                    # Default to a level that shows no logs unless explicitly configured.
+                    Logger::FATAL
+                  end
+
+          logger = Logger.new($stdout)
+          logger.level = level
+          logger.formatter = proc do |severity, _datetime, _progname, msg|
+            "[Paid SDK] #{severity}: #{msg}\n"
+          end
+          logger
+        end
+      end
+    end
+  end
+end

data/lib/paid_ruby/tracing/tracing.rb

@@ -0,0 +1,93 @@
+# frozen_string_literal: true
+
+require "opentelemetry/sdk"
+require "opentelemetry/exporter/otlp"
+require_relative "logging"
+
+module Paid
+  module Tracing
+    @token = nil
+
+    # Context keys to propagate external_customer_id and token to child spans.
+    # These are just keys, not the values themselves.
+    PAID_EXTERNAL_CUSTOMER_ID_KEY = OpenTelemetry::Context.create_key("paid.external_customer_id")
+    PAID_TOKEN_KEY = OpenTelemetry::Context.create_key("paid.token")
+
+    # @param api_key [String]
+    def self.initialize_tracing(api_key:)
+      endpoint = "https://collector.agentpaid.io:4318/v1/traces"
+      # endpoint = "http://localhost:4318/v1/traces"
+
+      @token = api_key
+
+      exporter = OpenTelemetry::Exporter::OTLP::Exporter.new(
+        endpoint: endpoint,
+        headers: {}
+      )
+      span_processor = OpenTelemetry::SDK::Trace::Export::BatchSpanProcessor.new(exporter)
+
+      OpenTelemetry::SDK.configure do |c|
+        c.add_span_processor(span_processor)
+      end
+
+      # Add an at_exit hook to ensure spans are flushed before the script exits.
+      at_exit do
+        OpenTelemetry.tracer_provider.shutdown
+      end
+
+      Logging.logger.info("Paid tracing initialized successfully")
+    rescue StandardError => e
+      Logging.logger.error("Failed to initialize Paid tracing: #{e.message}")
+      raise
+    end
+
+    def self.token
+      @token
+    end
+
+    # Getter for the OpenAI wrapper to retrieve the external_customer_id from the context.
+    def self.get_external_customer_id_from_context
+      OpenTelemetry::Context.current.value(PAID_EXTERNAL_CUSTOMER_ID_KEY)
+    end
+
+    # Getter for the OpenAI wrapper to retrieve the token from the context.
+    def self.get_token_from_context
+      OpenTelemetry::Context.current.value(PAID_TOKEN_KEY)
+    end
+
+    # @param external_customer_id [String]
+    # @param args [Array]
+    # @param block [Proc]
+    def self.capture(*args, external_customer_id:, &block)
+      token = self.token
+      unless token
+        Logging.logger.warn("No token found - tracing is not initialized and will not be captured")
+        return yield(*args) if block_given?
+      end
+
+      new_context_values = {
+        PAID_EXTERNAL_CUSTOMER_ID_KEY => external_customer_id,
+        PAID_TOKEN_KEY => token
+      }
+
+      # Execute the block within a new context containing our values.
+      OpenTelemetry::Context.with_values(new_context_values) do
+        tracer = OpenTelemetry.tracer_provider.tracer("paid.ruby")
+
+        tracer.in_span("paid.ruby:#{external_customer_id}") do |span|
+          span.set_attribute("external_customer_id", external_customer_id)
+          span.set_attribute("token", token)
+
+          begin
+            result = yield(*args) if block_given?
+            span.status = OpenTelemetry::Trace::Status.ok("Success")
+            result
+          rescue StandardError => e
+            span.status = OpenTelemetry::Trace::Status.error("Error: #{e.message}")
+            raise e
+          end
+        end
+      end
+    end
+  end
+end

data/lib/paid_ruby/tracing/wrappers/open_ai_wrapper.rb

@@ -0,0 +1,132 @@
+# frozen_string_literal: true
+
+require "openai"
+require_relative "../tracing"
+
+module Paid
+  module Tracing
+    module Wrappers
+      # A wrapper around the OpenAI::Client that provides automatic tracing for API calls.
+      class PaidOpenAI
+        def initialize(openai_client:)
+          @openai_client = openai_client
+          @tracer = OpenTelemetry.tracer_provider.tracer("paid.ruby")
+        end
+
+        # Wraps the OpenAI#chat method to create a child span.
+        def chat(parameters:)
+          wrap_call(operation: "chat", model: parameters[:model]) do
+            @openai_client.chat(parameters: parameters)
+          end
+        end
+
+        # Wraps the OpenAI#embeddings method to create a child span.
+        def embeddings(parameters:)
+          wrap_call(operation: "embeddings", model: parameters[:model]) do
+            @openai_client.embeddings(parameters: parameters)
+          end
+        end
+
+        # Returns a wrapper for the images API.
+        def images
+          ImagesWrapper.new(openai_client: @openai_client, tracer: @tracer)
+        end
+
+        private
+
+        # A private wrapper for the OpenAI Images API.
+        class ImagesWrapper
+          def initialize(openai_client:, tracer:)
+            @openai_client = openai_client
+            @tracer = tracer
+          end
+
+          def generate(parameters:)
+            current_span = OpenTelemetry::Trace.current_span
+            unless current_span.context.valid?
+              Paid::Tracing::Logging.logger.warn("No active span found, calling OpenAI directly without tracing.")
+              return @openai_client.images.generate(parameters: parameters)
+            end
+
+            external_customer_id = Paid::Tracing.get_external_customer_id_from_context
+            token = Paid::Tracing.get_token_from_context
+            model = parameters[:model] || "dall-e-3"
+            span_name = "trace.images #{model}"
+
+            @tracer.in_span(span_name) do |span|
+              attributes = {
+                "gen_ai.request.model" => model,
+                "gen_ai.system" => "openai",
+                "gen_ai.operation.name" => "image_generation"
+              }
+              attributes["external_customer_id"] = external_customer_id if external_customer_id
+              attributes["token"] = token if token
+              span.add_attributes(attributes)
+
+              begin
+                response = @openai_client.images.generate(parameters: parameters)
+                span.add_attributes({
+                                      "gen_ai.image.count" => parameters[:n] || 1,
+                                      "gen_ai.image.size" => parameters[:size] || "1024x1024",
+                                      "gen_ai.image.quality" => parameters[:quality] || "standard"
+                                    })
+                span.status = OpenTelemetry::Trace::Status.ok("Success")
+                response
+              rescue StandardError => e
+                span.record_exception(e)
+                span.status = OpenTelemetry::Trace::Status.error("Error: #{e.message}")
+                raise e
+              end
+            end
+          end
+        end
+
+        def wrap_call(operation:, model:, &block)
+          current_span = OpenTelemetry::Trace.current_span
+          unless current_span.context.valid?
+            Paid::Tracing::Logging.logger.warn("No active span found, calling OpenAI directly without tracing.")
+            return yield
+          end
+
+          external_customer_id = Paid::Tracing.get_external_customer_id_from_context
+          token = Paid::Tracing.get_token_from_context
+          model_name = model || "unknown"
+          span_name = "trace.#{operation} #{model_name}"
+
+          @tracer.in_span(span_name) do |span|
+            attributes = {
+              "gen_ai.system" => "openai",
+              "gen_ai.operation.name" => operation
+            }
+            attributes["external_customer_id"] = external_customer_id if external_customer_id
+            attributes["token"] = token if token
+            span.add_attributes(attributes)
+
+            begin
+              response = yield
+              add_response_attributes(span, response)
+              span.status = OpenTelemetry::Trace::Status.ok("Success")
+              response
+            rescue StandardError => e
+              span.record_exception(e)
+              span.status = OpenTelemetry::Trace::Status.error("Error: #{e.message}")
+              raise e
+            end
+          end
+        end
+
+        def add_response_attributes(span, response)
+          return unless response.is_a?(Hash) && response.dig("usage")
+
+          attributes = {
+            "gen_ai.usage.input_tokens" => response.dig("usage", "prompt_tokens"),
+            "gen_ai.usage.output_tokens" => response.dig("usage", "completion_tokens"),
+            "gen_ai.response.model" => response.dig("model")
+          }.compact
+
+          span.add_attributes(attributes)
+        end
+      end
+    end
+  end
+end

data/lib/paid_ruby/types/billing_frequency.rb

@@ -3,9 +3,9 @@
 module Paid
   class BillingFrequency
 
-    MONTHLY = "Monthly"
-    QUARTERLY = "Quarterly"
-    ANNUAL = "Annual"
+    MONTHLY = "monthly"
+    QUARTERLY = "quarterly"
+    ANNUAL = "annual"
 
   end
 end

data/lib/paid_ruby/types/pricing.rb

@@ -11,6 +11,8 @@ module Paid
     attr_reader :event_name
   # @return [Boolean] 
     attr_reader :taxable
+  # @return [Float] 
+    attr_reader :credit_cost
   # @return [Paid::ChargeType] 
     attr_reader :charge_type
   # @return [Paid::PricingModelType] 
@@ -29,21 +31,23 @@ module Paid
 
     # @param event_name [String] 
     # @param taxable [Boolean] 
+    # @param credit_cost [Float] 
     # @param charge_type [Paid::ChargeType] 
     # @param pricing_model [Paid::PricingModelType] 
     # @param billing_frequency [Paid::BillingFrequency] 
     # @param price_points [Hash{String => Paid::AgentPricePoint}] 
     # @param additional_properties [OpenStruct] Additional properties unmapped to the current class definition
     # @return [Paid::Pricing]
-    def initialize(event_name: OMIT, taxable:, charge_type:, pricing_model:, billing_frequency:, price_points:, additional_properties: nil)
+    def initialize(event_name: OMIT, taxable:, credit_cost: OMIT, charge_type:, pricing_model:, billing_frequency:, price_points:, additional_properties: nil)
       @event_name = event_name if event_name != OMIT
       @taxable = taxable
+      @credit_cost = credit_cost if credit_cost != OMIT
       @charge_type = charge_type
       @pricing_model = pricing_model
       @billing_frequency = billing_frequency
       @price_points = price_points
       @additional_properties = additional_properties
-      @_field_set = { "eventName": event_name, "taxable": taxable, "chargeType": charge_type, "pricingModel": pricing_model, "billingFrequency": billing_frequency, "pricePoints": price_points }.reject do | _k, v |
+      @_field_set = { "eventName": event_name, "taxable": taxable, "creditCost": credit_cost, "chargeType": charge_type, "pricingModel": pricing_model, "billingFrequency": billing_frequency, "pricePoints": price_points }.reject do | _k, v |
   v == OMIT
 end
     end
@@ -56,6 +60,7 @@ end
       parsed_json = JSON.parse(json_object)
       event_name = parsed_json["eventName"]
       taxable = parsed_json["taxable"]
+      credit_cost = parsed_json["creditCost"]
       charge_type = parsed_json["chargeType"]
       pricing_model = parsed_json["pricingModel"]
       billing_frequency = parsed_json["billingFrequency"]
@@ -66,6 +71,7 @@ end
       new(
         event_name: event_name,
         taxable: taxable,
+        credit_cost: credit_cost,
         charge_type: charge_type,
         pricing_model: pricing_model,
         billing_frequency: billing_frequency,
@@ -88,6 +94,7 @@ end
     def self.validate_raw(obj:)
       obj.event_name&.is_a?(String) != false || raise("Passed value for field obj.event_name is not the expected type, validation failed.")
       obj.taxable.is_a?(Boolean) != false || raise("Passed value for field obj.taxable is not the expected type, validation failed.")
+      obj.credit_cost&.is_a?(Float) != false || raise("Passed value for field obj.credit_cost is not the expected type, validation failed.")
       obj.charge_type.is_a?(Paid::ChargeType) != false || raise("Passed value for field obj.charge_type is not the expected type, validation failed.")
       obj.pricing_model.is_a?(Paid::PricingModelType) != false || raise("Passed value for field obj.pricing_model is not the expected type, validation failed.")
       obj.billing_frequency.is_a?(Paid::BillingFrequency) != false || raise("Passed value for field obj.billing_frequency is not the expected type, validation failed.")

data/lib/paid_ruby/types/pricing_model_type.rb

@@ -6,6 +6,7 @@ module Paid
     PER_UNIT = "PerUnit"
     VOLUME_PRICING = "VolumePricing"
     GRADUATED_PRICING = "GraduatedPricing"
+    PREPAID_CREDITS = "PrepaidCredits"
 
   end
 end

data/lib/paid_ruby/types/signal.rb

@@ -8,6 +8,8 @@ module Paid
     attr_reader :event_name
   # @return [String] 
     attr_reader :agent_id
+  # @return [String] 
+    attr_reader :external_agent_id
   # @return [String] 
     attr_reader :customer_id
   # @return [Hash{String => Object}] 
@@ -22,17 +24,19 @@ module Paid
 
     # @param event_name [String] 
     # @param agent_id [String] 
+    # @param external_agent_id [String] 
     # @param customer_id [String] 
     # @param data [Hash{String => Object}] 
     # @param additional_properties [OpenStruct] Additional properties unmapped to the current class definition
     # @return [Paid::Signal]
-    def initialize(event_name: OMIT, agent_id: OMIT, customer_id: OMIT, data: OMIT, additional_properties: nil)
+    def initialize(event_name: OMIT, agent_id: OMIT, external_agent_id: OMIT, customer_id: OMIT, data: OMIT, additional_properties: nil)
       @event_name = event_name if event_name != OMIT
       @agent_id = agent_id if agent_id != OMIT
+      @external_agent_id = external_agent_id if external_agent_id != OMIT
       @customer_id = customer_id if customer_id != OMIT
       @data = data if data != OMIT
       @additional_properties = additional_properties
-      @_field_set = { "event_name": event_name, "agent_id": agent_id, "customer_id": customer_id, "data": data }.reject do | _k, v |
+      @_field_set = { "event_name": event_name, "agent_id": agent_id, "external_agent_id": external_agent_id, "customer_id": customer_id, "data": data }.reject do | _k, v |
   v == OMIT
 end
     end
@@ -45,11 +49,13 @@ end
       parsed_json = JSON.parse(json_object)
       event_name = parsed_json["event_name"]
       agent_id = parsed_json["agent_id"]
+      external_agent_id = parsed_json["external_agent_id"]
       customer_id = parsed_json["customer_id"]
       data = parsed_json["data"]
       new(
         event_name: event_name,
         agent_id: agent_id,
+        external_agent_id: external_agent_id,
         customer_id: customer_id,
         data: data,
         additional_properties: struct
@@ -70,6 +76,7 @@ end
     def self.validate_raw(obj:)
       obj.event_name&.is_a?(String) != false || raise("Passed value for field obj.event_name is not the expected type, validation failed.")
       obj.agent_id&.is_a?(String) != false || raise("Passed value for field obj.agent_id is not the expected type, validation failed.")
+      obj.external_agent_id&.is_a?(String) != false || raise("Passed value for field obj.external_agent_id is not the expected type, validation failed.")
       obj.customer_id&.is_a?(String) != false || raise("Passed value for field obj.customer_id is not the expected type, validation failed.")
       obj.data&.is_a?(Hash) != false || raise("Passed value for field obj.data is not the expected type, validation failed.")
     end

data/lib/paid_ruby/usage/client.rb

@@ -19,6 +19,7 @@ module Paid
     # @param signals [Array<Hash>] Request of type Array<Paid::Signal>, as a Hash
     #   * :event_name (String) 
     #   * :agent_id (String) 
+    #   * :external_agent_id (String) 
     #   * :customer_id (String) 
     #   * :data (Hash{String => Object}) 
     # @param request_options [Paid::RequestOptions] 
@@ -29,7 +30,7 @@ module Paid
 #    environment: Paid::Environment::PRODUCTION,
 #    token: "YOUR_AUTH_TOKEN"
 #  )
-#  api.usage.record_bulk
+#  api.usage.record_bulk(signals: [{  }, {  }, {  }])
     def record_bulk(signals: nil, request_options: nil)
       response = @request_client.conn.post do | req |
   unless request_options&.timeout_in_seconds.nil?
@@ -62,6 +63,7 @@ end
     # @param signals [Array<Hash>] Request of type Array<Paid::Signal>, as a Hash
     #   * :event_name (String) 
     #   * :agent_id (String) 
+    #   * :external_agent_id (String) 
     #   * :customer_id (String) 
     #   * :data (Hash{String => Object}) 
     # @param request_options [Paid::RequestOptions] 
@@ -72,7 +74,7 @@ end
 #    environment: Paid::Environment::PRODUCTION,
 #    token: "YOUR_AUTH_TOKEN"
 #  )
-#  api.usage.record_bulk
+#  api.usage.record_bulk(signals: [{  }, {  }, {  }])
     def record_bulk(signals: nil, request_options: nil)
       Async do
         response = @request_client.conn.post do | req |

data/lib/paid_ruby.rb

@@ -9,6 +9,8 @@ require_relative "paid_ruby/contacts/client"
 require_relative "paid_ruby/orders/client"
 require_relative "paid_ruby/usage/client"
 require_relative "extensions/batch"
+require_relative "paid_ruby/tracing/tracing"
+require_relative "paid_ruby/tracing/wrappers/open_ai_wrapper"
 
 module Paid
   class Client
@@ -44,6 +46,19 @@ module Paid
       @orders = Paid::OrdersClient.new(request_client: @request_client)
       @usage = Paid::BatchUsageClient.new(request_client: @request_client)
     end
+
+    def initialize_tracing
+      token = @request_client.token
+      api_key = token.gsub(/^Bearer /, "")
+      Paid::Tracing.initialize_tracing(api_key: api_key)
+    end
+
+    # @param external_customer_id [String]
+    # @param args [Array]
+    # @param block [Proc]
+    def capture(*args, external_customer_id:, &block)
+      Paid::Tracing.capture(*args, external_customer_id: external_customer_id, &block)
+    end
   end
 
   class AsyncClient
@@ -79,5 +94,18 @@ module Paid
       @orders = Paid::AsyncOrdersClient.new(request_client: @async_request_client)
       @usage = Paid::AsyncBatchUsageClient.new(request_client: @async_request_client)
     end
+
+    def initialize_tracing
+      token = @async_request_client.token
+      api_key = token.gsub(/^Bearer /, "")
+      Paid::Tracing.initialize_tracing(api_key: api_key)
+    end
+
+    # @param external_customer_id [String]
+    # @param args [Array]
+    # @param block [Proc]
+    def capture(*args, external_customer_id:, &block)
+      Paid::Tracing.capture(*args, external_customer_id: external_customer_id, &block)
+    end
   end
 end

data/lib/requests.rb

@@ -47,7 +47,7 @@ end
     end
     # @return [Hash{String => String}]
     def get_headers
-      headers = { "X-Fern-Language": 'Ruby', "X-Fern-SDK-Name": 'paid_ruby', "X-Fern-SDK-Version": '0.1.1' }
+      headers = { "X-Fern-Language": 'Ruby', "X-Fern-SDK-Name": 'paid_ruby', "X-Fern-SDK-Version": '0.1.2' }
       headers["Authorization"] = ((@token.is_a? Method) ? @token.call : @token) unless @token.nil?
  headers
     end
@@ -92,7 +92,7 @@ end
     end
     # @return [Hash{String => String}]
     def get_headers
-      headers = { "X-Fern-Language": 'Ruby', "X-Fern-SDK-Name": 'paid_ruby', "X-Fern-SDK-Version": '0.1.1' }
+      headers = { "X-Fern-Language": 'Ruby', "X-Fern-SDK-Name": 'paid_ruby', "X-Fern-SDK-Version": '0.1.2' }
       headers["Authorization"] = ((@token.is_a? Method) ? @token.call : @token) unless @token.nil?
  headers
     end

metadata

@@ -1,15 +1,70 @@
 --- !ruby/object:Gem::Specification
 name: paid_ruby
 version: !ruby/object:Gem::Version
-  version: 0.1.1
+  version: 0.1.2
 platform: ruby
 authors:
 - ''
-autorequire:
 bindir: exe
 cert_chain: []
-date: 2025-06-12 00:00:00.000000000 Z
+date: 1980-01-02 00:00:00.000000000 Z
 dependencies:
+- !ruby/object:Gem::Dependency
+  name: opentelemetry-api
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.5'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.5'
+- !ruby/object:Gem::Dependency
+  name: opentelemetry-exporter-otlp
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 0.30.0
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 0.30.0
+- !ruby/object:Gem::Dependency
+  name: opentelemetry-sdk
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.8'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.8'
+- !ruby/object:Gem::Dependency
+  name: ruby-openai
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '8.1'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '8.1'
 - !ruby/object:Gem::Dependency
   name: faraday
   requirement: !ruby/object:Gem::Requirement
@@ -105,6 +160,9 @@ files:
 - lib/paid_ruby/customers/client.rb
 - lib/paid_ruby/orders/client.rb
 - lib/paid_ruby/orders/lines/client.rb
+- lib/paid_ruby/tracing/logging.rb
+- lib/paid_ruby/tracing/tracing.rb
+- lib/paid_ruby/tracing/wrappers/open_ai_wrapper.rb
 - lib/paid_ruby/types/address.rb
 - lib/paid_ruby/types/agent.rb
 - lib/paid_ruby/types/agent_attribute.rb
@@ -141,7 +199,6 @@ metadata:
   homepage_uri: https://github.com/paid-ai/paid-ruby
   source_code_uri: https://github.com/paid-ai/paid-ruby
   changelog_uri: https://github.com/paid-ai/paid-ruby/blob/master/CHANGELOG.md
-post_install_message:
 rdoc_options: []
 require_paths:
 - lib
@@ -156,8 +213,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.1.6
-signing_key:
+rubygems_version: 3.6.7
 specification_version: 4
 summary: ''
 test_files: []