pydantic-ai-examples 0.0.12__tar.gz → 1.39.1__tar.gz

pydantic_ai_examples-1.39.1/.gitignore

@@ -0,0 +1,25 @@
+site
+.venv
+dist
+__pycache__
+.env
+.dev.vars
+/scratch/
+/.coverage
+env*/
+/TODO.md
+/postgres-data/
+.DS_Store
+.chat_app_messages.sqlite
+.cache/
+.vscode/
+/question_graph_history.json
+/docs-site/.wrangler/
+node_modules/
+**.idea/
+.coverage*
+/test_tmp/
+.mcp.json
+.claude/
+/.cursor/
+/.devcontainer/

pydantic_ai_examples-1.39.1/LICENSE

@@ -0,0 +1,21 @@
+The MIT License (MIT)
+
+Copyright (c) Pydantic Services Inc. 2024 to present
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

{pydantic_ai_examples-0.0.12 → pydantic_ai_examples-1.39.1}/PKG-INFO

@@ -1,10 +1,11 @@
-Metadata-Version: 2.3
+Metadata-Version: 2.4
 Name: pydantic-ai-examples
-Version: 0.0.12
-Summary: Examples of how to use PydanticAI and what it can do.
-Author-email: Samuel Colvin <samuel@pydantic.dev>
-License: MIT
-Classifier: Development Status :: 4 - Beta
+Version: 1.39.1
+Summary: Examples of how to use Pydantic AI and what it can do.
+Author-email: Samuel Colvin <samuel@pydantic.dev>, Marcelo Trylesinski <marcelotryle@gmail.com>, David Montague <david@pydantic.dev>, Alex Hall <alex@pydantic.dev>, Douwe Maan <douwe@pydantic.dev>
+License-Expression: MIT
+License-File: LICENSE
+Classifier: Development Status :: 5 - Production/Stable
 Classifier: Environment :: Console
 Classifier: Environment :: MacOS X
 Classifier: Intended Audience :: Developers
@@ -16,24 +17,31 @@ Classifier: Operating System :: Unix
 Classifier: Programming Language :: Python
 Classifier: Programming Language :: Python :: 3
 Classifier: Programming Language :: Python :: 3 :: Only
-Classifier: Programming Language :: Python :: 3.9
 Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
 Classifier: Programming Language :: Python :: 3.13
 Classifier: Topic :: Internet
 Classifier: Topic :: Software Development :: Libraries :: Python Modules
-Requires-Python: >=3.9
+Requires-Python: >=3.10
 Requires-Dist: asyncpg>=0.30.0
+Requires-Dist: datasets>=4.0.0
+Requires-Dist: devtools>=0.12.2
+Requires-Dist: duckdb>=1.3.2
 Requires-Dist: fastapi>=0.115.4
-Requires-Dist: logfire[asyncpg,fastapi]>=2.3
-Requires-Dist: pydantic-ai-slim[groq,openai,vertexai]==0.0.12
+Requires-Dist: gradio>=5.9.0
+Requires-Dist: logfire[asyncpg,fastapi,httpx,sqlite3]>=3.14.1
+Requires-Dist: mcp[cli]>=1.4.1
+Requires-Dist: modal>=1.0.4
+Requires-Dist: pandas>=2.2.3
+Requires-Dist: pydantic-ai-slim[ag-ui,anthropic,groq,openai,vertexai]==1.39.1
+Requires-Dist: pydantic-evals==1.39.1
 Requires-Dist: python-multipart>=0.0.17
 Requires-Dist: rich>=13.9.2
 Requires-Dist: uvicorn>=0.32.0
 Description-Content-Type: text/markdown
 
