ai-utils.js
Version:
Build AI applications, chatbots, and agents with JavaScript and TypeScript.
23 lines (22 loc) • 872 B
JavaScript
;
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",
};