pydantic-claude-code 0.2.0__tar.gz → 0.3.0__tar.gz

{pydantic_claude_code-0.2.0 → pydantic_claude_code-0.3.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.5
 Name: pydantic-claude-code
-Version: 0.2.0
+Version: 0.3.0
 Summary: Use your Claude Code subscription from a Pydantic AI Agent, with full tool support
 Project-URL: Repository, https://github.com/mpfaffenberger/pydantic-ai-claude-code
 Project-URL: Homepage, https://pypi.org/project/pydantic-claude-code/
@@ -47,8 +47,9 @@ Two deliberate design choices distinguish this from a fork of pydantic-ai:
    executes your tools, and validates structured output. The wheel is a model
    + provider, not a second agent fighting for control.
 2. **Object-only resolution.** Instead of a `claude-code:` model-name string
-   (which would require patching pydantic-ai's internals), pass the model
-   object you build from the provider.
+   (which would require patching pydantic-ai's internals), construct a
+   `ClaudeCodeModel('claude-fable-5-1')` directly; it wires itself up to a
+   `ClaudeCodeProvider` by default.
 
 ## Quick start
 
@@ -57,15 +58,14 @@ import asyncio
 
 from pydantic_ai import Agent
 
-from pydantic_ai_claude_code import ClaudeCodeProvider, login
+from pydantic_ai_claude_code import ClaudeCodeModel, login
 
 async def main() -> None:
     # One-time: opens your browser, mints tokens, stores them
     # (only needs to run again when tokens are revoked).
     await login()
 
-    provider = ClaudeCodeProvider()  # loads the stored tokens
-    agent = Agent(provider.model('claude-fable-5-1'))
+    agent = Agent(ClaudeCodeModel('claude-fable-5-1'))
 
     result = await agent.run('Say hi in three words.')
     print(result.data)
@@ -77,9 +77,9 @@ asyncio.run(main())
 
 ```python
 from pydantic_ai import Agent
+from pydantic_ai_claude_code import ClaudeCodeModel
 
-provider = ClaudeCodeProvider()
-agent = Agent(provider.model('claude-fable-5-1'))
+agent = Agent(ClaudeCodeModel('claude-fable-5-1'))
 
 @agent.tool_plain
 def add(a: int, b: int) -> int:
@@ -96,12 +96,13 @@ definitions, and structured output works the same way as with the built-in
 ```python
 from pydantic import BaseModel
 from pydantic_ai import Agent
+from pydantic_ai_claude_code import ClaudeCodeModel
 
 class Weather(BaseModel):
     city: str
     temperature_c: float
 
-agent = Agent(ClaudeCodeProvider().model('claude-fable-5-1'), output_type=Weather)
+agent = Agent(ClaudeCodeModel('claude-fable-5-1'), output_type=Weather)
 result = await agent.run('Weather in Paris right now?')
 assert result.output.city == 'Paris'
 ```
@@ -110,8 +111,9 @@ assert result.output.city == 'Paris'
 
 ```python
 from pydantic_ai import Agent
+from pydantic_ai_claude_code import ClaudeCodeModel
 
-agent = Agent(ClaudeCodeProvider().model('claude-fable-5-1'))
+agent = Agent(ClaudeCodeModel('claude-fable-5-1'))
 async with agent.run_stream('Count from 1 to 3.') as stream:
     async for chunk in stream.stream_text():
         print(chunk, end='', flush=True)

{pydantic_claude_code-0.2.0 → pydantic_claude_code-0.3.0}/README.md

@@ -23,8 +23,9 @@ Two deliberate design choices distinguish this from a fork of pydantic-ai:
    executes your tools, and validates structured output. The wheel is a model
    + provider, not a second agent fighting for control.
 2. **Object-only resolution.** Instead of a `claude-code:` model-name string
-   (which would require patching pydantic-ai's internals), pass the model
-   object you build from the provider.
+   (which would require patching pydantic-ai's internals), construct a
+   `ClaudeCodeModel('claude-fable-5-1')` directly; it wires itself up to a
+   `ClaudeCodeProvider` by default.
 
 ## Quick start
 
@@ -33,15 +34,14 @@ import asyncio
 
 from pydantic_ai import Agent
 
-from pydantic_ai_claude_code import ClaudeCodeProvider, login
+from pydantic_ai_claude_code import ClaudeCodeModel, login
 
 async def main() -> None:
     # One-time: opens your browser, mints tokens, stores them
     # (only needs to run again when tokens are revoked).
     await login()
 
-    provider = ClaudeCodeProvider()  # loads the stored tokens
-    agent = Agent(provider.model('claude-fable-5-1'))
+    agent = Agent(ClaudeCodeModel('claude-fable-5-1'))
 
     result = await agent.run('Say hi in three words.')
     print(result.data)
@@ -53,9 +53,9 @@ asyncio.run(main())
 
 ```python
 from pydantic_ai import Agent
+from pydantic_ai_claude_code import ClaudeCodeModel
 
-provider = ClaudeCodeProvider()
-agent = Agent(provider.model('claude-fable-5-1'))
+agent = Agent(ClaudeCodeModel('claude-fable-5-1'))
 
 @agent.tool_plain
 def add(a: int, b: int) -> int:
@@ -72,12 +72,13 @@ definitions, and structured output works the same way as with the built-in
 ```python
 from pydantic import BaseModel
 from pydantic_ai import Agent
+from pydantic_ai_claude_code import ClaudeCodeModel
 
 class Weather(BaseModel):
     city: str
     temperature_c: float
 
-agent = Agent(ClaudeCodeProvider().model('claude-fable-5-1'), output_type=Weather)
+agent = Agent(ClaudeCodeModel('claude-fable-5-1'), output_type=Weather)
 result = await agent.run('Weather in Paris right now?')
 assert result.output.city == 'Paris'
 ```
@@ -86,8 +87,9 @@ assert result.output.city == 'Paris'
 
 ```python
 from pydantic_ai import Agent
+from pydantic_ai_claude_code import ClaudeCodeModel
 
-agent = Agent(ClaudeCodeProvider().model('claude-fable-5-1'))
+agent = Agent(ClaudeCodeModel('claude-fable-5-1'))
 async with agent.run_stream('Count from 1 to 3.') as stream:
     async for chunk in stream.stream_text():
         print(chunk, end='', flush=True)

{pydantic_claude_code-0.2.0 → pydantic_claude_code-0.3.0}/examples/basic.py

@@ -11,7 +11,7 @@ import asyncio
 
 from pydantic_ai import Agent
 
-from pydantic_ai_claude_code import ClaudeCodeProvider, default_store, login
+from pydantic_ai_claude_code import ClaudeCodeModel, default_store, login
 
 
 async def main() -> None:
@@ -19,8 +19,7 @@ async def main() -> None:
     if default_store().load() is None:
         await login()
 
-    provider = ClaudeCodeProvider()
-    agent = Agent(provider.model("claude-fable-5-1"))
+    agent = Agent(ClaudeCodeModel("claude-fable-5-1"))
 
     result = await agent.run("Say hi in exactly three words.")
     print("Agent said:", result.output)

{pydantic_claude_code-0.2.0 → pydantic_claude_code-0.3.0}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "pydantic-claude-code"
-version = "0.2.0"
+version = "0.3.0"
 description = "Use your Claude Code subscription from a Pydantic AI Agent, with full tool support"
 readme = "README.md"
 requires-python = ">=3.11,<3.15"

{pydantic_claude_code-0.2.0 → pydantic_claude_code-0.3.0}/src/pydantic_ai_claude_code/__init__.py

@@ -3,10 +3,9 @@
 Quick start:
 
     from pydantic_ai import Agent
-    from pydantic_ai_claude_code import ClaudeCodeProvider
+    from pydantic_ai_claude_code import ClaudeCodeModel
 
-    provider = ClaudeCodeProvider()
-    agent = Agent(provider.model('claude-fable-5-1'))
+    agent = Agent(ClaudeCodeModel('claude-fable-5-1'))
 """
 
 from .credentials import ClaudeCodeCredentials

{pydantic_claude_code-0.2.0 → pydantic_claude_code-0.3.0}/src/pydantic_ai_claude_code/model.py

@@ -2,8 +2,11 @@
 
 from __future__ import annotations
 
+from typing import Any
+
 from pydantic_ai.messages import ModelMessage, ModelRequest, SystemPromptPart
 from pydantic_ai.models.anthropic import AnthropicModel
+from pydantic_ai.providers import Provider
 
 from . import config
 
@@ -17,6 +20,26 @@ class ClaudeCodeModel(AnthropicModel):
     duplicate it.
     """
 
+    def __init__(
+        self,
+        model_name: str,
+        *,
+        provider: Provider[Any] | None = None,
+        **kwargs: Any,
+    ) -> None:
+        """Initialize a Claude Code model.
+
+        Args:
+            model_name: The model name to use, e.g. `'claude-fable-5-1'`.
+            provider: The provider to authenticate with. Defaults to a
+                `ClaudeCodeProvider` backed by the default token store.
+        """
+        if provider is None:
+            from .provider import ClaudeCodeProvider
+
+            provider = ClaudeCodeProvider()
+        super().__init__(model_name, provider=provider, **kwargs)
+
     def prepare_messages(
         self,
         messages: list[ModelMessage],

{pydantic_claude_code-0.2.0 → pydantic_claude_code-0.3.0}/src/pydantic_ai_claude_code/provider.py

@@ -12,7 +12,6 @@ from pydantic_ai.exceptions import UserError
 from pydantic_ai.providers.anthropic import AnthropicProvider
 
 from . import config
-from .model import ClaudeCodeModel
 from .auth import _ClaudeCodeAuth
 from .credentials import ClaudeCodeCredentials
 from .storage import TokenStore, default_store
@@ -85,15 +84,6 @@ class ClaudeCodeProvider(AnthropicProvider):
         """The current credentials; refreshed in place by the auth shim."""
         return self._claude_code_auth.credentials
 
-    def model(self, model_name: str) -> ClaudeCodeModel:
-        """Build a `ClaudeCodeModel` bound to this provider.
-
-        This is the object-only entry point agreed for the wheel: pass its result
-        to `Agent(model=...)` without modifying pydantic-ai. The persona is
-        prepended to the system context automatically.
-        """
-        return ClaudeCodeModel(model_name, provider=self)
-
     @classmethod
     def from_token_store(
         cls,

{pydantic_claude_code-0.2.0 → pydantic_claude_code-0.3.0}/tests/test_integration.py

@@ -16,6 +16,7 @@ import pytest
 from pydantic_ai import Agent
 
 from pydantic_ai_claude_code.credentials import ClaudeCodeCredentials
+from pydantic_ai_claude_code.model import ClaudeCodeModel
 from pydantic_ai_claude_code.provider import ClaudeCodeProvider
 
 
@@ -58,7 +59,7 @@ def messages_stub(monkeypatch) -> ThreadingHTTPServer:
 def test_agent_round_trip(messages_stub, model_name: str) -> None:
     creds = ClaudeCodeCredentials(access_token="sub-token", refresh_token="refresh")
     provider = ClaudeCodeProvider(credentials=creds, base_url=f"http://127.0.0.1:{messages_stub.server_address[1]}")
-    agent = Agent(provider.model(model_name))
+    agent = Agent(ClaudeCodeModel(model_name, provider=provider))
 
     result = asyncio_run(agent.run("Say hi."))
 

{pydantic_claude_code-0.2.0 → pydantic_claude_code-0.3.0}/tests/test_provider.py

@@ -8,6 +8,7 @@ import httpx2
 from pydantic_ai.exceptions import UserError
 
 from pydantic_ai_claude_code.credentials import ClaudeCodeCredentials
+from pydantic_ai_claude_code.model import ClaudeCodeModel
 from pydantic_ai_claude_code.provider import ClaudeCodeProvider
 from pydantic_ai_claude_code.storage import ClaudeCodeTokenStore
 
@@ -23,13 +24,21 @@ def test_credentials_property() -> None:
     assert provider.credentials.token == "a"
 
 
-def test_model_binding() -> None:
+def test_model_binds_explicit_provider() -> None:
     provider = ClaudeCodeProvider(credentials=ClaudeCodeCredentials(access_token="a", refresh_token="b"))
-    model = provider.model("claude-sonnet-4-5")
+    model = ClaudeCodeModel("claude-sonnet-4-5", provider=provider)
     assert model.model_name == "claude-sonnet-4-5"
     assert model._provider is provider  # type: ignore[attr-defined]
 
 
+def test_model_defaults_provider(tmp_path, monkeypatch) -> None:
+    monkeypatch.setenv("CLAUDE_CODE_CREDENTIALS", "file")
+    monkeypatch.setenv("CLAUDE_CODE_AUTH_FILE", str(tmp_path / "auth.json"))
+    ClaudeCodeTokenStore().save(ClaudeCodeCredentials(access_token="a", refresh_token="b"))
+    model = ClaudeCodeModel("claude-sonnet-4-5")
+    assert isinstance(model._provider, ClaudeCodeProvider)  # type: ignore[attr-defined]
+
+
 def test_missing_credentials_raises(tmp_path) -> None:
     store = ClaudeCodeTokenStore(path=tmp_path / "missing.json")
     with pytest.raises(UserError):