Add customization of Claude Code binary path in query and connect functions

src/claude_code_sdk/_internal/client.py

@@ -1,6 +1,7 @@
 """Internal client implementation."""
 
 from collections.abc import AsyncIterable, AsyncIterator
+from pathlib import Path
 from typing import Any
 
 from ..types import ClaudeCodeOptions, Message
@@ -15,12 +16,15 @@ def __init__(self) -> None:
         """Initialize the internal client."""
 
     async def process_query(
-        self, prompt: str | AsyncIterable[dict[str, Any]], options: ClaudeCodeOptions
+        self,
+        prompt: str | AsyncIterable[dict[str, Any]],
+        options: ClaudeCodeOptions,
+        cli_path: Path | None = None
     ) -> AsyncIterator[Message]:
         """Process a query through transport."""
 
         transport = SubprocessCLITransport(
-            prompt=prompt, options=options, close_stdin_after_prompt=True
+            prompt=prompt, options=options, cli_path=cli_path, close_stdin_after_prompt=True
         )
 
         try:

src/claude_code_sdk/client.py

@@ -2,6 +2,7 @@
 
 import os
 from collections.abc import AsyncIterable, AsyncIterator
+from pathlib import Path
 from typing import Any
 
 from ._errors import CLIConnectionError
@@ -99,7 +100,9 @@ def __init__(self, options: ClaudeCodeOptions | None = None):
         os.environ["CLAUDE_CODE_ENTRYPOINT"] = "sdk-py-client"
 
     async def connect(
-        self, prompt: str | AsyncIterable[dict[str, Any]] | None = None
+        self,
+        prompt: str | AsyncIterable[dict[str, Any]] | None = None,
+        cli_path: Path | None = None
     ) -> None:
         """Connect to Claude with a prompt or message stream."""
         from ._internal.transport.subprocess_cli import SubprocessCLITransport
@@ -115,6 +118,7 @@ async def _empty_stream() -> AsyncIterator[dict[str, Any]]:
         self._transport = SubprocessCLITransport(
             prompt=_empty_stream() if prompt is None else prompt,
             options=self.options,
+            cli_path=cli_path
         )
         await self._transport.connect()
 

src/claude_code_sdk/query.py

@@ -2,6 +2,7 @@
 
 import os
 from collections.abc import AsyncIterable, AsyncIterator
+from pathlib import Path
 from typing import Any
 
 from ._internal.client import InternalClient
@@ -12,6 +13,7 @@ async def query(
     *,
     prompt: str | AsyncIterable[dict[str, Any]],
     options: ClaudeCodeOptions | None = None,
+    cli_path: Path | None = None
 ) -> AsyncIterator[Message]:
     """
     Query Claude Code for one-shot or unidirectional streaming interactions.
@@ -56,6 +58,7 @@ async def query(
                  - 'acceptEdits': Auto-accept file edits
                  - 'bypassPermissions': Allow all tools (use with caution)
                  Set options.cwd for working directory.
+         cli_path: Optional Path of Claude Code binary.
 
     Yields:
         Messages from the conversation
@@ -98,5 +101,5 @@ async def prompts():
 
     client = InternalClient()
 
-    async for message in client.process_query(prompt=prompt, options=options):
+    async for message in client.process_query(prompt=prompt, options=options, cli_path=cli_path):
         yield message