agentbill-sdk 2.0.1 → 5.0.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 552854759cf5b06dca09f5a250402634f11f3077b60ec0306f7e1b61817b0281
-  data.tar.gz: 31ab1d0c1a940e0d1bfa66368d81a3cb3026c9b70868eda69b04b13d106134a0
+  metadata.gz: 7e9b487391274ecb94dbc031a40f6e30a167e6fa0e30bb257464f68e497871f2
+  data.tar.gz: cc6bc10bfdddf68981cd5454f80b4cffc6ea10a244f49f7e1f6582dd154e9b6c
 SHA512:
-  metadata.gz: e8cb6296cc8522aae786ffb0927379ddaed1a08d0b74cca6d11342b964851d99186878da310609e912c178d9a893f589314f802d0306327d6baa2d74cb9a8d0e
-  data.tar.gz: c47673c3b91a0c4e4924f52d14ec1d0821824d2d2bbddf75902332c5a383aa9a1b2eee5095cd07b83dd193715aaf69bed6993e8f4706f906021a16cb5614338c
+  metadata.gz: a4409ffa8bac425a994a09c45d513877d5167e7ca9706c86229de4a6a50f8bf1da729ec1ccab5448c1dda808ed43cb04b2c314840f894801bd203cd8ca608aff
+  data.tar.gz: 285d37fb82ae7b3173474c72c47647e2585d8cf983730b16a24adf068a7baae3004078e10c79c493f66960d19921fd531d4aac1b7b6f35688ec7b9e1b541c987

data/CHANGELOG.md

@@ -1,27 +1,57 @@
 # Changelog
-# Force sync - Updated to 1.0.2
 
 All notable changes to this project will be documented in this file.
 
 The format is based on [Keep a Changelog](https://keepachangelog.com/en/1.0.0/),
 and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0.html).
 
-## [1.0.2] - 2025-10-27
+## [5.0.1] - 2025-11-06
 
-### Changed
-- Version bump for consistency across SDKs
+### Added
+- **OTEL Correlation Support** - `track_signal()` now accepts optional `trace_id` and `span_id` parameters
+- Enables cost reconciliation by correlating OTEL spans (provider costs) with signals (customer revenue)
+- Parameters are optional - use only when OTEL tracking is enabled for profit margin analysis
+
+### Example
+```ruby
+# With OTEL correlation for cost reconciliation
+span = agentbill.tracer.start_span('ai_completion')
+# ... make AI call ...
+agentbill.track_signal(
+  event_name: 'ai_request',
+  revenue: 5.00,
+  trace_id: span.trace_id,  # Optional
+  span_id: span.span_id     # Optional
+)
+```
+
+## [2.0.2] - 2025-11-04
+
+### Fixed
+- **CRITICAL**: Fixed authentication header to use `X-API-Key` instead of `Authorization: Bearer` format
+- Updated `track_signal()` method in agentbill.rb to use correct authentication
+- Updated `flush()` method in tracer.rb to use correct authentication
+- Signals now properly authenticate with record-signals and otel-collector edge functions
+
+## [2.0.1] - 2025-10-25
+
+### Added
+- Enhanced error messages and validation
+
+## [2.0.0] - 2025-10-25
+
+### Added
+- Comprehensive input validation
+- Support for all signal parameters
 
 ## [1.0.0] - 2025-10-21
 
 ### Added
 - Initial release of AgentBill Ruby SDK
 - OpenAI client wrapping with automatic usage tracking
-- Anthropic client wrapping with automatic usage tracking
 - OpenTelemetry-based tracing infrastructure
 - Custom signal tracking support
 - Comprehensive test suite with RSpec
-- GitHub Actions CI/CD workflows
-- RuboCop code quality enforcement
 - Professional documentation and examples
 - MIT License
 
@@ -33,16 +63,11 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 - Error tracking and reporting
 - Customer-specific tracking support
 - Debug logging capabilities
-
-### Supported Providers
-- OpenAI (GPT-4, GPT-3.5, etc.)
-- Anthropic (Claude 3.5 Sonnet, etc.)
+- Gem packaging support
 
 ### Documentation
 - Complete README with usage examples
 - API documentation
 - Contributing guidelines
 - Security policy
-- Code of conduct
-
-[1.0.0]: https://github.com/Agent-Bill/Ruby/releases/tag/v1.0.0
+- Code examples

data/README.md

@@ -23,8 +23,9 @@ bundle install
 
 ### From Source
 ```bash
-git clone https://github.com/Agent-Bill/Ruby.git
-cd agentbill-ruby
+# Clone your repository and install
+git clone https://github.com/yourusername/agentbill-ruby-sdk.git
+cd agentbill-ruby-sdk
 gem build agentbill.gemspec
 gem install agentbill-1.0.0.gem
 ```
@@ -140,7 +141,7 @@ gem push agentbill-1.0.0.gem
 
 ## GitHub Repository Setup
 
-1. Create repository: `agentbill-ruby`
+1. Create your GitHub repository (e.g., `agentbill-ruby-sdk`)
 2. Push all files from `lib/` directory
 3. Tag releases: `git tag v1.0.0 && git push origin v1.0.0`
 

data/SECURITY.md

@@ -16,7 +16,7 @@ We take security vulnerabilities seriously. If you discover a security issue, pl
 
 **DO NOT** create a public GitHub issue for security vulnerabilities.
 
-Instead, please email: security@agentbill.com
+Instead, please email: support@agentbill.io
 
 Include:
 - Description of the vulnerability
@@ -68,6 +68,6 @@ The AgentBill SDK includes:
 
 ## Questions?
 
-For general security questions, email: security@agentbill.com
+For general security questions, email: support@agentbill.io
 
 Thank you for helping keep AgentBill secure! 🔒

data/examples/openai_custom_event.rb

@@ -0,0 +1,50 @@
+=begin
+OpenAI with Custom Event Names Example
+
+Shows how to pass custom event_name per AI call for better tracking
+=end
+
+require 'agentbill'
+require 'openai'
+
+# Initialize AgentBill
+agentbill = AgentBill::Client.init({
+  api_key: ENV['AGENTBILL_API_KEY'] || 'your-api-key',
+  agent_id: 'agent-uuid', # Your agent UUID
+  customer_id: 'customer-123',
+  debug: true
+})
+
+# Wrap your OpenAI client
+client = agentbill.wrap_openai(OpenAI::Client.new(
+  access_token: ENV['OPENAI_API_KEY']
+))
+
+# Make a request with custom event name for code generation
+response = client.chat({
+  model: 'gpt-4o-mini',
+  messages: [
+    { role: 'system', content: 'You are a code generation assistant.' },
+    { role: 'user', content: 'Write a function to calculate fibonacci numbers' }
+  ],
+  agentbill_options: {
+    event_name: 'code_generation'
+  }
+})
+
+puts response['choices'][0]['message']['content']
+
+# Different event for chat
+chat_response = client.chat({
+  model: 'gpt-4o-mini',
+  messages: [
+    { role: 'user', content: 'What is the weather like?' }
+  ],
+  agentbill_options: {
+    event_name: 'chat_response'
+  }
+})
+
+puts chat_response['choices'][0]['message']['content']
+
+# All usage (tokens, cost, latency) is automatically tracked with custom event names

data/lib/agentbill/tracer.rb

@@ -61,7 +61,7 @@ module AgentBill
       http.use_ssl = true
 
       request = Net::HTTP::Post.new(uri.path)
-      request['Authorization'] = "Bearer #{@api_key}"
+      request['X-API-Key'] = @api_key
       request['Content-Type'] = 'application/json'
       request.body = payload.to_json
 

data/lib/agentbill/version.rb

@@ -1,3 +1,3 @@
 module AgentBill
-  VERSION = "2.0.1"
+  VERSION = "5.0.1"
 end

data/lib/agentbill.rb

@@ -74,12 +74,14 @@ module AgentBill
       end
     end
 
