import { describe, expect, it, vi } from 'vitest';

const compile = vi.fn().mockReturnValue([{ role: 'user', content: 'hi' }]);
const promptGet = vi
  .fn()
  .mockResolvedValue({ name: 'greet', version: 2, config: { model: 'llama3' }, compile });

vi.mock('@langfuse/client', () => ({
  LangfuseClient: vi.fn().mockImplementation(() => ({ prompt: { get: promptGet } })),
}));

import { LangfusePromptGateway } from '../../../src/core/langfuse/LangfusePromptGateway';

describe('LangfusePromptGateway', () => {
  it('fetches a chat prompt from Langfuse and adapts it to RawPrompt', async () => {
    const gateway = new LangfusePromptGateway({ publicKey: 'pk', secretKey: 'sk', baseUrl: 'http://lf' });
    const raw = await gateway.fetch('greet');

    expect(promptGet).toHaveBeenCalledWith('greet', { type: 'chat' });
    expect(raw.name).toBe('greet');
    expect(raw.version).toBe(2);
    expect(raw.config).toEqual({ model: 'llama3' });
    expect(raw.compile({ who: 'alice' })).toEqual([{ role: 'user', content: 'hi' }]);
  });
});