agentbill-sdk 5.0.1 → 7.17.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 7e9b487391274ecb94dbc031a40f6e30a167e6fa0e30bb257464f68e497871f2
-  data.tar.gz: cc6bc10bfdddf68981cd5454f80b4cffc6ea10a244f49f7e1f6582dd154e9b6c
+  metadata.gz: 55eed0a2b3bf82cdd116d56d4fdd3f41c8ab60166be0d456d1a5791a2bf12dbd
+  data.tar.gz: aae409ee8bd40d1ba774364ce570c6d5e4da4ad81847dc466433e755a2dc0e7d
 SHA512:
-  metadata.gz: a4409ffa8bac425a994a09c45d513877d5167e7ca9706c86229de4a6a50f8bf1da729ec1ccab5448c1dda808ed43cb04b2c314840f894801bd203cd8ca608aff
-  data.tar.gz: 285d37fb82ae7b3173474c72c47647e2585d8cf983730b16a24adf068a7baae3004078e10c79c493f66960d19921fd531d4aac1b7b6f35688ec7b9e1b541c987
+  metadata.gz: 5ca68a5403e625c73793d0d26b68795b56660f348e70d49b977981dc78f46429804cc25a736e1146763566472c78e8749263c1e42ba36d36670b18bbc709ead7
+  data.tar.gz: 831a7b5f9170db9a40b52411771a992a07d683c73fc31a350c69e1070d37f6007d3e44902b5c503dd254aa08e21f87f68ea7ac69965025900f9de73cabeb490b

data/.github/workflows/ci.yml

@@ -1,10 +1,11 @@
 name: CI
 
 on:
-  push:
-    branches: [ main, develop ]
   pull_request:
     branches: [ main, develop ]
+    paths:
+      - 'sdks/ruby/**'
+      - '.github/workflows/ci.yml'
 
 jobs:
   test:
@@ -13,8 +14,8 @@ jobs:
     
     strategy:
       matrix:
-        ruby-version: ['3.0', '3.1', '3.2', '3.3']
-        os: [ubuntu-latest, macos-latest, windows-latest]
+        ruby-version: ['3.3']
+        os: [ubuntu-latest]
     
     steps:
       - uses: actions/checkout@v4

data/CHANGELOG.md

@@ -5,6 +5,35 @@ 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).
 
+## [7.6.3] - 2025-12-17
+
+### Changed
+- **Version Alignment**: Aligned with Python SDK v7.6.3 wrap_openai() semantic cache fix
+- Ruby SDK already had cache hit check and cache population in `wrap_openai()`
+
+## [6.8.8] - 2025-12-09
+
+### Fixed
+- **Version Bump**: Aligned with Python SDK v6.8.8 wrapper trace context auto-generation fix
+
+## [6.8.5] - 2025-12-09
+
+### Fixed
+- **Tracer Version Alignment**: Updated `service.version` and `scope.version` in OTLP payload to 6.8.5
+
+## [6.8.4] - 2025-12-08
+
+### Fixed
+- **CRITICAL**: Fixed `agent_id` and `customer_id` not being sent as OTEL resource attributes
+- The otel-collector extracts these from resource attributes, not top-level payload
+- Added `agent_id` config option to SDK initialization
+
+## [6.8.3] - 2025-12-08
+
+### Fixed
+- **Version Bump**: Aligned with Python SDK v6.8.3 trace context fix for decorator pattern
+- No functional changes in Ruby SDK (fix was Python-specific)
+
 ## [5.0.1] - 2025-11-06
 
 ### Added

data/examples/ollama_basic.rb

