daggerai/dist/src/llm/openai.js

A simple and powerful Typescript based agent framework to help businesses thrive in the AI Agent revolution.

"use strict";
var __importDefault = (this && this.__importDefault) || function (mod) {
    return (mod && mod.__esModule) ? mod : { "default": mod };
};
Object.defineProperty(exports, "__esModule", { value: true });
exports.ChatOpenAI = void 0;
const openai_1 = __importDefault(require("openai"));
const prompts_1 = require("../core/prompts");
class ChatOpenAI {
    api;
    params;
    constructor(params) {
        this.api = new openai_1.default({
            apiKey: params.apiKey || process.env.OPENAI_API_KEY,
        });
        this.params = params;
    }
    async invoke(prompt) {
        const completion = await this.api.chat.completions.create({
            model: this.params.model,
            max_tokens: this.params.maxTokens || 4000,
            stop: [prompts_1.SQUAD_PROMPTS.observation],
            messages: [{ role: 'user', content: prompt }],
        });
        return completion.choices[0]?.message.content || '';
    }
}
exports.ChatOpenAI = ChatOpenAI;
//# sourceMappingURL=openai.js.map