@awesome-fe/translate/utils/create-embedding.js

Translation utils

"use strict";
Object.defineProperty(exports, "__esModule", { value: true });
exports.createEmbedding = void 0;
const openai_1 = require("openai");
async function createEmbedding(inputs) {
    if (!inputs.length) {
        return [];
    }
    const configuration = new openai_1.Configuration({ apiKey: process.env.OPENAI_API_KEY });
    const openai = new openai_1.OpenAIApi(configuration);
    const response = await openai.createEmbedding({
        input: inputs,
        model: 'text-embedding-ada-002',
    });
    const tokens = response.data.usage.total_tokens;
    console.log(`Total tokens: ${tokens}`);
    const embeddings = response.data.data.map(embedding => embedding.embedding);
    return embeddings;
}
exports.createEmbedding = createEmbedding;
//# sourceMappingURL=create-embedding.js.map