agentbill-sdk 1.0.1 → 1.0.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 2a5d6d3184f1e30d721acd50e8b51ac947d086352bdd4f3f26702adc127b021c
-  data.tar.gz: 57eb001ae9a6ba61c002855ad1abbe17fe14fcc3c8fc269b55903b780da669d2
+  metadata.gz: eaa02ddbb2ec18543984f6db7fa7e579b30f94e4505469f6e1eb72cfb13b4048
+  data.tar.gz: 9ef5bdf26669e2b5f2d5594960c9b8f946a6e47875c7461dec37b4f3c477a79c
 SHA512:
-  metadata.gz: 686aef33e7ff8e70994ffe7ad034beddea13b5adf6fc9cd072288898c253f64c49aae92a28c7068f3de4252304b5b6ed5a86698886a87353ce3912afabb870b2
-  data.tar.gz: 2a86b18d8ea9a824fc2de991ae25f7f83473eda13f9ffaa96e6e451d1cca0729fa6cc3d3331e6a292a94e31fb604643b651e23697bbeec96dde6de6cae2d3c1f
+  metadata.gz: 4401d68f0d95b0d4df7f19c69c90c30ff497fcde8a9aa5177c24a2c52664610de3a1a5eff3407c6d61950d788265eca3c5d51eb05e1a2cde6a2c94337869fbd0
+  data.tar.gz: 65202696532413e194c27a8aafeb1e3f0874b58863408a13455fbe07a9e115c73687c29df8a9b58b98b987dff329fb000def1514bfa66454a2abbe03f513b4cb

data/.github/workflows/publish.yml

@@ -0,0 +1,56 @@
+name: Publish to RubyGems
+
+on:
+  push:
+    tags:
+      - 'ruby-v*'
+  workflow_dispatch:
+    inputs:
+      version:
+        description: 'Version to publish'
+        required: true
+
+jobs:
+  publish:
+    runs-on: ubuntu-latest
+    permissions:
+      contents: read
+      id-token: write
+    
+    steps:
+      - uses: actions/checkout@v4
+      
+      - name: Set up Ruby
+        uses: ruby/setup-ruby@v1
+        with:
+          ruby-version: '3.2'
+          bundler-cache: true
+      
+      - name: Install dependencies
+        run: bundle install
+      
+      - name: Run tests
+        run: bundle exec rake test
+      
+      - name: Build gem
+        run: gem build agentbill.gemspec
+      
+      - name: Publish to RubyGems
+        run: |
+          mkdir -p $HOME/.gem
+          touch $HOME/.gem/credentials
+          chmod 0600 $HOME/.gem/credentials
+          printf -- "---\n:rubygems_api_key: ${RUBYGEMS_API_KEY}\n" > $HOME/.gem/credentials
+          gem push *.gem
+        env:
+          RUBYGEMS_API_KEY: ${{ secrets.RUBYGEMS_API_KEY }}
+      
+      - name: Create GitHub Release
+        uses: actions/create-release@v1
+        env:
+          GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }}
+        with:
+          tag_name: ${{ github.ref }}
+          release_name: Ruby SDK ${{ github.ref }}
+          draft: false
+          prerelease: false

data/CHANGELOG.md

@@ -1,10 +1,16 @@
 # 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
+
+### Changed
+- Version bump for consistency across SDKs
+
 ## [1.0.0] - 2025-10-21
 
 ### Added

data/README.md

@@ -4,10 +4,10 @@ OpenTelemetry-based SDK for automatically tracking and billing AI agent usage.
 
 ## Installation
 
-### From GitHub (Recommended)
+### From RubyGems (Recommended)
 ```ruby
 # In your Gemfile
-gem 'agentbill', git: 'https://github.com/Agent-Bill/Ruby.git'
+gem 'agentbill-sdk'
 ```
 
 ### From RubyGems

data/agentbill.gemspec

@@ -4,7 +4,7 @@ Gem::Specification.new do |spec|
   spec.name          = "agentbill-sdk"
   spec.version       = AgentBill::VERSION
   spec.authors       = ["AgentBill"]
