Skip to content

Commit a9155e9

Browse files
committed
fix: upgrade to the latest on_initialize method signature
1 parent 208a950 commit a9155e9

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

mcp_proxy_for_aws/middleware/initialize_middleware.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -34,8 +34,8 @@ def __init__(self, client_factory: AWSMCPProxyClientFactory) -> None:
3434
async def on_initialize(
3535
self,
3636
context: MiddlewareContext[mt.InitializeRequest],
37-
call_next: CallNext[mt.InitializeRequest, None],
38-
) -> None:
37+
call_next: CallNext[mt.InitializeRequest, mt.InitializeResult | None],
38+
) -> mt.InitializeResult | None:
3939
try:
4040
logger.debug('Received initialize request %s.', context.message)
4141
self._client_factory.set_init_params(context.message)

0 commit comments

Comments
 (0)