seraph-agent/dist/llm/__tests__/openai.test.js

An extremely lightweight, SRE autonomous AI agent for seamless integration with common observability tasks.

"use strict";
var __importDefault = (this && this.__importDefault) || function (mod) {
    return (mod && mod.__esModule) ? mod : { "default": mod };
};
Object.defineProperty(exports, "__esModule", { value: true });
const openai_1 = require("../openai");
const openai_2 = __importDefault(require("openai"));
jest.mock('openai');
describe('OpenAIProvider', () => {
    const mockedOpenAI = openai_2.default;
    it('should call the chat.completions.create method with the correct prompt', async () => {
        const create = jest.fn().mockResolvedValue({
            choices: [{ message: { content: 'test response' } }],
        });
        mockedOpenAI.prototype.chat = { completions: { create } };
        const config = {
            port: 8080,
            workers: 4,
            apiKey: 'test-key',
            serverApiKey: null,
        };
        const provider = new openai_1.OpenAIProvider(config);
        const response = await provider.generate('test prompt');
        expect(create).toHaveBeenCalledWith({
            model: 'gpt-4-turbo',
            messages: [{ role: 'user', content: 'test prompt' }],
        });
        expect(response).toBe('test response');
    });
});