learning_ai_common_plat/packages/llm/src/testing.ts

19 lines
402 B
TypeScript

/**
* Test helpers for @bytelyst/llm.
*/
import { setLLM, _resetLLM } from './factory.js';
import { MockLLMProvider } from './providers/mock.js';
export function setTestLLMProvider(): MockLLMProvider {
const provider = new MockLLMProvider();
setLLM(provider);
return provider;
}
export function resetTestLLM(): void {
_resetLLM();
}
export { MockLLMProvider } from './providers/mock.js';