-  spec.email         = ["support@agentbill.com"]
+  spec.email         = ["dominic@agentbill.io"]
 
   spec.summary       = "OpenTelemetry-based SDK for tracking AI agent usage and billing"
   spec.description   = "Automatically track and bill AI agent usage with zero-config instrumentation for OpenAI, Anthropic, and more"

data/examples/manual_otel_tracking.rb

@@ -0,0 +1,45 @@
+require 'agentbill'
+
+# Initialize AgentBill
+agentbill = AgentBill::Client.init({
+  api_key: ENV['AGENTBILL_API_KEY'] || 'your-api-key',
+  base_url: ENV['AGENTBILL_BASE_URL'],
+  customer_id: 'customer-123',
+  debug: true
+})
+
+# Start a span for a database query
+span = agentbill.tracer.start_span('database_query', {
+  'db.system' => 'postgresql',
+  'db.operation' => 'SELECT',
+  'db.table' => 'users'
+})
+
+# Simulate work
+sleep(0.1)
+
+# Add more attributes during execution
+span.set_attributes({
+  'db.rows_returned' => 42,
+  'query.duration_ms' => 95
+})
+
+# Set status (1 = success, 2 = error)
+span.set_status(1)
+span.end
+
+# Start another span for an API call
+api_span = agentbill.tracer.start_span('external_api_call', {
+  'http.method' => 'POST',
+  'http.url' => 'https://api.example.com/endpoint',
+  'http.status_code' => 200
+})
+
+sleep(0.2)
+api_span.set_status(1)
+api_span.end
+
+# Flush all spans to AgentBill
+agentbill.flush
+
+puts "✅ OTEL spans tracked successfully!"

data/examples/zero_config.rb

@@ -0,0 +1,69 @@
+=begin
+Zero-Config AI Cost Guard Example
+
+Just wrap your AI client - that's it!
+
+What happens automatically:
+- Provider detection (OpenAI, Anthropic, etc.)
+- Budget validation BEFORE spending (when daily_budget or monthly_budget is set)
+- Automatic blocking if budget would be exceeded
+- Usage tracking AFTER completion
+=end
+
+require 'agentbill'
+require 'openai'
+
+# 1. Initialize AgentBill with your API key and budget
+agentbill = AgentBill::Client.init({
+  api_key: ENV['AGENTBILL_API_KEY'],
+  customer_id: 'customer-123',
+  daily_budget: 10.00,    # Optional: $10/day limit (enables Cost Guard)
+  monthly_budget: 200.00,  # Optional: $200/month limit (enables Cost Guard)
+  debug: true
+})
+
+# 2. Wrap your AI client - provider auto-detected!
+openai = agentbill.wrap_openai(OpenAI::Client.new(
+  access_token: ENV['OPENAI_API_KEY']
+))
+
+# 3. Use normally - Cost Guard protection happens automatically
+begin
+  response = openai.chat({
+    model: 'gpt-4o-mini',
+    messages: [
+      { role: 'user', content: 'What is the capital of France?' }
+    ]
+  })
+  
+  puts '✅ Response: ' + response['choices'][0]['message']['content']
+  puts '💰 Cost tracked automatically'
+  puts '🛡️ Budget validated BEFORE spending'
+  
+rescue => e
+  if e.respond_to?(:code)
+    case e.code
+    when 'BUDGET_EXCEEDED'
+      puts '❌ Budget limit reached - request blocked by Cost Guard'
+    when 'RATE_LIMIT_EXCEEDED'
+      puts '❌ Rate limit reached - request blocked by Cost Guard'
+    else
+      raise e
+    end
+  else
+    raise e
+  end
+end
+
+=begin
+That's it! Zero configuration needed.
+
+Cost Guard Features:
+- ✅ Hard budget limits (daily/monthly) - request blocked BEFORE spending
+- ✅ Real-time cost estimation and validation
+- ✅ Automatic usage tracking
+- ✅ Works across all providers (OpenAI, Anthropic, Bedrock, Azure, Mistral, Google AI)
+
+To disable Cost Guard: Simply remove daily_budget and monthly_budget from init.
+Advanced features (caching, fallback, etc.) coming in Pro tier.
+=end

