pydantic-ai-slim 0.4.3__py3-none-any.whl → 0.4.4__py3-none-any.whl

pydantic_ai/_cli.py

@@ -38,7 +38,7 @@ try:
     from rich.text import Text
 except ImportError as _import_error:
     raise ImportError(
-        'Please install `rich`, `prompt-toolkit` and `argcomplete` to use the PydanticAI CLI, '
+        'Please install `rich`, `prompt-toolkit` and `argcomplete` to use the Pydantic AI CLI, '
         'you can use the `cli` optional group — `pip install "pydantic-ai-slim[cli]"`'
     ) from _import_error
 
@@ -47,7 +47,7 @@ __all__ = 'cli', 'cli_exit'
 
 
 PYDANTIC_AI_HOME = Path.home() / '.pydantic-ai'
-"""The home directory for PydanticAI CLI.
+"""The home directory for Pydantic AI CLI.
 
 This folder is used to store the prompt history and configuration.
 """
@@ -101,12 +101,14 @@ def cli_exit(prog_name: str = 'pai'):  # pragma: no cover
     sys.exit(cli(prog_name=prog_name))
 
 
-def cli(args_list: Sequence[str] | None = None, *, prog_name: str = 'pai') -> int:  # noqa: C901
+def cli(  # noqa: C901
+    args_list: Sequence[str] | None = None, *, prog_name: str = 'pai', default_model: str = 'openai:gpt-4.1'
+) -> int:
     """Run the CLI and return the exit code for the process."""
     parser = argparse.ArgumentParser(
         prog=prog_name,
         description=f"""\
-PydanticAI CLI v{__version__}\n\n
+Pydantic AI CLI v{__version__}\n\n
 
 Special prompts:
 * `/exit` - exit the interactive mode (ctrl-c and ctrl-d also work)
@@ -120,7 +122,7 @@ Special prompts:
         '-m',
         '--model',
         nargs='?',
-        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".',
+        help=f'Model to use, in format "<provider>:<model>" e.g. "openai:gpt-4.1" or "anthropic:claude-sonnet-4-0". Defaults to "{default_model}".',
     )
     # we don't want to autocomplete or list models that don't include the provider,
     # e.g. we want to show `openai:gpt-4o` but not `gpt-4o`
@@ -151,7 +153,7 @@ Special prompts:
     args = parser.parse_args(args_list)
 
     console = Console()
-    name_version = f'[green]{prog_name} - PydanticAI CLI v{__version__}[/green]'
+    name_version = f'[green]{prog_name} - Pydantic AI CLI v{__version__}[/green]'
     if args.version:
         console.print(name_version, highlight=False)
         return 0
@@ -179,7 +181,7 @@ Special prompts:
     model_arg_set = args.model is not None
     if agent.model is None or model_arg_set:
         try:
-            agent.model = infer_model(args.model or 'openai:gpt-4o')
+            agent.model = infer_model(args.model or default_model)
         except UserError as e:
             console.print(f'Error initializing [magenta]{args.model}[/magenta]:\n[red]{e}[/red]')
             return 1