@@ -0,0 +1,81 @@
+# AgentBill v6.0.0 - Ollama Integration Example
+# Tracks local Ollama chat and generation (metrics only, $0 cost)
+
+require 'agentbill'
+require 'net/http'
+require 'json'
+
+# Mock Ollama client (replace with actual Ollama SDK)
+class OllamaClient
+  def chat(params)
+    uri = URI('http://localhost:11434/api/chat')
+    
+    http = Net::HTTP.new(uri.host, uri.port)
+    
+    request = Net::HTTP::Post.new(uri.path)
+    request['Content-Type'] = 'application/json'
+    request.body = params.to_json
+    
+    response = http.request(request)
+    JSON.parse(response.body, symbolize_names: true)
+  end
+
+  def generate(params)
+    uri = URI('http://localhost:11434/api/generate')
+    
+    http = Net::HTTP.new(uri.host, uri.port)
+    
+    request = Net::HTTP::Post.new(uri.path)
+    request['Content-Type'] = 'application/json'
+    request.body = params.to_json
+    
+    response = http.request(request)
+    JSON.parse(response.body, symbolize_names: true)
+  end
+end
+
+# Initialize AgentBill
+agentbill = AgentBill::Client.init({
+  api_key: ENV['AGENTBILL_API_KEY'],
+  customer_id: 'customer-123',
+  debug: true
+})
+
+# Create Ollama client
+ollama = OllamaClient.new
+
+# Wrap with AgentBill
+ollama_wrapper = AgentBill::OllamaWrapper.new(ollama, agentbill.config, agentbill.tracer)
+tracked_ollama = ollama_wrapper.wrap
+
+puts "🦙 Using Ollama (local, free)...\n\n"
+
+# Example 1: Chat
+puts "💬 Chat example:"
+chat_response = tracked_ollama.chat({
+  model: 'llama2',
+  messages: [
+    {
+      role: 'user',
+      content: 'Why is the sky blue? Give a brief answer.'
+    }
+  ]
+})
+
+puts "✅ Response: #{chat_response[:message][:content]}"
+puts "📊 Tokens: prompt=#{chat_response[:prompt_eval_count]}, completion=#{chat_response[:eval_count]}"
+puts "💰 Cost: $0.00 (local model)\n\n"
+
+# Example 2: Generate
+puts "📝 Generate example:"
+generate_response = tracked_ollama.generate({
+  model: 'llama2',
+  prompt: 'Write a haiku about coding:'
+})
+
+puts "✅ Response: #{generate_response[:response]}"
+puts "📊 Tokens: prompt=#{generate_response[:prompt_eval_count]}, completion=#{generate_response[:eval_count]}"
+puts "💰 Cost: $0.00 (local model)\n\n"
+
+puts "ℹ️  Ollama runs locally, so there's no API cost."
+puts "   AgentBill tracks metrics for monitoring and analysis.\n"

data/examples/perplexity_basic.rb

@@ -0,0 +1,66 @@
+# AgentBill v6.0.0 - Perplexity AI Integration Example
+# Tracks Perplexity AI chat completions with online search
+
+require 'agentbill'
+require 'net/http'
+require 'json'
+
+# Mock Perplexity client (replace with actual Perplexity SDK)
+class PerplexityClient
+  def initialize(api_key)
+    @api_key = api_key
+  end
+
+  def chat_create(params)
+    uri = URI('https://api.perplexity.ai/chat/completions')
+    
+    http = Net::HTTP.new(uri.host, uri.port)
+    http.use_ssl = true
+    
+    request = Net::HTTP::Post.new(uri.path)
+    request['Authorization'] = "Bearer #{@api_key}"
+    request['Content-Type'] = 'application/json'
+    request.body = params.to_json
+    
+    response = http.request(request)
+    JSON.parse(response.body, symbolize_names: true)
+  end
+end
+
+# Initialize AgentBill
+agentbill = AgentBill::Client.init({
+  api_key: ENV['AGENTBILL_API_KEY'],
+  customer_id: 'customer-123',
+  debug: true
+})
+
+# Create Perplexity client
+perplexity = PerplexityClient.new(ENV['PERPLEXITY_API_KEY'])
+
+# Wrap with AgentBill
+perplexity_wrapper = AgentBill::PerplexityWrapper.new(perplexity, agentbill.config, agentbill.tracer)
+tracked_perplexity = perplexity_wrapper.wrap
+
+puts "🔍 Querying Perplexity AI with online search...\n\n"
+
+# Make a request - automatically tracked
+response = tracked_perplexity.chat_create({
+  model: 'llama-3.1-sonar-small-128k-online',
+  messages: [
+    {
+      role: 'system',
+      content: 'Be precise and concise.'
+    },
+    {
+      role: 'user',
+      content: 'What are the latest developments in AI for 2025?'
+    }
+  ],
+  temperature: 0.2,
+  top_p: 0.9,
+  max_tokens: 1000
+})
+
+puts "✅ Response: #{response[:choices][0][:message][:content]}"
+puts "\n📊 Usage: #{response[:usage]}"
+puts "💰 Cost automatically tracked in AgentBill dashboard\n"

