fix: post rebase cleanup

Author: rushilpatel0

src/claude_code_sdk/__init__.py

@@ -7,12 +7,9 @@
     CLINotFoundError,
     ProcessError,
 )
-
 from ._internal.transport import Transport
-
 from .client import ClaudeSDKClient
 from .query import query
-
 from .types import (
     AssistantMessage,
     ClaudeCodeOptions,

src/claude_code_sdk/_internal/client.py

@@ -1,26 +1,13 @@
 """Internal client implementation."""
 
-from collections.abc import AsyncIterable, AsyncIterator
-from typing import Any
-
+from collections.abc import AsyncIterator
 
 from ..types import (
-    AssistantMessage,
     ClaudeCodeOptions,
-    ContentBlock,
     Message,
-    ResultMessage,
-    SystemMessage,
-    TextBlock,
-    ToolResultBlock,
-    ToolUseBlock,
-    UserMessage,
 )
-from .transport import Transport
-
-from ..types import ClaudeCodeOptions, Message
 from .message_parser import parse_message
-
+from .transport import Transport
 from .transport.subprocess_cli import SubprocessCLITransport
 
 
@@ -31,7 +18,10 @@ def __init__(self) -> None:
         """Initialize the internal client."""
 
     async def process_query(
-        self, prompt: str, options: ClaudeCodeOptions, transport: Transport | None = None
+        self,
+        prompt: str,
+        options: ClaudeCodeOptions,
+        transport: Transport | None = None,
     ) -> AsyncIterator[Message]:
         """Process a query through transport."""
 
@@ -40,8 +30,8 @@ async def process_query(
             chosen_transport = transport
         else:
             chosen_transport = SubprocessCLITransport(
-            prompt=prompt, options=options, close_stdin_after_prompt=True
-        )
+                prompt=prompt, options=options, close_stdin_after_prompt=True
+            )
 
         try:
             # Configure the transport with prompt and options
@@ -51,6 +41,5 @@ async def process_query(
             async for data in chosen_transport.receive_messages():
                 yield parse_message(data)
 
-
         finally:
             await chosen_transport.disconnect()

src/claude_code_sdk/_internal/transport/__init__.py

@@ -14,11 +14,6 @@ class Transport(ABC):
     must be updated to match interface changes.
     """
 
-    @abstractmethod
-    def configure(self, prompt: str, options: ClaudeCodeOptions) -> None:
-        """Configure transport with prompt and options."""
-        pass
-
     @abstractmethod
     async def connect(self) -> None:
         """Initialize connection."""

src/claude_code_sdk/_internal/transport/subprocess_cli.py

@@ -28,7 +28,6 @@
 class SubprocessCLITransport(Transport):
     """Subprocess transport using Claude Code CLI."""
 
-
     def __init__(
         self,
         prompt: str | AsyncIterable[dict[str, Any]],
@@ -51,12 +50,6 @@ def __init__(
         self._task_group: anyio.abc.TaskGroup | None = None
         self._stderr_file: Any = None  # tempfile.NamedTemporaryFile
 
-    def configure(self, prompt: str, options: ClaudeCodeOptions) -> None:
-        """Configure transport with prompt and options."""
-        self._prompt = prompt
-        self._options = options
-        self._cwd = str(options.cwd) if options.cwd else None
-
     def _find_cli(self) -> str:
         """Find Claude Code CLI binary."""
         if cli := shutil.which("claude"):
@@ -94,9 +87,6 @@ def _find_cli(self) -> str:
 
     def _build_command(self) -> list[str]:
         """Build CLI command with arguments."""
-        if not self._prompt or not self._options:
-            raise CLIConnectionError("Transport not configured. Call configure() first.")
-            
         cmd = [self._cli_path, "--output-format", "stream-json", "--verbose"]
 
         if self._options.system_prompt:

src/claude_code_sdk/query.py

@@ -99,11 +99,11 @@ async def prompts():
     Example - With custom transport:
         ```python
         from claude_code_sdk import query, Transport
-        
+
         class MyCustomTransport(Transport):
             # Implement custom transport logic
             pass
-        
+
         transport = MyCustomTransport()
         async for message in query(
             prompt="Hello",
@@ -119,5 +119,7 @@ class MyCustomTransport(Transport):
 
     client = InternalClient()
 
-    async for message in client.process_query(prompt=prompt, options=options, transport=transport):
+    async for message in client.process_query(
+        prompt=prompt, options=options, transport=transport
+    ):
         yield message