@@ -67,8 +67,11 @@ public final class CreateSessionRequest {
6767 @ JsonProperty ("streaming" )
6868 private Boolean streaming ;
6969
70+ @ JsonProperty ("includeSubAgentStreamingEvents" )
71+ private Boolean includeSubAgentStreamingEvents ;
72+
7073 @ JsonProperty ("mcpServers" )
71- private Map <String , Object > mcpServers ;
74+ private Map <String , McpServerConfig > mcpServers ;
7275
7376 @ JsonProperty ("envValueMode" )
7477 private String envValueMode ;
@@ -91,12 +94,18 @@ public final class CreateSessionRequest {
9194 @ JsonProperty ("configDir" )
9295 private String configDir ;
9396
97+ @ JsonProperty ("enableConfigDiscovery" )
98+ private Boolean enableConfigDiscovery ;
99+
94100 @ JsonProperty ("commands" )
95101 private List <CommandWireDefinition > commands ;
96102
97103 @ JsonProperty ("requestElicitation" )
98104 private Boolean requestElicitation ;
99105
106+ @ JsonProperty ("modelCapabilities" )
107+ private ModelCapabilitiesOverride modelCapabilities ;
108+
100109 /** Gets the model name. @return the model */
101110 public String getModel () {
102111 return model ;
@@ -240,12 +249,12 @@ public void setStreaming(Boolean streaming) {
240249 }
241250
242251 /** Gets MCP servers. @return the servers map */
243- public Map <String , Object > getMcpServers () {
252+ public Map <String , McpServerConfig > getMcpServers () {
244253 return mcpServers == null ? null : Collections .unmodifiableMap (mcpServers );
245254 }
246255
247256 /** Sets MCP servers. @param mcpServers the servers map */
248- public void setMcpServers (Map <String , Object > mcpServers ) {
257+ public void setMcpServers (Map <String , McpServerConfig > mcpServers ) {
249258 this .mcpServers = mcpServers ;
250259 }
251260
@@ -319,6 +328,29 @@ public void setConfigDir(String configDir) {
319328 this .configDir = configDir ;
320329 }
321330
331+ /** Gets enable config discovery flag. @return the flag */
332+ public Boolean getEnableConfigDiscovery () {
333+ return enableConfigDiscovery ;
334+ }
335+
336+ /** Sets enable config discovery flag. @param enableConfigDiscovery the flag */
337+ public void setEnableConfigDiscovery (Boolean enableConfigDiscovery ) {
338+ this .enableConfigDiscovery = enableConfigDiscovery ;
339+ }
340+
341+ /** Gets include sub-agent streaming events flag. @return the flag */
342+ public Boolean getIncludeSubAgentStreamingEvents () {
343+ return includeSubAgentStreamingEvents ;
344+ }
345+
346+ /**
347+ * Sets include sub-agent streaming events flag. @param
348+ * includeSubAgentStreamingEvents the flag
349+ */
350+ public void setIncludeSubAgentStreamingEvents (Boolean includeSubAgentStreamingEvents ) {
351+ this .includeSubAgentStreamingEvents = includeSubAgentStreamingEvents ;
352+ }
353+
322354 /** Gets the commands wire definitions. @return the commands */
323355 public List <CommandWireDefinition > getCommands () {
324356 return commands == null ? null : Collections .unmodifiableList (commands );
@@ -338,4 +370,16 @@ public Boolean getRequestElicitation() {
338370 public void setRequestElicitation (Boolean requestElicitation ) {
339371 this .requestElicitation = requestElicitation ;
340372 }
373+
374+ /** Gets the model capabilities override. @return the override */
375+ public ModelCapabilitiesOverride getModelCapabilities () {
376+ return modelCapabilities ;
377+ }
378+
379+ /**
380+ * Sets the model capabilities override. @param modelCapabilities the override
381+ */
382+ public void setModelCapabilities (ModelCapabilitiesOverride modelCapabilities ) {
383+ this .modelCapabilities = modelCapabilities ;
384+ }
341385}
0 commit comments