From f68abbb8314f833929df88cde05d81310da27783 Mon Sep 17 00:00:00 2001 From: Willem Jiang Date: Sat, 7 Nov 2026 21:44:05 +0800 Subject: [PATCH] chore(example): fix the deprecated contextmanager issue (#7974) --- .../deerflow_extension_example/plugin.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/examples/deerflow-extension-example/deerflow_extension_example/plugin.py b/examples/deerflow-extension-example/deerflow_extension_example/plugin.py index c99efc6164..90a9611d0c 100644 --- a/examples/deerflow-extension-example/deerflow_extension_example/plugin.py +++ b/examples/deerflow-extension-example/deerflow_extension_example/plugin.py @@ -3,7 +3,7 @@ from __future__ import annotations from asyncio import CancelledError -from collections.abc import Awaitable, Callable, Iterator, Sequence +from collections.abc import Awaitable, Callable, Generator, Sequence from contextlib import contextmanager from dataclasses import dataclass, field from threading import Lock @@ -95,7 +95,7 @@ def _stats(store: ExtensionData) -> ExampleStats: class ExampleMiddleware(AgentMiddleware): @contextmanager - def _observe(self, request: ToolCallRequest) -> Iterator[None]: + def _observe(self, request: ToolCallRequest) -> Generator[None, None, None]: task_store = task_store_from_runtime(getattr(request, "runtime", None)) task_stats = task_store.get(ExampleStats) if task_store is not None else None if task_stats is None: