deepinfra/dist/lib/models/base/custom-model.js

Official API wrapper for DeepInfra

"use strict";
Object.defineProperty(exports, "__esModule", { value: true });
exports.CustomModel = void 0;
const base_model_1 = require("../../../lib/models/base/base-model");
class CustomModel extends base_model_1.BaseModel {
    constructor(modelName, authToken, config) {
        super(modelName, authToken, config);
    }
    async generate(body) {
        const response = await this.client.post(body);
        return response.data;
    }
}
exports.CustomModel = CustomModel;