-# PydanticAI Examples
+# Pydantic AI Examples
 
 [![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)
@@ -41,6 +49,6 @@ Description-Content-Type: text/markdown
 [![versions](https://img.shields.io/pypi/pyversions/pydantic-ai.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)
 
-Examples of how to use PydanticAI and what it can do.
+Examples of how to use Pydantic AI and what it can do.
 
 For full documentation of these examples and how to run them, see [ai.pydantic.dev/examples/](https://ai.pydantic.dev/examples/).

{pydantic_ai_examples-0.0.12 → pydantic_ai_examples-1.39.1}/README.md

@@ -1,4 +1,4 @@
-# PydanticAI Examples
+# Pydantic AI Examples
 
 [![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)
@@ -6,6 +6,6 @@
 [![versions](https://img.shields.io/pypi/pyversions/pydantic-ai.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)
 
-Examples of how to use PydanticAI and what it can do.
+Examples of how to use Pydantic AI and what it can do.
 
 For full documentation of these examples and how to run them, see [ai.pydantic.dev/examples/](https://ai.pydantic.dev/examples/).

pydantic_ai_examples-1.39.1/pydantic_ai_examples/ag_ui/__init__.py

@@ -0,0 +1,41 @@
+"""Example usage of the AG-UI adapter for Pydantic AI.
+
+This provides a FastAPI application that demonstrates how to use the
+Pydantic AI agent with the AG-UI protocol. It includes examples for
+each of the AG-UI dojo features:
+- Agentic Chat
+- Human in the Loop
+- Agentic Generative UI
+- Tool Based Generative UI
+- Shared State
+- Predictive State Updates
+"""
+
+from __future__ import annotations
+
+from fastapi import FastAPI
+
+from .api import (
+    agentic_chat_app,
+    agentic_generative_ui_app,
+    human_in_the_loop_app,
+    predictive_state_updates_app,
+    shared_state_app,
+    tool_based_generative_ui_app,
+)
+
+app = FastAPI(title='Pydantic AI AG-UI server')
+app.mount('/agentic_chat', agentic_chat_app, 'Agentic Chat')
+app.mount('/agentic_generative_ui', agentic_generative_ui_app, 'Agentic Generative UI')
+app.mount('/human_in_the_loop', human_in_the_loop_app, 'Human in the Loop')
+app.mount(
+    '/predictive_state_updates',
+    predictive_state_updates_app,
+    'Predictive State Updates',
+)
+app.mount('/shared_state', shared_state_app, 'Shared State')
+app.mount(
+    '/tool_based_generative_ui',
+    tool_based_generative_ui_app,
+    'Tool Based Generative UI',
+)

pydantic_ai_examples-1.39.1/pydantic_ai_examples/ag_ui/__main__.py

@@ -0,0 +1,9 @@
+"""Very simply CLI to run the AG-UI example.
+
+See https://ai.pydantic.dev/examples/ag-ui/ for more information.
+"""
+
+if __name__ == '__main__':
+    import uvicorn
+
+    uvicorn.run('pydantic_ai_examples.ag_ui:app', port=9000)

pydantic_ai_examples-1.39.1/pydantic_ai_examples/ag_ui/api/__init__.py

@@ -0,0 +1,19 @@
+"""Example API for a AG-UI compatible Pydantic AI Agent UI."""
+
+from __future__ import annotations
+
+from .agentic_chat import app as agentic_chat_app
+from .agentic_generative_ui import app as agentic_generative_ui_app
+from .human_in_the_loop import app as human_in_the_loop_app
+from .predictive_state_updates import app as predictive_state_updates_app
+from .shared_state import app as shared_state_app
+from .tool_based_generative_ui import app as tool_based_generative_ui_app
+
+__all__ = [
+    'agentic_chat_app',
+    'agentic_generative_ui_app',
+    'human_in_the_loop_app',
+    'predictive_state_updates_app',
+    'shared_state_app',
+    'tool_based_generative_ui_app',
+]

pydantic_ai_examples-1.39.1/pydantic_ai_examples/ag_ui/api/agentic_chat.py

@@ -0,0 +1,28 @@
+"""Agentic Chat feature."""
+
+from __future__ import annotations
+
+from datetime import datetime
+from zoneinfo import ZoneInfo
+
+from pydantic_ai import Agent
+from pydantic_ai.ui.ag_ui.app import AGUIApp
+
+agent = Agent('openai:gpt-5-mini')
+
+
+@agent.tool_plain
+async def current_time(timezone: str = 'UTC') -> str:
+    """Get the current time in ISO format.
+
+    Args:
+        timezone: The timezone to use.
+
+    Returns:
+        The current time in ISO format string.
+    """
+    tz: ZoneInfo = ZoneInfo(timezone)
+    return datetime.now(tz=tz).isoformat()
+
+
+app = AGUIApp(agent)

pydantic_ai_examples-1.39.1/pydantic_ai_examples/ag_ui/api/agentic_generative_ui.py

@@ -0,0 +1,120 @@
+"""Agentic Generative UI feature."""
+
+from __future__ import annotations
+
+from textwrap import dedent
+from typing import Any, Literal
+
+from pydantic import BaseModel, Field
+
+from ag_ui.core import EventType, StateDeltaEvent, StateSnapshotEvent
+from pydantic_ai import Agent
+from pydantic_ai.ui.ag_ui.app import AGUIApp
+
+StepStatus = Literal['pending', 'completed']
+
+
+class Step(BaseModel):
+    """Represents a step in a plan."""
+
+    description: str = Field(description='The description of the step')
+    status: StepStatus = Field(
+        default='pending',
+        description='The status of the step (e.g., pending, completed)',
+    )
+
+
+class Plan(BaseModel):
+    """Represents a plan with multiple steps."""
+
+    steps: list[Step] = Field(default_factory=list, description='The steps in the plan')
+
+
+class JSONPatchOp(BaseModel):
+    """A class representing a JSON Patch operation (RFC 6902)."""
+
+    op: Literal['add', 'remove', 'replace', 'move', 'copy', 'test'] = Field(
+        description='The operation to perform: add, remove, replace, move, copy, or test',
+    )
+    path: str = Field(description='JSON Pointer (RFC 6901) to the target location')
+    value: Any = Field(
+        default=None,
+        description='The value to apply (for add, replace operations)',
+    )
+    from_: str | None = Field(
+        default=None,
+        alias='from',
+        description='Source path (for move, copy operations)',
+    )
+
+
+agent = Agent(
+    'openai:gpt-5-mini',
+    instructions=dedent(
+        """
+        When planning use tools only, without any other messages.
+        IMPORTANT:
+        - Use the `create_plan` tool to set the initial state of the steps
+        - Use the `update_plan_step` tool to update the status of each step
+        - Do NOT repeat the plan or summarise it in a message
+        - Do NOT confirm the creation or updates in a message
+        - Do NOT ask the user for additional information or next steps
+
+        Only one plan can be active at a time, so do not call the `create_plan` tool
+        again until all the steps in current plan are completed.
+        """
+    ),
+)
+
+
+@agent.tool_plain
+async def create_plan(steps: list[str]) -> StateSnapshotEvent:
+    """Create a plan with multiple steps.
+
+    Args:
+        steps: List of step descriptions to create the plan.
+
+    Returns:
+        StateSnapshotEvent containing the initial state of the steps.
+    """
+    plan: Plan = Plan(
+        steps=[Step(description=step) for step in steps],
+    )
+    return StateSnapshotEvent(
+        type=EventType.STATE_SNAPSHOT,
+        snapshot=plan.model_dump(),
+    )
+
+
+@agent.tool_plain
+async def update_plan_step(
+    index: int, description: str | None = None, status: StepStatus | None = None
+) -> StateDeltaEvent:
+    """Update the plan with new steps or changes.
+
+    Args:
+        index: The index of the step to update.
+        description: The new description for the step.
+        status: The new status for the step.
+
+    Returns:
+        StateDeltaEvent containing the changes made to the plan.
+    """
+    changes: list[JSONPatchOp] = []
+    if description is not None:
+        changes.append(
+            JSONPatchOp(
+                op='replace', path=f'/steps/{index}/description', value=description
+            )
+        )
+    if status is not None:
+        changes.append(
+            JSONPatchOp(op='replace', path=f'/steps/{index}/status', value=status)
+        )
+    return StateDeltaEvent(
+        type=EventType.STATE_DELTA,
+        delta=changes,
+    )
+
+
+app = AGUIApp(agent)

pydantic_ai_examples-1.39.1/pydantic_ai_examples/ag_ui/api/human_in_the_loop.py

@@ -0,0 +1,27 @@
+"""Human in the Loop Feature.
+
+No special handling is required for this feature.
+"""
+
+from __future__ import annotations
+
+from textwrap import dedent
+
+from pydantic_ai import Agent
+from pydantic_ai.ui.ag_ui.app import AGUIApp
+
+agent = Agent(
+    'openai:gpt-5-mini',
+    instructions=dedent(
+        """
+        When planning tasks use tools only, without any other messages.
+        IMPORTANT:
+        - Use the `generate_task_steps` tool to display the suggested steps to the user
+        - Never repeat the plan, or send a message detailing steps
+        - If accepted, confirm the creation of the plan and the number of selected (enabled) steps only
+        - If not accepted, ask the user for more information, DO NOT use the `generate_task_steps` tool again
+        """
+    ),
+)
+
+app = AGUIApp(agent)

pydantic_ai_examples-1.39.1/pydantic_ai_examples/ag_ui/api/predictive_state_updates.py

@@ -0,0 +1,78 @@
+"""Predictive State feature."""
+
+from __future__ import annotations
+
+from textwrap import dedent
+
+from pydantic import BaseModel
+
+from ag_ui.core import CustomEvent, EventType
+from pydantic_ai import Agent, RunContext
+from pydantic_ai.ui import StateDeps
+from pydantic_ai.ui.ag_ui.app import AGUIApp
+
+
+class DocumentState(BaseModel):
+    """State for the document being written."""
+
+    document: str = ''
+
+
+agent = Agent('openai:gpt-5-mini', deps_type=StateDeps[DocumentState])
+
+
+# Tools which return AG-UI events will be sent to the client as part of the
+# event stream, single events and iterables of events are supported.
+@agent.tool_plain
+async def document_predict_state() -> list[CustomEvent]:
+    """Enable document state prediction.
+
+    Returns:
+        CustomEvent containing the event to enable state prediction.
+    """
+    return [
+        CustomEvent(
+            type=EventType.CUSTOM,
+            name='PredictState',
+            value=[
+                {
+                    'state_key': 'document',
+                    'tool': 'write_document',
+                    'tool_argument': 'document',
+                },
+            ],
+        ),
+    ]
+
+
+@agent.instructions()
+async def story_instructions(ctx: RunContext[StateDeps[DocumentState]]) -> str:
+    """Provide instructions for writing document if present.
+
+    Args:
+        ctx: The run context containing document state information.
+
+    Returns:
+        Instructions string for the document writing agent.
+    """
+    return dedent(
+        f"""You are a helpful assistant for writing documents.
+
+        Before you start writing, you MUST call the `document_predict_state`
+        tool to enable state prediction.
+
+        To present the document to the user for review, you MUST use the
+        `write_document` tool.
+
+        When you have written the document, DO NOT repeat it as a message.
+        If accepted briefly summarize the changes you made, 2 sentences
+        max, otherwise ask the user to clarify what they want to change.
+
+        This is the current document:
+
+        {ctx.deps.state.document}
+        """
+    )
+
+
+app = AGUIApp(agent, deps=StateDeps(DocumentState()))

pydantic_ai_examples-1.39.1/pydantic_ai_examples/ag_ui/api/shared_state.py

@@ -0,0 +1,139 @@
+"""Shared State feature."""
+
+from __future__ import annotations
+
+from enum import Enum
+from textwrap import dedent
+
+from pydantic import BaseModel, Field
+
+from ag_ui.core import EventType, StateSnapshotEvent
+from pydantic_ai import Agent, RunContext
+from pydantic_ai.ui import StateDeps
+from pydantic_ai.ui.ag_ui.app import AGUIApp
+
+
+class SkillLevel(str, Enum):
+    """The level of skill required for the recipe."""
+
+    BEGINNER = 'Beginner'
+    INTERMEDIATE = 'Intermediate'
+    ADVANCED = 'Advanced'
+
+
+class SpecialPreferences(str, Enum):
+    """Special preferences for the recipe."""
+
+    HIGH_PROTEIN = 'High Protein'
+    LOW_CARB = 'Low Carb'
+    SPICY = 'Spicy'
+    BUDGET_FRIENDLY = 'Budget-Friendly'
+    ONE_POT_MEAL = 'One-Pot Meal'
+    VEGETARIAN = 'Vegetarian'
+    VEGAN = 'Vegan'
+
+
+class CookingTime(str, Enum):
+    """The cooking time of the recipe."""
+
+    FIVE_MIN = '5 min'
+    FIFTEEN_MIN = '15 min'
+    THIRTY_MIN = '30 min'
+    FORTY_FIVE_MIN = '45 min'
+    SIXTY_PLUS_MIN = '60+ min'
+
+
+class Ingredient(BaseModel):
+    """A class representing an ingredient in a recipe."""
+
+    icon: str = Field(
+        default='ingredient',
+        description="The icon emoji (not emoji code like '\x1f35e', but the actual emoji like 🥕) of the ingredient",
+    )
+    name: str
+    amount: str
+
+
+class Recipe(BaseModel):
+    """A class representing a recipe."""
+
+    skill_level: SkillLevel = Field(
+        default=SkillLevel.BEGINNER,
+        description='The skill level required for the recipe',
+    )
+    special_preferences: list[SpecialPreferences] = Field(
+        default_factory=list,
+        description='Any special preferences for the recipe',
+    )
+    cooking_time: CookingTime = Field(
+        default=CookingTime.FIVE_MIN, description='The cooking time of the recipe'
+    )
+    ingredients: list[Ingredient] = Field(
+        default_factory=list,
+        description='Ingredients for the recipe',
+    )
+    instructions: list[str] = Field(
+        default_factory=list, description='Instructions for the recipe'
+    )
+
+
+class RecipeSnapshot(BaseModel):
+    """A class representing the state of the recipe."""
+
+    recipe: Recipe = Field(
+        default_factory=Recipe, description='The current state of the recipe'
+    )
+
+
+agent = Agent('openai:gpt-5-mini', deps_type=StateDeps[RecipeSnapshot])
+
+
+@agent.tool_plain
+async def display_recipe(recipe: Recipe) -> StateSnapshotEvent:
+    """Display the recipe to the user.
+
+    Args:
+        recipe: The recipe to display.
+
+    Returns:
+        StateSnapshotEvent containing the recipe snapshot.
+    """
+    return StateSnapshotEvent(
+        type=EventType.STATE_SNAPSHOT,
+        snapshot={'recipe': recipe},
+    )
+
+
+@agent.instructions
+async def recipe_instructions(ctx: RunContext[StateDeps[RecipeSnapshot]]) -> str:
+    """Instructions for the recipe generation agent.
+
+    Args:
+        ctx: The run context containing recipe state information.
+
+    Returns:
+        Instructions string for the recipe generation agent.
+    """
+    return dedent(
+        f"""
+        You are a helpful assistant for creating recipes.
+
+        IMPORTANT:
+        - Create a complete recipe using the existing ingredients
+        - Append new ingredients to the existing ones
+        - Use the `display_recipe` tool to present the recipe to the user
+        - Do NOT repeat the recipe in the message, use the tool instead
+        - Do NOT run the `display_recipe` tool multiple times in a row
+
+        Once you have created the updated recipe and displayed it to the user,
+        summarise the changes in one sentence, don't describe the recipe in
+        detail or send it as a message to the user.
+
+        The current state of the recipe is:
+
+        {ctx.deps.state.recipe.model_dump_json(indent=2)}
+        """,
+    )
+
+
+app = AGUIApp(agent, deps=StateDeps(RecipeSnapshot()))

pydantic_ai_examples-1.39.1/pydantic_ai_examples/ag_ui/api/tool_based_generative_ui.py

@@ -0,0 +1,12 @@
+"""Tool Based Generative UI feature.
+
+No special handling is required for this feature.
+"""
+
+from __future__ import annotations
+
+from pydantic_ai import Agent
+from pydantic_ai.ui.ag_ui.app import AGUIApp
+
+agent = Agent('openai:gpt-5-mini')
+app = AGUIApp(agent)