data/lib/agentbill/version.rb

@@ -1,3 +1,3 @@
 module AgentBill
-  VERSION = "1.0.1"
+  VERSION = "1.0.3"
 end

data/lib/agentbill.rb

@@ -17,29 +17,142 @@ module AgentBill
       new(config)
     end
 
+    private
+
+    def estimate_tokens(text)
+      # Simple token estimation: ~4 chars per token
+      [1, text.to_s.length / 4].max
+    end
+
+    def estimate_cost(model, input_tokens, output_tokens)
+      # Model pricing (per 1K tokens)
+      pricing = {
+        'gpt-4' => { input: 0.03, output: 0.06 },
+        'gpt-4o' => { input: 0.005, output: 0.015 },
+        'gpt-4o-mini' => { input: 0.00015, output: 0.0006 },
+        'claude-sonnet-4-5' => { input: 0.003, output: 0.015 },
+        'claude-opus-4-1-20250805' => { input: 0.015, output: 0.075 },
+        'claude-3-5-sonnet-20241022' => { input: 0.003, output: 0.015 },
+        'mistral-large-latest' => { input: 0.004, output: 0.012 },
+        'gemini-pro' => { input: 0.00025, output: 0.0005 }
+      }
+      
+      model_price = pricing[model] || { input: 0.001, output: 0.002 }
+      (input_tokens / 1000.0 * model_price[:input]) + (output_tokens / 1000.0 * model_price[:output])
+    end
+
+    def validate_request(model, messages, estimated_output_tokens = 1000)
+      return { 'allowed' => true } unless @config[:daily_budget] || @config[:monthly_budget]
+      
+      uri = URI("#{@config[:base_url] || 'https://bgwyprqxtdreuutzpbgw.supabase.co'}/functions/v1/ai-cost-guard-router")
+      
+      payload = {
+        api_key: @config[:api_key],
+        customer_id: @config[:customer_id],
+        model: model,
+        messages: messages
+      }
+      
+      begin
+        http = Net::HTTP.new(uri.host, uri.port)
+        http.use_ssl = true
+        http.read_timeout = 10
+        
+        request = Net::HTTP::Post.new(uri.path)
+        request['Content-Type'] = 'application/json'
+        request.body = payload.to_json
+        
+        response = http.request(request)
+        result = JSON.parse(response.body)
+        
+        puts "[AgentBill Cost Guard] Router response: #{result}" if @config[:debug]
+        puts "[AgentBill] Tier: #{result['tier']}, Mode: #{result['mode']}" if @config[:debug] && result['tier']
+        result
+      rescue => e
+        puts "[AgentBill Cost Guard] Router failed: #{e.message}" if @config[:debug]
+        { 'allowed' => true }  # Fail open
+      end
+    end
+
+    def track_usage(model, provider, input_tokens, output_tokens, latency_ms, cost)
+      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],
+        model: model,
+        provider: provider,
+        prompt_tokens: input_tokens,
+        completion_tokens: output_tokens,
+        latency_ms: latency_ms,
+        cost: cost
+      }
+      
+      begin
+        http = Net::HTTP.new(uri.host, uri.port)
+        http.use_ssl = true
+        http.read_timeout = 10
+        
+        request = Net::HTTP::Post.new(uri.path)
+        request['Content-Type'] = 'application/json'
+        request.body = payload.to_json
+        
+        http.request(request)
+        puts "[AgentBill Cost Guard] Usage tracked: $#{format('%.4f', cost)}" if @config[:debug]
+      rescue => e
+        puts "[AgentBill Cost Guard] Tracking failed: #{e.message}" if @config[:debug]
+      end
+    end
+
+    public
+
     def wrap_openai(client)
       original_method = client.method(:chat)
