From 513a2031ff285a5b001cb55995c0e7085c5b3892 Mon Sep 17 00:00:00 2001 From: Willy Douhard Date: Tue, 29 Oct 2024 11:13:58 +0100 Subject: [PATCH] fix: logs all outputs --- literalai/callback/langchain_callback.py | 6 +----- literalai/version.py | 2 +- setup.py | 2 +- 3 files changed, 3 insertions(+), 7 deletions(-) diff --git a/literalai/callback/langchain_callback.py b/literalai/callback/langchain_callback.py index d60d341..1049343 100644 --- a/literalai/callback/langchain_callback.py +++ b/literalai/callback/langchain_callback.py @@ -512,13 +512,9 @@ def _on_run_update(self, run: Run) -> None: return outputs = run.outputs or {} - output_keys = list(outputs.keys()) - output = outputs - if output_keys: - output = outputs.get(output_keys[0], outputs) if current_step: - current_step.output = self.process_content(output) + current_step.output = self.process_content(outputs) current_step.end() def _on_error(self, error: BaseException, *, run_id: "UUID", **kwargs: Any): diff --git a/literalai/version.py b/literalai/version.py index de83e17..9cdd0a7 100644 --- a/literalai/version.py +++ b/literalai/version.py @@ -1 +1 @@ -__version__ = "0.0.627" +__version__ = "0.0.628" diff --git a/setup.py b/setup.py index baaf75b..3494b8d 100644 --- a/setup.py +++ b/setup.py @@ -2,7 +2,7 @@ setup( name="literalai", - version="0.0.627", # update version in literalai/version.py + version="0.0.628", # update version in literalai/version.py description="An SDK for observability in Python applications", long_description=open("README.md").read(), long_description_content_type="text/markdown",