Skip to content

Commit 496f9c4

Browse files
committed
fix: print successful data streaming
1 parent 57e6088 commit 496f9c4

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

src/openlayer/lib/tracing/tracer.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -474,11 +474,15 @@ def _handle_trace_completion(
474474
)
475475
client = _get_client()
476476
if client:
477-
client.inference_pipelines.data.stream(
477+
response = client.inference_pipelines.data.stream(
478478
inference_pipeline_id=inference_pipeline_id,
479479
rows=[trace_data],
480480
config=config,
481481
)
482+
print(
483+
"Successfully streamed data to Openlayer. Response:",
484+
response.to_json(),
485+
)
482486
except Exception as err: # pylint: disable=broad-except
483487
logger.error(traceback.format_exc())
484488
logger.error(

0 commit comments

Comments
 (0)