88from ..interfaces import Agent , Client
99from ..meta import AGENT_METHODS
1010from ..schema import (
11+ AcpMcpServer ,
1112 AudioContentBlock ,
1213 AuthenticateRequest ,
1314 AuthenticateResponse ,
@@ -101,7 +102,7 @@ async def new_session(
101102 self ,
102103 cwd : str ,
103104 additional_directories : list [str ] | None = None ,
104- mcp_servers : list [HttpMcpServer | SseMcpServer | McpServerStdio ] | None = None ,
105+ mcp_servers : list [HttpMcpServer | SseMcpServer | AcpMcpServer | McpServerStdio ] | None = None ,
105106 ** kwargs : Any ,
106107 ) -> NewSessionResponse :
107108 resolved_mcp_servers = mcp_servers or []
@@ -123,7 +124,7 @@ async def load_session(
123124 cwd : str ,
124125 session_id : str ,
125126 additional_directories : list [str ] | None = None ,
126- mcp_servers : list [HttpMcpServer | SseMcpServer | McpServerStdio ] | None = None ,
127+ mcp_servers : list [HttpMcpServer | SseMcpServer | AcpMcpServer | McpServerStdio ] | None = None ,
127128 ** kwargs : Any ,
128129 ) -> LoadSessionResponse :
129130 resolved_mcp_servers = mcp_servers or []
@@ -142,18 +143,12 @@ async def load_session(
142143
143144 @param_model (ListSessionsRequest )
144145 async def list_sessions (
145- self ,
146- additional_directories : list [str ] | None = None ,
147- cursor : str | None = None ,
148- cwd : str | None = None ,
149- ** kwargs : Any ,
146+ self , cursor : str | None = None , cwd : str | None = None , ** kwargs : Any
150147 ) -> ListSessionsResponse :
151148 return await request_model_from_dict (
152149 self ._conn ,
153150 AGENT_METHODS ["session_list" ],
154- ListSessionsRequest (
155- additional_directories = additional_directories , cursor = cursor , cwd = cwd , field_meta = kwargs or None
156- ),
151+ ListSessionsRequest (cursor = cursor , cwd = cwd , field_meta = kwargs or None ),
157152 ListSessionsResponse ,
158153 )
159154
@@ -228,7 +223,7 @@ async def fork_session(
228223 cwd : str ,
229224 session_id : str ,
230225 additional_directories : list [str ] | None = None ,
231- mcp_servers : list [HttpMcpServer | SseMcpServer | McpServerStdio ] | None = None ,
226+ mcp_servers : list [HttpMcpServer | SseMcpServer | AcpMcpServer | McpServerStdio ] | None = None ,
232227 ** kwargs : Any ,
233228 ) -> ForkSessionResponse :
234229 return await request_model (
@@ -250,7 +245,7 @@ async def resume_session(
250245 cwd : str ,
251246 session_id : str ,
252247 additional_directories : list [str ] | None = None ,
253- mcp_servers : list [HttpMcpServer | SseMcpServer | McpServerStdio ] | None = None ,
248+ mcp_servers : list [HttpMcpServer | SseMcpServer | AcpMcpServer | McpServerStdio ] | None = None ,
254249 ** kwargs : Any ,
255250 ) -> ResumeSessionResponse :
256251 return await request_model (
0 commit comments