Add unstable model selection to TS SDK

Signed-off-by: Ben Brandt <benjamin.j.brandt@gmail.com>

Author: benbrandt

rust/agent.rs

@@ -820,7 +820,7 @@ pub struct AgentMethodNames {
     pub session_cancel: &'static str,
     /// Method for selecting a model for a given session.
     #[cfg(feature = "unstable")]
-    pub model_select: &'static str,
+    pub session_set_model: &'static str,
 }
 
 /// Constant containing all agent method names.
@@ -833,7 +833,7 @@ pub const AGENT_METHOD_NAMES: AgentMethodNames = AgentMethodNames {
     session_prompt: SESSION_PROMPT_METHOD_NAME,
     session_cancel: SESSION_CANCEL_METHOD_NAME,
     #[cfg(feature = "unstable")]
-    model_select: SESSION_SET_MODEL_METHOD_NAME,
+    session_set_model: SESSION_SET_MODEL_METHOD_NAME,
 };
 
 /// Method name for the initialize request.

schema/meta.json

@@ -2,12 +2,12 @@
   "agentMethods": {
     "authenticate": "authenticate",
     "initialize": "initialize",
-    "model_select": "session/set_model",
     "session_cancel": "session/cancel",
     "session_load": "session/load",
     "session_new": "session/new",
     "session_prompt": "session/prompt",
-    "session_set_mode": "session/set_mode"
+    "session_set_mode": "session/set_mode",
+    "session_set_model": "session/set_model"
   },
   "clientMethods": {
     "fs_read_text_file": "fs/read_text_file",

typescript/acp.ts

@@ -79,7 +79,15 @@ export class AgentSideConnection {
         }
         case schema.AGENT_METHODS.session_prompt: {
           const validatedParams = schema.promptRequestSchema.parse(params);
-          return agent.prompt(validatedParams as schema.PromptRequest);
+          return agent.prompt(validatedParams);
+        }
+        case schema.AGENT_METHODS.session_set_model: {
+          if (!agent.setSessionModel) {
+            throw RequestError.methodNotFound(method);
+          }
+          const validatedParams =
+            schema.setSessionModelRequestSchema.parse(params);
+          return agent.setSessionModel(validatedParams);
         }
         default:
           if (method.startsWith("_")) {
@@ -1191,6 +1199,16 @@ export interface Agent {
   setSessionMode?(
     params: schema.SetSessionModeRequest,
   ): Promise<schema.SetSessionModeResponse | void>;
+  /**
+   * **UNSTABLE**
+   *
+   * This capability is not part of the spec yet, and may be removed or changed at any point.
+   *
+   * Select a model for a given session.
+   */
+  setSessionModel?(
+    params: schema.SetSessionModelRequest,
+  ): Promise<schema.SetSessionModelResponse | void>;
   /**
    * Authenticates the client using the specified authentication method.
    *

typescript/schema.ts

@@ -1,12 +1,12 @@
 export const AGENT_METHODS = {
   authenticate: "authenticate",
   initialize: "initialize",
-  model_select: "session/set_model",
   session_cancel: "session/cancel",
   session_load: "session/load",
   session_new: "session/new",
   session_prompt: "session/prompt",
   session_set_mode: "session/set_mode",
+  session_set_model: "session/set_model",
 } as const;
 
 export const CLIENT_METHODS = {