Skip to content

Commit 0ff4f6f

Browse files
authored
remove unnecessary awaits on telem decorator (#1095)
* add missing awaits * remove extra coroutine
1 parent 9b7cfc3 commit 0ff4f6f

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

guardrails/hub_telemetry/hub_tracing.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -3,6 +3,7 @@
33
Any,
44
Dict,
55
Optional,
6+
AsyncGenerator,
67
)
78

89
from opentelemetry.trace import Span
@@ -224,7 +225,7 @@ def wrapper(*args, **kwargs):
224225
return decorator
225226

226227

227-
async def _run_async_gen(fn, *args, **kwargs):
228+
async def _run_async_gen(fn, *args, **kwargs) -> AsyncGenerator[Any, None]:
228229
gen = fn(*args, **kwargs)
229230
async for item in gen:
230231
yield item
@@ -238,7 +239,7 @@ def async_trace_stream(
238239
):
239240
def decorator(fn):
240241
@wraps(fn)
241-
async def wrapper(*args, **kwargs):
242+
def wrapper(*args, **kwargs):
242243
hub_telemetry = HubTelemetry()
243244
if hub_telemetry._enabled and hub_telemetry._tracer is not None:
244245
with hub_telemetry._tracer.start_span(

0 commit comments

Comments
 (0)