feedback: None default for the val ctx and require kwargs after data in process functions

Author: NicolasPllr1

pydantic_ai_slim/pydantic_ai/_output.py

@@ -529,8 +529,9 @@ class BaseOutputProcessor(ABC, Generic[OutputDataT]):
     async def process(
         self,
         data: str,
+        *,
         run_context: RunContext[AgentDepsT],
-        validation_context: Any | None,
+        validation_context: Any | None = None,
         allow_partial: bool = False,
         wrap_validation_errors: bool = True,
     ) -> OutputDataT:
@@ -554,17 +555,18 @@ def __init__(self, wrapped: BaseObjectOutputProcessor[OutputDataT]):
     async def process(
         self,
         data: str,
+        *,
         run_context: RunContext[AgentDepsT],
-        validation_context: Any | None,
+        validation_context: Any | None = None,
         allow_partial: bool = False,
         wrap_validation_errors: bool = True,
     ) -> OutputDataT:
         text = _utils.strip_markdown_fences(data)
 
         return await self.wrapped.process(
             text,
-            run_context,
-            validation_context,
+            run_context=run_context,
+            validation_context=validation_context,
             allow_partial=allow_partial,
             wrap_validation_errors=wrap_validation_errors,
         )
@@ -644,8 +646,9 @@ def __init__(
     async def process(
         self,
         data: str | dict[str, Any] | None,
+        *,
         run_context: RunContext[AgentDepsT],
-        validation_context: Any | None,
+        validation_context: Any | None = None,
         allow_partial: bool = False,
         wrap_validation_errors: bool = True,
     ) -> OutputDataT:
@@ -809,15 +812,16 @@ def __init__(
     async def process(
         self,
         data: str,
+        *,
         run_context: RunContext[AgentDepsT],
         validation_context: Any | None = None,
         allow_partial: bool = False,
         wrap_validation_errors: bool = True,
     ) -> OutputDataT:
         union_object = await self._union_processor.process(
             data,
-            run_context,
-            validation_context,
+            run_context=run_context,
+            validation_context=validation_context,
             allow_partial=allow_partial,
             wrap_validation_errors=wrap_validation_errors,
         )
@@ -836,8 +840,8 @@ async def process(
 
         return await processor.process(
             inner_data,
-            run_context,
-            validation_context,
+            run_context=run_context,
+            validation_context=validation_context,
             allow_partial=allow_partial,
             wrap_validation_errors=wrap_validation_errors,
         )
@@ -847,6 +851,7 @@ class TextOutputProcessor(BaseOutputProcessor[OutputDataT]):
     async def process(
         self,
         data: str,
+        *,
         run_context: RunContext[AgentDepsT],
         validation_context: Any | None = None,
         allow_partial: bool = False,
@@ -879,6 +884,7 @@ def __init__(
     async def process(
         self,
         data: str,
+        *,
         run_context: RunContext[AgentDepsT],
         validation_context: Any | None = None,
         allow_partial: bool = False,
@@ -887,7 +893,13 @@ async def process(
         args = {self._str_argument_name: data}
         data = await execute_traced_output_function(self._function_schema, run_context, args, wrap_validation_errors)
 
-        return await super().process(data, run_context, validation_context, allow_partial, wrap_validation_errors)
+        return await super().process(
+            data,
+            run_context=run_context,
+            validation_context=validation_context,
+            allow_partial=allow_partial,
+            wrap_validation_errors=wrap_validation_errors,
+        )
 
 
 @dataclass(init=False)

pydantic_ai_slim/pydantic_ai/result.py

@@ -200,7 +200,11 @@ async def validate_response_output(
             validation_context = build_validation_context(self._tool_manager.validation_ctx, self._run_ctx)
 
             result_data = await text_processor.process(
-                text, self._run_ctx, validation_context, allow_partial=allow_partial, wrap_validation_errors=False
+                text,
+                run_context=self._run_ctx,
+                validation_context=validation_context,
+                allow_partial=allow_partial,
+                wrap_validation_errors=False,
             )
             for validator in self._output_validators:
                 result_data = await validator.validate(