floomai/dist/FloomClient.js

Floom orchestrates & executes Generative AI pipelines, Empowering Developers and DevOps to focus on what matters.

"use strict";
var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
    function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
    return new (P || (P = Promise))(function (resolve, reject) {
        function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
        function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
        function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
        step((generator = generator.apply(thisArg, _arguments || [])).next());
    });
};
var __importDefault = (this && this.__importDefault) || function (mod) {
    return (mod && mod.__esModule) ? mod : { "default": mod };
};
Object.defineProperty(exports, "__esModule", { value: true });
exports.FloomClient = void 0;
const axios_1 = __importDefault(require("axios"));
const FloomRequest_1 = require("./models/FloomRequest");
const FloomResponse_1 = require("./models/FloomResponse");
const DataTransferType_1 = require("./models/DataTransferType");
class FloomClient {
    constructor(endpoint, apiKey) {
        this._url = endpoint;
        this._apiKey = apiKey;
    }
    run(pipelineId, chatId = "", prompt = null, variables = null, dataTransfer = DataTransferType_1.DataTransferType.Base64) {
        return __awaiter(this, void 0, void 0, function* () {
            const httpClient = axios_1.default.create({
                timeout: 160000,
                headers: {
                    "Api-Key": this._apiKey,
                    "Content-Type": "application/json"
                }
            });
            const url = `${this._url}/v1/Pipelines/Run`;
            var floomRequest = new FloomRequest_1.FloomRequest(pipelineId, prompt, chatId, variables, dataTransfer);
            try {
                const response = yield httpClient.post(url, JSON.stringify(floomRequest));
                return new FloomResponse_1.FloomResponse(response.data.messageId, response.data.chatId, response.data.values, response.data.processingTime, response.data.tokenUsage, null // Assuming no error in this case
                );
            }
            catch (error) {
                let errorMessage;
                if (axios_1.default.isAxiosError(error)) {
                    errorMessage = error.response ?
                        `Failed with status code ${error.response.status}: ${error.response.statusText}` :
                        `Connection error: ${error.message}`;
                }
                else {
                    errorMessage = `An unknown error occurred: ${error}`;
                }
                // Return a new instance of FloomResponse indicating an error
                return new FloomResponse_1.FloomResponse(null, chatId, null, null, null, errorMessage);
            }
        });
    }
}
exports.FloomClient = FloomClient;