pydantic-ai-slim 0.1.10__tar.gz → 0.1.11__tar.gz

{pydantic_ai_slim-0.1.10 → pydantic_ai_slim-0.1.11}/PKG-INFO

@@ -1,8 +1,8 @@
 Metadata-Version: 2.4
 Name: pydantic-ai-slim
-Version: 0.1.10
+Version: 0.1.11
 Summary: Agent Framework / shim to use Pydantic with LLMs, slim package
-Author-email: Samuel Colvin <samuel@pydantic.dev>
+Author-email: Samuel Colvin <samuel@pydantic.dev>, Marcelo Trylesinski <marcelotryle@gmail.com>, David Montague <david@pydantic.dev>, Alex Hall <alex@pydantic.dev>
 License-Expression: MIT
 Classifier: Development Status :: 4 - Beta
 Classifier: Environment :: Console
@@ -29,7 +29,7 @@ Requires-Dist: exceptiongroup; python_version < '3.11'
 Requires-Dist: griffe>=1.3.2
 Requires-Dist: httpx>=0.27
 Requires-Dist: opentelemetry-api>=1.28.0
-Requires-Dist: pydantic-graph==0.1.10
+Requires-Dist: pydantic-graph==0.1.11
 Requires-Dist: pydantic>=2.10
 Requires-Dist: typing-inspection>=0.4.0
 Provides-Extra: anthropic
@@ -45,7 +45,7 @@ Requires-Dist: cohere>=5.13.11; (platform_system != 'Emscripten') and extra == '
 Provides-Extra: duckduckgo
 Requires-Dist: duckduckgo-search>=7.0.0; extra == 'duckduckgo'
 Provides-Extra: evals
-Requires-Dist: pydantic-evals==0.1.10; extra == 'evals'
+Requires-Dist: pydantic-evals==0.1.11; extra == 'evals'
 Provides-Extra: groq
 Requires-Dist: groq>=0.15.0; extra == 'groq'
 Provides-Extra: logfire
@@ -67,8 +67,8 @@ Description-Content-Type: text/markdown
 
 [![CI](https://github.com/pydantic/pydantic-ai/actions/workflows/ci.yml/badge.svg?event=push)](https://github.com/pydantic/pydantic-ai/actions/workflows/ci.yml?query=branch%3Amain)
 [![Coverage](https://coverage-badge.samuelcolvin.workers.dev/pydantic/pydantic-ai.svg)](https://coverage-badge.samuelcolvin.workers.dev/redirect/pydantic/pydantic-ai)
-[![PyPI](https://img.shields.io/pypi/v/pydantic-ai.svg)](https://pypi.python.org/pypi/pydantic-ai)
-[![versions](https://img.shields.io/pypi/pyversions/pydantic-ai.svg)](https://github.com/pydantic/pydantic-ai)
+[![PyPI](https://img.shields.io/pypi/v/pydantic-ai-slim.svg)](https://pypi.python.org/pypi/pydantic-ai-slim)
+[![versions](https://img.shields.io/pypi/pyversions/pydantic-ai-slim.svg)](https://github.com/pydantic/pydantic-ai)
 [![license](https://img.shields.io/github/license/pydantic/pydantic-ai.svg?v)](https://github.com/pydantic/pydantic-ai/blob/main/LICENSE)
 
 PydanticAI core logic with minimal required dependencies.

{pydantic_ai_slim-0.1.10 → pydantic_ai_slim-0.1.11}/README.md

@@ -2,8 +2,8 @@
 
 [![CI](https://github.com/pydantic/pydantic-ai/actions/workflows/ci.yml/badge.svg?event=push)](https://github.com/pydantic/pydantic-ai/actions/workflows/ci.yml?query=branch%3Amain)
 [![Coverage](https://coverage-badge.samuelcolvin.workers.dev/pydantic/pydantic-ai.svg)](https://coverage-badge.samuelcolvin.workers.dev/redirect/pydantic/pydantic-ai)
-[![PyPI](https://img.shields.io/pypi/v/pydantic-ai.svg)](https://pypi.python.org/pypi/pydantic-ai)
-[![versions](https://img.shields.io/pypi/pyversions/pydantic-ai.svg)](https://github.com/pydantic/pydantic-ai)
+[![PyPI](https://img.shields.io/pypi/v/pydantic-ai-slim.svg)](https://pypi.python.org/pypi/pydantic-ai-slim)
+[![versions](https://img.shields.io/pypi/pyversions/pydantic-ai-slim.svg)](https://github.com/pydantic/pydantic-ai)
 [![license](https://img.shields.io/github/license/pydantic/pydantic-ai.svg?v)](https://github.com/pydantic/pydantic-ai/blob/main/LICENSE)
 
 PydanticAI core logic with minimal required dependencies.

{pydantic_ai_slim-0.1.10 → pydantic_ai_slim-0.1.11}/pydantic_ai/__main__.py

@@ -1,6 +1,6 @@
 """This means `python -m pydantic_ai` should run the CLI."""
 
-from ._cli import app
+from ._cli import cli_exit
 
 if __name__ == '__main__':
-    app()
+    cli_exit()

{pydantic_ai_slim-0.1.10 → pydantic_ai_slim-0.1.11}/pydantic_ai/_cli.py

@@ -7,16 +7,16 @@ from asyncio import CancelledError
 from collections.abc import Sequence
 from contextlib import ExitStack
 from datetime import datetime, timezone
-from importlib.metadata import version
 from pathlib import Path
 from typing import Any, cast
 
 from typing_inspection.introspection import get_literal_values
 
-from pydantic_ai.agent import Agent
-from pydantic_ai.exceptions import UserError
-from pydantic_ai.messages import ModelMessage
-from pydantic_ai.models import KnownModelName, infer_model
+from . import __version__
+from .agent import Agent
+from .exceptions import UserError
+from .messages import ModelMessage
+from .models import KnownModelName, infer_model
 
 try:
     import argcomplete
@@ -39,7 +39,7 @@ except ImportError as _import_error:
     ) from _import_error
 
 
-__version__ = version('pydantic-ai-slim')
+__all__ = 'cli', 'cli_exit'
 
 
 class SimpleCodeBlock(CodeBlock):
@@ -83,14 +83,20 @@ The current date and time is {datetime.now()} {tzname}.
 The user is running {sys.platform}."""
 
 
-def cli(args_list: Sequence[str] | None = None) -> int:
+def cli_exit(prog_name: str = 'pai'):  # pragma: no cover
+    """Run the CLI and exit."""
+    sys.exit(cli(prog_name=prog_name))
+
+
+def cli(args_list: Sequence[str] | None = None, *, prog_name: str = 'pai') -> int:
+    """Run the CLI and return the exit code for the process."""
     parser = argparse.ArgumentParser(
-        prog='pai',
+        prog=prog_name,
         description=f"""\
 PydanticAI CLI v{__version__}\n\n
 
-Special prompt:
-* `/exit` - exit the interactive mode
+Special prompts:
+* `/exit` - exit the interactive mode (ctrl-c and ctrl-d also work)
 * `/markdown` - show the last markdown output of the last question
 * `/multiline` - toggle multiline mode
 """,
@@ -101,7 +107,7 @@ Special prompt:
         '-m',
         '--model',
         nargs='?',
-        help='Model to use, in format "<provider>:<model>" e.g. "openai:gpt-4o". Defaults to "openai:gpt-4o".',
+        help='Model to use, in format "<provider>:<model>" e.g. "openai:gpt-4o" or "anthropic:claude-3-7-sonnet-latest". Defaults to "openai:gpt-4o".',
         default='openai:gpt-4o',
     )
     # we don't want to autocomplete or list models that don't include the provider,
@@ -118,10 +124,10 @@ Special prompt:
         '-t',
         '--code-theme',
         nargs='?',
-        help='Which colors to use for code, can be "dark", "light" or any theme from pygments.org/styles/. Defaults to "monokai".',
-        default='monokai',
+        help='Which colors to use for code, can be "dark", "light" or any theme from pygments.org/styles/. Defaults to "dark" which works well on dark terminals.',
+        default='dark',
     )
-    parser.add_argument('--no-stream', action='store_true', help='Whether to stream responses from the model')
+    parser.add_argument('--no-stream', action='store_true', help='Disable streaming from the model')
     parser.add_argument('--version', action='store_true', help='Show version and exit')
 
     argcomplete.autocomplete(parser)
@@ -129,7 +135,8 @@ Special prompt:
 
     console = Console()
     console.print(
-        f'[green]pai - PydanticAI CLI v{__version__} using[/green] [magenta]{args.model}[/magenta]', highlight=False
+        f'[green]{prog_name} - PydanticAI CLI v{__version__} using[/green] [magenta]{args.model}[/magenta]',
+        highlight=False,
     )
     if args.version:
         return 0
@@ -160,23 +167,25 @@ Special prompt:
             pass
         return 0
 
-    history = Path.home() / '.pai-prompt-history.txt'
+    history = Path.home() / f'.{prog_name}-prompt-history.txt'
     # doing this instead of `PromptSession[Any](history=` allows mocking of PromptSession in tests
     session: PromptSession[Any] = PromptSession(history=FileHistory(str(history)))
     try:
-        return asyncio.run(run_chat(session, stream, cli_agent, console, code_theme))
+        return asyncio.run(run_chat(session, stream, cli_agent, console, code_theme, prog_name))
     except KeyboardInterrupt:  # pragma: no cover
         return 0
 
 
-async def run_chat(session: PromptSession[Any], stream: bool, agent: Agent, console: Console, code_theme: str) -> int:
+async def run_chat(
+    session: PromptSession[Any], stream: bool, agent: Agent, console: Console, code_theme: str, prog_name: str
+) -> int:
     multiline = False
     messages: list[ModelMessage] = []
 
     while True:
         try:
             auto_suggest = CustomAutoSuggest(['/markdown', '/multiline', '/exit'])
-            text = await session.prompt_async('pai ➤ ', auto_suggest=auto_suggest, multiline=multiline)
+            text = await session.prompt_async(f'{prog_name} ➤ ', auto_suggest=auto_suggest, multiline=multiline)
         except (KeyboardInterrupt, EOFError):  # pragma: no cover
             return 0
 
@@ -214,14 +223,14 @@ async def ask_agent(
 
     with status, ExitStack() as stack:
         async with agent.iter(prompt, message_history=messages) as agent_run:
-            live = Live('', refresh_per_second=15, console=console, vertical_overflow='visible')
+            live = Live('', refresh_per_second=15, console=console, vertical_overflow='ellipsis')
             async for node in agent_run:
                 if Agent.is_model_request_node(node):
                     async with node.stream(agent_run.ctx) as handle_stream:
                         status.stop()  # stopping multiple times is idempotent
                         stack.enter_context(live)  # entering multiple times is idempotent
 
-                        async for content in handle_stream.stream_output():
+                        async for content in handle_stream.stream_output(debounce_by=None):
                             live.update(Markdown(content, code_theme=code_theme))
 
         assert agent_run.result is not None
@@ -282,7 +291,3 @@ def handle_slash_command(
     else:
         console.print(f'[red]Unknown command[/red] [magenta]`{ident_prompt}`[/magenta]')
     return None, multiline
-
-
-def app():  # pragma: no cover
-    sys.exit(cli())

{pydantic_ai_slim-0.1.10 → pydantic_ai_slim-0.1.11}/pydantic_ai/exceptions.py

@@ -3,9 +3,9 @@ from __future__ import annotations as _annotations
 import json
 import sys
 
-if sys.version_info < (3, 11):  # pragma: no cover
+if sys.version_info < (3, 11):
     from exceptiongroup import ExceptionGroup
-else:  # pragma: no cover
+else:
     ExceptionGroup = ExceptionGroup
 
 __all__ = (

{pydantic_ai_slim-0.1.10 → pydantic_ai_slim-0.1.11}/pydantic_ai/messages.py

@@ -81,7 +81,7 @@ class VideoUrl:
     """Type identifier, this is available on all parts as a discriminator."""
 
     @property
-    def media_type(self) -> VideoMediaType:  # pragma: no cover
+    def media_type(self) -> VideoMediaType:  # pragma: lax no cover
         """Return the media type of the video, based on the url."""
         if self.url.endswith('.mkv'):
             return 'video/x-matroska'

{pydantic_ai_slim-0.1.10 → pydantic_ai_slim-0.1.11}/pydantic_ai/models/_json_schema.py

@@ -99,7 +99,7 @@ class WalkJsonSchema(ABC):
         if (additional_properties := schema.get('additionalProperties')) is not None:
             if isinstance(additional_properties, bool):
                 schema['additionalProperties'] = additional_properties
-            else:  # pragma: no cover
+            else:
                 schema['additionalProperties'] = self._handle(additional_properties)
 
         if (pattern_properties := schema.get('patternProperties')) is not None:

{pydantic_ai_slim-0.1.10 → pydantic_ai_slim-0.1.11}/pydantic_ai/models/function.py

@@ -282,7 +282,7 @@ def _estimate_string_tokens(content: str | Sequence[UserContent]) -> int:
         return 0
     if isinstance(content, str):
         return len(re.split(r'[\s",.:]+', content.strip()))
-    else:  # pragma: no cover
+    else:
         tokens = 0
         for part in content:
             if isinstance(part, str):

{pydantic_ai_slim-0.1.10 → pydantic_ai_slim-0.1.11}/pydantic_ai/models/gemini.py

@@ -805,7 +805,7 @@ class _GeminiJsonSchema(WalkJsonSchema):
         additional_properties = schema.pop(
             'additionalProperties', None
         )  # don't pop yet so it's included in the warning
-        if additional_properties:  # pragma: no cover
+        if additional_properties:
             original_schema = {**schema, 'additionalProperties': additional_properties}
             warnings.warn(
                 '`additionalProperties` is not supported by Gemini; it will be removed from the tool JSON schema.'

{pydantic_ai_slim-0.1.10 → pydantic_ai_slim-0.1.11}/pydantic_ai/models/openai.py

@@ -770,7 +770,7 @@ class OpenAIResponsesModel(Model):
                             file_data=f'data:{item.media_type};base64,{base64_encoded}',
                         )
                     )
-                elif isinstance(item, DocumentUrl):  # pragma: no cover
+                elif isinstance(item, DocumentUrl):
                     client = cached_async_http_client()
                     response = await client.get(item.url)
                     response.raise_for_status()

{pydantic_ai_slim-0.1.10 → pydantic_ai_slim-0.1.11}/pydantic_ai/providers/__init__.py

@@ -65,7 +65,7 @@ def infer_provider(provider: str) -> Provider[Any]:
 
         return GoogleGLAProvider()
     # NOTE: We don't test because there are many ways the `boto3.client` can retrieve the credentials.
-    elif provider == 'bedrock':  # pragma: no cover
+    elif provider == 'bedrock':
         from .bedrock import BedrockProvider
 
         return BedrockProvider()

{pydantic_ai_slim-0.1.10 → pydantic_ai_slim-0.1.11}/pydantic_ai/providers/azure.py

@@ -82,7 +82,7 @@ class AzureProvider(Provider[AsyncOpenAI]):
             self._client = openai_client
         else:
             azure_endpoint = azure_endpoint or os.getenv('AZURE_OPENAI_ENDPOINT')
-            if not azure_endpoint:  # pragma: no cover
+            if not azure_endpoint:
                 raise UserError(
                     'Must provide one of the `azure_endpoint` argument or the `AZURE_OPENAI_ENDPOINT` environment variable'
                 )

{pydantic_ai_slim-0.1.10 → pydantic_ai_slim-0.1.11}/pyproject.toml

@@ -14,7 +14,12 @@ bump = true
 name = "pydantic-ai-slim"
 dynamic = ["version", "dependencies", "optional-dependencies"]
 description = "Agent Framework / shim to use Pydantic with LLMs, slim package"
-authors = [{ name = "Samuel Colvin", email = "samuel@pydantic.dev" }]
+authors = [
+  { name = "Samuel Colvin", email = "samuel@pydantic.dev" },
+  { name = "Marcelo Trylesinski", email = "marcelotryle@gmail.com" },
+  { name = "David Montague", email = "david@pydantic.dev" },
+  { name = "Alex Hall", email = "alex@pydantic.dev" },
+]
 license = "MIT"
 readme = "README.md"
 classifiers = [
@@ -87,13 +92,14 @@ dev = [
     "pytest-recording>=0.13.2",
     "diff-cover>=9.2.0",
     "boto3-stubs[bedrock-runtime]",
+    "strict-no-cover>=0.1.1",
 ]
 
 [tool.hatch.metadata]
 allow-direct-references = true
 
 [project.scripts]
-pai = "pydantic_ai._cli:app"
+pai = "pydantic_ai._cli:cli_exit"  # TODO remove this when clai has been out for a while
 
 [tool.hatch.build.targets.wheel]
 packages = ["pydantic_ai"]