import { GetRunContext, MiddlewareType } from '../../../context.js';
import { B as BaseAgent, c as AgentMeta, a as BaseAgentRunOptions } from '../../../base-BjzyQ8k6.js';
import { E as Emitter, C as Callback } from '../../../emitter-kHxkUxco.js';
import { BaseMemory } from '../../../memory/base.js';
import { Message, UserMessage } from '../../../backend/message.js';
import { StreamlitAgentTemplates } from './prompts.js';
import { TokenMemory } from '../../../memory/tokenMemory.js';
import { a as ChatModel, i as ChatModelOutput } from '../../../chat-CfTNfR60.js';
import '../../../internals/serializable.js';
import '../../../internals/types.js';
import '../../../internals/helpers/guards.js';
import '../../../internals/helpers/promise.js';
import '../../../errors.js';
import '../../../tools/base.js';
import 'ajv';
import 'promise-based-task';
import '../../../cache/base.js';
import '../../../internals/helpers/schema.js';
import 'zod';
import 'zod-to-json-schema';
import 'ai';
import '../../../template.js';
import '../../../backend/constants.js';
import '../../../logger/logger.js';
import 'pino';

interface StreamlitAgentInput {
    llm: ChatModel;
    memory: BaseMemory;
    templates?: Partial<StreamlitAgentTemplates>;
}
interface StreamlitRunInput {
    prompt: string | null;
}
interface Block {
    name: "text" | "app";
    content: string;
    start: number;
    end: number;
}
interface Result {
    raw: string;
    blocks: Block[];
}
interface StreamlitRunOutput {
    result: Result;
    message: Message;
    memory: BaseMemory;
}
interface StreamlitEvents {
    newToken: Callback<{
        delta: string;
        state: Readonly<{
            content: string;
        }>;
        chunk: ChatModelOutput;
    }>;
}
declare class StreamlitAgent extends BaseAgent<StreamlitRunInput, StreamlitRunOutput> {
    protected readonly input: StreamlitAgentInput;
    emitter: Emitter<StreamlitEvents>;
    constructor(input: StreamlitAgentInput);
    get meta(): AgentMeta;
    set memory(memory: BaseMemory);
    get memory(): BaseMemory;
    protected _run(input: StreamlitRunInput, _options: BaseAgentRunOptions, run: GetRunContext<typeof this>): Promise<StreamlitRunOutput>;
    protected prepare(input: StreamlitRunInput): Promise<{
        runMemory: TokenMemory;
        userMessage: UserMessage | null;
    }>;
    protected parse(raw: string): Result;
    createSnapshot(): {
        input: StreamlitAgentInput;
        isRunning: boolean;
        emitter: Emitter<unknown>;
        middlewares: MiddlewareType<any>[];
    };
}

export { StreamlitAgent, type StreamlitAgentInput, type StreamlitEvents, type StreamlitRunInput, type StreamlitRunOutput };