Skip to content

Commit e993d69

Browse files
committed
use the batch exporter instead of the simple one
1 parent 0285131 commit e993d69

File tree

2 files changed

+4
-4
lines changed

2 files changed

+4
-4
lines changed

guardrails/run/async_runner.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -281,7 +281,7 @@ async def async_call(
281281
return llm_response
282282

283283
# TODO: Refactor this to use inheritance and overrides
284-
# @async_trace(name="/validation", origin="AsyncRunner.async_validate")
284+
@async_trace(name="/validation", origin="AsyncRunner.async_validate")
285285
async def async_validate(
286286
self,
287287
iteration: Iteration,

guardrails/utils/hub_telemetry_utils.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,7 @@
88
)
99
from opentelemetry.sdk.resources import SERVICE_NAME, Resource
1010
from opentelemetry.sdk.trace import TracerProvider
11-
from opentelemetry.sdk.trace.export import ConsoleSpanExporter, SimpleSpanProcessor
11+
from opentelemetry.sdk.trace.export import ConsoleSpanExporter, BatchSpanProcessor
1212
from opentelemetry.trace.propagation.tracecontext import TraceContextTextMapPropagator
1313
from opentelemetry.trace.propagation import set_span_in_context
1414

@@ -74,9 +74,9 @@ def initialize_tracer(
7474
self._tracer_provider = TracerProvider(resource=self._resource)
7575

7676
if export_locally:
77-
self._processor = SimpleSpanProcessor(ConsoleSpanExporter())
77+
self._processor = BatchSpanProcessor(ConsoleSpanExporter())
7878
else:
79-
self._processor = SimpleSpanProcessor(
79+
self._processor = BatchSpanProcessor(
8080
OTLPSpanExporter(endpoint=self._endpoint)
8181
)
8282

0 commit comments

Comments
 (0)