Skip to content

Commit 66ac6cd

Browse files
committed
style(deepagents): apply ruff formatting
1 parent 8aa32ec commit 66ac6cd

2 files changed

Lines changed: 10 additions & 2 deletions

File tree

  • instrumentation-loongsuite/loongsuite-instrumentation-deepagents

instrumentation-loongsuite/loongsuite-instrumentation-deepagents/src/opentelemetry/instrumentation/deepagents/internal/patch.py

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -190,7 +190,13 @@ def _wrap_graph_methods(graph: Any) -> None:
190190
return
191191

192192
originals: dict[str, Callable[..., Any]] = {}
193-
for method_name in ("invoke", "ainvoke", "stream", "astream", "with_config"):
193+
for method_name in (
194+
"invoke",
195+
"ainvoke",
196+
"stream",
197+
"astream",
198+
"with_config",
199+
):
194200
original = getattr(graph, method_name, None)
195201
if original is None:
196202
continue

instrumentation-loongsuite/loongsuite-instrumentation-deepagents/tests/test_instrumentor.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -130,7 +130,9 @@ def test_uninstrument_restores_wrapped_graph_methods():
130130
assert not hasattr(child_graph, REACT_AGENT_METADATA_KEY)
131131
assert not hasattr(child_graph, DEEPAGENTS_METADATA_KEY)
132132
assert graph.invoke("hello") is None
133-
assert child_graph.invoke("hello") == {"metadata": {"customer": "kept"}}
133+
assert child_graph.invoke("hello") == {
134+
"metadata": {"customer": "kept"}
135+
}
134136

135137
graph.seen_stream_config = "not-called"
136138
stream_iter = graph.stream("hello")

0 commit comments

Comments
 (0)