@wrcb/cb-common 1.0.837 → 1.0.839

package/build/events/ai/aiRequestMadeEvent.d.ts

@@ -0,0 +1,23 @@
+import { Subjects } from '../subjects';
+export type AIProvider = 'openai' | 'cerebras';
+export type AIRequestType = 'chat' | 'tts' | 'transcribe' | 'image-generate' | 'image-edit';
+export interface AIRequestMadeEvent {
+    subject: Subjects.AIRequestMade;
+    data: {
+        service: string;
+        provider: AIProvider;
+        type: AIRequestType;
+        model: string;
+        promptTokens: number;
+        completionTokens: number;
+        totalTokens: number;
+        costUSD: number;
+        durationMs: number;
+        requestedAt: string;
+        prompt: string;
+        inputChars?: number;
+        imageSize?: string;
+        success: boolean;
+        errorMessage?: string;
+    };
+}

package/build/events/ai/aiRequestMadeEvent.js

@@ -0,0 +1,2 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });

package/build/events/ai/aiRequestMadePublisher.d.ts

@@ -0,0 +1,6 @@
+import { Publisher } from '../basePublisher';
+import { AIRequestMadeEvent } from './aiRequestMadeEvent';
+import { Subjects } from '../subjects';
+export declare class AIRequestMadePublisher extends Publisher<AIRequestMadeEvent> {
+    subject: Subjects.AIRequestMade;
+}

package/build/events/ai/aiRequestMadePublisher.js

@@ -0,0 +1,12 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.AIRequestMadePublisher = void 0;
+const basePublisher_1 = require("../basePublisher");
+const subjects_1 = require("../subjects");
+class AIRequestMadePublisher extends basePublisher_1.Publisher {
+    constructor() {
+        super(...arguments);
+        this.subject = subjects_1.Subjects.AIRequestMade;
+    }
+}
+exports.AIRequestMadePublisher = AIRequestMadePublisher;

package/build/events/ai/index.d.ts

@@ -0,0 +1,2 @@
+export * from './aiRequestMadeEvent';
+export * from './aiRequestMadePublisher';

package/build/events/ai/index.js

@@ -0,0 +1,18 @@
+"use strict";
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __exportStar = (this && this.__exportStar) || function(m, exports) {
+    for (var p in m) if (p !== "default" && !Object.prototype.hasOwnProperty.call(exports, p)) __createBinding(exports, m, p);
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+__exportStar(require("./aiRequestMadeEvent"), exports);
+__exportStar(require("./aiRequestMadePublisher"), exports);

package/build/events/subjects.d.ts

@@ -61,5 +61,6 @@ export declare enum Subjects {
     SMATopicApproved = "sma:topic:approved",
     SMAVideoReady = "sma:video:ready",
     SMAPostPublished = "sma:post:published",
-    SMAPostFailed = "sma:post:failed"
+    SMAPostFailed = "sma:post:failed",
+    AIRequestMade = "ai:request:made"
 }

package/build/events/subjects.js

@@ -73,4 +73,6 @@ var Subjects;
     Subjects["SMAVideoReady"] = "sma:video:ready";
     Subjects["SMAPostPublished"] = "sma:post:published";
     Subjects["SMAPostFailed"] = "sma:post:failed";
+    // ai tracking
+    Subjects["AIRequestMade"] = "ai:request:made";
 })(Subjects || (exports.Subjects = Subjects = {}));

package/build/index.d.ts

@@ -56,6 +56,8 @@ export * from './events/baseListener';
 export * from './events/basePublisher';
 export * from './events/youtube-dark';
 export * from './events/social-media-automator';
+export * from './events/ai';
+export * from './services/ai/AIClient';
 export * from './services/TenantDataService';
 export * from './services/CurrencyService';
 export * from './helpers/calculateBonus';

package/build/index.js

