@qbraid-core/chat/dist/src/client.js

Client for the qBraid AI Chat service.

"use strict";
// Copyright (c) 2025, qBraid Development Team
// All rights reserved.
Object.defineProperty(exports, "__esModule", { value: true });
exports.ChatClient = void 0;
const base_1 = require("@qbraid-core/base");
class ChatClient extends base_1.QbraidClient {
    constructor(session) {
        super(session);
    }
    async createMessage(prompt, model = 'gpt-4o-mini', stream = false) {
        const response = await this.session.client.post('/chat', {
            prompt,
            model,
            stream,
        }, {
            responseType: stream ? 'stream' : 'json',
        });
        // stream: false, return type: ChatResponse
        // stream: true, return type: string
        if (stream) {
            return new Promise((resolve, reject) => {
                let data = '';
                response.data.on('data', (chunk) => {
                    data += chunk.toString();
                });
                response.data.on('end', () => {
                    console.log('Stream ended');
                    resolve(data);
                });
                response.data.on('error', (err) => {
                    reject(err);
                });
            });
        }
        else {
            return response.data;
        }
    }
    async getModels() {
        const response = await this.session.client.get('/chat/models');
        return response.data;
    }
}
exports.ChatClient = ChatClient;
//# sourceMappingURL=client.js.map