AI Service

A package for wrapping AI entities (Functions, Workflows, or Agents) with pricing capabilities and Stripe integration for billing.

Installation

pnpm add ai-service

Usage

import { Service } from 'ai-service';
import { ai } from 'ai-functions';

// Create a priced function service
const pricedFunction = Service({
  entity: ai.generateText,
  type: 'function',
  pricing: {
    model: 'payPerUse',
    pricePerUse: 0.05 // $0.05 per use
  }
});

// Use the function as normal
const result = await pricedFunction('Generate a summary of this text...');

// Calculate the price for usage
const price = await pricedFunction.calculatePrice({
  inputs: 10,
  outputs: 5
});

// Record usage for billing
const usage = await pricedFunction.recordUsage({
  inputs: 10,
  outputs: 5
});

// Create a subscription for a customer
const subscription = await pricedFunction.createSubscription('cus_123456');

Pricing Models

The Service function supports various pricing models:

• Pay Per Use: Fixed price per function call
• Prepaid: Pay for a fixed amount of usage in advance
• Postpaid: Pay for usage after the fact
• Subscription: Recurring billing (per user or per instance)
• Outcome: Price based on the outcome of the function call

Stripe Integration

The package integrates with Stripe for billing. Make sure to set the STRIPE_API_KEY environment variable.