Merge pull request #140 from Chainlit/willy/fix-lg-outputs

fix: logs all outputs

Author: willydouhard

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):

literalai/version.py

@@ -1 +1 @@
-__version__ = "0.0.627"
+__version__ = "0.0.628"

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",