@@ -78,6 +78,8 @@ __exportStar(require("./events/baseListener"), exports);
 __exportStar(require("./events/basePublisher"), exports);
 __exportStar(require("./events/youtube-dark"), exports);
 __exportStar(require("./events/social-media-automator"), exports);
+__exportStar(require("./events/ai"), exports);
+__exportStar(require("./services/ai/AIClient"), exports);
 __exportStar(require("./services/TenantDataService"), exports);
 __exportStar(require("./services/CurrencyService"), exports);
 __exportStar(require("./helpers/calculateBonus"), exports);

package/build/services/ai/AIClient.d.ts

@@ -0,0 +1,58 @@
+import OpenAI from 'openai';
+import { Stan } from 'node-nats-streaming';
+import { AIProvider } from '../../events/ai/aiRequestMadeEvent';
+export interface ChatParams {
+    provider: AIProvider;
+    model: string;
+    messages: OpenAI.ChatCompletionMessageParam[];
+    maxTokens?: number;
+    temperature?: number;
+    responseFormat?: {
+        type: 'json_object' | 'text';
+    };
+}
+export interface TTSParams {
+    model: string;
+    voice: string;
+    input: string;
+    speed?: number;
+    instructions?: string;
+    responseFormat?: string;
+}
+export interface TranscribeParams {
+    audioBuffer: Buffer;
+    mimeType?: string;
+}
+export interface ImageGenerateParams {
+    model: string;
+    prompt: string;
+    size: string;
+    quality?: string;
+    n?: number;
+}
+export interface ImageEditParams {
+    model: string;
+    imageBuffer: Buffer;
+    prompt: string;
+    size: string;
+    n?: number;
+}
+export interface ChatResult {
+    content: string;
+    promptTokens: number;
+    completionTokens: number;
+    totalTokens: number;
+}
+export declare class AIClient {
+    private readonly natsClient;
+    private readonly service;
+    private readonly openai;
+    private readonly cerebras;
+    constructor(natsClient: Stan, service: string, openaiApiKey: string, cerebrasApiKey?: string);
+    private emit;
+    chat(params: ChatParams): Promise<ChatResult>;
+    tts(params: TTSParams): Promise<Buffer>;
+    transcribe(params: TranscribeParams): Promise<string>;
+    generateImage(params: ImageGenerateParams): Promise<string>;
+    editImage(params: ImageEditParams): Promise<string>;
+}

package/build/services/ai/AIClient.js

