diff --git a/src/openlayer/lib/tracing/tracer.py b/src/openlayer/lib/tracing/tracer.py index 1824df97..83af81fb 100644 --- a/src/openlayer/lib/tracing/tracer.py +++ b/src/openlayer/lib/tracing/tracer.py @@ -474,11 +474,15 @@ def _handle_trace_completion( ) client = _get_client() if client: - client.inference_pipelines.data.stream( + response = client.inference_pipelines.data.stream( inference_pipeline_id=inference_pipeline_id, rows=[trace_data], config=config, ) + print( + "Successfully streamed data to Openlayer. Response:", + response.to_json(), + ) except Exception as err: # pylint: disable=broad-except logger.error(traceback.format_exc()) logger.error(