Skip to content
16 changes: 9 additions & 7 deletions src/openlayer/lib/integrations/langchain_callback.py
Original file line number Diff line number Diff line change
Expand Up @@ -163,13 +163,15 @@ def _process_and_upload_trace(self, root_step: steps.Step) -> None:

if tracer._publish:
try:
tracer._client.inference_pipelines.data.stream(
inference_pipeline_id=utils.get_env_variable(
"OPENLAYER_INFERENCE_PIPELINE_ID"
),
rows=[trace_data],
config=config,
)
client = tracer._get_client()
if client:
client.inference_pipelines.data.stream(
inference_pipeline_id=utils.get_env_variable(
"OPENLAYER_INFERENCE_PIPELINE_ID"
),
rows=[trace_data],
config=config,
)
except Exception as err: # pylint: disable=broad-except
tracer.logger.error("Could not stream data to Openlayer %s", err)

Expand Down
Loading