Update agents.md

Author: muhammadhamidrazasidtechno

docs/agents.md

@@ -206,17 +206,17 @@ agent = Agent(
 - `ToolsToFinalOutputFunction`: A custom function that processes tool results and decides whether to stop or continue with the LLM.
 
 ```python
-from agents import Agent, Runner, function_tool, FunctionToolResult, RunContextWrapper
+from agents import Agent, Runner, function_tool, FunctionToolResult, RunContextWrapper, TContext
 from agents.agent import ToolsToFinalOutputResult
-from typing import List, Any
+from typing import List
 
 @function_tool
 def get_weather(city: str) -> str:
     """Returns weather info for the specified city."""
     return f"The weather in {city} is sunny"
 
 def custom_tool_handler(
-    context: RunContextWrapper[Any],
+    context: RunContextWrapper[TContext],
     tool_results: List[FunctionToolResult]
 ) -> ToolsToFinalOutputResult:
     """Processes tool results to decide final output."""