import { Message, PromptFunctions, PromptMemory, PromptSection, Tokenizer } from "promptrix"; import { PromptCompletionModel, PromptResponse, PromptResponseStatus } from "./types"; /** * A test model that can be used to test the prompt completion system. */ export declare class TestModel implements PromptCompletionModel { /** * * @param status Optional. Status of the prompt response. Defaults to `success`. * @param response Optional. Response to the prompt. Defaults to `{ role: 'assistant', content: 'Hello World' }`. */ constructor(status?: PromptResponseStatus, response?: string | Message); /** * Status of the prompt response. */ status: PromptResponseStatus; /** * Response to the prompt. */ response: string | Message; /** * Returns the response to the prompt. * @param memory Memory to use when rendering the prompt. * @param functions Functions to use when rendering the prompt. * @param tokenizer Tokenizer to use when rendering the prompt. * @param prompt Prompt to complete. * @param prompt_options Options for completing the prompt. * @returns A `PromptResponse` with the status and message. */ completePrompt(memory: PromptMemory, functions: PromptFunctions, tokenizer: Tokenizer, prompt: PromptSection): Promise; } //# sourceMappingURL=TestModel.d.ts.map