Fix lint and test

Signed-off-by: Rushil Patel <[email protected]>

Author: dicksontsai

CLAUDE.md

@@ -3,8 +3,8 @@
 ```bash
 # Lint and style
 # Check for issues and fix automatically
-python -m ruff check src/ test/ --fix
-python -m ruff format src/ test/
+python -m ruff check src/ tests/ --fix
+python -m ruff format src/ tests/
 
 # Typecheck (only done for src/)
 python -m mypy src/

src/claude_code_sdk/_internal/transport/subprocess_cli.py

@@ -306,12 +306,9 @@ async def receive_messages(self) -> AsyncIterator[dict[str, Any]]:
                             yield data
                         except GeneratorExit:
                             return
-                    except json.JSONDecodeError as e:
-                        logger.warning(
-                            f"Failed to parse JSON from CLI output: {e}. Buffer content: {json_buffer[:200]}..."
-                        )
-                        # Clear buffer to avoid repeated parse attempts on malformed data
-                        json_buffer = ""
+                    except json.JSONDecodeError:
+                        # Don't clear buffer - we might be in the middle of a split JSON message
+                        # The buffer will be cleared when we successfully parse or hit size limit
                         continue
 
         except anyio.ClosedResourceError:

src/claude_code_sdk/client.py

@@ -98,7 +98,9 @@ def __init__(self, options: ClaudeCodeOptions | None = None):
         self._transport: Any | None = None
         os.environ["CLAUDE_CODE_ENTRYPOINT"] = "sdk-py-client"
 
-    async def connect(self, prompt: str | AsyncIterable[dict[str, Any]] | None = None) -> None:
+    async def connect(
+        self, prompt: str | AsyncIterable[dict[str, Any]] | None = None
+    ) -> None:
         """Connect to Claude with a prompt or message stream."""
         from ._internal.transport.subprocess_cli import SubprocessCLITransport
 
@@ -128,7 +130,9 @@ async def receive_messages(self) -> AsyncIterator[Message]:
             if message:
                 yield message
 
-    async def query(self, prompt: str | AsyncIterable[dict[str, Any]], session_id: str = "default") -> None:
+    async def query(
+        self, prompt: str | AsyncIterable[dict[str, Any]], session_id: str = "default"
+    ) -> None:
         """
         Send a new request in streaming mode.
 

src/claude_code_sdk/query.py

@@ -9,7 +9,9 @@
 
 
 async def query(
-    *, prompt: str | AsyncIterable[dict[str, Any]], options: ClaudeCodeOptions | None = None
+    *,
+    prompt: str | AsyncIterable[dict[str, Any]],
+    options: ClaudeCodeOptions | None = None,
 ) -> AsyncIterator[Message]:
     """
     Query Claude Code for one-shot or unidirectional streaming interactions.