data/lib/agentbill/agents.rb

@@ -0,0 +1,226 @@
+# AgentBill Agents Resource - CRUD operations with signal type management
+# 
+# Example:
+#   ab = AgentBill::Client.init(api_key: 'agb_...')
+#   
+#   # Create agent
+#   agent = ab.agents.create(name: 'Support Bot', description: 'Customer support')
+#   puts "API Key (save this!): #{agent['api_key']}"
+#   
+#   # Assign signal types
+#   ab.agents.assign_signal_types(
+#     agent_id: agent['id'],
+#     signal_type_names: ['ai_call', 'completion']
+#   )
+#   
+#   # List agents
+#   result = ab.agents.list(status: 'active')
+#   
+#   # Get signal types for agent
+#   signal_types = ab.agents.get_signal_types(agent['id'])
+
+require 'net/http'
+require 'json'
+
+module AgentBill
+  class AgentsResource
+    def initialize(config)
+      @config = config
+      @base_url = config[:base_url] || 'https://api.agentbill.io'
+      @api_key = config[:api_key]
+      @debug = config[:debug] || false
+    end
+
+    # List agents with optional status filter and pagination
+    #
+    # @param status [String, nil] Filter by status ('active', 'inactive', 'archived')
+    # @param limit [Integer] Number of agents to return (default: 50)
+    # @param offset [Integer] Offset for pagination (default: 0)
+    # @param search [String, nil] Search term to filter by name or description
+    # @return [Hash] Hash with 'data' (list of agents) and 'pagination' info
+    def list(status: nil, limit: 50, offset: 0, search: nil)
+      uri = URI("#{@base_url}/functions/v1/api-agents")
+      params = { limit: limit.to_s, offset: offset.to_s }
+      params[:status] = status if status
+      params[:search] = search if search
+      uri.query = URI.encode_www_form(params)
+
+      response = make_request(:get, uri)
+      handle_response(response, 'list agents')
+    end
+
+    # Create a new agent
+    # IMPORTANT: The response includes a one-time visible API key.
+    # Save this key securely - it cannot be retrieved again.
+    #
+    # @param name [String] Agent name (required)
+    # @param description [String, nil] Agent description
+    # @param agent_type [String, nil] Type of agent (e.g., 'support', 'sales', 'assistant')
+    # @param model [String, nil] Default model for this agent
+    # @param external_id [String, nil] Your external ID for this agent
+    # @param metadata [Hash, nil] Additional metadata
+    # @param signal_type_names [Array<String>, nil] Signal types to auto-assign (e.g., ['ai_call', 'completion'])
+    # @return [Hash] Created agent object with api_key (one-time visible)
+    def create(name:, description: nil, agent_type: nil, model: nil, external_id: nil, metadata: nil, signal_type_names: nil)
+      uri = URI("#{@base_url}/functions/v1/api-agents")
+      
+      payload = { name: name }
+      payload[:description] = description if description
+      payload[:agent_type] = agent_type if agent_type
+      payload[:model] = model if model
+      payload[:external_id] = external_id if external_id
+      payload[:metadata] = metadata if metadata
+      payload[:signal_type_names] = signal_type_names if signal_type_names
+
+      response = make_request(:post, uri, payload)
+      handle_response(response, 'create agent')
+    end
+
+    # Get an agent by ID or external_id
+    #
+    # @param agent_id [String] Agent UUID or external_id
+    # @param by_external_id [Boolean] If true, treat agent_id as external_id
+    # @return [Hash] Agent object
+    def get(agent_id, by_external_id: false)
+      uri = URI("#{@base_url}/functions/v1/api-agents")
+      params = by_external_id ? { external_id: agent_id } : { id: agent_id }
+      uri.query = URI.encode_www_form(params)
+
+      response = make_request(:get, uri)
+      result = handle_response(response, 'get agent')
+      
+      if result['data'].is_a?(Array)
+        raise "Agent not found" if result['data'].empty?
+        return result['data'][0]
+      end
+      result
+    end
+
+    # Update an agent
+    #
+    # @param agent_id [String] Agent UUID
+    # @param name [String, nil] New name
+    # @param description [String, nil] New description
+    # @param agent_type [String, nil] New agent type
+    # @param model [String, nil] New default model
+    # @param status [String, nil] New status ('active', 'inactive', 'archived')
+    # @param metadata [Hash, nil] New metadata (replaces existing)
+    # @return [Hash] Updated agent object
+    def update(agent_id, name: nil, description: nil, agent_type: nil, model: nil, status: nil, metadata: nil)
+      uri = URI("#{@base_url}/functions/v1/api-agents")
+      
+      payload = { id: agent_id }
+      payload[:name] = name if name
+      payload[:description] = description if description
+      payload[:agent_type] = agent_type if agent_type
+      payload[:model] = model if model
+      payload[:status] = status if status
+      payload[:metadata] = metadata if metadata
+
+      response = make_request(:patch, uri, payload)
+      handle_response(response, 'update agent')
+    end
+
+    # Delete an agent
+    #
+    # @param agent_id [String] Agent UUID to delete
+    def delete(agent_id)
+      uri = URI("#{@base_url}/functions/v1/api-agents")
+      payload = { id: agent_id }
+
+      response = make_request(:delete, uri, payload)
+      handle_response(response, 'delete agent')
+    end
+
+    # Assign signal types to an agent (explicit assignment)
+    # This method explicitly configures which signal types the agent can emit.
+    #
+    # @param agent_id [String, nil] Agent UUID (provide this OR external_agent_id)
+    # @param external_agent_id [String, nil] External agent ID (provide this OR agent_id)
+    # @param signal_type_ids [Array<String>, nil] List of signal type UUIDs to assign
+    # @param signal_type_names [Array<String>, nil] List of signal type names (e.g., ['ai_call', 'completion'])
+    # @return [Hash] Hash with agent_id, signal_type_ids, and assigned_count
+    def assign_signal_types(agent_id: nil, external_agent_id: nil, signal_type_ids: nil, signal_type_names: nil)
+      raise ArgumentError, "Either agent_id or external_agent_id is required" unless agent_id || external_agent_id
+      raise ArgumentError, "Either signal_type_ids or signal_type_names is required" unless signal_type_ids || signal_type_names
+
+      uri = URI("#{@base_url}/functions/v1/api-agent-signal-types")
+      
+      payload = {}
+      payload[:agent_id] = agent_id if agent_id
+      payload[:external_agent_id] = external_agent_id if external_agent_id
+      payload[:signal_type_ids] = signal_type_ids if signal_type_ids
+      payload[:signal_type_names] = signal_type_names if signal_type_names
+
+      response = make_request(:post, uri, payload)
+      handle_response(response, 'assign signal types')
+    end
+
+    # Get signal types assigned to an agent
+    #
+    # @param agent_id [String] Agent UUID or external_id
+    # @param by_external_id [Boolean] If true, treat agent_id as external_id
+    # @return [Hash] Hash with agent_id and list of signal_types
+    def get_signal_types(agent_id, by_external_id: false)
+      uri = URI("#{@base_url}/functions/v1/api-agent-signal-types")
+      params = by_external_id ? { external_agent_id: agent_id } : { agent_id: agent_id }
+      uri.query = URI.encode_www_form(params)
+
+      response = make_request(:get, uri)
+      handle_response(response, 'get signal types')
+    end
+
+    # Remove signal types from an agent
+    #
+    # @param agent_id [String] Agent UUID
+    # @param signal_type_ids [Array<String>, nil] Specific signal type IDs to remove. If nil, removes all.
+    def remove_signal_types(agent_id, signal_type_ids: nil)
+      uri = URI("#{@base_url}/functions/v1/api-agent-signal-types")
+      
+      payload = { agent_id: agent_id }
+      payload[:signal_type_ids] = signal_type_ids if signal_type_ids
+
+      response = make_request(:delete, uri, payload)
+      handle_response(response, 'remove signal types')
+    end
+
+    private
+
+    def make_request(method, uri, payload = nil)
+      http = Net::HTTP.new(uri.host, uri.port)
+      http.use_ssl = true
+      http.read_timeout = 30
+
+      case method
+      when :get
+        request = Net::HTTP::Get.new(uri)
+      when :post
+        request = Net::HTTP::Post.new(uri.path)
+        request.body = payload.to_json if payload
+      when :patch
+        request = Net::HTTP::Patch.new(uri.path)
+        request.body = payload.to_json if payload
+      when :delete
+        request = Net::HTTP::Delete.new(uri.path)
+        request.body = payload.to_json if payload
+      end
+
+      request['Content-Type'] = 'application/json'
+      request['X-API-Key'] = @api_key
+
+      http.request(request)
+    end
+
+    def handle_response(response, operation)
+      body = response.body ? JSON.parse(response.body) : {}
+      
+      unless %w[200 201 204].include?(response.code)
+        error_msg = body['error'] || "Failed to #{operation}"
+        raise error_msg
+      end
+
+      puts "[AgentBill] #{operation} successful" if @debug
+      body
+    end
+  end
+end