@@ -0,0 +1,214 @@
+"use strict";
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
+    Object.defineProperty(o, "default", { enumerable: true, value: v });
+}) : function(o, v) {
+    o["default"] = v;
+});
+var __importStar = (this && this.__importStar) || function (mod) {
+    if (mod && mod.__esModule) return mod;
+    var result = {};
+    if (mod != null) for (var k in mod) if (k !== "default" && Object.prototype.hasOwnProperty.call(mod, k)) __createBinding(result, mod, k);
+    __setModuleDefault(result, mod);
+    return result;
+};
+var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
+    function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
+    return new (P || (P = Promise))(function (resolve, reject) {
+        function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
+        function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
+        function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
+        step((generator = generator.apply(thisArg, _arguments || [])).next());
+    });
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.AIClient = void 0;
+const openai_1 = __importStar(require("openai"));
+const aiRequestMadePublisher_1 = require("../../events/ai/aiRequestMadePublisher");
+const COST_PER_1K = {
+    'gpt-4o': { input: 0.0025, output: 0.010 },
+    'gpt-4o-mini': { input: 0.00015, output: 0.0006 },
+    'gpt-oss-120b': { input: 0, output: 0 },
+    'whisper-1': { input: 0.006, output: 0 },
+    'tts-1': { input: 0.015, output: 0 },
+    'tts-1-hd': { input: 0.030, output: 0 },
+    'gpt-4o-mini-tts': { input: 0.012, output: 0 },
+    'dall-e-3': { input: 0.040, output: 0 },
+    'dall-e-2': { input: 0.020, output: 0 },
+    'gpt-image-1': { input: 0.067, output: 0 },
+};
+function calcCost(model, promptTokens, completionTokens) {
+    var _a;
+    const rates = (_a = COST_PER_1K[model]) !== null && _a !== void 0 ? _a : { input: 0, output: 0 };
+    return (promptTokens / 1000) * rates.input + (completionTokens / 1000) * rates.output;
+}
+class AIClient {
+    constructor(natsClient, service, openaiApiKey, cerebrasApiKey) {
+        this.natsClient = natsClient;
+        this.service = service;
+        this.openai = new openai_1.default({ apiKey: openaiApiKey });
+        this.cerebras = new openai_1.default({
+            apiKey: cerebrasApiKey !== null && cerebrasApiKey !== void 0 ? cerebrasApiKey : '',
+            baseURL: 'https://api.cerebras.ai/v1',
+        });
+    }
+    emit(data) {
+        return __awaiter(this, void 0, void 0, function* () {
+            try {
+                yield new aiRequestMadePublisher_1.AIRequestMadePublisher(this.natsClient).publish({
+                    service: this.service,
+                    provider: data.provider,
+                    type: data.type,
+                    model: data.model,
+                    promptTokens: data.promptTokens,
+                    completionTokens: data.completionTokens,
+                    totalTokens: data.promptTokens + data.completionTokens,
+                    costUSD: data.costUSD,
+                    durationMs: data.durationMs,
+                    requestedAt: data.requestedAt,
+                    prompt: data.prompt,
+                    inputChars: data.inputChars,
+                    imageSize: data.imageSize,
+                    success: data.success,
+                    errorMessage: data.errorMessage,
+                });
+            }
+            catch (_a) {
+                // nunca deixa o emit de evento quebrar a requisição principal
+            }
+        });
+    }
+    chat(params) {
+        return __awaiter(this, void 0, void 0, function* () {
+            var _a, _b, _c, _d, _e, _f, _g;
+            const { provider, model, messages, maxTokens, temperature, responseFormat } = params;
+            const client = provider === 'cerebras' ? this.cerebras : this.openai;
+            const requestedAt = new Date().toISOString();
+            const t0 = Date.now();
+            const promptSummary = messages.map(m => { var _a; return `[${m.role}] ${String((_a = m.content) !== null && _a !== void 0 ? _a : '').slice(0, 200)}`; }).join(' | ').slice(0, 500);
+            try {
+                const response = yield client.chat.completions.create(Object.assign(Object.assign(Object.assign({ model,
+                    messages }, (maxTokens !== undefined && { max_tokens: maxTokens })), (temperature !== undefined && { temperature })), (responseFormat !== undefined && { response_format: responseFormat })));
+                const promptTokens = (_b = (_a = response.usage) === null || _a === void 0 ? void 0 : _a.prompt_tokens) !== null && _b !== void 0 ? _b : 0;
+                const completionTokens = (_d = (_c = response.usage) === null || _c === void 0 ? void 0 : _c.completion_tokens) !== null && _d !== void 0 ? _d : 0;
+                const costUSD = calcCost(model, promptTokens, completionTokens);
+                yield this.emit({ provider, type: 'chat', model, promptTokens, completionTokens, costUSD, durationMs: Date.now() - t0, requestedAt, prompt: promptSummary, success: true });
+                return {
+                    content: (_g = (_f = (_e = response.choices[0]) === null || _e === void 0 ? void 0 : _e.message) === null || _f === void 0 ? void 0 : _f.content) !== null && _g !== void 0 ? _g : '',
+                    promptTokens,
+                    completionTokens,
+                    totalTokens: promptTokens + completionTokens,
+                };
+            }
+            catch (err) {
+                const errorMessage = err instanceof Error ? err.message : String(err);
+                yield this.emit({ provider, type: 'chat', model, promptTokens: 0, completionTokens: 0, costUSD: 0, durationMs: Date.now() - t0, requestedAt, prompt: promptSummary, success: false, errorMessage });
+                throw err;
+            }
+        });
+    }
+    tts(params) {
+        return __awaiter(this, void 0, void 0, function* () {
+            var _a, _b;
+            const { model, voice, input, speed, instructions, responseFormat } = params;
+            const requestedAt = new Date().toISOString();
+            const t0 = Date.now();
+            const costUSD = (input.length / 1000) * ((_b = (_a = COST_PER_1K[model]) === null || _a === void 0 ? void 0 : _a.input) !== null && _b !== void 0 ? _b : 0.030);
+            try {
+                const response = yield this.openai.audio.speech.create(Object.assign(Object.assign({ model, voice: voice, input, response_format: (responseFormat !== null && responseFormat !== void 0 ? responseFormat : 'mp3') }, (speed !== undefined && { speed })), (instructions !== undefined && { instructions })));
+                const buffer = Buffer.from(yield response.arrayBuffer());
+                yield this.emit({ provider: 'openai', type: 'tts', model, promptTokens: 0, completionTokens: 0, costUSD, durationMs: Date.now() - t0, requestedAt, prompt: input.slice(0, 200), inputChars: input.length, success: true });
+                return buffer;
+            }
+            catch (err) {
+                const errorMessage = err instanceof Error ? err.message : String(err);
+                yield this.emit({ provider: 'openai', type: 'tts', model, promptTokens: 0, completionTokens: 0, costUSD: 0, durationMs: Date.now() - t0, requestedAt, prompt: input.slice(0, 200), inputChars: input.length, success: false, errorMessage });
+                throw err;
+            }
+        });
+    }
+    transcribe(params) {
+        return __awaiter(this, void 0, void 0, function* () {
+            const { audioBuffer, mimeType } = params;
+            const requestedAt = new Date().toISOString();
+            const t0 = Date.now();
+            const estimatedMinutes = audioBuffer.length / 40000 / 60;
+            const costUSD = estimatedMinutes * 0.006;
+            try {
+                const file = yield (0, openai_1.toFile)(audioBuffer, 'audio.m4a', { type: mimeType !== null && mimeType !== void 0 ? mimeType : 'audio/m4a' });
+                const response = yield this.openai.audio.transcriptions.create({ model: 'whisper-1', file });
+                yield this.emit({ provider: 'openai', type: 'transcribe', model: 'whisper-1', promptTokens: 0, completionTokens: 0, costUSD, durationMs: Date.now() - t0, requestedAt, prompt: `audio ${audioBuffer.length}B`, success: true });
+                return response.text;
+            }
+            catch (err) {
+                const errorMessage = err instanceof Error ? err.message : String(err);
+                yield this.emit({ provider: 'openai', type: 'transcribe', model: 'whisper-1', promptTokens: 0, completionTokens: 0, costUSD: 0, durationMs: Date.now() - t0, requestedAt, prompt: `audio ${audioBuffer.length}B`, success: false, errorMessage });
+                throw err;
+            }
+        });
+    }
+    generateImage(params) {
+        return __awaiter(this, void 0, void 0, function* () {
+            var _a, _b;
+            const { model, prompt, size, quality, n } = params;
+            const requestedAt = new Date().toISOString();
+            const t0 = Date.now();
+            const costUSD = (_b = (_a = COST_PER_1K[model]) === null || _a === void 0 ? void 0 : _a.input) !== null && _b !== void 0 ? _b : 0.040;
+            try {
+                const response = yield this.openai.images.generate(Object.assign(Object.assign({ model,
+                    prompt, size: size }, (quality !== undefined && { quality: quality })), { n: n !== null && n !== void 0 ? n : 1, response_format: 'b64_json' }));
+                const b64 = response.data[0].b64_json;
+                yield this.emit({ provider: 'openai', type: 'image-generate', model, promptTokens: 0, completionTokens: 0, costUSD, durationMs: Date.now() - t0, requestedAt, prompt: prompt.slice(0, 300), imageSize: size, success: true });
+                return b64;
+            }
+            catch (err) {
+                const errorMessage = err instanceof Error ? err.message : String(err);
+                yield this.emit({ provider: 'openai', type: 'image-generate', model, promptTokens: 0, completionTokens: 0, costUSD: 0, durationMs: Date.now() - t0, requestedAt, prompt: prompt.slice(0, 300), imageSize: size, success: false, errorMessage });
+                throw err;
+            }
+        });
+    }
+    editImage(params) {
+        return __awaiter(this, void 0, void 0, function* () {
+            var _a, _b;
+            const { model, imageBuffer, prompt, size, n } = params;
+            const requestedAt = new Date().toISOString();
+            const t0 = Date.now();
+            const costUSD = (_b = (_a = COST_PER_1K[model]) === null || _a === void 0 ? void 0 : _a.input) !== null && _b !== void 0 ? _b : 0.020;
+            try {
+                if (!globalThis.File) {
+                    const { File: NodeFile } = yield Promise.resolve().then(() => __importStar(require('node:buffer')));
+                    globalThis.File = NodeFile;
+                }
+                const imageFile = yield (0, openai_1.toFile)(imageBuffer, 'image.png', { type: 'image/png' });
+                const response = yield this.openai.images.edit({
+                    model,
+                    image: imageFile,
+                    prompt,
+                    size: size,
+                    n: n !== null && n !== void 0 ? n : 1,
+                    response_format: 'b64_json',
+                });
+                const b64 = response.data[0].b64_json;
+                yield this.emit({ provider: 'openai', type: 'image-edit', model, promptTokens: 0, completionTokens: 0, costUSD, durationMs: Date.now() - t0, requestedAt, prompt: prompt.slice(0, 300), imageSize: size, success: true });
+                return b64;
+            }
+            catch (err) {
+                const errorMessage = err instanceof Error ? err.message : String(err);
+                yield this.emit({ provider: 'openai', type: 'image-edit', model, promptTokens: 0, completionTokens: 0, costUSD: 0, durationMs: Date.now() - t0, requestedAt, prompt: prompt.slice(0, 300), imageSize: size, success: false, errorMessage });
+                throw err;
+            }
+        });
+    }
+}
+exports.AIClient = AIClient;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@wrcb/cb-common",
-  "version": "1.0.837",
+  "version": "1.0.839",
   "description": "Common resources between services",
   "main": "./build/index.js",
   "types": "./build/index.d.ts",
