Skip to content

Commit 994b2aa

Browse files
Formatting
1 parent bc990b4 commit 994b2aa

4 files changed

Lines changed: 21 additions & 24 deletions

File tree

go/internal/e2e/permissions_test.go

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -161,7 +161,7 @@ func TestPermissions(t *testing.T) {
161161
ctx.ConfigureForTest(t)
162162

163163
session, err := client.CreateSession(t.Context(), &copilot.SessionConfig{
164-
OnPermissionRequest: func(request copilot.PermissionRequest, invocation copilot.ToolInvocation) (copilot.PermissionRequestResult, error) {
164+
OnPermissionRequest: func(request copilot.PermissionRequest, invocation copilot.PermissionInvocation) (copilot.PermissionRequestResult, error) {
165165
return copilot.PermissionRequestResult{Kind: "denied-no-approval-rule-and-could-not-request-from-user"}, nil
166166
},
167167
})
@@ -211,7 +211,7 @@ func TestPermissions(t *testing.T) {
211211
}
212212

213213
session2, err := client.ResumeSession(t.Context(), sessionID, &copilot.ResumeSessionConfig{
214-
OnPermissionRequest: func(request copilot.PermissionRequest, invocation copilot.ToolInvocation) (copilot.PermissionRequestResult, error) {
214+
OnPermissionRequest: func(request copilot.PermissionRequest, invocation copilot.PermissionInvocation) (copilot.PermissionRequestResult, error) {
215215
return copilot.PermissionRequestResult{Kind: "denied-no-approval-rule-and-could-not-request-from-user"}, nil
216216
},
217217
})

nodejs/test/client.test.ts

Lines changed: 10 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,9 @@ describe("CopilotClient", () => {
1010
await client.start();
1111
onTestFinished(() => client.forceStop());
1212

13-
await expect((client as any).createSession({})).rejects.toThrow(/onPermissionRequest.*is required/);
13+
await expect((client as any).createSession({})).rejects.toThrow(
14+
/onPermissionRequest.*is required/
15+
);
1416
});
1517

1618
it("throws when resumeSession is called without onPermissionRequest", async () => {
@@ -19,9 +21,9 @@ describe("CopilotClient", () => {
1921
onTestFinished(() => client.forceStop());
2022

2123
const session = await client.createSession({ onPermissionRequest: approveAll });
22-
await expect(
23-
(client as any).resumeSession(session.sessionId, {})
24-
).rejects.toThrow(/onPermissionRequest.*is required/);
24+
await expect((client as any).resumeSession(session.sessionId, {})).rejects.toThrow(
25+
/onPermissionRequest.*is required/
26+
);
2527
});
2628

2729
it("returns a standardized failure result when a tool is not registered", async () => {
@@ -67,7 +69,10 @@ describe("CopilotClient", () => {
6769

6870
const session = await client.createSession({ onPermissionRequest: approveAll });
6971
const spy = vi.spyOn((client as any).connection!, "sendRequest");
70-
await client.resumeSession(session.sessionId, { clientName: "my-app", onPermissionRequest: approveAll });
72+
await client.resumeSession(session.sessionId, {
73+
clientName: "my-app",
74+
onPermissionRequest: approveAll,
75+
});
7176

7277
expect(spy).toHaveBeenCalledWith(
7378
"session.resume",

python/copilot/client.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -434,7 +434,8 @@ async def create_session(self, config: SessionConfig) -> CopilotSession:
434434
435435
Example:
436436
>>> # Basic session
437-
>>> session = await client.create_session({"on_permission_request": PermissionHandler.approve_all})
437+
>>> config = {"on_permission_request": PermissionHandler.approve_all}
438+
>>> session = await client.create_session(config)
438439
>>>
439440
>>> # Session with model and streaming
440441
>>> session = await client.create_session({
@@ -606,7 +607,8 @@ async def resume_session(self, session_id: str, config: ResumeSessionConfig) ->
606607
607608
Example:
608609
>>> # Resume a previous session
609-
>>> session = await client.resume_session("session-123", {"on_permission_request": PermissionHandler.approve_all})
610+
>>> config = {"on_permission_request": PermissionHandler.approve_all}
611+
>>> session = await client.resume_session("session-123", config)
610612
>>>
611613
>>> # Resume with new tools
612614
>>> session = await client.resume_session("session-123", {

python/e2e/test_permissions.py

Lines changed: 5 additions & 15 deletions
Original file line numberDiff line numberDiff line change
@@ -76,9 +76,7 @@ async def test_should_deny_tool_operations_when_handler_explicitly_denies(
7676
def deny_all(request, invocation):
7777
return {"kind": "denied-no-approval-rule-and-could-not-request-from-user"}
7878

79-
session = await ctx.client.create_session(
80-
{"on_permission_request": deny_all}
81-
)
79+
session = await ctx.client.create_session({"on_permission_request": deny_all})
8280

8381
denied_events = []
8482
done_event = asyncio.Event()
@@ -89,9 +87,7 @@ def on_event(event):
8987
msg = (
9088
error
9189
if isinstance(error, str)
92-
else (
93-
getattr(error, "message", None) if error is not None else None
94-
)
90+
else (getattr(error, "message", None) if error is not None else None)
9591
)
9692
if msg and "Permission denied" in msg:
9793
denied_events.append(event)
@@ -120,9 +116,7 @@ async def test_should_deny_tool_operations_when_handler_explicitly_denies_after_
120116
def deny_all(request, invocation):
121117
return {"kind": "denied-no-approval-rule-and-could-not-request-from-user"}
122118

123-
session2 = await ctx.client.resume_session(
124-
session_id, {"on_permission_request": deny_all}
125-
)
119+
session2 = await ctx.client.resume_session(session_id, {"on_permission_request": deny_all})
126120

127121
denied_events = []
128122
done_event = asyncio.Event()
@@ -133,9 +127,7 @@ def on_event(event):
133127
msg = (
134128
error
135129
if isinstance(error, str)
136-
else (
137-
getattr(error, "message", None) if error is not None else None
138-
)
130+
else (getattr(error, "message", None) if error is not None else None)
139131
)
140132
if msg and "Permission denied" in msg:
141133
denied_events.append(event)
@@ -151,9 +143,7 @@ def on_event(event):
151143

152144
await session2.destroy()
153145

154-
async def test_should_work_with_approve_all_permission_handler(
155-
self, ctx: E2ETestContext
156-
):
146+
async def test_should_work_with_approve_all_permission_handler(self, ctx: E2ETestContext):
157147
"""Test that sessions work with approve-all permission handler"""
158148
session = await ctx.client.create_session(
159149
{"on_permission_request": PermissionHandler.approve_all}

0 commit comments

Comments
 (0)