import type { OldShortcut, Shortcut } from '../../types';
import type { MessageContentType, MessageRole, MessageStatus } from './constants';
import type { ContentMap, InputContent } from './contents';
import type { InterruptMessage } from './interrupt';
export interface ActivityMessage extends BaseMessage<MessageRole.Activity, ContentMap[MessageContentType.FlowAgent] | ContentMap[MessageContentType.KnowledgeRag] | ContentMap[MessageContentType.ReferenceDocument]> {
    activityType: MessageContentType.FlowAgent | MessageContentType.KnowledgeRag | MessageContentType.ReferenceDocument | string;
}
export interface AssistantMessage extends BaseMessage<MessageRole.Assistant> {
    toolCalls?: ToolCall[];
}
declare global {
    interface AIBluekingMessageMap {
    }
}
export interface BaseMessage<T extends MessageType, C = string> {
    content: C;
    id: number | string;
    messageId: number | string;
    name?: string;
    role: T;
    status: MessageStatus;
    uid?: string;
    property?: {
        extra?: {
            cite: {
                data: {
                    key: string;
                    value: string;
                }[];
                title: string;
                type: 'structured';
            } | string;
            command: string;
            context: Partial<{
                [key: string]: string;
                __key: string;
                __label: string;
                __value: string;
                context_type: 'checkbox' | 'input' | 'number' | 'radioGroup' | 'select' | 'switcher' | 'text' | 'textarea';
            } & Partial<OldShortcut>>[];
            pause?: boolean;
            shortcut?: Partial<Shortcut>;
        };
    };
}
export interface Context {
    description: string;
    value: string;
}
export type DeveloperMessage = BaseMessage<MessageRole.Developer, string>;
export type FunctionCall = {
    arguments: string;
    description?: string;
    mcpName?: string;
    name: string;
};
export type GuideMessage = BaseMessage<MessageRole.Guide, string>;
export type HiddenAssistantMessage = BaseMessage<MessageRole.HiddenAssistant, string>;
export type HiddenGuideMessage = BaseMessage<MessageRole.HiddenGuide, string>;
export type HiddenMessage = BaseMessage<MessageRole.Hidden, string>;
export type HiddenSystemMessage = BaseMessage<MessageRole.HiddenSystem, string>;
export type HiddenUserMessage = BaseMessage<MessageRole.HiddenUser, string>;
export type InfoMessage = BaseMessage<MessageRole.Info, string>;
export type LoadingMessage = BaseMessage<MessageRole.Loading, string>;
export type Message = MessageMap[MessageType];
export type MessageMap = AIBluekingMessageMap & {
    [MessageRole.Activity]: ActivityMessage;
    [MessageRole.Assistant]: AssistantMessage;
    [MessageRole.Developer]: DeveloperMessage;
    [MessageRole.Guide]: GuideMessage;
    [MessageRole.Hidden]: HiddenMessage;
    [MessageRole.HiddenAssistant]: HiddenAssistantMessage;
    [MessageRole.HiddenGuide]: HiddenGuideMessage;
    [MessageRole.HiddenSystem]: HiddenSystemMessage;
    [MessageRole.HiddenUser]: HiddenUserMessage;
    [MessageRole.Info]: InfoMessage;
    [MessageRole.Interrupt]: InterruptMessage;
    [MessageRole.Loading]: LoadingMessage;
    [MessageRole.Pause]: PauseMessage;
    [MessageRole.Placeholder]: PlaceholderMessage;
    [MessageRole.Reasoning]: ReasoningMessage;
    [MessageRole.System]: SystemMessage;
    [MessageRole.TemplateAssistant]: TemplateAssistantMessage;
    [MessageRole.TemplateGuide]: TemplateGuideMessage;
    [MessageRole.TemplateHidden]: TemplateHiddenMessage;
    [MessageRole.TemplateSystem]: TemplateSystemMessage;
    [MessageRole.TemplateUser]: TemplateUserMessage;
    [MessageRole.Tool]: ToolMessage;
    [MessageRole.User]: UserMessage;
};
export type MessageType = keyof MessageMap;
export type PauseMessage = BaseMessage<MessageRole.Pause, string>;
export type PlaceholderMessage = BaseMessage<MessageRole.Placeholder, string>;
export interface ReasoningMessage extends BaseMessage<MessageRole.Reasoning, string[]> {
    duration?: number;
}
export type SystemMessage = BaseMessage<MessageRole.System, string>;
export type TemplateAssistantMessage = BaseMessage<MessageRole.TemplateAssistant, string>;
export type TemplateGuideMessage = BaseMessage<MessageRole.TemplateGuide, string>;
export type TemplateHiddenMessage = BaseMessage<MessageRole.TemplateHidden, string>;
export type TemplateSystemMessage = BaseMessage<MessageRole.TemplateSystem, string>;
export type TemplateUserMessage = BaseMessage<MessageRole.TemplateUser, string>;
export type Tool = {
    description: string;
    name: string;
    parameters: unknown;
};
export type ToolCall = {
    function: FunctionCall;
    id: string;
    toolMessage?: Partial<ToolMessage>;
    type: MessageContentType.Function;
};
export interface ToolMessage extends BaseMessage<MessageRole.Tool, string> {
    duration: number;
    error?: boolean | string;
    toolCallId: string;
}
export type UserMessage = BaseMessage<MessageRole.User, InputContent[] | string>;