Merge pull request #141 from Chainlit/willy/fix-lc-structured-output

fix: lc structured output json ser

Author: willydouhard

literalai/callback/langchain_callback.py

@@ -2,6 +2,8 @@
 from importlib.metadata import version
 from typing import TYPE_CHECKING, Any, Dict, List, Optional, TypedDict, Union, cast
 
+from pydantic import BaseModel
+
 from literalai.helper import ensure_values_serializable
 from literalai.observability.generation import (
     ChatGeneration,
@@ -148,6 +150,8 @@ def process_content(self, content: Any, root=True):
                     return [self._convert_message(m) for m in content]
             elif self._is_message(content):
                 return self._convert_message(content)
+            elif isinstance(content, BaseModel):
+                return content.model_dump()
             elif isinstance(content, dict):
                 processed_dict = {}
                 for key, value in content.items():
@@ -186,7 +190,9 @@ def _build_llm_settings(
             }
 
             # make sure there is no api key specification
-            settings = {k: v for k, v in merged.items() if not k.endswith("_api_key")}
+            settings = self.process_content(
+                {k: v for k, v in merged.items() if not k.endswith("_api_key")}
+            )
             model_keys = ["azure_deployment", "deployment_name", "model", "model_name"]
             model = next((settings[k] for k in model_keys if k in settings), None)
             tools = None
@@ -203,6 +209,7 @@ def _build_llm_settings(
         "RunnableParallel",
         "RunnableAssign",
         "RunnableLambda",
+        "structured_outputs_parser",
         "<lambda>",
     ]
     DEFAULT_TO_KEEP = ["retriever", "llm", "agent", "chain", "tool"]

literalai/event_processor.py

@@ -32,6 +32,7 @@ class EventProcessor:
     batch_timeout: float = 5.0
 
     def __init__(self, api: "LiteralAPI", batch_size: int = 1, disabled: bool = False):
+        self.stop_event = threading.Event()
         self.batch_size = batch_size
         self.api = api
         self.event_queue = queue.Queue()
@@ -44,7 +45,6 @@ def __init__(self, api: "LiteralAPI", batch_size: int = 1, disabled: bool = Fals
         )
         if not self.disabled:
             self.processing_thread.start()
-        self.stop_event = threading.Event()
 
     def add_event(self, event: "StepDict"):
         with self.counter_lock:

literalai/version.py

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

setup.py

@@ -2,7 +2,7 @@
 
 setup(
     name="literalai",
-    version="0.0.628",  # update version in literalai/version.py
+    version="0.0.629",  # 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",