data/lib/agentbill/customers.rb

@@ -0,0 +1,164 @@
+# AgentBill Customers Resource - CRUD operations
+# 
+# Example:
+#   ab = AgentBill::Client.init(api_key: 'agb_...')
+#   
+#   # List customers
+#   result = ab.customers.list(limit: 10)
+#   
+#   # Create customer
+#   customer = ab.customers.create(name: 'Acme', email: 'a@b.com')
+#   
+#   # Get customer by ID
+#   customer = ab.customers.get('cust-123')
+#   
+#   # Update customer
+#   customer = ab.customers.update('cust-123', name: 'Acme Inc')
+#   
+#   # Delete customer
+#   ab.customers.delete('cust-123')
+
+require 'net/http'
+require 'json'
+
+module AgentBill
+  class CustomersResource
+    def initialize(config)
+      @config = config
+      @base_url = config[:base_url] || 'https://api.agentbill.io'
+      @api_key = config[:api_key]
+      @debug = config[:debug] || false
+    end
+
+    # List customers with pagination and search
+    #
+    # @param limit [Integer] Number of customers to return (default: 50)
+    # @param offset [Integer] Offset for pagination (default: 0)
+    # @param search [String, nil] Search term to filter by name or email
+    # @return [Hash] Hash with 'data' (list of customers) and 'pagination' info
+    def list(limit: 50, offset: 0, search: nil)
+      uri = URI("#{@base_url}/functions/v1/api-customers")
+      params = { limit: limit.to_s, offset: offset.to_s }
+      params[:search] = search if search
+      uri.query = URI.encode_www_form(params)
+
+      response = make_request(:get, uri)
+      handle_response(response, 'list customers')
+    end
+
+    # Create a new customer
+    #
+    # @param name [String] Customer name (required)
+    # @param email [String] Customer email (required)
+    # @param phone [String, nil] Phone number
+    # @param website [String, nil] Website URL
+    # @param external_id [String, nil] Your external ID for this customer
+    # @param metadata [Hash, nil] Additional metadata
+    # @return [Hash] Created customer object
+    def create(name:, email:, phone: nil, website: nil, external_id: nil, metadata: nil)
+      uri = URI("#{@base_url}/functions/v1/api-customers")
+      
+      payload = { name: name, email: email }
+      payload[:phone] = phone if phone
+      payload[:website] = website if website
+      payload[:external_id] = external_id if external_id
+      payload[:metadata] = metadata if metadata
+
+      response = make_request(:post, uri, payload)
+      handle_response(response, 'create customer')
+    end
+
+    # Get a customer by ID or external_id
+    #
+    # @param customer_id [String] Customer UUID or external_id
+    # @param by_external_id [Boolean] If true, treat customer_id as external_id
+    # @return [Hash] Customer object
+    def get(customer_id, by_external_id: false)
+      uri = URI("#{@base_url}/functions/v1/api-customers")
+      params = by_external_id ? { external_id: customer_id } : { id: customer_id }
+      uri.query = URI.encode_www_form(params)
+
+      response = make_request(:get, uri)
+      result = handle_response(response, 'get customer')
+      
+      if result['data'].is_a?(Array)
+        raise "Customer not found" if result['data'].empty?
+        return result['data'][0]
+      end
+      result
+    end
+
+    # Update a customer
+    #
+    # @param customer_id [String] Customer UUID
+    # @param name [String, nil] New name
+    # @param email [String, nil] New email
+    # @param phone [String, nil] New phone
+    # @param website [String, nil] New website
+    # @param metadata [Hash, nil] New metadata (replaces existing)
+    # @return [Hash] Updated customer object
+    def update(customer_id, name: nil, email: nil, phone: nil, website: nil, metadata: nil)
+      uri = URI("#{@base_url}/functions/v1/api-customers")
+      
+      payload = { id: customer_id }
+      payload[:name] = name if name
+      payload[:email] = email if email
+      payload[:phone] = phone if phone
+      payload[:website] = website if website
+      payload[:metadata] = metadata if metadata
+
+      response = make_request(:patch, uri, payload)
+      handle_response(response, 'update customer')
+    end
+
+    # Delete a customer
+    #
+    # @param customer_id [String] Customer UUID to delete
+    def delete(customer_id)
+      uri = URI("#{@base_url}/functions/v1/api-customers")
+      payload = { id: customer_id }
+
+      response = make_request(:delete, uri, payload)
+      handle_response(response, 'delete customer')
+    end
+
+    private
+
+    def make_request(method, uri, payload = nil)
+      http = Net::HTTP.new(uri.host, uri.port)
+      http.use_ssl = true
+      http.read_timeout = 30
+
+      case method
+      when :get
+        request = Net::HTTP::Get.new(uri)
+      when :post
+        request = Net::HTTP::Post.new(uri.path)
+        request.body = payload.to_json if payload
+      when :patch
+        request = Net::HTTP::Patch.new(uri.path)
+        request.body = payload.to_json if payload
+      when :delete
+        request = Net::HTTP::Delete.new(uri.path)
+        request.body = payload.to_json if payload
+      end
+
+      request['Content-Type'] = 'application/json'
+      request['X-API-Key'] = @api_key
+
+      http.request(request)
+    end
+
+    def handle_response(response, operation)
+      body = response.body ? JSON.parse(response.body) : {}
+      
+      unless %w[200 201 204].include?(response.code)
+        error_msg = body['error'] || "Failed to #{operation}"
+        raise error_msg
+      end
+
+      puts "[AgentBill] #{operation} successful" if @debug
+      body
+    end
+  end
+end