github/deer-flow
1
0
Fork 0
mirror of https://github.com/bytedance/deer-flow.git synced 2026-11-07 19:20:34 +00:00
Code Issues Packages Projects Releases Wiki Activity

chore(example): fix the deprecated contextmanager issue (#7974)

Browse Source
This commit is contained in:
Willem Jiang 2026-11-07 21:44:05 +08:00 committed by GitHub
parent 372baf762f
commit f68abbb831
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
examples/deerflow-extension-example/deerflow_extension_example/plugin.py
View File

@ -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: