pydantic-ai-slim 0.2.3__tar.gz → 0.2.4__tar.gz

{pydantic_ai_slim-0.2.3 → pydantic_ai_slim-0.2.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pydantic-ai-slim
-Version: 0.2.3
+Version: 0.2.4
 Summary: Agent Framework / shim to use Pydantic with LLMs, slim package
 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
@@ -26,15 +26,14 @@ Classifier: Topic :: Software Development :: Libraries :: Python Modules
 Requires-Python: >=3.9
 Requires-Dist: eval-type-backport>=0.2.0
 Requires-Dist: exceptiongroup; python_version < '3.11'
-Requires-Dist: fasta2a==0.2.3
 Requires-Dist: griffe>=1.3.2
 Requires-Dist: httpx>=0.27
 Requires-Dist: opentelemetry-api>=1.28.0
-Requires-Dist: pydantic-graph==0.2.3
+Requires-Dist: pydantic-graph==0.2.4
 Requires-Dist: pydantic>=2.10
 Requires-Dist: typing-inspection>=0.4.0
 Provides-Extra: a2a
-Requires-Dist: fasta2a==0.2.3; extra == 'a2a'
+Requires-Dist: fasta2a==0.2.4; extra == 'a2a'
 Provides-Extra: anthropic
 Requires-Dist: anthropic>=0.49.0; extra == 'anthropic'
 Provides-Extra: bedrock
@@ -48,7 +47,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.2.3; extra == 'evals'
+Requires-Dist: pydantic-evals==0.2.4; extra == 'evals'
 Provides-Extra: groq
 Requires-Dist: groq>=0.15.0; extra == 'groq'
 Provides-Extra: logfire

{pydantic_ai_slim-0.2.3 → pydantic_ai_slim-0.2.4}/pydantic_ai/_cli.py

@@ -102,7 +102,7 @@ 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:
+def cli(args_list: Sequence[str] | None = None, *, prog_name: str = 'pai') -> int:  # noqa: C901
     """Run the CLI and return the exit code for the process."""
     parser = argparse.ArgumentParser(
         prog=prog_name,
@@ -122,7 +122,6 @@ Special prompts:
         '--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".',
-        default='openai:gpt-4o',
     )
     # 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`
@@ -153,40 +152,49 @@ Special prompts:
     args = parser.parse_args(args_list)
 
     console = Console()
-    console.print(
-        f'[green]{prog_name} - PydanticAI CLI v{__version__} using[/green] [magenta]{args.model}[/magenta]',
-        highlight=False,
-    )
+    name_version = f'[green]{prog_name} - PydanticAI CLI v{__version__}[/green]'
     if args.version:
+        console.print(name_version, highlight=False)
         return 0
     if args.list_models:
-        console.print('Available models:', style='green bold')
+        console.print(f'{name_version}\n\n[green]Available models:[/green]')
         for model in qualified_model_names:
             console.print(f'  {model}', highlight=False)
         return 0
 
     agent: Agent[None, str] = cli_agent
     if args.agent:
+        sys.path.append(os.getcwd())
         try:
-            current_path = os.getcwd()
-            sys.path.append(current_path)
-
             module_path, variable_name = args.agent.split(':')
-            module = importlib.import_module(module_path)
-            agent = getattr(module, variable_name)
-            if not isinstance(agent, Agent):
-                console.print(f'[red]Error: {args.agent} is not an Agent instance[/red]')
-                return 1
-            console.print(f'[green]Using custom agent:[/green] [magenta]{args.agent}[/magenta]', highlight=False)
         except ValueError:
             console.print('[red]Error: Agent must be specified in "module:variable" format[/red]')
             return 1
 
-    try:
-        agent.model = infer_model(args.model)
-    except UserError as e:
-        console.print(f'Error initializing [magenta]{args.model}[/magenta]:\n[red]{e}[/red]')
-        return 1
+        module = importlib.import_module(module_path)
+        agent = getattr(module, variable_name)
+        if not isinstance(agent, Agent):
+            console.print(f'[red]Error: {args.agent} is not an Agent instance[/red]')
+            return 1
+
+    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')
+        except UserError as e:
+            console.print(f'Error initializing [magenta]{args.model}[/magenta]:\n[red]{e}[/red]')
+            return 1
+
+    model_name = agent.model if isinstance(agent.model, str) else f'{agent.model.system}:{agent.model.model_name}'
+    if args.agent and model_arg_set:
+        console.print(
+            f'{name_version} using custom agent [magenta]{args.agent}[/magenta] with [magenta]{model_name}[/magenta]',
+            highlight=False,
+        )
+    elif args.agent:
+        console.print(f'{name_version} using custom agent [magenta]{args.agent}[/magenta]', highlight=False)
+    else:
+        console.print(f'{name_version} with [magenta]{model_name}[/magenta]', highlight=False)
 
     stream = not args.no_stream
     if args.code_theme == 'light':

{pydantic_ai_slim-0.2.3 → pydantic_ai_slim-0.2.4}/pyproject.toml

@@ -50,7 +50,6 @@ dependencies = [
     "eval-type-backport>=0.2.0",
     "griffe>=1.3.2",
     "httpx>=0.27",
-    "fasta2a=={{ version }}",
     "pydantic>=2.10",
     "pydantic-graph=={{ version }}",
     "exceptiongroup; python_version < '3.11'",