fix: appease the type checker

Author: mrkaye97

pydantic_ai_slim/pydantic_ai/durable_exec/hatchet/_agent.py

@@ -11,7 +11,8 @@
 from typing_extensions import Never
 
 from pydantic_ai import _utils, messages as _messages, models, usage as _usage
-from pydantic_ai.agent import AbstractAgent, AgentRun, AgentRunResult, EventStreamHandler, RunOutputDataT, WrapperAgent
+from pydantic_ai.agent import AbstractAgent, AgentRun, AgentRunResult, EventStreamHandler, WrapperAgent
+from pydantic_ai.agent.abstract import Instructions, RunOutputDataT
 from pydantic_ai.exceptions import UserError
 from pydantic_ai.messages import AgentStreamEvent
 from pydantic_ai.models import Model
@@ -652,6 +653,7 @@ def override(
         model: models.Model | models.KnownModelName | str | _utils.Unset = _utils.UNSET,
         toolsets: Sequence[AbstractToolset[AgentDepsT]] | _utils.Unset = _utils.UNSET,
         tools: Sequence[Tool[AgentDepsT] | ToolFuncEither[AgentDepsT, ...]] | _utils.Unset = _utils.UNSET,
+        instructions: Instructions[AgentDepsT] | _utils.Unset = _utils.UNSET,
     ) -> Iterator[None]:
         """Context manager to temporarily override agent dependencies, model, toolsets, or tools.
 
@@ -663,6 +665,7 @@ def override(
             model: The model to use instead of the model passed to the agent run.
             toolsets: The toolsets to use instead of the toolsets passed to the agent constructor and agent run.
             tools: The tools to use instead of the tools registered with the agent.
+            instructions: The instructions to use instead of the instructions registered with the agent.
         """
         with super().override(deps=deps, model=model, toolsets=toolsets, tools=tools):
             yield

pydantic_ai_slim/pydantic_ai/durable_exec/hatchet/_function_toolset.py

@@ -66,7 +66,7 @@ async def tool_task(
                     _ctx: Context,
                 ) -> ToolOutput:
                     run_context = self.run_context_type.deserialize_run_context(
-                        input.serialized_run_context, deps=input.deps, hatchet_context=_ctx
+                        input.serialized_run_context, deps=input.deps
                     )
                     tool = (await wrapped.get_tools(run_context))[current_tool_name]
 

pydantic_ai_slim/pydantic_ai/durable_exec/hatchet/_mcp_server.py

@@ -87,9 +87,7 @@ async def wrapped_get_tools_task(
             input: GetToolsInput[AgentDepsT],
             _ctx: Context,
         ) -> dict[str, ToolDefinition]:
-            run_context = self.run_context_type.deserialize_run_context(
-                input.serialized_run_context, deps=input.deps, hatchet_context=_ctx
-            )
+            run_context = self.run_context_type.deserialize_run_context(input.serialized_run_context, deps=input.deps)
 
             # ToolsetTool is not serializable as it holds a SchemaValidator (which is also the same for every MCP tool so unnecessary to pass along the wire every time),
             # so we just return the ToolDefinitions and wrap them in ToolsetTool outside of the activity.
@@ -120,9 +118,7 @@ async def wrapped_call_tool_task(
             input: CallToolInput[AgentDepsT],
             _ctx: Context,
         ) -> CallToolOutput[AgentDepsT]:
-            run_context = self.run_context_type.deserialize_run_context(
-                input.serialized_run_context, deps=input.deps, hatchet_context=_ctx
-            )
+            run_context = self.run_context_type.deserialize_run_context(input.serialized_run_context, deps=input.deps)
             tool = self.tool_for_tool_def(input.tool_def)
 
             result = await super(HatchetMCPServer, self).call_tool(input.name, input.tool_args, run_context, tool)

pydantic_ai_slim/pydantic_ai/durable_exec/hatchet/_model.py

@@ -118,7 +118,8 @@ async def wrapped_request_stream_task(
             assert self.event_stream_handler
 
             run_context = self.run_context_type.deserialize_run_context(
-                input.serialized_run_context, deps=input.serialized_run_context, hatchet_context=ctx
+                input.serialized_run_context,
+                deps=input.serialized_run_context,
             )
 
             async with self.wrapped.request_stream(