agentbill-sdk 2.0.1 → 7.17.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 552854759cf5b06dca09f5a250402634f11f3077b60ec0306f7e1b61817b0281
-  data.tar.gz: 31ab1d0c1a940e0d1bfa66368d81a3cb3026c9b70868eda69b04b13d106134a0
+  metadata.gz: 55eed0a2b3bf82cdd116d56d4fdd3f41c8ab60166be0d456d1a5791a2bf12dbd
+  data.tar.gz: aae409ee8bd40d1ba774364ce570c6d5e4da4ad81847dc466433e755a2dc0e7d
 SHA512:
-  metadata.gz: e8cb6296cc8522aae786ffb0927379ddaed1a08d0b74cca6d11342b964851d99186878da310609e912c178d9a893f589314f802d0306327d6baa2d74cb9a8d0e
-  data.tar.gz: c47673c3b91a0c4e4924f52d14ec1d0821824d2d2bbddf75902332c5a383aa9a1b2eee5095cd07b83dd193715aaf69bed6993e8f4706f906021a16cb5614338c
+  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

@@ -1,27 +1,86 @@
 # 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
+## [7.6.3] - 2025-12-17
 
 ### Changed
-- Version bump for consistency across SDKs
+- **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
+- **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 +92,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/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/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/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