agentbill-sdk 1.0.2 → 2.0.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 17b96a78039bd5ad557ee6abc8606a2240e9e4d9e69a496fd1fa99bd3d1814a1
-  data.tar.gz: e126680a47d95e27ec984f94cba1631ed86c056111a1b40f2e7226c9cb274432
+  metadata.gz: d9f3faf536ea2cdd78d8d97f3200ab80bab8889654b924a2f035a35685aa210f
+  data.tar.gz: 4397d5f95b5e13bee80a28f7a27165a4f8b5e60c5f478bbfd19b11030651413f
 SHA512:
-  metadata.gz: 2d465e3290f46ce834684c4960d78b61041d562f6378b2c234df7713d41d9927809559f2c907ab49fa10110c24c7ab16f50ec4a3203e2d0b8e64cb605c980637
-  data.tar.gz: c29f3c706608e6e43659aff341a89d6f809ebda394c18edf1e7f6bd99c3256569bb2dbb2714bd855ec8c248cea3d7a0c7ade528e902f87da8f9064d90a5076c6
+  metadata.gz: 625bad2690c2dd452d299a5664cca0db3d2e806ab0318ce0a0ce7330b6ce6b873db4213b63e2123286ade6593b13ece94de3b1550e96f5f26b4dd2122c6921ba
+  data.tar.gz: 3257333096fda49a819738485c457e69741c556b35281648890533e58a3819d631289e9fc170575a55bf598d41a6d2aafe3c7e42bee8c7ea4c25b95bf9937672

data/CHANGELOG.md

@@ -1,4 +1,5 @@
 # Changelog
+# Force sync - Updated to 1.0.2
 
 All notable changes to this project will be documented in this file.
 

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/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.2"
+  VERSION = "2.0.0"
 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
         
-        span = @tracer.start_span('anthropic.message', {
-          'model' => params[:model] || 'unknown',
+        # 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
+        
+        # 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,6 +222,8 @@ module AgentBill
         end
       end
       
+      # Store reference to self for helper methods
+      @config[:_client] = self
       client
     end
 

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: agentbill-sdk
 version: !ruby/object:Gem::Version
-  version: 1.0.2
+  version: 2.0.0
 platform: ruby
 authors:
 - AgentBill
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2025-10-27 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: []
@@ -66,6 +66,7 @@ files:
 - 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