coverage

Author: DouweM

pydantic_ai_slim/pydantic_ai/ui/vercel_ai/_adapter.py

@@ -70,7 +70,7 @@ class VercelAIAdapter(BaseAdapter[RequestData, UIMessage, BaseChunk, AgentDepsT,
     @classmethod
     async def validate_request(cls, request: Request) -> RequestData:
         """Validate a Vercel AI request."""
-        return request_data_ta.validate_json(await request.body())  # TODO (DouweM): coverage
+        return request_data_ta.validate_json(await request.body())
 
     @property
     def event_stream(self) -> BaseEventStream[RequestData, BaseChunk, AgentDepsT, OutputDataT]:
@@ -79,7 +79,7 @@ def event_stream(self) -> BaseEventStream[RequestData, BaseChunk, AgentDepsT, Ou
     @property
     def response_headers(self) -> Mapping[str, str] | None:
         """Get the response headers for the adapter."""
-        return VERCEL_AI_DSP_HEADERS  # TODO (DouweM): coverage
+        return VERCEL_AI_DSP_HEADERS
 
     @cached_property
     def messages(self) -> list[ModelMessage]:

pydantic_ai_slim/pydantic_ai/ui/vercel_ai/_event_stream.py

@@ -91,7 +91,7 @@ async def after_stream(self) -> AsyncIterator[BaseChunk]:
 
     async def on_error(self, error: Exception) -> AsyncIterator[BaseChunk]:
         """Handle errors during streaming."""
-        yield ErrorChunk(error_text=str(error))  # TODO (DouweM): coverage
+        yield ErrorChunk(error_text=str(error))
 
     async def handle_text_start(self, part: TextPart, follows_text: bool = False) -> AsyncIterator[BaseChunk]:
         """Handle a TextPart at start."""
@@ -136,7 +136,7 @@ async def handle_thinking_end(
 
     def handle_tool_call_start(self, part: ToolCallPart | BuiltinToolCallPart) -> AsyncIterator[BaseChunk]:
         """Handle a ToolCallPart or BuiltinToolCallPart at start."""
-        return self._handle_tool_call_start(part)  # TODO (DouweM): coverage
+        return self._handle_tool_call_start(part)
 
     def handle_builtin_tool_call_start(self, part: BuiltinToolCallPart) -> AsyncIterator[BaseChunk]:
         """Handle a BuiltinToolCallEvent, emitting tool input events."""