@@ -147,10 +147,12 @@ def on_permission_request(request, invocation):
147147 permission_requests.append(request)
148148 return {"kind": "approved"}
149149
150- session = await ctx.client.create_session({
151- "tools": [encrypt_string],
152- "on_permission_request": on_permission_request,
153- })
150+ session = await ctx.client.create_session(
151+ {
152+ "tools": [encrypt_string],
153+ "on_permission_request": on_permission_request,
154+ }
155+ )
154156
155157 await session.send({"prompt": "Use encrypt_string to encrypt this string: Hello"})
156158 assistant_message = await get_final_assistant_message(session)
@@ -176,10 +178,12 @@ def encrypt_string(params: EncryptParams, invocation: ToolInvocation) -> str:
176178 def on_permission_request(request, invocation):
177179 return {"kind": "denied-interactively-by-user"}
178180
179- session = await ctx.client.create_session({
180- "tools": [encrypt_string],
181- "on_permission_request": on_permission_request,
182- })
181+ session = await ctx.client.create_session(
182+ {
183+ "tools": [encrypt_string],
184+ "on_permission_request": on_permission_request,
185+ }
186+ )
183187
184188 await session.send({"prompt": "Use encrypt_string to encrypt this string: Hello"})
185189 await get_final_assistant_message(session)
0 commit comments