runtimes/protocol/chat.ts

import {
    ChatParams,
    ChatResult,
    EndChatParams,
    EndChatResult,
    FeedbackParams,
    FollowUpClickParams,
    InfoLinkClickParams,
    InsertToCursorPositionParams,
    LinkClickParams,
    QuickActionParams,
    SourceLinkClickParams,
    TabAddParams,
    TabChangeParams,
    TabRemoveParams,
    ProtocolNotificationType,
    ProtocolRequestType,
    CHAT_REQUEST_METHOD,
    END_CHAT_REQUEST_METHOD,
    FEEDBACK_NOTIFICATION_METHOD,
    FOLLOW_UP_CLICK_NOTIFICATION_METHOD,
    INFO_LINK_CLICK_NOTIFICATION_METHOD,
    INSERT_TO_CURSOR_POSITION_NOTIFICATION_METHOD,
    LINK_CLICK_NOTIFICATION_METHOD,
    QUICK_ACTION_REQUEST_METHOD,
    READY_NOTIFICATION_METHOD,
    SOURCE_LINK_CLICK_NOTIFICATION_METHOD,
    TAB_ADD_NOTIFICATION_METHOD,
    TAB_CHANGE_NOTIFICATION_METHOD,
    TAB_REMOVE_NOTIFICATION_METHOD,
    AutoParameterStructuresProtocolRequestType,
    EncryptedChatParams,
    EncryptedQuickActionParams,
    QuickActionResult,
    OPEN_TAB_REQUEST_METHOD,
    OpenTabParams,
    OpenTabResult,
    CHAT_UPDATE_NOTIFICATION_METHOD,
    FILE_CLICK_NOTIFICATION_METHOD,
    ChatUpdateParams,
    FileClickParams,
    INLINE_CHAT_REQUEST_METHOD,
    InlineChatParams,
    InlineChatResult,
    CONTEXT_COMMAND_NOTIFICATION_METHOD,
    ContextCommandParams,
    CREATE_PROMPT_NOTIFICATION_METHOD,
    CreatePromptParams,
    INLINE_CHAT_RESULT_NOTIFICATION_METHOD,
    InlineChatResultParams,
    ListConversationsParams,
    ListConversationsResult,
    LIST_CONVERSATIONS_REQUEST_METHOD,
    ConversationClickParams,
    ConversationClickResult,
    CONVERSATION_CLICK_REQUEST_METHOD,
    GetSerializedChatParams,
    GetSerializedChatResult,
    GET_SERIALIZED_CHAT_REQUEST_METHOD,
    TabBarActionParams,
    TabBarActionResult,
    TAB_BAR_ACTION_REQUEST_METHOD,
    ChatOptionsUpdateParams,
    CHAT_OPTIONS_UPDATE_NOTIFICATION_METHOD,
    PromptInputOptionChangeParams,
    PROMPT_INPUT_OPTION_CHANGE_METHOD,
    ButtonClickParams,
    ButtonClickResult,
    BUTTON_CLICK_REQUEST_METHOD,
} from './lsp'

export const chatRequestType = new AutoParameterStructuresProtocolRequestType<
    ChatParams | EncryptedChatParams,
    ChatResult | string,
    ChatResult | string,
    void,
    void
>(CHAT_REQUEST_METHOD)
export const inlineChatRequestType = new AutoParameterStructuresProtocolRequestType<
    InlineChatParams | EncryptedChatParams,
    InlineChatResult | string,
    InlineChatResult | string,
    void,
    void
>(INLINE_CHAT_REQUEST_METHOD)
export const endChatRequestType = new ProtocolRequestType<EndChatParams, EndChatResult, never, void, void>(
    END_CHAT_REQUEST_METHOD
)
export const quickActionRequestType = new AutoParameterStructuresProtocolRequestType<
    QuickActionParams | EncryptedQuickActionParams,
    QuickActionResult | string,
    QuickActionResult | string,
    void,
    void
>(QUICK_ACTION_REQUEST_METHOD)
export const readyNotificationType = new ProtocolNotificationType<void, void>(READY_NOTIFICATION_METHOD)
export const feedbackNotificationType = new ProtocolNotificationType<FeedbackParams, void>(FEEDBACK_NOTIFICATION_METHOD)
export const tabAddNotificationType = new ProtocolNotificationType<TabAddParams, void>(TAB_ADD_NOTIFICATION_METHOD)
export const tabChangeNotificationType = new ProtocolNotificationType<TabChangeParams, void>(
    TAB_CHANGE_NOTIFICATION_METHOD
)
export const tabRemoveNotificationType = new ProtocolNotificationType<TabRemoveParams, void>(
    TAB_REMOVE_NOTIFICATION_METHOD
)
export const insertToCursorPositionNotificationType = new ProtocolNotificationType<InsertToCursorPositionParams, void>(
    INSERT_TO_CURSOR_POSITION_NOTIFICATION_METHOD
)
export const linkClickNotificationType = new ProtocolNotificationType<LinkClickParams, void>(
    LINK_CLICK_NOTIFICATION_METHOD
)
export const infoLinkClickNotificationType = new ProtocolNotificationType<InfoLinkClickParams, void>(
    INFO_LINK_CLICK_NOTIFICATION_METHOD
)
export const sourceLinkClickNotificationType = new ProtocolNotificationType<SourceLinkClickParams, void>(
    SOURCE_LINK_CLICK_NOTIFICATION_METHOD
)
export const followUpClickNotificationType = new ProtocolNotificationType<FollowUpClickParams, void>(
    FOLLOW_UP_CLICK_NOTIFICATION_METHOD
)
export const openTabRequestType = new ProtocolRequestType<OpenTabParams, OpenTabResult, never, void, void>(
    OPEN_TAB_REQUEST_METHOD
)
export const buttonClickRequestType = new ProtocolRequestType<ButtonClickParams, ButtonClickResult, never, void, void>(
    BUTTON_CLICK_REQUEST_METHOD
)
export const chatOptionsUpdateType = new ProtocolNotificationType<ChatOptionsUpdateParams, void>(
    CHAT_OPTIONS_UPDATE_NOTIFICATION_METHOD
)
export const chatUpdateNotificationType = new ProtocolNotificationType<ChatUpdateParams, void>(
    CHAT_UPDATE_NOTIFICATION_METHOD
)
export const fileClickNotificationType = new ProtocolNotificationType<FileClickParams, void>(
    FILE_CLICK_NOTIFICATION_METHOD
)

// context
export const contextCommandsNotificationType = new ProtocolNotificationType<ContextCommandParams, void>(
    CONTEXT_COMMAND_NOTIFICATION_METHOD
)
export const createPromptNotificationType = new ProtocolNotificationType<CreatePromptParams, void>(
    CREATE_PROMPT_NOTIFICATION_METHOD
)
/**
 * The inline chat result notification is sent from client to server to notify the action taken by the user
 * from the suggested response returned by the server.
 */
export const inlineChatResultNotificationType = new ProtocolNotificationType<InlineChatResultParams, void>(
    INLINE_CHAT_RESULT_NOTIFICATION_METHOD
)

// history
export const listConversationsRequestType = new AutoParameterStructuresProtocolRequestType<
    ListConversationsParams,
    ListConversationsResult,
    never,
    void,
    void
>(LIST_CONVERSATIONS_REQUEST_METHOD)
export const conversationClickRequestType = new AutoParameterStructuresProtocolRequestType<
    ConversationClickParams,
    ConversationClickResult,
    never,
    void,
    void
>(CONVERSATION_CLICK_REQUEST_METHOD)

/**
 * The tab bar action request is sent from client to server to execute action from Chat tab bar UI.
 * Tab bar action may have TabId attached to indicate that action is performed on specific tab.
 *
 * See `TabBarActionParams` for supported actions.
 */
export const tabBarActionRequestType = new ProtocolRequestType<
    TabBarActionParams,
    TabBarActionResult,
    never,
    void,
    void
>(TAB_BAR_ACTION_REQUEST_METHOD)

/**
 * The get serialized chat request is sent from server to client to retrieve chat conversation messages serialized to specified format.
 */
export const getSerializedChatRequestType = new ProtocolRequestType<
    GetSerializedChatParams,
    GetSerializedChatResult,
    never,
    void,
    void
>(GET_SERIALIZED_CHAT_REQUEST_METHOD)

export const promptInputOptionChangeNotificationType = new ProtocolNotificationType<
    PromptInputOptionChangeParams,
    void
>(PROMPT_INPUT_OPTION_CHANGE_METHOD)