Commit 6597529
authored
fix: deserialize permission_suggestions into PermissionUpdate instances (#920)
## Problem
`ToolPermissionContext.suggestions` is typed as
`list[PermissionUpdate]`, but `_internal/query.py` assigns the raw
`permission_suggestions` list from the control protocol without
deserializing, so at runtime it's `list[dict]`.
This breaks the round-trip pattern where a `can_use_tool` callback
echoes a suggestion back to persist it:
```python
async def can_use_tool(tool_name, input_data, context):
persist = [s for s in context.suggestions if s.destination == "localSettings"]
# AttributeError: 'dict' object has no attribute 'destination'
return PermissionResultAllow(updated_permissions=persist)
# and even with dict access, the response path calls .to_dict() on each item:
# AttributeError: 'dict' object has no attribute 'to_dict'
```
The equivalent works in the TypeScript SDK.
## Fix
- Add `PermissionUpdate.from_dict()` as the inverse of `to_dict()`,
rebuilding `PermissionRuleValue` entries from the camelCase wire keys.
- Map `permission_suggestions` through it when constructing
`ToolPermissionContext`.
Suggestions now arrive as dataclass instances and can be returned
directly in `PermissionResultAllow(updated_permissions=...)`.
## Tests
- `TestPermissionUpdate` round-trip unit tests for `addRules`,
`setMode`, `addDirectories`.
- `test_permission_callback_suggestions_roundtrip` sends a wire-format
suggestion, asserts the callback receives a `PermissionUpdate`, echoes
it back, and checks the serialized response matches the input.
- Fixed a fixture in `test_permission_callback_deny` that passed
`["deny"]` as a suggestion; it only worked before because nothing parsed
the list.
Related: #624 tightens the type annotation on the control-protocol
TypedDict but doesn't change runtime behavior; this PR makes the runtime
match the annotation.1 parent bdb3291 commit 6597529
4 files changed
Lines changed: 124 additions & 3 deletions
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
19 | 19 | | |
20 | 20 | | |
21 | 21 | | |
| 22 | + | |
22 | 23 | | |
23 | 24 | | |
24 | 25 | | |
| |||
350 | 351 | | |
351 | 352 | | |
352 | 353 | | |
353 | | - | |
354 | | - | |
| 354 | + | |
| 355 | + | |
| 356 | + | |
| 357 | + | |
| 358 | + | |
| 359 | + | |
355 | 360 | | |
356 | 361 | | |
357 | 362 | | |
| |||
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
169 | 169 | | |
170 | 170 | | |
171 | 171 | | |
| 172 | + | |
| 173 | + | |
| 174 | + | |
| 175 | + | |
| 176 | + | |
| 177 | + | |
| 178 | + | |
| 179 | + | |
| 180 | + | |
| 181 | + | |
| 182 | + | |
| 183 | + | |
| 184 | + | |
| 185 | + | |
| 186 | + | |
| 187 | + | |
| 188 | + | |
| 189 | + | |
| 190 | + | |
| 191 | + | |
| 192 | + | |
172 | 193 | | |
173 | 194 | | |
174 | 195 | | |
| |||
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
97 | 97 | | |
98 | 98 | | |
99 | 99 | | |
| 100 | + | |
| 101 | + | |
| 102 | + | |
| 103 | + | |
| 104 | + | |
| 105 | + | |
| 106 | + | |
| 107 | + | |
| 108 | + | |
| 109 | + | |
| 110 | + | |
| 111 | + | |
| 112 | + | |
| 113 | + | |
| 114 | + | |
| 115 | + | |
| 116 | + | |
| 117 | + | |
| 118 | + | |
| 119 | + | |
| 120 | + | |
| 121 | + | |
| 122 | + | |
| 123 | + | |
| 124 | + | |
| 125 | + | |
| 126 | + | |
| 127 | + | |
| 128 | + | |
| 129 | + | |
| 130 | + | |
| 131 | + | |
| 132 | + | |
| 133 | + | |
| 134 | + | |
| 135 | + | |
| 136 | + | |
| 137 | + | |
| 138 | + | |
| 139 | + | |
| 140 | + | |
| 141 | + | |
| 142 | + | |
| 143 | + | |
| 144 | + | |
| 145 | + | |
| 146 | + | |
| 147 | + | |
| 148 | + | |
| 149 | + | |
| 150 | + | |
| 151 | + | |
100 | 152 | | |
101 | 153 | | |
102 | 154 | | |
| |||
121 | 173 | | |
122 | 174 | | |
123 | 175 | | |
124 | | - | |
| 176 | + | |
125 | 177 | | |
126 | 178 | | |
127 | 179 | | |
| |||
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
12 | 12 | | |
13 | 13 | | |
14 | 14 | | |
| 15 | + | |
| 16 | + | |
15 | 17 | | |
16 | 18 | | |
17 | 19 | | |
| |||
22 | 24 | | |
23 | 25 | | |
24 | 26 | | |
| 27 | + | |
| 28 | + | |
| 29 | + | |
| 30 | + | |
| 31 | + | |
| 32 | + | |
| 33 | + | |
| 34 | + | |
| 35 | + | |
| 36 | + | |
| 37 | + | |
| 38 | + | |
| 39 | + | |
| 40 | + | |
| 41 | + | |
| 42 | + | |
| 43 | + | |
| 44 | + | |
| 45 | + | |
| 46 | + | |
| 47 | + | |
| 48 | + | |
| 49 | + | |
| 50 | + | |
| 51 | + | |
| 52 | + | |
| 53 | + | |
| 54 | + | |
| 55 | + | |
| 56 | + | |
| 57 | + | |
| 58 | + | |
| 59 | + | |
| 60 | + | |
| 61 | + | |
| 62 | + | |
| 63 | + | |
| 64 | + | |
| 65 | + | |
| 66 | + | |
| 67 | + | |
25 | 68 | | |
26 | 69 | | |
27 | 70 | | |
| |||
0 commit comments