Make WalkJsonSchema private (#1490)

Author: dmontagu

pydantic_ai_slim/pydantic_ai/models/json_schema.py renamed to pydantic_ai_slim/pydantic_ai/models/_json_schema.py

@@ -11,7 +11,11 @@
 
 @dataclass(init=False)
 class WalkJsonSchema(ABC):
-    """Walks a JSON schema, applying transformations to it at each level."""
+    """Walks a JSON schema, applying transformations to it at each level.
+
+    Note: We may eventually want to rework tools to build the JSON schema from the type directly, using a subclass of
+    pydantic.json_schema.GenerateJsonSchema, rather than making use of this machinery.
+    """
 
     def __init__(
         self, schema: JsonSchema, *, prefer_inlined_defs: bool = False, simplify_nullable_unions: bool = False

pydantic_ai_slim/pydantic_ai/models/gemini.py

@@ -44,7 +44,7 @@
     check_allow_model_requests,
     get_user_agent,
 )
-from .json_schema import JsonSchema, WalkJsonSchema
+from ._json_schema import JsonSchema, WalkJsonSchema
 
 LatestGeminiModelNames = Literal[
     'gemini-1.5-flash',

pydantic_ai_slim/pydantic_ai/models/openai.py

@@ -42,7 +42,7 @@
     check_allow_model_requests,
     get_user_agent,
 )
-from .json_schema import JsonSchema, WalkJsonSchema
+from ._json_schema import JsonSchema, WalkJsonSchema
 
 try:
     from openai import NOT_GIVEN, APIStatusError, AsyncOpenAI, AsyncStream, NotGiven