Lint fixes

Author: abhishekbhakat

pydantic_ai_slim/pydantic_ai/_agent_graph.py

@@ -1119,7 +1119,8 @@ async def _process_message_history(
 
         if is_async_callable(processor):
             if takes_ctx:
-                messages = await processor(run_context, messages)
+                async_processor_with_ctx = cast(_HistoryProcessorAsyncWithCtx[DepsT], processor)
+                messages = await async_processor_with_ctx(run_context, messages)
             else:
                 async_processor = cast(_HistoryProcessorAsync, processor)
                 messages = await async_processor(messages)
@@ -1131,7 +1132,6 @@ async def _process_message_history(
                 sync_processor = cast(_HistoryProcessorSync, processor)
                 messages = await run_in_executor(sync_processor, messages)
 
-    messages = cast(list[_messages.ModelMessage], messages)
     if len(messages) == 0:
         raise exceptions.UserError('Processed history cannot be empty.')
 

pydantic_ai_slim/pydantic_ai/ag_ui.py

@@ -6,6 +6,7 @@
 
 from __future__ import annotations
 
+import asyncio
 import json
 import uuid
 from collections.abc import AsyncIterator, Awaitable, Callable, Iterable, Mapping, Sequence
@@ -24,7 +25,6 @@
 
 from pydantic import BaseModel, ValidationError
 
-from . import _utils
 from ._agent_graph import CallToolsNode, ModelRequestNode
 from .agent import AbstractAgent, AgentRun, AgentRunResult
 from .exceptions import UserError
@@ -378,11 +378,9 @@ async def run_ag_ui(
                 yield encoder.encode(event)
 
         if on_complete is not None and run.result is not None:
-            if _utils.is_async_callable(on_complete):
-                if on_complete is not None:
-                    await on_complete(run.result)
-            else:
-                await _utils.run_in_executor(on_complete, run.result)
+            result = on_complete(run.result)
+            if asyncio.iscoroutine(result):
+                await result
     except _RunError as e:
         yield encoder.encode(
             RunErrorEvent(message=e.message, code=e.code),