UNPKG

ai-utils.js

Version:

Build AI applications, chatbots, and agents with JavaScript and TypeScript.

23 lines (22 loc) 872 B
"use strict"; Object.defineProperty(exports, "__esModule", { value: true }); exports.extractSuccessfulModelCalls = void 0; function extractSuccessfulModelCalls(modelCallEvents) { return modelCallEvents .filter((event) => "status" in event && event.status === "success") .map((event) => ({ model: event.metadata.model, settings: event.settings, response: event.response, type: eventTypeToCostType[event.type], })); } exports.extractSuccessfulModelCalls = extractSuccessfulModelCalls; const eventTypeToCostType = { "image-generation-finished": "image-generation", "json-generation-finished": "json-generation", "text-embedding-finished": "text-embedding", "text-generation-finished": "text-generation", "text-streaming-finished": "text-streaming", "transcription-finished": "transcription", };