Skip to content

Commit 0d23841

Browse files
dicksontsairushilpatel0
authored andcommitted
Ruff
1 parent 4175e4a commit 0d23841

File tree

4 files changed

+18
-12
lines changed

4 files changed

+18
-12
lines changed

src/claude_code_sdk/__init__.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,5 @@
11
"""Claude SDK for Python."""
22

3-
43
from ._errors import (
54
ClaudeSDKError,
65
CLIConnectionError,

src/claude_code_sdk/_internal/client.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -40,9 +40,7 @@ async def process_query(
4040
chosen_transport = transport
4141
else:
4242
chosen_transport = SubprocessCLITransport(
43-
prompt=prompt,
44-
options=options,
45-
close_stdin_after_prompt=True
43+
prompt=prompt, options=options, close_stdin_after_prompt=True
4644
)
4745

4846
try:

src/claude_code_sdk/_internal/transport/subprocess_cli.py

Lines changed: 9 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -171,6 +171,7 @@ async def connect(self) -> None:
171171
self._stdin_stream = TextSendStream(self._process.stdin)
172172
# Start streaming messages to stdin in background
173173
import asyncio
174+
174175
asyncio.create_task(self._stream_to_stdin())
175176
else:
176177
# String mode: close stdin immediately (backward compatible)
@@ -224,7 +225,7 @@ async def send_request(self, messages: list[Any], options: dict[str, Any]) -> No
224225
"type": "user",
225226
"message": {"role": "user", "content": str(message)},
226227
"parent_tool_use_id": None,
227-
"session_id": options.get("session_id", "default")
228+
"session_id": options.get("session_id", "default"),
228229
}
229230

230231
await self._stdin_stream.send(json.dumps(message) + "\n")
@@ -372,7 +373,9 @@ def is_connected(self) -> bool:
372373
async def interrupt(self) -> None:
373374
"""Send interrupt control request (only works in streaming mode)."""
374375
if not self._is_streaming:
375-
raise CLIConnectionError("Interrupt requires streaming mode (AsyncIterable prompt)")
376+
raise CLIConnectionError(
377+
"Interrupt requires streaming mode (AsyncIterable prompt)"
378+
)
376379

377380
if not self._stdin_stream:
378381
raise CLIConnectionError("Not connected or stdin not available")
@@ -392,7 +395,7 @@ async def _send_control_request(self, request: dict[str, Any]) -> dict[str, Any]
392395
control_request = {
393396
"type": "control_request",
394397
"request_id": request_id,
395-
"request": request
398+
"request": request,
396399
}
397400

398401
# Send request
@@ -407,7 +410,9 @@ async def _send_control_request(self, request: dict[str, Any]) -> dict[str, Any]
407410
response = self._pending_control_responses.pop(request_id)
408411

409412
if response.get("subtype") == "error":
410-
raise CLIConnectionError(f"Control request failed: {response.get('error')}")
413+
raise CLIConnectionError(
414+
f"Control request failed: {response.get('error')}"
415+
)
411416

412417
return response
413418
except TimeoutError:

tests/test_streaming_client.py

Lines changed: 8 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -389,7 +389,8 @@ def _create_test_script(
389389
if response is None:
390390
response = '{"type": "result", "subtype": "success", "duration_ms": 100, "duration_api_ms": 50, "is_error": false, "num_turns": 1, "session_id": "test", "total_cost_usd": 0.001}'
391391

392-
script_content = textwrap.dedent("""
392+
script_content = textwrap.dedent(
393+
"""
393394
#!/usr/bin/env python3
394395
import sys
395396
import json
@@ -416,7 +417,8 @@ def _create_test_script(
416417
)
417418

418419
if expected_messages is not None:
419-
script_content += textwrap.dedent(f"""
420+
script_content += textwrap.dedent(
421+
f"""
420422
# Verify we got the expected messages
421423
assert len(stdin_messages) == {len(expected_messages)}
422424
""",
@@ -425,12 +427,14 @@ def _create_test_script(
425427
script_content += f'''assert '"{msg}"' in stdin_messages[{i}]\n'''
426428

427429
if should_error:
428-
script_content += textwrap.dedent("""
430+
script_content += textwrap.dedent(
431+
"""
429432
sys.exit(1)
430433
""",
431434
)
432435
else:
433-
script_content += textwrap.dedent(f"""
436+
script_content += textwrap.dedent(
437+
f"""
434438
# Output response
435439
print('{response}')
436440
""",

0 commit comments

Comments
 (0)