import { type AssistantMessage, type AssistantMessageEventStream } from "@earendil-works/pi-ai";
import type { ResponseStreamEvent } from "openai/resources/responses/responses.js";
import type { GrammarToolInputJsonBuffer } from "../../constrained-sampling.js";
import { type ResponsesToolCall } from "../native-items.ts";
import type { OpenAIResponsesStreamOptions } from "../shared.ts";
type ThinkingBlock = Extract<AssistantMessage["content"][number], {
    type: "thinking";
}>;
type TextBlock = Extract<AssistantMessage["content"][number], {
    type: "text";
}>;
type ToolCallBlock = ResponsesToolCall & {
    partialJson?: string | undefined;
};
type ReasoningState = {
    kind: "reasoning";
    blockIndex: number;
    block: ThinkingBlock;
    summaryParts: Map<number, {
        text: string;
    }>;
    contentParts: Map<number, {
        text: string;
    }>;
};
type MessageState = {
    kind: "message";
    blockIndex: number;
    block: TextBlock;
    parts: Map<number, {
        type: "output_text" | "refusal";
        text: string;
    }>;
};
type FunctionCallState = {
    kind: "function_call";
    blockIndex: number;
    block: ToolCallBlock;
};
export type CustomToolCallState = {
    kind: "custom_tool_call";
    blockIndex: number;
    block: ToolCallBlock;
    input: string;
    property: string;
    jsonBuffer: GrammarToolInputJsonBuffer;
};
export type OutputState = ReasoningState | MessageState | FunctionCallState | CustomToolCallState;
export interface ResponsesStreamOutput {
    output: AssistantMessage;
    stream: Pick<AssistantMessageEventStream, "push">;
    outputStates: Map<number, OutputState>;
    options: OpenAIResponsesStreamOptions | undefined;
}
export declare function startResponsesOutputItem(event: Extract<ResponseStreamEvent, {
    type: "response.output_item.added";
}>, state: ResponsesStreamOutput): void;
export declare function finishResponsesOutputItem(event: Extract<ResponseStreamEvent, {
    type: "response.output_item.done";
}>, state: ResponsesStreamOutput): void;
export {};