import { createGateway, wrapLanguageModel } from 'ai';
import { createGatewayV4Middleware } from '@ai-billing/gateway';
import { consoleDestination } from '@ai-billing/core';

const gateway = createGateway({ apiKey: process.env.AI_GATEWAY_API_KEY });

const billingMiddleware = createGatewayV4Middleware({
  destinations: [consoleDestination()],
});

const wrappedModel = wrapLanguageModel({
  model: gateway('gpt-5'),
  middleware: billingMiddleware,
});