@@ -27,16 +27,16 @@
   "author": "Willian Rios",
   "license": "ISC",
   "devDependencies": {
+    "@types/cookie-session": "^2.0.49",
+    "@types/express": "^4.17.21",
     "@types/express-rate-limit": "^5.1.3",
     "@types/fluent-ffmpeg": "^2.1.27",
+    "@types/jsonwebtoken": "^9.0.6",
     "@types/mime-types": "^3.0.1",
     "@types/multer": "^2.0.0",
+    "@types/supertest": "^6.0.2",
     "@typescript-eslint/eslint-plugin": "^8.29.1",
     "@typescript-eslint/parser": "^8.29.1",
-    "@types/cookie-session": "^2.0.49",
-    "@types/express": "^4.17.21",
-    "@types/jsonwebtoken": "^9.0.6",
-    "@types/supertest": "^6.0.2",
     "eslint": "^9.24.0",
     "eslint-config-prettier": "^10.1.2",
     "eslint-plugin-prettier": "^5.2.6",
@@ -59,6 +59,7 @@
     "mongoose": "^8.6.3",
     "multer": "^2.0.2",
     "node-nats-streaming": "^0.3.2",
+    "openai": "^6.42.0",
     "rate-limit-redis": "^4.2.0",
     "sharp": "^0.34.3",
     "uuid": "^11.1.0"