deepinfra/dist/lib/models/base/text-generation.js

Official API wrapper for DeepInfra

"use strict";
Object.defineProperty(exports, "__esModule", { value: true });
exports.TextGeneration = void 0;
const axios_1 = require("axios");
const base_1 = require("../../../lib/models/base");
class TextGeneration extends base_1.BaseModel {
    constructor(modelName, authToken, config) {
        super(modelName, authToken, config);
    }
    async generate(body) {
        try {
            const response = await this.client.post(body);
            return response.data;
        }
        catch (error) {
            this.handleError(error);
            throw new Error("Failed to generate text");
        }
    }
    handleError(error) {
        if (error instanceof axios_1.AxiosError) {
            if (error.response) {
                console.error("Response data:", error.response.data);
                console.error("Status:", error.response.status);
                console.error("Headers:", error.response.headers);
            }
            else if (error.request) {
                console.error("No response received:", error.request);
            }
            else {
                console.error("Error message:", error.message);
            }
        }
        else {
            console.error("An unexpected error occurred:", error);
        }
    }
}
exports.TextGeneration = TextGeneration;