import type { BaseLanguageModelInput } from '@langchain/core/language_models/base';
import { type BaseMessage } from '@langchain/core/messages';
import type { ChatLunaChatModel } from 'koishi-plugin-chatluna/llm-core/platform/model';
import { type LivingMemoryLogger } from './logger';
export interface LivingMemoryModelCallContext {
    logger: LivingMemoryLogger;
    stage: string;
    attempt: number | (() => number);
    onResponse?: (response: BaseMessage) => void;
    promptLogging?: 'all' | 'first' | 'none';
    logResponseText?: boolean;
}
export declare const modelStageLabels: Record<string, string>;
export declare const invokeLoggedModel: (model: ChatLunaChatModel, input: BaseLanguageModelInput, runConfig: Parameters<ChatLunaChatModel['invoke']>[1], context: LivingMemoryModelCallContext) => Promise<import("@langchain/core/messages").AIMessageChunk>;
export declare const createLoggedModel: (model: ChatLunaChatModel, context: LivingMemoryModelCallContext) => ChatLunaChatModel;