+      config = @config
+      tracer = @tracer
       
       client.define_singleton_method(:chat) do |params|
-        start_time = Time.now
+        model = params[:model] || 'unknown'
+        messages = params[:messages] || []
+        max_tokens = params[:max_tokens] || params[:max_completion_tokens] || 1000
+        
+        # Phase 1: Validate budget BEFORE API call
+        validation = config[:_client].send(:validate_request, model, messages, max_tokens)
+        unless validation['allowed']
+          error_msg = validation['reason'] || 'Budget limit reached'
+          puts "[AgentBill Cost Guard] ❌ Request blocked: #{error_msg}" if config[:debug]
+          error = StandardError.new(error_msg)
+          error.define_singleton_method(:code) { 'BUDGET_EXCEEDED' }
+          raise error
+        end
         
-        span = @tracer.start_span('openai.chat.completion', {
-          'model' => params[:model] || 'unknown',
+        # Phase 2: Execute AI call
+        start_time = Time.now
+        span = tracer.start_span('openai.chat.completion', {
+          'model' => model,
           'provider' => 'openai'
         })
         
         begin
           response = original_method.call(params)
-          
           latency = ((Time.now - start_time) * 1000).round
+          
+          # Phase 3: Track actual usage
+          input_tokens = response.dig(:usage, :prompt_tokens) || 0
+          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)
+          
           span.set_attributes({
-            'response.prompt_tokens' => response.dig(:usage, :prompt_tokens),
-            'response.completion_tokens' => response.dig(:usage, :completion_tokens),
+            'response.prompt_tokens' => input_tokens,
+            'response.completion_tokens' => output_tokens,
             'response.total_tokens' => response.dig(:usage, :total_tokens),
             'latency_ms' => latency
           })
           span.set_status(0)
           
+          puts "[AgentBill Cost Guard] ✓ Protected call completed: $#{format('%.4f', cost)}" if config[:debug]
           response
         rescue => e
           span.set_status(1, e.message)
@@ -49,31 +162,57 @@ module AgentBill
         end
       end
       
+      # Store reference to self for helper methods
+      @config[:_client] = self
       client
     end
 
     def wrap_anthropic(client)
       original_method = client.method(:messages)
+      config = @config
+      tracer = @tracer
       
       client.define_singleton_method(:messages) do |params|
-        start_time = Time.now
+        model = params[:model] || 'unknown'
+        messages = params[:messages] || []
+        max_tokens = params[:max_tokens] || 1000
+        
+        # Phase 1: Validate budget BEFORE API call
+        validation = config[:_client].send(:validate_request, model, messages, max_tokens)
+        unless validation['allowed']
+          error_msg = validation['reason'] || 'Budget limit reached'
+          puts "[AgentBill Cost Guard] ❌ Request blocked: #{error_msg}" if config[:debug]
+          error = StandardError.new(error_msg)
+          error.define_singleton_method(:code) { 'BUDGET_EXCEEDED' }
+          raise error
+        end
         
-        span = @tracer.start_span('anthropic.message', {
-          'model' => params[:model] || 'unknown',
+        # Phase 2: Execute AI call
+        start_time = Time.now
+        span = tracer.start_span('anthropic.message', {
+          'model' => model,
           'provider' => 'anthropic'
         })
         
         begin
           response = original_method.call(params)
-          
           latency = ((Time.now - start_time) * 1000).round
+          
+          # Phase 3: Track actual usage
+          input_tokens = response.dig(:usage, :input_tokens) || 0
+          output_tokens = response.dig(:usage, :output_tokens) || 0
+          cost = config[:_client].send(:estimate_cost, model, input_tokens, output_tokens)
+          
+          config[:_client].send(:track_usage, model, 'anthropic', input_tokens, output_tokens, latency, cost)
+          
           span.set_attributes({
-            'response.input_tokens' => response.dig(:usage, :input_tokens),
-            'response.output_tokens' => response.dig(:usage, :output_tokens),
+            'response.input_tokens' => input_tokens,
+            'response.output_tokens' => output_tokens,
             'latency_ms' => latency
           })
           span.set_status(0)
           
+          puts "[AgentBill Cost Guard] ✓ Protected call completed: $#{format('%.4f', cost)}" if config[:debug]
           response
         rescue => e
           span.set_status(1, e.message)
@@ -83,19 +222,49 @@ module AgentBill
         end
       end
       
+      # Store reference to self for helper methods
+      @config[:_client] = self
       client
     end
 
-    def track_signal(event_name:, revenue: 0, data: {})
+    # Track a comprehensive signal with all 68 parameters
+    #
+    # Supports all parameters:
+    # - event_name (required)
+    # - data_source, timestamp
+    # - agent_external_id, customer_external_id, account_external_id, user_external_id,
+    #   order_external_id, session_id, conversation_id, thread_id
+    # - model, provider, prompt_hash, prompt_sample, response_sample, function_name, tool_name
+    # - prompt_tokens, completion_tokens, total_tokens, streaming_tokens, cached_tokens, reasoning_tokens
+    # - latency_ms, time_to_first_token, time_to_action_ms, queue_time_ms, processing_time_ms
+    # - revenue, cost, conversion_value, revenue_source
+    # - experiment_id, experiment_group, variant_id, ab_test_name
+    # - conversion_type, conversion_step, funnel_stage, goal_achieved
+    # - 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
+    # - custom_dimensions, metadata, data
+    #
+    # Example:
+    #   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"]
+    #   )
+    def track_signal(**params)
+      raise ArgumentError, "event_name is required" unless params[:event_name]
+      
       uri = URI("#{@config[:base_url] || 'https://bgwyprqxtdreuutzpbgw.supabase.co'}/functions/v1/record-signals")
       
-      payload = {
-        event_name: event_name,
-        revenue: revenue,
-        customer_id: @config[:customer_id],
-        timestamp: Time.now.to_i,
-        data: data
-      }
+      # Add timestamp if not provided
+      params[:timestamp] ||= Time.now.to_f
+      
+      # Remove nil values
+      payload = params.reject { |_, v| v.nil? }
       
       begin
         http = Net::HTTP.new(uri.host, uri.port)
@@ -109,12 +278,15 @@ module AgentBill
         response = http.request(request)
         
         if @config[:debug]
-          puts "[AgentBill] Signal tracked: #{event_name}, revenue: $#{revenue}"
+          puts "[AgentBill] Signal tracked: #{params[:event_name]}"
         end
+        
+        response.code == '200'
       rescue => e
         if @config[:debug]
           puts "[AgentBill] Failed to track signal: #{e.message}"
         end
+        false
       end
     end
 

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: agentbill-sdk
 version: !ruby/object:Gem::Version
-  version: 1.0.1
+  version: 1.0.3
 platform: ruby
 authors:
 - AgentBill
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2025-10-24 00:00:00.000000000 Z
+date: 2025-10-31 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bundler
@@ -41,7 +41,7 @@ dependencies:
 description: Automatically track and bill AI agent usage with zero-config instrumentation
   for OpenAI, Anthropic, and more
 email:
-- support@agentbill.com
+- dominic@agentbill.io
 executables: []
 extensions: []
 extra_rdoc_files: []
@@ -50,6 +50,7 @@ files:
 - ".github/ISSUE_TEMPLATE/feature_request.md"
 - ".github/pull_request_template.md"
 - ".github/workflows/ci.yml"
+- ".github/workflows/publish.yml"
 - ".gitignore"
 - ".rspec"
 - ".rubocop.yml"
@@ -63,7 +64,9 @@ files:
 - SECURITY.md
 - agentbill.gemspec
 - examples/anthropic_basic.rb
+- examples/manual_otel_tracking.rb
 - examples/openai_basic.rb
+- examples/zero_config.rb
 - lib/agentbill.rb
 - lib/agentbill/tracer.rb
 - lib/agentbill/version.rb