From d37e322862b696afd7a535d0ada67431d032a510 Mon Sep 17 00:00:00 2001 From: Douwe Maan Date: Wed, 6 Aug 2025 18:43:34 +0000 Subject: [PATCH] Make InlineDefsJsonSchemaTransformer public --- docs/models/openai.md | 2 +- pydantic_ai_slim/pydantic_ai/profiles/__init__.py | 10 +++++++++- pydantic_ai_slim/pydantic_ai/profiles/amazon.py | 3 +-- pydantic_ai_slim/pydantic_ai/profiles/meta.py | 3 +-- pydantic_ai_slim/pydantic_ai/profiles/qwen.py | 3 +-- 5 files changed, 13 insertions(+), 8 deletions(-) diff --git a/docs/models/openai.md b/docs/models/openai.md index 5dd131068d..19fe8711eb 100644 --- a/docs/models/openai.md +++ b/docs/models/openai.md @@ -177,7 +177,7 @@ If the model you're using is not working correctly out of the box, you can tweak ```py from pydantic_ai import Agent from pydantic_ai.models.openai import OpenAIModel -from pydantic_ai.profiles._json_schema import InlineDefsJsonSchemaTransformer +from pydantic_ai.profiles import InlineDefsJsonSchemaTransformer from pydantic_ai.profiles.openai import OpenAIModelProfile from pydantic_ai.providers.openai import OpenAIProvider diff --git a/pydantic_ai_slim/pydantic_ai/profiles/__init__.py b/pydantic_ai_slim/pydantic_ai/profiles/__init__.py index 68c98fcde4..088169441a 100644 --- a/pydantic_ai_slim/pydantic_ai/profiles/__init__.py +++ b/pydantic_ai_slim/pydantic_ai/profiles/__init__.py @@ -7,7 +7,15 @@ from typing_extensions import Self from ..output import StructuredOutputMode -from ._json_schema import JsonSchemaTransformer +from ._json_schema import InlineDefsJsonSchemaTransformer, JsonSchemaTransformer + +__all__ = [ + 'ModelProfile', + 'ModelProfileSpec', + 'DEFAULT_PROFILE', + 'InlineDefsJsonSchemaTransformer', + 'JsonSchemaTransformer', +] @dataclass diff --git a/pydantic_ai_slim/pydantic_ai/profiles/amazon.py b/pydantic_ai_slim/pydantic_ai/profiles/amazon.py index 8cac0f11d9..88fbc5dc18 100644 --- a/pydantic_ai_slim/pydantic_ai/profiles/amazon.py +++ b/pydantic_ai_slim/pydantic_ai/profiles/amazon.py @@ -1,7 +1,6 @@ from __future__ import annotations as _annotations -from . import ModelProfile -from ._json_schema import InlineDefsJsonSchemaTransformer +from . import InlineDefsJsonSchemaTransformer, ModelProfile def amazon_model_profile(model_name: str) -> ModelProfile | None: diff --git a/pydantic_ai_slim/pydantic_ai/profiles/meta.py b/pydantic_ai_slim/pydantic_ai/profiles/meta.py index bfaec611d0..497e6ca683 100644 --- a/pydantic_ai_slim/pydantic_ai/profiles/meta.py +++ b/pydantic_ai_slim/pydantic_ai/profiles/meta.py @@ -1,7 +1,6 @@ from __future__ import annotations as _annotations -from . import ModelProfile -from ._json_schema import InlineDefsJsonSchemaTransformer +from . import InlineDefsJsonSchemaTransformer, ModelProfile def meta_model_profile(model_name: str) -> ModelProfile | None: diff --git a/pydantic_ai_slim/pydantic_ai/profiles/qwen.py b/pydantic_ai_slim/pydantic_ai/profiles/qwen.py index fa3e75827c..5b6f351cb8 100644 --- a/pydantic_ai_slim/pydantic_ai/profiles/qwen.py +++ b/pydantic_ai_slim/pydantic_ai/profiles/qwen.py @@ -1,7 +1,6 @@ from __future__ import annotations as _annotations -from . import ModelProfile -from ._json_schema import InlineDefsJsonSchemaTransformer +from . import InlineDefsJsonSchemaTransformer, ModelProfile def qwen_model_profile(model_name: str) -> ModelProfile | None: