Skip to content

Commit c4d1382

Browse files
Formatting
1 parent b57d0d5 commit c4d1382

File tree

1 file changed

+12
-8
lines changed

1 file changed

+12
-8
lines changed

python/e2e/test_tools.py

Lines changed: 12 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)