Skip to content

Commit 7fb7cbe

Browse files
fix: update client retrieval for LangChain callback handler
1 parent bada5eb commit 7fb7cbe

File tree

1 file changed

+9
-7
lines changed

1 file changed

+9
-7
lines changed

src/openlayer/lib/integrations/langchain_callback.py

Lines changed: 9 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -163,13 +163,15 @@ def _process_and_upload_trace(self, root_step: steps.Step) -> None:
163163

164164
if tracer._publish:
165165
try:
166-
tracer._client.inference_pipelines.data.stream(
167-
inference_pipeline_id=utils.get_env_variable(
168-
"OPENLAYER_INFERENCE_PIPELINE_ID"
169-
),
170-
rows=[trace_data],
171-
config=config,
172-
)
166+
client = tracer._get_client()
167+
if client:
168+
client.inference_pipelines.data.stream(
169+
inference_pipeline_id=utils.get_env_variable(
170+
"OPENLAYER_INFERENCE_PIPELINE_ID"
171+
),
172+
rows=[trace_data],
173+
config=config,
174+
)
173175
except Exception as err: # pylint: disable=broad-except
174176
tracer.logger.error("Could not stream data to Openlayer %s", err)
175177

0 commit comments

Comments
 (0)