-    def track_usage(model, provider, input_tokens, output_tokens, latency_ms, cost)
+    def track_usage(model, provider, input_tokens, output_tokens, latency_ms, cost, event_name = 'ai_request')
       uri = URI("#{@config[:base_url] || 'https://bgwyprqxtdreuutzpbgw.supabase.co'}/functions/v1/track-ai-usage")
       
       payload = {
         api_key: @config[:api_key],
         customer_id: @config[:customer_id],
+        agent_id: @config[:agent_id],
+        event_name: event_name,
         model: model,
         provider: provider,
         prompt_tokens: input_tokens,
@@ -116,6 +118,10 @@ module AgentBill
         messages = params[:messages] || []
         max_tokens = params[:max_tokens] || params[:max_completion_tokens] || 1000
         
+        # Extract event_name from agentbill_options if provided (don't pass to OpenAI)
+        agentbill_options = params.delete(:agentbill_options) || {}
+        event_name = agentbill_options[:event_name] || 'ai_request'
+        
         # Phase 1: Validate budget BEFORE API call
         validation = config[:_client].send(:validate_request, model, messages, max_tokens)
         unless validation['allowed']
@@ -142,7 +148,7 @@ module AgentBill
           output_tokens = response.dig(:usage, :completion_tokens) || 0
           cost = config[:_client].send(:estimate_cost, model, input_tokens, output_tokens)
           
-          config[:_client].send(:track_usage, model, 'openai', input_tokens, output_tokens, latency, cost)
+          config[:_client].send(:track_usage, model, 'openai', input_tokens, output_tokens, latency, cost, event_name)
           
           span.set_attributes({
             'response.prompt_tokens' => input_tokens,
@@ -227,10 +233,12 @@ module AgentBill
       client
     end
 
-    # Track a comprehensive signal with all 68 parameters
+    # Track a comprehensive signal with all 68 parameters including trace_id and span_id
     #
-    # Supports all parameters:
+    # Supports optional trace_id and span_id for OTEL correlation:
     # - event_name (required)
+    # - trace_id (optional) - For correlating with OTEL spans for cost reconciliation
+    # - span_id (optional) - For correlating with OTEL spans for cost reconciliation
     # - data_source, timestamp
     # - agent_external_id, customer_external_id, account_external_id, user_external_id,
     #   order_external_id, session_id, conversation_id, thread_id
@@ -243,17 +251,25 @@ module AgentBill
     # - feedback_score, user_satisfaction, error_type, error_message, retry_count, success_rate
     # - tags, category, priority, severity, compliance_flag, data_classification
     # - product_id, feature_flag, environment, deployment_version, region, tenant_id
-    # - parent_span_id, trace_id
+    # - parent_span_id
     # - custom_dimensions, metadata, data
     #
     # Example:
+    #   # Basic tracking
     #   agentbill.track_signal(
     #     event_name: "user_conversion",
     #     revenue: 99.99,
-    #     customer_external_id: "cust_123",
-    #     experiment_id: "exp_abc",
-    #     conversion_type: "purchase",
-    #     tags: ["checkout", "success"]
+    #     customer_external_id: "cust_123"
+    #   )
+    #   
+    #   # With OTEL correlation
+    #   span = agentbill.tracer.start_span('ai_completion')
+    #   # ... make AI call ...
+    #   agentbill.track_signal(
+    #     event_name: "ai_request",
+    #     revenue: 5.00,
+    #     trace_id: span.trace_id,  # Optional
+    #     span_id: span.span_id     # Optional
     #   )
     def track_signal(**params)
       raise ArgumentError, "event_name is required" unless params[:event_name]
@@ -263,6 +279,18 @@ module AgentBill
       # Add timestamp if not provided
       params[:timestamp] ||= Time.now.to_f
       
+      # Auto-fill customer_id or customer_external_id from config if not provided
+      if !params[:customer_id] && !params[:customer_external_id] && @config[:customer_id]
+        # Check if it's a UUID format - send as customer_id, else customer_external_id
+        uuid_regex = /^[0-9a-f]{8}-[0-9a-f]{4}-[0-9a-f]{4}-[0-9a-f]{4}-[0-9a-f]{12}$/i
+        is_uuid = @config[:customer_id].match?(uuid_regex)
+        if is_uuid
+          params[:customer_id] = @config[:customer_id]
+        else
+          params[:customer_external_id] = @config[:customer_id]
+        end
+      end
+      
       # Remove nil values
       payload = params.reject { |_, v| v.nil? }
       
@@ -271,14 +299,15 @@ module AgentBill
         http.use_ssl = true
         
         request = Net::HTTP::Post.new(uri.path)
-        request['Authorization'] = "Bearer #{@config[:api_key]}"
+        request['X-API-Key'] = @config[:api_key]
         request['Content-Type'] = 'application/json'
         request.body = payload.to_json
         
         response = http.request(request)
         
         if @config[:debug]
-          puts "[AgentBill] Signal tracked: #{params[:event_name]}"
+          trace_info = params[:trace_id] ? " (trace: #{params[:trace_id]})" : ""
+          puts "[AgentBill] Signal tracked: #{params[:event_name]}#{trace_info}"
         end
         
         response.code == '200'

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: agentbill-sdk
 version: !ruby/object:Gem::Version
-  version: 2.0.1
+  version: 5.0.1
 platform: ruby
 authors:
 - AgentBill
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2025-10-31 00:00:00.000000000 Z
+date: 2025-11-06 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bundler
@@ -66,6 +66,7 @@ files:
 - examples/anthropic_basic.rb
 - examples/manual_otel_tracking.rb
 - examples/openai_basic.rb
+- examples/openai_custom_event.rb
 - examples/zero_config.rb
 - lib/agentbill.rb
 - lib/agentbill/tracer.rb