import { GenericCapabilitySelections } from '@theia/ai-core';
import { ChatAgentLocation } from './chat-agents';
import { ChatSessionSettings, ResponseTokenUsage } from './chat-model';
export interface SerializableChangeSetElement {
    kind?: string;
    uri: string;
    name?: string;
    icon?: string;
    additionalInfo?: string;
    state?: 'pending' | 'applied' | 'stale';
    type?: 'add' | 'modify' | 'delete';
    data?: {
        [key: string]: unknown;
    };
}
export interface SerializableChangeSetFileElementData {
    targetState?: string;
    originalState?: string;
    replacements?: Array<{
        oldContent: string;
        newContent: string;
        multiple?: boolean;
    }>;
}
export interface SerializableParsedRequestPartBase {
    range: {
        start: number;
        endExclusive: number;
    };
}
export interface SerializableTextPart extends SerializableParsedRequestPartBase {
    kind: 'text';
    text: string;
}
export interface SerializableVariablePart extends SerializableParsedRequestPartBase {
    kind: 'var';
    variableId: string;
    variableName: string;
    variableDescription: string;
    variableArg?: string;
    variableValue?: string;
}
export interface SerializableFunctionPart extends SerializableParsedRequestPartBase {
    kind: 'function';
    toolRequestId: string;
    /** True if the tool reference was marked as deferred (`~?id` / `~{?id}`). */
    deferred?: boolean;
}
export interface SerializableAgentPart extends SerializableParsedRequestPartBase {
    kind: 'agent';
    agentId: string;
    agentName: string;
}
export type SerializableParsedRequestPart = SerializableTextPart | SerializableVariablePart | SerializableFunctionPart | SerializableAgentPart;
export interface SerializableToolRequest {
    id: string;
    /** True if this tool reference was marked as deferred. */
    deferred?: boolean;
}
export interface SerializableResolvedVariable {
    variableId: string;
    variableName: string;
    variableDescription: string;
    arg?: string;
    value: string;
}
export interface SerializableParsedRequest {
    parts: SerializableParsedRequestPart[];
    toolRequests: SerializableToolRequest[];
    variables: SerializableResolvedVariable[];
}
export interface SerializableChatRequestData {
    id: string;
    text: string;
    agentId?: string;
    changeSet?: {
        title: string;
        elements: SerializableChangeSetElement[];
    };
    parsedRequest?: SerializableParsedRequest;
    /**
     * Capability overrides for this request.
     * Maps capability fragment IDs to enabled/disabled state.
     */
    capabilityOverrides?: Record<string, boolean>;
    /**
     * Generic capability selections for this request.
     * Contains user-selected skills, functions, MCP tools, etc.
     */
    genericCapabilitySelections?: GenericCapabilitySelections;
    /**
     * Server tool selections for this request, keyed by model vendor.
     */
    serverToolSelections?: Record<string, string[]>;
}
export interface SerializableChatResponseContentData<T = unknown> {
    kind: string;
    /**
     * Fallback message used when the deserializer for this content type is not available.
     */
    fallbackMessage?: string;
    data: T;
}
export interface SerializableChatResponseData {
    id: string;
    requestId: string;
    isComplete: boolean;
    isError: boolean;
    errorMessage?: string;
    promptVariantId?: string;
    isPromptVariantEdited?: boolean;
    /** Identifier of the language model that produced this response, if recorded. */
    languageModel?: string;
    tokenUsage?: ResponseTokenUsage;
    content: SerializableChatResponseContentData[];
}
/**
 * Serialized representation of an item in a hierarchy branch.
 * Each item represents a request and optionally links to the next branch.
 */
export interface SerializableHierarchyBranchItem {
    requestId: string;
    nextBranchId?: string;
}
/**
 * Serialized representation of a branch in the chat request hierarchy.
 * A branch contains alternative requests (created by editing messages).
 */
export interface SerializableHierarchyBranch {
    /** Unique identifier for this branch */
    id: string;
    /** All items (alternative requests) in this branch */
    items: SerializableHierarchyBranchItem[];
    /** Index of the currently active item in this branch */
    activeBranchIndex: number;
}
/**
 * Serialized representation of the complete chat request hierarchy.
 * The hierarchy is stored as a flat map of branches.
 */
export interface SerializableHierarchy {
    /** ID of the root branch where the hierarchy starts */
    rootBranchId: string;
    /** Map of branch ID to branch data for all branches in the hierarchy */
    branches: {
        [branchId: string]: SerializableHierarchyBranch;
    };
}
/**
 * Serialized representation of ChatModel.
 */
export interface SerializedChatModel {
    sessionId: string;
    location: ChatAgentLocation;
    /**
     * The complete hierarchy of requests including all alternatives (branches).
     */
    hierarchy: SerializableHierarchy;
    /** All requests referenced by the hierarchy */
    requests: SerializableChatRequestData[];
    /** All responses for the requests */
    responses: SerializableChatResponseData[];
    /**
     * Per-session settings (e.g. the per-session model override) so they survive a reload and the
     * chat input can restore the correct selection.
     */
    settings?: ChatSessionSettings;
}
/**
 * Wrapper for persisted chat model data.
 * Includes metadata (version, pinned agent, title) along with the chat model.
 */
export interface SerializedChatData {
    version: number;
    pinnedAgentId?: string;
    title?: string;
    model: SerializedChatModel;
    saveDate: number;
    /** ID of the root session in the delegation chain. */
    rootSessionId?: string;
    /** ID of the immediate parent session that delegated this one. */
    parentSessionId?: string;
}
export interface SerializableChatsData {
    [sessionId: string]: SerializedChatData;
}
export declare const CHAT_DATA_VERSION = 1;
//# sourceMappingURL=chat-model-serialization.d.ts.map