move OutputObjectDefinition from private _output.py to public output.py (#2991)

Author: moritzwilksch

pydantic_ai_slim/pydantic_ai/_output.py

@@ -19,6 +19,7 @@
     NativeOutput,
     OutputDataT,
     OutputMode,
+    OutputObjectDefinition,
     OutputSpec,
     OutputTypeOrFunction,
     PromptedOutput,
@@ -581,14 +582,6 @@ def mode(self) -> OutputMode:
         return 'tool_or_text'
 
 
-@dataclass
-class OutputObjectDefinition:
-    json_schema: ObjectJsonSchema
-    name: str | None = None
-    description: str | None = None
-    strict: bool | None = None
-
-
 @dataclass(init=False)
 class BaseOutputProcessor(ABC, Generic[OutputDataT]):
     @abstractmethod

pydantic_ai_slim/pydantic_ai/output.py

@@ -11,7 +11,7 @@
 
 from . import _utils
 from .messages import ToolCallPart
-from .tools import DeferredToolRequests, RunContext, ToolDefinition
+from .tools import DeferredToolRequests, ObjectJsonSchema, RunContext, ToolDefinition
 
 __all__ = (
     # classes
@@ -20,6 +20,7 @@
     'PromptedOutput',
     'TextOutput',
     'StructuredDict',
+    'OutputObjectDefinition',
     # types
     'OutputDataT',
     'OutputMode',
@@ -242,6 +243,16 @@ def __init__(
         self.template = template
 
 
+@dataclass
+class OutputObjectDefinition:
+    """Definition of an output object used for structured output generation."""
+
+    json_schema: ObjectJsonSchema
+    name: str | None = None
+    description: str | None = None
+    strict: bool | None = None
+
+
 @dataclass
 class TextOutput(Generic[OutputDataT]):
     """Marker class to use text output for an output function taking a string argument.