fix/chat history case (#76)

Author: PierpaoloSorbellini

nebuly/__init__.py

@@ -2,4 +2,4 @@
 from .init import init
 
 __all__ = ["init", "new_interaction"]
-__version__ = "0.3.37"
+__version__ = "0.3.38"

nebuly/providers/langchain.py

@@ -153,6 +153,10 @@ def _parse_langchain_data(  # pylint: disable=too-many-return-statements
             data["messages"][-1], (HumanMessage, AIMessage, ToolMessage)
         ):
             return str(data["messages"][-1].content)
+        if "chat_history" in data:
+            messages: list[HumanMessage | AIMessage] = data["chat_history"]
+            human_messages = [m for m in messages if m.type == "human"]
+            return str(human_messages[-1].content)
         return "\n".join([f"{key}: {value}" for key, value in data.items()])
     if isinstance(data, list):
         data = data[-1]

nebuly/requests.py

@@ -90,7 +90,6 @@ def post_message(
     if watched.api_key:
         api_key = watched.api_key
     post_json_data(url, message, api_key)
-    
 
 
 def post_json_data(url: str, json_data: str, api_key: str) -> Any:
@@ -145,10 +144,7 @@ def post_json_data(url: str, json_data: str, api_key: str) -> Any:
     return None
 
 
-def _crop_spans(
-    message: str,
-    max_size: float = 0.5 
-) -> str:
+def _crop_spans(message: str, max_size: float = 0.5) -> str:
     message_size = len(str(message).encode("utf-8")) / 1_000_000
     if message_size < max_size:
         return message
@@ -157,12 +153,10 @@ def _crop_spans(
         _remove_long_text(message_dict["body"]["spans"])
         return json.dumps(message_dict)
     except (json.JSONDecodeError, KeyError, AttributeError) as e:
-        logger.warning(
-            "Failed to crop spans. Error: %s",
-            str(e)
-        )
+        logger.warning("Failed to crop spans. Error: %s", str(e))
         return message
-    
+
+
 def _remove_long_text(
     item: dict[str, Any] | list[Any] | Any, max_words: int = 20
 ) -> None:
@@ -177,4 +171,4 @@ def _remove_long_text(
                 item[k] = item[k][:50] + "..."
     elif isinstance(item, list):
         for v in item:
-            _remove_long_text(v)
+            _remove_long_text(v)

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "nebuly"
-version = "0.3.37"
+version = "0.3.38"
 description = "The SDK for instrumenting applications for tracking AI costs."
 authors = ["Nebuly"]
 readme = "README.md"

tests/test_crop_spans.py

@@ -28,7 +28,7 @@ def test_crop_spans__is_working() -> None:
         _ = InteractionWatch(**message["body"])
     except (TypeError, ValueError, KeyError) as e:
         pytest.fail(f"Creating InteractionWatch raised an error: {e}")
-        
+
 
 def test_crop_spans__is_doing_nothing_if_size_is_small() -> None:
     with open("./tests/huge_message.json", encoding="utf-8") as f: