feat: add listAvailableModels request method (#593)

Author: ctlai95

runtimes/README.md

@@ -196,6 +196,7 @@ The runtime supports chat by default
 | Send conversation or conversation action click event. If no action is provided, the default action is "open". | `aws/chat/conversationClick`          | `ConversationClickParams`          | [Request](https://microsoft.github.io/language-server-protocol/specifications/lsp/3.17/specification/#requestMessage) Client to Server | `ConversationClickResult`             |
 | Send request to list the MCP servers available: all or based on filter if provided. Similar to conversations, the filter in the request is a map of filter option ids to corresponding values. | `aws/chat/listMcpServers`          | `ListMcpServersParams`          | [Request](https://microsoft.github.io/language-server-protocol/specifications/lsp/3.17/specification/#requestMessage) Client to Server | `ListMcpServersResult`             |
 | Send MCP server or MCP server action click event. If no action is provided, the default action is "select". | `aws/chat/mcpServerClick`          | `McpServerClickParams`          | [Request](https://microsoft.github.io/language-server-protocol/specifications/lsp/3.17/specification/#requestMessage) Client to Server | `McpServerClickResult`             |
+| Send request to list available models for the current chat tab. | `aws/chat/listAvailableModels`     | `ListAvailableModelsParams`     | [Request](https://microsoft.github.io/language-server-protocol/specifications/lsp/3.17/specification/#requestMessage) Client to Server | `ListAvailableModelsResult`        |
 | Send pinned context to client. Pinned context contains context commands that are pinned to the current tab. | `aws/chat/sendPinnedContext`          | `PinnedContextParams`          | [Notification](https://microsoft.github.io/language-server-protocol/specifications/lsp/3.17/specification/#notificationMessage) Server to Client | n/a             |
 | Notify server that pinned context has been added by the client. | `aws/chat/pinnedContextAdd`          | `PinnedContextParams`          | [Notification](https://microsoft.github.io/language-server-protocol/specifications/lsp/3.17/specification/#notificationMessage) Client to Server | n/a             |
 | Notify server that pinned context has been removed by the client. | `aws/chat/pinnedContextRemove`          | `PinnedContextParams`          | [Notification](https://microsoft.github.io/language-server-protocol/specifications/lsp/3.17/specification/#notificationMessage) Client to Server | n/a             |

runtimes/protocol/chat.ts

@@ -88,6 +88,9 @@ import {
     OPEN_FILE_DIALOG_METHOD,
     OpenFileDialogParams,
     OpenFileDialogResult,
+    LIST_AVAILABLE_MODELS_REQUEST_METHOD,
+    ListAvailableModelsResult,
+    ListAvailableModelsParams,
 } from './lsp'
 
 export const chatRequestType = new AutoParameterStructuresProtocolRequestType<
@@ -268,3 +271,11 @@ export const promptInputOptionChangeNotificationType = new ProtocolNotificationT
     PromptInputOptionChangeParams,
     void
 >(PROMPT_INPUT_OPTION_CHANGE_METHOD)
+
+export const listAvailableModelsRequestType = new ProtocolRequestType<
+    ListAvailableModelsParams,
+    ListAvailableModelsResult,
+    never,
+    void,
+    void
+>(LIST_AVAILABLE_MODELS_REQUEST_METHOD)

runtimes/runtimes/base-runtime.ts

@@ -59,6 +59,7 @@ import {
     onPinnedContextAddNotificationType,
     onPinnedContextRemoveNotificationType,
     openFileDialogRequestType,
+    listAvailableModelsRequestType,
 } from '../protocol'
 import { createConnection } from 'vscode-languageserver/browser'
 import {
@@ -194,6 +195,7 @@ export const baseRuntime = (connections: { reader: MessageReader; writer: Messag
             lspConnection.onNotification(promptInputOptionChangeNotificationType.method, handler),
         onOpenFileDialog: handler => lspConnection.onRequest(openFileDialogRequestType.method, handler),
         onRuleClick: handler => lspConnection.onRequest(ruleClickRequestType.method, handler),
+        onListAvailableModels: handler => lspConnection.onRequest(listAvailableModelsRequestType.method, handler),
     }
 
     const identityManagement: IdentityManagement = {

runtimes/runtimes/chat/baseChat.ts

@@ -85,6 +85,9 @@ import {
     OpenFileDialogParams,
     OpenFileDialogResult,
     openFileDialogRequestType,
+    ListAvailableModelsParams,
+    ListAvailableModelsResult,
+    listAvailableModelsRequestType,
 } from '../../protocol'
 import { Chat } from '../../server-interface'
 
@@ -236,4 +239,8 @@ export class BaseChat implements Chat {
     public onRuleClick(handler: RequestHandler<RuleClickParams, RuleClickResult, void>) {
         this.connection.onRequest(ruleClickRequestType.method, handler)
     }
+
+    public onListAvailableModels(handler: RequestHandler<ListAvailableModelsParams, ListAvailableModelsResult, void>) {
+        this.connection.onRequest(listAvailableModelsRequestType.method, handler)
+    }
 }

runtimes/server-interface/chat.ts

@@ -49,6 +49,8 @@ import {
     RuleClickResult,
     PinnedContextParams,
     ActiveEditorChangedParams,
+    ListAvailableModelsParams,
+    ListAvailableModelsResult,
 } from '../protocol'
 
 /**
@@ -75,6 +77,7 @@ export type Chat = {
     onOpenFileDialog: (
         handler: RequestHandler<OpenFileDialogParams, OpenFileDialogResult, OpenFileDialogResult>
     ) => void
+    onListAvailableModels: (handler: RequestHandler<ListAvailableModelsParams, ListAvailableModelsResult, void>) => void
     // Notifications
     onSendFeedback: (handler: NotificationHandler<FeedbackParams>) => void
     onReady: (handler: NotificationHandler<void>) => void

types/chat.ts

@@ -48,6 +48,9 @@ export const MCP_SERVER_CLICK_REQUEST_METHOD = 'aws/chat/mcpServerClick'
 // export
 export const GET_SERIALIZED_CHAT_REQUEST_METHOD = 'aws/chat/getSerializedChat'
 
+// model selection
+export const LIST_AVAILABLE_MODELS_REQUEST_METHOD = 'aws/chat/listAvailableModels'
+
 // button ids
 export const OPEN_WORKSPACE_INDEX_SETTINGS_BUTTON_ID = 'open-settings-for-ws-index'
 
@@ -671,3 +674,18 @@ export interface PromptInputOptionChangeParams {
     optionsValues: Record<string, string>
     eventId?: string
 }
+
+export interface Model {
+    id: string
+    name: string
+}
+
+export interface ListAvailableModelsParams {
+    tabId: string
+}
+
+export interface ListAvailableModelsResult {
+    tabId: string
+    models: Model[]
+    selectedModelId?: string
+}